|
|
#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
|
