Add files via upload

LegacyUpdate/ElevationHelper.cpp

@@ -0,0 +1,74 @@
+// ElevationHelper.cpp : Implementation of CElevationHelper
+#include "stdafx.h"
+#include "ElevationHelper.h"
+#include "Utils.h"
+#include <strsafe.h>
+
+const BSTR permittedProgIDs[] = {
+	L"Microsoft.Update.",
+	NULL
+};
+const int permittedProgIDsMax = 1;
+
+BOOL ProgIDIsPermitted(PWSTR progID) {
+	if (progID == NULL) {
+		return FALSE;
+	}
+
+	for (int i = 0; i < permittedProgIDsMax; i++) {
+		if (wcsncmp(progID, permittedProgIDs[i], wcslen(permittedProgIDs[i])) == 0) {
+			return TRUE;
+		}
+	}
+	return FALSE;
+}
+
+STDMETHODIMP CoCreateInstanceAsAdmin(HWND hwnd, __in REFCLSID rclsid, __in REFIID riid, __deref_out void **ppv) {
+	WCHAR clsidString[45];
+	StringFromGUID2(rclsid, clsidString, ARRAYSIZE(clsidString));
+
+	WCHAR monikerName[75];
+	HRESULT hr = StringCchPrintf(monikerName, ARRAYSIZE(monikerName), L"Elevation:Administrator!new:%s", clsidString);
+	if (!SUCCEEDED(hr)) {
+		return hr;
+	}
+
+	BIND_OPTS3 bindOpts;
+	memset(&bindOpts, 0, sizeof(bindOpts));
+	bindOpts.cbStruct = sizeof(bindOpts);
+	bindOpts.hwnd = hwnd;
+	bindOpts.dwClassContext = CLSCTX_LOCAL_SERVER;
+	return CoGetObject(monikerName, &bindOpts, riid, ppv);
+}
+
+HRESULT CElevationHelper::CreateObject(BSTR progID, IDispatch **retval) {
+	if (progID == NULL) {
+		return E_INVALIDARG;
+	}
+
+	HRESULT hr = S_OK;
+	CComPtr<IDispatch> object;
+	if (!ProgIDIsPermitted(progID)) {
+		hr = E_ACCESSDENIED;
+		goto end;
+	}
+
+	CLSID clsid;
+	hr = CLSIDFromProgID(progID, &clsid);
+	if (!SUCCEEDED(hr)) {
+		goto end;
+	}
+
+	hr = object.CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER);
+	if (!SUCCEEDED(hr)) {
+		goto end;
+	}
+
+	*retval = object.Detach();
+
+end:
+	if (!SUCCEEDED(hr)) {
+		TRACE("CreateObject(%ls) failed: %ls\n", progID, GetMessageForHresult(hr));
+	}
+	return hr;
+}