petos/libLicenseClient
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[impl] Check if the file from download cache really exists before returning it

Browse Source 
[impl] Remove downloaded update file if checksum is invalid
This commit is contained in:
Peter Sykora 2018-03-31 23:43:47 +02:00
parent 4e1edd2a33
commit 2b751fe84e
2 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/CachedDownloader.cpp
View File

@ -66,20 +66,22 @@ boost::filesystem::path CachedDownloader::download(const std::string & url)
auto it = m_cache.find(url);
if (it != m_cache.end())
{
return m_cacheDir / it->second;
}
else
{
auto fileName = boost::filesystem::unique_path();
auto tmpFile = m_cacheDir / fileName;
auto cachedFilename = m_cacheDir / it->second;
if (boost::filesystem::exists(cachedFilename))
{
std::ofstream os(tmpFile.string(), std::ofstream::binary);
m_httpClient.get(url, os);
return cachedFilename;
}
m_cache[url] = fileName.string();
addToCache(m_cacheDir / cacheListFilename, url, fileName.string());
return tmpFile;
}
auto fileName = boost::filesystem::unique_path();
auto tmpFile = m_cacheDir / fileName;
{
std::ofstream os(tmpFile.string(), std::ofstream::binary);
m_httpClient.get(url, os);
}
m_cache[url] = fileName.string();
addToCache(m_cacheDir / cacheListFilename, url, fileName.string());
return tmpFile;
}
void CachedDownloader::clearCache()

1
src/ModuleManager.cpp
View File

@ -27,6 +27,7 @@ void ModuleManager::applyUpdate(Module & module, const ModuleUpdate & update)
auto hash = calcSHA256(updatePath);
if (hash != boost::algorithm::to_upper_copy(update.checksum))
{
fs::remove(updatePath);
throw std::runtime_error("Integrity check of the update has failed");
}