Core: download to tmp file first.

hivivo · Jan 30, 2014 · f17abdf · f17abdf
1 parent e792831
commit f17abdf
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/Core/Downloader/HttpClient.cpp b/Core/Downloader/HttpClient.cpp
@@ -81,6 +81,9 @@ bool HttpClient::downloadAs(string url, string filepath)
     // forward all data to this func
     curl_easy_setopt(m_curl, CURLOPT_WRITEFUNCTION, &HttpClient::writeToFile);
 
+    // try to write a temp file instead of the real file
+    string tmpFile = filepath + ".tmp";
+
     // open the file
     FILE * file = fopen(filepath.c_str(), "wb");
     if (file)
@@ -93,6 +96,9 @@ bool HttpClient::downloadAs(string url, string filepath)
 
         fclose(file);
 
+        // rename it to real filename // TODO: handle exceptions (interrupt or bad request)
+        rename(tmpFile.c_str(), filepath.c_str());
+
         return res == CURLE_OK;
     }