-/* PROJECT: ReactOS Applications Manager
- * LICENSE: GPL - See COPYING in the top level directory
+/*
+ * PROJECT: ReactOS Applications Manager
+ * LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
* FILE: base/applications/rapps/loaddlg.cpp
* PURPOSE: Displaying a download dialog
* COPYRIGHT: Copyright 2001 John R. Sheets (for CodeWeavers)
* Copyright 2005 Ge van Geldorp (gvg@reactos.org)
* Copyright 2009 Dmitry Chapyshev (dmitry@reactos.org)
* Copyright 2015 Ismael Ferreras Morezuelas (swyterzone+ros@gmail.com)
- * Copyright 2017 Alexander Shaposhnikov (chaez.san@gmail.com)
- */
-
-/*
- * Based on Wine dlls/shdocvw/shdocvw_main.c
- *
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 2.1 of the License, or (at your option) any later version.
- *
- * This library is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with this library; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ * Copyright 2017 Alexander Shaposhnikov (sanchaez@reactos.org)
*/
+ /*
+ * Based on Wine dlls/shdocvw/shdocvw_main.c
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
#include "rapps.h"
#include <shlobj_undoc.h>
#include <atlbase.h>
#include <atlcom.h>
+#include <atlwin.h>
#include <wininet.h>
#include <shellutils.h>
+
+#include <rosctrls.h>
#include <windowsx.h>
+#include "rosui.h"
+#include "dialogs.h"
+#include "misc.h"
+
+#ifdef USE_CERT_PINNING
+#define CERT_ISSUER_INFO "BE\r\nGlobalSign nv-sa\r\nGlobalSign Domain Validation CA - SHA256 - G2"
+#define CERT_SUBJECT_INFO "Domain Control Validated\r\n*.reactos.org"
+#endif
+
+enum DownloadStatus
+{
+ DLSTATUS_WAITING = IDS_STATUS_WAITING,
+ DLSTATUS_DOWNLOADING = IDS_STATUS_DOWNLOADING,
+ DLSTATUS_WAITING_INSTALL = IDS_STATUS_DOWNLOADED,
+ DLSTATUS_INSTALLING = IDS_STATUS_INSTALLING,
+ DLSTATUS_INSTALLED = IDS_STATUS_INSTALLED,
+ DLSTATUS_FINISHED = IDS_STATUS_FINISHED
+};
+
+ATL::CStringW LoadStatusString(DownloadStatus StatusParam)
+{
+ ATL::CStringW szString;
+ szString.LoadStringW(StatusParam);
+ return szString;
+}
+
+struct DownloadInfo
+{
+ DownloadInfo() {}
+ DownloadInfo(const CAvailableApplicationInfo& AppInfo)
+ :szUrl(AppInfo.m_szUrlDownload), szName(AppInfo.m_szName), szSHA1(AppInfo.m_szSHA1)
+ {
+ }
+
+ ATL::CStringW szUrl;
+ ATL::CStringW szName;
+ ATL::CStringW szSHA1;
+};
+
+struct DownloadParam
+{
+ DownloadParam() : Dialog(NULL), AppInfo(), szCaption(NULL) {}
+ DownloadParam(HWND dlg, const ATL::CSimpleArray<DownloadInfo> &info, LPCWSTR caption)
+ : Dialog(dlg), AppInfo(info), szCaption(caption)
+ {
+ }
+
+ HWND Dialog;
+ ATL::CSimpleArray<DownloadInfo> AppInfo;
+ LPCWSTR szCaption;
+};
+
+
class CDownloadDialog :
public CComObjectRootEx<CComMultiThreadModelNoCS>,
public IBindStatusCallback
public:
~CDownloadDialog()
{
- DestroyWindow(m_hDialog);
+ //DestroyWindow(m_hDialog);
}
HRESULT Initialize(HWND Dlg, BOOL *pbCancelled)
m_UrlHasBeenCopied = TRUE;
}
- SetLastError(0);
+ SetLastError(ERROR_SUCCESS);
r = GetWindowLongPtrW(m_hDialog, GWLP_USERDATA);
- if (0 != r || 0 != GetLastError())
+ if (r || GetLastError() != ERROR_SUCCESS)
{
*m_pbCancelled = TRUE;
return E_ABORT;
END_COM_MAP()
};
+class CDowloadingAppsListView
+ : public CUiWindow<CListView>
+{
+public:
+ HWND Create(HWND hwndParent)
+ {
+ RECT r = {10, 150, 320, 350};
+ const DWORD style = WS_CHILD | WS_VISIBLE | LVS_REPORT | LVS_SINGLESEL
+ | LVS_SHOWSELALWAYS | LVS_NOSORTHEADER | LVS_NOCOLUMNHEADER;
+
+ HWND hwnd = CListView::Create(hwndParent, r, NULL, style, WS_EX_CLIENTEDGE);
+
+ AddColumn(0, 150, LVCFMT_LEFT);
+ AddColumn(1, 120, LVCFMT_LEFT);
+
+ return hwnd;
+ }
+
+ VOID LoadList(ATL::CSimpleArray<DownloadInfo> arrInfo)
+ {
+ for (INT i = 0; i < arrInfo.GetSize(); ++i)
+ {
+ AddRow(i, arrInfo[i].szName.GetString(), DLSTATUS_WAITING);
+ }
+ }
+
+ VOID SetDownloadStatus(INT ItemIndex, DownloadStatus Status)
+ {
+ HWND hListView = GetWindow();
+ ATL::CStringW szBuffer = LoadStatusString(Status);
+ ListView_SetItemText(hListView, ItemIndex, 1, const_cast<LPWSTR>(szBuffer.GetString()));
+ }
+
+ BOOL AddItem(INT ItemIndex, LPWSTR lpText)
+ {
+ LVITEMW Item;
+
+ ZeroMemory(&Item, sizeof(Item));
+
+ Item.mask = LVIF_TEXT | LVIF_STATE;
+ Item.pszText = lpText;
+ Item.iItem = ItemIndex;
+
+ return InsertItem(&Item);
+ }
+
+ VOID AddRow(INT RowIndex, LPCWSTR szAppName, const DownloadStatus Status)
+ {
+ ATL::CStringW szStatus = LoadStatusString(Status);
+ AddItem(RowIndex,
+ const_cast<LPWSTR>(szAppName));
+ SetDownloadStatus(RowIndex, Status);
+ }
+
+ BOOL AddColumn(INT Index, INT Width, INT Format)
+ {
+ LVCOLUMNW Column;
+ ZeroMemory(&Column, sizeof(Column));
+
+ Column.mask = LVCF_FMT | LVCF_WIDTH | LVCF_SUBITEM;
+ Column.iSubItem = Index;
+ Column.cx = Width;
+ Column.fmt = Format;
+
+ return (InsertColumn(Index, &Column) == -1) ? FALSE : TRUE;
+ }
+};
+
extern "C"
HRESULT WINAPI CDownloadDialog_Constructor(HWND Dlg, BOOL *pbCancelled, REFIID riid, LPVOID *ppv)
{
}
#endif
-inline VOID
-MessageBox_LoadString(HWND hMainWnd, INT StringID)
+inline VOID MessageBox_LoadString(HWND hMainWnd, INT StringID)
{
ATL::CString szMsgText;
- if (szMsgText.LoadStringW(hInst, StringID))
+ if (szMsgText.LoadStringW(StringID))
+ {
MessageBoxW(hMainWnd, szMsgText.GetString(), NULL, MB_OK | MB_ICONERROR);
+ }
}
-// DownloadManager
+// CDownloadManager
+ATL::CSimpleArray<DownloadInfo> CDownloadManager::AppsToInstallList;
+CDowloadingAppsListView CDownloadManager::DownloadsListView;
-PAPPLICATION_INFO DownloadManager::AppInfo = NULL;
+VOID CDownloadManager::Download(const DownloadInfo &DLInfo, BOOL bIsModal)
+{
+ AppsToInstallList.RemoveAll();
+ AppsToInstallList.Add(DLInfo);
+ LaunchDownloadDialog(bIsModal);
+}
-INT_PTR CALLBACK DownloadManager::DownloadDlgProc(HWND Dlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
+INT_PTR CALLBACK CDownloadManager::DownloadDlgProc(HWND Dlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
- HANDLE Thread;
- DWORD ThreadId;
- HWND Item;
+ static WCHAR szCaption[MAX_PATH];
switch (uMsg)
{
case WM_INITDIALOG:
{
HICON hIconSm, hIconBg;
- WCHAR szCaption[MAX_PATH];
- ATL::CStringW szNewCaption;
hIconBg = (HICON) GetClassLongW(hMainWnd, GCLP_HICON);
hIconSm = (HICON) GetClassLongW(hMainWnd, GCLP_HICONSM);
SendMessageW(Dlg, WM_SETICON, ICON_SMALL, (LPARAM) hIconSm);
}
- // Change caption to show the currently downloaded app
- GetWindowTextW(Dlg, szCaption, MAX_PATH);
- szNewCaption.Format(szCaption, AppInfo->szName.GetString());
- SetWindowTextW(Dlg, szNewCaption.GetString());
-
SetWindowLongW(Dlg, GWLP_USERDATA, 0);
- Item = GetDlgItem(Dlg, IDC_DOWNLOAD_PROGRESS);
+ HWND Item = GetDlgItem(Dlg, IDC_DOWNLOAD_PROGRESS);
if (Item)
{
// initialize the default values for our nifty progress bar
// and subclass it so that it learns to print a status text
SendMessageW(Item, PBM_SETRANGE, 0, MAKELPARAM(0, 100));
SendMessageW(Item, PBM_SETPOS, 0, 0);
-
+
SetWindowSubclass(Item, DownloadProgressProc, 0, 0);
}
- // add a neat placeholder until the download URL is retrieved
- SetDlgItemTextW(Dlg, IDC_DOWNLOAD_STATUS, L"\x2022 \x2022 \x2022");
+ // Add a ListView
+ HWND hListView = DownloadsListView.Create(Dlg);
+ if (!hListView)
+ {
+ return FALSE;
+ }
+ DownloadsListView.LoadList(AppsToInstallList);
+
+ ShowWindow(Dlg, SW_SHOW);
+
+ // Get a dlg string for later use
+ GetWindowTextW(Dlg, szCaption, MAX_PATH);
+
+ // Start download process
+ DownloadParam *param = new DownloadParam(Dlg, AppsToInstallList, szCaption);
+ DWORD ThreadId;
+ HANDLE Thread = CreateThread(NULL, 0, ThreadFunc, (LPVOID) param, 0, &ThreadId);
- Thread = CreateThread(NULL, 0, ThreadFunc, Dlg, 0, &ThreadId);
if (!Thread)
+ {
return FALSE;
+ }
+
CloseHandle(Thread);
+ AppsToInstallList.RemoveAll();
return TRUE;
}
+
case WM_COMMAND:
if (wParam == IDCANCEL)
{
- SetWindowLongPtrW(Dlg, GWLP_USERDATA, 1);
+ SetWindowLongW(Dlg, GWLP_USERDATA, 1);
PostMessageW(Dlg, WM_CLOSE, 0, 0);
}
return FALSE;
case WM_CLOSE:
- DestroyWindow(Dlg);
+ EndDialog(Dlg, 0);
+ //DestroyWindow(Dlg);
return TRUE;
default:
}
}
-LRESULT CALLBACK DownloadManager::DownloadProgressProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam, UINT_PTR uIdSubclass, DWORD_PTR dwRefData)
+LRESULT CALLBACK CDownloadManager::DownloadProgressProc(HWND hWnd,
+ UINT uMsg,
+ WPARAM wParam,
+ LPARAM lParam,
+ UINT_PTR uIdSubclass,
+ DWORD_PTR dwRefData)
{
static ATL::CStringW szProgressText;
/* Raymond Chen says that we should safely unsubclass all the things!
(http://blogs.msdn.com/b/oldnewthing/archive/2003/11/11/55653.aspx) */
+
case WM_NCDESTROY:
{
szProgressText.Empty();
}
}
-DWORD WINAPI DownloadManager::ThreadFunc(LPVOID Context)
+DWORD WINAPI CDownloadManager::ThreadFunc(LPVOID param)
{
CComPtr<IBindStatusCallback> dl;
ATL::CStringW Path;
PWSTR p, q;
- HWND Dlg = (HWND) Context;
+
+ HWND hDlg = static_cast<DownloadParam*>(param)->Dialog;
+ HWND Item;
+ INT iAppId;
+
ULONG dwContentLen, dwBytesWritten, dwBytesRead, dwStatus;
ULONG dwCurrentBytesRead = 0;
ULONG dwStatusLen = sizeof(dwStatus);
+
BOOL bCancelled = FALSE;
BOOL bTempfile = FALSE;
BOOL bCab = FALSE;
+
HINTERNET hOpen = NULL;
HINTERNET hFile = NULL;
HANDLE hOut = INVALID_HANDLE_VALUE;
+
unsigned char lpBuffer[4096];
LPCWSTR lpszAgent = L"RApps/1.0";
URL_COMPONENTS urlComponents;
size_t urlLength, filenameLength;
- if (!AppInfo)
+ const ATL::CSimpleArray<DownloadInfo> &InfoArray = static_cast<DownloadParam*>(param)->AppInfo;
+ LPCWSTR szCaption = static_cast<DownloadParam*>(param)->szCaption;
+ ATL::CStringW szNewCaption;
+
+ if (InfoArray.GetSize() <= 0)
{
MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD);
goto end;
}
- /* build the path for the download */
- p = wcsrchr(AppInfo->szUrlDownload.GetString(), L'/');
- q = wcsrchr(AppInfo->szUrlDownload.GetString(), L'?');
+ for (iAppId = 0; iAppId < InfoArray.GetSize(); ++iAppId)
+ {
+ // build the path for the download
+ p = wcsrchr(InfoArray[iAppId].szUrl.GetString(), L'/');
+ q = wcsrchr(InfoArray[iAppId].szUrl.GetString(), L'?');
- /* do we have a final slash separator? */
- if (!p)
- goto end;
+ // do we have a final slash separator?
+ if (!p)
+ goto end;
- /* prepare the tentative length of the filename, maybe we've to remove part of it later on */
- filenameLength = wcslen(p) * sizeof(WCHAR);
+ // prepare the tentative length of the filename, maybe we've to remove part of it later on
+ filenameLength = wcslen(p) * sizeof(WCHAR);
- /* do we have query arguments in the target URL after the filename? account for them
- (e.g. https://example.org/myfile.exe?no_adware_plz) */
- if (q && q > p && (q - p) > 0)
- filenameLength -= wcslen(q - 1) * sizeof(WCHAR);
+ /* do we have query arguments in the target URL after the filename? account for them
+ (e.g. https://example.org/myfile.exe?no_adware_plz) */
+ if (q && q > p && (q - p) > 0)
+ filenameLength -= wcslen(q - 1) * sizeof(WCHAR);
- /* is this URL an update package for RAPPS? if so store it in a different place */
- if (AppInfo->szUrlDownload == APPLICATION_DATABASE_URL)
- {
- bCab = TRUE;
- if (!GetStorageDirectory(Path))
- goto end;
- }
- else
- {
- Path = SettingsInfo.szDownloadDir;
- }
+ // is this URL an update package for RAPPS? if so store it in a different place
+ if (InfoArray[iAppId].szUrl == APPLICATION_DATABASE_URL)
+ {
+ bCab = TRUE;
+ if (!GetStorageDirectory(Path))
+ goto end;
+ }
+ else
+ {
+ Path = SettingsInfo.szDownloadDir;
+ }
- /* is the path valid? can we access it? */
- if (GetFileAttributesW(Path.GetString()) == INVALID_FILE_ATTRIBUTES)
- {
- if (!CreateDirectoryW(Path.GetString(), NULL))
- goto end;
- }
+ // is the path valid? can we access it?
+ if (GetFileAttributesW(Path.GetString()) == INVALID_FILE_ATTRIBUTES)
+ {
+ if (!CreateDirectoryW(Path.GetString(), NULL))
+ goto end;
+ }
- /* append a \ to the provided file system path, and the filename portion from the URL after that */
- Path.Format(L"\\%ls", (p + 1));
+ // append a \ to the provided file system path, and the filename portion from the URL after that
+ Path += L"\\";
+ Path += (LPWSTR) (p + 1);
- if (!bCab && AppInfo->szSHA1[0] && GetFileAttributesW(Path.GetString()) != INVALID_FILE_ATTRIBUTES)
- {
- /* only open it in case of total correctness */
- if (VerifyInteg(AppInfo->szSHA1, Path))
- goto run;
- }
+ if (!bCab && InfoArray[iAppId].szSHA1[0] && GetFileAttributesW(Path.GetString()) != INVALID_FILE_ATTRIBUTES)
+ {
+ // only open it in case of total correctness
+ if (VerifyInteg(InfoArray[iAppId].szSHA1.GetString(), Path))
+ goto run;
+ }
- /* download it */
- bTempfile = TRUE;
- CDownloadDialog_Constructor(Dlg, &bCancelled, IID_PPV_ARG(IBindStatusCallback, &dl));
+ // Reset progress bar
+ Item = GetDlgItem(hDlg, IDC_DOWNLOAD_PROGRESS);
+ if (Item)
+ {
+ SendMessageW(Item, PBM_SETPOS, 0, 0);
+ }
- if (dl == NULL)
- goto end;
+ // Change caption to show the currently downloaded app
+ if (!bCab)
+ {
+ szNewCaption.Format(szCaption, InfoArray[iAppId].szName.GetString());
+ }
+ else
+ {
+ szNewCaption.LoadStringW(IDS_DL_DIALOG_DB_DOWNLOAD_DISP);
+ }
- /* FIXME: this should just be using the system-wide proxy settings */
- switch (SettingsInfo.Proxy)
- {
- case 0: /* preconfig */
- hOpen = InternetOpenW(lpszAgent, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
- break;
- case 1: /* direct (no proxy) */
- hOpen = InternetOpenW(lpszAgent, INTERNET_OPEN_TYPE_DIRECT, NULL, NULL, 0);
- break;
- case 2: /* use proxy */
- hOpen = InternetOpenW(lpszAgent, INTERNET_OPEN_TYPE_PROXY, SettingsInfo.szProxyServer, SettingsInfo.szNoProxyFor, 0);
- break;
- default: /* preconfig */
- hOpen = InternetOpenW(lpszAgent, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
- break;
- }
+ SetWindowTextW(hDlg, szNewCaption.GetString());
- if (!hOpen)
- goto end;
+ // Add the download URL
+ SetDlgItemTextW(hDlg, IDC_DOWNLOAD_STATUS, InfoArray[iAppId].szUrl.GetString());
- hFile = InternetOpenUrlW(hOpen, AppInfo->szUrlDownload, NULL, 0, INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_KEEP_CONNECTION, 0);
- if (!hFile)
- {
- MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD2);
- goto end;
- }
+ DownloadsListView.SetDownloadStatus(iAppId, DLSTATUS_DOWNLOADING);
- if (!HttpQueryInfoW(hFile, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, &dwStatus, &dwStatusLen, NULL))
- goto end;
+ // download it
+ bTempfile = TRUE;
+ CDownloadDialog_Constructor(hDlg, &bCancelled, IID_PPV_ARG(IBindStatusCallback, &dl));
- if (dwStatus != HTTP_STATUS_OK)
- {
- MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD);
- goto end;
- }
+ if (dl == NULL)
+ goto end;
+
+ /* FIXME: this should just be using the system-wide proxy settings */
+ switch (SettingsInfo.Proxy)
+ {
+ case 0: // preconfig
+ hOpen = InternetOpenW(lpszAgent, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
+ break;
+ case 1: // direct (no proxy)
+ hOpen = InternetOpenW(lpszAgent, INTERNET_OPEN_TYPE_DIRECT, NULL, NULL, 0);
+ break;
+ case 2: // use proxy
+ hOpen = InternetOpenW(lpszAgent, INTERNET_OPEN_TYPE_PROXY, SettingsInfo.szProxyServer, SettingsInfo.szNoProxyFor, 0);
+ break;
+ default: // preconfig
+ hOpen = InternetOpenW(lpszAgent, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
+ break;
+ }
- dwStatusLen = sizeof(dwStatus);
+ if (!hOpen)
+ goto end;
- memset(&urlComponents, 0, sizeof(urlComponents));
- urlComponents.dwStructSize = sizeof(urlComponents);
+ hFile = InternetOpenUrlW(hOpen, InfoArray[iAppId].szUrl.GetString(), NULL, 0, INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_KEEP_CONNECTION, 0);
- urlLength = AppInfo->szUrlDownload.GetLength();
- urlComponents.dwSchemeLength = urlLength + 1;
- urlComponents.lpszScheme = (LPWSTR) malloc(urlComponents.dwSchemeLength * sizeof(WCHAR));
- urlComponents.dwHostNameLength = urlLength + 1;
- urlComponents.lpszHostName = (LPWSTR) malloc(urlComponents.dwHostNameLength * sizeof(WCHAR));
+ if (!hFile)
+ {
+ MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD2);
+ goto end;
+ }
- if (!InternetCrackUrlW(AppInfo->szUrlDownload, urlLength + 1, ICU_DECODE | ICU_ESCAPE, &urlComponents))
- goto end;
+ if (!HttpQueryInfoW(hFile, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, &dwStatus, &dwStatusLen, NULL))
+ goto end;
- if (urlComponents.nScheme == INTERNET_SCHEME_HTTP || urlComponents.nScheme == INTERNET_SCHEME_HTTPS)
- HttpQueryInfo(hFile, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, &dwContentLen, &dwStatus, 0);
+ if (dwStatus != HTTP_STATUS_OK)
+ {
+ MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD);
+ goto end;
+ }
- if (urlComponents.nScheme == INTERNET_SCHEME_FTP)
- dwContentLen = FtpGetFileSize(hFile, &dwStatus);
+ dwStatusLen = sizeof(dwStatus);
-#ifdef USE_CERT_PINNING
- /* are we using HTTPS to download the RAPPS update package? check if the certificate is original */
- if ((urlComponents.nScheme == INTERNET_SCHEME_HTTPS) &&
- (wcscmp(AppInfo->szUrlDownload, APPLICATION_DATABASE_URL) == 0) &&
- (!CertIsValid(hOpen, urlComponents.lpszHostName)))
- {
- MessageBox_LoadString(hMainWnd, IDS_CERT_DOES_NOT_MATCH);
- goto end;
- }
-#endif
+ memset(&urlComponents, 0, sizeof(urlComponents));
+ urlComponents.dwStructSize = sizeof(urlComponents);
- free(urlComponents.lpszScheme);
- free(urlComponents.lpszHostName);
+ urlLength = InfoArray[iAppId].szUrl.GetLength();
+ urlComponents.dwSchemeLength = urlLength + 1;
+ urlComponents.lpszScheme = (LPWSTR) malloc(urlComponents.dwSchemeLength * sizeof(WCHAR));
+ urlComponents.dwHostNameLength = urlLength + 1;
+ urlComponents.lpszHostName = (LPWSTR) malloc(urlComponents.dwHostNameLength * sizeof(WCHAR));
- hOut = CreateFileW(Path.GetString(), GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, 0, NULL);
+ if (!InternetCrackUrlW(InfoArray[iAppId].szUrl, urlLength + 1, ICU_DECODE | ICU_ESCAPE, &urlComponents))
+ goto end;
- if (hOut == INVALID_HANDLE_VALUE)
- goto end;
+ if (urlComponents.nScheme == INTERNET_SCHEME_HTTP || urlComponents.nScheme == INTERNET_SCHEME_HTTPS)
+ HttpQueryInfoW(hFile, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, &dwContentLen, &dwStatus, 0);
- do
- {
- if (!InternetReadFile(hFile, lpBuffer, _countof(lpBuffer), &dwBytesRead))
+ if (urlComponents.nScheme == INTERNET_SCHEME_FTP)
+ dwContentLen = FtpGetFileSize(hFile, &dwStatus);
+
+#ifdef USE_CERT_PINNING
+ // are we using HTTPS to download the RAPPS update package? check if the certificate is original
+ if ((urlComponents.nScheme == INTERNET_SCHEME_HTTPS) &&
+ (wcscmp(InfoArray[iAppId].szUrl, APPLICATION_DATABASE_URL) == 0) &&
+ (!CertIsValid(hOpen, urlComponents.lpszHostName)))
{
- MessageBox_LoadString(hMainWnd, IDS_INTERRUPTED_DOWNLOAD);
+ MessageBox_LoadString(hMainWnd, IDS_CERT_DOES_NOT_MATCH);
goto end;
}
+#endif
- if (!WriteFile(hOut, &lpBuffer[0], dwBytesRead, &dwBytesWritten, NULL))
- {
- MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_WRITE);
+ free(urlComponents.lpszScheme);
+ free(urlComponents.lpszHostName);
+
+ hOut = CreateFileW(Path.GetString(), GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, 0, NULL);
+
+ if (hOut == INVALID_HANDLE_VALUE)
goto end;
- }
- dwCurrentBytesRead += dwBytesRead;
- dl->OnProgress(dwCurrentBytesRead, dwContentLen, 0, AppInfo->szUrlDownload);
- } while (dwBytesRead && !bCancelled);
+ dwCurrentBytesRead = 0;
+ do
+ {
+ if (!InternetReadFile(hFile, lpBuffer, _countof(lpBuffer), &dwBytesRead))
+ {
+ MessageBox_LoadString(hMainWnd, IDS_INTERRUPTED_DOWNLOAD);
+ goto end;
+ }
- CloseHandle(hOut);
- hOut = INVALID_HANDLE_VALUE;
+ if (!WriteFile(hOut, &lpBuffer[0], dwBytesRead, &dwBytesWritten, NULL))
+ {
+ MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_WRITE);
+ goto end;
+ }
- if (bCancelled)
- goto end;
+ dwCurrentBytesRead += dwBytesRead;
+ dl->OnProgress(dwCurrentBytesRead, dwContentLen, 0, InfoArray[iAppId].szUrl.GetString());
+ } while (dwBytesRead && !bCancelled);
- /* if this thing isn't a RAPPS update and it has a SHA-1 checksum
- verify its integrity by using the native advapi32.A_SHA1 functions */
- if (!bCab && AppInfo->szSHA1[0] != 0)
- {
- ATL::CStringW szMsgText;
+ CloseHandle(hOut);
+ hOut = INVALID_HANDLE_VALUE;
- /* change a few strings in the download dialog to reflect the verification process */
- if (!szMsgText.LoadStringW(hInst, IDS_INTEG_CHECK_TITLE))
+ if (bCancelled)
goto end;
- SetWindowText(Dlg, szMsgText.GetString());
- SendMessageW(GetDlgItem(Dlg, IDC_DOWNLOAD_STATUS), WM_SETTEXT, 0, (LPARAM) Path.GetString());
-
- /* this may take a while, depending on the file size */
- if (!VerifyInteg(AppInfo->szSHA1, Path.GetString()))
+ /* if this thing isn't a RAPPS update and it has a SHA-1 checksum
+ verify its integrity by using the native advapi32.A_SHA1 functions */
+ if (!bCab && InfoArray[iAppId].szSHA1[0] != 0)
{
- if (!szMsgText.LoadStringW(hInst, IDS_INTEG_CHECK_FAIL))
+ ATL::CStringW szMsgText;
+
+ // change a few strings in the download dialog to reflect the verification process
+ if (!szMsgText.LoadStringW(IDS_INTEG_CHECK_TITLE))
goto end;
- MessageBoxW(Dlg, szMsgText.GetString(), NULL, MB_OK | MB_ICONERROR);
- goto end;
- }
- }
+ SetWindowTextW(hDlg, szMsgText.GetString());
+ SendMessageW(GetDlgItem(hDlg, IDC_DOWNLOAD_STATUS), WM_SETTEXT, 0, (LPARAM) Path.GetString());
- ShowWindow(Dlg, SW_HIDE);
+ // this may take a while, depending on the file size
+ if (!VerifyInteg(InfoArray[iAppId].szSHA1.GetString(), Path.GetString()))
+ {
+ if (!szMsgText.LoadStringW(IDS_INTEG_CHECK_FAIL))
+ goto end;
+
+ MessageBoxW(hDlg, szMsgText.GetString(), NULL, MB_OK | MB_ICONERROR);
+ goto end;
+ }
+ }
run:
- /* run it */
- if (!bCab)
- ShellExecuteW(NULL, L"open", Path.GetString(), NULL, NULL, SW_SHOWNORMAL);
+ DownloadsListView.SetDownloadStatus(iAppId, DLSTATUS_WAITING_INSTALL);
+
+ // run it
+ if (!bCab)
+ {
+ SHELLEXECUTEINFOW shExInfo = {0};
+ shExInfo.cbSize = sizeof(shExInfo);
+ shExInfo.fMask = SEE_MASK_NOCLOSEPROCESS;
+ shExInfo.lpVerb = L"open";
+ shExInfo.lpFile = Path.GetString();
+ shExInfo.lpParameters = L"";
+ shExInfo.nShow = SW_SHOW;
+
+ if (ShellExecuteExW(&shExInfo))
+ {
+ DownloadsListView.SetDownloadStatus(iAppId, DLSTATUS_INSTALLING);
+ //TODO: issue an install operation separately so that the apps could be downloaded in the background
+ WaitForSingleObject(shExInfo.hProcess, INFINITE);
+ CloseHandle(shExInfo.hProcess);
+ }
+ else
+ {
+ MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_INSTALL);
+ }
+ }
end:
- if (hOut != INVALID_HANDLE_VALUE)
- CloseHandle(hOut);
+ if (hOut != INVALID_HANDLE_VALUE)
+ CloseHandle(hOut);
- InternetCloseHandle(hFile);
- InternetCloseHandle(hOpen);
+ InternetCloseHandle(hFile);
+ InternetCloseHandle(hOpen);
- if (bTempfile)
- {
- if (bCancelled || (SettingsInfo.bDelInstaller && !bCab))
- DeleteFileW(Path.GetString());
- }
+ if (bTempfile)
+ {
+ if (bCancelled || (SettingsInfo.bDelInstaller && !bCab))
+ DeleteFileW(Path.GetString());
+ }
- EndDialog(Dlg, 0);
+ DownloadsListView.SetDownloadStatus(iAppId, DLSTATUS_FINISHED);
+ }
+ delete static_cast<DownloadParam*>(param);
+ SendMessageW(hDlg, WM_CLOSE, 0, 0);
return 0;
}
-//TODO: Maybe launch this (similar function) in a seperate thread, so the list could be updated
-BOOL DownloadManager::DownloadListOfApplications(const ATL::CSimpleArray<PAPPLICATION_INFO>& AppsList)
+BOOL CDownloadManager::DownloadListOfApplications(const ATL::CSimpleArray<CAvailableApplicationInfo*>& AppsList, BOOL bIsModal)
{
- BOOL bResult = TRUE;
-
- for (INT i = 0; i < AppsList.GetSize(); ++i)
+ if (AppsList.GetSize() == 0)
{
- bResult = DownloadApplication(AppsList[i]) && bResult;
+ return FALSE;
}
- return bResult;
-}
-BOOL DownloadManager::DownloadApplication(PAPPLICATION_INFO pAppInfo)
-{
- if (!pAppInfo)
+ // Initialize shared variables
+ for (INT i = 0; i < AppsList.GetSize(); ++i)
{
- return FALSE;
+ if (AppsList[i])
+ {
+ AppsToInstallList.Add(*(AppsList[i]));
+ }
}
// Create a dialog and issue a download process
- AppInfo = pAppInfo;
- LaunchDownloadDialog();
-
- WriteLogMessage(EVENTLOG_SUCCESS, MSG_SUCCESS_INSTALL, AppInfo->szName.GetString());
+ LaunchDownloadDialog(bIsModal);
return TRUE;
}
-VOID DownloadManager::DownloadApplicationsDB(LPCWSTR lpUrl)
+BOOL CDownloadManager::DownloadApplication(CAvailableApplicationInfo* pAppInfo, BOOL bIsModal)
{
- APPLICATION_INFO IntInfo;
- IntInfo.szUrlDownload = lpUrl;
+ if (!pAppInfo)
+ return FALSE;
- AppInfo = &IntInfo;
+ Download(*pAppInfo, bIsModal);
+ return TRUE;
+}
- LaunchDownloadDialog();
+VOID CDownloadManager::DownloadApplicationsDB(LPCWSTR lpUrl)
+{
+ static DownloadInfo DatabaseDLInfo;
+ DatabaseDLInfo.szUrl = lpUrl;
+ DatabaseDLInfo.szName.LoadStringW(IDS_DL_DIALOG_DB_DISP);
+ Download(DatabaseDLInfo, TRUE);
}
//TODO: Reuse the dialog
-VOID DownloadManager::LaunchDownloadDialog()
+VOID CDownloadManager::LaunchDownloadDialog(BOOL bIsModal)
{
- CreateDialogW(hInst,
- MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG),
- hMainWnd,
- DownloadDlgProc);
-}
\ No newline at end of file
+ if (bIsModal)
+ {
+ DialogBoxW(hInst,
+ MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG),
+ hMainWnd,
+ DownloadDlgProc);
+ }
+ else
+ {
+ CreateDialogW(hInst,
+ MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG),
+ hMainWnd,
+ DownloadDlgProc);
+ }
+}
+// CDownloadManager