From 8074cb39623f844ac3e24b2c4b11a1165fa67900 Mon Sep 17 00:00:00 2001
From: Rogier Polak <rogier.l.a.polak@gmail.com>
Date: Fri, 16 Feb 2007 15:07:56 +0000
Subject: Fixed update bug (hopefully)

---
 src/gui/updatewindow.cpp | 24 +++++++++++++++++++-----
 1 file changed, 19 insertions(+), 5 deletions(-)

(limited to 'src/gui')

diff --git a/src/gui/updatewindow.cpp b/src/gui/updatewindow.cpp
index c763e488..45211b3e 100644
--- a/src/gui/updatewindow.cpp
+++ b/src/gui/updatewindow.cpp
@@ -268,6 +268,7 @@ int UpdaterWindow::downloadThread(void *ptr)
     while (attempts < 3 && !uw->mDownloadComplete)
     {
         FILE *outfile = NULL;
+        FILE *newfile = NULL;
         uw->setLabel(uw->mCurrentFile + " (0%)");
 
         curl = curl_easy_init();
@@ -328,10 +329,9 @@ int UpdaterWindow::downloadThread(void *ptr)
 
             curl_easy_cleanup(curl);
 
-            uw->mDownloadComplete = true;
-
             if (!uw->mStoreInMemory)
             {
+
                 // Don't check resources2.txt checksum
                 if (uw->mDownloadStatus == UPDATE_RESOURCES)
                 {
@@ -339,16 +339,18 @@ int UpdaterWindow::downloadThread(void *ptr)
 
                     if (uw->mCurrentChecksum != adler)
                     {
-                        uw->mDownloadComplete = false;
+                        fclose(outfile);
+
                         // Remove the corrupted file
                         ::remove(outFilename.c_str());
                         logger->log(
                             "Checksum for file %s failed: (%lx/%lx)",
                             uw->mCurrentFile.c_str(),
                             adler, uw->mCurrentChecksum);
+                        attempts++;
+                        continue; //Bail out her to avoid the renaming
                     }
                 }
-
                 fclose(outfile);
 
                 // Give the file the proper name
@@ -359,6 +361,18 @@ int UpdaterWindow::downloadThread(void *ptr)
                 // the rename will fail on Windows.
                 ::remove(newName.c_str());
                 ::rename(outFilename.c_str(), newName.c_str());
+
+                //Check if we can open it and no errors were encountered during renaming
+                newfile = fopen(newName.c_str(), "rb");
+                if (newfile)
+                {
+                    fclose(newfile);
+                    uw->mDownloadComplete = true;
+                }
+            }
+            else
+            {
+                uw->mDownloadComplete = true; // it's stored in memory, we're done
             }
         }
         attempts++;
@@ -420,7 +434,7 @@ void UpdaterWindow::logic()
                 mCurrentFile = "resources2.txt";
                 mStoreInMemory = false;
                 mDownloadStatus = UPDATE_LIST;
-                download();
+                download(); //download() changes mDownloadComplete to false
             }
             break;
         case UPDATE_LIST:
-- 
cgit v1.2.3-60-g2f50