VBoxManageControlVM.cpp@ 50996

Last change on this file since 50996 was 48538, checked in by vboxsync, 11 years ago
Main/NetworkAdapter+Machine+Appliance+SystemProperties+Medium+Console+Settings+IDL: make NAT networking a separate network attachment type which improves the user experience, store the necessary settings, plus changing the design of the methods which will move images and entire VMs, they lacked a progress object Frontends/VirtualBox: adapted fully, can configure NAT networks with proper drop down list support Frontends/VBoxManage: also supports NAT networks completely, and adds the long missing code to list intnets
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 55.7 KB

Line
1	/* $Id: VBoxManageControlVM.cpp 48538 2013-09-19 15:17:43Z vboxsync $ */
2	/** @file
3	* VBoxManage - Implementation of the controlvm command.
4	*/
5
6	/*
7	* Copyright (C) 2006-2013 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/uuid.h>
36	#include <iprt/file.h>
37	#include <VBox/log.h>
38
39	#include "VBoxManage.h"
40
41	#include <list>
42
43
44	/**
45	* Parses a number.
46	*
47	* @returns Valid number on success.
48	* @returns 0 if invalid number. All necessary bitching has been done.
49	* @param psz Pointer to the nic number.
50	*/
51	static unsigned parseNum(const char psz, unsigned cMaxNum, const char name)
52	{
53	uint32_t u32;
54	char *pszNext;
55	int rc = RTStrToUInt32Ex(psz, &pszNext, 10, &u32);
56	if ( RT_SUCCESS(rc)
57	&& *pszNext == '\0'
58	&& u32 >= 1
59	&& u32 <= cMaxNum)
60	return (unsigned)u32;
61	errorArgument("Invalid %s number '%s'", name, psz);
62	return 0;
63	}
64
65	unsigned int getMaxNics(IVirtualBox* vbox, IMachine* mach)
66	{
67	ComPtr<ISystemProperties> info;
68	ChipsetType_T aChipset;
69	ULONG NetworkAdapterCount = 0;
70	HRESULT rc;
71
72	do {
73	CHECK_ERROR_BREAK(vbox, COMGETTER(SystemProperties)(info.asOutParam()));
74	CHECK_ERROR_BREAK(mach, COMGETTER(ChipsetType)(&aChipset));
75	CHECK_ERROR_BREAK(info, GetMaxNetworkAdapters(aChipset, &NetworkAdapterCount));
76
77	return (unsigned int)NetworkAdapterCount;
78	} while (0);
79
80	return 0;
81	}
82
83
84	int handleControlVM(HandlerArg *a)
85	{
86	using namespace com;
87	HRESULT rc;
88
89	if (a->argc < 2)
90	return errorSyntax(USAGE_CONTROLVM, "Not enough parameters");
91
92	/* try to find the given machine */
93	ComPtr<IMachine> machine;
94	CHECK_ERROR(a->virtualBox, FindMachine(Bstr(a->argv[0]).raw(),
95	machine.asOutParam()));
96	if (FAILED(rc))
97	return 1;
98
99	/* open a session for the VM */
100	CHECK_ERROR_RET(machine, LockMachine(a->session, LockType_Shared), 1);
101
102	do
103	{
104	/* get the associated console */
105	ComPtr<IConsole> console;
106	CHECK_ERROR_BREAK(a->session, COMGETTER(Console)(console.asOutParam()));
107	/* ... and session machine */
108	ComPtr<IMachine> sessionMachine;
109	CHECK_ERROR_BREAK(a->session, COMGETTER(Machine)(sessionMachine.asOutParam()));
110
111	/* which command? */
112	if (!strcmp(a->argv[1], "pause"))
113	{
114	CHECK_ERROR_BREAK(console, Pause());
115	}
116	else if (!strcmp(a->argv[1], "resume"))
117	{
118	CHECK_ERROR_BREAK(console, Resume());
119	}
120	else if (!strcmp(a->argv[1], "reset"))
121	{
122	CHECK_ERROR_BREAK(console, Reset());
123	}
124	else if (!strcmp(a->argv[1], "unplugcpu"))
125	{
126	if (a->argc <= 1 + 1)
127	{
128	errorArgument("Missing argument to '%s'. Expected CPU number.", a->argv[1]);
129	rc = E_FAIL;
130	break;
131	}
132
133	unsigned n = parseNum(a->argv[2], 32, "CPU");
134
135	CHECK_ERROR_BREAK(sessionMachine, HotUnplugCPU(n));
136	}
137	else if (!strcmp(a->argv[1], "plugcpu"))
138	{
139	if (a->argc <= 1 + 1)
140	{
141	errorArgument("Missing argument to '%s'. Expected CPU number.", a->argv[1]);
142	rc = E_FAIL;
143	break;
144	}
145
146	unsigned n = parseNum(a->argv[2], 32, "CPU");
147
148	CHECK_ERROR_BREAK(sessionMachine, HotPlugCPU(n));
149	}
150	else if (!strcmp(a->argv[1], "cpuexecutioncap"))
151	{
152	if (a->argc <= 1 + 1)
153	{
154	errorArgument("Missing argument to '%s'. Expected execution cap number.", a->argv[1]);
155	rc = E_FAIL;
156	break;
157	}
158
159	unsigned n = parseNum(a->argv[2], 100, "ExecutionCap");
160
161	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(CPUExecutionCap)(n));
162	}
163	else if (!strcmp(a->argv[1], "clipboard"))
164	{
165	if (a->argc <= 1 + 1)
166	{
167	errorArgument("Missing argument to '%s'. Expected clipboard mode.", a->argv[1]);
168	rc = E_FAIL;
169	break;
170	}
171
172	ClipboardMode_T mode;
173	if (!strcmp(a->argv[2], "disabled"))
174	mode = ClipboardMode_Disabled;
175	else if (!strcmp(a->argv[2], "hosttoguest"))
176	mode = ClipboardMode_HostToGuest;
177	else if (!strcmp(a->argv[2], "guesttohost"))
178	mode = ClipboardMode_GuestToHost;
179	else if (!strcmp(a->argv[2], "bidirectional"))
180	mode = ClipboardMode_Bidirectional;
181	else
182	{
183	errorArgument("Invalid '%s' argument '%s'.", a->argv[1], a->argv[2]);
184	rc = E_FAIL;
185	}
186	if (SUCCEEDED(rc))
187	{
188	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(ClipboardMode)(mode));
189	}
190	}
191	else if (!strcmp(a->argv[1], "draganddrop"))
192	{
193	if (a->argc <= 1 + 1)
194	{
195	errorArgument("Missing argument to '%s'. Expected drag'n'drop mode.", a->argv[1]);
196	rc = E_FAIL;
197	break;
198	}
199
200	DragAndDropMode_T mode;
201	if (!strcmp(a->argv[2], "disabled"))
202	mode = DragAndDropMode_Disabled;
203	else if (!strcmp(a->argv[2], "hosttoguest"))
204	mode = DragAndDropMode_HostToGuest;
205	else if (!strcmp(a->argv[2], "guesttohost"))
206	mode = DragAndDropMode_GuestToHost;
207	else if (!strcmp(a->argv[2], "bidirectional"))
208	mode = DragAndDropMode_Bidirectional;
209	else
210	{
211	errorArgument("Invalid '%s' argument '%s'.", a->argv[1], a->argv[2]);
212	rc = E_FAIL;
213	}
214	if (SUCCEEDED(rc))
215	{
216	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(DragAndDropMode)(mode));
217	}
218	}
219	else if (!strcmp(a->argv[1], "poweroff"))
220	{
221	ComPtr<IProgress> progress;
222	CHECK_ERROR_BREAK(console, PowerDown(progress.asOutParam()));
223
224	rc = showProgress(progress);
225	CHECK_PROGRESS_ERROR(progress, ("Failed to power off machine"));
226	}
227	else if (!strcmp(a->argv[1], "savestate"))
228	{
229	/* first pause so we don't trigger a live save which needs more time/resources */
230	bool fPaused = false;
231	rc = console->Pause();
232	if (FAILED(rc))
233	{
234	bool fError = true;
235	if (rc == VBOX_E_INVALID_VM_STATE)
236	{
237	/* check if we are already paused */
238	MachineState_T machineState;
239	CHECK_ERROR_BREAK(console, COMGETTER(State)(&machineState));
240	/* the error code was lost by the previous instruction */
241	rc = VBOX_E_INVALID_VM_STATE;
242	if (machineState != MachineState_Paused)
243	{
244	RTMsgError("Machine in invalid state %d -- %s\n",
245	machineState, machineStateToName(machineState, false));
246	}
247	else
248	{
249	fError = false;
250	fPaused = true;
251	}
252	}
253	if (fError)
254	break;
255	}
256
257	ComPtr<IProgress> progress;
258	CHECK_ERROR(console, SaveState(progress.asOutParam()));
259	if (FAILED(rc))
260	{
261	if (!fPaused)
262	console->Resume();
263	break;
264	}
265
266	rc = showProgress(progress);
267	CHECK_PROGRESS_ERROR(progress, ("Failed to save machine state"));
268	if (FAILED(rc))
269	{
270	if (!fPaused)
271	console->Resume();
272	}
273	}
274	else if (!strcmp(a->argv[1], "acpipowerbutton"))
275	{
276	CHECK_ERROR_BREAK(console, PowerButton());
277	}
278	else if (!strcmp(a->argv[1], "acpisleepbutton"))
279	{
280	CHECK_ERROR_BREAK(console, SleepButton());
281	}
282	else if (!strcmp(a->argv[1], "keyboardputscancode"))
283	{
284	ComPtr<IKeyboard> pKeyboard;
285	CHECK_ERROR_BREAK(console, COMGETTER(Keyboard)(pKeyboard.asOutParam()));
286	if (!pKeyboard)
287	{
288	RTMsgError("Guest not running");
289	rc = E_FAIL;
290	break;
291	}
292
293	if (a->argc <= 1 + 1)
294	{
295	errorArgument("Missing argument to '%s'. Expected IBM PC AT set 2 keyboard scancode(s) as hex byte(s).", a->argv[1]);
296	rc = E_FAIL;
297	break;
298	}
299
300	std::list<LONG> llScancodes;
301
302	/* Process the command line. */
303	int i;
304	for (i = 1 + 1; i < a->argc; i++)
305	{
306	if ( RT_C_IS_XDIGIT (a->argv[i][0])
307	&& RT_C_IS_XDIGIT (a->argv[i][1])
308	&& a->argv[i][2] == 0)
309	{
310	uint8_t u8Scancode;
311	int irc = RTStrToUInt8Ex(a->argv[i], NULL, 16, &u8Scancode);
312	if (RT_FAILURE (irc))
313	{
314	RTMsgError("Converting '%s' returned %Rrc!", a->argv[i], rc);
315	rc = E_FAIL;
316	break;
317	}
318
319	llScancodes.push_back(u8Scancode);
320	}
321	else
322	{
323	RTMsgError("Error: '%s' is not a hex byte!", a->argv[i]);
324	rc = E_FAIL;
325	break;
326	}
327	}
328
329	if (FAILED(rc))
330	break;
331
332	/* Send scancodes to the VM. */
333	com::SafeArray<LONG> saScancodes(llScancodes);
334	ULONG codesStored = 0;
335	CHECK_ERROR_BREAK(pKeyboard, PutScancodes(ComSafeArrayAsInParam(saScancodes),
336	&codesStored));
337	if (codesStored < saScancodes.size())
338	{
339	RTMsgError("Only %d scancodes were stored", codesStored);
340	rc = E_FAIL;
341	break;
342	}
343	}
344	else if (!strncmp(a->argv[1], "setlinkstate", 12))
345	{
346	/* Get the number of network adapters */
347	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
348
349	unsigned n = parseNum(&a->argv[1][12], NetworkAdapterCount, "NIC");
350	if (!n)
351	{
352	rc = E_FAIL;
353	break;
354	}
355	if (a->argc <= 1 + 1)
356	{
357	errorArgument("Missing argument to '%s'", a->argv[1]);
358	rc = E_FAIL;
359	break;
360	}
361	/* get the corresponding network adapter */
362	ComPtr<INetworkAdapter> adapter;
363	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
364	if (adapter)
365	{
366	if (!strcmp(a->argv[2], "on"))
367	{
368	CHECK_ERROR_BREAK(adapter, COMSETTER(CableConnected)(TRUE));
369	}
370	else if (!strcmp(a->argv[2], "off"))
371	{
372	CHECK_ERROR_BREAK(adapter, COMSETTER(CableConnected)(FALSE));
373	}
374	else
375	{
376	errorArgument("Invalid link state '%s'", Utf8Str(a->argv[2]).c_str());
377	rc = E_FAIL;
378	break;
379	}
380	}
381	}
382	/* here the order in which strncmp is called is important
383	* cause nictracefile can be very well compared with
384	* nictrace and nic and thus everything will always fail
385	* if the order is changed
386	*/
387	else if (!strncmp(a->argv[1], "nictracefile", 12))
388	{
389	/* Get the number of network adapters */
390	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
391	unsigned n = parseNum(&a->argv[1][12], NetworkAdapterCount, "NIC");
392	if (!n)
393	{
394	rc = E_FAIL;
395	break;
396	}
397	if (a->argc <= 2)
398	{
399	errorArgument("Missing argument to '%s'", a->argv[1]);
400	rc = E_FAIL;
401	break;
402	}
403
404	/* get the corresponding network adapter */
405	ComPtr<INetworkAdapter> adapter;
406	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
407	if (adapter)
408	{
409	BOOL fEnabled;
410	adapter->COMGETTER(Enabled)(&fEnabled);
411	if (fEnabled)
412	{
413	if (a->argv[2])
414	{
415	CHECK_ERROR_RET(adapter, COMSETTER(TraceFile)(Bstr(a->argv[2]).raw()), 1);
416	}
417	else
418	{
419	errorArgument("Invalid filename or filename not specified for NIC %lu", n);
420	rc = E_FAIL;
421	break;
422	}
423	}
424	else
425	RTMsgError("The NIC %d is currently disabled and thus its tracefile can't be changed", n);
426	}
427	}
428	else if (!strncmp(a->argv[1], "nictrace", 8))
429	{
430	/* Get the number of network adapters */
431	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
432
433	unsigned n = parseNum(&a->argv[1][8], NetworkAdapterCount, "NIC");
434	if (!n)
435	{
436	rc = E_FAIL;
437	break;
438	}
439	if (a->argc <= 2)
440	{
441	errorArgument("Missing argument to '%s'", a->argv[1]);
442	rc = E_FAIL;
443	break;
444	}
445
446	/* get the corresponding network adapter */
447	ComPtr<INetworkAdapter> adapter;
448	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
449	if (adapter)
450	{
451	BOOL fEnabled;
452	adapter->COMGETTER(Enabled)(&fEnabled);
453	if (fEnabled)
454	{
455	if (!strcmp(a->argv[2], "on"))
456	{
457	CHECK_ERROR_RET(adapter, COMSETTER(TraceEnabled)(TRUE), 1);
458	}
459	else if (!strcmp(a->argv[2], "off"))
460	{
461	CHECK_ERROR_RET(adapter, COMSETTER(TraceEnabled)(FALSE), 1);
462	}
463	else
464	{
465	errorArgument("Invalid nictrace%lu argument '%s'", n, Utf8Str(a->argv[2]).c_str());
466	rc = E_FAIL;
467	break;
468	}
469	}
470	else
471	RTMsgError("The NIC %d is currently disabled and thus its trace flag can't be changed", n);
472	}
473	}
474	else if( a->argc > 2
475	&& !strncmp(a->argv[1], "natpf", 5))
476	{
477	/* Get the number of network adapters */
478	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox, sessionMachine);
479	ComPtr<INATEngine> engine;
480	unsigned n = parseNum(&a->argv[1][5], NetworkAdapterCount, "NIC");
481	if (!n)
482	{
483	rc = E_FAIL;
484	break;
485	}
486	if (a->argc <= 2)
487	{
488	errorArgument("Missing argument to '%s'", a->argv[1]);
489	rc = E_FAIL;
490	break;
491	}
492
493	/* get the corresponding network adapter */
494	ComPtr<INetworkAdapter> adapter;
495	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
496	if (!adapter)
497	{
498	rc = E_FAIL;
499	break;
500	}
501	CHECK_ERROR(adapter, COMGETTER(NATEngine)(engine.asOutParam()));
502	if (!engine)
503	{
504	rc = E_FAIL;
505	break;
506	}
507
508	if (!strcmp(a->argv[2], "delete"))
509	{
510	if (a->argc >= 3)
511	CHECK_ERROR(engine, RemoveRedirect(Bstr(a->argv[3]).raw()));
512	}
513	else
514	{
515	#define ITERATE_TO_NEXT_TERM(ch) \
516	do { \
517	while (*ch != ',') \
518	{ \
519	if (*ch == 0) \
520	{ \
521	return errorSyntax(USAGE_CONTROLVM, \
522	"Missing or invalid argument to '%s'", \
523	a->argv[1]); \
524	} \
525	ch++; \
526	} \
527	*ch = '\0'; \
528	ch++; \
529	} while(0)
530
531	char *strName;
532	char *strProto;
533	char *strHostIp;
534	char *strHostPort;
535	char *strGuestIp;
536	char *strGuestPort;
537	char *strRaw = RTStrDup(a->argv[2]);
538	char *ch = strRaw;
539	strName = RTStrStrip(ch);
540	ITERATE_TO_NEXT_TERM(ch);
541	strProto = RTStrStrip(ch);
542	ITERATE_TO_NEXT_TERM(ch);
543	strHostIp = RTStrStrip(ch);
544	ITERATE_TO_NEXT_TERM(ch);
545	strHostPort = RTStrStrip(ch);
546	ITERATE_TO_NEXT_TERM(ch);
547	strGuestIp = RTStrStrip(ch);
548	ITERATE_TO_NEXT_TERM(ch);
549	strGuestPort = RTStrStrip(ch);
550	NATProtocol_T proto;
551	if (RTStrICmp(strProto, "udp") == 0)
552	proto = NATProtocol_UDP;
553	else if (RTStrICmp(strProto, "tcp") == 0)
554	proto = NATProtocol_TCP;
555	else
556	{
557	return errorSyntax(USAGE_CONTROLVM,
558	"Wrong rule proto '%s' specified -- only 'udp' and 'tcp' are allowed.",
559	strProto);
560	}
561	CHECK_ERROR(engine, AddRedirect(Bstr(strName).raw(), proto, Bstr(strHostIp).raw(),
562	RTStrToUInt16(strHostPort), Bstr(strGuestIp).raw(), RTStrToUInt16(strGuestPort)));
563	#undef ITERATE_TO_NEXT_TERM
564	}
565	/* commit changes */
566	if (SUCCEEDED(rc))
567	CHECK_ERROR(sessionMachine, SaveSettings());
568	}
569	else if (!strncmp(a->argv[1], "nicproperty", 11))
570	{
571	/* Get the number of network adapters */
572	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox,sessionMachine) ;
573	unsigned n = parseNum(&a->argv[1][11], NetworkAdapterCount, "NIC");
574	if (!n)
575	{
576	rc = E_FAIL;
577	break;
578	}
579	if (a->argc <= 2)
580	{
581	errorArgument("Missing argument to '%s'", a->argv[1]);
582	rc = E_FAIL;
583	break;
584	}
585
586	/* get the corresponding network adapter */
587	ComPtr<INetworkAdapter> adapter;
588	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
589	if (adapter)
590	{
591	BOOL fEnabled;
592	adapter->COMGETTER(Enabled)(&fEnabled);
593	if (fEnabled)
594	{
595	/* Parse 'name=value' */
596	char *pszProperty = RTStrDup(a->argv[2]);
597	if (pszProperty)
598	{
599	char *pDelimiter = strchr(pszProperty, '=');
600	if (pDelimiter)
601	{
602	*pDelimiter = '\0';
603
604	Bstr bstrName = pszProperty;
605	Bstr bstrValue = &pDelimiter[1];
606	CHECK_ERROR(adapter, SetProperty(bstrName.raw(), bstrValue.raw()));
607	}
608	else
609	{
610	errorArgument("Invalid nicproperty%d argument '%s'", n, a->argv[2]);
611	rc = E_FAIL;
612	}
613	RTStrFree(pszProperty);
614	}
615	else
616	{
617	RTStrmPrintf(g_pStdErr, "Error: Failed to allocate memory for nicproperty%d '%s'\n", n, a->argv[2]);
618	rc = E_FAIL;
619	}
620	if (FAILED(rc))
621	break;
622	}
623	else
624	RTMsgError("The NIC %d is currently disabled and thus its properties can't be changed", n);
625	}
626	}
627	else if (!strncmp(a->argv[1], "nicpromisc", 10))
628	{
629	/* Get the number of network adapters */
630	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox,sessionMachine) ;
631	unsigned n = parseNum(&a->argv[1][10], NetworkAdapterCount, "NIC");
632	if (!n)
633	{
634	rc = E_FAIL;
635	break;
636	}
637	if (a->argc <= 2)
638	{
639	errorArgument("Missing argument to '%s'", a->argv[1]);
640	rc = E_FAIL;
641	break;
642	}
643
644	/* get the corresponding network adapter */
645	ComPtr<INetworkAdapter> adapter;
646	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
647	if (adapter)
648	{
649	BOOL fEnabled;
650	adapter->COMGETTER(Enabled)(&fEnabled);
651	if (fEnabled)
652	{
653	NetworkAdapterPromiscModePolicy_T enmPromiscModePolicy;
654	if (!strcmp(a->argv[2], "deny"))
655	enmPromiscModePolicy = NetworkAdapterPromiscModePolicy_Deny;
656	else if ( !strcmp(a->argv[2], "allow-vms")
657	\|\| !strcmp(a->argv[2], "allow-network"))
658	enmPromiscModePolicy = NetworkAdapterPromiscModePolicy_AllowNetwork;
659	else if (!strcmp(a->argv[2], "allow-all"))
660	enmPromiscModePolicy = NetworkAdapterPromiscModePolicy_AllowAll;
661	else
662	{
663	errorArgument("Unknown promiscuous mode policy '%s'", a->argv[2]);
664	rc = E_INVALIDARG;
665	break;
666	}
667
668	CHECK_ERROR(adapter, COMSETTER(PromiscModePolicy)(enmPromiscModePolicy));
669	}
670	else
671	RTMsgError("The NIC %d is currently disabled and thus its promiscuous mode can't be changed", n);
672	}
673	}
674	else if (!strncmp(a->argv[1], "nic", 3))
675	{
676	/* Get the number of network adapters */
677	ULONG NetworkAdapterCount = getMaxNics(a->virtualBox,sessionMachine) ;
678	unsigned n = parseNum(&a->argv[1][3], NetworkAdapterCount, "NIC");
679	if (!n)
680	{
681	rc = E_FAIL;
682	break;
683	}
684	if (a->argc <= 2)
685	{
686	errorArgument("Missing argument to '%s'", a->argv[1]);
687	rc = E_FAIL;
688	break;
689	}
690
691	/* get the corresponding network adapter */
692	ComPtr<INetworkAdapter> adapter;
693	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
694	if (adapter)
695	{
696	BOOL fEnabled;
697	adapter->COMGETTER(Enabled)(&fEnabled);
698	if (fEnabled)
699	{
700	if (!strcmp(a->argv[2], "null"))
701	{
702	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
703	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_Null), 1);
704	}
705	else if (!strcmp(a->argv[2], "nat"))
706	{
707	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
708	if (a->argc == 4)
709	CHECK_ERROR_RET(adapter, COMSETTER(NATNetwork)(Bstr(a->argv[3]).raw()), 1);
710	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_NAT), 1);
711	}
712	else if ( !strcmp(a->argv[2], "bridged")
713	\|\| !strcmp(a->argv[2], "hostif")) /* backward compatibility */
714	{
715	if (a->argc <= 3)
716	{
717	errorArgument("Missing argument to '%s'", a->argv[2]);
718	rc = E_FAIL;
719	break;
720	}
721	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
722	CHECK_ERROR_RET(adapter, COMSETTER(BridgedInterface)(Bstr(a->argv[3]).raw()), 1);
723	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_Bridged), 1);
724	}
725	else if (!strcmp(a->argv[2], "intnet"))
726	{
727	if (a->argc <= 3)
728	{
729	errorArgument("Missing argument to '%s'", a->argv[2]);
730	rc = E_FAIL;
731	break;
732	}
733	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
734	CHECK_ERROR_RET(adapter, COMSETTER(InternalNetwork)(Bstr(a->argv[3]).raw()), 1);
735	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_Internal), 1);
736	}
737	#if defined(VBOX_WITH_NETFLT)
738	else if (!strcmp(a->argv[2], "hostonly"))
739	{
740	if (a->argc <= 3)
741	{
742	errorArgument("Missing argument to '%s'", a->argv[2]);
743	rc = E_FAIL;
744	break;
745	}
746	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
747	CHECK_ERROR_RET(adapter, COMSETTER(HostOnlyInterface)(Bstr(a->argv[3]).raw()), 1);
748	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_HostOnly), 1);
749	}
750	#endif
751	else if (!strcmp(a->argv[2], "generic"))
752	{
753	if (a->argc <= 3)
754	{
755	errorArgument("Missing argument to '%s'", a->argv[2]);
756	rc = E_FAIL;
757	break;
758	}
759	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
760	CHECK_ERROR_RET(adapter, COMSETTER(GenericDriver)(Bstr(a->argv[3]).raw()), 1);
761	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_Generic), 1);
762	}
763	else if (!strcmp(a->argv[2], "natnetwork"))
764	{
765	if (a->argc <= 3)
766	{
767	errorArgument("Missing argument to '%s'", a->argv[2]);
768	rc = E_FAIL;
769	break;
770	}
771	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
772	CHECK_ERROR_RET(adapter, COMSETTER(NATNetwork)(Bstr(a->argv[3]).raw()), 1);
773	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_NATNetwork), 1);
774	}
775	/** @todo obsolete, remove eventually */
776	else if (!strcmp(a->argv[2], "vde"))
777	{
778	if (a->argc <= 3)
779	{
780	errorArgument("Missing argument to '%s'", a->argv[2]);
781	rc = E_FAIL;
782	break;
783	}
784	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
785	CHECK_ERROR_RET(adapter, COMSETTER(AttachmentType)(NetworkAttachmentType_Generic), 1);
786	CHECK_ERROR_RET(adapter, SetProperty(Bstr("name").raw(), Bstr(a->argv[3]).raw()), 1);
787	}
788	else
789	{
790	errorArgument("Invalid type '%s' specfied for NIC %lu", Utf8Str(a->argv[2]).c_str(), n);
791	rc = E_FAIL;
792	break;
793	}
794	}
795	else
796	RTMsgError("The NIC %d is currently disabled and thus its attachment type can't be changed", n);
797	}
798	}
799	else if ( !strcmp(a->argv[1], "vrde")
800	\|\| !strcmp(a->argv[1], "vrdp"))
801	{
802	if (!strcmp(a->argv[1], "vrdp"))
803	RTStrmPrintf(g_pStdErr, "Warning: 'vrdp' is deprecated. Use 'vrde'.\n");
804
805	if (a->argc <= 1 + 1)
806	{
807	errorArgument("Missing argument to '%s'", a->argv[1]);
808	rc = E_FAIL;
809	break;
810	}
811	ComPtr<IVRDEServer> vrdeServer;
812	sessionMachine->COMGETTER(VRDEServer)(vrdeServer.asOutParam());
813	ASSERT(vrdeServer);
814	if (vrdeServer)
815	{
816	if (!strcmp(a->argv[2], "on"))
817	{
818	CHECK_ERROR_BREAK(vrdeServer, COMSETTER(Enabled)(TRUE));
819	}
820	else if (!strcmp(a->argv[2], "off"))
821	{
822	CHECK_ERROR_BREAK(vrdeServer, COMSETTER(Enabled)(FALSE));
823	}
824	else
825	{
826	errorArgument("Invalid remote desktop server state '%s'", Utf8Str(a->argv[2]).c_str());
827	rc = E_FAIL;
828	break;
829	}
830	}
831	}
832	else if ( !strcmp(a->argv[1], "vrdeport")
833	\|\| !strcmp(a->argv[1], "vrdpport"))
834	{
835	if (!strcmp(a->argv[1], "vrdpport"))
836	RTStrmPrintf(g_pStdErr, "Warning: 'vrdpport' is deprecated. Use 'vrdeport'.\n");
837
838	if (a->argc <= 1 + 1)
839	{
840	errorArgument("Missing argument to '%s'", a->argv[1]);
841	rc = E_FAIL;
842	break;
843	}
844
845	ComPtr<IVRDEServer> vrdeServer;
846	sessionMachine->COMGETTER(VRDEServer)(vrdeServer.asOutParam());
847	ASSERT(vrdeServer);
848	if (vrdeServer)
849	{
850	Bstr ports;
851
852	if (!strcmp(a->argv[2], "default"))
853	ports = "0";
854	else
855	ports = a->argv[2];
856
857	CHECK_ERROR_BREAK(vrdeServer, SetVRDEProperty(Bstr("TCP/Ports").raw(), ports.raw()));
858	}
859	}
860	else if ( !strcmp(a->argv[1], "vrdevideochannelquality")
861	\|\| !strcmp(a->argv[1], "vrdpvideochannelquality"))
862	{
863	if (!strcmp(a->argv[1], "vrdpvideochannelquality"))
864	RTStrmPrintf(g_pStdErr, "Warning: 'vrdpvideochannelquality' is deprecated. Use 'vrdevideochannelquality'.\n");
865
866	if (a->argc <= 1 + 1)
867	{
868	errorArgument("Missing argument to '%s'", a->argv[1]);
869	rc = E_FAIL;
870	break;
871	}
872	ComPtr<IVRDEServer> vrdeServer;
873	sessionMachine->COMGETTER(VRDEServer)(vrdeServer.asOutParam());
874	ASSERT(vrdeServer);
875	if (vrdeServer)
876	{
877	Bstr value = a->argv[2];
878
879	CHECK_ERROR(vrdeServer, SetVRDEProperty(Bstr("VideoChannel/Quality").raw(), value.raw()));
880	}
881	}
882	else if (!strcmp(a->argv[1], "vrdeproperty"))
883	{
884	if (a->argc <= 1 + 1)
885	{
886	errorArgument("Missing argument to '%s'", a->argv[1]);
887	rc = E_FAIL;
888	break;
889	}
890	ComPtr<IVRDEServer> vrdeServer;
891	sessionMachine->COMGETTER(VRDEServer)(vrdeServer.asOutParam());
892	ASSERT(vrdeServer);
893	if (vrdeServer)
894	{
895	/* Parse 'name=value' */
896	char *pszProperty = RTStrDup(a->argv[2]);
897	if (pszProperty)
898	{
899	char *pDelimiter = strchr(pszProperty, '=');
900	if (pDelimiter)
901	{
902	*pDelimiter = '\0';
903
904	Bstr bstrName = pszProperty;
905	Bstr bstrValue = &pDelimiter[1];
906	CHECK_ERROR(vrdeServer, SetVRDEProperty(bstrName.raw(), bstrValue.raw()));
907	}
908	else
909	{
910	errorArgument("Invalid vrdeproperty argument '%s'", a->argv[2]);
911	rc = E_FAIL;
912	}
913	RTStrFree(pszProperty);
914	}
915	else
916	{
917	RTStrmPrintf(g_pStdErr, "Error: Failed to allocate memory for VRDE property '%s'\n", a->argv[2]);
918	rc = E_FAIL;
919	}
920	}
921	if (FAILED(rc))
922	{
923	break;
924	}
925	}
926	else if ( !strcmp(a->argv[1], "usbattach")
927	\|\| !strcmp(a->argv[1], "usbdetach"))
928	{
929	if (a->argc < 3)
930	{
931	errorSyntax(USAGE_CONTROLVM, "Not enough parameters");
932	rc = E_FAIL;
933	break;
934	}
935
936	bool attach = !strcmp(a->argv[1], "usbattach");
937
938	Bstr usbId = a->argv[2];
939
940	Guid guid(usbId);
941	if (!guid.isValid())
942	{
943	// assume address
944	if (attach)
945	{
946	ComPtr<IHost> host;
947	CHECK_ERROR_BREAK(a->virtualBox, COMGETTER(Host)(host.asOutParam()));
948	SafeIfaceArray <IHostUSBDevice> coll;
949	CHECK_ERROR_BREAK(host, COMGETTER(USBDevices)(ComSafeArrayAsOutParam(coll)));
950	ComPtr<IHostUSBDevice> dev;
951	CHECK_ERROR_BREAK(host, FindUSBDeviceByAddress(Bstr(a->argv[2]).raw(),
952	dev.asOutParam()));
953	CHECK_ERROR_BREAK(dev, COMGETTER(Id)(usbId.asOutParam()));
954	}
955	else
956	{
957	SafeIfaceArray <IUSBDevice> coll;
958	CHECK_ERROR_BREAK(console, COMGETTER(USBDevices)(ComSafeArrayAsOutParam(coll)));
959	ComPtr<IUSBDevice> dev;
960	CHECK_ERROR_BREAK(console, FindUSBDeviceByAddress(Bstr(a->argv[2]).raw(),
961	dev.asOutParam()));
962	CHECK_ERROR_BREAK(dev, COMGETTER(Id)(usbId.asOutParam()));
963	}
964	}
965	else if (guid.isZero())
966	{
967	errorArgument("Zero UUID argument '%s'", a->argv[2]);
968	rc = E_FAIL;
969	break;
970	}
971
972	if (attach)
973	CHECK_ERROR_BREAK(console, AttachUSBDevice(usbId.raw()));
974	else
975	{
976	ComPtr<IUSBDevice> dev;
977	CHECK_ERROR_BREAK(console, DetachUSBDevice(usbId.raw(),
978	dev.asOutParam()));
979	}
980	}
981	else if (!strcmp(a->argv[1], "setvideomodehint"))
982	{
983	if (a->argc != 5 && a->argc != 6 && a->argc != 7 && a->argc != 9)
984	{
985	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
986	rc = E_FAIL;
987	break;
988	}
989	bool fEnabled = true;
990	uint32_t uXRes = RTStrToUInt32(a->argv[2]);
991	uint32_t uYRes = RTStrToUInt32(a->argv[3]);
992	uint32_t uBpp = RTStrToUInt32(a->argv[4]);
993	uint32_t uDisplayIdx = 0;
994	bool fChangeOrigin = false;
995	int32_t iOriginX = 0;
996	int32_t iOriginY = 0;
997	if (a->argc >= 6)
998	uDisplayIdx = RTStrToUInt32(a->argv[5]);
999	if (a->argc >= 7)
1000	{
1001	int vrc = parseBool(a->argv[6], &fEnabled);
1002	if (RT_FAILURE(vrc))
1003	{
1004	errorSyntax(USAGE_CONTROLVM, "Either \"yes\" or \"no\" is expected");
1005	rc = E_FAIL;
1006	break;
1007	}
1008	fEnabled = !RTStrICmp(a->argv[6], "yes");
1009	}
1010	if (a->argc == 9)
1011	{
1012	iOriginX = RTStrToInt32(a->argv[7]);
1013	iOriginY = RTStrToInt32(a->argv[8]);
1014	fChangeOrigin = true;
1015	}
1016
1017	ComPtr<IDisplay> pDisplay;
1018	CHECK_ERROR_BREAK(console, COMGETTER(Display)(pDisplay.asOutParam()));
1019	if (!pDisplay)
1020	{
1021	RTMsgError("Guest not running");
1022	rc = E_FAIL;
1023	break;
1024	}
1025	CHECK_ERROR_BREAK(pDisplay, SetVideoModeHint(uDisplayIdx, fEnabled,
1026	fChangeOrigin, iOriginX, iOriginY,
1027	uXRes, uYRes, uBpp));
1028	}
1029	else if (!strcmp(a->argv[1], "setcredentials"))
1030	{
1031	bool fAllowLocalLogon = true;
1032	if ( a->argc == 7
1033	\|\| ( a->argc == 8
1034	&& ( !strcmp(a->argv[3], "-p")
1035	\|\| !strcmp(a->argv[3], "--passwordfile"))))
1036	{
1037	if ( strcmp(a->argv[5 + (a->argc - 7)], "--allowlocallogon")
1038	&& strcmp(a->argv[5 + (a->argc - 7)], "-allowlocallogon"))
1039	{
1040	errorArgument("Invalid parameter '%s'", a->argv[5]);
1041	rc = E_FAIL;
1042	break;
1043	}
1044	if (!strcmp(a->argv[6 + (a->argc - 7)], "no"))
1045	fAllowLocalLogon = false;
1046	}
1047	else if ( a->argc != 5
1048	&& ( a->argc != 6
1049	\|\| ( strcmp(a->argv[3], "-p")
1050	&& strcmp(a->argv[3], "--passwordfile"))))
1051	{
1052	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1053	rc = E_FAIL;
1054	break;
1055	}
1056	Utf8Str passwd, domain;
1057	if (a->argc == 5 \|\| a->argc == 7)
1058	{
1059	passwd = a->argv[3];
1060	domain = a->argv[4];
1061	}
1062	else
1063	{
1064	RTEXITCODE rcExit = readPasswordFile(a->argv[4], &passwd);
1065	if (rcExit != RTEXITCODE_SUCCESS)
1066	{
1067	rc = E_FAIL;
1068	break;
1069	}
1070	domain = a->argv[5];
1071	}
1072
1073	ComPtr<IGuest> pGuest;
1074	CHECK_ERROR_BREAK(console, COMGETTER(Guest)(pGuest.asOutParam()));
1075	if (!pGuest)
1076	{
1077	RTMsgError("Guest not running");
1078	rc = E_FAIL;
1079	break;
1080	}
1081	CHECK_ERROR_BREAK(pGuest, SetCredentials(Bstr(a->argv[2]).raw(),
1082	Bstr(passwd).raw(),
1083	Bstr(domain).raw(),
1084	fAllowLocalLogon));
1085	}
1086	#if 0 /* TODO: review & remove */
1087	else if (!strcmp(a->argv[1], "dvdattach"))
1088	{
1089	Bstr uuid;
1090	if (a->argc != 3)
1091	{
1092	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1093	rc = E_FAIL;
1094	break;
1095	}
1096
1097	ComPtr<IMedium> dvdMedium;
1098
1099	/* unmount? */
1100	if (!strcmp(a->argv[2], "none"))
1101	{
1102	/* nothing to do, NULL object will cause unmount */
1103	}
1104	/* host drive? */
1105	else if (!strncmp(a->argv[2], "host:", 5))
1106	{
1107	ComPtr<IHost> host;
1108	CHECK_ERROR(a->virtualBox, COMGETTER(Host)(host.asOutParam()));
1109
1110	rc = host->FindHostDVDDrive(Bstr(a->argv[2] + 5), dvdMedium.asOutParam());
1111	if (!dvdMedium)
1112	{
1113	errorArgument("Invalid host DVD drive name \"%s\"",
1114	a->argv[2] + 5);
1115	rc = E_FAIL;
1116	break;
1117	}
1118	}
1119	else
1120	{
1121	/* first assume it's a UUID */
1122	uuid = a->argv[2];
1123	rc = a->virtualBox->GetDVDImage(uuid, dvdMedium.asOutParam());
1124	if (FAILED(rc) \|\| !dvdMedium)
1125	{
1126	/* must be a filename, check if it's in the collection */
1127	rc = a->virtualBox->FindDVDImage(Bstr(a->argv[2]), dvdMedium.asOutParam());
1128	/* not registered, do that on the fly */
1129	if (!dvdMedium)
1130	{
1131	Bstr emptyUUID;
1132	CHECK_ERROR(a->virtualBox, OpenDVDImage(Bstr(a->argv[2]), emptyUUID, dvdMedium.asOutParam()));
1133	}
1134	}
1135	if (!dvdMedium)
1136	{
1137	rc = E_FAIL;
1138	break;
1139	}
1140	}
1141
1142	/** @todo generalize this, allow arbitrary number of DVD drives
1143	* and as a consequence multiple attachments and different
1144	* storage controllers. */
1145	if (dvdMedium)
1146	dvdMedium->COMGETTER(Id)(uuid.asOutParam());
1147	else
1148	uuid = Guid().toString();
1149	CHECK_ERROR(machine, MountMedium(Bstr("IDE Controller"), 1, 0, uuid, FALSE /* aForce */));
1150	}
1151	else if (!strcmp(a->argv[1], "floppyattach"))
1152	{
1153	Bstr uuid;
1154	if (a->argc != 3)
1155	{
1156	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1157	rc = E_FAIL;
1158	break;
1159	}
1160
1161	ComPtr<IMedium> floppyMedium;
1162
1163	/* unmount? */
1164	if (!strcmp(a->argv[2], "none"))
1165	{
1166	/* nothing to do, NULL object will cause unmount */
1167	}
1168	/* host drive? */
1169	else if (!strncmp(a->argv[2], "host:", 5))
1170	{
1171	ComPtr<IHost> host;
1172	CHECK_ERROR(a->virtualBox, COMGETTER(Host)(host.asOutParam()));
1173	host->FindHostFloppyDrive(Bstr(a->argv[2] + 5), floppyMedium.asOutParam());
1174	if (!floppyMedium)
1175	{
1176	errorArgument("Invalid host floppy drive name \"%s\"",
1177	a->argv[2] + 5);
1178	rc = E_FAIL;
1179	break;
1180	}
1181	}
1182	else
1183	{
1184	/* first assume it's a UUID */
1185	uuid = a->argv[2];
1186	rc = a->virtualBox->GetFloppyImage(uuid, floppyMedium.asOutParam());
1187	if (FAILED(rc) \|\| !floppyMedium)
1188	{
1189	/* must be a filename, check if it's in the collection */
1190	rc = a->virtualBox->FindFloppyImage(Bstr(a->argv[2]), floppyMedium.asOutParam());
1191	/* not registered, do that on the fly */
1192	if (!floppyMedium)
1193	{
1194	Bstr emptyUUID;
1195	CHECK_ERROR(a->virtualBox, OpenFloppyImage(Bstr(a->argv[2]), emptyUUID, floppyMedium.asOutParam()));
1196	}
1197	}
1198	if (!floppyMedium)
1199	{
1200	rc = E_FAIL;
1201	break;
1202	}
1203	}
1204	floppyMedium->COMGETTER(Id)(uuid.asOutParam());
1205	CHECK_ERROR(machine, MountMedium(Bstr("Floppy Controller"), 0, 0, uuid, FALSE /* aForce */));
1206	}
1207	#endif /* obsolete dvdattach/floppyattach */
1208	else if (!strcmp(a->argv[1], "guestmemoryballoon"))
1209	{
1210	if (a->argc != 3)
1211	{
1212	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1213	rc = E_FAIL;
1214	break;
1215	}
1216	uint32_t uVal;
1217	int vrc;
1218	vrc = RTStrToUInt32Ex(a->argv[2], NULL, 0, &uVal);
1219	if (vrc != VINF_SUCCESS)
1220	{
1221	errorArgument("Error parsing guest memory balloon size '%s'", a->argv[2]);
1222	rc = E_FAIL;
1223	break;
1224	}
1225	/* guest is running; update IGuest */
1226	ComPtr<IGuest> pGuest;
1227	rc = console->COMGETTER(Guest)(pGuest.asOutParam());
1228	if (SUCCEEDED(rc))
1229	{
1230	if (!pGuest)
1231	{
1232	RTMsgError("Guest not running");
1233	rc = E_FAIL;
1234	break;
1235	}
1236	CHECK_ERROR(pGuest, COMSETTER(MemoryBalloonSize)(uVal));
1237	}
1238	}
1239	else if (!strcmp(a->argv[1], "teleport"))
1240	{
1241	Bstr bstrHostname;
1242	uint32_t uMaxDowntime = 250 /ms/;
1243	uint32_t uPort = UINT32_MAX;
1244	uint32_t cMsTimeout = 0;
1245	Utf8Str strPassword;
1246	static const RTGETOPTDEF s_aTeleportOptions[] =
1247	{
1248	{ "--host", 'h', RTGETOPT_REQ_STRING }, /** @todo RTGETOPT_FLAG_MANDATORY */
1249	{ "--hostname", 'h', RTGETOPT_REQ_STRING }, /** @todo remove this */
1250	{ "--maxdowntime", 'd', RTGETOPT_REQ_UINT32 },
1251	{ "--port", 'P', RTGETOPT_REQ_UINT32 }, /** @todo RTGETOPT_FLAG_MANDATORY */
1252	{ "--passwordfile", 'p', RTGETOPT_REQ_STRING },
1253	{ "--password", 'W', RTGETOPT_REQ_STRING },
1254	{ "--timeout", 't', RTGETOPT_REQ_UINT32 },
1255	{ "--detailed-progress", 'D', RTGETOPT_REQ_NOTHING }
1256	};
1257	RTGETOPTSTATE GetOptState;
1258	RTGetOptInit(&GetOptState, a->argc, a->argv, s_aTeleportOptions, RT_ELEMENTS(s_aTeleportOptions), 2, RTGETOPTINIT_FLAGS_NO_STD_OPTS);
1259	int ch;
1260	RTGETOPTUNION Value;
1261	while ( SUCCEEDED(rc)
1262	&& (ch = RTGetOpt(&GetOptState, &Value)))
1263	{
1264	switch (ch)
1265	{
1266	case 'h': bstrHostname = Value.psz; break;
1267	case 'd': uMaxDowntime = Value.u32; break;
1268	case 'D': g_fDetailedProgress = true; break;
1269	case 'P': uPort = Value.u32; break;
1270	case 'p':
1271	{
1272	RTEXITCODE rcExit = readPasswordFile(Value.psz, &strPassword);
1273	if (rcExit != RTEXITCODE_SUCCESS)
1274	rc = E_FAIL;
1275	break;
1276	}
1277	case 'W': strPassword = Value.psz; break;
1278	case 't': cMsTimeout = Value.u32; break;
1279	default:
1280	errorGetOpt(USAGE_CONTROLVM, ch, &Value);
1281	rc = E_FAIL;
1282	break;
1283	}
1284	}
1285	if (FAILED(rc))
1286	break;
1287
1288	ComPtr<IProgress> progress;
1289	CHECK_ERROR_BREAK(console, Teleport(bstrHostname.raw(), uPort,
1290	Bstr(strPassword).raw(),
1291	uMaxDowntime,
1292	progress.asOutParam()));
1293
1294	if (cMsTimeout)
1295	{
1296	rc = progress->COMSETTER(Timeout)(cMsTimeout);
1297	if (FAILED(rc) && rc != VBOX_E_INVALID_OBJECT_STATE)
1298	CHECK_ERROR_BREAK(progress, COMSETTER(Timeout)(cMsTimeout)); /* lazyness */
1299	}
1300
1301	rc = showProgress(progress);
1302	CHECK_PROGRESS_ERROR(progress, ("Teleportation failed"));
1303	}
1304	else if (!strcmp(a->argv[1], "screenshotpng"))
1305	{
1306	if (a->argc <= 2 \|\| a->argc > 4)
1307	{
1308	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
1309	rc = E_FAIL;
1310	break;
1311	}
1312	int vrc;
1313	uint32_t iScreen = 0;
1314	if (a->argc == 4)
1315	{
1316	vrc = RTStrToUInt32Ex(a->argv[3], NULL, 0, &iScreen);
1317	if (vrc != VINF_SUCCESS)
1318	{
1319	errorArgument("Error parsing display number '%s'", a->argv[3]);
1320	rc = E_FAIL;
1321	break;
1322	}
1323	}
1324	ComPtr<IDisplay> pDisplay;
1325	CHECK_ERROR_BREAK(console, COMGETTER(Display)(pDisplay.asOutParam()));
1326	if (!pDisplay)
1327	{
1328	RTMsgError("Guest not running");
1329	rc = E_FAIL;
1330	break;
1331	}
1332	ULONG width, height, bpp;
1333	LONG xOrigin, yOrigin;
1334	CHECK_ERROR_BREAK(pDisplay, GetScreenResolution(iScreen, &width, &height, &bpp, &xOrigin, &yOrigin));
1335	com::SafeArray<BYTE> saScreenshot;
1336	CHECK_ERROR_BREAK(pDisplay, TakeScreenShotPNGToArray(iScreen, width, height, ComSafeArrayAsOutParam(saScreenshot)));
1337	RTFILE pngFile = NIL_RTFILE;
1338	vrc = RTFileOpen(&pngFile, a->argv[2], RTFILE_O_OPEN_CREATE \| RTFILE_O_WRITE \| RTFILE_O_TRUNCATE \| RTFILE_O_DENY_ALL);
1339	if (RT_FAILURE(vrc))
1340	{
1341	RTMsgError("Failed to create file '%s'. rc=%Rrc", a->argv[2], vrc);
1342	rc = E_FAIL;
1343	break;
1344	}
1345	vrc = RTFileWrite(pngFile, saScreenshot.raw(), saScreenshot.size(), NULL);
1346	if (RT_FAILURE(vrc))
1347	{
1348	RTMsgError("Failed to write screenshot to file '%s'. rc=%Rrc", a->argv[2], vrc);
1349	rc = E_FAIL;
1350	}
1351	RTFileClose(pngFile);
1352	}
1353	else if ( !strcmp(a->argv[1], "vcpenabled"))
1354	{
1355	if (a->argc != 3)
1356	{
1357	errorArgument("Missing argument to '%s'", a->argv[1]);
1358	rc = E_FAIL;
1359	break;
1360	}
1361	if (!strcmp(a->argv[2], "on"))
1362	{
1363	CHECK_ERROR_RET(sessionMachine, COMSETTER(VideoCaptureEnabled)(TRUE), 1);
1364	}
1365	else if (!strcmp(a->argv[2], "off"))
1366	{
1367	CHECK_ERROR_RET(sessionMachine, COMSETTER(VideoCaptureEnabled)(FALSE), 1);
1368	}
1369	else
1370	{
1371	errorArgument("Invalid state '%s'", Utf8Str(a->argv[2]).c_str());
1372	rc = E_FAIL;
1373	break;
1374	}
1375	}
1376	else if ( !strcmp(a->argv[1], "videocapturescreens"))
1377	{
1378	ULONG cMonitors = 64;
1379	CHECK_ERROR_BREAK(machine, COMGETTER(MonitorCount)(&cMonitors));
1380	com::SafeArray<BOOL> saScreens(cMonitors);
1381	if ( a->argc == 3
1382	&& !strcmp(a->argv[2], "all"))
1383	{
1384	/* enable all screens */
1385	for (unsigned i = 0; i < cMonitors; i++)
1386	saScreens[i] = true;
1387	}
1388	else if ( a->argc == 3
1389	&& !strcmp(a->argv[2], "none"))
1390	{
1391	/* disable all screens */
1392	for (unsigned i = 0; i < cMonitors; i++)
1393	saScreens[i] = false;
1394	}
1395	else
1396	{
1397	/* enable selected screens */
1398	for (unsigned i = 0; i < cMonitors; i++)
1399	saScreens[i] = false;
1400	for (int i = 2; SUCCEEDED(rc) && i < a->argc; i++)
1401	{
1402	uint32_t iScreen;
1403	int vrc = RTStrToUInt32Ex(a->argv[i], NULL, 0, &iScreen);
1404	if (vrc != VINF_SUCCESS)
1405	{
1406	errorArgument("Error parsing display number '%s'", a->argv[i]);
1407	rc = E_FAIL;
1408	break;
1409	}
1410	if (iScreen >= cMonitors)
1411	{
1412	errorArgument("Invalid screen ID specified '%u'", iScreen);
1413	rc = E_FAIL;
1414	break;
1415	}
1416	saScreens[iScreen] = true;
1417	}
1418	}
1419
1420	CHECK_ERROR_BREAK(sessionMachine, COMSETTER(VideoCaptureScreens)(ComSafeArrayAsInParam(saScreens)));
1421	}
1422	else if (!strcmp(a->argv[1], "webcam"))
1423	{
1424	if (a->argc < 3)
1425	{
1426	errorArgument("Missing argument to '%s'", a->argv[1]);
1427	rc = E_FAIL;
1428	break;
1429	}
1430
1431	ComPtr<IEmulatedUSB> pEmulatedUSB;
1432	CHECK_ERROR_BREAK(console, COMGETTER(EmulatedUSB)(pEmulatedUSB.asOutParam()));
1433	if (!pEmulatedUSB)
1434	{
1435	RTMsgError("Guest not running");
1436	rc = E_FAIL;
1437	break;
1438	}
1439
1440	if (!strcmp(a->argv[2], "attach"))
1441	{
1442	Bstr path("");
1443	if (a->argc >= 4)
1444	path = a->argv[3];
1445	Bstr settings("");
1446	if (a->argc >= 5)
1447	settings = a->argv[4];
1448	CHECK_ERROR_BREAK(pEmulatedUSB, WebcamAttach(path.raw(), settings.raw()));
1449	}
1450	else if (!strcmp(a->argv[2], "detach"))
1451	{
1452	Bstr path("");
1453	if (a->argc >= 4)
1454	path = a->argv[3];
1455	CHECK_ERROR_BREAK(pEmulatedUSB, WebcamDetach(path.raw()));
1456	}
1457	else if (!strcmp(a->argv[2], "list"))
1458	{
1459	com::SafeArray <BSTR> webcams;
1460	CHECK_ERROR_BREAK(pEmulatedUSB, COMGETTER(Webcams)(ComSafeArrayAsOutParam(webcams)));
1461	for (size_t i = 0; i < webcams.size(); ++i)
1462	{
1463	RTPrintf("%ls\n", webcams[i][0]? webcams[i]: Bstr("default").raw());
1464	}
1465	}
1466	else
1467	{
1468	errorArgument("Invalid argument to '%s'", a->argv[1]);
1469	rc = E_FAIL;
1470	break;
1471	}
1472
1473	}
1474	else
1475	{
1476	errorSyntax(USAGE_CONTROLVM, "Invalid parameter '%s'", a->argv[1]);
1477	rc = E_FAIL;
1478	}
1479	} while (0);
1480
1481	a->session->UnlockMachine();
1482
1483	return SUCCEEDED(rc) ? 0 : 1;
1484	}

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

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

Download in other formats: