VBoxVRDP.cpp@ 63311

Last change on this file since 63311 was 63100, checked in by vboxsync, 8 years ago
GA/NT/VBoxTray: warnings
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 16.1 KB

Line
1	/* $Id: VBoxVRDP.cpp 63100 2016-08-06 15:44:22Z vboxsync $ */
2	/** @file
3	* VBoxVRDP - VBox VRDP connection notification
4	*/
5
6	/*
7	* Copyright (C) 2006-2016 Oracle Corporation
8	*
9	* This file is part of VirtualBox Open Source Edition (OSE), as
10	* available from http://www.virtualbox.org. This file is free software;
11	* you can redistribute it and/or modify it under the terms of the GNU
12	* General Public License (GPL) as published by the Free Software
13	* Foundation, in version 2 as it comes in the "COPYING" file of the
14	* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15	* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16	*/
17
18	/* 0x0501 for SPI_SETDROPSHADOW */
19	#define _WIN32_WINNT 0x0501
20	#include <iprt/win/windows.h>
21	#include "VBoxTray.h"
22	#include "VBoxHelpers.h"
23	#include "VBoxVRDP.h"
24
25	#include <VBox/VMMDev.h>
26	#ifdef DEBUG
27	# define LOG_ENABLED
28	# define LOG_GROUP LOG_GROUP_DEFAULT
29	#endif
30	#include <VBox/log.h>
31
32	#include <iprt/assert.h>
33	#include <iprt/ldr.h>
34
35
36
37	/* The guest receives VRDP_ACTIVE/VRDP_INACTIVE notifications.
38	*
39	* When VRDP_ACTIVE is received, the guest asks host about the experience level.
40	* The experience level is an integer value, different values disable some GUI effects.
41	*
42	* On VRDP_INACTIVE the original values are restored.
43	*
44	* Note: that this is not controlled from the client, that is a per VM settings.
45	*
46	* Note: theming is disabled separately by EnableTheming.
47	*/
48
49	#define VBOX_SPI_STRING 0
50	#define VBOX_SPI_BOOL_PTR 1
51	#define VBOX_SPI_BOOL 2
52	#define VBOX_SPI_PTR 3
53
54	static ANIMATIONINFO animationInfoDisable =
55	{
56	sizeof (ANIMATIONINFO),
57	FALSE
58	};
59
60	typedef struct _VBOXVRDPEXPPARAM
61	{
62	const char *name;
63	UINT uActionSet;
64	UINT uActionGet;
65	uint32_t level; /* The parameter remain enabled at this or higher level. */
66	int type;
67	void *pvDisable;
68	UINT cbSavedValue;
69	char achSavedValue[2 * MAX_PATH]; /* Large enough to save the bitmap path. */
70	} VBOXVRDPEXPPARAM, *PVBOXVRDPEXPPARAM;
71
72	typedef struct _VBOXVRDPCONTEXT
73	{
74	const VBOXSERVICEENV *pEnv;
75
76	uint32_t level;
77	BOOL fSavedThemeEnabled;
78
79	RTLDRMOD hModUxTheme;
80
81	HRESULT (* pfnEnableTheming)(BOOL fEnable);
82	BOOL (* pfnIsThemeActive)(VOID);
83	} VBOXVRDPCONTEXT, *PVBOXVRDPCONTEXT;
84
85	static VBOXVRDPCONTEXT g_Ctx = { 0 };
86
87	#define SPI_(l, a) #a, SPI_SET##a, SPI_GET##a, VRDP_EXPERIENCE_LEVEL_##l
88
89	static VBOXVRDPEXPPARAM s_aSPIParams[] =
90	{
91	{ SPI_(MEDIUM, DESKWALLPAPER), VBOX_SPI_STRING, "" },
92	{ SPI_(FULL, DROPSHADOW), VBOX_SPI_BOOL_PTR, },
93	{ SPI_(HIGH, FONTSMOOTHING), VBOX_SPI_BOOL, },
94	{ SPI_(FULL, MENUFADE), VBOX_SPI_BOOL_PTR, },
95	{ SPI_(FULL, COMBOBOXANIMATION), VBOX_SPI_BOOL_PTR, },
96	{ SPI_(FULL, CURSORSHADOW), VBOX_SPI_BOOL_PTR, },
97	{ SPI_(HIGH, GRADIENTCAPTIONS), VBOX_SPI_BOOL_PTR, },
98	{ SPI_(FULL, LISTBOXSMOOTHSCROLLING), VBOX_SPI_BOOL_PTR, },
99	{ SPI_(FULL, MENUANIMATION), VBOX_SPI_BOOL_PTR, },
100	{ SPI_(FULL, SELECTIONFADE), VBOX_SPI_BOOL_PTR, },
101	{ SPI_(FULL, TOOLTIPANIMATION), VBOX_SPI_BOOL_PTR, },
102	{ SPI_(FULL, ANIMATION), VBOX_SPI_PTR, &animationInfoDisable, sizeof (ANIMATIONINFO) },
103	{ SPI_(MEDIUM, DRAGFULLWINDOWS), VBOX_SPI_BOOL, }
104	};
105
106	#undef SPI_
107
108	static void vboxExperienceSet(uint32_t uLevel)
109	{
110	for (size_t i = 0; i < RT_ELEMENTS(s_aSPIParams); i++)
111	{
112	PVBOXVRDPEXPPARAM pParam = &s_aSPIParams[i];
113	if (pParam->level > uLevel)
114	{
115	/*
116	* The parameter has to be disabled.
117	*/
118	LogFlowFunc(("Saving %s\n", pParam->name));
119
120	/* Save the current value. */
121	switch (pParam->type)
122	{
123	case VBOX_SPI_STRING:
124	{
125	/* The 2nd parameter is size in characters of the buffer.
126	* The 3rd parameter points to the buffer.
127	*/
128	SystemParametersInfo (pParam->uActionGet,
129	MAX_PATH,
130	pParam->achSavedValue,
131	0);
132	} break;
133
134	case VBOX_SPI_BOOL:
135	case VBOX_SPI_BOOL_PTR:
136	{
137	/* The 3rd parameter points to BOOL. */
138	SystemParametersInfo (pParam->uActionGet,
139	0,
140	pParam->achSavedValue,
141	0);
142	} break;
143
144	case VBOX_SPI_PTR:
145	{
146	/* The 3rd parameter points to the structure.
147	* The cbSize member of this structure must be set.
148	* The uiParam parameter must alos be set.
149	*/
150	if (pParam->cbSavedValue > sizeof (pParam->achSavedValue))
151	{
152	LogFlowFunc(("Not enough space %d > %d\n", pParam->cbSavedValue, sizeof (pParam->achSavedValue)));
153	break;
154	}
155
156	(UINT )&pParam->achSavedValue[0] = pParam->cbSavedValue;
157
158	SystemParametersInfo (s_aSPIParams[i].uActionGet,
159	s_aSPIParams[i].cbSavedValue,
160	s_aSPIParams[i].achSavedValue,
161	0);
162	} break;
163
164	default:
165	break;
166	}
167
168	LogFlowFunc(("Disabling %s\n", pParam->name));
169
170	/* Disable the feature. */
171	switch (pParam->type)
172	{
173	case VBOX_SPI_STRING:
174	{
175	/* The 3rd parameter points to the string. */
176	SystemParametersInfo (pParam->uActionSet,
177	0,
178	pParam->pvDisable,
179	SPIF_SENDCHANGE);
180	} break;
181
182	case VBOX_SPI_BOOL:
183	{
184	/* The 2nd parameter is BOOL. */
185	SystemParametersInfo (pParam->uActionSet,
186	FALSE,
187	NULL,
188	SPIF_SENDCHANGE);
189	} break;
190
191	case VBOX_SPI_BOOL_PTR:
192	{
193	/* The 3rd parameter is NULL to disable. */
194	SystemParametersInfo (pParam->uActionSet,
195	0,
196	NULL,
197	SPIF_SENDCHANGE);
198	} break;
199
200	case VBOX_SPI_PTR:
201	{
202	/* The 3rd parameter points to the structure. */
203	SystemParametersInfo (pParam->uActionSet,
204	0,
205	pParam->pvDisable,
206	SPIF_SENDCHANGE);
207	} break;
208
209	default:
210	break;
211	}
212	}
213	}
214	}
215
216	static void vboxExperienceRestore(uint32_t uLevel)
217	{
218	int i;
219	for (i = 0; i < RT_ELEMENTS(s_aSPIParams); i++)
220	{
221	PVBOXVRDPEXPPARAM pParam = &s_aSPIParams[i];
222	if (pParam->level > uLevel)
223	{
224	LogFlowFunc(("Restoring %s\n", pParam->name));
225
226	/* Restore the feature. */
227	switch (pParam->type)
228	{
229	case VBOX_SPI_STRING:
230	{
231	/* The 3rd parameter points to the string. */
232	SystemParametersInfo (pParam->uActionSet,
233	0,
234	pParam->achSavedValue,
235	SPIF_SENDCHANGE);
236	} break;
237
238	case VBOX_SPI_BOOL:
239	{
240	/* The 2nd parameter is BOOL. */
241	SystemParametersInfo (pParam->uActionSet,
242	(BOOL )&pParam->achSavedValue[0],
243	NULL,
244	SPIF_SENDCHANGE);
245	} break;
246
247	case VBOX_SPI_BOOL_PTR:
248	{
249	/* The 3rd parameter is NULL to disable. */
250	BOOL fSaved = (BOOL )&pParam->achSavedValue[0];
251
252	SystemParametersInfo (pParam->uActionSet,
253	0,
254	fSaved? &fSaved: NULL,
255	SPIF_SENDCHANGE);
256	} break;
257
258	case VBOX_SPI_PTR:
259	{
260	/* The 3rd parameter points to the structure. */
261	SystemParametersInfo (pParam->uActionSet,
262	0,
263	pParam->achSavedValue,
264	SPIF_SENDCHANGE);
265	} break;
266
267	default:
268	break;
269	}
270	}
271	}
272	}
273
274	static DECLCALLBACK(int) VBoxVRDPInit(const PVBOXSERVICEENV pEnv, void **ppInstance)
275	{
276	AssertPtrReturn(pEnv, VERR_INVALID_POINTER);
277	AssertPtrReturn(ppInstance, VERR_INVALID_POINTER);
278
279	LogFlowFuncEnter();
280
281	PVBOXVRDPCONTEXT pCtx = &g_Ctx; /* Only one instance at the moment. */
282	AssertPtr(pCtx);
283
284	pCtx->pEnv = pEnv;
285	pCtx->level = VRDP_EXPERIENCE_LEVEL_FULL;
286	pCtx->fSavedThemeEnabled = FALSE;
287
288	int rc = RTLdrLoadSystem("UxTheme.dll", false /fNoUnload/, &g_Ctx.hModUxTheme);
289	if (RT_SUCCESS(rc))
290	{
291	(PFNRT )&pCtx->pfnEnableTheming = RTLdrGetFunction(g_Ctx.hModUxTheme, "EnableTheming");
292	(PFNRT )&pCtx->pfnIsThemeActive = RTLdrGetFunction(g_Ctx.hModUxTheme, "IsThemeActive");
293
294	*ppInstance = &g_Ctx;
295	}
296	else
297	{
298	g_Ctx.hModUxTheme = NIL_RTLDRMOD;
299	g_Ctx.pfnEnableTheming = NULL;
300	g_Ctx.pfnIsThemeActive = NULL;
301	}
302
303	/* Tell the caller that the service does not work but it is OK to continue. */
304	if (RT_FAILURE(rc))
305	rc = VERR_NOT_SUPPORTED;
306
307	LogFlowFuncLeaveRC(rc);
308	return rc;
309	}
310
311	static DECLCALLBACK(void) VBoxVRDPDestroy(void *pInstance)
312	{
313	AssertPtrReturnVoid(pInstance);
314
315	LogFlowFuncEnter();
316
317	PVBOXVRDPCONTEXT pCtx = (PVBOXVRDPCONTEXT)pInstance;
318
319	vboxExperienceRestore (pCtx->level);
320	if (pCtx->hModUxTheme != NIL_RTLDRMOD)
321	{
322	RTLdrClose(g_Ctx.hModUxTheme);
323	pCtx->hModUxTheme = NIL_RTLDRMOD;
324	}
325
326	return;
327	}
328
329	/**
330	* Thread function to wait for and process mode change requests
331	*/
332	static DECLCALLBACK(int) VBoxVRDPWorker(void pInstance, bool volatile pfShutdown)
333	{
334	AssertPtrReturn(pInstance, VERR_INVALID_POINTER);
335
336	LogFlowFuncEnter();
337
338	/*
339	* Tell the control thread that it can continue
340	* spawning services.
341	*/
342	RTThreadUserSignal(RTThreadSelf());
343
344	PVBOXVRDPCONTEXT pCtx = (PVBOXVRDPCONTEXT)pInstance;
345	AssertPtr(pCtx);
346
347	HANDLE gVBoxDriver = pCtx->pEnv->hDriver;
348	VBoxGuestFilterMaskInfo maskInfo;
349	DWORD cbReturned;
350
351	maskInfo.u32OrMask = VMMDEV_EVENT_VRDP;
352	maskInfo.u32NotMask = 0;
353	if (!DeviceIoControl(gVBoxDriver, VBOXGUEST_IOCTL_CTL_FILTER_MASK, &maskInfo, sizeof (maskInfo), NULL, 0, &cbReturned, NULL))
354	{
355	DWORD dwErr = GetLastError();
356	LogFlowFunc(("DeviceIOControl(CtlMask) failed with %ld, exiting\n", dwErr));
357	return RTErrConvertFromWin32(dwErr);
358	}
359
360	int rc = VINF_SUCCESS;
361
362	for (;;)
363	{
364	/* wait for the event */
365	VBoxGuestWaitEventInfo waitEvent;
366	waitEvent.u32TimeoutIn = 5000;
367	waitEvent.u32EventMaskIn = VMMDEV_EVENT_VRDP;
368	if (DeviceIoControl(gVBoxDriver, VBOXGUEST_IOCTL_WAITEVENT, &waitEvent, sizeof(waitEvent), &waitEvent, sizeof(waitEvent), &cbReturned, NULL))
369	{
370	/* are we supposed to stop? */
371	if (*pfShutdown)
372	break;
373
374	/* did we get the right event? */
375	if (waitEvent.u32EventFlagsOut & VMMDEV_EVENT_VRDP)
376	{
377	/* Call the host to get VRDP status and the experience level. */
378	VMMDevVRDPChangeRequest vrdpChangeRequest = {0};
379
380	vrdpChangeRequest.header.size = sizeof(VMMDevVRDPChangeRequest);
381	vrdpChangeRequest.header.version = VMMDEV_REQUEST_HEADER_VERSION;
382	vrdpChangeRequest.header.requestType = VMMDevReq_GetVRDPChangeRequest;
383	vrdpChangeRequest.u8VRDPActive = 0;
384	vrdpChangeRequest.u32VRDPExperienceLevel = 0;
385
386	if (DeviceIoControl (gVBoxDriver,
387	VBOXGUEST_IOCTL_VMMREQUEST(sizeof(VMMDevVRDPChangeRequest)),
388	&vrdpChangeRequest,
389	sizeof(VMMDevVRDPChangeRequest),
390	&vrdpChangeRequest,
391	sizeof(VMMDevVRDPChangeRequest),
392	&cbReturned, NULL))
393	{
394	LogFlowFunc(("u8VRDPActive = %d, level %d\n", vrdpChangeRequest.u8VRDPActive, vrdpChangeRequest.u32VRDPExperienceLevel));
395
396	if (vrdpChangeRequest.u8VRDPActive)
397	{
398	pCtx->level = vrdpChangeRequest.u32VRDPExperienceLevel;
399	vboxExperienceSet (pCtx->level);
400
401	if (pCtx->level == VRDP_EXPERIENCE_LEVEL_ZERO
402	&& pCtx->pfnEnableTheming
403	&& pCtx->pfnIsThemeActive)
404	{
405	pCtx->fSavedThemeEnabled = pCtx->pfnIsThemeActive ();
406
407	LogFlowFunc(("pCtx->fSavedThemeEnabled = %d\n", pCtx->fSavedThemeEnabled));
408
409	if (pCtx->fSavedThemeEnabled)
410	{
411	pCtx->pfnEnableTheming (FALSE);
412	}
413	}
414	}
415	else
416	{
417	if (pCtx->level == VRDP_EXPERIENCE_LEVEL_ZERO
418	&& pCtx->pfnEnableTheming
419	&& pCtx->pfnIsThemeActive)
420	{
421	if (pCtx->fSavedThemeEnabled)
422	{
423	/** @todo the call returns S_OK but theming remains disabled. */
424	HRESULT hrc = pCtx->pfnEnableTheming (TRUE);
425	LogFlowFunc(("enabling theme rc = 0x%08X\n", hrc)); NOREF(hrc);
426	pCtx->fSavedThemeEnabled = FALSE;
427	}
428	}
429
430	vboxExperienceRestore (pCtx->level);
431
432	pCtx->level = VRDP_EXPERIENCE_LEVEL_FULL;
433	}
434	}
435	else
436	{
437	/* sleep a bit to not eat too much CPU in case the above call always fails */
438	RTThreadSleep(10);
439
440	if (*pfShutdown)
441	break;
442	}
443	}
444	}
445	else
446	{
447	/* sleep a bit to not eat too much CPU in case the above call always fails */
448	RTThreadSleep(50);
449
450	if (*pfShutdown)
451	break;
452	}
453	}
454
455	maskInfo.u32OrMask = 0;
456	maskInfo.u32NotMask = VMMDEV_EVENT_VRDP;
457	if (!DeviceIoControl(gVBoxDriver, VBOXGUEST_IOCTL_CTL_FILTER_MASK, &maskInfo, sizeof (maskInfo), NULL, 0, &cbReturned, NULL))
458	LogFlowFunc(("DeviceIOControl(CtlMask) failed\n"));
459
460	LogFlowFuncLeaveRC(rc);
461	return rc;
462	}
463
464	/**
465	* The service description.
466	*/
467	VBOXSERVICEDESC g_SvcDescVRDP =
468	{
469	/* pszName. */
470	"VRDP",
471	/* pszDescription. */
472	"VRDP Connection Notification",
473	/* methods */
474	VBoxVRDPInit,
475	VBoxVRDPWorker,
476	NULL /* pfnStop */,
477	VBoxVRDPDestroy
478	};
479

Note: See TracBrowser for help on using the repository browser.

source: vbox/trunk/src/VBox/Additions/WINNT/VBoxTray/VBoxVRDP.cpp@ 63311

Download in other formats: