|
|
// OcarptMain.h : Declaration of the COcarptMain
#ifndef __OCARPTMAIN_H_
#define __OCARPTMAIN_H_
#include "resource.h" // main symbols
#include <atlctl.h>
#include <time.h>
#include "inetupld.h"
#include <exdisp.h>
#include <shlguid.h>
#include <strsafe.h>
#define _USE_WINHTTP 1
#ifdef _USE_WINHTTP
#include <winhttp.h>
#include <winhttpi.h>
#define MAX_URL_LENGTH 2176
#else
#include <wininet.h>
#define MAX_URL_LENGTH INTERNET_MAX_URL_LENGTH
#endif // _USE_WINHTTP
#define ARRAYSIZE(a) (sizeof(a)/sizeof(a[0]))
/////////////////////////////////////////////////////////////////////////////
// COcarptMain
class ATL_NO_VTABLE COcarptMain : public CComObjectRootEx<CComSingleThreadModel>, public IDispatchImpl<IOcarptMain, &IID_IOcarptMain, &LIBID_OCARPTLib>, public CComControl<COcarptMain>, public IPersistStreamInitImpl<COcarptMain>, public IOleControlImpl<COcarptMain>, public IOleObjectImpl<COcarptMain>, public IOleInPlaceActiveObjectImpl<COcarptMain>, public IViewObjectExImpl<COcarptMain>, public IOleInPlaceObjectWindowlessImpl<COcarptMain>, public IPersistStorageImpl<COcarptMain>, public ISpecifyPropertyPagesImpl<COcarptMain>, public IQuickActivateImpl<COcarptMain>, public IDataObjectImpl<COcarptMain>, public IProvideClassInfo2Impl<&CLSID_OcarptMain, NULL, &LIBID_OCARPTLib>, public CComCoClass<COcarptMain, &CLSID_OcarptMain>, public IObjectSafetyImpl<COcarptMain, INTERFACESAFE_FOR_UNTRUSTED_CALLER |INTERFACESAFE_FOR_UNTRUSTED_DATA>
{public: COcarptMain() { m_pUploadFile = NULL; m_b_SetSiteCalled = FALSE; }
DECLARE_REGISTRY_RESOURCEID(IDR_OCARPTMAIN)
DECLARE_PROTECT_FINAL_CONSTRUCT()
BEGIN_COM_MAP(COcarptMain) COM_INTERFACE_ENTRY(IOcarptMain) COM_INTERFACE_ENTRY(IDispatch) COM_INTERFACE_ENTRY(IViewObjectEx) COM_INTERFACE_ENTRY(IViewObject2) COM_INTERFACE_ENTRY(IViewObject) COM_INTERFACE_ENTRY(IOleInPlaceObjectWindowless) COM_INTERFACE_ENTRY(IOleInPlaceObject) COM_INTERFACE_ENTRY2(IOleWindow, IOleInPlaceObjectWindowless) COM_INTERFACE_ENTRY(IOleInPlaceActiveObject) COM_INTERFACE_ENTRY(IOleControl) COM_INTERFACE_ENTRY(IOleObject) COM_INTERFACE_ENTRY(IPersistStreamInit) COM_INTERFACE_ENTRY2(IPersist, IPersistStreamInit) COM_INTERFACE_ENTRY(ISpecifyPropertyPages) COM_INTERFACE_ENTRY(IQuickActivate) COM_INTERFACE_ENTRY(IPersistStorage) COM_INTERFACE_ENTRY(IDataObject) COM_INTERFACE_ENTRY(IProvideClassInfo) COM_INTERFACE_ENTRY(IProvideClassInfo2) COM_INTERFACE_ENTRY(IObjectSafety)END_COM_MAP()
BEGIN_PROP_MAP(COcarptMain) PROP_DATA_ENTRY("_cx", m_sizeExtent.cx, VT_UI4) PROP_DATA_ENTRY("_cy", m_sizeExtent.cy, VT_UI4) // Example entries
// PROP_ENTRY("Property Description", dispid, clsid)
// PROP_PAGE(CLSID_StockColorPage)
END_PROP_MAP()
BEGIN_MSG_MAP(COcarptMain) CHAIN_MSG_MAP(CComControl<COcarptMain>) DEFAULT_REFLECTION_HANDLER()END_MSG_MAP()// Handler prototypes:
// LRESULT MessageHandler(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
// LRESULT CommandHandler(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled);
// LRESULT NotifyHandler(int idCtrl, LPNMHDR pnmh, BOOL& bHandled);
// IViewObjectEx
DECLARE_VIEW_STATUS(VIEWSTATUS_SOLIDBKGND | VIEWSTATUS_OPAQUE)
// IOcarptMain
public: STDMETHOD(RetrieveFileContents)(/*[in]*/BSTR *FileName, /*[out,retval]*/ VARIANT *pvContents); STDMETHOD(ValidateDump)(/*[in]*/ BSTR *FileName, /*[out,retval]*/VARIANT *Result); STDMETHOD(Browse)(/*[in]*/ BSTR *pbstrTitle, /*[in]*/BSTR *Lang, /*[out,retval]*/ VARIANT *Path); STDMETHOD(Search)(/*[out,retval]*/ VARIANT *pvFileList); STDMETHOD(Upload)(/*[in]*/ BSTR *SourceFile, /*[in]*/BSTR *DestFile, /*[in]*/BSTR *Langage, /*[in]*/ BSTR *OptionCode, /*[in]*/ int ConvertToMini, /*[out,retval]*/ VARIANT *ReturnCode); STDMETHOD(GetUploadStatus)(/*[out,retval]*/ VARIANT *PercentDone); STDMETHOD(GetUploadResult)(/*[out,retval]*/ VARIANT *UploadResult); STDMETHOD(CancelUpload)(/*[out,retval]*/ VARIANT *ReturnCode);
HRESULT OnDraw(ATL_DRAWINFO& di) { RECT& rc = *(RECT*)di.prcBounds; Rectangle(di.hdcDraw, rc.left, rc.top, rc.right, rc.bottom);
SetTextAlign(di.hdcDraw, TA_CENTER|TA_BASELINE); LPCTSTR pszText = _T(""); TextOut(di.hdcDraw, (rc.left + rc.right) / 2, (rc.top + rc.bottom) / 2, pszText, lstrlen(pszText));
return S_OK; }
STDMETHODIMP SetClientSite (IOleClientSite *pClientSite) { _spUnkSite = pClientSite; m_b_SetSiteCalled = TRUE; return S_OK; }
STDMETHODIMP GetSite (REFIID riid, LPVOID* ppvSite) { if (m_b_SetSiteCalled) return _spUnkSite->QueryInterface(riid,ppvSite); else return E_FAIL; }
bool InApprovedDomain() { TCHAR ourUrl[MAX_URL_LENGTH];
return true; if (!GetOurUrl(ourUrl, sizeof ourUrl)) return false; return IsApprovedDomain(ourUrl); }
bool GetOurUrl(TCHAR* pszURL, int cbBuf) { HRESULT hr; CComPtr<IServiceProvider> spSrvProv; CComPtr<IWebBrowser2> spWebBrowser;
hr = GetSite(IID_IServiceProvider, (void**)&spSrvProv); if (FAILED(hr)) return false;
hr = spSrvProv->QueryService(SID_SWebBrowserApp, IID_IWebBrowser2, (void**)&spWebBrowser); if (FAILED(hr)) return false;
CComBSTR bstrURL; if (FAILED(spWebBrowser->get_LocationURL(&bstrURL))) return false;
#ifdef UNICODE
StringCbCopy(pszURL, cbBuf, bstrURL); #else
WideCharToMultiByte(CP_ACP, 0, bstrURL, -1, pszURL, cbBuf, NULL, NULL); #endif
return true; }
bool IsApprovedDomain(TCHAR* ourUrl) { // Only allow http access.
// You can change this to allow file:// access.
//
if (GetScheme(ourUrl) != INTERNET_SCHEME_HTTPS) return false;
TCHAR ourDomain[256]; if (!GetDomain(ourUrl, ourDomain, sizeof(ourDomain))) return false;
for (int i = 0; i < ARRAYSIZE(_approvedDomains); i++) { if (MatchDomains(const_cast<TCHAR*>(_approvedDomains[i]), ourDomain)) { return true; } }
return false; }
INTERNET_SCHEME GetScheme(TCHAR* url) { TCHAR buf[32]; URL_COMPONENTS uc; ZeroMemory(&uc, sizeof uc);
uc.dwStructSize = sizeof uc; uc.lpszScheme = buf; uc.dwSchemeLength = sizeof buf;
#ifdef _USE_WINHTTP
if (WinHttpCrackUrl(url, lstrlen(url), ICU_DECODE, &uc))#else
if (InternetCrackUrl(url, lstrlen(url), ICU_DECODE, &uc))#endif
return uc.nScheme; else return INTERNET_SCHEME_UNKNOWN; }
bool GetDomain(TCHAR* url, TCHAR* buf, int cbBuf) { URL_COMPONENTS uc; ZeroMemory(&uc, sizeof uc);
uc.dwStructSize = sizeof uc; uc.lpszHostName = buf; uc.dwHostNameLength = cbBuf;
#ifdef _USE_WINHTTP
return (WinHttpCrackUrl(url, lstrlen(url), ICU_DECODE, &uc)#else
return (InternetCrackUrl(url, lstrlen(url), ICU_DECODE, &uc)#endif
!= FALSE); }
// Return if ourDomain is within approvedDomain.
// approvedDomain must either match ourDomain
// or be a suffix preceded by a dot.
//
bool MatchDomains(TCHAR* approvedDomain, TCHAR* ourDomain) { int apDomLen = lstrlen(approvedDomain); int ourDomLen = lstrlen(ourDomain);
if (apDomLen > ourDomLen) return false;
if (lstrcmpi(ourDomain+ourDomLen-apDomLen, approvedDomain) != 0) return false;
if (apDomLen == ourDomLen) return true;
if (ourDomain[ourDomLen - apDomLen - 1] == '.') return true;
return false; }
void GetFileHandle(wchar_t *FileName, HANDLE *hFile); BOOL DeleteTempDir(wchar_t *TempDirectory,wchar_t *FileName,wchar_t *CabName); BOOL CreateTempDir(wchar_t *TempDirectory); BOOL ConvertFullDumpInternal (BSTR *Source, BSTR *Destination); DWORD GetResponseURL(wchar_t *HostName, wchar_t *RemoteFileName, BOOL fFullDump, wchar_t *ResponseURL);
private: static TCHAR* _approvedDomains[8];
private:
IOleClientSite *_spUnkSite; BOOL m_b_SetSiteCalled; POCA_UPLOADFILE m_pUploadFile; void FormatMiniDate(SYSTEMTIME *pTimeStruct, CComBSTR &strDate); void FormatDate(SYSTEMTIME *pTimeStruct, CComBSTR &strDate); void FormatDate(tm *pTimeStruct, CComBSTR &strDate); BOOL FindFullDumps( BSTR *FileLists); BOOL FindMiniDumps( BSTR *FileLists); BOOL ValidMiniDump(BSTR FileName); BOOL ValidMiniDump(LPCTSTR FileName);};#endif //__OCARPTMAIN_H_
|
