mirror of
https://github.com/reactos/reactos.git
synced 2024-10-25 15:36:07 +00:00
057a69972f
CORE-8176 #resolve #comment Trunk commit r63174 should fix this issue. I merged from trunk to bring it into Shell-experiments. svn path=/branches/shell-experiments/; revision=63175
407 lines
10 KiB
C
407 lines
10 KiB
C
/* PROJECT: ReactOS Applications Manager
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* FILE: base/applications/rapps/loaddlg.c
|
|
* 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)
|
|
*/
|
|
/*
|
|
* 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 <wininet.h>
|
|
#include <shellapi.h>
|
|
|
|
static PAPPLICATION_INFO AppInfo;
|
|
static HICON hIcon = NULL;
|
|
|
|
typedef struct _IBindStatusCallbackImpl
|
|
{
|
|
const IBindStatusCallbackVtbl *vtbl;
|
|
LONG ref;
|
|
HWND hDialog;
|
|
BOOL *pbCancelled;
|
|
} IBindStatusCallbackImpl;
|
|
|
|
static
|
|
HRESULT WINAPI
|
|
dlQueryInterface(IBindStatusCallback* This, REFIID riid, void** ppvObject)
|
|
{
|
|
if (!ppvObject) return E_POINTER;
|
|
|
|
if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IBindStatusCallback))
|
|
{
|
|
IBindStatusCallback_AddRef(This);
|
|
*ppvObject = This;
|
|
return S_OK;
|
|
}
|
|
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static
|
|
ULONG WINAPI
|
|
dlAddRef(IBindStatusCallback* iface)
|
|
{
|
|
IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl*) iface;
|
|
return InterlockedIncrement(&This->ref);
|
|
}
|
|
|
|
static
|
|
ULONG WINAPI
|
|
dlRelease(IBindStatusCallback* iface)
|
|
{
|
|
IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl*) iface;
|
|
DWORD ref = InterlockedDecrement(&This->ref);
|
|
|
|
if (!ref)
|
|
{
|
|
DestroyWindow(This->hDialog);
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static
|
|
HRESULT WINAPI
|
|
dlOnStartBinding(IBindStatusCallback* iface, DWORD dwReserved, IBinding* pib)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
static
|
|
HRESULT WINAPI
|
|
dlGetPriority(IBindStatusCallback* iface, LONG* pnPriority)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
static
|
|
HRESULT WINAPI
|
|
dlOnLowResource( IBindStatusCallback* iface, DWORD reserved)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
static
|
|
HRESULT WINAPI
|
|
dlOnProgress(IBindStatusCallback* iface,
|
|
ULONG ulProgress,
|
|
ULONG ulProgressMax,
|
|
ULONG ulStatusCode,
|
|
LPCWSTR szStatusText)
|
|
{
|
|
IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl *) iface;
|
|
HWND Item;
|
|
LONG r;
|
|
WCHAR OldText[100];
|
|
|
|
Item = GetDlgItem(This->hDialog, IDC_DOWNLOAD_PROGRESS);
|
|
if (Item && ulProgressMax)
|
|
{
|
|
SendMessageW(Item, PBM_SETPOS, ((ULONGLONG)ulProgress * 100) / ulProgressMax, 0);
|
|
}
|
|
|
|
Item = GetDlgItem(This->hDialog, IDC_DOWNLOAD_STATUS);
|
|
if (Item && szStatusText)
|
|
{
|
|
SendMessageW(Item, WM_GETTEXT, sizeof(OldText) / sizeof(OldText[0]), (LPARAM) OldText);
|
|
if (sizeof(OldText) / sizeof(OldText[0]) - 1 <= wcslen(OldText) || 0 != wcscmp(OldText, szStatusText))
|
|
{
|
|
SendMessageW(Item, WM_SETTEXT, 0, (LPARAM) szStatusText);
|
|
}
|
|
}
|
|
|
|
SetLastError(0);
|
|
r = GetWindowLongPtrW(This->hDialog, GWLP_USERDATA);
|
|
if (0 != r || 0 != GetLastError())
|
|
{
|
|
*This->pbCancelled = TRUE;
|
|
return E_ABORT;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static
|
|
HRESULT WINAPI
|
|
dlOnStopBinding(IBindStatusCallback* iface, HRESULT hresult, LPCWSTR szError)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
static
|
|
HRESULT WINAPI
|
|
dlGetBindInfo(IBindStatusCallback* iface, DWORD* grfBINDF, BINDINFO* pbindinfo)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
static
|
|
HRESULT WINAPI
|
|
dlOnDataAvailable(IBindStatusCallback* iface, DWORD grfBSCF,
|
|
DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
static
|
|
HRESULT WINAPI
|
|
dlOnObjectAvailable(IBindStatusCallback* iface, REFIID riid, IUnknown* punk)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
static const IBindStatusCallbackVtbl dlVtbl =
|
|
{
|
|
dlQueryInterface,
|
|
dlAddRef,
|
|
dlRelease,
|
|
dlOnStartBinding,
|
|
dlGetPriority,
|
|
dlOnLowResource,
|
|
dlOnProgress,
|
|
dlOnStopBinding,
|
|
dlGetBindInfo,
|
|
dlOnDataAvailable,
|
|
dlOnObjectAvailable
|
|
};
|
|
|
|
static IBindStatusCallback*
|
|
CreateDl(HWND Dlg, BOOL *pbCancelled)
|
|
{
|
|
IBindStatusCallbackImpl *This;
|
|
|
|
This = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IBindStatusCallbackImpl));
|
|
if (!This) return NULL;
|
|
|
|
This->vtbl = &dlVtbl;
|
|
This->ref = 1;
|
|
This->hDialog = Dlg;
|
|
This->pbCancelled = pbCancelled;
|
|
|
|
return (IBindStatusCallback*) This;
|
|
}
|
|
|
|
static
|
|
DWORD WINAPI
|
|
ThreadFunc(LPVOID Context)
|
|
{
|
|
IBindStatusCallback *dl = NULL;
|
|
WCHAR path[MAX_PATH];
|
|
LPWSTR p;
|
|
HWND Dlg = (HWND) Context;
|
|
DWORD len, dwContentLen, dwBytesWritten, dwBytesRead;
|
|
DWORD dwCurrentBytesRead = 0;
|
|
DWORD dwBufLen = sizeof(dwContentLen);
|
|
BOOL bCancelled = FALSE;
|
|
BOOL bTempfile = FALSE;
|
|
BOOL bCab = FALSE;
|
|
HINTERNET hOpen = NULL;
|
|
HINTERNET hFile = NULL;
|
|
HANDLE hOut = INVALID_HANDLE_VALUE;
|
|
unsigned char lpBuffer[4096];
|
|
const LPWSTR lpszAgent = L"RApps/1.0";
|
|
|
|
/* built the path for the download */
|
|
p = wcsrchr(AppInfo->szUrlDownload, L'/');
|
|
if (!p) goto end;
|
|
|
|
len = wcslen(AppInfo->szUrlDownload);
|
|
if (len > 4)
|
|
{
|
|
if (AppInfo->szUrlDownload[len - 4] == '.' &&
|
|
AppInfo->szUrlDownload[len - 3] == 'c' &&
|
|
AppInfo->szUrlDownload[len - 2] == 'a' &&
|
|
AppInfo->szUrlDownload[len - 1] == 'b')
|
|
{
|
|
bCab = TRUE;
|
|
if (!GetStorageDirectory(path, sizeof(path) / sizeof(path[0])))
|
|
goto end;
|
|
}
|
|
else
|
|
{
|
|
if (FAILED(StringCbCopyW(path, sizeof(path),
|
|
SettingsInfo.szDownloadDir)))
|
|
{
|
|
goto end;
|
|
}
|
|
}
|
|
}
|
|
else goto end;
|
|
|
|
if (GetFileAttributesW(path) == INVALID_FILE_ATTRIBUTES)
|
|
{
|
|
if (!CreateDirectoryW(path, NULL))
|
|
goto end;
|
|
}
|
|
|
|
if (FAILED(StringCbCatW(path, sizeof(path), L"\\")))
|
|
goto end;
|
|
if (FAILED(StringCbCatW(path, sizeof(path), p + 1)))
|
|
goto end;
|
|
|
|
/* download it */
|
|
bTempfile = TRUE;
|
|
dl = CreateDl(Context, &bCancelled);
|
|
if (dl == NULL) goto end;
|
|
|
|
hOpen = InternetOpenW(lpszAgent, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
|
|
if (!hOpen) goto end;
|
|
|
|
hFile = InternetOpenUrlW(hOpen, AppInfo->szUrlDownload, NULL, 0, INTERNET_FLAG_PRAGMA_NOCACHE|INTERNET_FLAG_KEEP_CONNECTION, 0);
|
|
if(!hFile) goto end;
|
|
|
|
hOut = CreateFileW(path, GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, 0, NULL);
|
|
if (hOut == INVALID_HANDLE_VALUE) goto end;
|
|
|
|
HttpQueryInfo(hFile, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, &dwContentLen, &dwBufLen, 0);
|
|
|
|
do
|
|
{
|
|
if (!InternetReadFile(hFile, lpBuffer, _countof(lpBuffer), &dwBytesRead)) goto end;
|
|
if (!WriteFile(hOut, &lpBuffer[0], dwBytesRead, &dwBytesWritten, NULL)) goto end;
|
|
dwCurrentBytesRead += dwBytesRead;
|
|
IBindStatusCallback_OnProgress(dl, dwCurrentBytesRead, dwContentLen, 0, AppInfo->szUrlDownload);
|
|
}
|
|
while (dwBytesRead);
|
|
|
|
CloseHandle(hOut);
|
|
hOut = INVALID_HANDLE_VALUE;
|
|
|
|
if (bCancelled) goto end;
|
|
|
|
ShowWindow(Dlg, SW_HIDE);
|
|
|
|
/* run it */
|
|
if (!bCab)
|
|
{
|
|
ShellExecuteW( NULL, L"open", path, NULL, NULL, SW_SHOWNORMAL );
|
|
}
|
|
end:
|
|
if (hOut != INVALID_HANDLE_VALUE) CloseHandle(hOut);
|
|
InternetCloseHandle(hFile);
|
|
InternetCloseHandle(hOpen);
|
|
|
|
if (dl) IBindStatusCallback_Release(dl);
|
|
|
|
if (bTempfile)
|
|
{
|
|
if (bCancelled || (SettingsInfo.bDelInstaller && !bCab))
|
|
DeleteFileW(path);
|
|
}
|
|
|
|
EndDialog(Dlg, 0);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static
|
|
INT_PTR CALLBACK
|
|
DownloadDlgProc(HWND Dlg, UINT Msg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
HANDLE Thread;
|
|
DWORD ThreadId;
|
|
HWND Item;
|
|
|
|
switch (Msg)
|
|
{
|
|
case WM_INITDIALOG:
|
|
|
|
hIcon = LoadIconW(hInst, MAKEINTRESOURCEW(IDI_MAIN));
|
|
if (hIcon)
|
|
{
|
|
SendMessageW(Dlg, WM_SETICON, ICON_BIG, (LPARAM) hIcon);
|
|
SendMessageW(Dlg, WM_SETICON, ICON_SMALL, (LPARAM) hIcon);
|
|
}
|
|
|
|
SetWindowLongPtrW(Dlg, GWLP_USERDATA, 0);
|
|
Item = GetDlgItem(Dlg, IDC_DOWNLOAD_PROGRESS);
|
|
if (Item)
|
|
{
|
|
SendMessageW(Item, PBM_SETRANGE, 0, MAKELPARAM(0, 100));
|
|
SendMessageW(Item, PBM_SETPOS, 0, 0);
|
|
}
|
|
|
|
Thread = CreateThread(NULL, 0, ThreadFunc, Dlg, 0, &ThreadId);
|
|
if (!Thread) return FALSE;
|
|
CloseHandle(Thread);
|
|
return TRUE;
|
|
|
|
case WM_COMMAND:
|
|
if (wParam == IDCANCEL)
|
|
{
|
|
SetWindowLongPtrW(Dlg, GWLP_USERDATA, 1);
|
|
PostMessageW(Dlg, WM_CLOSE, 0, 0);
|
|
}
|
|
return FALSE;
|
|
|
|
case WM_CLOSE:
|
|
if (hIcon) DestroyIcon(hIcon);
|
|
EndDialog(Dlg, 0);
|
|
return TRUE;
|
|
|
|
default:
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
DownloadApplication(INT Index)
|
|
{
|
|
if (!IS_AVAILABLE_ENUM(SelectedEnumType))
|
|
return FALSE;
|
|
|
|
AppInfo = (PAPPLICATION_INFO) ListViewGetlParam(Index);
|
|
if (!AppInfo) return FALSE;
|
|
|
|
WriteLogMessage(EVENTLOG_SUCCESS, MSG_SUCCESS_INSTALL, AppInfo->szName);
|
|
|
|
DialogBoxW(hInst,
|
|
MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG),
|
|
hMainWnd,
|
|
DownloadDlgProc);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
VOID
|
|
DownloadApplicationsDB(LPWSTR lpUrl)
|
|
{
|
|
APPLICATION_INFO IntInfo;
|
|
|
|
ZeroMemory(&IntInfo, sizeof(APPLICATION_INFO));
|
|
if (FAILED(StringCbCopyW(IntInfo.szUrlDownload,
|
|
sizeof(IntInfo.szUrlDownload),
|
|
lpUrl)))
|
|
{
|
|
return;
|
|
}
|
|
|
|
AppInfo = &IntInfo;
|
|
|
|
DialogBoxW(hInst,
|
|
MAKEINTRESOURCEW(IDD_DOWNLOAD_DIALOG),
|
|
hMainWnd,
|
|
DownloadDlgProc);
|
|
}
|
|
|