CDownloadManager Class Reference

Inheritance diagram for CDownloadManager:

Collaboration diagram for CDownloadManager:

Public Types
enum	{ WM_ISCANCELLED = WM_APP , WM_SETSTATUS , WM_GETINSTANCE , WM_GETNEXT }
Public Types inherited from IUnknown
typedef IUnknown *	LPUNKNOWN

Public Member Functions
	CDownloadManager ()
BOOL	IsCancelled ()
BOOL	IsSilentDialog ()
void	StartWorkerThread ()
void	Add (const DownloadInfo &Info)
void	Show ()
INT_PTR	RealDlgProc (HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
void	UpdateProgress (ULONG ulProgress, ULONG ulProgressMax)
void	PerformDownloadAndInstall (const DownloadInfo &Info)
Public Member Functions inherited from ATL::CComObjectRootEx< CComMultiThreadModelNoCS >
	~CComObjectRootEx ()
ULONG	InternalAddRef ()
ULONG	InternalRelease ()
void	Lock ()
void	Unlock ()
HRESULT	_AtlInitialConstruct ()
Public Member Functions inherited from ATL::CComObjectRootBase
	CComObjectRootBase ()
	~CComObjectRootBase ()
void	SetVoid (void *)
HRESULT	_AtlFinalConstruct ()
HRESULT	FinalConstruct ()
void	InternalFinalConstructAddRef ()
void	InternalFinalConstructRelease ()
void	FinalRelease ()
Public Member Functions inherited from IUnknown
HRESULT	QueryInterface ([in] REFIID riid, [out, iid_is(riid)] void **ppvObject)
ULONG	AddRef ()
ULONG	Release ()

Static Public Member Functions
static CDownloadManager *	CreateInstanceHelper (UINT Flags)
static BOOL	CreateInstance (BOOL Modal, CComPtr< CDownloadManager > &Obj)
static CDownloadManager *	FindInstance ()
static INT_PTR CALLBACK	DlgProc (HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
static unsigned int CALLBACK	ThreadFunc (void *ThreadParam)
Static Public Member Functions inherited from ATL::CComCoClass< CDownloadManager, &CLSID_NULL >
static LPCTSTR WINAPI	GetObjectDescription ()
Static Public Member Functions inherited from ATL::CComObjectRootBase
static void WINAPI	ObjectMain (bool)
static const struct _ATL_CATMAP_ENTRY *	GetCategoryMap ()
static HRESULT WINAPI	InternalQueryInterface (void *pThis, const _ATL_INTMAP_ENTRY *pEntries, REFIID iid, void **ppvObject)

Protected Attributes
HWND	m_hDlg
UINT	m_Threads
UINT	m_Index
BOOL	m_bCancelled
BOOL	m_bModal
UINT	m_fDaf = 0
WCHAR	m_szCaptionFmt [100]
ATL::CSimpleArray< DownloadInfo >	m_List
CDowloadingAppsListView	m_ListView
CDownloaderProgress	m_ProgressBar

Additional Inherited Members
Public Attributes inherited from ATL::CComObjectRootBase
LONG	m_dwRef

Detailed Description

Definition at line 469 of file loaddlg.cpp.

Member Enumeration Documentation

anonymous enum

anonymous enum

Enumerator
WM_ISCANCELLED
WM_SETSTATUS
WM_GETINSTANCE
WM_GETNEXT

Definition at line 475 of file loaddlg.cpp.

         {
        WM_ISCANCELLED = WM_APP, // Return BOOL
        WM_SETSTATUS, // wParam DownloadStatus
        WM_GETINSTANCE, // Return CDownloadManager*
        WM_GETNEXT, // Return DownloadInfo* or NULL
    };

Constructor & Destructor Documentation

CDownloadManager()

CDownloadManager::CDownloadManager ( )

inline

Definition at line 482 of file loaddlg.cpp.

: m_hDlg(NULL), m_Threads(0), m_Index(0), m_bCancelled(FALSE) {}

Member Function Documentation

Add()

void CDownloadManager::Add

(

const DownloadInfo &

Info

)

Definition at line 574 of file loaddlg.cpp.

{
    const UINT count = m_List.GetSize(), start = count;
    for (UINT i = 0; i < count; ++i)
    {
        if (Info.Equal(m_List[i]))
            return; // Already in the list
    }
    m_List.Add(Info);
    if (m_hDlg)
        m_ListView.LoadList(m_List, start);
}

CreateInstance()

static BOOL CDownloadManager::CreateInstance ( BOOL  Modal, CComPtr< CDownloadManager > &  Obj  )

inlinestatic

Definition at line 505 of file loaddlg.cpp.

    {
        CDownloadManager *p = CreateInstanceHelper(Modal);
        if (!p)
            return FALSE;
        Obj.Attach(p);
        return TRUE;
    }

Referenced by DownloadApplicationsDB(), and DownloadListOfApplications().

CreateInstanceHelper()

static CDownloadManager * CDownloadManager::CreateInstanceHelper ( UINT  Flags )

inlinestatic

Definition at line 485 of file loaddlg.cpp.

    {
        if (!(Flags & DAF_MODAL))
        {
            CDownloadManager* pExisting = CDownloadManager::FindInstance();
            if (pExisting)
            {
                pExisting->AddRef();
                return pExisting;
            }
        }
        CComPtr<CDownloadManager> obj;
        if (FAILED(ShellObjectCreator(obj)))
            return NULL;
        obj->m_fDaf = Flags;
        obj->m_bModal = !!(Flags & DAF_MODAL);
        return obj.Detach();
    }

Referenced by CreateInstance().

DlgProc()

INT_PTR CALLBACK CDownloadManager::DlgProc ( HWND  hDlg, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 604 of file loaddlg.cpp.

{
    CDownloadManager* pThis = (CDownloadManager*)GetWindowLongPtrW(hDlg, DWLP_USER);
    if (!pThis)
    {
        if (uMsg != WM_INITDIALOG)
            return FALSE;
        SetWindowLongPtrW(hDlg, DWLP_USER, lParam);
        pThis = (CDownloadManager*)lParam;
    }
    return pThis->RealDlgProc(hDlg, uMsg, wParam, lParam);
}

FindInstance()

static CDownloadManager * CDownloadManager::FindInstance ( )

inlinestatic

Definition at line 515 of file loaddlg.cpp.

    {
        if (g_hDownloadWnd && IsWindowVisible(g_hDownloadWnd))
            return (CDownloadManager*)SendMessageW(g_hDownloadWnd, WM_GETINSTANCE, 0, 0);
        return NULL;
    }

Referenced by CreateInstanceHelper().

IsCancelled()

BOOL CDownloadManager::IsCancelled ( )

inline

Definition at line 523 of file loaddlg.cpp.

    {
        return !IsWindow(m_hDlg) || SendMessageW(m_hDlg, WM_ISCANCELLED, 0, 0);
    }

Referenced by PerformDownloadAndInstall().

IsSilentDialog()

BOOL CDownloadManager::IsSilentDialog ( )

inline

Definition at line 529 of file loaddlg.cpp.

    {
        return m_fDaf & DAF_SILENT;
    }

Referenced by RealDlgProc(), and Show().

PerformDownloadAndInstall()

void CDownloadManager::PerformDownloadAndInstall

(

const DownloadInfo &

Info

)

Definition at line 739 of file loaddlg.cpp.

{
    const HWND hDlg = m_hDlg;
    const HWND hStatus = GetDlgItem(m_hDlg, IDC_DOWNLOAD_STATUS);
    SetFriendlyUrl(hStatus, Info.szUrl);
 
    m_ProgressBar.SetMarquee(FALSE);
    m_ProgressBar.SendMessageW(PBM_SETPOS, 0, 0);
    m_ProgressBar.SetProgress(0, Info.SizeInBytes);
 
    CStringW str;
    CPathW Path;
    PCWSTR p;
 
    ULONG dwContentLen, dwBytesWritten, dwBytesRead, dwStatus, dwStatusLen;
    ULONG dwCurrentBytesRead = 0;
    BOOL bTempfile = FALSE, bCancelled = FALSE;
 
    HINTERNET hOpen = NULL;
    HINTERNET hFile = NULL;
    HANDLE hOut = INVALID_HANDLE_VALUE;
 
    
    LPCWSTR lpszAgent = L"RApps/1.1";
    const DWORD dwUrlConnectFlags =
        INTERNET_FLAG_DONT_CACHE | INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_KEEP_CONNECTION;
    URL_COMPONENTSW urlComponents;
    size_t urlLength;
    unsigned char lpBuffer[4096];
 
    // Change caption to show the currently downloaded app
    switch (Info.DLType)
    {
        case DLTYPE_APPLICATION:
            str.Format(m_szCaptionFmt, Info.szName.GetString());
            break;
        case DLTYPE_DBUPDATE:
            str.LoadStringW(IDS_DL_DIALOG_DB_DOWNLOAD_DISP);
            break;
        case DLTYPE_DBUPDATE_UNOFFICIAL:
            str.LoadStringW(IDS_DL_DIALOG_DB_UNOFFICIAL_DOWNLOAD_DISP);
            break;
    }
    SetWindowTextW(hDlg, str);
 
    // is this URL an update package for RAPPS? if so store it in a different place
    if (Info.DLType != DLTYPE_APPLICATION)
    {
        if (!GetStorageDirectory(Path))
        {
            ShowLastError(hMainWnd, FALSE, GetLastError());
            goto end;
        }
    }
    else
    {
        Path = SettingsInfo.szDownloadDir;
    }
 
    // build the path for the download
    p = wcsrchr(Info.szUrl.GetString(), L'/');
 
    // do we have a final slash separator?
    if (!p)
    {
        MessageBox_LoadString(hMainWnd, IDS_UNABLE_PATH);
        goto end;
    }
 
    // is the path valid? can we access it?
    if (GetFileAttributesW(Path) == INVALID_FILE_ATTRIBUTES)
    {
        if (!CreateDirectoryW(Path, NULL))
        {
            ShowLastError(hMainWnd, FALSE, GetLastError());
            goto end;
        }
    }
 
    switch (Info.DLType)
    {
        case DLTYPE_DBUPDATE:
        case DLTYPE_DBUPDATE_UNOFFICIAL:
            Path += APPLICATION_DATABASE_NAME;
            break;
        case DLTYPE_APPLICATION:
        {
            CStringW name = Info.szFileName;
            if (name.IsEmpty())
                name = p + 1; // use the filename retrieved from URL
            UrlUnescapeAndMakeFileNameValid(name);
            Path += name;
            break;
        }
    }
 
    if ((Info.DLType == DLTYPE_APPLICATION) && Info.szSHA1[0] &&
        GetFileAttributesW(Path) != INVALID_FILE_ATTRIBUTES)
    {
        // only open it in case of total correctness
        if (VerifyInteg(Info.szSHA1.GetString(), Path))
            goto run;
    }
 
    // Download it
    SendMessageW(hDlg, WM_SETSTATUS, DLSTATUS_DOWNLOADING, 0);
    /* 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;
    }
 
    bTempfile = TRUE;
    dwContentLen = 0;
    dwStatusLen = sizeof(dwStatus);
    ZeroMemory(&urlComponents, sizeof(urlComponents));
    urlComponents.dwStructSize = sizeof(urlComponents);
 
    urlLength = Info.szUrl.GetLength();
    urlComponents.dwSchemeLength = urlLength + 1;
    urlComponents.lpszScheme = (LPWSTR)malloc(urlComponents.dwSchemeLength * sizeof(WCHAR));
 
    if (!InternetCrackUrlW(Info.szUrl, urlLength + 1, ICU_DECODE | ICU_ESCAPE, &urlComponents))
    {
        ShowLastError(hMainWnd, TRUE, GetLastError());
        goto end;
    }
 
    if (urlComponents.nScheme == INTERNET_SCHEME_HTTP || urlComponents.nScheme == INTERNET_SCHEME_HTTPS)
    {
        hFile = InternetOpenUrlW(hOpen, Info.szUrl, 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, Info.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(Info.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 (Info.SizeInBytes)
        {
            dwContentLen = Info.SizeInBytes;
        }
        else
        {
            // content-length is not known, enable marquee mode
            m_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) && (Info.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 (!IsTrustedPinnedCert(subjectName, issuerName))
        {
            szMsgText.Format(IDS_MISMATCH_CERT_INFO, (LPCSTR)subjectName, (LPCSTR)issuerName);
            bAskQuestion = true;
        }
 
        if (bAskQuestion)
        {
            if (MessageBoxA(hDlg, szMsgText, NULL, MB_YESNO | MB_ICONERROR) != IDYES)
            {
                goto end;
            }
        }
    }
#endif
 
    hOut = CreateFileW(Path, GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, 0, NULL);
    if (hOut == INVALID_HANDLE_VALUE)
    {
        ShowLastError(hDlg, FALSE, GetLastError());
        goto end;
    }
 
    dwCurrentBytesRead = 0;
    do
    {
        bCancelled = IsCancelled();
        if (bCancelled)
            break;
 
        if (!InternetReadFile(hFile, lpBuffer, _countof(lpBuffer), &dwBytesRead))
        {
            ShowLastError(hDlg, TRUE, GetLastError());
            goto end;
        }
 
        if (!WriteFile(hOut, &lpBuffer[0], dwBytesRead, &dwBytesWritten, NULL))
        {
            ShowLastError(hDlg, FALSE, GetLastError());
            goto end;
        }
 
        dwCurrentBytesRead += dwBytesRead;
        UpdateProgress(dwCurrentBytesRead, dwContentLen);
        
    } while (dwBytesRead);
 
    CloseHandle(hOut);
    hOut = INVALID_HANDLE_VALUE;
 
    if (bCancelled)
    {
        DPRINT1("Operation cancelled\n");
        goto end;
    }
 
    if (!dwContentLen)
    {
        // set progress bar to 100%
        m_ProgressBar.SetMarquee(FALSE);
 
        dwContentLen = dwCurrentBytesRead;
        UpdateProgress(dwCurrentBytesRead, dwContentLen);
    }
 
    /* 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 ((Info.DLType == DLTYPE_APPLICATION) && Info.szSHA1[0] != 0)
    {
        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;
        }
 
        SetWindowTextW(hDlg, szMsgText);
        SetWindowTextW(hStatus, Path);
 
        // this may take a while, depending on the file size
        if (!VerifyInteg(Info.szSHA1, Path))
        {
            if (!szMsgText.LoadStringW(IDS_INTEG_CHECK_FAIL))
            {
                DPRINT1("Unable to load string\n");
                goto end;
            }
 
            MessageBoxW(hDlg, szMsgText, NULL, MB_OK | MB_ICONERROR);
            goto end;
        }
    }
 
run:
    SendMessageW(hDlg, WM_SETSTATUS, DLSTATUS_WAITING_INSTALL, 0);
 
    // run it
    if (Info.DLType == DLTYPE_APPLICATION)
    {
        BOOL bSilentInstall = Info.Flags & DAF_SILENT;
        CStringW app, params, tempdir;
        SHELLEXECUTEINFOW shExInfo = { sizeof(shExInfo), SEE_MASK_NOCLOSEPROCESS, hDlg };
        shExInfo.lpVerb = L"open";
        shExInfo.lpFile = Path;
        shExInfo.lpParameters = L"";
        shExInfo.nShow = SW_SHOW;
 
        if (Info.IType == INSTALLER_GENERATE)
        {
            params = L"/" CMD_KEY_GENINST + CStringW(bSilentInstall ? L" /S" : L"") +
                     L" \"" + Info.szPackageName + L"\" \"" + shExInfo.lpFile + L"\"";
            shExInfo.lpParameters = params;
            shExInfo.lpFile = app.GetBuffer(MAX_PATH);
            GetModuleFileNameW(NULL, const_cast<LPWSTR>(shExInfo.lpFile), MAX_PATH);
            app.ReleaseBuffer();
        }
        else if (Info.IType == INSTALLER_EXEINZIP)
        {
            HRESULT hr = ExtractArchiveForExecution(Path, Info.szPackageName, tempdir, app);
            if (FAILED(hr))
            {
                ShowLastError(hDlg, FALSE, hr);
                goto end;
            }
            shExInfo.lpFile = app;
        }
 
        if (bSilentInstall)
        {
            if (!Info.szSilentInstallArgs.IsEmpty())
            {
                // The package wants to force specific parameters
                shExInfo.lpParameters = Info.szSilentInstallArgs.GetString();
            }
            else
            {
                UINT extrainfo = 0;
                InstallerType it = Info.IExecType;
 
                if (it == INSTALLER_UNKNOWN || it == INSTALLER_EXEINZIP)
                    it = GuessInstallerType(shExInfo.lpFile, extrainfo);
 
                if (GetSilentInstallParameters(it, extrainfo, shExInfo.lpFile, params))
                {
                    shExInfo.lpParameters = params;
                    if (it == INSTALLER_MSI)
                        shExInfo.lpFile = L"msiexec.exe"; // params contains the .msi path
                }
            }
        }
 
        /* FIXME: Do we want to log installer status? */
        WriteLogMessage(EVENTLOG_SUCCESS, MSG_SUCCESS_INSTALL, Info.szName);
 
        if (ShellExecuteExW(&shExInfo))
        {
            // reflect installation progress in the titlebar
            // TODO: make a separate string with a placeholder to include app name?
            CStringW szMsgText = LoadStatusString(DLSTATUS_INSTALLING);
            SetWindowTextW(hDlg, szMsgText);
 
            SendMessageW(hDlg, WM_SETSTATUS, DLSTATUS_INSTALLING, 0);
 
            // TODO: issue an install operation separately so that the apps could be downloaded in the background
            if (shExInfo.hProcess)
            {
                WaitForSingleObject(shExInfo.hProcess, INFINITE);
                CloseHandle(shExInfo.hProcess);
                SendMessageW(hMainWnd, WM_NOTIFY_INSTALLERFINISHED, 0, (LPARAM)(PCWSTR)Info.szPackageName);
            }
        }
        else
        {
            ShowLastError(hMainWnd, FALSE, GetLastError());
        }
 
        if (!tempdir.IsEmpty())
        {
            DeleteDirectoryTree(tempdir, hDlg);
        }
    }
 
end:
    if (hOut != INVALID_HANDLE_VALUE)
        CloseHandle(hOut);
 
    if (hFile)
        InternetCloseHandle(hFile);
    InternetCloseHandle(hOpen);
 
    if (bTempfile)
    {
        if (bCancelled || (SettingsInfo.bDelInstaller && Info.DLType == DLTYPE_APPLICATION))
        {
            // Don't delete .zip/.cab files so the user can extract from them
            if (bCancelled || Info.IType == INSTALLER_GENERATE || Info.IType == INSTALLER_EXEINZIP ||
                !OpensWithExplorer(Path) || HIBYTE(ClassifyFile(Path)) != PERCEIVED_TYPE_COMPRESSED)
            {
                DeleteFileW(Path);
            }
        }
    }
 
    SendMessageW(hDlg, WM_SETSTATUS, DLSTATUS_FINISHED, 0);
}

Referenced by ThreadFunc().

RealDlgProc()

INT_PTR CDownloadManager::RealDlgProc	(	HWND	hDlg,
		UINT	uMsg,
		WPARAM	wParam,
		LPARAM	lParam
	)

Definition at line 618 of file loaddlg.cpp.

{
    switch (uMsg)
    {
        case WM_INITDIALOG:
        {
            g_Busy++;
            AddRef();
            m_hDlg = hDlg;
            if (!m_bModal)
                g_hDownloadWnd = hDlg;
 
            HICON hIconSm, hIconBg;
            if (hMainWnd)
            {
                hIconBg = (HICON)GetClassLongPtrW(hMainWnd, GCLP_HICON);
                hIconSm = (HICON)GetClassLongPtrW(hMainWnd, GCLP_HICONSM);
            }
            if (!hMainWnd || (!hIconBg || !hIconSm))
            {
                hIconBg = hIconSm = LoadIconW(hInst, MAKEINTRESOURCEW(IDI_MAIN));
            }
            SendMessageW(hDlg, WM_SETICON, ICON_BIG, (LPARAM)hIconBg);
            SendMessageW(hDlg, WM_SETICON, ICON_SMALL, (LPARAM)hIconSm);
 
            GetWindowTextW(hDlg, m_szCaptionFmt, _countof(m_szCaptionFmt));
            CStringW buf;
            buf = m_szCaptionFmt;
            buf.Replace(L"%ls", L"");
            SetWindowTextW(hDlg, buf); // "Downloading..."
 
            HWND hItem = GetDlgItem(hDlg, IDC_DOWNLOAD_PROGRESS);
            if (hItem)
            {
                // initialize the default values for our nifty progress bar
                // and subclass it so that it learns to print a status text
                m_ProgressBar.SubclassWindow(hItem);
                m_ProgressBar.SendMessageW(PBM_SETRANGE, 0, MAKELPARAM(0, 100));
                m_ProgressBar.SendMessageW(PBM_SETPOS, 0, 0);
                if (m_List.GetSize() > 0)
                    m_ProgressBar.SetProgress(0, m_List[0].SizeInBytes);
            }
 
            if (!m_ListView.Create(hDlg))
                return FALSE;
            m_ListView.LoadList(m_List);
 
            ShowWindow(hDlg, IsSilentDialog() ? SW_HIDE : SW_SHOW);
            StartWorkerThread();
            return TRUE;
        }
 
        case WM_COMMAND:
            if (LOWORD(wParam) == IDCANCEL)
            {
                m_bCancelled = TRUE;
                PostMessageW(hDlg, WM_CLOSE, 0, 0);
            }
            return FALSE;
 
        case WM_CLOSE:
            m_bCancelled = TRUE;
            if (m_ProgressBar)
                m_ProgressBar.UnsubclassWindow(TRUE);
            return m_bModal && !IsSilentDialog() ? ::EndDialog(hDlg, 0) : ::DestroyWindow(hDlg);
 
        case WM_DESTROY:
            if (g_hDownloadWnd == hDlg)
                g_hDownloadWnd = NULL;
            g_Busy--;
            if (hMainWnd)
                PostMessage(hMainWnd, WM_NOTIFY_OPERATIONCOMPLETED, 0, 0);
            if (m_bModal && IsSilentDialog())
                PostQuitMessage(0);
            Release();
            break;
 
        case WM_ISCANCELLED:
            return SetDlgMsgResult(hDlg, uMsg, m_bCancelled);
 
        case WM_SETSTATUS:
            m_ListView.SetDownloadStatus(m_Index - 1, (DownloadStatus)wParam);
            break;
 
        case WM_GETINSTANCE:
            return SetDlgMsgResult(hDlg, uMsg, (INT_PTR)this);
 
        case WM_GETNEXT:
        {
            DownloadInfo *pItem = NULL;
            if (!m_bCancelled && m_Index < (SIZE_T)m_List.GetSize())
                pItem = &m_List[m_Index++];
            return SetDlgMsgResult(hDlg, uMsg, (INT_PTR)pItem);
        }
    }
    return FALSE;
}

Referenced by DlgProc().

Show()

void CDownloadManager::Show

(

)

Definition at line 588 of file loaddlg.cpp.

{
    HWND hDlg = NULL;
    const BOOL bSilent = IsSilentDialog(), bModal = m_bModal;
    if (bModal && !bSilent)
        DialogBoxParamW(hInst, MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG), hMainWnd, DlgProc, (LPARAM)this);
    else if (!m_hDlg || !IsWindow(m_hDlg))
        hDlg = CreateDialogParamW(hInst, MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG), hMainWnd, DlgProc, (LPARAM)this);
 
    // A DialogBox dialog cannot be invisible, it is forced visible after WM_INITDIALOG returns.
    // We therefore use a modeless dialog when we are both modal and silent.
    for (MSG msg; bModal && bSilent && hDlg && GetMessageW(&msg, NULL, 0, 0);)
        DispatchMessage(&msg);
}

StartWorkerThread()

void CDownloadManager::StartWorkerThread

(

)

Definition at line 562 of file loaddlg.cpp.

{
    AddRef(); // To keep m_List alive in thread
    unsigned int ThreadId;
    HANDLE Thread = (HANDLE)_beginthreadex(NULL, 0, ThreadFunc, this, 0, &ThreadId);
    if (Thread)
        CloseHandle(Thread);
    else
        Release();
}

Referenced by RealDlgProc().

ThreadFunc()

unsigned int CALLBACK CDownloadManager::ThreadFunc ( void *  ThreadParam )

static

Definition at line 723 of file loaddlg.cpp.

{
    CDownloadManager *pThis = (CDownloadManager*)ThreadParam;
    HWND hDlg = pThis->m_hDlg;
    for (;;)
    {
        DownloadInfo *pItem = (DownloadInfo*)SendMessageW(hDlg, WM_GETNEXT, 0, 0);
        if (!pItem)
            break;
        pThis->PerformDownloadAndInstall(*pItem);
    }
    SendMessageW(hDlg, WM_CLOSE, 0, 0);
    return pThis->Release();
}

Referenced by StartWorkerThread().

UpdateProgress()

void CDownloadManager::UpdateProgress	(	ULONG	ulProgress,
		ULONG	ulProgressMax
	)

Definition at line 717 of file loaddlg.cpp.

{
    m_ProgressBar.SetProgress(ulProgress, ulProgressMax);
}

Referenced by PerformDownloadAndInstall().

Member Data Documentation

m_bCancelled

BOOL CDownloadManager::m_bCancelled

protected

Definition at line 552 of file loaddlg.cpp.

Referenced by RealDlgProc().

m_bModal

BOOL CDownloadManager::m_bModal

protected

Definition at line 553 of file loaddlg.cpp.

Referenced by RealDlgProc(), and Show().

m_fDaf

UINT CDownloadManager::m_fDaf = 0

protected

Definition at line 554 of file loaddlg.cpp.

Referenced by IsSilentDialog().

m_hDlg

HWND CDownloadManager::m_hDlg

protected

Definition at line 549 of file loaddlg.cpp.

Referenced by Add(), IsCancelled(), PerformDownloadAndInstall(), RealDlgProc(), Show(), and ThreadFunc().

m_Index

UINT CDownloadManager::m_Index

protected

Definition at line 551 of file loaddlg.cpp.

Referenced by RealDlgProc().

m_List

ATL::CSimpleArray<DownloadInfo> CDownloadManager::m_List

protected

Definition at line 556 of file loaddlg.cpp.

Referenced by Add(), and RealDlgProc().

m_ListView

CDowloadingAppsListView CDownloadManager::m_ListView

protected

Definition at line 557 of file loaddlg.cpp.

Referenced by Add(), and RealDlgProc().

m_ProgressBar

CDownloaderProgress CDownloadManager::m_ProgressBar

protected

Definition at line 558 of file loaddlg.cpp.

Referenced by PerformDownloadAndInstall(), RealDlgProc(), and UpdateProgress().

m_szCaptionFmt

WCHAR CDownloadManager::m_szCaptionFmt[100]

protected

Definition at line 555 of file loaddlg.cpp.

Referenced by PerformDownloadAndInstall(), and RealDlgProc().

m_Threads

UINT CDownloadManager::m_Threads

protected

Definition at line 550 of file loaddlg.cpp.

---

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

• base/applications/rapps/loaddlg.cpp