+++ /dev/null
-/*
- * 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 2004 Mike McCormack (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 (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 <shlguid_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
-{
- HWND m_hDialog;
- PBOOL m_pbCancelled;
- BOOL m_UrlHasBeenCopied;
-
-public:
- ~CDownloadDialog()
- {
- //DestroyWindow(m_hDialog);
- }
-
- HRESULT Initialize(HWND Dlg, BOOL *pbCancelled)
- {
- m_hDialog = Dlg;
- m_pbCancelled = pbCancelled;
- m_UrlHasBeenCopied = FALSE;
- return S_OK;
- }
-
- virtual HRESULT STDMETHODCALLTYPE OnStartBinding(
- DWORD dwReserved,
- IBinding *pib)
- {
- return S_OK;
- }
-
- virtual HRESULT STDMETHODCALLTYPE GetPriority(
- LONG *pnPriority)
- {
- return S_OK;
- }
-
- virtual HRESULT STDMETHODCALLTYPE OnLowResource(
- DWORD reserved)
- {
- return S_OK;
- }
-
- virtual HRESULT STDMETHODCALLTYPE OnProgress(
- ULONG ulProgress,
- ULONG ulProgressMax,
- ULONG ulStatusCode,
- LPCWSTR szStatusText)
- {
- HWND Item;
- LONG r;
-
- Item = GetDlgItem(m_hDialog, IDC_DOWNLOAD_PROGRESS);
- if (Item && ulProgressMax)
- {
- WCHAR szProgress[100];
- WCHAR szProgressMax[100];
- UINT uiPercentage = ((ULONGLONG) ulProgress * 100) / ulProgressMax;
-
- /* send the current progress to the progress bar */
- SendMessageW(Item, PBM_SETPOS, uiPercentage, 0);
-
- /* format the bits and bytes into pretty and accessible units... */
- StrFormatByteSizeW(ulProgress, szProgress, _countof(szProgress));
- StrFormatByteSizeW(ulProgressMax, szProgressMax, _countof(szProgressMax));
-
- /* ...and post all of it to our subclassed progress bar text subroutine */
- ATL::CStringW m_ProgressText;
- m_ProgressText.Format(L"%u%% \x2014 %ls / %ls",
- uiPercentage,
- szProgress,
- szProgressMax);
- SendMessageW(Item, WM_SETTEXT, 0, (LPARAM) m_ProgressText.GetString());
- }
-
- Item = GetDlgItem(m_hDialog, IDC_DOWNLOAD_STATUS);
- if (Item && szStatusText && wcslen(szStatusText) > 0 && m_UrlHasBeenCopied == FALSE)
- {
- DWORD len = wcslen(szStatusText) + 1;
- ATL::CStringW buf;
-
- /* beautify our url for display purposes */
- if (!InternetCanonicalizeUrlW(szStatusText, buf.GetBuffer(len), &len, ICU_DECODE | ICU_NO_ENCODE))
- {
- /* just use the original */
- buf.ReleaseBuffer();
- buf = szStatusText;
- }
- else
- {
- buf.ReleaseBuffer();
- }
-
- /* paste it into our dialog and don't do it again in this instance */
- SendMessageW(Item, WM_SETTEXT, 0, (LPARAM) buf.GetString());
- m_UrlHasBeenCopied = TRUE;
- }
-
- SetLastError(ERROR_SUCCESS);
- r = GetWindowLongPtrW(m_hDialog, GWLP_USERDATA);
- if (r || GetLastError() != ERROR_SUCCESS)
- {
- *m_pbCancelled = TRUE;
- return E_ABORT;
- }
-
- return S_OK;
- }
-
- virtual HRESULT STDMETHODCALLTYPE OnStopBinding(
- HRESULT hresult,
- LPCWSTR szError)
- {
- return S_OK;
- }
-
- virtual HRESULT STDMETHODCALLTYPE GetBindInfo(
- DWORD *grfBINDF,
- BINDINFO *pbindinfo)
- {
- return S_OK;
- }
-
- virtual HRESULT STDMETHODCALLTYPE OnDataAvailable(
- DWORD grfBSCF,
- DWORD dwSize,
- FORMATETC *pformatetc,
- STGMEDIUM *pstgmed)
- {
- return S_OK;
- }
-
- virtual HRESULT STDMETHODCALLTYPE OnObjectAvailable(
- REFIID riid,
- IUnknown *punk)
- {
- return S_OK;
- }
-
- BEGIN_COM_MAP(CDownloadDialog)
- COM_INTERFACE_ENTRY_IID(IID_IBindStatusCallback, IBindStatusCallback)
- 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)
-{
- return ShellObjectCreatorInit<CDownloadDialog>(Dlg, pbCancelled, riid, ppv);
-}
-
-#ifdef USE_CERT_PINNING
-static BOOL CertIsValid(HINTERNET hInternet, LPWSTR lpszHostName)
-{
- HINTERNET hConnect;
- HINTERNET hRequest;
- DWORD certInfoLength;
- BOOL Ret = FALSE;
- INTERNET_CERTIFICATE_INFOW certInfo;
-
- hConnect = InternetConnectW(hInternet, lpszHostName, INTERNET_DEFAULT_HTTPS_PORT, NULL, NULL, INTERNET_SERVICE_HTTP, INTERNET_FLAG_SECURE, 0);
- if (hConnect)
- {
- hRequest = HttpOpenRequestW(hConnect, L"HEAD", NULL, NULL, NULL, NULL, INTERNET_FLAG_SECURE, 0);
- if (hRequest != NULL)
- {
- Ret = HttpSendRequestW(hRequest, L"", 0, NULL, 0);
- if (Ret)
- {
- certInfoLength = sizeof(certInfo);
- Ret = InternetQueryOptionW(hRequest,
- INTERNET_OPTION_SECURITY_CERTIFICATE_STRUCT,
- &certInfo,
- &certInfoLength);
- if (Ret)
- {
- if (certInfo.lpszEncryptionAlgName)
- LocalFree(certInfo.lpszEncryptionAlgName);
- if (certInfo.lpszIssuerInfo)
- {
- if (strcmp((LPSTR) certInfo.lpszIssuerInfo, CERT_ISSUER_INFO) != 0)
- Ret = FALSE;
- LocalFree(certInfo.lpszIssuerInfo);
- }
- if (certInfo.lpszProtocolName)
- LocalFree(certInfo.lpszProtocolName);
- if (certInfo.lpszSignatureAlgName)
- LocalFree(certInfo.lpszSignatureAlgName);
- if (certInfo.lpszSubjectInfo)
- {
- if (strcmp((LPSTR) certInfo.lpszSubjectInfo, CERT_SUBJECT_INFO) != 0)
- Ret = FALSE;
- LocalFree(certInfo.lpszSubjectInfo);
- }
- }
- }
- InternetCloseHandle(hRequest);
- }
- InternetCloseHandle(hConnect);
- }
- return Ret;
-}
-#endif
-
-inline VOID MessageBox_LoadString(HWND hMainWnd, INT StringID)
-{
- ATL::CString szMsgText;
- if (szMsgText.LoadStringW(StringID))
- {
- MessageBoxW(hMainWnd, szMsgText.GetString(), NULL, MB_OK | MB_ICONERROR);
- }
-}
-
-// CDownloadManager
-ATL::CSimpleArray<DownloadInfo> CDownloadManager::AppsToInstallList;
-CDowloadingAppsListView CDownloadManager::DownloadsListView;
-
-VOID CDownloadManager::Download(const DownloadInfo &DLInfo, BOOL bIsModal)
-{
- AppsToInstallList.RemoveAll();
- AppsToInstallList.Add(DLInfo);
- LaunchDownloadDialog(bIsModal);
-}
-
-INT_PTR CALLBACK CDownloadManager::DownloadDlgProc(HWND Dlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
-{
- static WCHAR szCaption[MAX_PATH];
-
- switch (uMsg)
- {
- case WM_INITDIALOG:
- {
- HICON hIconSm, hIconBg;
-
- hIconBg = (HICON) GetClassLongW(hMainWnd, GCLP_HICON);
- hIconSm = (HICON) GetClassLongW(hMainWnd, GCLP_HICONSM);
-
- if (hIconBg && hIconSm)
- {
- SendMessageW(Dlg, WM_SETICON, ICON_BIG, (LPARAM) hIconBg);
- SendMessageW(Dlg, WM_SETICON, ICON_SMALL, (LPARAM) hIconSm);
- }
-
- SetWindowLongW(Dlg, GWLP_USERDATA, 0);
- 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 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);
-
- if (!Thread)
- {
- return FALSE;
- }
-
- CloseHandle(Thread);
- AppsToInstallList.RemoveAll();
- return TRUE;
- }
-
- case WM_COMMAND:
- if (wParam == IDCANCEL)
- {
- SetWindowLongW(Dlg, GWLP_USERDATA, 1);
- PostMessageW(Dlg, WM_CLOSE, 0, 0);
- }
- return FALSE;
-
- case WM_CLOSE:
- EndDialog(Dlg, 0);
- //DestroyWindow(Dlg);
- return TRUE;
-
- default:
- return FALSE;
- }
-}
-
-LRESULT CALLBACK CDownloadManager::DownloadProgressProc(HWND hWnd,
- UINT uMsg,
- WPARAM wParam,
- LPARAM lParam,
- UINT_PTR uIdSubclass,
- DWORD_PTR dwRefData)
-{
- static ATL::CStringW szProgressText;
-
- switch (uMsg)
- {
- case WM_SETTEXT:
- {
- if (lParam)
- {
- szProgressText = (PCWSTR) lParam;
- }
- return TRUE;
- }
-
- case WM_ERASEBKGND:
- case WM_PAINT:
- {
- PAINTSTRUCT ps;
- HDC hDC = BeginPaint(hWnd, &ps), hdcMem;
- HBITMAP hbmMem;
- HANDLE hOld;
- RECT myRect;
- UINT win_width, win_height;
-
- GetClientRect(hWnd, &myRect);
-
- /* grab the progress bar rect size */
- win_width = myRect.right - myRect.left;
- win_height = myRect.bottom - myRect.top;
-
- /* create an off-screen DC for double-buffering */
- hdcMem = CreateCompatibleDC(hDC);
- hbmMem = CreateCompatibleBitmap(hDC, win_width, win_height);
-
- hOld = SelectObject(hdcMem, hbmMem);
-
- /* call the original draw code and redirect it to our memory buffer */
- DefSubclassProc(hWnd, uMsg, (WPARAM) hdcMem, lParam);
-
- /* draw our nifty progress text over it */
- SelectFont(hdcMem, GetStockFont(DEFAULT_GUI_FONT));
- DrawShadowText(hdcMem, szProgressText.GetString(), szProgressText.GetLength(),
- &myRect,
- DT_CENTER | DT_VCENTER | DT_NOPREFIX | DT_SINGLELINE,
- GetSysColor(COLOR_CAPTIONTEXT),
- GetSysColor(COLOR_3DSHADOW),
- 1, 1);
-
- /* transfer the off-screen DC to the screen */
- BitBlt(hDC, 0, 0, win_width, win_height, hdcMem, 0, 0, SRCCOPY);
-
- /* free the off-screen DC */
- SelectObject(hdcMem, hOld);
- DeleteObject(hbmMem);
- DeleteDC(hdcMem);
-
- EndPaint(hWnd, &ps);
- return 0;
- }
-
- /* 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();
- RemoveWindowSubclass(hWnd, DownloadProgressProc, uIdSubclass);
- }
- /* Fall-through */
- default:
- return DefSubclassProc(hWnd, uMsg, wParam, lParam);
- }
-}
-
-DWORD WINAPI CDownloadManager::ThreadFunc(LPVOID param)
-{
- CComPtr<IBindStatusCallback> dl;
- ATL::CStringW Path;
- PWSTR p, q;
-
- 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;
-
- 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;
- }
-
- 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;
-
- // 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);
-
- // 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;
- }
-
- // 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 && 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;
- }
-
- // Reset progress bar
- Item = GetDlgItem(hDlg, IDC_DOWNLOAD_PROGRESS);
- if (Item)
- {
- SendMessageW(Item, PBM_SETPOS, 0, 0);
- }
-
- // 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);
- }
-
- SetWindowTextW(hDlg, szNewCaption.GetString());
-
- // Add the download URL
- SetDlgItemTextW(hDlg, IDC_DOWNLOAD_STATUS, InfoArray[iAppId].szUrl.GetString());
-
- DownloadsListView.SetDownloadStatus(iAppId, DLSTATUS_DOWNLOADING);
-
- // download it
- bTempfile = TRUE;
- CDownloadDialog_Constructor(hDlg, &bCancelled, IID_PPV_ARG(IBindStatusCallback, &dl));
-
- 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;
- }
-
- if (!hOpen)
- goto end;
-
- hFile = InternetOpenUrlW(hOpen, InfoArray[iAppId].szUrl.GetString(), NULL, 0, INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_KEEP_CONNECTION, 0);
-
- if (!hFile)
- {
- MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD2);
- goto end;
- }
-
- if (!HttpQueryInfoW(hFile, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, &dwStatus, &dwStatusLen, NULL))
- goto end;
-
- if (dwStatus != HTTP_STATUS_OK)
- {
- MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD);
- goto end;
- }
-
- dwStatusLen = sizeof(dwStatus);
-
- memset(&urlComponents, 0, sizeof(urlComponents));
- urlComponents.dwStructSize = sizeof(urlComponents);
-
- 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));
-
- if (!InternetCrackUrlW(InfoArray[iAppId].szUrl, urlLength + 1, ICU_DECODE | ICU_ESCAPE, &urlComponents))
- 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);
-
- 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_CERT_DOES_NOT_MATCH);
- goto end;
- }
-#endif
-
- 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 = 0;
- do
- {
- if (!InternetReadFile(hFile, lpBuffer, _countof(lpBuffer), &dwBytesRead))
- {
- MessageBox_LoadString(hMainWnd, IDS_INTERRUPTED_DOWNLOAD);
- goto end;
- }
-
- if (!WriteFile(hOut, &lpBuffer[0], dwBytesRead, &dwBytesWritten, NULL))
- {
- MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_WRITE);
- goto end;
- }
-
- dwCurrentBytesRead += dwBytesRead;
- dl->OnProgress(dwCurrentBytesRead, dwContentLen, 0, InfoArray[iAppId].szUrl.GetString());
- } while (dwBytesRead && !bCancelled);
-
- CloseHandle(hOut);
- hOut = INVALID_HANDLE_VALUE;
-
- if (bCancelled)
- goto end;
-
- /* 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)
- {
- 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;
-
- SetWindowTextW(hDlg, szMsgText.GetString());
- SendMessageW(GetDlgItem(hDlg, IDC_DOWNLOAD_STATUS), WM_SETTEXT, 0, (LPARAM) Path.GetString());
-
- // 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:
- 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);
-
- InternetCloseHandle(hFile);
- InternetCloseHandle(hOpen);
-
- if (bTempfile)
- {
- if (bCancelled || (SettingsInfo.bDelInstaller && !bCab))
- DeleteFileW(Path.GetString());
- }
-
- DownloadsListView.SetDownloadStatus(iAppId, DLSTATUS_FINISHED);
- }
-
- delete static_cast<DownloadParam*>(param);
- SendMessageW(hDlg, WM_CLOSE, 0, 0);
- return 0;
-}
-
-BOOL CDownloadManager::DownloadListOfApplications(const ATL::CSimpleArray<CAvailableApplicationInfo*>& AppsList, BOOL bIsModal)
-{
- if (AppsList.GetSize() == 0)
- {
- return FALSE;
- }
-
- // Initialize shared variables
- for (INT i = 0; i < AppsList.GetSize(); ++i)
- {
- if (AppsList[i])
- {
- AppsToInstallList.Add(*(AppsList[i]));
- }
- }
-
- // Create a dialog and issue a download process
- LaunchDownloadDialog(bIsModal);
-
- return TRUE;
-}
-
-BOOL CDownloadManager::DownloadApplication(CAvailableApplicationInfo* pAppInfo, BOOL bIsModal)
-{
- if (!pAppInfo)
- return FALSE;
-
- Download(*pAppInfo, bIsModal);
- return TRUE;
-}
-
-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 CDownloadManager::LaunchDownloadDialog(BOOL bIsModal)
-{
- if (bIsModal)
- {
- DialogBoxW(hInst,
- MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG),
- hMainWnd,
- DownloadDlgProc);
- }
- else
- {
- CreateDialogW(hInst,
- MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG),
- hMainWnd,
- DownloadDlgProc);
- }
-}
-// CDownloadManager