246 lines
9.8 KiB
C
246 lines
9.8 KiB
C
|
#ifndef _ASINSCTL_H
|
||
|
|
#define _ASINSCTL_H
|
||
|
|
|
||
|
|
#include "ipserver.h"
|
||
|
|
#include "ctrlobj.h"
|
||
|
|
#include "internet.h"
|
||
|
|
#include "iasctrls.h"
|
||
|
|
#include "inseng.h"
|
||
|
|
#include "progdlg.h"
|
||
|
|
|
||
|
|
|
||
|
|
#define RESID_TOOLBOX_BITMAP 1
|
||
|
|
|
||
|
|
#define FINALIZE_DOREBOOT 0x00000001
|
||
|
|
#define FINALIZE_NOREBOOTPROMPT 0x00000002
|
||
|
|
|
||
|
|
#define PROCESSCOMPONENT_DELAYREBOOT 0x00000001
|
||
|
|
#define PROCESSCOMPONENT_NOPROGRESSUI 0x00000002
|
||
|
|
#define PROCESSCOMPONENT_NOSUMMARYUI 0x00000004
|
||
|
|
|
||
|
|
#define PROCESSCOMPONENT_NOINSTALLUI 0x00000080
|
||
|
|
|
||
|
|
#define MAX_URLS 5
|
||
|
|
#define ERROR_STRING_SIZE 2048
|
||
|
|
#define ERROR_STRING_INCREMENT 2048
|
||
|
|
|
||
|
|
//---------------------------------------------
|
||
|
|
// AUTORUN SETTINGS FOR WIN
|
||
|
|
//---------------------------------------------
|
||
|
|
#define WINDOWS_DEFAULT_AUTOPLAY_VALUE 0x095
|
||
|
|
#define WINDOWS_AUTOPLAY_OFF 0x0FF
|
||
|
|
|
||
|
|
typedef struct
|
||
|
|
{
|
||
|
|
DWORD dwSize;
|
||
|
|
DWORD dwStatus;
|
||
|
|
DWORD dwSubstatus;
|
||
|
|
DWORD dwPhase;
|
||
|
|
DWORD dwResult;
|
||
|
|
BSTR strID;
|
||
|
|
BSTR strName;
|
||
|
|
BSTR strString;
|
||
|
|
BSTR chWin;
|
||
|
|
BSTR chInstall;
|
||
|
|
BSTR chDL;
|
||
|
|
DWORD dwWin;
|
||
|
|
DWORD dwInstall;
|
||
|
|
DWORD dwDL;
|
||
|
|
} CALLBACK_PARAMS;
|
||
|
|
|
||
|
|
|
||
|
|
// Global needed in CInstallEngine for suppressing CD Auto Splash.
|
||
|
|
extern UINT g_uCDAutorunMsg;
|
||
|
|
extern unsigned long g_ulOldAutorunSetting;
|
||
|
|
unsigned long SetAutorunSetting(unsigned long ulNewSettting);
|
||
|
|
|
||
|
|
//=--------------------------------------------------------------------------=
|
||
|
|
// CInstallEngine
|
||
|
|
//=--------------------------------------------------------------------------=
|
||
|
|
// our control.
|
||
|
|
//
|
||
|
|
class CInstallEngineCtl : public COleControl, public IInstallEngineCtl, public IInstallEngineCallback, public ISupportErrorInfo
|
||
|
|
{
|
||
|
|
|
||
|
|
friend class CProgressDlg;
|
||
|
|
friend INT_PTR CALLBACK ProgressDlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
|
||
|
|
friend DWORD WINAPI DoInstall(LPVOID pv);
|
||
|
|
friend INT_PTR CALLBACK LocationDlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
|
||
|
|
friend INT_PTR CALLBACK DiskSpaceDlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
|
||
|
|
|
||
|
|
public:
|
||
|
|
// IUnknown methods
|
||
|
|
//
|
||
|
|
DECLARE_STANDARD_UNKNOWN();
|
||
|
|
|
||
|
|
// IDispatch methods
|
||
|
|
//
|
||
|
|
DECLARE_STANDARD_DISPATCH();
|
||
|
|
|
||
|
|
// ISupportErrorInfo methods
|
||
|
|
//
|
||
|
|
DECLARE_STANDARD_SUPPORTERRORINFO();
|
||
|
|
|
||
|
|
// IInstallEngine methods
|
||
|
|
//
|
||
|
|
STDMETHOD(get_ReadyState)(THIS_ long FAR* thestate);
|
||
|
|
STDMETHOD(SetCifFile)(THIS_ BSTR strCabName, BSTR strCifName);
|
||
|
|
STDMETHOD(put_BaseUrl)(THIS_ BSTR strBaseUrl);
|
||
|
|
STDMETHOD(put_DownloadDir)(THIS_ BSTR strDownloadDir);
|
||
|
|
STDMETHOD(SetAction)(THIS_ BSTR ComponentID, long action, long *lResult);
|
||
|
|
STDMETHOD(IsComponentInstalled)(THIS_ BSTR ComponentID, long *lResult);
|
||
|
|
STDMETHOD(ProcessComponents)(THIS_ long lFlag);
|
||
|
|
STDMETHOD(get_Size)(THIS_ BSTR ComponentID, long FAR* thestate);
|
||
|
|
STDMETHOD(get_DisplayName)(THIS_ BSTR ComponentID, BSTR *name);
|
||
|
|
STDMETHOD(Abort)(THIS_ long lFlags);
|
||
|
|
STDMETHOD(get_TotalDownloadSize)(THIS_ long FAR* totalsize);
|
||
|
|
STDMETHOD(get_TotalDependencySize)(THIS_ long FAR* totaldepsize);
|
||
|
|
STDMETHOD(FinalizeInstall)(THIS_ long lFlag);
|
||
|
|
STDMETHOD(get_EngineStatus)(THIS_ long FAR* theenginestatus);
|
||
|
|
STDMETHOD(HandleEngineProblem)(THIS_ long lFlag);
|
||
|
|
STDMETHOD(CheckFreeSpace)(THIS_ long lPad, long FAR* lEnough);
|
||
|
|
STDMETHOD(SetLocalCif)(THIS_ BSTR strCif, long FAR* lResult);
|
||
|
|
STDMETHOD(SetSitesFile)(THIS_ BSTR strUrl, BSTR strRegion, BSTR strLocale, long FAR* lResult);
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
// Install engine callbacks
|
||
|
|
STDMETHOD(OnStartInstall)(DWORD dwDLSize, DWORD dwInstallSize);
|
||
|
|
STDMETHOD(OnStartComponent)(LPCSTR pszID, DWORD dwDLSize, DWORD dwInstallSize, LPCSTR pszName);
|
||
|
|
STDMETHOD(OnComponentProgress)(LPCSTR pszID, DWORD dwPhase, LPCSTR pszName, LPCSTR pszMsgString, ULONG progress, ULONG dwMax );
|
||
|
|
STDMETHOD(OnStopComponent)(LPCSTR pszID, HRESULT hrError, DWORD dwPhase, LPCSTR pszString, DWORD dwStatus);
|
||
|
|
STDMETHOD(OnStopInstall)(HRESULT hrError, LPCSTR szError, DWORD dwStatus);
|
||
|
|
STDMETHOD(OnEngineStatusChange)(DWORD dwStatus, DWORD substatus);
|
||
|
|
STDMETHOD(OnEngineProblem)(DWORD dwProblem, LPDWORD pdwAction);
|
||
|
|
|
||
|
|
// OLE Control stuff follows:
|
||
|
|
//
|
||
|
|
CInstallEngineCtl(IUnknown *pUnkOuter, BOOL *pbSuccess);
|
||
|
|
virtual ~CInstallEngineCtl();
|
||
|
|
// static creation function. all controls must have one of these!
|
||
|
|
//
|
||
|
|
static IUnknown *Create(IUnknown *);
|
||
|
|
|
||
|
|
// Helper functions to fire our events
|
||
|
|
void _FireOnStartInstallEvent(DWORD dwTotalSize);
|
||
|
|
void _FireOnStartComponentEvent(LPCSTR pszID, DWORD dwTotalSize, LPCSTR pszName);
|
||
|
|
void _FireOnStopComponentEvent(LPCSTR pszID, HRESULT hrError, DWORD dwPhase, LPCSTR pszString, DWORD dwStatus);
|
||
|
|
void _FireOnStopInstallEvent(HRESULT hrError, LPCSTR szError, DWORD dwStatus);
|
||
|
|
void _FireEngineStatusChange(DWORD dwEngineStatus, DWORD substatus);
|
||
|
|
void _FireOnEngineProblem(DWORD dwProblem);
|
||
|
|
void _FireOnComponentProgressEvent(LPCSTR pszID, DWORD dwPhase, LPCSTR pszString, LPCSTR pszMsgString, ULONG progress, ULONG themax);
|
||
|
|
void _FireOnComponentProgress(DWORD lPhase, DWORD lSoFar, DWORD lTotal);
|
||
|
|
void _FireOnStartInstallExEvent(DWORD dwDLSize, DWORD dwInsSize);
|
||
|
|
|
||
|
|
private:
|
||
|
|
// overridables that the control must implement.
|
||
|
|
//
|
||
|
|
STDMETHOD(LoadBinaryState)(IStream *pStream);
|
||
|
|
STDMETHOD(SaveBinaryState)(IStream *pStream);
|
||
|
|
STDMETHOD(LoadTextState)(IPropertyBag *pPropertyBag, IErrorLog *pErrorLog);
|
||
|
|
STDMETHOD(SaveTextState)(IPropertyBag *pPropertyBag, BOOL fWriteDefault);
|
||
|
|
STDMETHOD(OnDraw)(DWORD dvAspect, HDC hdcDraw, LPCRECTL prcBounds, LPCRECTL prcWBounds, HDC hicTargetDev, BOOL fOptimize);
|
||
|
|
|
||
|
|
STDMETHOD(FreezeEvents)(BOOL bFreeze);
|
||
|
|
|
||
|
|
// Window stuff
|
||
|
|
virtual LRESULT WindowProc(UINT msg, WPARAM wParam, LPARAM lParam);
|
||
|
|
virtual BOOL RegisterClassData(void);
|
||
|
|
|
||
|
|
// internal QI
|
||
|
|
virtual HRESULT InternalQueryInterface(REFIID, void **);
|
||
|
|
|
||
|
|
// before and after window is created
|
||
|
|
virtual void BeforeCreateWindow(void);
|
||
|
|
virtual BOOL AfterCreateWindow(void);
|
||
|
|
|
||
|
|
|
||
|
|
void _InitProgressDialog();
|
||
|
|
void _DoInstall();
|
||
|
|
HRESULT _CheckInstallPath(BOOL *pfNeedDownload);
|
||
|
|
BOOL _PathIsIEInstallPoint(LPCSTR pszPath);
|
||
|
|
HRESULT _PickWebSites(LPCSTR pszSites, LPCSTR pszLocale, LPCSTR pszRegion, BOOL bKeepExisting);
|
||
|
|
BOOL _IsEnoughSpace(LPSTR szSpace1, DWORD dwSize1, LPSTR szSpace2, DWORD dwSize2,
|
||
|
|
LPSTR szSpace3, DWORD dwSize3);
|
||
|
|
HRESULT _CheckForDiskSpace();
|
||
|
|
HRESULT _ShowDiskSpaceDialog();
|
||
|
|
void _FireCancel(DWORD dwCancel);
|
||
|
|
LONG _OpenJITKey(HKEY *hKey, REGSAM samAttr);
|
||
|
|
void _DeleteURLList();
|
||
|
|
void _WriteURLList();
|
||
|
|
void _WriteRegionToReg(LPSTR szRegion);
|
||
|
|
void _PickRegionAndFirstSite(IDownloadSite **ppISite, UINT uNumSites, LPSTR pszRegion, UINT *puFirstSite);
|
||
|
|
void _PickRandomSites(IDownloadSite **ppISite, BYTE *pPicks, UINT uNumSites, UINT uNumToPick, LPSTR pszRegion);
|
||
|
|
void MarkJITInstall();
|
||
|
|
|
||
|
|
BOOL _fInOnEngineProblem:1;
|
||
|
|
BOOL _fEventToFire:1;
|
||
|
|
BOOL _fNeedReboot:1;
|
||
|
|
BOOL _fLocalCifSet:1;
|
||
|
|
BOOL _fReconcileCif:1;
|
||
|
|
BOOL _fDoingIEInstall:1;
|
||
|
|
BOOL _fInstalling:1;
|
||
|
|
BOOL _fJITInstall:1;
|
||
|
|
|
||
|
|
|
||
|
|
DWORD _dwOldStatus;
|
||
|
|
DWORD _dwAction;
|
||
|
|
DWORD _dwInstallStatus;
|
||
|
|
UINT _uInstallMode;
|
||
|
|
UINT _uInstallPad;
|
||
|
|
|
||
|
|
HRESULT _hResult;
|
||
|
|
|
||
|
|
DWORD _dwSavedEngineStatus;
|
||
|
|
DWORD _dwSavedSubStatus;
|
||
|
|
DWORD _dwFreezeEvents;
|
||
|
|
DWORD _uAllowGrovel;
|
||
|
|
|
||
|
|
|
||
|
|
BSTR _strCurrentID;
|
||
|
|
BSTR _strCurrentName;
|
||
|
|
BSTR _strCurrentString;
|
||
|
|
DWORD m_readyState;
|
||
|
|
LPSTR _pszErrorString;
|
||
|
|
int _iErrorStringSize;
|
||
|
|
DWORD _dwProcessComponentsFlags;
|
||
|
|
HICON _hIcon;
|
||
|
|
DWORD _dwLastPhase;
|
||
|
|
CProgressDlg *_pProgDlg;
|
||
|
|
IInstallEngine2 *_pinseng;
|
||
|
|
DWORD _dwMSTrustKey;
|
||
|
|
HANDLE _hDone;
|
||
|
|
UINT _uCurrentUrl;
|
||
|
|
LPSTR _rpszUrlList[MAX_URLS];
|
||
|
|
char _szCifCab[MAX_PATH];
|
||
|
|
char _szCifFile[MAX_PATH];
|
||
|
|
char _szDownloadDir[MAX_PATH];
|
||
|
|
BOOL _bCancelPending;
|
||
|
|
BOOL _bDeleteURLList;
|
||
|
|
BOOL _bNewWebSites;
|
||
|
|
};
|
||
|
|
|
||
|
|
DWORD WINAPI DoInstall(LPVOID pv);
|
||
|
|
|
||
|
|
DEFINE_CONTROLOBJECT(InstallEngineCtl, // Name of object
|
||
|
|
&CLSID_InstallEngineCtl, // CLSID of object
|
||
|
|
"InstallEngineCtl", // ProgID
|
||
|
|
CInstallEngineCtl::Create, // Static create function
|
||
|
|
1, // Version
|
||
|
|
&IID_IInstallEngineCtl, // Primary dispatch
|
||
|
|
NULL, // name of help file
|
||
|
|
&DIID_DInstallEngineCtlEvents, // event interface
|
||
|
|
OLEMISC_SETCLIENTSITEFIRST|OLEMISC_ACTIVATEWHENVISIBLE|OLEMISC_RECOMPOSEONRESIZE|OLEMISC_CANTLINKINSIDE|OLEMISC_INSIDEOUT,
|
||
|
|
0, // Activation policy
|
||
|
|
RESID_TOOLBOX_BITMAP, // toolbox id
|
||
|
|
"InstallEngineCtlWndClass", // Window class
|
||
|
|
0, // number of pages
|
||
|
|
NULL, // array of property pages
|
||
|
|
0, // Number of verbs
|
||
|
|
NULL); // Array of verbs
|
||
|
|
|
||
|
|
#define OBJECT_INSTALLENGINECTL 0
|
||
|
|
|
||
|
|
#endif
|