VBoxManageControlVM.cpp@ 33296

Last change on this file since 33296 was 33294, checked in by vboxsync, 15 years ago
Main: API change, merge IVirtualBox::getMachine() with findMachine()
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 32.5 KB

Line
1	/* $Id: VBoxManageControlVM.cpp 33294 2010-10-21 10:45:26Z vboxsync $ */
2	/** @file
3	* VBoxManage - Implementation of controlvm command.
4	*/
5
6	/*
7	* Copyright (C) 2006-2010 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/EventQueue.h>
29
30	#include <VBox/com/VirtualBox.h>
31
32	#include <iprt/ctype.h>
33	#include <VBox/err.h>
34	#include <iprt/getopt.h>
35	#include <iprt/stream.h>
36	#include <iprt/string.h>
37	#include <iprt/uuid.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 necesary 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
67	int handleControlVM(HandlerArg *a)
68	{
69	using namespace com;
70	HRESULT rc;
71
72	if (a->argc < 2)
73	return errorSyntax(USAGE_CONTROLVM, "Not enough parameters");
74
75	/* try to find the given machine */
76	ComPtr <IMachine> machine;
77	CHECK_ERROR(a->virtualBox, FindMachine(Bstr(a->argv[0]).raw(),
78	machine.asOutParam()));
79	if (FAILED(rc))
80	return 1;
81
82	/* open a session for the VM */
83	CHECK_ERROR_RET(machine, LockMachine(a->session, LockType_Shared), 1);
84
85	do
86	{
87	/* get the associated console */
88	ComPtr<IConsole> console;
89	CHECK_ERROR_BREAK(a->session, COMGETTER(Console)(console.asOutParam()));
90	/* ... and session machine */
91	ComPtr<IMachine> sessionMachine;
92	CHECK_ERROR_BREAK(a->session, COMGETTER(Machine)(sessionMachine.asOutParam()));
93
94	/* which command? */
95	if (!strcmp(a->argv[1], "pause"))
96	{
97	CHECK_ERROR_BREAK(console, Pause());
98	}
99	else if (!strcmp(a->argv[1], "resume"))
100	{
101	CHECK_ERROR_BREAK(console, Resume());
102	}
103	else if (!strcmp(a->argv[1], "reset"))
104	{
105	CHECK_ERROR_BREAK(console, Reset());
106	}
107	else if (!strcmp(a->argv[1], "unplugcpu"))
108	{
109	if (a->argc <= 1 + 1)
110	{
111	errorArgument("Missing argument to '%s'. Expected CPU number.", a->argv[1]);
112	rc = E_FAIL;
113	break;
114	}
115
116	unsigned n = parseNum(a->argv[2], 32, "CPU");
117
118	CHECK_ERROR_BREAK(sessionMachine, HotUnplugCPU(n));
119	}
120	else if (!strcmp(a->argv[1], "plugcpu"))
121	{
122	if (a->argc <= 1 + 1)
123	{
124	errorArgument("Missing argument to '%s'. Expected CPU number.", a->argv[1]);
125	rc = E_FAIL;
126	break;
127	}
128
129	unsigned n = parseNum(a->argv[2], 32, "CPU");
130
131	CHECK_ERROR_BREAK(sessionMachine, HotPlugCPU(n));
132	}
133	else if (!strcmp(a->argv[1], "cpuexecutioncap"))
134	{
135	if (a->argc <= 1 + 1)
136	{
137	errorArgument("Missing argument to '%s'. Expected execution cap number.", a->argv[1]);
138	rc = E_FAIL;
139	break;
140	}
141
142	unsigned n = parseNum(a->argv[2], 100, "ExecutionCap");
143
144	CHECK_ERROR_BREAK(machine, COMSETTER(CPUExecutionCap)(n));
145	}
146	else if (!strcmp(a->argv[1], "poweroff"))
147	{
148	ComPtr<IProgress> progress;
149	CHECK_ERROR_BREAK(console, PowerDown(progress.asOutParam()));
150
151	rc = showProgress(progress);
152	if (FAILED(rc))
153	{
154	com::ProgressErrorInfo info(progress);
155	if (info.isBasicAvailable())
156	RTMsgError("Failed to power off machine. Error message: %lS", info.getText().raw());
157	else
158	RTMsgError("Failed to power off machine. No error message available!");
159	}
160	}
161	else if (!strcmp(a->argv[1], "savestate"))
162	{
163	/* first pause so we don't trigger a live save which needs more time/resources */
164	CHECK_ERROR_BREAK(console, Pause());
165
166	ComPtr<IProgress> progress;
167	CHECK_ERROR(console, SaveState(progress.asOutParam()));
168	if (FAILED(rc))
169	{
170	console->Resume();
171	break;
172	}
173
174	rc = showProgress(progress);
175	if (FAILED(rc))
176	{
177	com::ProgressErrorInfo info(progress);
178	if (info.isBasicAvailable())
179	RTMsgError("Failed to save machine state. Error message: %lS", info.getText().raw());
180	else
181	RTMsgError("Failed to save machine state. No error message available!");
182	console->Resume();
183	}
184	}
185	else if (!strcmp(a->argv[1], "acpipowerbutton"))
186	{
187	CHECK_ERROR_BREAK(console, PowerButton());
188	}
189	else if (!strcmp(a->argv[1], "acpisleepbutton"))
190	{
191	CHECK_ERROR_BREAK(console, SleepButton());
192	}
193	else if (!strcmp(a->argv[1], "injectnmi"))
194	{
195	/* get the machine debugger. */
196	ComPtr <IMachineDebugger> debugger;
197	CHECK_ERROR_BREAK(console, COMGETTER(Debugger)(debugger.asOutParam()));
198	CHECK_ERROR_BREAK(debugger, InjectNMI());
199	}
200	else if (!strcmp(a->argv[1], "keyboardputscancode"))
201	{
202	ComPtr<IKeyboard> keyboard;
203	CHECK_ERROR_BREAK(console, COMGETTER(Keyboard)(keyboard.asOutParam()));
204
205	if (a->argc <= 1 + 1)
206	{
207	errorArgument("Missing argument to '%s'. Expected IBM PC AT set 2 keyboard scancode(s) as hex byte(s).", a->argv[1]);
208	rc = E_FAIL;
209	break;
210	}
211
212	std::list<LONG> llScancodes;
213
214	/* Process the command line. */
215	int i;
216	for (i = 1 + 1; i < a->argc; i++)
217	{
218	if ( RT_C_IS_XDIGIT (a->argv[i][0])
219	&& RT_C_IS_XDIGIT (a->argv[i][1])
220	&& a->argv[i][2] == 0)
221	{
222	uint8_t u8Scancode;
223	int irc = RTStrToUInt8Ex(a->argv[i], NULL, 16, &u8Scancode);
224	if (RT_FAILURE (irc))
225	{
226	RTMsgError("Converting '%s' returned %Rrc!", a->argv[i], rc);
227	rc = E_FAIL;
228	break;
229	}
230
231	llScancodes.push_back(u8Scancode);
232	}
233	else
234	{
235	RTMsgError("Error: '%s' is not a hex byte!", a->argv[i]);
236	rc = E_FAIL;
237	break;
238	}
239	}
240
241	if (FAILED(rc))
242	break;
243
244	/* Send scancodes to the VM. */
245	com::SafeArray<LONG> saScancodes(llScancodes);
246	ULONG codesStored = 0;
247	CHECK_ERROR_BREAK(keyboard, PutScancodes(ComSafeArrayAsInParam(saScancodes),
248	&codesStored));
249	if (codesStored < saScancodes.size())
250	{
251	RTMsgError("Only %d scancodes were stored", codesStored);
252	rc = E_FAIL;
253	break;
254	}
255	}
256	else if (!strncmp(a->argv[1], "setlinkstate", 12))
257	{
258	/* Get the number of network adapters */
259	ULONG NetworkAdapterCount = 0;
260	ComPtr <ISystemProperties> info;
261	CHECK_ERROR_BREAK(a->virtualBox, COMGETTER(SystemProperties)(info.asOutParam()));
262	CHECK_ERROR_BREAK(info, COMGETTER(NetworkAdapterCount)(&NetworkAdapterCount));
263
264	unsigned n = parseNum(&a->argv[1][12], NetworkAdapterCount, "NIC");
265	if (!n)
266	{
267	rc = E_FAIL;
268	break;
269	}
270	if (a->argc <= 1 + 1)
271	{
272	errorArgument("Missing argument to '%s'", a->argv[1]);
273	rc = E_FAIL;
274	break;
275	}
276	/* get the corresponding network adapter */
277	ComPtr<INetworkAdapter> adapter;
278	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
279	if (adapter)
280	{
281	if (!strcmp(a->argv[2], "on"))
282	{
283	CHECK_ERROR_BREAK(adapter, COMSETTER(CableConnected)(TRUE));
284	}
285	else if (!strcmp(a->argv[2], "off"))
286	{
287	CHECK_ERROR_BREAK(adapter, COMSETTER(CableConnected)(FALSE));
288	}
289	else
290	{
291	errorArgument("Invalid link state '%s'", Utf8Str(a->argv[2]).c_str());
292	rc = E_FAIL;
293	break;
294	}
295	}
296	}
297	#ifdef VBOX_DYNAMIC_NET_ATTACH
298	/* here the order in which strncmp is called is important
299	* cause nictracefile can be very well compared with
300	* nictrace and nic and thus everything will always fail
301	* if the order is changed
302	*/
303	else if (!strncmp(a->argv[1], "nictracefile", 12))
304	{
305	/* Get the number of network adapters */
306	ULONG NetworkAdapterCount = 0;
307	ComPtr <ISystemProperties> info;
308	CHECK_ERROR_BREAK(a->virtualBox, COMGETTER(SystemProperties)(info.asOutParam()));
309	CHECK_ERROR_BREAK(info, COMGETTER(NetworkAdapterCount)(&NetworkAdapterCount));
310
311	unsigned n = parseNum(&a->argv[1][12], NetworkAdapterCount, "NIC");
312	if (!n)
313	{
314	rc = E_FAIL;
315	break;
316	}
317	if (a->argc <= 2)
318	{
319	errorArgument("Missing argument to '%s'", a->argv[1]);
320	rc = E_FAIL;
321	break;
322	}
323
324	/* get the corresponding network adapter */
325	ComPtr<INetworkAdapter> adapter;
326	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
327	if (adapter)
328	{
329	BOOL fEnabled;
330	adapter->COMGETTER(Enabled)(&fEnabled);
331	if (fEnabled)
332	{
333	if (a->argv[2])
334	{
335	CHECK_ERROR_RET(adapter, COMSETTER(TraceFile)(Bstr(a->argv[2]).raw()), 1);
336	}
337	else
338	{
339	errorArgument("Invalid filename or filename not specified for NIC %lu", n);
340	rc = E_FAIL;
341	break;
342	}
343	}
344	else
345	RTMsgError("The NIC %d is currently disabled and thus can't change its tracefile", n);
346	}
347	}
348	else if (!strncmp(a->argv[1], "nictrace", 8))
349	{
350	/* Get the number of network adapters */
351	ULONG NetworkAdapterCount = 0;
352	ComPtr <ISystemProperties> info;
353	CHECK_ERROR_BREAK(a->virtualBox, COMGETTER(SystemProperties)(info.asOutParam()));
354	CHECK_ERROR_BREAK(info, COMGETTER(NetworkAdapterCount)(&NetworkAdapterCount));
355
356	unsigned n = parseNum(&a->argv[1][8], NetworkAdapterCount, "NIC");
357	if (!n)
358	{
359	rc = E_FAIL;
360	break;
361	}
362	if (a->argc <= 2)
363	{
364	errorArgument("Missing argument to '%s'", a->argv[1]);
365	rc = E_FAIL;
366	break;
367	}
368
369	/* get the corresponding network adapter */
370	ComPtr<INetworkAdapter> adapter;
371	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
372	if (adapter)
373	{
374	BOOL fEnabled;
375	adapter->COMGETTER(Enabled)(&fEnabled);
376	if (fEnabled)
377	{
378	if (!strcmp(a->argv[2], "on"))
379	{
380	CHECK_ERROR_RET(adapter, COMSETTER(TraceEnabled)(TRUE), 1);
381	}
382	else if (!strcmp(a->argv[2], "off"))
383	{
384	CHECK_ERROR_RET(adapter, COMSETTER(TraceEnabled)(FALSE), 1);
385	}
386	else
387	{
388	errorArgument("Invalid nictrace%lu argument '%s'", n, Utf8Str(a->argv[2]).c_str());
389	rc = E_FAIL;
390	break;
391	}
392	}
393	else
394	RTMsgError("The NIC %d is currently disabled and thus can't change its trace flag", n);
395	}
396	}
397	else if (!strncmp(a->argv[1], "nic", 3))
398	{
399	/* Get the number of network adapters */
400	ULONG NetworkAdapterCount = 0;
401	ComPtr <ISystemProperties> info;
402	CHECK_ERROR_BREAK(a->virtualBox, COMGETTER(SystemProperties)(info.asOutParam()));
403	CHECK_ERROR_BREAK(info, COMGETTER(NetworkAdapterCount)(&NetworkAdapterCount));
404
405	unsigned n = parseNum(&a->argv[1][3], NetworkAdapterCount, "NIC");
406	if (!n)
407	{
408	rc = E_FAIL;
409	break;
410	}
411	if (a->argc <= 2)
412	{
413	errorArgument("Missing argument to '%s'", a->argv[1]);
414	rc = E_FAIL;
415	break;
416	}
417
418	/* get the corresponding network adapter */
419	ComPtr<INetworkAdapter> adapter;
420	CHECK_ERROR_BREAK(sessionMachine, GetNetworkAdapter(n - 1, adapter.asOutParam()));
421	if (adapter)
422	{
423	BOOL fEnabled;
424	adapter->COMGETTER(Enabled)(&fEnabled);
425	if (fEnabled)
426	{
427	if (!strcmp(a->argv[2], "null"))
428	{
429	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
430	CHECK_ERROR_RET(adapter, Detach(), 1);
431	}
432	else if (!strcmp(a->argv[2], "nat"))
433	{
434	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
435	if (a->argc == 4)
436	CHECK_ERROR_RET(adapter, COMSETTER(NATNetwork)(Bstr(a->argv[3]).raw()), 1);
437	CHECK_ERROR_RET(adapter, AttachToNAT(), 1);
438	}
439	else if ( !strcmp(a->argv[2], "bridged")
440	\|\| !strcmp(a->argv[2], "hostif")) /* backward compatibility */
441	{
442	if (a->argc <= 3)
443	{
444	errorArgument("Missing argument to '%s'", a->argv[2]);
445	rc = E_FAIL;
446	break;
447	}
448	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
449	CHECK_ERROR_RET(adapter, COMSETTER(HostInterface)(Bstr(a->argv[3]).raw()), 1);
450	CHECK_ERROR_RET(adapter, AttachToBridgedInterface(), 1);
451	}
452	else if (!strcmp(a->argv[2], "intnet"))
453	{
454	if (a->argc <= 3)
455	{
456	errorArgument("Missing argument to '%s'", a->argv[2]);
457	rc = E_FAIL;
458	break;
459	}
460	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
461	CHECK_ERROR_RET(adapter, COMSETTER(InternalNetwork)(Bstr(a->argv[3]).raw()), 1);
462	CHECK_ERROR_RET(adapter, AttachToInternalNetwork(), 1);
463	}
464	#if defined(VBOX_WITH_NETFLT)
465	else if (!strcmp(a->argv[2], "hostonly"))
466	{
467	if (a->argc <= 3)
468	{
469	errorArgument("Missing argument to '%s'", a->argv[2]);
470	rc = E_FAIL;
471	break;
472	}
473	CHECK_ERROR_RET(adapter, COMSETTER(Enabled)(TRUE), 1);
474	CHECK_ERROR_RET(adapter, COMSETTER(HostInterface)(Bstr(a->argv[3]).raw()), 1);
475	CHECK_ERROR_RET(adapter, AttachToHostOnlyInterface(), 1);
476	}
477	#endif
478	else
479	{
480	errorArgument("Invalid type '%s' specfied for NIC %lu", Utf8Str(a->argv[2]).c_str(), n);
481	rc = E_FAIL;
482	break;
483	}
484	}
485	else
486	RTMsgError("The NIC %d is currently disabled and thus can't change its attachment type", n);
487	}
488	}
489	#endif /* VBOX_DYNAMIC_NET_ATTACH */
490	#ifdef VBOX_WITH_VRDP
491	else if (!strcmp(a->argv[1], "vrdp"))
492	{
493	if (a->argc <= 1 + 1)
494	{
495	errorArgument("Missing argument to '%s'", a->argv[1]);
496	rc = E_FAIL;
497	break;
498	}
499	/* get the corresponding VRDP server */
500	ComPtr<IVRDPServer> vrdpServer;
501	sessionMachine->COMGETTER(VRDPServer)(vrdpServer.asOutParam());
502	ASSERT(vrdpServer);
503	if (vrdpServer)
504	{
505	if (!strcmp(a->argv[2], "on"))
506	{
507	CHECK_ERROR_BREAK(vrdpServer, COMSETTER(Enabled)(TRUE));
508	}
509	else if (!strcmp(a->argv[2], "off"))
510	{
511	CHECK_ERROR_BREAK(vrdpServer, COMSETTER(Enabled)(FALSE));
512	}
513	else
514	{
515	errorArgument("Invalid vrdp server state '%s'", Utf8Str(a->argv[2]).c_str());
516	rc = E_FAIL;
517	break;
518	}
519	}
520	}
521	else if (!strcmp(a->argv[1], "vrdpport"))
522	{
523	if (a->argc <= 1 + 1)
524	{
525	errorArgument("Missing argument to '%s'", a->argv[1]);
526	rc = E_FAIL;
527	break;
528	}
529	/* get the corresponding VRDP server */
530	ComPtr<IVRDPServer> vrdpServer;
531	sessionMachine->COMGETTER(VRDPServer)(vrdpServer.asOutParam());
532	ASSERT(vrdpServer);
533	if (vrdpServer)
534	{
535	Bstr vrdpports;
536
537	if (!strcmp(a->argv[2], "default"))
538	vrdpports = "0";
539	else
540	vrdpports = a->argv[2];
541
542	CHECK_ERROR_BREAK(vrdpServer, COMSETTER(Ports)(vrdpports.raw()));
543	}
544	}
545	else if (!strcmp(a->argv[1], "vrdpvideochannelquality"))
546	{
547	if (a->argc <= 1 + 1)
548	{
549	errorArgument("Missing argument to '%s'", a->argv[1]);
550	rc = E_FAIL;
551	break;
552	}
553	/* get the corresponding VRDP server */
554	ComPtr<IVRDPServer> vrdpServer;
555	sessionMachine->COMGETTER(VRDPServer)(vrdpServer.asOutParam());
556	ASSERT(vrdpServer);
557	if (vrdpServer)
558	{
559	unsigned n = parseNum(a->argv[2], 100, "VRDP video channel quality in percent");
560
561	CHECK_ERROR(vrdpServer, COMSETTER(VideoChannelQuality)(n));
562	}
563	}
564	#endif /* VBOX_WITH_VRDP */
565	else if ( !strcmp(a->argv[1], "usbattach")
566	\|\| !strcmp(a->argv[1], "usbdetach"))
567	{
568	if (a->argc < 3)
569	{
570	errorSyntax(USAGE_CONTROLVM, "Not enough parameters");
571	rc = E_FAIL;
572	break;
573	}
574
575	bool attach = !strcmp(a->argv[1], "usbattach");
576
577	Bstr usbId = a->argv[2];
578	if (Guid(usbId).isEmpty())
579	{
580	// assume address
581	if (attach)
582	{
583	ComPtr <IHost> host;
584	CHECK_ERROR_BREAK(a->virtualBox, COMGETTER(Host)(host.asOutParam()));
585	SafeIfaceArray <IHostUSBDevice> coll;
586	CHECK_ERROR_BREAK(host, COMGETTER(USBDevices)(ComSafeArrayAsOutParam(coll)));
587	ComPtr <IHostUSBDevice> dev;
588	CHECK_ERROR_BREAK(host, FindUSBDeviceByAddress(Bstr(a->argv[2]).raw(),
589	dev.asOutParam()));
590	CHECK_ERROR_BREAK(dev, COMGETTER(Id)(usbId.asOutParam()));
591	}
592	else
593	{
594	SafeIfaceArray <IUSBDevice> coll;
595	CHECK_ERROR_BREAK(console, COMGETTER(USBDevices)(ComSafeArrayAsOutParam(coll)));
596	ComPtr <IUSBDevice> dev;
597	CHECK_ERROR_BREAK(console, FindUSBDeviceByAddress(Bstr(a->argv[2]).raw(),
598	dev.asOutParam()));
599	CHECK_ERROR_BREAK(dev, COMGETTER(Id)(usbId.asOutParam()));
600	}
601	}
602
603	if (attach)
604	CHECK_ERROR_BREAK(console, AttachUSBDevice(usbId.raw()));
605	else
606	{
607	ComPtr <IUSBDevice> dev;
608	CHECK_ERROR_BREAK(console, DetachUSBDevice(usbId.raw(),
609	dev.asOutParam()));
610	}
611	}
612	else if (!strcmp(a->argv[1], "setvideomodehint"))
613	{
614	if (a->argc != 5 && a->argc != 6)
615	{
616	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
617	rc = E_FAIL;
618	break;
619	}
620	uint32_t xres = RTStrToUInt32(a->argv[2]);
621	uint32_t yres = RTStrToUInt32(a->argv[3]);
622	uint32_t bpp = RTStrToUInt32(a->argv[4]);
623	uint32_t displayIdx = 0;
624	if (a->argc == 6)
625	displayIdx = RTStrToUInt32(a->argv[5]);
626
627	ComPtr<IDisplay> display;
628	CHECK_ERROR_BREAK(console, COMGETTER(Display)(display.asOutParam()));
629	CHECK_ERROR_BREAK(display, SetVideoModeHint(xres, yres, bpp, displayIdx));
630	}
631	else if (!strcmp(a->argv[1], "setcredentials"))
632	{
633	bool fAllowLocalLogon = true;
634	if (a->argc == 7)
635	{
636	if ( strcmp(a->argv[5], "--allowlocallogon")
637	&& strcmp(a->argv[5], "-allowlocallogon"))
638	{
639	errorArgument("Invalid parameter '%s'", a->argv[5]);
640	rc = E_FAIL;
641	break;
642	}
643	if (!strcmp(a->argv[6], "no"))
644	fAllowLocalLogon = false;
645	}
646	else if (a->argc != 5)
647	{
648	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
649	rc = E_FAIL;
650	break;
651	}
652
653	ComPtr<IGuest> guest;
654	CHECK_ERROR_BREAK(console, COMGETTER(Guest)(guest.asOutParam()));
655	CHECK_ERROR_BREAK(guest, SetCredentials(Bstr(a->argv[2]).raw(),
656	Bstr(a->argv[3]).raw(),
657	Bstr(a->argv[4]).raw(),
658	fAllowLocalLogon));
659	}
660	#if 0 /* TODO: review & remove */
661	else if (!strcmp(a->argv[1], "dvdattach"))
662	{
663	Bstr uuid;
664	if (a->argc != 3)
665	{
666	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
667	rc = E_FAIL;
668	break;
669	}
670
671	ComPtr<IMedium> dvdMedium;
672
673	/* unmount? */
674	if (!strcmp(a->argv[2], "none"))
675	{
676	/* nothing to do, NULL object will cause unmount */
677	}
678	/* host drive? */
679	else if (!strncmp(a->argv[2], "host:", 5))
680	{
681	ComPtr<IHost> host;
682	CHECK_ERROR(a->virtualBox, COMGETTER(Host)(host.asOutParam()));
683
684	rc = host->FindHostDVDDrive(Bstr(a->argv[2] + 5), dvdMedium.asOutParam());
685	if (!dvdMedium)
686	{
687	errorArgument("Invalid host DVD drive name \"%s\"",
688	a->argv[2] + 5);
689	rc = E_FAIL;
690	break;
691	}
692	}
693	else
694	{
695	/* first assume it's a UUID */
696	uuid = a->argv[2];
697	rc = a->virtualBox->GetDVDImage(uuid, dvdMedium.asOutParam());
698	if (FAILED(rc) \|\| !dvdMedium)
699	{
700	/* must be a filename, check if it's in the collection */
701	rc = a->virtualBox->FindDVDImage(Bstr(a->argv[2]), dvdMedium.asOutParam());
702	/* not registered, do that on the fly */
703	if (!dvdMedium)
704	{
705	Bstr emptyUUID;
706	CHECK_ERROR(a->virtualBox, OpenDVDImage(Bstr(a->argv[2]), emptyUUID, dvdMedium.asOutParam()));
707	}
708	}
709	if (!dvdMedium)
710	{
711	rc = E_FAIL;
712	break;
713	}
714	}
715
716	/** @todo generalize this, allow arbitrary number of DVD drives
717	* and as a consequence multiple attachments and different
718	* storage controllers. */
719	if (dvdMedium)
720	dvdMedium->COMGETTER(Id)(uuid.asOutParam());
721	else
722	uuid = Guid().toString();
723	CHECK_ERROR(machine, MountMedium(Bstr("IDE Controller"), 1, 0, uuid, FALSE /* aForce */));
724	}
725	else if (!strcmp(a->argv[1], "floppyattach"))
726	{
727	Bstr uuid;
728	if (a->argc != 3)
729	{
730	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
731	rc = E_FAIL;
732	break;
733	}
734
735	ComPtr<IMedium> floppyMedium;
736
737	/* unmount? */
738	if (!strcmp(a->argv[2], "none"))
739	{
740	/* nothing to do, NULL object will cause unmount */
741	}
742	/* host drive? */
743	else if (!strncmp(a->argv[2], "host:", 5))
744	{
745	ComPtr<IHost> host;
746	CHECK_ERROR(a->virtualBox, COMGETTER(Host)(host.asOutParam()));
747	host->FindHostFloppyDrive(Bstr(a->argv[2] + 5), floppyMedium.asOutParam());
748	if (!floppyMedium)
749	{
750	errorArgument("Invalid host floppy drive name \"%s\"",
751	a->argv[2] + 5);
752	rc = E_FAIL;
753	break;
754	}
755	}
756	else
757	{
758	/* first assume it's a UUID */
759	uuid = a->argv[2];
760	rc = a->virtualBox->GetFloppyImage(uuid, floppyMedium.asOutParam());
761	if (FAILED(rc) \|\| !floppyMedium)
762	{
763	/* must be a filename, check if it's in the collection */
764	rc = a->virtualBox->FindFloppyImage(Bstr(a->argv[2]), floppyMedium.asOutParam());
765	/* not registered, do that on the fly */
766	if (!floppyMedium)
767	{
768	Bstr emptyUUID;
769	CHECK_ERROR(a->virtualBox, OpenFloppyImage(Bstr(a->argv[2]), emptyUUID, floppyMedium.asOutParam()));
770	}
771	}
772	if (!floppyMedium)
773	{
774	rc = E_FAIL;
775	break;
776	}
777	}
778	floppyMedium->COMGETTER(Id)(uuid.asOutParam());
779	CHECK_ERROR(machine, MountMedium(Bstr("Floppy Controller"), 0, 0, uuid, FALSE /* aForce */));
780	}
781	#endif /* obsolete dvdattach/floppyattach */
782	else if (!strcmp(a->argv[1], "guestmemoryballoon"))
783	{
784	if (a->argc != 3)
785	{
786	errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
787	rc = E_FAIL;
788	break;
789	}
790	uint32_t uVal;
791	int vrc;
792	vrc = RTStrToUInt32Ex(a->argv[2], NULL, 0, &uVal);
793	if (vrc != VINF_SUCCESS)
794	{
795	errorArgument("Error parsing guest memory balloon size '%s'", a->argv[2]);
796	rc = E_FAIL;
797	break;
798	}
799	/* guest is running; update IGuest */
800	ComPtr <IGuest> guest;
801	rc = console->COMGETTER(Guest)(guest.asOutParam());
802	if (SUCCEEDED(rc))
803	CHECK_ERROR(guest, COMSETTER(MemoryBalloonSize)(uVal));
804	}
805	else if (!strcmp(a->argv[1], "teleport"))
806	{
807	Bstr bstrHostname;
808	uint32_t uMaxDowntime = 250 /ms/;
809	uint32_t uPort = UINT32_MAX;
810	uint32_t cMsTimeout = 0;
811	Bstr bstrPassword("");
812	static const RTGETOPTDEF s_aTeleportOptions[] =
813	{
814	{ "--host", 'h', RTGETOPT_REQ_STRING }, /** @todo RTGETOPT_FLAG_MANDATORY */
815	{ "--hostname", 'h', RTGETOPT_REQ_STRING }, /** @todo remove this */
816	{ "--maxdowntime", 'd', RTGETOPT_REQ_UINT32 },
817	{ "--port", 'p', RTGETOPT_REQ_UINT32 }, /** @todo RTGETOPT_FLAG_MANDATORY */
818	{ "--password", 'P', RTGETOPT_REQ_STRING },
819	{ "--timeout", 't', RTGETOPT_REQ_UINT32 },
820	{ "--detailed-progress", 'D', RTGETOPT_REQ_NOTHING }
821	};
822	RTGETOPTSTATE GetOptState;
823	RTGetOptInit(&GetOptState, a->argc, a->argv, s_aTeleportOptions, RT_ELEMENTS(s_aTeleportOptions), 2, RTGETOPTINIT_FLAGS_NO_STD_OPTS);
824	int ch;
825	RTGETOPTUNION Value;
826	while ( SUCCEEDED(rc)
827	&& (ch = RTGetOpt(&GetOptState, &Value)))
828	{
829	switch (ch)
830	{
831	case 'h': bstrHostname = Value.psz; break;
832	case 'd': uMaxDowntime = Value.u32; break;
833	case 'D': g_fDetailedProgress = true; break;
834	case 'p': uPort = Value.u32; break;
835	case 'P': bstrPassword = Value.psz; break;
836	case 't': cMsTimeout = Value.u32; break;
837	default:
838	errorGetOpt(USAGE_CONTROLVM, ch, &Value);
839	rc = E_FAIL;
840	break;
841	}
842	}
843	if (FAILED(rc))
844	break;
845
846	ComPtr<IProgress> progress;
847	CHECK_ERROR_BREAK(console, Teleport(bstrHostname.raw(), uPort,
848	bstrPassword.raw(),
849	uMaxDowntime,
850	progress.asOutParam()));
851
852	if (cMsTimeout)
853	{
854	rc = progress->COMSETTER(Timeout)(cMsTimeout);
855	if (FAILED(rc) && rc != VBOX_E_INVALID_OBJECT_STATE)
856	CHECK_ERROR_BREAK(progress, COMSETTER(Timeout)(cMsTimeout)); /* lazyness */
857	}
858
859	rc = showProgress(progress);
860	if (FAILED(rc))
861	{
862	com::ProgressErrorInfo info(progress);
863	if (info.isBasicAvailable())
864	RTMsgError("Teleportation failed. Error message: %lS", info.getText().raw());
865	else
866	RTMsgError("Teleportation failed. No error message available!");
867	}
868	}
869	else
870	{
871	errorSyntax(USAGE_CONTROLVM, "Invalid parameter '%s'", Utf8Str(a->argv[1]).c_str());
872	rc = E_FAIL;
873	}
874	} while (0);
875
876	a->session->UnlockMachine();
877
878	return SUCCEEDED(rc) ? 0 : 1;
879	}
880

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

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

Download in other formats: