VirtualBox

source: vbox/trunk/src/VBox/Frontends/VBoxHeadless/testcase/tstHeadless.cpp@ 98187

Last change on this file since 98187 was 98103, checked in by vboxsync, 23 months ago

Copyright year updates by scm.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 6.4 KB
Line 
1/** @file
2 *
3 * VBox frontends: VBoxHeadless frontend:
4 * Testcases
5 */
6
7/*
8 * Copyright (C) 2006-2023 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
38using 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 */
50int 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
Note: See TracBrowser for help on using the repository browser.

© 2024 Oracle Support Privacy / Do Not Sell My Info Terms of Use Trademark Policy Automated Access Etiquette