updater: Add windows updater module

2017-02-20 04:50:17 -08:00
parent 1355243697
commit 86862b672e
9 changed files with 2582 additions and 0 deletions
--- a/UI/win-update/updater/updater.hpp
+++ b/UI/win-update/updater/updater.hpp
@@ -0,0 +1,103 @@
+#pragma once
+
+#define WINVER 0x0600
+#define _WIN32_WINDOWS 0x0600
+#define _WIN32_WINNT 0x0600
+#define WIN32_LEAN_AND_MEAN
+
+#define ZLIB_CONST
+
+#include <windows.h>
+#include <winhttp.h>
+#include <commctrl.h>
+#include <Wincrypt.h>
+#include <shlobj.h>
+#include <shellapi.h>
+#include <malloc.h>
+#include <stdlib.h>
+#include <tchar.h>
+#include <strsafe.h>
+#include <zlib.h>
+#include <ctype.h>
+#include <blake2.h>
+
+#include <string>
+
+#include "../win-update-helpers.hpp"
+
+#define BLAKE2_HASH_LENGTH 20
+#define BLAKE2_HASH_STR_LENGTH ((BLAKE2_HASH_LENGTH * 2) + 1)
+
+#if defined _M_IX86
+#pragma comment(linker, \
+                "/manifestdependency:\"type='win32' " \
+                "name='Microsoft.Windows.Common-Controls' " \
+                "version='6.0.0.0' " \
+                "processorArchitecture='x86' " \
+                "publicKeyToken='6595b64144ccf1df' " \
+                "language='*'\"")
+#elif defined _M_IA64
+#pragma comment(linker, \
+                "/manifestdependency:\"type='win32' " \
+                "name='Microsoft.Windows.Common-Controls' " \
+                "version='6.0.0.0' " \
+                "processorArchitecture='ia64' " \
+                "publicKeyToken='6595b64144ccf1df' " \
+                "language='*'\"")
+#elif defined _M_X64
+#pragma comment(linker, \
+                "/manifestdependency:\"type='win32' " \
+                "name='Microsoft.Windows.Common-Controls' " \
+                "version='6.0.0.0' " \
+                "processorArchitecture='amd64' " \
+                "publicKeyToken='6595b64144ccf1df' " \
+                "language='*'\"")
+#else
+#pragma comment(linker, \
+                "/manifestdependency:\"type='win32' " \
+                "name='Microsoft.Windows.Common-Controls' " \
+                "version='6.0.0.0' processorArchitecture='*' " \
+                "publicKeyToken='6595b64144ccf1df' " \
+                "language='*'\"")
+#endif
+
+#include <util/windows/WinHandle.hpp>
+#include <jansson.h>
+#include "resource.h"
+
+bool HTTPGetFile(HINTERNET      hConnect,
+                 const wchar_t *url,
+                 const wchar_t *outputPath,
+                 const wchar_t *extraHeaders,
+                 int *          responseCode);
+bool HTTPPostData(const wchar_t *url,
+                  const BYTE *   data,
+                  int            dataLen,
+                  const wchar_t *extraHeaders,
+                  int *          responseCode,
+                  std::string &  response);
+
+void HashToString(const BYTE *in, wchar_t *out);
+void StringToHash(const wchar_t *in, BYTE *out);
+
+bool CalculateFileHash(const wchar_t *path, BYTE *hash);
+
+int ApplyPatch(LPCTSTR patchFile, LPCTSTR targetFile);
+
+extern HWND       hwndMain;
+extern HCRYPTPROV hProvider;
+extern int        totalFileSize;
+extern int        completedFileSize;
+extern HANDLE     cancelRequested;
+
+#pragma pack(push, r1, 1)
+
+typedef struct {
+	BLOBHEADER blobheader;
+	RSAPUBKEY  rsapubkey;
+} PUBLICKEYHEADER;
+
+#pragma pack(pop, r1)
+
+void FreeWinHttpHandle(HINTERNET handle);
+using HttpHandle = CustomHandle<HINTERNET, FreeWinHttpHandle>;