336 lines
8.8 KiB
C
336 lines
8.8 KiB
C
|
#ifndef _WEBJIT_H_
|
||
|
|
#define _WEBJIT_H_
|
||
|
|
|
||
|
|
#include <commctrl.h>
|
||
|
|
|
||
|
|
#define WM_DOWNLOAD_DONE WM_USER + 11
|
||
|
|
#define WM_SETUP_ERROR WM_USER + 12
|
||
|
|
#define WM_DOWNLOAD_ERROR WM_USER + 13
|
||
|
|
#define WM_PROCESS_ERROR WM_USER + 14
|
||
|
|
#define WM_INTERNAL_ERROR WM_USER + 15
|
||
|
|
#define WM_START_TIMER WM_USER + 16
|
||
|
|
#define WM_DONT_WAIT WM_USER + 17
|
||
|
|
#define WM_DATA_AVAILABLE WM_USER + 18
|
||
|
|
|
||
|
|
#define SUCCESS 0 // Component successfully downloaded and installed
|
||
|
|
#define CANCELLED 2 // User canceled dialog
|
||
|
|
#define NEVERASK 3 // User requested never to be asked to d/l this component again
|
||
|
|
|
||
|
|
#define NO_MIME_MATCH 11
|
||
|
|
#define NO_EXT_MATCH 12
|
||
|
|
#define EXEC_ERROR 13
|
||
|
|
|
||
|
|
#define TIMER_EXEC_POLL 1
|
||
|
|
#define TIMER_EXEC_POLL_INTERVAL 1000
|
||
|
|
#define TIMER_DOWNLOAD 2
|
||
|
|
#define TIMER_DOWNLOAD_INTERVAL 10
|
||
|
|
|
||
|
|
typedef enum
|
||
|
|
{
|
||
|
|
WJSTATE_INIT, // obj constructed
|
||
|
|
WJSTATE_BINDING, // download in progress
|
||
|
|
WJSTATE_DOWNLOADED, // OnStopBinding called
|
||
|
|
WJSTATE_FINISHED_READ, // all data read
|
||
|
|
WJSTATE_VERIFYING, // into VerifyTrust call
|
||
|
|
WJSTATE_READY_TO_EXEC, // read to shell exec.
|
||
|
|
WJSTATE_WAITING_PROCESS, // Start DoSetup
|
||
|
|
WJSTATE_DONE, // all done, ready to free obj
|
||
|
|
} WJSTATE;
|
||
|
|
|
||
|
|
class CWebJit;
|
||
|
|
typedef struct _WebJitParam
|
||
|
|
{
|
||
|
|
LPCSTR lpszResource;
|
||
|
|
DWORD dwFlags;
|
||
|
|
LPCWSTR pwszComponentId;
|
||
|
|
CWebJit* pWebJit;
|
||
|
|
HWND hWndParent;
|
||
|
|
QUERYCONTEXT *pQueryInstalled;
|
||
|
|
}
|
||
|
|
WEBJIT_PARAM;
|
||
|
|
|
||
|
|
extern IInternetSecurityManager* g_pSecurityManager;
|
||
|
|
class CWebJit : public IBindStatusCallback
|
||
|
|
, public IAuthenticate
|
||
|
|
, public IInternetHostSecurityManager
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
|
||
|
|
// data members
|
||
|
|
BOOL m_fInited;
|
||
|
|
HMODULE m_hWintrustMod;
|
||
|
|
BOOL m_fInitedCC;
|
||
|
|
HMODULE m_hComCtlMod;
|
||
|
|
DWORD m_dwFlags;
|
||
|
|
LPCWSTR m_pwszComponentId;
|
||
|
|
HWND m_hWndParent;
|
||
|
|
QUERYCONTEXT* m_pQueryInstalled;
|
||
|
|
|
||
|
|
DWORD m_cRef;
|
||
|
|
DWORD m_dwTotal;
|
||
|
|
DWORD m_dwCurrent;
|
||
|
|
LPCSTR m_szResource; //don't own: don't free.
|
||
|
|
LPWSTR m_pwszUrl;
|
||
|
|
DWORD m_dwDownloadSpeed;
|
||
|
|
|
||
|
|
IBindCtx* m_pbc;
|
||
|
|
IMoniker* m_pMk;
|
||
|
|
IStream* m_pStm;
|
||
|
|
IBinding* m_pBinding;
|
||
|
|
|
||
|
|
HWND m_hDialog;
|
||
|
|
HANDLE m_hProcess;
|
||
|
|
HANDLE m_hCacheFile;
|
||
|
|
DWORD m_dwRetVal;
|
||
|
|
|
||
|
|
HRESULT m_hDownloadResult;
|
||
|
|
BOOL m_fResultIn;
|
||
|
|
WJSTATE m_State;
|
||
|
|
BOOL m_fAborted;
|
||
|
|
BOOL m_fCalledAbort;
|
||
|
|
|
||
|
|
HRESULT m_hrInternal;
|
||
|
|
LPWSTR m_pwszMimeType;
|
||
|
|
LPWSTR m_pwszRedirectUrl;
|
||
|
|
LPWSTR m_pwszCacheFile;
|
||
|
|
BOOL m_fHtml;
|
||
|
|
BOOL m_fDownloadInited;
|
||
|
|
|
||
|
|
#define TEMPREADBUFFERSIZE 8192
|
||
|
|
CHAR* m_pTempBuffer;
|
||
|
|
|
||
|
|
BOOL m_bReading;
|
||
|
|
BOOL m_bStartedReadTimer;
|
||
|
|
|
||
|
|
public:
|
||
|
|
// constructors/destructors
|
||
|
|
CWebJit(WEBJIT_PARAM* pWebJitParam);
|
||
|
|
~CWebJit();
|
||
|
|
VOID ReleaseAll();
|
||
|
|
|
||
|
|
// IUnknown methods
|
||
|
|
STDMETHODIMP QueryInterface(REFIID riid, void ** ppv);
|
||
|
|
STDMETHODIMP_(ULONG) AddRef();
|
||
|
|
STDMETHODIMP_(ULONG) Release();
|
||
|
|
|
||
|
|
// IBindStatusCallback methods
|
||
|
|
STDMETHODIMP OnStartBinding(DWORD dwReserved, IBinding* pbinding);
|
||
|
|
STDMETHODIMP GetPriority(LONG* pnPriority)
|
||
|
|
{
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
STDMETHODIMP OnLowResource(DWORD dwReserved)
|
||
|
|
{
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
STDMETHODIMP OnProgress(ULONG ulProgress, ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR pwzStatusText);
|
||
|
|
STDMETHODIMP OnStopBinding(HRESULT hrResult, LPCWSTR szError);
|
||
|
|
STDMETHODIMP GetBindInfo(DWORD* pgrfBINDF, BINDINFO* pbindInfo);
|
||
|
|
STDMETHODIMP OnDataAvailable(DWORD grfBSCF, DWORD dwSize, FORMATETC *pfmtetc, STGMEDIUM* pstgmed);
|
||
|
|
STDMETHODIMP OnObjectAvailable(REFIID riid, IUnknown* punk);
|
||
|
|
|
||
|
|
// IAuthenticate methods
|
||
|
|
STDMETHODIMP Authenticate(HWND* phwnd, LPWSTR *pszUsername,LPWSTR *pszPassword);
|
||
|
|
|
||
|
|
// IInternetHostSecurityManager
|
||
|
|
|
||
|
|
STDMETHODIMP GetSecurityId( BYTE *pbSecurityId,
|
||
|
|
DWORD *pcbSecurityId,
|
||
|
|
DWORD_PTR dwReserved)
|
||
|
|
{
|
||
|
|
DEBUG_ENTER((DBG_DOWNLOAD,
|
||
|
|
Hresult,
|
||
|
|
"CWebJit::GetSecurityId",
|
||
|
|
"this=%#x",
|
||
|
|
this
|
||
|
|
));
|
||
|
|
|
||
|
|
HRESULT hr = E_FAIL;
|
||
|
|
|
||
|
|
IInternetSecurityManager *pSecMgr = g_pSecurityManager;
|
||
|
|
|
||
|
|
if (pSecMgr)
|
||
|
|
{
|
||
|
|
hr = pSecMgr->GetSecurityId(m_pwszUrl, pbSecurityId,
|
||
|
|
pcbSecurityId, dwReserved);
|
||
|
|
}
|
||
|
|
|
||
|
|
DEBUG_LEAVE(hr);
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP ProcessUrlAction( DWORD dwAction,
|
||
|
|
BYTE *pPolicy,
|
||
|
|
DWORD cbPolicy,
|
||
|
|
BYTE *pContext,
|
||
|
|
DWORD cbContext,
|
||
|
|
DWORD dwFlags,
|
||
|
|
DWORD dwReserved)
|
||
|
|
{
|
||
|
|
DEBUG_ENTER((DBG_DOWNLOAD,
|
||
|
|
Hresult,
|
||
|
|
"CWebJit::ProcessUrlAction",
|
||
|
|
"this=%#x, %d(%#x)",
|
||
|
|
this, dwAction, dwAction
|
||
|
|
));
|
||
|
|
|
||
|
|
HRESULT hr = E_FAIL;
|
||
|
|
|
||
|
|
IInternetSecurityManager *pSecMgr = g_pSecurityManager;
|
||
|
|
|
||
|
|
if (pSecMgr)
|
||
|
|
{
|
||
|
|
hr = pSecMgr->ProcessUrlAction(m_pwszUrl, dwAction, pPolicy,
|
||
|
|
cbPolicy, pContext, cbContext,
|
||
|
|
dwFlags, dwReserved);
|
||
|
|
}
|
||
|
|
|
||
|
|
DEBUG_LEAVE(hr);
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP QueryCustomPolicy( REFGUID guidKey,
|
||
|
|
BYTE **ppPolicy,
|
||
|
|
DWORD *pcbPolicy,
|
||
|
|
BYTE *pContext,
|
||
|
|
DWORD cbContext,
|
||
|
|
DWORD dwReserved)
|
||
|
|
{
|
||
|
|
DEBUG_ENTER((DBG_DOWNLOAD,
|
||
|
|
Hresult,
|
||
|
|
"CWebJit::ProcessUrlAction",
|
||
|
|
"this=%#x",
|
||
|
|
this
|
||
|
|
));
|
||
|
|
|
||
|
|
HRESULT hr = E_FAIL;
|
||
|
|
|
||
|
|
IInternetSecurityManager *pSecMgr = g_pSecurityManager;
|
||
|
|
|
||
|
|
if (pSecMgr)
|
||
|
|
{
|
||
|
|
hr = pSecMgr->QueryCustomPolicy(m_pwszUrl, guidKey, ppPolicy,
|
||
|
|
pcbPolicy, pContext, cbContext,
|
||
|
|
dwReserved);
|
||
|
|
}
|
||
|
|
|
||
|
|
DEBUG_LEAVE(hr);
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
VOID UpdateDownloadResult(HRESULT hr, BOOL fFromOnStopBinding);
|
||
|
|
VOID ProcessFile();
|
||
|
|
HRESULT SetupDownload();
|
||
|
|
HRESULT StartDownload();
|
||
|
|
HRESULT ExecFile();
|
||
|
|
HRESULT DisplayError(HRESULT hr, UINT nMsgError);
|
||
|
|
HRESULT VerifyMimeAndExtension();
|
||
|
|
BOOL NeedHostSecMgr();
|
||
|
|
HRESULT VerifyTrust(BOOL fUseHostSecMgr);
|
||
|
|
BOOL IsConnected(BOOL* pfIsOffline);
|
||
|
|
VOID ProcessAbort();
|
||
|
|
VOID Terminate(DWORD iRetVal);
|
||
|
|
HRESULT CanWebJit();
|
||
|
|
BOOL SetupWindow();
|
||
|
|
DWORD GetDownloadSpeed();
|
||
|
|
BOOL UpdateProgressUI();
|
||
|
|
VOID UpdateStatusString();
|
||
|
|
VOID ReadData();
|
||
|
|
|
||
|
|
BOOL IsReadTimerStarted()
|
||
|
|
{
|
||
|
|
return m_bStartedReadTimer;
|
||
|
|
}
|
||
|
|
|
||
|
|
VOID SetReadTimerStarted()
|
||
|
|
{
|
||
|
|
m_bStartedReadTimer = TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
VOID SetState(WJSTATE state)
|
||
|
|
{
|
||
|
|
m_State = state;
|
||
|
|
}
|
||
|
|
|
||
|
|
WJSTATE GetState()
|
||
|
|
{
|
||
|
|
return m_State;
|
||
|
|
}
|
||
|
|
|
||
|
|
HANDLE GetProcessHandle()
|
||
|
|
{
|
||
|
|
return m_hProcess;
|
||
|
|
}
|
||
|
|
|
||
|
|
VOID SetWindowHandle(HWND hDlg)
|
||
|
|
{
|
||
|
|
m_hDialog = hDlg;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD GetResult()
|
||
|
|
{
|
||
|
|
return m_dwRetVal;
|
||
|
|
}
|
||
|
|
|
||
|
|
LPCWSTR GetComponentIdName()
|
||
|
|
{
|
||
|
|
return m_pwszComponentId;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL IsDownloadInited()
|
||
|
|
{
|
||
|
|
return m_fDownloadInited;
|
||
|
|
}
|
||
|
|
|
||
|
|
VOID SetDownloadInited()
|
||
|
|
{
|
||
|
|
m_fDownloadInited = TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL IsAborted()
|
||
|
|
{
|
||
|
|
return m_fAborted;
|
||
|
|
}
|
||
|
|
|
||
|
|
// to delayload InitCommonControlsEx
|
||
|
|
HRESULT InitCC(void);
|
||
|
|
#define DELAYCCAPI(_fn, _args, _nargs, ret) \
|
||
|
|
ret _fn _args \
|
||
|
|
{ \
|
||
|
|
HRESULT hres; \
|
||
|
|
ret retval = 0; \
|
||
|
|
hres = InitCC(); \
|
||
|
|
if (SUCCEEDED(hres) && (_pfn##_fn)) \
|
||
|
|
{ \
|
||
|
|
retval = _pfn##_fn _nargs; \
|
||
|
|
} \
|
||
|
|
return retval; \
|
||
|
|
} \
|
||
|
|
ret (STDAPICALLTYPE* _pfn##_fn) _args;
|
||
|
|
|
||
|
|
DELAYCCAPI(InitCommonControlsEx,
|
||
|
|
(LPINITCOMMONCONTROLSEX lpInitCtrls),
|
||
|
|
(lpInitCtrls),
|
||
|
|
BOOL);
|
||
|
|
BOOL CWebJit::InitCommonControlsForWebJit();
|
||
|
|
|
||
|
|
// to delayload WinVerifyTrust
|
||
|
|
HRESULT Init(void);
|
||
|
|
#define DELAYWVTAPI(_fn, _args, _nargs) \
|
||
|
|
HRESULT _fn _args { \
|
||
|
|
HRESULT hres = Init(); \
|
||
|
|
if (SUCCEEDED(hres)) { \
|
||
|
|
hres = _pfn##_fn _nargs; \
|
||
|
|
} \
|
||
|
|
return hres; } \
|
||
|
|
HRESULT (STDAPICALLTYPE* _pfn##_fn) _args;
|
||
|
|
|
||
|
|
DELAYWVTAPI(WinVerifyTrust,
|
||
|
|
(HWND hwnd, GUID * ActionID, LPVOID ActionData),
|
||
|
|
(hwnd, ActionID, ActionData));
|
||
|
|
};
|
||
|
|
|
||
|
|
INT_PTR CALLBACK WebJitProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam);
|
||
|
|
|
||
|
|
#endif _WEBJIT_H_
|