reactos/rosapps/downloader/download.c

/* PROJECT:     ReactOS Downloader (was GetFirefox)
 * LICENSE:     GPL - See COPYING in the top level directory
 * FILE:        base/applications/downloader/download.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 2007 Dmitry Chapyshev ()lentind@yandex.ru)
 */
/*
 * 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#define COBJMACROS
#define WIN32_NO_STATUS
#include <windows.h>
#include <commctrl.h>
#include <shlobj.h>
#include <shlwapi.h>
#include <urlmon.h>

#include "resources.h"
#include "structures.h"

#define NDEBUG
#include <debug.h>

extern struct Application* SelectedApplication;
extern WCHAR Strings [STRING_COUNT][MAX_STRING_LENGHT];

typedef struct _IBindStatusCallbackImpl
  {
    const IBindStatusCallbackVtbl *vtbl;
    LONG ref;
    HWND hDialog;
    BOOL *pbCancelled;
  } IBindStatusCallbackImpl;

static HRESULT WINAPI
dlQueryInterface(IBindStatusCallback* This, REFIID riid, void** ppvObject)
{
	if (NULL == 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)
{
	DPRINT1("OnStartBinding not implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlGetPriority(IBindStatusCallback* iface, LONG* pnPriority)
{
	DPRINT1("GetPriority not implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlOnLowResource( IBindStatusCallback* iface, DWORD reserved)
{
	DPRINT1("OnLowResource not implemented\n");

	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_PROGRESS);
	if (NULL != Item && 0 != ulProgressMax)
    {
		SendMessageW(Item, PBM_SETPOS, (ulProgress * 100) / ulProgressMax, 0);
    }

	Item = GetDlgItem(This->hDialog, IDC_STATUS);
	if (NULL != Item && NULL != 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;
		DPRINT("Cancelled\n");
		return E_ABORT;
    }

	return S_OK;
}

static HRESULT WINAPI
dlOnStopBinding(IBindStatusCallback* iface, HRESULT hresult, LPCWSTR szError)
{
	DPRINT1("OnStopBinding not implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlGetBindInfo(IBindStatusCallback* iface, DWORD* grfBINDF, BINDINFO* pbindinfo)
{
	DPRINT1("GetBindInfo not implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlOnDataAvailable(IBindStatusCallback* iface, DWORD grfBSCF,
                  DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed)
{
	DPRINT1("OnDataAvailable implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlOnObjectAvailable(IBindStatusCallback* iface, REFIID riid, IUnknown* punk)
{
	DPRINT1("OnObjectAvailable implemented\n");

	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(), 0, sizeof(IBindStatusCallbackImpl));
	This->vtbl = &dlVtbl;
	This->ref = 1;
	This->hDialog = Dlg;
	This->pbCancelled = pbCancelled;

	return (IBindStatusCallback*) This;
}

static DWORD WINAPI
ThreadFunc(LPVOID Context)
{
	//static const WCHAR szUrl[] = DownloadUrl;
	IBindStatusCallback *dl;
	WCHAR path[MAX_PATH];
	LPWSTR p;
	STARTUPINFOW si;
	PROCESS_INFORMATION pi;
	HWND Dlg = (HWND) Context;
	DWORD r;
	BOOL bCancelled = FALSE;
	BOOL bTempfile = FALSE;
	HKEY hKey;

	/* built the path for the download */
	p = wcsrchr(SelectedApplication->Location, L'/');
	if (NULL == p)
    {
		goto end;
    }
	if (RegOpenKey(HKEY_LOCAL_MACHINE,
				  TEXT("Software\\ReactOS\\Download!"),
                  &hKey) != ERROR_SUCCESS)
	{
		goto end;
	}

	DWORD dwSize = MAX_PATH;
	if (RegQueryValueEx(hKey,
						L"DownloadFolder",
						NULL,
						NULL,
						(LPBYTE)&path,
						&dwSize) != ERROR_SUCCESS)
	{
		goto end;
	}

	if (GetFileAttributes(path) == 0xFFFFFFFF)
		if (!CreateDirectory((LPCTSTR)path,NULL))
		{
			goto end;
		}
	wcscat(path, L"\\");
	wcscat(path, p + 1);

	/* download it */
	bTempfile = TRUE;
	dl = CreateDl(Context, &bCancelled);
	r = URLDownloadToFileW(NULL, SelectedApplication->Location, path, 0, dl);
	if (NULL != dl)
    {
		IBindStatusCallback_Release(dl);
    }
	if (S_OK != r)
    {
		MessageBoxW(0,Strings[IDS_DOWNLOAD_ERROR],0,0);
		goto end;
    }
	else if (bCancelled)
    {
		goto end;
    }
	ShowWindow(Dlg, SW_HIDE);

	/* run it */
	memset(&si, 0, sizeof(si));
	si.cb = sizeof(si);
	r = CreateProcessW(path, NULL, NULL, NULL, 0, 0, NULL, NULL, &si, &pi);
	if (0 == r)
    {
		goto end;
    }
	CloseHandle(pi.hThread);
	WaitForSingleObject(pi.hProcess, INFINITE);
	CloseHandle(pi.hProcess);

	end:
		if (bTempfile)
		{
			DWORD dwSize = sizeof(DWORD);
			DWORD dwValue, dwType = REG_DWORD;
			if (RegQueryValueEx(hKey,
								L"DeleteInstaller",
								NULL,
								&dwType,
								(LPBYTE)&dwValue,
								&dwSize) == ERROR_SUCCESS)
				if (dwValue == 0x1)
					DeleteFileW(path);
			RegCloseKey(hKey);
		}
	EndDialog(Dlg, 0);
	return 0;
}

INT_PTR CALLBACK
DownloadProc(HWND Dlg, UINT Msg, WPARAM wParam, LPARAM lParam)
{
	HANDLE Thread;
	DWORD ThreadId;
	HWND Item;

	switch (Msg)
    {
		case WM_INITDIALOG:/*
      Icon = LoadIconW((HINSTANCE) GetWindowLongPtr(Dlg, GWLP_HINSTANCE),
                       MAKEINTRESOURCEW(IDI_ICON_MAIN));
      if (NULL != Icon)
        {
          SendMessageW(Dlg, WM_SETICON, ICON_BIG, (LPARAM) Icon);
          SendMessageW(Dlg, WM_SETICON, ICON_SMALL, (LPARAM) Icon);
        }*/
		SetWindowLongPtrW(Dlg, GWLP_USERDATA, 0);
		Item = GetDlgItem(Dlg, IDC_PROGRESS);
		if (NULL != Item)
        {
			SendMessageW(Item, PBM_SETRANGE, 0, MAKELPARAM(0,100));
			SendMessageW(Item, PBM_SETPOS, 0, 0);
        }/*
      Item = GetDlgItem(Dlg, IDC_REMOVE);
      if (NULL != Item)
        {
          if (GetShortcutName(ShortcutName) &&
              INVALID_FILE_ATTRIBUTES != GetFileAttributesW(ShortcutName))
            {
              SendMessageW(Item, BM_SETCHECK, BST_CHECKED, 0);
            }
          else
            {
              SendMessageW(Item, BM_SETCHECK, BST_UNCHECKED, 0);
              ShowWindow(Item, SW_HIDE);
            }
        }*/
		Thread = CreateThread(NULL, 0, ThreadFunc, Dlg, 0, &ThreadId);
		if (NULL == Thread)
		{
			return FALSE;
		}
		CloseHandle(Thread);
		return TRUE;

    case WM_COMMAND:
		if (wParam == IDCANCEL)
		{
			SetWindowLongPtrW(Dlg, GWLP_USERDATA, 1);
			PostMessage(Dlg, WM_CLOSE, 0, 0);
		}
		return FALSE;

    case WM_CLOSE:
        EndDialog(Dlg, 0);
        return TRUE;

	default:
		return FALSE;
    }
}