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_ICOMAdminCatalog
= { 0xDD662187, 0xDFC2, 0x11d1,
29 {0xA2, 0xCF, 0x00, 0x80, 0x5F, 0xC7, 0x92, 0x35} };
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
<ICOMAdminCatalog
> 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_ICOMAdminCatalog
,
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
<ICOMAdminCatalog
> 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_ICOMAdminCatalog
,
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
->InstallComponent(_bstr_t(QGA_PROVIDER_LNAME
),
277 _bstr_t(dllPath
), _bstr_t(tlbPath
),
280 /* Setup roles of the applicaion */
282 chk(pApps
->GetCollection(_bstr_t(L
"Roles"), key
,
283 (IDispatch
**)pRoles
.replace()));
284 chk(pRoles
->Populate());
285 chk(pRoles
->Add((IDispatch
**)pObj
.replace()));
286 chk(put_Value(pObj
, L
"Name", L
"Administrators"));
287 chk(put_Value(pObj
, L
"Description", L
"Administrators group"));
288 chk(pRoles
->SaveChanges(&n
));
289 chk(pObj
->get_Key(&key
));
291 /* Setup users in the role */
293 chk(pRoles
->GetCollection(_bstr_t(L
"UsersInRole"), key
,
294 (IDispatch
**)pUsersInRole
.replace()));
295 chk(pUsersInRole
->Populate());
297 chk(pUsersInRole
->Add((IDispatch
**)pObj
.replace()));
298 chk(GetAdminName(&name
));
299 chk(put_Value(pObj
, L
"User", _bstr_t(".\\") + name
));
301 chk(pUsersInRole
->Add((IDispatch
**)pObj
.replace()));
302 chk(put_Value(pObj
, L
"User", L
"SYSTEM"));
303 chk(pUsersInRole
->SaveChanges(&n
));
306 if (unregisterOnFailure
&& FAILED(hr
)) {
314 static BOOL
CreateRegistryKey(LPCTSTR key
, LPCTSTR value
, LPCTSTR data
)
320 ret
= RegCreateKeyEx(HKEY_CLASSES_ROOT
, key
, 0, NULL
,
321 REG_OPTION_NON_VOLATILE
, KEY_WRITE
, NULL
, &hKey
, NULL
);
322 if (ret
!= ERROR_SUCCESS
) {
327 size
= strlen(data
) + 1;
332 ret
= RegSetValueEx(hKey
, value
, 0, REG_SZ
, (LPBYTE
)data
, size
);
336 if (ret
!= ERROR_SUCCESS
) {
337 /* As we cannot printf within DllRegisterServer(), show a dialog. */
338 errmsg_dialog(ret
, "Cannot add registry", key
);
344 /* Register this dll as a VSS provider */
345 STDAPI
DllRegisterServer(void)
347 COMInitializer initializer
;
348 COMPointer
<IVssAdmin
> pVssAdmin
;
350 char dllPath
[MAX_PATH
];
354 errmsg_dialog(hr
, "Module instance is not available");
358 /* Add this module to registery */
360 sprintf(key
, "CLSID\\%s", g_szClsid
);
361 if (!CreateRegistryKey(key
, NULL
, g_szClsid
)) {
365 if (!GetModuleFileName(g_hinstDll
, dllPath
, sizeof(dllPath
))) {
366 errmsg_dialog(GetLastError(), "GetModuleFileName failed");
370 sprintf(key
, "CLSID\\%s\\InprocServer32", g_szClsid
);
371 if (!CreateRegistryKey(key
, NULL
, dllPath
)) {
375 if (!CreateRegistryKey(key
, "ThreadingModel", "Apartment")) {
379 sprintf(key
, "CLSID\\%s\\ProgID", g_szClsid
);
380 if (!CreateRegistryKey(key
, NULL
, g_szProgid
)) {
384 if (!CreateRegistryKey(g_szProgid
, NULL
, QGA_PROVIDER_NAME
)) {
388 sprintf(key
, "%s\\CLSID", g_szProgid
);
389 if (!CreateRegistryKey(key
, NULL
, g_szClsid
)) {
393 hr
= CoCreateInstance(CLSID_VSSCoordinator
, NULL
, CLSCTX_ALL
,
394 IID_IVssAdmin
, (void **)pVssAdmin
.replace());
396 errmsg_dialog(hr
, "CoCreateInstance(VSSCoordinator) failed");
400 hr
= pVssAdmin
->RegisterProvider(g_gProviderId
, CLSID_QGAVSSProvider
,
401 const_cast<WCHAR
*>(QGA_PROVIDER_LNAME
),
403 const_cast<WCHAR
*>(QGA_PROVIDER_VERSION
),
406 errmsg_dialog(hr
, "RegisterProvider failed");
411 DllUnregisterServer();
417 /* Unregister this VSS hardware provider from the system */
418 STDAPI
DllUnregisterServer(void)
421 COMInitializer initializer
;
422 COMPointer
<IVssAdmin
> pVssAdmin
;
424 HRESULT hr
= CoCreateInstance(CLSID_VSSCoordinator
,
425 NULL
, CLSCTX_ALL
, IID_IVssAdmin
,
426 (void **)pVssAdmin
.replace());
428 hr
= pVssAdmin
->UnregisterProvider(g_gProviderId
);
430 errmsg(hr
, "CoCreateInstance(VSSCoordinator) failed");
433 sprintf(key
, "CLSID\\%s", g_szClsid
);
434 SHDeleteKey(HKEY_CLASSES_ROOT
, key
);
435 SHDeleteKey(HKEY_CLASSES_ROOT
, g_szProgid
);
437 return S_OK
; /* Uninstall should never fail */
441 /* Support function to convert ASCII string into BSTR (used in _bstr_t) */
444 BSTR WINAPI
ConvertStringToBSTR(const char *ascii
) {
445 int len
= strlen(ascii
);
446 BSTR bstr
= SysAllocStringLen(NULL
, len
);
452 if (mbstowcs(bstr
, ascii
, len
) == (size_t)-1) {
453 fprintf(stderr
, "Failed to convert string '%s' into BSTR", ascii
);