VBoxManageControlVM.cpp@ 69198

Last change on this file since 69198 was 68487, checked in by vboxsync, 7 years ago
Build fix.
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 79.0 KB

Line
1	/* $Id: VBoxManageControlVM.cpp 68487 2017-08-21 14:06:12Z vboxsync $ */
2	/** @file
3	* VBoxManage - Implementation of the controlvm command.
4	*/
5
6	/*
7	* Copyright (C) 2006-2017 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
19	/*********************************************************************************************************************************
20	* Header Files *
21	*********************************************************************************************************************************/
22	#include <VBox/com/com.h>
23	#include <VBox/com/string.h>
24	#include <VBox/com/Guid.h>
25	#include <VBox/com/array.h>
26	#include <VBox/com/ErrorInfo.h>
27	#include <VBox/com/errorprint.h>
28	#include <VBox/com/VirtualBox.h>
29
30	#include <iprt/ctype.h>
31	#include <VBox/err.h>
32	#include <iprt/getopt.h>
33	#include <iprt/stream.h>
34	#include <iprt/string.h>
35	#include <iprt/thread.h>
36	#include <iprt/uuid.h>
37	#include <iprt/file.h>
38	#include <VBox/log.h>
39
40	#include "VBoxManage.h"
41
42	#include <list>
43
44
45	/**
46	* Parses a number.
47	*
48	* @returns Valid number on success.
49	* @returns 0 if invalid number. All necessary bitching has been done.
50	* @param psz Pointer to the nic number.
51	*/
52	static unsigned parseNum(const char psz, unsigned cMaxNum, const char name)
53	{
54	uint32_t u32;
55	char *pszNext;
56	int rc = RTStrToUInt32Ex(psz, &pszNext, 10, &u32);
57	if ( RT_SUCCESS(rc)
58	&& *pszNext == '\0'
59	&& u32 >= 1
60	&& u32 <= cMaxNum)
61	return (unsigned)u32;
62	errorArgument("Invalid %s number '%s'", name, psz);
63	return 0;
64	}
65
66	unsigned int getMaxNics(IVirtualBox* vbox, IMachine* mach)
67	{
68	ComPtr<ISystemProperties> info;
69	ChipsetType_T aChipset;
70	ULONG NetworkAdapterCount = 0;
71	HRESULT rc;
72
73	do {
74	CHECK_ERROR_BREAK(vbox, COMGETTER(SystemProperties)(info.asOutParam()));
75	CHECK_ERROR_BREAK(mach, COMGETTER(ChipsetType)(&aChipset));
76	CHECK_ERROR_BREAK(info, GetMaxNetworkAdapters(aChipset, &NetworkAdapterCount));
77
78	return (unsigned int)NetworkAdapterCount;
79	} while (0);
80
81	return 0;
82	}
83
84	#define KBDCHARDEF_MOD_NONE 0x00
85	#define KBDCHARDEF_MOD_SHIFT 0x01
86
87	typedef struct KBDCHARDEF
88	{
89	uint8_t u8Scancode;
90	uint8_t u8Modifiers;
91	} KBDCHARDEF;
92
93	static const KBDCHARDEF g_aASCIIChars[0x80] =
94	{
95	/* 0x00 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
96	/* 0x01 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
97	/* 0x02 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
98	/* 0x03 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
99	/* 0x04 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
100	/* 0x05 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
101	/* 0x06 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
102	/* 0x07 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
103	/* 0x08 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
104	/* 0x09 ' ' */ {0x0f, KBDCHARDEF_MOD_NONE},
105	/* 0x0A ' ' */ {0x1c, KBDCHARDEF_MOD_NONE},
106	/* 0x0B ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
107	/* 0x0C ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
108	/* 0x0D ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
109	/* 0x0E ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
110	/* 0x0F ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
111	/* 0x10 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
112	/* 0x11 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
113	/* 0x12 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
114	/* 0x13 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
115	/* 0x14 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
116	/* 0x15 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
117	/* 0x16 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
118	/* 0x17 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
119	/* 0x18 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
120	/* 0x19 ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
121	/* 0x1A ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
122	/* 0x1B ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
123	/* 0x1C ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
124	/* 0x1D ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
125	/* 0x1E ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
126	/* 0x1F ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
127	/* 0x20 ' ' */ {0x39, KBDCHARDEF_MOD_NONE},
128	/* 0x21 '!' */ {0x02, KBDCHARDEF_MOD_SHIFT},
129	/* 0x22 '"' */ {0x28, KBDCHARDEF_MOD_SHIFT},
130	/* 0x23 '#' */ {0x04, KBDCHARDEF_MOD_SHIFT},
131	/* 0x24 '$' */ {0x05, KBDCHARDEF_MOD_SHIFT},
132	/* 0x25 '%' */ {0x06, KBDCHARDEF_MOD_SHIFT},
133	/* 0x26 '&' */ {0x08, KBDCHARDEF_MOD_SHIFT},
134	/* 0x27 ''' */ {0x28, KBDCHARDEF_MOD_NONE},
135	/* 0x28 '(' */ {0x0a, KBDCHARDEF_MOD_SHIFT},
136	/* 0x29 ')' */ {0x0b, KBDCHARDEF_MOD_SHIFT},
137	/* 0x2A '' / {0x09, KBDCHARDEF_MOD_SHIFT},
138	/* 0x2B '+' */ {0x0d, KBDCHARDEF_MOD_SHIFT},
139	/* 0x2C ',' */ {0x33, KBDCHARDEF_MOD_NONE},
140	/* 0x2D '-' */ {0x0c, KBDCHARDEF_MOD_NONE},
141	/* 0x2E '.' */ {0x34, KBDCHARDEF_MOD_NONE},
142	/* 0x2F '/' */ {0x35, KBDCHARDEF_MOD_NONE},
143	/* 0x30 '0' */ {0x0b, KBDCHARDEF_MOD_NONE},
144	/* 0x31 '1' */ {0x02, KBDCHARDEF_MOD_NONE},
145	/* 0x32 '2' */ {0x03, KBDCHARDEF_MOD_NONE},
146	/* 0x33 '3' */ {0x04, KBDCHARDEF_MOD_NONE},
147	/* 0x34 '4' */ {0x05, KBDCHARDEF_MOD_NONE},
148	/* 0x35 '5' */ {0x06, KBDCHARDEF_MOD_NONE},
149	/* 0x36 '6' */ {0x07, KBDCHARDEF_MOD_NONE},
150	/* 0x37 '7' */ {0x08, KBDCHARDEF_MOD_NONE},
151	/* 0x38 '8' */ {0x09, KBDCHARDEF_MOD_NONE},
152	/* 0x39 '9' */ {0x0a, KBDCHARDEF_MOD_NONE},
153	/* 0x3A ':' */ {0x27, KBDCHARDEF_MOD_SHIFT},
154	/* 0x3B ';' */ {0x27, KBDCHARDEF_MOD_NONE},
155	/* 0x3C '<' */ {0x33, KBDCHARDEF_MOD_SHIFT},
156	/* 0x3D '=' */ {0x0d, KBDCHARDEF_MOD_NONE},
157	/* 0x3E '>' */ {0x34, KBDCHARDEF_MOD_SHIFT},
158	/* 0x3F '?' */ {0x35, KBDCHARDEF_MOD_SHIFT},
159	/* 0x40 '@' */ {0x03, KBDCHARDEF_MOD_SHIFT},
160	/* 0x41 'A' */ {0x1e, KBDCHARDEF_MOD_SHIFT},
161	/* 0x42 'B' */ {0x30, KBDCHARDEF_MOD_SHIFT},
162	/* 0x43 'C' */ {0x2e, KBDCHARDEF_MOD_SHIFT},
163	/* 0x44 'D' */ {0x20, KBDCHARDEF_MOD_SHIFT},
164	/* 0x45 'E' */ {0x12, KBDCHARDEF_MOD_SHIFT},
165	/* 0x46 'F' */ {0x21, KBDCHARDEF_MOD_SHIFT},
166	/* 0x47 'G' */ {0x22, KBDCHARDEF_MOD_SHIFT},
167	/* 0x48 'H' */ {0x23, KBDCHARDEF_MOD_SHIFT},
168	/* 0x49 'I' */ {0x17, KBDCHARDEF_MOD_SHIFT},
169	/* 0x4A 'J' */ {0x24, KBDCHARDEF_MOD_SHIFT},
170	/* 0x4B 'K' */ {0x25, KBDCHARDEF_MOD_SHIFT},
171	/* 0x4C 'L' */ {0x26, KBDCHARDEF_MOD_SHIFT},
172	/* 0x4D 'M' */ {0x32, KBDCHARDEF_MOD_SHIFT},
173	/* 0x4E 'N' */ {0x31, KBDCHARDEF_MOD_SHIFT},
174	/* 0x4F 'O' */ {0x18, KBDCHARDEF_MOD_SHIFT},
175	/* 0x50 'P' */ {0x19, KBDCHARDEF_MOD_SHIFT},
176	/* 0x51 'Q' */ {0x10, KBDCHARDEF_MOD_SHIFT},
177	/* 0x52 'R' */ {0x13, KBDCHARDEF_MOD_SHIFT},
178	/* 0x53 'S' */ {0x1f, KBDCHARDEF_MOD_SHIFT},
179	/* 0x54 'T' */ {0x14, KBDCHARDEF_MOD_SHIFT},
180	/* 0x55 'U' */ {0x16, KBDCHARDEF_MOD_SHIFT},
181	/* 0x56 'V' */ {0x2f, KBDCHARDEF_MOD_SHIFT},
182	/* 0x57 'W' */ {0x11, KBDCHARDEF_MOD_SHIFT},
183	/* 0x58 'X' */ {0x2d, KBDCHARDEF_MOD_SHIFT},
184	/* 0x59 'Y' */ {0x15, KBDCHARDEF_MOD_SHIFT},
185	/* 0x5A 'Z' */ {0x2c, KBDCHARDEF_MOD_SHIFT},
186	/* 0x5B '[' */ {0x1a, KBDCHARDEF_MOD_NONE},
187	/* 0x5C '\' */ {0x2b, KBDCHARDEF_MOD_NONE},
188	/* 0x5D ']' */ {0x1b, KBDCHARDEF_MOD_NONE},
189	/* 0x5E '^' */ {0x07, KBDCHARDEF_MOD_SHIFT},
190	/* 0x5F '_' */ {0x0c, KBDCHARDEF_MOD_SHIFT},
191	/* 0x60 '`' */ {0x28, KBDCHARDEF_MOD_NONE},
192	/* 0x61 'a' */ {0x1e, KBDCHARDEF_MOD_NONE},
193	/* 0x62 'b' */ {0x30, KBDCHARDEF_MOD_NONE},
194	/* 0x63 'c' */ {0x2e, KBDCHARDEF_MOD_NONE},
195	/* 0x64 'd' */ {0x20, KBDCHARDEF_MOD_NONE},
196	/* 0x65 'e' */ {0x12, KBDCHARDEF_MOD_NONE},
197	/* 0x66 'f' */ {0x21, KBDCHARDEF_MOD_NONE},
198	/* 0x67 'g' */ {0x22, KBDCHARDEF_MOD_NONE},
199	/* 0x68 'h' */ {0x23, KBDCHARDEF_MOD_NONE},
200	/* 0x69 'i' */ {0x17, KBDCHARDEF_MOD_NONE},
201	/* 0x6A 'j' */ {0x24, KBDCHARDEF_MOD_NONE},
202	/* 0x6B 'k' */ {0x25, KBDCHARDEF_MOD_NONE},
203	/* 0x6C 'l' */ {0x26, KBDCHARDEF_MOD_NONE},
204	/* 0x6D 'm' */ {0x32, KBDCHARDEF_MOD_NONE},
205	/* 0x6E 'n' */ {0x31, KBDCHARDEF_MOD_NONE},
206	/* 0x6F 'o' */ {0x18, KBDCHARDEF_MOD_NONE},
207	/* 0x70 'p' */ {0x19, KBDCHARDEF_MOD_NONE},
208	/* 0x71 'q' */ {0x10, KBDCHARDEF_MOD_NONE},
209	/* 0x72 'r' */ {0x13, KBDCHARDEF_MOD_NONE},
210	/* 0x73 's' */ {0x1f, KBDCHARDEF_MOD_NONE},
211	/* 0x74 't' */ {0x14, KBDCHARDEF_MOD_NONE},
212	/* 0x75 'u' */ {0x16, KBDCHARDEF_MOD_NONE},
213	/* 0x76 'v' */ {0x2f, KBDCHARDEF_MOD_NONE},
214	/* 0x77 'w' */ {0x11, KBDCHARDEF_MOD_NONE},
215	/* 0x78 'x' */ {0x2d, KBDCHARDEF_MOD_NONE},
216	/* 0x79 'y' */ {0x15, KBDCHARDEF_MOD_NONE},
217	/* 0x7A 'z' */ {0x2c, KBDCHARDEF_MOD_NONE},
218	/* 0x7B '{' */ {0x1a, KBDCHARDEF_MOD_SHIFT},
219	/* 0x7C '\|' */ {0x2b, KBDCHARDEF_MOD_SHIFT},
220	/* 0x7D '}' */ {0x1b, KBDCHARDEF_MOD_SHIFT},
221	/* 0x7E '~' */ {0x29, KBDCHARDEF_MOD_SHIFT},
222	/* 0x7F ' ' */ {0x00, KBDCHARDEF_MOD_NONE},
223	};
224
225	static HRESULT keyboardPutScancodes(IKeyboard *pKeyboard, const std::list<LONG> &llScancodes)
226	{
227	/* Send scancodes to the VM. */
228	com::SafeArray<LONG> saScancodes(llScancodes);
229
230	#if 1
231	HRESULT rc = S_OK;
232	size_t i;
233	for (i = 0; i < saScancodes.size(); ++i)
234	{
235	rc = pKeyboard->PutScancode(saScancodes[i]);
236	if (FAILED(rc))
237	{
238	RTMsgError("Failed to send a scancode");
239	break;
240	}
241
242	RTThreadSleep(10); /* "Typing" too fast causes lost characters. */
243	}
244	#else
245	/** @todo PutScancodes does not deliver more than 20 scancodes. */
246	ULONG codesStored = 0;
247	HRESULT rc = pKeyboard->PutScancodes(ComSafeArrayAsInParam(saScancodes),
248	&codesStored);
249	if (SUCCEEDED(rc) && codesStored < saScancodes.size())
250	{
251	RTMsgError("Only %d scancodes were stored", codesStored);
252	rc = E_FAIL;
253	}
254	#endif
255
256	return rc;
257	}
258
259	static void keyboardCharsToScancodes(const char pch, size_t cchMax, std::list<LONG> &llScancodes, bool pfShift)
260	{
261	size_t cchProcessed = 0;
262	const char *p = pch;
263	while (cchProcessed < cchMax)
264	{
265	++cchProcessed;
266	const uint8_t c = (uint8_t)*p++;
267	if (c < RT_ELEMENTS(g_aASCIIChars))
268	{
269	const KBDCHARDEF *d = &g_aASCIIChars[c];
270	if (d->u8Scancode)
271	{
272	const bool fNeedShift = RT_BOOL(d->u8Modifiers & KBDCHARDEF_MOD_SHIFT);
273	if (*pfShift != fNeedShift)
274	{
275	*pfShift = fNeedShift;
276	/* Press or release the SHIFT key. */
277	llScancodes.push_back(0x2a \| (fNeedShift? 0x00: 0x80));
278	}
279
280	llScancodes.push_back(d->u8Scancode);
281	llScancodes.push_back(d->u8Scancode \| 0x80);
282	}
283	}
284	}
285	}
286
287	static HRESULT keyboardPutString(IKeyboard pKeyboard, int argc, char *argv)
288	{
289	std::list<LONG> llScancodes;
290	bool fShift = false;
291
292	/* Convert command line string(s) to the en-us keyboard scancodes. */
293	int i;
294	for (i = 1 + 1; i < argc; ++i)
295	{
296	if (llScancodes.size() > 0)
297	{
298	/* Insert a SPACE before the next string. */
299	llScancodes.push_back(0x39);
300	llScancodes.push_back(0x39 \| 0x80);
301	}
302
303	keyboardCharsToScancodes(argv[i], strlen(argv[i]), llScancodes, &fShift);
304	}
305
306	/* Release SHIFT if pressed. */
307	if (fShift)
308	llScancodes.push_back(0x2a \| 0x80);
309
310	return keyboardPutScancodes(pKeyboard, llScancodes);
311	}
312
313	static HRESULT keyboardPutFile(IKeyboard pKeyboard, const char pszFilename)
314	{
315	std::list<LONG> llScancodes;
316	bool fShift = false;
317
318	RTFILE File = NIL_RTFILE;
319	int vrc = RTFileOpen(&File, pszFilename, RTFILE_O_READ \| RTFILE_O_OPEN \| RTFILE_O_DENY_WRITE);
320	if (RT_SUCCESS(vrc))
321	{
322	uint64_t cbFile = 0;
323	vrc = RTFileGetSize(File, &cbFile);
324	if (RT_SUCCESS(vrc))
325	{
326	const uint64_t cbFileMax = _64K;
327	if (cbFile <= cbFileMax)
328	{
329	const size_t cbBuffer = _4K;
330	char pchBuf = (char )RTMemAlloc(cbBuffer);
331	if (pchBuf)
332	{
333	size_t cbRemaining = (size_t)cbFile;
334	while (cbRemaining > 0)
335	{
336	const size_t cbToRead = cbRemaining > cbBuffer ? cbBuffer : cbRemaining;
337
338	size_t cbRead = 0;
339	vrc = RTFileRead(File, pchBuf, cbToRead, &cbRead);
340	if (RT_FAILURE(vrc) \|\| cbRead == 0)
341	break;
342
343	keyboardCharsToScancodes(pchBuf, cbRead, llScancodes, &fShift);
344	cbRemaining -= cbRead;
345	}
346
347	RTMemFree(pchBuf);
348	}
349	else
350	RTMsgError("Out of memory allocating %d bytes", cbBuffer);
351	}
352	else
353	RTMsgError("File size %RI64 is greater than %RI64: '%s'", cbFile, cbFileMax, pszFilename);
354	}
355	else
356	RTMsgError("Cannot get size of file '%s': %Rrc", pszFilename, vrc);
357
358	RTFileClose(File);
359	}
360	else
361	RTMsgError("Cannot open file '%s': %Rrc", pszFilename, vrc);
362
363	/* Release SHIFT if pressed. */
364	if (fShift)
365	llScancodes.push_back(0x2a \| 0x80);
366
367	return keyboardPutScancodes(pKeyboard, llScancodes);
368	}
369
370
371	RTEXITCODE handleControlVM(HandlerArg *a)
372	{
373	using namespace com;
374	bool fNeedsSaving = false;
375	HRESULT rc;
376
377	if (a->argc < 2)
378	return errorSyntax(USAGE_CONTROLVM, "Not enough parameters");
379
380	/* try to find the given machine */
381	ComPtr<IMachine> machine;
382	CHECK_ERROR(a->virtualBox, FindMachine(Bstr(a->argv[0]).raw(),
383	machine.asOutParam()));
384	if (FAILED(rc))
385	return RTEXITCODE_FAILURE;
386
387	/* open a session for the VM */
388	CHECK_ERROR_RET(machine, LockMachine(a->session, LockType_Shared), RTEXITCODE_FAILURE);
389
390	ComPtr<IConsole> console;
391	ComPtr<IMachine> sessionMachine;
392
393	do
394	{
395	/* get the associated console */
396	CHECK_ERROR_BREAK(a->session, COMGETTER(Console)(console.asOutParam()));
397	/* ... and session machine */
398	CHECK_ERROR_BREAK(a->session, COMGETTER(Machine)(sessionMachine.asOutParam()));
399
400	if (!console)
401	return RTMsgErrorExit(RTEXITCODE_FAILURE, "Machine '%s' is not currently running", a->argv[0]);
402
403	/* which command? */
404	if (!strcmp(a->argv[1], "pause"))
405	{
406	CHECK_ERROR_BREAK(console, Pause());
407	}
408	else if (!strcmp(a->argv[1], "resume"))
409	{
410	CHECK_ERROR_BREAK(console, Resume());
411	}
412	else if (!strcmp(a->argv[1], "reset"))
413	{
414	CHECK_ERROR_BREAK(console, Reset());
415	}
416	else if (!strcmp(a->argv[1], "unplugcpu"))
417	{
418	if (a->argc <= 1 + 1)
419	{
420	errorArgument("Missing argument to '%s'. Expected CPU number.", a->argv[1]);
421	rc = E_FAIL;
422	break;
423	}
424
425	unsigned n = parseNum(a->argv[2], 32, "CPU");
426
427	CHECK_ERROR_BREAK(sessionMachine, HotUnplugCPU(n));
428	}
429	else if (!strcmp(a->argv[1], "plugcpu"))
430	{
431	if (a->argc <= 1 + 1)
432	{
433	errorArgument("Missing argument to '%s'. Expected CPU number.", a->argv[1]);
434	rc = E_FAIL;
435	break;
436	}
437
438	unsigned n = parseNum(a->argv[2], 32, "CPU");
439
440	CHECK_ERROR_BREAK(sessionMachine, HotPlugCPU(n));
441	}
442	else if (!strcmp(a->argv[1], "cpuexecutioncap"))
443	{
444	if (a->argc <= 1 + 1)
445	{
446	errorArgument("Missing argument to '%s'. Expected execution cap number.", a->argv[1]);
447	rc = E_FAIL;
448	break;
449	}
450
451	unsigned n = parseNum(a->argv[2], 100, "ExecutionCap");
452
453	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(CPUExecutionCap)(n));
454	}
455	else if (!strcmp(a->argv[1], "audioin"))
456	{
457	ComPtr<IAudioAdapter> adapter;
458	CHECK_ERROR_BREAK(sessionMachine, COMGETTER(AudioAdapter)(adapter.asOutParam()));
459	if (adapter)
460	{
461	if (!strcmp(a->argv[2], "on"))
462	{
463	CHECK_ERROR_RET(adapter, COMSETTER(EnabledIn)(TRUE), RTEXITCODE_FAILURE);
464	}
465	else if (!strcmp(a->argv[2], "off"))
466	{
467	CHECK_ERROR_RET(adapter, COMSETTER(EnabledIn)(FALSE), RTEXITCODE_FAILURE);
468	}
469	else
470	{
471	errorArgument("Invalid value '%s'", Utf8Str(a->argv[2]).c_str());
472	rc = E_FAIL;
473	break;
474	}
475	if (SUCCEEDED(rc))
476	fNeedsSaving = true;
477	}
478	else
479	{
480	errorArgument("audio adapter not enabled in VM configuration");
481	rc = E_FAIL;
482	break;
483	}
484	}
485	else if (!strcmp(a->argv[1], "audioout"))
486	{
487	ComPtr<IAudioAdapter> adapter;
488	CHECK_ERROR_BREAK(sessionMachine, COMGETTER(AudioAdapter)(adapter.asOutParam()));
489	if (adapter)
490	{
491	if (!strcmp(a->argv[2], "on"))
492	{
493	CHECK_ERROR_RET(adapter, COMSETTER(EnabledOut)(TRUE), RTEXITCODE_FAILURE);
494	}
495	else if (!strcmp(a->argv[2], "off"))
496	{
497	CHECK_ERROR_RET(adapter, COMSETTER(EnabledOut)(FALSE), RTEXITCODE_FAILURE);
498	}
499	else
500	{
501	errorArgument("Invalid value '%s'", Utf8Str(a->argv[2]).c_str());
502	rc = E_FAIL;
503	break;
504	}
505	if (SUCCEEDED(rc))
506	fNeedsSaving = true;
507	}
508	else
509	{
510	errorArgument("audio adapter not enabled in VM configuration");
511	rc = E_FAIL;
512	break;
513	}
514	}
515	else if (!strcmp(a->argv[1], "clipboard"))
516	{
517	if (a->argc <= 1 + 1)
518	{
519	errorArgument("Missing argument to '%s'. Expected clipboard mode.", a->argv[1]);
520	rc = E_FAIL;
521	break;
522	}
523
524	ClipboardMode_T mode = ClipboardMode_Disabled; /* Shut up MSC */
525	if (!strcmp(a->argv[2], "disabled"))
526	mode = ClipboardMode_Disabled;
527	else if (!strcmp(a->argv[2], "hosttoguest"))
528	mode = ClipboardMode_HostToGuest;
529	else if (!strcmp(a->argv[2], "guesttohost"))
530	mode = ClipboardMode_GuestToHost;
531	else if (!strcmp(a->argv[2], "bidirectional"))
532	mode = ClipboardMode_Bidirectional;
533	else
534	{
535	errorArgument("Invalid '%s' argument '%s'.", a->argv[1], a->argv[2]);
536	rc = E_FAIL;
537	}
538	if (SUCCEEDED(rc))
539	{
540	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(ClipboardMode)(mode));
541	if (SUCCEEDED(rc))
542	fNeedsSaving = true;
543	}
544	}
545	else if (!strcmp(a->argv[1], "draganddrop"))
546	{
547	if (a->argc <= 1 + 1)
548	{
549	errorArgument("Missing argument to '%s'. Expected drag and drop mode.", a->argv[1]);
550	rc = E_FAIL;
551	break;
552	}
553
554	DnDMode_T mode = DnDMode_Disabled; /* Shup up MSC. */
555	if (!strcmp(a->argv[2], "disabled"))
556	mode = DnDMode_Disabled;
557	else if (!strcmp(a->argv[2], "hosttoguest"))
558	mode = DnDMode_HostToGuest;
559	else if (!strcmp(a->argv[2], "guesttohost"))
560	mode = DnDMode_GuestToHost;
561	else if (!strcmp(a->argv[2], "bidirectional"))
562	mode = DnDMode_Bidirectional;
563	else
564	{
565	errorArgument("Invalid '%s' argument '%s'.", a->argv[1], a->argv[2]);
566	rc = E_FAIL;
567	}
568	if (SUCCEEDED(rc))
569	{
570	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(DnDMode)(mode));
571	if (SUCCEEDED(rc))
572	fNeedsSaving = true;
573	}
574	}
575	else if (!strcmp(a->argv[1], "poweroff"))
576	{
577	ComPtr<IProgress> progress;
578	CHECK_ERROR_BREAK(console, PowerDown(progress.asOutParam()));
579
580	rc = showProgress(progress);
581	CHECK_PROGRESS_ERROR(progress, ("Failed to power off machine"));
582	}
583	else if (!strcmp(a->argv[1], "savestate"))
584	{
585	/* first pause so we don't trigger a live save which needs more time/resources */
586	bool fPaused = false;
587	rc = console->Pause();
588	if (FAILED(rc))
589	{
590	bool fError = true;
591	if (rc == VBOX_E_INVALID_VM_STATE)
592	{
593	/* check if we are already paused */
594	MachineState_T machineState;
595	CHECK_ERROR_BREAK(console, COMGETTER(State)(&machineState));
596	/* the error code was lost by the previous instruction */
597	rc = VBOX_E_INVALID_VM_STATE;
598	if (machineState != MachineState_Paused)
599	{
600	RTMsgError("Machine in invalid state %d -- %s\n",
601	machineState, machineStateToName(machineState, false));
602	}
603	else
604	{
605	fError = false;
606	fPaused = true;
607	}
608	}
609	if (fError)
610	break;
611	}
612
613	ComPtr<IProgress> progress;
614	CHECK_ERROR(sessionMachine, SaveState(progress.asOutParam()));
615	if (FAILED(rc))
616	{
617	if (!fPaused)
618	console->Resume();
619	break;
620	}
621
622	rc = showProgress(progress);
623	CHECK_PROGRESS_ERROR(progress, ("Failed to save machine state"));
624	if (FAILED(rc))
625	{
626	if (!fPaused)
627	console->Resume();
628	}
629	}
630	else if (!strcmp(a->argv[1], "acpipowerbutton"))
631	{
632	CHECK_ERROR_BREAK(console, PowerButton());
633	}
634	else if (!strcmp(a->argv[1], "acpisleepbutton"))
635	{
636	CHECK_ERROR_BREAK(console, SleepButton());
637	}
638	else if (!strcmp(a->argv[1], "keyboardputscancode"))
639	{
640	ComPtr<IKeyboard> pKeyboard;
641	CHECK_ERROR_BREAK(console, COMGETTER(Keyboard)(pKeyboard.asOutParam()));
642	if (!pKeyboard)
643	{
644	RTMsgError("Guest not running");
645	rc = E_FAIL;
646	break;
647	}
648
649	if (a->argc <= 1 + 1)
650	{
651	errorArgument("Missing argument to '%s'. Expected IBM PC AT set 2 keyboard scancode(s) as hex byte(s).", a->argv[1]);
652	rc = E_FAIL;
653	break;
654	}
655
656	std::list<LONG> llScancodes;
657
658	/* Process the command line. */
659	int i;
660	for (i = 1 + 1; i < a->argc; i++)
661	{
662	if ( RT_C_IS_XDIGIT (a->argv[i][0])
663	&& RT_C_IS_XDIGIT (a->argv[i][1])
664	&& a->argv[i][2] == 0)
665	{
666	uint8_t u8Scancode;
667	int irc = RTStrToUInt8Ex(a->argv[i], NULL, 16, &u8Scancode);
668	if (RT_FAILURE (irc))
669	{
670	RTMsgError("Converting '%s' returned %Rrc!", a->argv[i], rc);
671	rc = E_FAIL;
672	break;
673	}
674
675	llScancodes.push_back(u8Scancode);
676	}
677	else
678	{
679	RTMsgError("Error: '%s' is not a hex byte!", a->argv[i]);
680	rc = E_FAIL;
681	break;
682	}
683	}
684
685	if (FAILED(rc))
686	break;
687
688	rc = keyboardPutScancodes(pKeyboard, llScancodes);
689	}
690	else if (!strcmp(a->argv[1], "keyboardputstring"))
691	{
692	ComPtr<IKeyboard> pKeyboard;
693	CHECK_ERROR_BREAK(console, COMGETTER(Keyboard)(pKeyboard.asOutParam()));
694	if (!pKeyboard)
695	{
696	RTMsgError("Guest not running");
697	rc = E_FAIL;
698	break;
699	}
700
701	if (a->argc <= 1 + 1)
702	{
703	errorArgument("Missing argument to '%s'. Expected ASCII string(s).", a->argv[1]);
704	rc = E_FAIL;
705	break;
706	}
707
708	rc = keyboardPutString(pKeyboard, a->argc, a->argv);
709	}
710	else if (!strcmp(a->argv[1], "keyboardputfile"))
711	{
712	ComPtr<IKeyboard> pKeyboard;
713	CHECK_ERROR_BREAK(console, COMGETTER(Keyboard)(pKeyboard.asOutParam()));
714	if (!pKeyboard)
715	{
716	RTMsgError("Guest not running");
717	rc = E_FAIL;
718	break;
719	}
720
721	if (a->argc <= 1 + 1)
722	{
723	errorArgument("Missing argument to '%s'. Expected file name.", a->argv[1]);
724	rc = E_FAIL;
725	break;
726	}
727
728	rc = keyboardPutFile(pKeyboard, a->argv[2]);
729	}
730	else if (!strncmp(a->argv[1], "setlinkstate", 12))
731	{
732	/* Get the number of network adapters */
733	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
734	unsigned n = parseNum(&a->argv[1][12], NetworkAdapterCount, "NIC");
735	if (!n)
736	{
737	rc = E_FAIL;
738	break;
739	}
740	if (a->argc <= 1 + 1)
741	{
742	errorArgument("Missing argument to '%s'", a->argv[1]);
743	rc = E_FAIL;
744	break;
745	}
746	/* get the corresponding network adapter */
747	ComPtr<INetworkAdapter> adapter;
748	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
749	if (adapter)
750	{
751	if (!strcmp(a->argv[2], "on"))
752	{
753	CHECK_ERROR_BREAK(adapter, COMSETTER(CableConnected)(TRUE));
754	}
755	else if (!strcmp(a->argv[2], "off"))
756	{
757	CHECK_ERROR_BREAK(adapter, COMSETTER(CableConnected)(FALSE));
758	}
759	else
760	{
761	errorArgument("Invalid link state '%s'", Utf8Str(a->argv[2]).c_str());
762	rc = E_FAIL;
763	break;
764	}
765	if (SUCCEEDED(rc))
766	fNeedsSaving = true;
767	}
768	}
769	/* here the order in which strncmp is called is important
770	* cause nictracefile can be very well compared with
771	* nictrace and nic and thus everything will always fail
772	* if the order is changed
773	*/
774	else if (!strncmp(a->argv[1], "nictracefile", 12))
775	{
776	/* Get the number of network adapters */
777	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
778	unsigned n = parseNum(&a->argv[1][12], NetworkAdapterCount, "NIC");
779	if (!n)
780	{
781	rc = E_FAIL;
782	break;
783	}
784	if (a->argc <= 2)
785	{
786	errorArgument("Missing argument to '%s'", a->argv[1]);
787	rc = E_FAIL;
788	break;
789	}
790
791	/* get the corresponding network adapter */
792	ComPtr<INetworkAdapter> adapter;
793	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
794	if (adapter)
795	{
796	BOOL fEnabled;
797	adapter->COMGETTER(Enabled)(&fEnabled);
798	if (fEnabled)
799	{
800	if (a->argv[2])
801	{
802	CHECK_ERROR_RET(adapter, COMSETTER(TraceFile)(Bstr(a->argv[2]).raw()), RTEXITCODE_FAILURE);
803	}
804	else
805	{
806	errorArgument("Invalid filename or filename not specified for NIC %lu", n);
807	rc = E_FAIL;
808	break;
809	}
810	if (SUCCEEDED(rc))
811	fNeedsSaving = true;
812	}
813	else
814	RTMsgError("The NIC %d is currently disabled and thus its tracefile can't be changed", n);
815	}
816	}
817	else if (!strncmp(a->argv[1], "nictrace", 8))
818	{
819	/* Get the number of network adapters */
820	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
821	unsigned n = parseNum(&a->argv[1][8], NetworkAdapterCount, "NIC");
822	if (!n)
823	{
824	rc = E_FAIL;
825	break;
826	}
827	if (a->argc <= 2)
828	{
829	errorArgument("Missing argument to '%s'", a->argv[1]);
830	rc = E_FAIL;
831	break;
832	}
833
834	/* get the corresponding network adapter */
835	ComPtr<INetworkAdapter> adapter;
836	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
837	if (adapter)
838	{
839	BOOL fEnabled;
840	adapter->COMGETTER(Enabled)(&fEnabled);
841	if (fEnabled)
842	{
843	if (!strcmp(a->argv[2], "on"))
844	{
845	CHECK_ERROR_RET(adapter, COMSETTER(TraceEnabled)(TRUE), RTEXITCODE_FAILURE);
846	}
847	else if (!strcmp(a->argv[2], "off"))
848	{
849	CHECK_ERROR_RET(adapter, COMSETTER(TraceEnabled)(FALSE), RTEXITCODE_FAILURE);
850	}
851	else
852	{
853	errorArgument("Invalid nictrace%lu argument '%s'", n, Utf8Str(a->argv[2]).c_str());
854	rc = E_FAIL;
855	break;
856	}
857	if (SUCCEEDED(rc))
858	fNeedsSaving = true;
859	}
860	else
861	RTMsgError("The NIC %d is currently disabled and thus its trace flag can't be changed", n);
862	}
863	}
864	else if( a->argc > 2
865	&& !strncmp(a->argv[1], "natpf", 5))
866	{
867	/* Get the number of network adapters */
868	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
869	unsigned n = parseNum(&a->argv[1][5], NetworkAdapterCount, "NIC");
870	if (!n)
871	{
872	rc = E_FAIL;
873	break;
874	}
875	if (a->argc <= 2)
876	{
877	errorArgument("Missing argument to '%s'", a->argv[1]);
878	rc = E_FAIL;
879	break;
880	}
881
882	/* get the corresponding network adapter */
883	ComPtr<INetworkAdapter> adapter;
884	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
885	if (!adapter)
886	{
887	rc = E_FAIL;
888	break;
889	}
890	ComPtr<INATEngine> engine;
891	CHECK_ERROR(adapter, COMGETTER(NATEngine)(engine.asOutParam()));
892	if (!engine)
893	{
894	rc = E_FAIL;
895	break;
896	}
897
898	if (!strcmp(a->argv[2], "delete"))
899	{
900	if (a->argc >= 3)
901	CHECK_ERROR(engine, RemoveRedirect(Bstr(a->argv[3]).raw()));
902	}
903	else
904	{
905	#define ITERATE_TO_NEXT_TERM(ch) \
906	do { \
907	while (*ch != ',') \
908	{ \
909	if (*ch == 0) \
910	{ \
911	return errorSyntax(USAGE_CONTROLVM, \
912	"Missing or invalid argument to '%s'", \
913	a->argv[1]); \
914	} \
915	ch++; \
916	} \
917	*ch = '\0'; \
918	ch++; \
919	} while(0)
920
921	char *strName;
922	char *strProto;
923	char *strHostIp;
924	char *strHostPort;
925	char *strGuestIp;
926	char *strGuestPort;
927	char *strRaw = RTStrDup(a->argv[2]);
928	char *ch = strRaw;
929	strName = RTStrStrip(ch);
930	ITERATE_TO_NEXT_TERM(ch);
931	strProto = RTStrStrip(ch);
932	ITERATE_TO_NEXT_TERM(ch);
933	strHostIp = RTStrStrip(ch);
934	ITERATE_TO_NEXT_TERM(ch);
935	strHostPort = RTStrStrip(ch);
936	ITERATE_TO_NEXT_TERM(ch);
937	strGuestIp = RTStrStrip(ch);
938	ITERATE_TO_NEXT_TERM(ch);
939	strGuestPort = RTStrStrip(ch);
940	NATProtocol_T proto;
941	if (RTStrICmp(strProto, "udp") == 0)
942	proto = NATProtocol_UDP;
943	else if (RTStrICmp(strProto, "tcp") == 0)
944	proto = NATProtocol_TCP;
945	else
946	{
947	return errorSyntax(USAGE_CONTROLVM,
948	"Wrong rule proto '%s' specified -- only 'udp' and 'tcp' are allowed.",
949	strProto);
950	}
951	CHECK_ERROR(engine, AddRedirect(Bstr(strName).raw(), proto, Bstr(strHostIp).raw(),
952	RTStrToUInt16(strHostPort), Bstr(strGuestIp).raw(), RTStrToUInt16(strGuestPort)));
953	#undef ITERATE_TO_NEXT_TERM
954	}
955	if (SUCCEEDED(rc))
956	fNeedsSaving = true;
957	}
958	else if (!strncmp(a->argv[1], "nicproperty", 11))
959	{
960	/* Get the number of network adapters */
961	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
962	unsigned n = parseNum(&a->argv[1][11], NetworkAdapterCount, "NIC");
963	if (!n)
964	{
965	rc = E_FAIL;
966	break;
967	}
968	if (a->argc <= 2)
969	{
970	errorArgument("Missing argument to '%s'", a->argv[1]);
971	rc = E_FAIL;
972	break;
973	}
974
975	/* get the corresponding network adapter */
976	ComPtr<INetworkAdapter> adapter;
977	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
978	if (adapter)
979	{
980	BOOL fEnabled;
981	adapter->COMGETTER(Enabled)(&fEnabled);
982	if (fEnabled)
983	{
984	/* Parse 'name=value' */
985	char *pszProperty = RTStrDup(a->argv[2]);
986	if (pszProperty)
987	{
988	char *pDelimiter = strchr(pszProperty, '=');
989	if (pDelimiter)
990	{
991	*pDelimiter = '\0';
992
993	Bstr bstrName = pszProperty;
994	Bstr bstrValue = &pDelimiter[1];
995	CHECK_ERROR(adapter, SetProperty(bstrName.raw(), bstrValue.raw()));
996	if (SUCCEEDED(rc))
997	fNeedsSaving = true;
998	}
999	else
1000	{
1001	errorArgument("Invalid nicproperty%d argument '%s'", n, a->argv[2]);
1002	rc = E_FAIL;
1003	}
1004	RTStrFree(pszProperty);
1005	}
1006	else
1007	{
1008	RTStrmPrintf(g_pStdErr, "Error: Failed to allocate memory for nicproperty%d '%s'\n", n, a->argv[2]);
1009	rc = E_FAIL;
1010	}
1011	if (FAILED(rc))
1012	break;
1013	}
1014	else
1015	RTMsgError("The NIC %d is currently disabled and thus its properties can't be changed", n);
1016	}
1017	}
1018	else if (!strncmp(a->argv[1], "nicpromisc", 10))
1019	{
1020	/* Get the number of network adapters */
1021	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
1022	unsigned n = parseNum(&a->argv[1][10], NetworkAdapterCount, "NIC");
1023	if (!n)
1024	{
1025	rc = E_FAIL;
1026	break;
1027	}
1028	if (a->argc <= 2)
1029	{
1030	errorArgument("Missing argument to '%s'", a->argv[1]);
1031	rc = E_FAIL;
1032	break;
1033	}
1034
1035	/* get the corresponding network adapter */
1036	ComPtr<INetworkAdapter> adapter;
1037	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
1038	if (adapter)
1039	{
1040	BOOL fEnabled;
1041	adapter->COMGETTER(Enabled)(&fEnabled);
1042	if (fEnabled)
1043	{
1044	NetworkAdapterPromiscModePolicy_T enmPromiscModePolicy;
1045	if (!strcmp(a->argv[2], "deny"))
1046	enmPromiscModePolicy = NetworkAdapterPromiscModePolicy_Deny;
1047	else if ( !strcmp(a->argv[2], "allow-vms")
1048	\|\| !strcmp(a->argv[2], "allow-network"))
1049	enmPromiscModePolicy = NetworkAdapterPromiscModePolicy_AllowNetwork;
1050	else if (!strcmp(a->argv[2], "allow-all"))
1051	enmPromiscModePolicy = NetworkAdapterPromiscModePolicy_AllowAll;
1052	else
1053	{
1054	errorArgument("Unknown promiscuous mode policy '%s'", a->argv[2]);
1055	rc = E_INVALIDARG;
1056	break;
1057	}
1058
1059	CHECK_ERROR(adapter, COMSETTER(PromiscModePolicy)(enmPromiscModePolicy));
1060	if (SUCCEEDED(rc))
1061	fNeedsSaving = true;
1062	}
1063	else
1064	RTMsgError("The NIC %d is currently disabled and thus its promiscuous mode can't be changed", n);
1065	}
1066	}
1067	else if (!strncmp(a->argv[1], "nic", 3))
1068	{
1069	/* Get the number of network adapters */
1070	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
1071	unsigned n = parseNum(&a->argv[1][3], NetworkAdapterCount, "NIC");
1072	if (!n)
1073	{
1074	rc = E_FAIL;
1075	break;
1076	}
1077	if (a->argc <= 2)
1078	{
1079	errorArgument("Missing argument to '%s'", a->argv[1]);
1080	rc = E_FAIL;
1081	break;
1082	}
1083
1084	/* get the corresponding network adapter */
1085	ComPtr<INetworkAdapter> adapter;
1086	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
1087	if (adapter)
1088	{
1089	BOOL fEnabled;
1090	adapter->COMGETTER(Enabled)(&fEnabled);
1091	if (fEnabled)
1092	{
1093	if (!strcmp(a->argv[2], "null"))
1094	{
1095	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_Null), RTEXITCODE_FAILURE);
1096	}
1097	else if (!strcmp(a->argv[2], "nat"))
1098	{
1099	if (a->argc == 4)
1100	CHECK_ERROR_RET(adapter, COMSETTER(NATNetwork)(Bstr(a->argv[3]).raw()), RTEXITCODE_FAILURE);
1101	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_NAT), RTEXITCODE_FAILURE);
1102	}
1103	else if ( !strcmp(a->argv[2], "bridged")
1104	\|\| !strcmp(a->argv[2], "hostif")) /* backward compatibility */
1105	{
1106	if (a->argc <= 3)
1107	{
1108	errorArgument("Missing argument to '%s'", a->argv[2]);
1109	rc = E_FAIL;
1110	break;
1111	}
1112	CHECK_ERROR_RET(adapter, COMSETTER(BridgedInterface)(Bstr(a->argv[3]).raw()), RTEXITCODE_FAILURE);
1113	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_Bridged), RTEXITCODE_FAILURE);
1114	}
1115	else if (!strcmp(a->argv[2], "intnet"))
1116	{
1117	if (a->argc <= 3)
1118	{
1119	errorArgument("Missing argument to '%s'", a->argv[2]);
1120	rc = E_FAIL;
1121	break;
1122	}
1123	CHECK_ERROR_RET(adapter, COMSETTER(InternalNetwork)(Bstr(a->argv[3]).raw()), RTEXITCODE_FAILURE);
1124	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_Internal), RTEXITCODE_FAILURE);
1125	}
1126	#if defined(VBOX_WITH_NETFLT)
1127	else if (!strcmp(a->argv[2], "hostonly"))
1128	{
1129	if (a->argc <= 3)
1130	{
1131	errorArgument("Missing argument to '%s'", a->argv[2]);
1132	rc = E_FAIL;
1133	break;
1134	}
1135	CHECK_ERROR_RET(adapter, COMSETTER(HostOnlyInterface)(Bstr(a->argv[3]).raw()), RTEXITCODE_FAILURE);
1136	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_HostOnly), RTEXITCODE_FAILURE);
1137	}
1138	#endif
1139	else if (!strcmp(a->argv[2], "generic"))
1140	{
1141	if (a->argc <= 3)
1142	{
1143	errorArgument("Missing argument to '%s'", a->argv[2]);
1144	rc = E_FAIL;
1145	break;
1146	}
1147	CHECK_ERROR_RET(adapter, COMSETTER(GenericDriver)(Bstr(a->argv[3]).raw()), RTEXITCODE_FAILURE);
1148	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_Generic), RTEXITCODE_FAILURE);
1149	}
1150	else if (!strcmp(a->argv[2], "natnetwork"))
1151	{
1152	if (a->argc <= 3)
1153	{
1154	errorArgument("Missing argument to '%s'", a->argv[2]);
1155	rc = E_FAIL;
1156	break;
1157	}
1158	CHECK_ERROR_RET(adapter, COMSETTER(NATNetwork)(Bstr(a->argv[3]).raw()), RTEXITCODE_FAILURE);
1159	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_NATNetwork), RTEXITCODE_FAILURE);
1160	}
1161	/** @todo obsolete, remove eventually */
1162	else if (!strcmp(a->argv[2], "vde"))
1163	{
1164	if (a->argc <= 3)
1165	{
1166	errorArgument("Missing argument to '%s'", a->argv[2]);
1167	rc = E_FAIL;
1168	break;
1169	}
1170	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_Generic), RTEXITCODE_FAILURE);
1171	CHECK_ERROR_RET(adapter, SetProperty(Bstr("name").raw(), Bstr(a->argv[3]).raw()), RTEXITCODE_FAILURE);
1172	}
1173	else
1174	{
1175	errorArgument("Invalid type '%s' specfied for NIC %lu", Utf8Str(a->argv[2]).c_str(), n);
1176	rc = E_FAIL;
1177	break;
1178	}
1179	if (SUCCEEDED(rc))
1180	fNeedsSaving = true;
1181	}
1182	else
1183	RTMsgError("The NIC %d is currently disabled and thus its attachment type can't be changed", n);
1184	}
1185	}
1186	else if ( !strcmp(a->argv[1], "vrde")
1187	\|\| !strcmp(a->argv[1], "vrdp"))
1188	{
1189	if (!strcmp(a->argv[1], "vrdp"))
1190	RTStrmPrintf(g_pStdErr, "Warning: 'vrdp' is deprecated. Use 'vrde'.\n");
1191
1192	if (a->argc <= 1 + 1)
1193	{
1194	errorArgument("Missing argument to '%s'", a->argv[1]);
1195	rc = E_FAIL;
1196	break;
1197	}
1198	ComPtr<IVRDEServer> vrdeServer;
1199	sessionMachine->COMGETTER(VRDEServer)(vrdeServer.asOutParam());
1200	ASSERT(vrdeServer);
1201	if (vrdeServer)
1202	{
1203	if (!strcmp(a->argv[2], "on"))
1204	{
1205	CHECK_ERROR_BREAK(vrdeServer, COMSETTER(Enabled)(TRUE));
1206	}
1207	else if (!strcmp(a->argv[2], "off"))
1208	{
1209	CHECK_ERROR_BREAK(vrdeServer, COMSETTER(Enabled)(FALSE));
1210	}
1211	else
1212	{
1213	errorArgument("Invalid remote desktop server state '%s'", Utf8Str(a->argv[2]).c_str());
1214	rc = E_FAIL;
1215	break;
1216	}
1217	if (SUCCEEDED(rc))
1218	fNeedsSaving = true;
1219	}
1220	}
1221	else if ( !strcmp(a->argv[1], "vrdeport")
1222	\|\| !strcmp(a->argv[1], "vrdpport"))
1223	{
1224	if (!strcmp(a->argv[1], "vrdpport"))
1225	RTStrmPrintf(g_pStdErr, "Warning: 'vrdpport' is deprecated. Use 'vrdeport'.\n");
1226
1227	if (a->argc <= 1 + 1)
1228	{
1229	errorArgument("Missing argument to '%s'", a->argv[1]);
1230	rc = E_FAIL;
1231	break;
1232	}
1233
1234	ComPtr<IVRDEServer> vrdeServer;
1235	sessionMachine->COMGETTER(VRDEServer)(vrdeServer.asOutParam());
1236	ASSERT(vrdeServer);
1237	if (vrdeServer)
1238	{
1239	Bstr ports;
1240
1241	if (!strcmp(a->argv[2], "default"))
1242	ports = "0";
1243	else
1244	ports = a->argv[2];
1245
1246	CHECK_ERROR_BREAK(vrdeServer, SetVRDEProperty(Bstr("TCP/Ports").raw(), ports.raw()));
1247	if (SUCCEEDED(rc))
1248	fNeedsSaving = true;
1249	}
1250	}
1251	else if ( !strcmp(a->argv[1], "vrdevideochannelquality")
1252	\|\| !strcmp(a->argv[1], "vrdpvideochannelquality"))
1253	{
1254	if (!strcmp(a->argv[1], "vrdpvideochannelquality"))
1255	RTStrmPrintf(g_pStdErr, "Warning: 'vrdpvideochannelquality' is deprecated. Use 'vrdevideochannelquality'.\n");
1256
1257	if (a->argc <= 1 + 1)
1258	{
1259	errorArgument("Missing argument to '%s'", a->argv[1]);
1260	rc = E_FAIL;
1261	break;
1262	}
1263	ComPtr<IVRDEServer> vrdeServer;
1264	sessionMachine->COMGETTER(VRDEServer)(vrdeServer.asOutParam());
1265	ASSERT(vrdeServer);
1266	if (vrdeServer)
1267	{
1268	Bstr value = a->argv[2];
1269
1270	CHECK_ERROR(vrdeServer, SetVRDEProperty(Bstr("VideoChannel/Quality").raw(), value.raw()));
1271	if (SUCCEEDED(rc))
1272	fNeedsSaving = true;
1273	}
1274	}
1275	else if (!strcmp(a->argv[1], "vrdeproperty"))
1276	{
1277	if (a->argc <= 1 + 1)
1278	{
1279	errorArgument("Missing argument to '%s'", a->argv[1]);
1280	rc = E_FAIL;
1281	break;
1282	}
1283	ComPtr<IVRDEServer> vrdeServer;
1284	sessionMachine->COMGETTER(VRDEServer)(vrdeServer.asOutParam());
1285	ASSERT(vrdeServer);
1286	if (vrdeServer)
1287	{
1288	/* Parse 'name=value' */
1289	char *pszProperty = RTStrDup(a->argv[2]);
1290	if (pszProperty)
1291	{
1292	char *pDelimiter = strchr(pszProperty, '=');
1293	if (pDelimiter)
1294	{
1295	*pDelimiter = '\0';
1296
1297	Bstr bstrName = pszProperty;
1298	Bstr bstrValue = &pDelimiter[1];
1299	CHECK_ERROR(vrdeServer, SetVRDEProperty(bstrName.raw(), bstrValue.raw()));
1300	if (SUCCEEDED(rc))
1301	fNeedsSaving = true;
1302	}
1303	else
1304	{
1305	errorArgument("Invalid vrdeproperty argument '%s'", a->argv[2]);
1306	rc = E_FAIL;
1307	}
1308	RTStrFree(pszProperty);
1309	}
1310	else
1311	{
1312	RTStrmPrintf(g_pStdErr, "Error: Failed to allocate memory for VRDE property '%s'\n", a->argv[2]);
1313	rc = E_FAIL;
1314	}
1315	}
1316	if (FAILED(rc))
1317	{
1318	break;
1319	}
1320	}
1321	else if ( !strcmp(a->argv[1], "usbattach")
1322	\|\| !strcmp(a->argv[1], "usbdetach"))
1323	{
1324	if (a->argc < 3)
1325	{
1326	errorSyntax(USAGE_CONTROLVM, "Not enough parameters");
1327	rc = E_FAIL;
1328	break;
1329	}
1330	else if (a->argc == 4 \|\| a->argc > 5)
1331	{
1332	errorSyntax(USAGE_CONTROLVM, "Wrong number of arguments");
1333	rc = E_FAIL;
1334	break;
1335	}
1336
1337	bool attach = !strcmp(a->argv[1], "usbattach");
1338
1339	Bstr usbId = a->argv[2];
1340	Bstr captureFilename;
1341
1342	if (a->argc == 5)
1343	{
1344	if (!strcmp(a->argv[3], "--capturefile"))
1345	captureFilename = a->argv[4];
1346	else
1347	{
1348	errorArgument("Invalid parameter '%s'", a->argv[3]);
1349	rc = E_FAIL;
1350	break;
1351	}
1352	}
1353
1354	Guid guid(usbId);
1355	if (!guid.isValid())
1356	{
1357	// assume address
1358	if (attach)
1359	{
1360	ComPtr<IHost> host;
1361	CHECK_ERROR_BREAK(a->virtualBox, COMGETTER(Host)(host.asOutParam()));
1362	SafeIfaceArray <IHostUSBDevice> coll;
1363	CHECK_ERROR_BREAK(host, COMGETTER(USBDevices)(ComSafeArrayAsOutParam(coll)));
1364	ComPtr<IHostUSBDevice> dev;
1365	CHECK_ERROR_BREAK(host, FindUSBDeviceByAddress(Bstr(a->argv[2]).raw(),
1366	dev.asOutParam()));
1367	CHECK_ERROR_BREAK(dev, COMGETTER(Id)(usbId.asOutParam()));
1368	}
1369	else
1370	{
1371	SafeIfaceArray <IUSBDevice> coll;
1372	CHECK_ERROR_BREAK(console, COMGETTER(USBDevices)(ComSafeArrayAsOutParam(coll)));
1373	ComPtr<IUSBDevice> dev;
1374	CHECK_ERROR_BREAK(console, FindUSBDeviceByAddress(Bstr(a->argv[2]).raw(),
1375	dev.asOutParam()));
1376	CHECK_ERROR_BREAK(dev, COMGETTER(Id)(usbId.asOutParam()));
1377	}
1378	}
1379	else if (guid.isZero())
1380	{
1381	errorArgument("Zero UUID argument '%s'", a->argv[2]);
1382	rc = E_FAIL;
1383	break;
1384	}
1385
1386	if (attach)
1387	CHECK_ERROR_BREAK(console, AttachUSBDevice(usbId.raw(), captureFilename.raw()));
1388	else
1389	{
1390	ComPtr<IUSBDevice> dev;
1391	CHECK_ERROR_BREAK(console, DetachUSBDevice(usbId.raw(),
1392	dev.asOutParam()));
1393	}
1394	}
1395	else if (!strcmp(a->argv[1], "setvideomodehint"))
1396	{
1397	if (a->argc != 5 && a->argc != 6 && a->argc != 7 && a->argc != 9)
1398	{
1399	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1400	rc = E_FAIL;
1401	break;
1402	}
1403	bool fEnabled = true;
1404	uint32_t uXRes = RTStrToUInt32(a->argv[2]);
1405	uint32_t uYRes = RTStrToUInt32(a->argv[3]);
1406	uint32_t uBpp = RTStrToUInt32(a->argv[4]);
1407	uint32_t uDisplayIdx = 0;
1408	bool fChangeOrigin = false;
1409	int32_t iOriginX = 0;
1410	int32_t iOriginY = 0;
1411	if (a->argc >= 6)
1412	uDisplayIdx = RTStrToUInt32(a->argv[5]);
1413	if (a->argc >= 7)
1414	{
1415	int vrc = parseBool(a->argv[6], &fEnabled);
1416	if (RT_FAILURE(vrc))
1417	{
1418	errorSyntax(USAGE_CONTROLVM, "Either \"yes\" or \"no\" is expected");
1419	rc = E_FAIL;
1420	break;
1421	}
1422	fEnabled = !RTStrICmp(a->argv[6], "yes");
1423	}
1424	if (a->argc == 9)
1425	{
1426	iOriginX = RTStrToInt32(a->argv[7]);
1427	iOriginY = RTStrToInt32(a->argv[8]);
1428	fChangeOrigin = true;
1429	}
1430
1431	ComPtr<IDisplay> pDisplay;
1432	CHECK_ERROR_BREAK(console, COMGETTER(Display)(pDisplay.asOutParam()));
1433	if (!pDisplay)
1434	{
1435	RTMsgError("Guest not running");
1436	rc = E_FAIL;
1437	break;
1438	}
1439	CHECK_ERROR_BREAK(pDisplay, SetVideoModeHint(uDisplayIdx, fEnabled,
1440	fChangeOrigin, iOriginX, iOriginY,
1441	uXRes, uYRes, uBpp));
1442	}
1443	else if (!strcmp(a->argv[1], "setcredentials"))
1444	{
1445	bool fAllowLocalLogon = true;
1446	if ( a->argc == 7
1447	\|\| ( a->argc == 8
1448	&& ( !strcmp(a->argv[3], "-p")
1449	\|\| !strcmp(a->argv[3], "--passwordfile"))))
1450	{
1451	if ( strcmp(a->argv[5 + (a->argc - 7)], "--allowlocallogon")
1452	&& strcmp(a->argv[5 + (a->argc - 7)], "-allowlocallogon"))
1453	{
1454	errorArgument("Invalid parameter '%s'", a->argv[5]);
1455	rc = E_FAIL;
1456	break;
1457	}
1458	if (!strcmp(a->argv[6 + (a->argc - 7)], "no"))
1459	fAllowLocalLogon = false;
1460	}
1461	else if ( a->argc != 5
1462	&& ( a->argc != 6
1463	\|\| ( strcmp(a->argv[3], "-p")
1464	&& strcmp(a->argv[3], "--passwordfile"))))
1465	{
1466	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1467	rc = E_FAIL;
1468	break;
1469	}
1470	Utf8Str passwd, domain;
1471	if (a->argc == 5 \|\| a->argc == 7)
1472	{
1473	passwd = a->argv[3];
1474	domain = a->argv[4];
1475	}
1476	else
1477	{
1478	RTEXITCODE rcExit = readPasswordFile(a->argv[4], &passwd);
1479	if (rcExit != RTEXITCODE_SUCCESS)
1480	{
1481	rc = E_FAIL;
1482	break;
1483	}
1484	domain = a->argv[5];
1485	}
1486
1487	ComPtr<IGuest> pGuest;
1488	CHECK_ERROR_BREAK(console, COMGETTER(Guest)(pGuest.asOutParam()));
1489	if (!pGuest)
1490	{
1491	RTMsgError("Guest not running");
1492	rc = E_FAIL;
1493	break;
1494	}
1495	CHECK_ERROR_BREAK(pGuest, SetCredentials(Bstr(a->argv[2]).raw(),
1496	Bstr(passwd).raw(),
1497	Bstr(domain).raw(),
1498	fAllowLocalLogon));
1499	}
1500	#if 0 /** @todo review & remove */
1501	else if (!strcmp(a->argv[1], "dvdattach"))
1502	{
1503	Bstr uuid;
1504	if (a->argc != 3)
1505	{
1506	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1507	rc = E_FAIL;
1508	break;
1509	}
1510
1511	ComPtr<IMedium> dvdMedium;
1512
1513	/* unmount? */
1514	if (!strcmp(a->argv[2], "none"))
1515	{
1516	/* nothing to do, NULL object will cause unmount */
1517	}
1518	/* host drive? */
1519	else if (!strncmp(a->argv[2], "host:", 5))
1520	{
1521	ComPtr<IHost> host;
1522	CHECK_ERROR(a->virtualBox, COMGETTER(Host)(host.asOutParam()));
1523
1524	rc = host->FindHostDVDDrive(Bstr(a->argv[2] + 5), dvdMedium.asOutParam());
1525	if (!dvdMedium)
1526	{
1527	errorArgument("Invalid host DVD drive name \"%s\"",
1528	a->argv[2] + 5);
1529	rc = E_FAIL;
1530	break;
1531	}
1532	}
1533	else
1534	{
1535	/* first assume it's a UUID */
1536	uuid = a->argv[2];
1537	rc = a->virtualBox->GetDVDImage(uuid, dvdMedium.asOutParam());
1538	if (FAILED(rc) \|\| !dvdMedium)
1539	{
1540	/* must be a filename, check if it's in the collection */
1541	rc = a->virtualBox->FindDVDImage(Bstr(a->argv[2]), dvdMedium.asOutParam());
1542	/* not registered, do that on the fly */
1543	if (!dvdMedium)
1544	{
1545	Bstr emptyUUID;
1546	CHECK_ERROR(a->virtualBox, OpenDVDImage(Bstr(a->argv[2]), emptyUUID, dvdMedium.asOutParam()));
1547	}
1548	}
1549	if (!dvdMedium)
1550	{
1551	rc = E_FAIL;
1552	break;
1553	}
1554	}
1555
1556	/** @todo generalize this, allow arbitrary number of DVD drives
1557	* and as a consequence multiple attachments and different
1558	* storage controllers. */
1559	if (dvdMedium)
1560	dvdMedium->COMGETTER(Id)(uuid.asOutParam());
1561	else
1562	uuid = Guid().toString();
1563	CHECK_ERROR(machine, MountMedium(Bstr("IDE Controller"), 1, 0, uuid, FALSE /* aForce */));
1564	}
1565	else if (!strcmp(a->argv[1], "floppyattach"))
1566	{
1567	Bstr uuid;
1568	if (a->argc != 3)
1569	{
1570	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1571	rc = E_FAIL;
1572	break;
1573	}
1574
1575	ComPtr<IMedium> floppyMedium;
1576
1577	/* unmount? */
1578	if (!strcmp(a->argv[2], "none"))
1579	{
1580	/* nothing to do, NULL object will cause unmount */
1581	}
1582	/* host drive? */
1583	else if (!strncmp(a->argv[2], "host:", 5))
1584	{
1585	ComPtr<IHost> host;
1586	CHECK_ERROR(a->virtualBox, COMGETTER(Host)(host.asOutParam()));
1587	host->FindHostFloppyDrive(Bstr(a->argv[2] + 5), floppyMedium.asOutParam());
1588	if (!floppyMedium)
1589	{
1590	errorArgument("Invalid host floppy drive name \"%s\"",
1591	a->argv[2] + 5);
1592	rc = E_FAIL;
1593	break;
1594	}
1595	}
1596	else
1597	{
1598	/* first assume it's a UUID */
1599	uuid = a->argv[2];
1600	rc = a->virtualBox->GetFloppyImage(uuid, floppyMedium.asOutParam());
1601	if (FAILED(rc) \|\| !floppyMedium)
1602	{
1603	/* must be a filename, check if it's in the collection */
1604	rc = a->virtualBox->FindFloppyImage(Bstr(a->argv[2]), floppyMedium.asOutParam());
1605	/* not registered, do that on the fly */
1606	if (!floppyMedium)
1607	{
1608	Bstr emptyUUID;
1609	CHECK_ERROR(a->virtualBox, OpenFloppyImage(Bstr(a->argv[2]), emptyUUID, floppyMedium.asOutParam()));
1610	}
1611	}
1612	if (!floppyMedium)
1613	{
1614	rc = E_FAIL;
1615	break;
1616	}
1617	}
1618	floppyMedium->COMGETTER(Id)(uuid.asOutParam());
1619	CHECK_ERROR(machine, MountMedium(Bstr("Floppy Controller"), 0, 0, uuid, FALSE /* aForce */));
1620	}
1621	#endif /* obsolete dvdattach/floppyattach */
1622	else if (!strcmp(a->argv[1], "guestmemoryballoon"))
1623	{
1624	if (a->argc != 3)
1625	{
1626	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1627	rc = E_FAIL;
1628	break;
1629	}
1630	uint32_t uVal;
1631	int vrc;
1632	vrc = RTStrToUInt32Ex(a->argv[2], NULL, 0, &uVal);
1633	if (vrc != VINF_SUCCESS)
1634	{
1635	errorArgument("Error parsing guest memory balloon size '%s'", a->argv[2]);
1636	rc = E_FAIL;
1637	break;
1638	}
1639	/* guest is running; update IGuest */
1640	ComPtr<IGuest> pGuest;
1641	rc = console->COMGETTER(Guest)(pGuest.asOutParam());
1642	if (SUCCEEDED(rc))
1643	{
1644	if (!pGuest)
1645	{
1646	RTMsgError("Guest not running");
1647	rc = E_FAIL;
1648	break;
1649	}
1650	CHECK_ERROR(pGuest, COMSETTER(MemoryBalloonSize)(uVal));
1651	}
1652	}
1653	else if (!strcmp(a->argv[1], "teleport"))
1654	{
1655	Bstr bstrHostname;
1656	uint32_t uMaxDowntime = 250 /ms/;
1657	uint32_t uPort = UINT32_MAX;
1658	uint32_t cMsTimeout = 0;
1659	Utf8Str strPassword;
1660	static const RTGETOPTDEF s_aTeleportOptions[] =
1661	{
1662	{ "--host", 'h', RTGETOPT_REQ_STRING }, /** @todo RTGETOPT_FLAG_MANDATORY */
1663	{ "--hostname", 'h', RTGETOPT_REQ_STRING }, /** @todo remove this */
1664	{ "--maxdowntime", 'd', RTGETOPT_REQ_UINT32 },
1665	{ "--port", 'P', RTGETOPT_REQ_UINT32 }, /** @todo RTGETOPT_FLAG_MANDATORY */
1666	{ "--passwordfile", 'p', RTGETOPT_REQ_STRING },
1667	{ "--password", 'W', RTGETOPT_REQ_STRING },
1668	{ "--timeout", 't', RTGETOPT_REQ_UINT32 },
1669	{ "--detailed-progress", 'D', RTGETOPT_REQ_NOTHING }
1670	};
1671	RTGETOPTSTATE GetOptState;
1672	RTGetOptInit(&GetOptState, a->argc, a->argv, s_aTeleportOptions, RT_ELEMENTS(s_aTeleportOptions), 2, RTGETOPTINIT_FLAGS_NO_STD_OPTS);
1673	int ch;
1674	RTGETOPTUNION Value;
1675	while ( SUCCEEDED(rc)
1676	&& (ch = RTGetOpt(&GetOptState, &Value)))
1677	{
1678	switch (ch)
1679	{
1680	case 'h': bstrHostname = Value.psz; break;
1681	case 'd': uMaxDowntime = Value.u32; break;
1682	case 'D': g_fDetailedProgress = true; break;
1683	case 'P': uPort = Value.u32; break;
1684	case 'p':
1685	{
1686	RTEXITCODE rcExit = readPasswordFile(Value.psz, &strPassword);
1687	if (rcExit != RTEXITCODE_SUCCESS)
1688	rc = E_FAIL;
1689	break;
1690	}
1691	case 'W': strPassword = Value.psz; break;
1692	case 't': cMsTimeout = Value.u32; break;
1693	default:
1694	errorGetOpt(USAGE_CONTROLVM, ch, &Value);
1695	rc = E_FAIL;
1696	break;
1697	}
1698	}
1699	if (FAILED(rc))
1700	break;
1701
1702	ComPtr<IProgress> progress;
1703	CHECK_ERROR_BREAK(console, Teleport(bstrHostname.raw(), uPort,
1704	Bstr(strPassword).raw(),
1705	uMaxDowntime,
1706	progress.asOutParam()));
1707
1708	if (cMsTimeout)
1709	{
1710	rc = progress->COMSETTER(Timeout)(cMsTimeout);
1711	if (FAILED(rc) && rc != VBOX_E_INVALID_OBJECT_STATE)
1712	CHECK_ERROR_BREAK(progress, COMSETTER(Timeout)(cMsTimeout)); /* lazyness */
1713	}
1714
1715	rc = showProgress(progress);
1716	CHECK_PROGRESS_ERROR(progress, ("Teleportation failed"));
1717	}
1718	else if (!strcmp(a->argv[1], "screenshotpng"))
1719	{
1720	if (a->argc <= 2 \|\| a->argc > 4)
1721	{
1722	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1723	rc = E_FAIL;
1724	break;
1725	}
1726	int vrc;
1727	uint32_t iScreen = 0;
1728	if (a->argc == 4)
1729	{
1730	vrc = RTStrToUInt32Ex(a->argv[3], NULL, 0, &iScreen);
1731	if (vrc != VINF_SUCCESS)
1732	{
1733	errorArgument("Error parsing display number '%s'", a->argv[3]);
1734	rc = E_FAIL;
1735	break;
1736	}
1737	}
1738	ComPtr<IDisplay> pDisplay;
1739	CHECK_ERROR_BREAK(console, COMGETTER(Display)(pDisplay.asOutParam()));
1740	if (!pDisplay)
1741	{
1742	RTMsgError("Guest not running");
1743	rc = E_FAIL;
1744	break;
1745	}
1746	ULONG width, height, bpp;
1747	LONG xOrigin, yOrigin;
1748	GuestMonitorStatus_T monitorStatus;
1749	CHECK_ERROR_BREAK(pDisplay, GetScreenResolution(iScreen, &width, &height, &bpp, &xOrigin, &yOrigin, &monitorStatus));
1750	com::SafeArray<BYTE> saScreenshot;
1751	CHECK_ERROR_BREAK(pDisplay, TakeScreenShotToArray(iScreen, width, height, BitmapFormat_PNG, ComSafeArrayAsOutParam(saScreenshot)));
1752	RTFILE pngFile = NIL_RTFILE;
1753	vrc = RTFileOpen(&pngFile, a->argv[2], RTFILE_O_OPEN_CREATE \| RTFILE_O_WRITE \| RTFILE_O_TRUNCATE \| RTFILE_O_DENY_ALL);
1754	if (RT_FAILURE(vrc))
1755	{
1756	RTMsgError("Failed to create file '%s' (%Rrc)", a->argv[2], vrc);
1757	rc = E_FAIL;
1758	break;
1759	}
1760	vrc = RTFileWrite(pngFile, saScreenshot.raw(), saScreenshot.size(), NULL);
1761	if (RT_FAILURE(vrc))
1762	{
1763	RTMsgError("Failed to write screenshot to file '%s' (%Rrc)", a->argv[2], vrc);
1764	rc = E_FAIL;
1765	}
1766	RTFileClose(pngFile);
1767	}
1768	#ifdef VBOX_WITH_VIDEOREC
1769	/*
1770	* Note: Commands starting with "vcp" are the deprecated versions and are
1771	* kept to ensure backwards compatibility.
1772	*/
1773	else if ( !strcmp(a->argv[1], "videocap")
1774	\|\| !strcmp(a->argv[1], "vcpenabled"))
1775	{
1776	if (a->argc != 3)
1777	{
1778	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1779	rc = E_FAIL;
1780	break;
1781	}
1782	if (!strcmp(a->argv[2], "on"))
1783	{
1784	CHECK_ERROR_RET(sessionMachine, COMSETTER(VideoCaptureEnabled)(TRUE), RTEXITCODE_FAILURE);
1785	}
1786	else if (!strcmp(a->argv[2], "off"))
1787	{
1788	CHECK_ERROR_RET(sessionMachine, COMSETTER(VideoCaptureEnabled)(FALSE), RTEXITCODE_FAILURE);
1789	}
1790	else
1791	{
1792	errorArgument("Invalid state '%s'", Utf8Str(a->argv[2]).c_str());
1793	rc = E_FAIL;
1794	break;
1795	}
1796	}
1797	else if ( !strcmp(a->argv[1], "videocapscreens")
1798	\|\| !strcmp(a->argv[1], "vcpscreens"))
1799	{
1800	ULONG cMonitors = 64;
1801	CHECK_ERROR_BREAK(machine, COMGETTER(MonitorCount)(&cMonitors));
1802	com::SafeArray<BOOL> saScreens(cMonitors);
1803	if ( a->argc == 3
1804	&& !strcmp(a->argv[2], "all"))
1805	{
1806	/* enable all screens */
1807	for (unsigned i = 0; i < cMonitors; i++)
1808	saScreens[i] = true;
1809	}
1810	else if ( a->argc == 3
1811	&& !strcmp(a->argv[2], "none"))
1812	{
1813	/* disable all screens */
1814	for (unsigned i = 0; i < cMonitors; i++)
1815	saScreens[i] = false;
1816
1817	/** @todo r=andy What if this is specified? */
1818	}
1819	else
1820	{
1821	/* enable selected screens */
1822	for (unsigned i = 0; i < cMonitors; i++)
1823	saScreens[i] = false;
1824	for (int i = 2; SUCCEEDED(rc) && i < a->argc; i++)
1825	{
1826	uint32_t iScreen;
1827	int vrc = RTStrToUInt32Ex(a->argv[i], NULL, 0, &iScreen);
1828	if (vrc != VINF_SUCCESS)
1829	{
1830	errorArgument("Error parsing display number '%s'", a->argv[i]);
1831	rc = E_FAIL;
1832	break;
1833	}
1834	if (iScreen >= cMonitors)
1835	{
1836	errorArgument("Invalid screen ID specified '%u'", iScreen);
1837	rc = E_FAIL;
1838	break;
1839	}
1840	saScreens[iScreen] = true;
1841	}
1842	}
1843
1844	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureScreens)(ComSafeArrayAsInParam(saScreens)));
1845	}
1846	else if ( !strcmp(a->argv[1], "videocapfile")
1847	\|\| !strcmp(a->argv[1], "vcpfile"))
1848	{
1849	if (a->argc != 3)
1850	{
1851	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1852	rc = E_FAIL;
1853	break;
1854	}
1855
1856	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureFile)(Bstr(a->argv[2]).raw()));
1857	}
1858	else if (!strcmp(a->argv[1], "videocapres"))
1859	{
1860	if (a->argc != 4)
1861	{
1862	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1863	rc = E_FAIL;
1864	break;
1865	}
1866
1867	uint32_t uVal;
1868	int vrc = RTStrToUInt32Ex(a->argv[2], NULL, 0, &uVal);
1869	if (RT_FAILURE(vrc))
1870	{
1871	errorArgument("Error parsing width '%s'", a->argv[2]);
1872	rc = E_FAIL;
1873	break;
1874	}
1875	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureWidth)(uVal));
1876
1877	vrc = RTStrToUInt32Ex(a->argv[3], NULL, 0, &uVal);
1878	if (RT_FAILURE(vrc))
1879	{
1880	errorArgument("Error parsing height '%s'", a->argv[3]);
1881	rc = E_FAIL;
1882	break;
1883	}
1884	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureHeight)(uVal));
1885	}
1886	else if (!strcmp(a->argv[1], "vcpwidth")) /* Deprecated; keeping for compatibility. */
1887	{
1888	uint32_t uVal;
1889	int vrc = RTStrToUInt32Ex(a->argv[2], NULL, 0, &uVal);
1890	if (RT_FAILURE(vrc))
1891	{
1892	errorArgument("Error parsing width '%s'", a->argv[2]);
1893	rc = E_FAIL;
1894	break;
1895	}
1896	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureWidth)(uVal));
1897	}
1898	else if (!strcmp(a->argv[1], "vcpheight")) /* Deprecated; keeping for compatibility. */
1899	{
1900	uint32_t uVal;
1901	int vrc = RTStrToUInt32Ex(a->argv[2], NULL, 0, &uVal);
1902	if (RT_FAILURE(vrc))
1903	{
1904	errorArgument("Error parsing height '%s'", a->argv[2]);
1905	rc = E_FAIL;
1906	break;
1907	}
1908	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureHeight)(uVal));
1909	}
1910	else if ( !strcmp(a->argv[1], "videocaprate")
1911	\|\| !strcmp(a->argv[1], "vcprate"))
1912	{
1913	if (a->argc != 3)
1914	{
1915	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1916	rc = E_FAIL;
1917	break;
1918	}
1919
1920	uint32_t uVal;
1921	int vrc = RTStrToUInt32Ex(a->argv[2], NULL, 0, &uVal);
1922	if (RT_FAILURE(vrc))
1923	{
1924	errorArgument("Error parsing rate '%s'", a->argv[2]);
1925	rc = E_FAIL;
1926	break;
1927	}
1928	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureRate)(uVal));
1929	}
1930	else if ( !strcmp(a->argv[1], "videocapfps")
1931	\|\| !strcmp(a->argv[1], "vcpfps"))
1932	{
1933	if (a->argc != 3)
1934	{
1935	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1936	rc = E_FAIL;
1937	break;
1938	}
1939
1940	uint32_t uVal;
1941	int vrc = RTStrToUInt32Ex(a->argv[2], NULL, 0, &uVal);
1942	if (RT_FAILURE(vrc))
1943	{
1944	errorArgument("Error parsing FPS '%s'", a->argv[2]);
1945	rc = E_FAIL;
1946	break;
1947	}
1948	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureFPS)(uVal));
1949	}
1950	else if ( !strcmp(a->argv[1], "videocapmaxtime")
1951	\|\| !strcmp(a->argv[1], "vcpmaxtime"))
1952	{
1953	if (a->argc != 3)
1954	{
1955	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1956	rc = E_FAIL;
1957	break;
1958	}
1959
1960	uint32_t uVal;
1961	int vrc = RTStrToUInt32Ex(a->argv[2], NULL, 0, &uVal);
1962	if (RT_FAILURE(vrc))
1963	{
1964	errorArgument("Error parsing maximum time '%s'", a->argv[2]);
1965	rc = E_FAIL;
1966	break;
1967	}
1968	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureMaxTime)(uVal));
1969	}
1970	else if ( !strcmp(a->argv[1], "videocapmaxsize")
1971	\|\| !strcmp(a->argv[1], "vcpmaxsize"))
1972	{
1973	if (a->argc != 3)
1974	{
1975	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1976	rc = E_FAIL;
1977	break;
1978	}
1979
1980	uint32_t uVal;
1981	int vrc = RTStrToUInt32Ex(a->argv[2], NULL, 0, &uVal);
1982	if (RT_FAILURE(vrc))
1983	{
1984	errorArgument("Error parsing maximum file size '%s'", a->argv[2]);
1985	rc = E_FAIL;
1986	break;
1987	}
1988	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureMaxFileSize)(uVal));
1989	}
1990	else if ( !strcmp(a->argv[1], "videocapopts")
1991	\|\| !strcmp(a->argv[1], "vcpoptions"))
1992	{
1993	if (a->argc != 3)
1994	{
1995	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1996	rc = E_FAIL;
1997	break;
1998	}
1999
2000	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureOptions)(Bstr(a->argv[3]).raw()));
2001	}
2002	#endif /* VBOX_WITH_VIDEOREC */
2003	else if (!strcmp(a->argv[1], "webcam"))
2004	{
2005	if (a->argc < 3)
2006	{
2007	errorArgument("Missing argument to '%s'", a->argv[1]);
2008	rc = E_FAIL;
2009	break;
2010	}
2011
2012	ComPtr<IEmulatedUSB> pEmulatedUSB;
2013	CHECK_ERROR_BREAK(console, COMGETTER(EmulatedUSB)(pEmulatedUSB.asOutParam()));
2014	if (!pEmulatedUSB)
2015	{
2016	RTMsgError("Guest not running");
2017	rc = E_FAIL;
2018	break;
2019	}
2020
2021	if (!strcmp(a->argv[2], "attach"))
2022	{
2023	Bstr path("");
2024	if (a->argc >= 4)
2025	path = a->argv[3];
2026	Bstr settings("");
2027	if (a->argc >= 5)
2028	settings = a->argv[4];
2029	CHECK_ERROR_BREAK(pEmulatedUSB, WebcamAttach(path.raw(), settings.raw()));
2030	}
2031	else if (!strcmp(a->argv[2], "detach"))
2032	{
2033	Bstr path("");
2034	if (a->argc >= 4)
2035	path = a->argv[3];
2036	CHECK_ERROR_BREAK(pEmulatedUSB, WebcamDetach(path.raw()));
2037	}
2038	else if (!strcmp(a->argv[2], "list"))
2039	{
2040	com::SafeArray <BSTR> webcams;
2041	CHECK_ERROR_BREAK(pEmulatedUSB, COMGETTER(Webcams)(ComSafeArrayAsOutParam(webcams)));
2042	for (size_t i = 0; i < webcams.size(); ++i)
2043	{
2044	RTPrintf("%ls\n", webcams[i][0]? webcams[i]: Bstr("default").raw());
2045	}
2046	}
2047	else
2048	{
2049	errorArgument("Invalid argument to '%s'", a->argv[1]);
2050	rc = E_FAIL;
2051	break;
2052	}
2053	}
2054	else if (!strcmp(a->argv[1], "addencpassword"))
2055	{
2056	if ( a->argc != 4
2057	&& a->argc != 6)
2058	{
2059	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
2060	break;
2061	}
2062
2063	BOOL fRemoveOnSuspend = FALSE;
2064	if (a->argc == 6)
2065	{
2066	if ( strcmp(a->argv[4], "--removeonsuspend")
2067	\|\| ( strcmp(a->argv[5], "yes")
2068	&& strcmp(a->argv[5], "no")))
2069	{
2070	errorSyntax(USAGE_CONTROLVM, "Invalid parameters");
2071	break;
2072	}
2073	if (!strcmp(a->argv[5], "yes"))
2074	fRemoveOnSuspend = TRUE;
2075	}
2076
2077	Bstr bstrPwId(a->argv[2]);
2078	Utf8Str strPassword;
2079
2080	if (!RTStrCmp(a->argv[3], "-"))
2081	{
2082	/* Get password from console. */
2083	RTEXITCODE rcExit = readPasswordFromConsole(&strPassword, "Enter password:");
2084	if (rcExit == RTEXITCODE_FAILURE)
2085	break;
2086	}
2087	else
2088	{
2089	RTEXITCODE rcExit = readPasswordFile(a->argv[3], &strPassword);
2090	if (rcExit == RTEXITCODE_FAILURE)
2091	{
2092	RTMsgError("Failed to read new password from file");
2093	break;
2094	}
2095	}
2096
2097	CHECK_ERROR_BREAK(console, AddDiskEncryptionPassword(bstrPwId.raw(), Bstr(strPassword).raw(), fRemoveOnSuspend));
2098	}
2099	else if (!strcmp(a->argv[1], "removeencpassword"))
2100	{
2101	if (a->argc != 3)
2102	{
2103	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
2104	break;
2105	}
2106	Bstr bstrPwId(a->argv[2]);
2107	CHECK_ERROR_BREAK(console, RemoveDiskEncryptionPassword(bstrPwId.raw()));
2108	}
2109	else if (!strcmp(a->argv[1], "removeallencpasswords"))
2110	{
2111	CHECK_ERROR_BREAK(console, ClearAllDiskEncryptionPasswords());
2112	}
2113	else
2114	{
2115	errorSyntax(USAGE_CONTROLVM, "Invalid parameter '%s'", a->argv[1]);
2116	rc = E_FAIL;
2117	}
2118	} while (0);
2119
2120	/* The client has to trigger saving the state explicitely. */
2121	if (fNeedsSaving)
2122	CHECK_ERROR(sessionMachine, SaveSettings());
2123
2124	a->session->UnlockMachine();
2125
2126	return SUCCEEDED(rc) ? RTEXITCODE_SUCCESS : RTEXITCODE_FAILURE;
2127	}

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

source: vbox/trunk/src/VBox/Frontends/VBoxManage/VBoxManageControlVM.cpp@ 69198

Download in other formats: