Jacek Caban : appwiz.cpl: Added checksum validation of downloaded file.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: b00117d4b65e55461b7c54f36ffbf9c0cdf825a8
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=b00117d4b65e55461b7c54f36ffbf9c0cdf825a8

Author: Jacek Caban <jacek at codeweavers.com>
Date:   Wed Dec 29 00:53:38 2010 +0100

appwiz.cpl: Added checksum validation of downloaded file.

---

 dlls/appwiz.cpl/En.rc    |    1 +
 dlls/appwiz.cpl/addons.c |   68 +++++++++++++++++++++++++++++++++++++++++++++-
 dlls/appwiz.cpl/res.h    |    1 +
 3 files changed, 69 insertions(+), 1 deletions(-)

diff --git a/dlls/appwiz.cpl/En.rc b/dlls/appwiz.cpl/En.rc
index f7ffa02..0c5cf76 100644
--- a/dlls/appwiz.cpl/En.rc
+++ b/dlls/appwiz.cpl/En.rc
@@ -46,6 +46,7 @@ STRINGTABLE
 {
     IDS_DOWNLOADING     "Downloading..."
     IDS_INSTALLING      "Installing..."
+    IDS_INVALID_SHA     "Unexpected checksum of downloaded file. Aborting installation of corrupted file."
 }
 
 IDD_MAIN DIALOG 0, 0, 320, 220
diff --git a/dlls/appwiz.cpl/addons.c b/dlls/appwiz.cpl/addons.c
index 4478719..d19af66 100644
--- a/dlls/appwiz.cpl/addons.c
+++ b/dlls/appwiz.cpl/addons.c
@@ -20,6 +20,7 @@
 
 #include <stdarg.h>
 #include <fcntl.h>
+#include <stdio.h>
 #ifdef HAVE_UNISTD_H
 # include <unistd.h>
 #endif
@@ -53,10 +54,13 @@ WINE_DEFAULT_DEBUG_CHANNEL(appwizcpl);
 
 #ifdef __i386__
 #define ARCH_STRING "x86"
+#define GECKO_SHA "1b6c637207b6f032ae8a52841db9659433482714"
 #elif defined(__x86_64__)
 #define ARCH_STRING "x86_64"
+#define GECKO_SHA "55b4b60cd2a48631d6236fb411c3a94d806d9906"
 #else
 #define ARCH_STRING ""
+#define GECKO_SHA "???"
 #endif
 
 #define GECKO_FILE_NAME "wine_gecko-" GECKO_VERSION "-" ARCH_STRING ".cab"
@@ -82,6 +86,67 @@ static inline char *heap_strdupWtoA(LPCWSTR str)
     return ret;
 }
 
+/* SHA definitions are copied from advapi32. They aren't available in headers. */
+
+typedef struct {
+   ULONG Unknown[6];
+   ULONG State[5];
+   ULONG Count[2];
+   UCHAR Buffer[64];
+} SHA_CTX, *PSHA_CTX;
+
+void WINAPI A_SHAInit(PSHA_CTX);
+void WINAPI A_SHAUpdate(PSHA_CTX,const unsigned char*,UINT);
+void WINAPI A_SHAFinal(PSHA_CTX,PULONG);
+
+static BOOL sha_check(const WCHAR *file_name)
+{
+    const unsigned char *file_map;
+    HANDLE file, map;
+    ULONG sha[5];
+    char buf[2*sizeof(sha)+1];
+    SHA_CTX ctx;
+    DWORD size, i;
+
+    file = CreateFileW(file_name, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
+    if(file == INVALID_HANDLE_VALUE)
+        return FALSE;
+
+    size = GetFileSize(file, NULL);
+
+    map = CreateFileMappingW(file, NULL, PAGE_READONLY, 0, 0, NULL);
+    CloseHandle(file);
+    if(!map)
+        return FALSE;
+
+    file_map = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0);
+    CloseHandle(map);
+    if(!file_map)
+        return FALSE;
+
+    A_SHAInit(&ctx);
+    A_SHAUpdate(&ctx, file_map, size);
+    A_SHAFinal(&ctx, sha);
+
+    UnmapViewOfFile(file_map);
+
+    for(i=0; i < sizeof(sha); i++)
+        sprintf(buf + i*2, "%02x", *((unsigned char*)sha+i));
+
+    if(strcmp(buf, GECKO_SHA)) {
+        WCHAR message[256];
+
+        WARN("Got %s, expected %s\n", buf, GECKO_SHA);
+
+        if(LoadStringW(hInst, IDS_INVALID_SHA, message, sizeof(message)/sizeof(WCHAR)))
+            MessageBoxW(NULL, message, NULL, MB_ICONERROR);
+
+        return FALSE;
+    }
+
+    return TRUE;
+}
+
 static void set_status(DWORD id)
 {
     HWND status = GetDlgItem(install_dialog, ID_DWL_STATUS);
@@ -447,7 +512,8 @@ static DWORD WINAPI download_proc(PVOID arg)
         return 0;
     }
 
-    install_cab(tmp_file);
+    if(sha_check(tmp_file))
+        install_cab(tmp_file);
     DeleteFileW(tmp_file);
     EndDialog(install_dialog, 0);
     return 0;
diff --git a/dlls/appwiz.cpl/res.h b/dlls/appwiz.cpl/res.h
index 390c292..3bda4b6 100644
--- a/dlls/appwiz.cpl/res.h
+++ b/dlls/appwiz.cpl/res.h
@@ -68,3 +68,4 @@
 #define IDS_MODIFY_REMOVE      11
 #define IDS_DOWNLOADING        12
 #define IDS_INSTALLING         13
+#define IDS_INVALID_SHA        14