2 * QEMU Guest Agent win32 VSS Provider installer
4 * Copyright Hitachi Data Systems Corp. 2013
7 * Tomoki Sekiyama <tomoki.sekiyama@hds.com>
9 * This work is licensed under the terms of the GNU GPL, version 2 or later.
10 * See the COPYING file in the top-level directory.
16 #include "vss-common.h"
17 #include "inc/win2003/vscoordint.h"
24 extern HINSTANCE g_hinstDll
;
26 const GUID CLSID_COMAdminCatalog
= { 0xF618C514, 0xDFB8, 0x11d1,
27 {0xA2, 0xCF, 0x00, 0x80, 0x5F, 0xC7, 0x92, 0x35} };
28 const GUID IID_ICOMAdminCatalog2
= { 0x790C6E0B, 0x9194, 0x4cc9,
29 {0x94, 0x26, 0xA4, 0x8A, 0x63, 0x18, 0x56, 0x96} };
30 const GUID CLSID_WbemLocator
= { 0x4590f811, 0x1d3a, 0x11d0,
31 {0x89, 0x1f, 0x00, 0xaa, 0x00, 0x4b, 0x2e, 0x24} };
32 const GUID IID_IWbemLocator
= { 0xdc12a687, 0x737f, 0x11cf,
33 {0x88, 0x4d, 0x00, 0xaa, 0x00, 0x4b, 0x2e, 0x24} };
35 void errmsg(DWORD err
, const char *text
)
38 * `text' contains function call statement when errmsg is called via chk().
39 * To make error message more readable, we cut off the text after '('.
40 * If text doesn't contains '(', negative precision is given, which is
41 * treated as though it were missing.
43 char *msg
= NULL
, *nul
= strchr(text
, '(');
44 int len
= nul
? nul
- text
: -1;
46 FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER
|
47 FORMAT_MESSAGE_FROM_SYSTEM
| FORMAT_MESSAGE_IGNORE_INSERTS
,
48 NULL
, err
, MAKELANGID(LANG_NEUTRAL
, SUBLANG_DEFAULT
),
49 (char *)&msg
, 0, NULL
);
50 fprintf(stderr
, "%.*s. (Error: %lx) %s\n", len
, text
, err
, msg
);
54 static void errmsg_dialog(DWORD err
, const char *text
, const char *opt
= "")
58 FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER
|
59 FORMAT_MESSAGE_FROM_SYSTEM
| FORMAT_MESSAGE_IGNORE_INSERTS
,
60 NULL
, err
, MAKELANGID(LANG_NEUTRAL
, SUBLANG_DEFAULT
),
61 (char *)&msg
, 0, NULL
);
62 snprintf(buf
, sizeof(buf
), "%s%s. (Error: %lx) %s", text
, opt
, err
, msg
);
63 MessageBox(NULL
, buf
, "Error from " QGA_PROVIDER_NAME
, MB_OK
|MB_ICONERROR
);
67 #define _chk(hr, status, msg, err_label) \
76 #define chk(status) _chk(hr, status, "Failed to " #status, out)
78 void __stdcall
_com_issue_error(HRESULT hr
)
80 errmsg(hr
, "Unexpected error in COM");
84 HRESULT
put_Value(ICatalogObject
*pObj
, LPCWSTR name
, T val
)
86 return pObj
->put_Value(_bstr_t(name
), _variant_t(val
));
89 /* Lookup Administrators group name from winmgmt */
90 static HRESULT
GetAdminName(_bstr_t
*name
)
93 COMPointer
<IWbemLocator
> pLoc
;
94 COMPointer
<IWbemServices
> pSvc
;
95 COMPointer
<IEnumWbemClassObject
> pEnum
;
96 COMPointer
<IWbemClassObject
> pWobj
;
100 chk(CoCreateInstance(CLSID_WbemLocator
, NULL
, CLSCTX_INPROC_SERVER
,
101 IID_IWbemLocator
, (LPVOID
*)pLoc
.replace()));
102 chk(pLoc
->ConnectServer(_bstr_t(L
"ROOT\\CIMV2"), NULL
, NULL
, NULL
,
103 0, 0, 0, pSvc
.replace()));
104 chk(CoSetProxyBlanket(pSvc
, RPC_C_AUTHN_WINNT
, RPC_C_AUTHZ_NONE
,
105 NULL
, RPC_C_AUTHN_LEVEL_CALL
,
106 RPC_C_IMP_LEVEL_IMPERSONATE
, NULL
, EOAC_NONE
));
107 chk(pSvc
->ExecQuery(_bstr_t(L
"WQL"),
108 _bstr_t(L
"select * from Win32_Account where "
109 "SID='S-1-5-32-544' and localAccount=TRUE"),
110 WBEM_FLAG_RETURN_IMMEDIATELY
| WBEM_FLAG_FORWARD_ONLY
,
111 NULL
, pEnum
.replace()));
114 errmsg(hr
, "Failed to query for Administrators");
117 chk(pEnum
->Next(WBEM_INFINITE
, 1, pWobj
.replace(), &returned
));
120 errmsg(hr
, "No Administrators found");
124 chk(pWobj
->Get(_bstr_t(L
"Name"), 0, &var
, 0, 0));
129 errmsg(hr
, "Failed to get name of Administrators");
137 /* Find and iterate QGA VSS provider in COM+ Application Catalog */
138 static HRESULT
QGAProviderFind(
139 HRESULT (*found
)(ICatalogCollection
*, int, void *), void *arg
)
142 COMInitializer initializer
;
143 COMPointer
<IUnknown
> pUnknown
;
144 COMPointer
<ICOMAdminCatalog2
> pCatalog
;
145 COMPointer
<ICatalogCollection
> pColl
;
146 COMPointer
<ICatalogObject
> pObj
;
150 chk(CoCreateInstance(CLSID_COMAdminCatalog
, NULL
, CLSCTX_INPROC_SERVER
,
151 IID_IUnknown
, (void **)pUnknown
.replace()));
152 chk(pUnknown
->QueryInterface(IID_ICOMAdminCatalog2
,
153 (void **)pCatalog
.replace()));
154 chk(pCatalog
->GetCollection(_bstr_t(L
"Applications"),
155 (IDispatch
**)pColl
.replace()));
156 chk(pColl
->Populate());
158 chk(pColl
->get_Count(&n
));
159 for (i
= n
- 1; i
>= 0; i
--) {
160 chk(pColl
->get_Item(i
, (IDispatch
**)pObj
.replace()));
161 chk(pObj
->get_Value(_bstr_t(L
"Name"), &var
));
162 if (var
== _variant_t(QGA_PROVIDER_LNAME
)) {
163 if (FAILED(found(pColl
, i
, arg
))) {
168 chk(pColl
->SaveChanges(&n
));
174 /* Count QGA VSS provider in COM+ Application Catalog */
175 static HRESULT
QGAProviderCount(ICatalogCollection
*coll
, int i
, void *arg
)
181 /* Remove QGA VSS provider from COM+ Application Catalog Collection */
182 static HRESULT
QGAProviderRemove(ICatalogCollection
*coll
, int i
, void *arg
)
186 fprintf(stderr
, "Removing COM+ Application: %s\n", QGA_PROVIDER_NAME
);
187 chk(coll
->Remove(i
));
192 /* Unregister this module from COM+ Applications Catalog */
193 STDAPI
COMUnregister(void)
197 DllUnregisterServer();
198 chk(QGAProviderFind(QGAProviderRemove
, NULL
));
203 /* Register this module to COM+ Applications Catalog */
204 STDAPI
COMRegister(void)
207 COMInitializer initializer
;
208 COMPointer
<IUnknown
> pUnknown
;
209 COMPointer
<ICOMAdminCatalog2
> pCatalog
;
210 COMPointer
<ICatalogCollection
> pApps
, pRoles
, pUsersInRole
;
211 COMPointer
<ICatalogObject
> pObj
;
215 CHAR dllPath
[MAX_PATH
], tlbPath
[MAX_PATH
];
216 bool unregisterOnFailure
= false;
220 errmsg(E_FAIL
, "Failed to initialize DLL");
224 chk(QGAProviderFind(QGAProviderCount
, (void *)&count
));
226 errmsg(E_ABORT
, "QGA VSS Provider is already installed");
230 chk(CoCreateInstance(CLSID_COMAdminCatalog
, NULL
, CLSCTX_INPROC_SERVER
,
231 IID_IUnknown
, (void **)pUnknown
.replace()));
232 chk(pUnknown
->QueryInterface(IID_ICOMAdminCatalog2
,
233 (void **)pCatalog
.replace()));
235 /* Install COM+ Component */
237 chk(pCatalog
->GetCollection(_bstr_t(L
"Applications"),
238 (IDispatch
**)pApps
.replace()));
239 chk(pApps
->Populate());
240 chk(pApps
->Add((IDispatch
**)&pObj
));
241 chk(put_Value(pObj
, L
"Name", QGA_PROVIDER_LNAME
));
242 chk(put_Value(pObj
, L
"Description", QGA_PROVIDER_LNAME
));
243 chk(put_Value(pObj
, L
"ApplicationAccessChecksEnabled", true));
244 chk(put_Value(pObj
, L
"Authentication", short(6)));
245 chk(put_Value(pObj
, L
"AuthenticationCapability", short(2)));
246 chk(put_Value(pObj
, L
"ImpersonationLevel", short(2)));
247 chk(pApps
->SaveChanges(&n
));
249 /* The app should be deleted if something fails after SaveChanges */
250 unregisterOnFailure
= true;
252 chk(pObj
->get_Key(&key
));
254 if (!GetModuleFileName(g_hinstDll
, dllPath
, sizeof(dllPath
))) {
255 hr
= HRESULT_FROM_WIN32(GetLastError());
256 errmsg(hr
, "GetModuleFileName failed");
262 errmsg(hr
, "Failed to lookup dll");
265 strcpy(tlbPath
, dllPath
);
266 strcpy(tlbPath
+n
-3, "tlb");
267 fprintf(stderr
, "Registering " QGA_PROVIDER_NAME
":\n");
268 fprintf(stderr
, " %s\n", dllPath
);
269 fprintf(stderr
, " %s\n", tlbPath
);
270 if (!PathFileExists(tlbPath
)) {
271 hr
= HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND
);
272 errmsg(hr
, "Failed to lookup tlb");
276 chk(pCatalog
->CreateServiceForApplication(
277 _bstr_t(QGA_PROVIDER_LNAME
), _bstr_t(QGA_PROVIDER_LNAME
),
278 _bstr_t(L
"SERVICE_AUTO_START"), _bstr_t(L
"SERVICE_ERROR_NORMAL"),
279 _bstr_t(L
""), _bstr_t(L
".\\localsystem"), _bstr_t(L
""), FALSE
));
280 chk(pCatalog
->InstallComponent(_bstr_t(QGA_PROVIDER_LNAME
),
281 _bstr_t(dllPath
), _bstr_t(tlbPath
),
284 /* Setup roles of the applicaion */
286 chk(pApps
->GetCollection(_bstr_t(L
"Roles"), key
,
287 (IDispatch
**)pRoles
.replace()));
288 chk(pRoles
->Populate());
289 chk(pRoles
->Add((IDispatch
**)pObj
.replace()));
290 chk(put_Value(pObj
, L
"Name", L
"Administrators"));
291 chk(put_Value(pObj
, L
"Description", L
"Administrators group"));
292 chk(pRoles
->SaveChanges(&n
));
293 chk(pObj
->get_Key(&key
));
295 /* Setup users in the role */
297 chk(pRoles
->GetCollection(_bstr_t(L
"UsersInRole"), key
,
298 (IDispatch
**)pUsersInRole
.replace()));
299 chk(pUsersInRole
->Populate());
301 chk(pUsersInRole
->Add((IDispatch
**)pObj
.replace()));
302 chk(GetAdminName(&name
));
303 chk(put_Value(pObj
, L
"User", _bstr_t(".\\") + name
));
305 chk(pUsersInRole
->Add((IDispatch
**)pObj
.replace()));
306 chk(put_Value(pObj
, L
"User", L
"SYSTEM"));
307 chk(pUsersInRole
->SaveChanges(&n
));
310 if (unregisterOnFailure
&& FAILED(hr
)) {
318 static BOOL
CreateRegistryKey(LPCTSTR key
, LPCTSTR value
, LPCTSTR data
)
324 ret
= RegCreateKeyEx(HKEY_CLASSES_ROOT
, key
, 0, NULL
,
325 REG_OPTION_NON_VOLATILE
, KEY_WRITE
, NULL
, &hKey
, NULL
);
326 if (ret
!= ERROR_SUCCESS
) {
331 size
= strlen(data
) + 1;
336 ret
= RegSetValueEx(hKey
, value
, 0, REG_SZ
, (LPBYTE
)data
, size
);
340 if (ret
!= ERROR_SUCCESS
) {
341 /* As we cannot printf within DllRegisterServer(), show a dialog. */
342 errmsg_dialog(ret
, "Cannot add registry", key
);
348 /* Register this dll as a VSS provider */
349 STDAPI
DllRegisterServer(void)
351 COMInitializer initializer
;
352 COMPointer
<IVssAdmin
> pVssAdmin
;
354 char dllPath
[MAX_PATH
];
358 errmsg_dialog(hr
, "Module instance is not available");
362 /* Add this module to registery */
364 sprintf(key
, "CLSID\\%s", g_szClsid
);
365 if (!CreateRegistryKey(key
, NULL
, g_szClsid
)) {
369 if (!GetModuleFileName(g_hinstDll
, dllPath
, sizeof(dllPath
))) {
370 errmsg_dialog(GetLastError(), "GetModuleFileName failed");
374 sprintf(key
, "CLSID\\%s\\InprocServer32", g_szClsid
);
375 if (!CreateRegistryKey(key
, NULL
, dllPath
)) {
379 if (!CreateRegistryKey(key
, "ThreadingModel", "Apartment")) {
383 sprintf(key
, "CLSID\\%s\\ProgID", g_szClsid
);
384 if (!CreateRegistryKey(key
, NULL
, g_szProgid
)) {
388 if (!CreateRegistryKey(g_szProgid
, NULL
, QGA_PROVIDER_NAME
)) {
392 sprintf(key
, "%s\\CLSID", g_szProgid
);
393 if (!CreateRegistryKey(key
, NULL
, g_szClsid
)) {
397 hr
= CoCreateInstance(CLSID_VSSCoordinator
, NULL
, CLSCTX_ALL
,
398 IID_IVssAdmin
, (void **)pVssAdmin
.replace());
400 errmsg_dialog(hr
, "CoCreateInstance(VSSCoordinator) failed");
404 hr
= pVssAdmin
->RegisterProvider(g_gProviderId
, CLSID_QGAVSSProvider
,
405 const_cast<WCHAR
*>(QGA_PROVIDER_LNAME
),
407 const_cast<WCHAR
*>(QGA_PROVIDER_VERSION
),
410 errmsg_dialog(hr
, "RegisterProvider failed");
415 DllUnregisterServer();
421 /* Unregister this VSS hardware provider from the system */
422 STDAPI
DllUnregisterServer(void)
425 COMInitializer initializer
;
426 COMPointer
<IVssAdmin
> pVssAdmin
;
428 HRESULT hr
= CoCreateInstance(CLSID_VSSCoordinator
,
429 NULL
, CLSCTX_ALL
, IID_IVssAdmin
,
430 (void **)pVssAdmin
.replace());
432 hr
= pVssAdmin
->UnregisterProvider(g_gProviderId
);
434 errmsg(hr
, "CoCreateInstance(VSSCoordinator) failed");
437 sprintf(key
, "CLSID\\%s", g_szClsid
);
438 SHDeleteKey(HKEY_CLASSES_ROOT
, key
);
439 SHDeleteKey(HKEY_CLASSES_ROOT
, g_szProgid
);
441 return S_OK
; /* Uninstall should never fail */
445 /* Support function to convert ASCII string into BSTR (used in _bstr_t) */
448 BSTR WINAPI
ConvertStringToBSTR(const char *ascii
) {
449 int len
= strlen(ascii
);
450 BSTR bstr
= SysAllocStringLen(NULL
, len
);
456 if (mbstowcs(bstr
, ascii
, len
) == (size_t)-1) {
457 fprintf(stderr
, "Failed to convert string '%s' into BSTR", ascii
);