| /* |
| * QEMU Guest Agent win32 VSS Provider installer |
| * |
| * Copyright Hitachi Data Systems Corp. 2013 |
| * |
| * Authors: |
| * Tomoki Sekiyama <tomoki.sekiyama@hds.com> |
| * |
| * This work is licensed under the terms of the GNU GPL, version 2 or later. |
| * See the COPYING file in the top-level directory. |
| */ |
| |
| #include <stdio.h> |
| #include <string.h> |
| |
| #include "vss-common.h" |
| #include "inc/win2003/vscoordint.h" |
| |
| #include <comadmin.h> |
| #include <wbemidl.h> |
| #include <comdef.h> |
| #include <comutil.h> |
| |
| extern HINSTANCE g_hinstDll; |
| |
| const GUID CLSID_COMAdminCatalog = { 0xF618C514, 0xDFB8, 0x11d1, |
| {0xA2, 0xCF, 0x00, 0x80, 0x5F, 0xC7, 0x92, 0x35} }; |
| const GUID IID_ICOMAdminCatalog = { 0xDD662187, 0xDFC2, 0x11d1, |
| {0xA2, 0xCF, 0x00, 0x80, 0x5F, 0xC7, 0x92, 0x35} }; |
| const GUID CLSID_WbemLocator = { 0x4590f811, 0x1d3a, 0x11d0, |
| {0x89, 0x1f, 0x00, 0xaa, 0x00, 0x4b, 0x2e, 0x24} }; |
| const GUID IID_IWbemLocator = { 0xdc12a687, 0x737f, 0x11cf, |
| {0x88, 0x4d, 0x00, 0xaa, 0x00, 0x4b, 0x2e, 0x24} }; |
| |
| void errmsg(DWORD err, const char *text) |
| { |
| /* |
| * `text' contains function call statement when errmsg is called via chk(). |
| * To make error message more readable, we cut off the text after '('. |
| * If text doesn't contains '(', negative precision is given, which is |
| * treated as though it were missing. |
| */ |
| char *msg = NULL, *nul = strchr(text, '('); |
| int len = nul ? nul - text : -1; |
| |
| FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | |
| FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, |
| NULL, err, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), |
| (char *)&msg, 0, NULL); |
| fprintf(stderr, "%.*s. (Error: %lx) %s\n", len, text, err, msg); |
| LocalFree(msg); |
| } |
| |
| static void errmsg_dialog(DWORD err, const char *text, const char *opt = "") |
| { |
| char *msg, buf[512]; |
| |
| FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | |
| FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, |
| NULL, err, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), |
| (char *)&msg, 0, NULL); |
| snprintf(buf, sizeof(buf), "%s%s. (Error: %lx) %s", text, opt, err, msg); |
| MessageBox(NULL, buf, "Error from " QGA_PROVIDER_NAME, MB_OK|MB_ICONERROR); |
| LocalFree(msg); |
| } |
| |
| #define _chk(hr, status, msg, err_label) \ |
| do { \ |
| hr = (status); \ |
| if (FAILED(hr)) { \ |
| errmsg(hr, msg); \ |
| goto err_label; \ |
| } \ |
| } while (0) |
| |
| #define chk(status) _chk(hr, status, "Failed to " #status, out) |
| |
| void __stdcall _com_issue_error(HRESULT hr) |
| { |
| errmsg(hr, "Unexpected error in COM"); |
| } |
| |
| template<class T> |
| HRESULT put_Value(ICatalogObject *pObj, LPCWSTR name, T val) |
| { |
| return pObj->put_Value(_bstr_t(name), _variant_t(val)); |
| } |
| |
| /* Lookup Administrators group name from winmgmt */ |
| static HRESULT GetAdminName(_bstr_t *name) |
| { |
| HRESULT hr; |
| COMPointer<IWbemLocator> pLoc; |
| COMPointer<IWbemServices> pSvc; |
| COMPointer<IEnumWbemClassObject> pEnum; |
| COMPointer<IWbemClassObject> pWobj; |
| ULONG returned; |
| _variant_t var; |
| |
| chk(CoCreateInstance(CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER, |
| IID_IWbemLocator, (LPVOID *)pLoc.replace())); |
| chk(pLoc->ConnectServer(_bstr_t(L"ROOT\\CIMV2"), NULL, NULL, NULL, |
| 0, 0, 0, pSvc.replace())); |
| chk(CoSetProxyBlanket(pSvc, RPC_C_AUTHN_WINNT, RPC_C_AUTHZ_NONE, |
| NULL, RPC_C_AUTHN_LEVEL_CALL, |
| RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE)); |
| chk(pSvc->ExecQuery(_bstr_t(L"WQL"), |
| _bstr_t(L"select * from Win32_Account where " |
| "SID='S-1-5-32-544' and localAccount=TRUE"), |
| WBEM_FLAG_RETURN_IMMEDIATELY | WBEM_FLAG_FORWARD_ONLY, |
| NULL, pEnum.replace())); |
| if (!pEnum) { |
| hr = E_FAIL; |
| errmsg(hr, "Failed to query for Administrators"); |
| goto out; |
| } |
| chk(pEnum->Next(WBEM_INFINITE, 1, pWobj.replace(), &returned)); |
| if (returned == 0) { |
| hr = E_FAIL; |
| errmsg(hr, "No Administrators found"); |
| goto out; |
| } |
| |
| chk(pWobj->Get(_bstr_t(L"Name"), 0, &var, 0, 0)); |
| try { |
| *name = var; |
| } catch(...) { |
| hr = E_FAIL; |
| errmsg(hr, "Failed to get name of Administrators"); |
| goto out; |
| } |
| |
| out: |
| return hr; |
| } |
| |
| /* Find and iterate QGA VSS provider in COM+ Application Catalog */ |
| static HRESULT QGAProviderFind( |
| HRESULT (*found)(ICatalogCollection *, int, void *), void *arg) |
| { |
| HRESULT hr; |
| COMInitializer initializer; |
| COMPointer<IUnknown> pUnknown; |
| COMPointer<ICOMAdminCatalog> pCatalog; |
| COMPointer<ICatalogCollection> pColl; |
| COMPointer<ICatalogObject> pObj; |
| _variant_t var; |
| long i, n; |
| |
| chk(CoCreateInstance(CLSID_COMAdminCatalog, NULL, CLSCTX_INPROC_SERVER, |
| IID_IUnknown, (void **)pUnknown.replace())); |
| chk(pUnknown->QueryInterface(IID_ICOMAdminCatalog, |
| (void **)pCatalog.replace())); |
| chk(pCatalog->GetCollection(_bstr_t(L"Applications"), |
| (IDispatch **)pColl.replace())); |
| chk(pColl->Populate()); |
| |
| chk(pColl->get_Count(&n)); |
| for (i = n - 1; i >= 0; i--) { |
| chk(pColl->get_Item(i, (IDispatch **)pObj.replace())); |
| chk(pObj->get_Value(_bstr_t(L"Name"), &var)); |
| if (var == _variant_t(QGA_PROVIDER_LNAME)) { |
| if (FAILED(found(pColl, i, arg))) { |
| goto out; |
| } |
| } |
| } |
| chk(pColl->SaveChanges(&n)); |
| |
| out: |
| return hr; |
| } |
| |
| /* Count QGA VSS provider in COM+ Application Catalog */ |
| static HRESULT QGAProviderCount(ICatalogCollection *coll, int i, void *arg) |
| { |
| (*(int *)arg)++; |
| return S_OK; |
| } |
| |
| /* Remove QGA VSS provider from COM+ Application Catalog Collection */ |
| static HRESULT QGAProviderRemove(ICatalogCollection *coll, int i, void *arg) |
| { |
| HRESULT hr; |
| |
| fprintf(stderr, "Removing COM+ Application: %s\n", QGA_PROVIDER_NAME); |
| chk(coll->Remove(i)); |
| out: |
| return hr; |
| } |
| |
| /* Unregister this module from COM+ Applications Catalog */ |
| STDAPI COMUnregister(void) |
| { |
| HRESULT hr; |
| |
| DllUnregisterServer(); |
| chk(QGAProviderFind(QGAProviderRemove, NULL)); |
| out: |
| return hr; |
| } |
| |
| /* Register this module to COM+ Applications Catalog */ |
| STDAPI COMRegister(void) |
| { |
| HRESULT hr; |
| COMInitializer initializer; |
| COMPointer<IUnknown> pUnknown; |
| COMPointer<ICOMAdminCatalog> pCatalog; |
| COMPointer<ICatalogCollection> pApps, pRoles, pUsersInRole; |
| COMPointer<ICatalogObject> pObj; |
| long n; |
| _bstr_t name; |
| _variant_t key; |
| CHAR dllPath[MAX_PATH], tlbPath[MAX_PATH]; |
| bool unregisterOnFailure = false; |
| int count = 0; |
| |
| if (!g_hinstDll) { |
| errmsg(E_FAIL, "Failed to initialize DLL"); |
| return E_FAIL; |
| } |
| |
| chk(QGAProviderFind(QGAProviderCount, (void *)&count)); |
| if (count) { |
| errmsg(E_ABORT, "QGA VSS Provider is already installed"); |
| return E_ABORT; |
| } |
| |
| chk(CoCreateInstance(CLSID_COMAdminCatalog, NULL, CLSCTX_INPROC_SERVER, |
| IID_IUnknown, (void **)pUnknown.replace())); |
| chk(pUnknown->QueryInterface(IID_ICOMAdminCatalog, |
| (void **)pCatalog.replace())); |
| |
| /* Install COM+ Component */ |
| |
| chk(pCatalog->GetCollection(_bstr_t(L"Applications"), |
| (IDispatch **)pApps.replace())); |
| chk(pApps->Populate()); |
| chk(pApps->Add((IDispatch **)&pObj)); |
| chk(put_Value(pObj, L"Name", QGA_PROVIDER_LNAME)); |
| chk(put_Value(pObj, L"Description", QGA_PROVIDER_LNAME)); |
| chk(put_Value(pObj, L"ApplicationAccessChecksEnabled", true)); |
| chk(put_Value(pObj, L"Authentication", short(6))); |
| chk(put_Value(pObj, L"AuthenticationCapability", short(2))); |
| chk(put_Value(pObj, L"ImpersonationLevel", short(2))); |
| chk(pApps->SaveChanges(&n)); |
| |
| /* The app should be deleted if something fails after SaveChanges */ |
| unregisterOnFailure = true; |
| |
| chk(pObj->get_Key(&key)); |
| |
| if (!GetModuleFileName(g_hinstDll, dllPath, sizeof(dllPath))) { |
| hr = HRESULT_FROM_WIN32(GetLastError()); |
| errmsg(hr, "GetModuleFileName failed"); |
| goto out; |
| } |
| n = strlen(dllPath); |
| if (n < 3) { |
| hr = E_FAIL; |
| errmsg(hr, "Failed to lookup dll"); |
| goto out; |
| } |
| strcpy(tlbPath, dllPath); |
| strcpy(tlbPath+n-3, "tlb"); |
| fprintf(stderr, "Registering " QGA_PROVIDER_NAME ":\n"); |
| fprintf(stderr, " %s\n", dllPath); |
| fprintf(stderr, " %s\n", tlbPath); |
| if (!PathFileExists(tlbPath)) { |
| hr = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND); |
| errmsg(hr, "Failed to lookup tlb"); |
| goto out; |
| } |
| |
| chk(pCatalog->InstallComponent(_bstr_t(QGA_PROVIDER_LNAME), |
| _bstr_t(dllPath), _bstr_t(tlbPath), |
| _bstr_t(""))); |
| |
| /* Setup roles of the applicaion */ |
| |
| chk(pApps->GetCollection(_bstr_t(L"Roles"), key, |
| (IDispatch **)pRoles.replace())); |
| chk(pRoles->Populate()); |
| chk(pRoles->Add((IDispatch **)pObj.replace())); |
| chk(put_Value(pObj, L"Name", L"Administrators")); |
| chk(put_Value(pObj, L"Description", L"Administrators group")); |
| chk(pRoles->SaveChanges(&n)); |
| chk(pObj->get_Key(&key)); |
| |
| /* Setup users in the role */ |
| |
| chk(pRoles->GetCollection(_bstr_t(L"UsersInRole"), key, |
| (IDispatch **)pUsersInRole.replace())); |
| chk(pUsersInRole->Populate()); |
| |
| chk(pUsersInRole->Add((IDispatch **)pObj.replace())); |
| chk(GetAdminName(&name)); |
| chk(put_Value(pObj, L"User", _bstr_t(".\\") + name)); |
| |
| chk(pUsersInRole->Add((IDispatch **)pObj.replace())); |
| chk(put_Value(pObj, L"User", L"SYSTEM")); |
| chk(pUsersInRole->SaveChanges(&n)); |
| |
| out: |
| if (unregisterOnFailure && FAILED(hr)) { |
| COMUnregister(); |
| } |
| |
| return hr; |
| } |
| |
| |
| static BOOL CreateRegistryKey(LPCTSTR key, LPCTSTR value, LPCTSTR data) |
| { |
| HKEY hKey; |
| LONG ret; |
| DWORD size; |
| |
| ret = RegCreateKeyEx(HKEY_CLASSES_ROOT, key, 0, NULL, |
| REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hKey, NULL); |
| if (ret != ERROR_SUCCESS) { |
| goto out; |
| } |
| |
| if (data != NULL) { |
| size = strlen(data) + 1; |
| } else { |
| size = 0; |
| } |
| |
| ret = RegSetValueEx(hKey, value, 0, REG_SZ, (LPBYTE)data, size); |
| RegCloseKey(hKey); |
| |
| out: |
| if (ret != ERROR_SUCCESS) { |
| /* As we cannot printf within DllRegisterServer(), show a dialog. */ |
| errmsg_dialog(ret, "Cannot add registry", key); |
| return FALSE; |
| } |
| return TRUE; |
| } |
| |
| /* Register this dll as a VSS provider */ |
| STDAPI DllRegisterServer(void) |
| { |
| COMInitializer initializer; |
| COMPointer<IVssAdmin> pVssAdmin; |
| HRESULT hr = E_FAIL; |
| char dllPath[MAX_PATH]; |
| char key[256]; |
| |
| if (!g_hinstDll) { |
| errmsg_dialog(hr, "Module instance is not available"); |
| goto out; |
| } |
| |
| /* Add this module to registery */ |
| |
| sprintf(key, "CLSID\\%s", g_szClsid); |
| if (!CreateRegistryKey(key, NULL, g_szClsid)) { |
| goto out; |
| } |
| |
| if (!GetModuleFileName(g_hinstDll, dllPath, sizeof(dllPath))) { |
| errmsg_dialog(GetLastError(), "GetModuleFileName failed"); |
| goto out; |
| } |
| |
| sprintf(key, "CLSID\\%s\\InprocServer32", g_szClsid); |
| if (!CreateRegistryKey(key, NULL, dllPath)) { |
| goto out; |
| } |
| |
| if (!CreateRegistryKey(key, "ThreadingModel", "Apartment")) { |
| goto out; |
| } |
| |
| sprintf(key, "CLSID\\%s\\ProgID", g_szClsid); |
| if (!CreateRegistryKey(key, NULL, g_szProgid)) { |
| goto out; |
| } |
| |
| if (!CreateRegistryKey(g_szProgid, NULL, QGA_PROVIDER_NAME)) { |
| goto out; |
| } |
| |
| sprintf(key, "%s\\CLSID", g_szProgid); |
| if (!CreateRegistryKey(key, NULL, g_szClsid)) { |
| goto out; |
| } |
| |
| hr = CoCreateInstance(CLSID_VSSCoordinator, NULL, CLSCTX_ALL, |
| IID_IVssAdmin, (void **)pVssAdmin.replace()); |
| if (FAILED(hr)) { |
| errmsg_dialog(hr, "CoCreateInstance(VSSCoordinator) failed"); |
| goto out; |
| } |
| |
| hr = pVssAdmin->RegisterProvider(g_gProviderId, CLSID_QGAVSSProvider, |
| const_cast<WCHAR*>(QGA_PROVIDER_LNAME), |
| VSS_PROV_SOFTWARE, |
| const_cast<WCHAR*>(QGA_PROVIDER_VERSION), |
| g_gProviderVersion); |
| if (FAILED(hr)) { |
| errmsg_dialog(hr, "RegisterProvider failed"); |
| } |
| |
| out: |
| if (FAILED(hr)) { |
| DllUnregisterServer(); |
| } |
| |
| return hr; |
| } |
| |
| /* Unregister this VSS hardware provider from the system */ |
| STDAPI DllUnregisterServer(void) |
| { |
| TCHAR key[256]; |
| COMInitializer initializer; |
| COMPointer<IVssAdmin> pVssAdmin; |
| |
| HRESULT hr = CoCreateInstance(CLSID_VSSCoordinator, |
| NULL, CLSCTX_ALL, IID_IVssAdmin, |
| (void **)pVssAdmin.replace()); |
| if (SUCCEEDED(hr)) { |
| hr = pVssAdmin->UnregisterProvider(g_gProviderId); |
| } else { |
| errmsg(hr, "CoCreateInstance(VSSCoordinator) failed"); |
| } |
| |
| sprintf(key, "CLSID\\%s", g_szClsid); |
| SHDeleteKey(HKEY_CLASSES_ROOT, key); |
| SHDeleteKey(HKEY_CLASSES_ROOT, g_szProgid); |
| |
| return S_OK; /* Uninstall should never fail */ |
| } |
| |
| |
| /* Support function to convert ASCII string into BSTR (used in _bstr_t) */ |
| namespace _com_util |
| { |
| BSTR WINAPI ConvertStringToBSTR(const char *ascii) { |
| int len = strlen(ascii); |
| BSTR bstr = SysAllocStringLen(NULL, len); |
| |
| if (!bstr) { |
| return NULL; |
| } |
| |
| if (mbstowcs(bstr, ascii, len) == (size_t)-1) { |
| fprintf(stderr, "Failed to convert string '%s' into BSTR", ascii); |
| bstr[0] = 0; |
| } |
| return bstr; |
| } |
| } |