1 | /** @file
|
---|
2 | *
|
---|
3 | * VBox frontends: VBoxHeadless frontend:
|
---|
4 | * Testcases
|
---|
5 | */
|
---|
6 |
|
---|
7 | /*
|
---|
8 | * Copyright (C) 2006-2024 Oracle and/or its affiliates.
|
---|
9 | *
|
---|
10 | * This file is part of VirtualBox base platform packages, as
|
---|
11 | * available from https://www.virtualbox.org.
|
---|
12 | *
|
---|
13 | * This program is free software; you can redistribute it and/or
|
---|
14 | * modify it under the terms of the GNU General Public License
|
---|
15 | * as published by the Free Software Foundation, in version 3 of the
|
---|
16 | * License.
|
---|
17 | *
|
---|
18 | * This program is distributed in the hope that it will be useful, but
|
---|
19 | * WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
20 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
---|
21 | * General Public License for more details.
|
---|
22 | *
|
---|
23 | * You should have received a copy of the GNU General Public License
|
---|
24 | * along with this program; if not, see <https://www.gnu.org/licenses>.
|
---|
25 | *
|
---|
26 | * SPDX-License-Identifier: GPL-3.0-only
|
---|
27 | */
|
---|
28 |
|
---|
29 | #include <VBox/com/array.h>
|
---|
30 | #include <VBox/com/com.h>
|
---|
31 | #include <VBox/com/string.h>
|
---|
32 | #include <VBox/com/Guid.h>
|
---|
33 | #include <VBox/com/ErrorInfo.h>
|
---|
34 | #include <VBox/com/errorprint.h>
|
---|
35 |
|
---|
36 | #include <VBox/com/VirtualBox.h>
|
---|
37 |
|
---|
38 | using namespace com;
|
---|
39 |
|
---|
40 | #include <VBox/log.h>
|
---|
41 | #include <iprt/initterm.h>
|
---|
42 | #include <iprt/stream.h>
|
---|
43 |
|
---|
44 |
|
---|
45 | ////////////////////////////////////////////////////////////////////////////////
|
---|
46 |
|
---|
47 | /**
|
---|
48 | * Entry point.
|
---|
49 | */
|
---|
50 | int main(int argc, char **argv)
|
---|
51 | {
|
---|
52 | // initialize VBox Runtime
|
---|
53 | RTR3InitExe(argc, &argv, 0);
|
---|
54 |
|
---|
55 | // the below cannot be Bstr because on Linux Bstr doesn't work
|
---|
56 | // until XPCOM (nsMemory) is initialized
|
---|
57 | const char *name = NULL;
|
---|
58 | const char *operation = NULL;
|
---|
59 |
|
---|
60 | // parse the command line
|
---|
61 | if (argc > 1)
|
---|
62 | name = argv [1];
|
---|
63 | if (argc > 2)
|
---|
64 | operation = argv [2];
|
---|
65 |
|
---|
66 | if (!name || !operation)
|
---|
67 | {
|
---|
68 | RTPrintf("\nUsage:\n\n"
|
---|
69 | "%s <machine_name> [on|off|pause|resume]\n\n",
|
---|
70 | argv [0]);
|
---|
71 | return 0;
|
---|
72 | }
|
---|
73 |
|
---|
74 | RTPrintf("\n");
|
---|
75 | RTPrintf("tstHeadless STARTED.\n");
|
---|
76 |
|
---|
77 | RTPrintf("VM name : {%s}\n"
|
---|
78 | "Operation : %s\n\n",
|
---|
79 | name, operation);
|
---|
80 |
|
---|
81 | HRESULT hrc;
|
---|
82 |
|
---|
83 | hrc = com::Initialize();
|
---|
84 | if (FAILED(hrc))
|
---|
85 | {
|
---|
86 | RTPrintf("ERROR: failed to initialize COM!\n");
|
---|
87 | return hrc;
|
---|
88 | }
|
---|
89 |
|
---|
90 | do
|
---|
91 | {
|
---|
92 | ComPtr <IVirtualBoxClient> virtualBoxClient;
|
---|
93 | ComPtr <IVirtualBox> virtualBox;
|
---|
94 | ComPtr <ISession> session;
|
---|
95 |
|
---|
96 | RTPrintf("Creating VirtualBox object...\n");
|
---|
97 | hrc = virtualBoxClient.createInprocObject(CLSID_VirtualBoxClient);
|
---|
98 | if (SUCCEEDED(hrc))
|
---|
99 | hrc = virtualBoxClient->COMGETTER(VirtualBox)(virtualBox.asOutParam());
|
---|
100 | if (FAILED(hrc))
|
---|
101 | RTPrintf("ERROR: failed to create the VirtualBox object!\n");
|
---|
102 | else
|
---|
103 | {
|
---|
104 | hrc = session.createInprocObject(CLSID_Session);
|
---|
105 | if (FAILED(hrc))
|
---|
106 | RTPrintf("ERROR: failed to create a session object!\n");
|
---|
107 | }
|
---|
108 |
|
---|
109 | if (FAILED(hrc))
|
---|
110 | {
|
---|
111 | com::ErrorInfo info;
|
---|
112 | if (!info.isFullAvailable() && !info.isBasicAvailable())
|
---|
113 | {
|
---|
114 | com::GluePrintRCMessage(hrc);
|
---|
115 | RTPrintf("Most likely, the VirtualBox COM server is not running or failed to start.\n");
|
---|
116 | }
|
---|
117 | else
|
---|
118 | com::GluePrintErrorInfo(info);
|
---|
119 | break;
|
---|
120 | }
|
---|
121 |
|
---|
122 | ComPtr <IMachine> m;
|
---|
123 |
|
---|
124 | // find ID by name
|
---|
125 | CHECK_ERROR_BREAK(virtualBox, FindMachine(Bstr(name).raw(),
|
---|
126 | m.asOutParam()));
|
---|
127 |
|
---|
128 | if (!strcmp(operation, "on"))
|
---|
129 | {
|
---|
130 | ComPtr <IProgress> progress;
|
---|
131 | RTPrintf("Opening a new (remote) session...\n");
|
---|
132 | CHECK_ERROR_BREAK(m,
|
---|
133 | LaunchVMProcess(session, Bstr("vrdp").raw(),
|
---|
134 | ComSafeArrayNullInParam(), progress.asOutParam()));
|
---|
135 |
|
---|
136 | RTPrintf("Waiting for the remote session to open...\n");
|
---|
137 | CHECK_ERROR_BREAK(progress, WaitForCompletion(-1));
|
---|
138 |
|
---|
139 | BOOL completed;
|
---|
140 | CHECK_ERROR_BREAK(progress, COMGETTER(Completed)(&completed));
|
---|
141 | ASSERT(completed);
|
---|
142 |
|
---|
143 | LONG resultCode;
|
---|
144 | CHECK_ERROR_BREAK(progress, COMGETTER(ResultCode)(&resultCode));
|
---|
145 | if (FAILED(resultCode))
|
---|
146 | {
|
---|
147 | ProgressErrorInfo info(progress);
|
---|
148 | com::GluePrintErrorInfo(info);
|
---|
149 | }
|
---|
150 | else
|
---|
151 | {
|
---|
152 | RTPrintf("Remote session has been successfully opened.\n");
|
---|
153 | }
|
---|
154 | }
|
---|
155 | else
|
---|
156 | {
|
---|
157 | RTPrintf("Opening an existing session...\n");
|
---|
158 | CHECK_ERROR_BREAK(m, LockMachine(session, LockType_Shared));
|
---|
159 |
|
---|
160 | ComPtr <IConsole> console;
|
---|
161 | CHECK_ERROR_BREAK(session, COMGETTER(Console)(console.asOutParam()));
|
---|
162 |
|
---|
163 | if (!strcmp(operation, "off"))
|
---|
164 | {
|
---|
165 | ComPtr <IProgress> progress;
|
---|
166 | RTPrintf("Powering the VM off...\n");
|
---|
167 | CHECK_ERROR_BREAK(console, PowerDown(progress.asOutParam()));
|
---|
168 |
|
---|
169 | RTPrintf("Waiting for the VM to power down...\n");
|
---|
170 | CHECK_ERROR_BREAK(progress, WaitForCompletion(-1));
|
---|
171 |
|
---|
172 | BOOL completed;
|
---|
173 | CHECK_ERROR_BREAK(progress, COMGETTER(Completed)(&completed));
|
---|
174 | ASSERT(completed);
|
---|
175 |
|
---|
176 | LONG resultCode;
|
---|
177 | CHECK_ERROR_BREAK(progress, COMGETTER(ResultCode)(&resultCode));
|
---|
178 | if (FAILED(resultCode))
|
---|
179 | {
|
---|
180 | ProgressErrorInfo info(progress);
|
---|
181 | com::GluePrintErrorInfo(info);
|
---|
182 | }
|
---|
183 | else
|
---|
184 | {
|
---|
185 | RTPrintf("VM is powered down.\n");
|
---|
186 | }
|
---|
187 | }
|
---|
188 | else
|
---|
189 | if (!strcmp(operation, "pause"))
|
---|
190 | {
|
---|
191 | RTPrintf("Pausing the VM...\n");
|
---|
192 | CHECK_ERROR_BREAK(console, Pause());
|
---|
193 | }
|
---|
194 | else
|
---|
195 | if (!strcmp(operation, "resume"))
|
---|
196 | {
|
---|
197 | RTPrintf("Resuming the VM...\n");
|
---|
198 | CHECK_ERROR_BREAK(console, Resume());
|
---|
199 | }
|
---|
200 | else
|
---|
201 | {
|
---|
202 | RTPrintf("Invalid operation!\n");
|
---|
203 | }
|
---|
204 | }
|
---|
205 |
|
---|
206 | RTPrintf("Closing the session (may fail after power off)...\n");
|
---|
207 | CHECK_ERROR(session, UnlockMachine());
|
---|
208 | }
|
---|
209 | while (0);
|
---|
210 | RTPrintf("\n");
|
---|
211 |
|
---|
212 | com::Shutdown();
|
---|
213 |
|
---|
214 | RTPrintf("tstHeadless FINISHED.\n");
|
---|
215 |
|
---|
216 | return SUCCEEDED(hrc) ? RTEXITCODE_SUCCESS : RTEXITCODE_FAILURE;
|
---|
217 | }
|
---|
218 |
|
---|