Add files via upload

LegacyUpdate/LegacyUpdateCtrl.h

@@ -0,0 +1,114 @@
+#pragma once
+
+// LegacyUpdateCtrl.h : Declaration of the CLegacyUpdateCtrl ActiveX Control class.
+
+// CLegacyUpdateCtrl : See LegacyUpdateCtrl.cpp for implementation.
+
+#include <atlctl.h>
+#include <MsHTML.h>
+#include "resource.h"
+#include "LegacyUpdate_i.h"
+
+// CLegacyUpdateCtrl
+class ATL_NO_VTABLE CLegacyUpdateCtrl :
+	public CComObjectRootEx<CComSingleThreadModel>,
+	public IDispatchImpl<ILegacyUpdateCtrl, &IID_ILegacyUpdateCtrl, &LIBID_LegacyUpdateLib, /*wMajor =*/ 1, /*wMinor =*/ 0>,
+	public IPersistStreamInitImpl<CLegacyUpdateCtrl>,
+	public IOleControlImpl<CLegacyUpdateCtrl>,
+	public IOleObjectImpl<CLegacyUpdateCtrl>,
+	public IOleInPlaceActiveObjectImpl<CLegacyUpdateCtrl>,
+	public IViewObjectExImpl<CLegacyUpdateCtrl>,
+	public IOleInPlaceObjectWindowlessImpl<CLegacyUpdateCtrl>,
+	public ISupportErrorInfo,
+	public IObjectWithSiteImpl<CLegacyUpdateCtrl>,
+	public IPersistStorageImpl<CLegacyUpdateCtrl>,
+	public IQuickActivateImpl<CLegacyUpdateCtrl>,
+	public IProvideClassInfo2Impl<&CLSID_LegacyUpdateCtrl, NULL, &LIBID_LegacyUpdateLib>,
+	public CComCoClass<CLegacyUpdateCtrl, &CLSID_LegacyUpdateCtrl>,
+	public CComControl<CLegacyUpdateCtrl> {
+private:
+	CComPtr<IElevationHelper> m_elevatedHelper;
+	CComPtr<IElevationHelper> m_nonElevatedHelper;
+
+public:
+	CLegacyUpdateCtrl() {
+		m_bWindowOnly = TRUE;
+	}
+
+	DECLARE_OLEMISC_STATUS(
+		OLEMISC_INVISIBLEATRUNTIME |
+		OLEMISC_ACTIVATEWHENVISIBLE |
+		OLEMISC_SETCLIENTSITEFIRST |
+		OLEMISC_INSIDEOUT |
+		OLEMISC_CANTLINKINSIDE |
+		OLEMISC_RECOMPOSEONRESIZE
+	)
+
+	DECLARE_REGISTRY_RESOURCEID(IDR_LEGACYUPDATECTRL)
+
+	BEGIN_COM_MAP(CLegacyUpdateCtrl)
+		COM_INTERFACE_ENTRY(ILegacyUpdateCtrl)
+		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(ISupportErrorInfo)
+		COM_INTERFACE_ENTRY(IQuickActivate)
+		COM_INTERFACE_ENTRY(IPersistStorage)
+		COM_INTERFACE_ENTRY(IProvideClassInfo)
+		COM_INTERFACE_ENTRY(IProvideClassInfo2)
+	END_COM_MAP()
+
+	BEGIN_PROP_MAP(CLegacyUpdateCtrl)
+	END_PROP_MAP()
+
+	BEGIN_MSG_MAP(CLegacyUpdateCtrl)
+		CHAIN_MSG_MAP(CComControl<CLegacyUpdateCtrl>)
+		DEFAULT_REFLECTION_HANDLER()
+	END_MSG_MAP()
+
+	// ISupportsErrorInfo
+	STDMETHOD(InterfaceSupportsErrorInfo)(REFIID riid) {
+		return IsEqualGUID(riid, IID_ILegacyUpdateCtrl) ? S_OK : S_FALSE;
+	}
+
+	// IViewObjectEx
+	DECLARE_VIEW_STATUS(VIEWSTATUS_SOLIDBKGND | VIEWSTATUS_OPAQUE)
+
+	// ILegacyUpdateCtrl
+	DECLARE_PROTECT_FINAL_CONSTRUCT()
+
+	HRESULT FinalConstruct() { return S_OK; }
+	void FinalRelease() {}
+
+private:
+	IHTMLDocument2 *GetHTMLDocument();
+	HWND GetIEWindowHWND();
+	BOOL IsPermitted();
+
+public:
+	STDMETHODIMP CheckControl(VARIANT_BOOL *retval);
+	STDMETHODIMP MessageForHresult(LONG inHresult, BSTR *retval);
+	STDMETHODIMP GetOSVersionInfo(OSVersionField osField, LONG systemMetric, VARIANT *retval);
+	STDMETHODIMP RequestElevation();
+	STDMETHODIMP CreateObject(BSTR progID, IDispatch **retval);
+	STDMETHODIMP GetUserType(UserType *retval);
+	STDMETHODIMP get_IsRebootRequired(VARIANT_BOOL *retval);
+	STDMETHODIMP get_IsWindowsUpdateDisabled(VARIANT_BOOL *retval);
+	STDMETHODIMP RebootIfRequired(void);
+	STDMETHODIMP ViewWindowsUpdateLog(void);
+	STDMETHODIMP OpenWindowsUpdateSettings(void);
+	STDMETHODIMP get_IsUsingWsusServer(VARIANT_BOOL *retval);
+	STDMETHODIMP get_WsusServerUrl(BSTR *retval);
+	STDMETHODIMP get_WsusStatusServerUrl(BSTR *retval);
+};
+
+OBJECT_ENTRY_AUTO(__uuidof(LegacyUpdateCtrl), CLegacyUpdateCtrl)