CDownloadManager Class Reference

Collaboration diagram for CDownloadManager:

Static Public Member Functions
static VOID	Add (DownloadInfo info)
static VOID	Download (const DownloadInfo &DLInfo, BOOL bIsModal=FALSE)
static INT_PTR CALLBACK	DownloadDlgProc (HWND Dlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
static unsigned int WINAPI	ThreadFunc (LPVOID Context)
static VOID	LaunchDownloadDialog (BOOL)

Static Private Member Functions
static VOID	UpdateProgress (HWND hDlg, ULONG ulProgress, ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)

Static Private Attributes
static ATL::CSimpleArray< DownloadInfo >	AppsDownloadList
static CDowloadingAppsListView	DownloadsListView
static CDownloaderProgress	ProgressBar
static BOOL	bCancelled = FALSE
static BOOL	bModal = FALSE

Detailed Description

Definition at line 359 of file loaddlg.cpp.

Member Function Documentation

Add()

VOID CDownloadManager::Add ( DownloadInfo  info )

static

Definition at line 383 of file loaddlg.cpp.

{
    AppsDownloadList.Add(info);
}

Referenced by DownloadListOfApplications().

Download()

VOID CDownloadManager::Download ( const DownloadInfo &  DLInfo, BOOL  bIsModal = FALSE  )

static

Definition at line 388 of file loaddlg.cpp.

{
    AppsDownloadList.RemoveAll();
    AppsDownloadList.Add(DLInfo);
    LaunchDownloadDialog(bIsModal);
}

Referenced by DownloadApplication(), and DownloadApplicationsDB().

DownloadDlgProc()

INT_PTR CALLBACK CDownloadManager::DownloadDlgProc ( HWND  Dlg, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 395 of file loaddlg.cpp.

{
    static WCHAR szCaption[MAX_PATH];

    switch (uMsg)
    {
    case WM_INITDIALOG:
    {
        HICON hIconSm, hIconBg;
        ATL::CStringW szTempCaption;

        bCancelled = FALSE;

        hIconBg = (HICON) GetClassLongPtrW(hMainWnd, GCLP_HICON);
        hIconSm = (HICON) GetClassLongPtrW(hMainWnd, GCLP_HICONSM);

        if (hIconBg && hIconSm)
        {
            SendMessageW(Dlg, WM_SETICON, ICON_BIG, (LPARAM) hIconBg);
            SendMessageW(Dlg, WM_SETICON, ICON_SMALL, (LPARAM) hIconSm);
        }

        SetWindowLongPtrW(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
            ProgressBar.SubclassWindow(Item);
            ProgressBar.SendMessage(PBM_SETRANGE, 0, MAKELPARAM(0, 100));
            ProgressBar.SendMessage(PBM_SETPOS, 0, 0);
        }

        // Add a ListView
        HWND hListView = DownloadsListView.Create(Dlg);
        if (!hListView)
        {
            return FALSE;
        }
        DownloadsListView.LoadList(AppsDownloadList);

        // Get a dlg string for later use
        GetWindowTextW(Dlg, szCaption, _countof(szCaption));

        // Hide a placeholder from displaying
        szTempCaption = szCaption;
        szTempCaption.Replace(L"%ls", L"");
        SetWindowText(Dlg, szTempCaption.GetString());

        ShowWindow(Dlg, SW_SHOW);

        // Start download process
        DownloadParam *param = new DownloadParam(Dlg, AppsDownloadList, szCaption);
        unsigned int ThreadId;
        HANDLE Thread = (HANDLE)_beginthreadex(NULL, 0, ThreadFunc, (void *) param, 0, &ThreadId);

        if (!Thread)
        {
            return FALSE;
        }

        CloseHandle(Thread);
        AppsDownloadList.RemoveAll();
        return TRUE;
    }

    case WM_COMMAND:
        if (wParam == IDCANCEL)
        {
            bCancelled = TRUE;
            PostMessageW(Dlg, WM_CLOSE, 0, 0);
        }
        return FALSE;

    case WM_CLOSE:
        if (CDownloadManager::bModal)
        {
            ::EndDialog(Dlg, 0);
        }
        else
        {
            ::DestroyWindow(Dlg);
        }
        return TRUE;

    default:
        return FALSE;
    }
}

Referenced by LaunchDownloadDialog().

LaunchDownloadDialog()

VOID CDownloadManager::LaunchDownloadDialog ( BOOL  bIsModal )

static

Definition at line 990 of file loaddlg.cpp.

{
    CDownloadManager::bModal = bIsModal;
    if (bIsModal)
    {
        DialogBoxW(hInst,
                   MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG),
                   hMainWnd,
                   DownloadDlgProc);
    }
    else
    {
        CreateDialogW(hInst,
                      MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG),
                      hMainWnd,
                      DownloadDlgProc);
    }
}

Referenced by Download(), and DownloadListOfApplications().

ThreadFunc()

unsigned int WINAPI CDownloadManager::ThreadFunc ( LPVOID  Context )

static

Definition at line 549 of file loaddlg.cpp.

{
    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 bTempfile = FALSE;

    HINTERNET hOpen = NULL;
    HINTERNET hFile = NULL;
    HANDLE hOut = INVALID_HANDLE_VALUE;

    unsigned char lpBuffer[4096];
    LPCWSTR lpszAgent = L"RApps/1.1";
    URL_COMPONENTSW 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;

    const DWORD dwUrlConnectFlags = INTERNET_FLAG_DONT_CACHE | INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_KEEP_CONNECTION;

    if (InfoArray.GetSize() <= 0)
    {
        MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD);
        goto end;
    }

    for (iAppId = 0; iAppId < InfoArray.GetSize(); ++iAppId)
    {
        // Reset progress bar
        if (!IsWindow(hDlg)) break;
        Item = GetDlgItem(hDlg, IDC_DOWNLOAD_PROGRESS);
        if (Item)
        {
            ProgressBar.SetMarquee(FALSE);
            ProgressBar.SendMessage(WM_SETTEXT, 0, (LPARAM) L"");
            ProgressBar.SendMessage(PBM_SETPOS, 0, 0);
        }

        // is this URL an update package for RAPPS? if so store it in a different place
        if (InfoArray[iAppId].DLType != DLTYPE_APPLICATION)
        {
            if (!GetStorageDirectory(Path))
            {
                ShowLastError(hMainWnd, FALSE, GetLastError());
                goto end;
            }
        }
        else
        {
            Path = SettingsInfo.szDownloadDir;
        }

        // Change caption to show the currently downloaded app
        switch(InfoArray[iAppId].DLType)
        {
        case DLTYPE_APPLICATION:
            szNewCaption.Format(szCaption, InfoArray[iAppId].szName.GetString());
            break;
        case DLTYPE_DBUPDATE:
            szNewCaption.LoadStringW(IDS_DL_DIALOG_DB_DOWNLOAD_DISP);
            break;
        case DLTYPE_DBUPDATE_UNOFFICIAL:
            szNewCaption.LoadStringW(IDS_DL_DIALOG_DB_UNOFFICIAL_DOWNLOAD_DISP);
            break;
        }
        
        if (!IsWindow(hDlg)) goto end;
        SetWindowTextW(hDlg, szNewCaption.GetString());

        // 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)
        {
            MessageBox_LoadString(hMainWnd, IDS_UNABLE_PATH);
            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 the path valid? can we access it?
        if (GetFileAttributesW(Path.GetString()) == INVALID_FILE_ATTRIBUTES)
        {
            if (!CreateDirectoryW(Path.GetString(), NULL))
            {
                ShowLastError(hMainWnd, FALSE, GetLastError());
                goto end;
            }
        }

        // append a \ to the provided file system path, and the filename portion from the URL after that

        PathAddBackslashW(Path.GetBuffer(MAX_PATH));
        switch (InfoArray[iAppId].DLType)
        {
        case DLTYPE_DBUPDATE:
        case DLTYPE_DBUPDATE_UNOFFICIAL:
            PathAppendW(Path.GetBuffer(), APPLICATION_DATABASE_NAME);
            break;
        case DLTYPE_APPLICATION:
            PathAppendW(Path.GetBuffer(), (LPWSTR)(p + 1)); // use the filename retrieved from URL
            break;
        }
        Path.ReleaseBuffer();

        if ((InfoArray[iAppId].DLType == DLTYPE_APPLICATION) && 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;
        }

        // Add the download URL
        if (!IsWindow(hDlg)) goto end;
        SetDlgItemTextW(hDlg, IDC_DOWNLOAD_STATUS, InfoArray[iAppId].szUrl.GetString());

        DownloadsListView.SetDownloadStatus(iAppId, DLSTATUS_DOWNLOADING);

        // download it
        UrlHasBeenCopied = FALSE;
        bTempfile = TRUE;

        /* FIXME: this should just be using the system-wide proxy settings */
        switch (SettingsInfo.Proxy)
        {
        case 0: // preconfig
        default:
            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;
        }

        if (!hOpen)
        {
            ShowLastError(hMainWnd, TRUE, GetLastError());
            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));

        if (!InternetCrackUrlW(InfoArray[iAppId].szUrl, urlLength + 1, ICU_DECODE | ICU_ESCAPE, &urlComponents))
        {
            ShowLastError(hMainWnd, TRUE, GetLastError());
            goto end;
        }

        dwContentLen = 0;

        if (urlComponents.nScheme == INTERNET_SCHEME_HTTP ||
            urlComponents.nScheme == INTERNET_SCHEME_HTTPS)
        {
            hFile = InternetOpenUrlW(hOpen, InfoArray[iAppId].szUrl.GetString(), NULL, 0,
                                     dwUrlConnectFlags,
                                     0);
            if (!hFile)
            {
                if (!ShowLastError(hMainWnd, TRUE, GetLastError()))
                {
                    /* Workaround for CORE-17377 */
                    MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD2);
                }
                goto end;
            }

            // query connection
            if (!HttpQueryInfoW(hFile, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, &dwStatus, &dwStatusLen, NULL))
            {
                ShowLastError(hMainWnd, TRUE, GetLastError());
                goto end;
            }

            if (dwStatus != HTTP_STATUS_OK)
            {
                MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD);
                goto end;
            }

            // query content length
            HttpQueryInfoW(hFile, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, &dwContentLen, &dwStatusLen, NULL);
        }
        else if (urlComponents.nScheme == INTERNET_SCHEME_FTP)
        {
            // force passive mode on FTP
            hFile = InternetOpenUrlW(hOpen, InfoArray[iAppId].szUrl, NULL, 0,
                                     dwUrlConnectFlags | INTERNET_FLAG_PASSIVE,
                                     0);
            if (!hFile)
            {
                if (!ShowLastError(hMainWnd, TRUE, GetLastError()))
                {
                    /* Workaround for CORE-17377 */
                    MessageBox_LoadString(hMainWnd, IDS_UNABLE_TO_DOWNLOAD2);
                }
                goto end;
            }

            dwContentLen = FtpGetFileSize(hFile, &dwStatus);
        }
        else if (urlComponents.nScheme == INTERNET_SCHEME_FILE)
        {
            // Add support for the file scheme so testing locally is simpler
            WCHAR LocalFilePath[MAX_PATH];
            DWORD cchPath = _countof(LocalFilePath);
            // Ideally we would use PathCreateFromUrlAlloc here, but that is not exported (yet)
            HRESULT hr = PathCreateFromUrlW(InfoArray[iAppId].szUrl, LocalFilePath, &cchPath, 0);
            if (SUCCEEDED(hr))
            {
                if (CopyFileW(LocalFilePath, Path, FALSE))
                {
                    goto run;
                }
                else
                {
                    ShowLastError(hMainWnd, FALSE, GetLastError());
                    goto end;
                }
            }
            else
            {
                ShowLastError(hMainWnd, FALSE, hr);
                goto end;
            }
        }

        if (!dwContentLen)
        {
            // Someone was nice enough to add this, let's use it
            if (InfoArray[iAppId].SizeInBytes)
            {
                dwContentLen = InfoArray[iAppId].SizeInBytes;
            }
            else
            {
                // content-length is not known, enable marquee mode
                ProgressBar.SetMarquee(TRUE);
            }
        }

        free(urlComponents.lpszScheme);

#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) &&
            (InfoArray[iAppId].DLType == DLTYPE_DBUPDATE))
        {
            CLocalPtr<char> subjectName, issuerName;
            CStringA szMsgText;
            bool bAskQuestion = false;
            if (!CertGetSubjectAndIssuer(hFile, subjectName, issuerName))
            {
                szMsgText.LoadStringW(IDS_UNABLE_TO_QUERY_CERT);
                bAskQuestion = true;
            }
            else
            {
                if (strcmp(subjectName, CERT_SUBJECT_INFO) ||
                    strcmp(issuerName, CERT_ISSUER_INFO))
                {
                    szMsgText.Format(IDS_MISMATCH_CERT_INFO, (char*)subjectName, (const char*)issuerName);
                    bAskQuestion = true;
                }
            }

            if (bAskQuestion)
            {
                if (MessageBoxA(hMainWnd, szMsgText.GetString(), NULL, MB_YESNO | MB_ICONERROR) != IDYES)
                {
                    goto end;
                }
            }
        }
#endif

        hOut = CreateFileW(Path.GetString(), GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, 0, NULL);

        if (hOut == INVALID_HANDLE_VALUE)
        {
            ShowLastError(hMainWnd, FALSE, GetLastError());
            goto end;
        }

        dwCurrentBytesRead = 0;
        do
        {
            if (!InternetReadFile(hFile, lpBuffer, _countof(lpBuffer), &dwBytesRead))
            {
                ShowLastError(hMainWnd, TRUE, GetLastError());
                goto end;
            }

            if (!WriteFile(hOut, &lpBuffer[0], dwBytesRead, &dwBytesWritten, NULL))
            {
                ShowLastError(hMainWnd, FALSE, GetLastError());
                goto end;
            }

            dwCurrentBytesRead += dwBytesRead;
            if (!IsWindow(hDlg)) goto end;
            UpdateProgress(hDlg, dwCurrentBytesRead, dwContentLen, 0, InfoArray[iAppId].szUrl.GetString());
        } while (dwBytesRead && !bCancelled);

        CloseHandle(hOut);
        hOut = INVALID_HANDLE_VALUE;

        if (bCancelled)
        {
            DPRINT1("Operation cancelled\n");
            goto end;
        }

        if (!dwContentLen)
        {
            // set progress bar to 100%
            ProgressBar.SetMarquee(FALSE);

            dwContentLen = dwCurrentBytesRead;
            if (!IsWindow(hDlg)) goto end;
            UpdateProgress(hDlg, dwCurrentBytesRead, dwContentLen, 0, InfoArray[iAppId].szUrl.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 ((InfoArray[iAppId].DLType == DLTYPE_APPLICATION) && 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))
            {
                DPRINT1("Unable to load string\n");
                goto end;
            }

            if (!IsWindow(hDlg)) 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))
                {
                    DPRINT1("Unable to load string\n");
                    goto end;
                }

                if (!IsWindow(hDlg)) goto end;
                MessageBoxW(hDlg, szMsgText.GetString(), NULL, MB_OK | MB_ICONERROR);
                goto end;
            }
        }

run:
        DownloadsListView.SetDownloadStatus(iAppId, DLSTATUS_WAITING_INSTALL);

        // run it
        if (InfoArray[iAppId].DLType == DLTYPE_APPLICATION)
        {
            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))
            {
                //reflect installation progress in the titlebar
                //TODO: make a separate string with a placeholder to include app name?
                ATL::CStringW szMsgText = LoadStatusString(DLSTATUS_INSTALLING);
                if (!IsWindow(hDlg)) goto end;
                SetWindowTextW(hDlg, szMsgText.GetString());

                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
            {
                ShowLastError(hMainWnd, FALSE, GetLastError());
            }
        }

end:
        if (hOut != INVALID_HANDLE_VALUE)
            CloseHandle(hOut);

        if (hFile)
            InternetCloseHandle(hFile);
        InternetCloseHandle(hOpen);

        if (bTempfile)
        {
            if (bCancelled || (SettingsInfo.bDelInstaller && (InfoArray[iAppId].DLType == DLTYPE_APPLICATION)))
                DeleteFileW(Path.GetString());
        }

        if (!IsWindow(hDlg)) return 0;
        DownloadsListView.SetDownloadStatus(iAppId, DLSTATUS_FINISHED);
    }

    delete static_cast<DownloadParam*>(param);
    if (!IsWindow(hDlg)) return 0;
    SendMessageW(hDlg, WM_CLOSE, 0, 0);
    return 0;
}

Referenced by DownloadDlgProc().

UpdateProgress()

VOID CDownloadManager::UpdateProgress ( HWND  hDlg, ULONG  ulProgress, ULONG  ulProgressMax, ULONG  ulStatusCode, LPCWSTR  szStatusText  )

staticprivate

Definition at line 487 of file loaddlg.cpp.

{
    HWND Item;

    if (!IsWindow(hDlg)) return;
    ProgressBar.SetProgress(ulProgress, ulProgressMax);

    if (!IsWindow(hDlg)) return;
    Item = GetDlgItem(hDlg, IDC_DOWNLOAD_STATUS);
    if (Item && szStatusText && wcslen(szStatusText) > 0 && UrlHasBeenCopied == FALSE)
    {
        SIZE_T len = wcslen(szStatusText) + 1;
        ATL::CStringW buf;
        DWORD dummyLen;

        /* beautify our url for display purposes */
        if (!InternetCanonicalizeUrlW(szStatusText, buf.GetBuffer(len), &dummyLen, 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());
        UrlHasBeenCopied = TRUE;
    }
}

Referenced by ThreadFunc().

Member Data Documentation

AppsDownloadList

ATL::CSimpleArray< DownloadInfo > CDownloadManager::AppsDownloadList

staticprivate

Definition at line 361 of file loaddlg.cpp.

Referenced by Add(), Download(), and DownloadDlgProc().

bCancelled

BOOL CDownloadManager::bCancelled = FALSE

staticprivate

Definition at line 364 of file loaddlg.cpp.

Referenced by DownloadDlgProc(), and ThreadFunc().

bModal

BOOL CDownloadManager::bModal = FALSE

staticprivate

Definition at line 365 of file loaddlg.cpp.

Referenced by DownloadDlgProc(), and LaunchDownloadDialog().

DownloadsListView

CDowloadingAppsListView CDownloadManager::DownloadsListView

staticprivate

Definition at line 362 of file loaddlg.cpp.

Referenced by DownloadDlgProc(), and ThreadFunc().

ProgressBar

CDownloaderProgress CDownloadManager::ProgressBar

staticprivate

Definition at line 363 of file loaddlg.cpp.

Referenced by DownloadDlgProc(), ThreadFunc(), and UpdateProgress().

---

The documentation for this class was generated from the following file:

• base/applications/rapps/loaddlg.cpp