diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-11 08:17:27 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-11 08:17:27 +0000 |
commit | f215e02bf85f68d3a6106c2a1f4f7f063f819064 (patch) | |
tree | 6bb5b92c046312c4e95ac2620b10ddf482d3fa8b /src/VBox/Frontends/VBoxHeadless/testcase | |
parent | Initial commit. (diff) | |
download | virtualbox-f215e02bf85f68d3a6106c2a1f4f7f063f819064.tar.xz virtualbox-f215e02bf85f68d3a6106c2a1f4f7f063f819064.zip |
Adding upstream version 7.0.14-dfsg.upstream/7.0.14-dfsg
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/VBox/Frontends/VBoxHeadless/testcase')
-rw-r--r-- | src/VBox/Frontends/VBoxHeadless/testcase/Makefile.kmk | 38 | ||||
-rw-r--r-- | src/VBox/Frontends/VBoxHeadless/testcase/tstHeadless.cpp | 218 |
2 files changed, 256 insertions, 0 deletions
diff --git a/src/VBox/Frontends/VBoxHeadless/testcase/Makefile.kmk b/src/VBox/Frontends/VBoxHeadless/testcase/Makefile.kmk new file mode 100644 index 00000000..1f6b6af2 --- /dev/null +++ b/src/VBox/Frontends/VBoxHeadless/testcase/Makefile.kmk @@ -0,0 +1,38 @@ +# $Id: Makefile.kmk $ +## @file +# Sub-Makefile for the VRDP (headless RDP server) frontend testcase. +# + +# +# Copyright (C) 2006-2023 Oracle and/or its affiliates. +# +# This file is part of VirtualBox base platform packages, as +# available from https://www.virtualbox.org. +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation, in version 3 of the +# License. +# +# This program is distributed in the hope that it will be useful, but +# WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, see <https://www.gnu.org/licenses>. +# +# SPDX-License-Identifier: GPL-3.0-only +# + +SUB_DEPTH = ../../../../.. +include $(KBUILD_PATH)/subheader.kmk + +ifdef VBOX_WITH_TESTCASES + PROGRAMS += tstHeadless + tstHeadless_TEMPLATE = VBoxMainClientTstExe + tstHeadless_SOURCES = tstHeadless.cpp +endif + +include $(FILE_KBUILD_SUB_FOOTER) + diff --git a/src/VBox/Frontends/VBoxHeadless/testcase/tstHeadless.cpp b/src/VBox/Frontends/VBoxHeadless/testcase/tstHeadless.cpp new file mode 100644 index 00000000..180a952e --- /dev/null +++ b/src/VBox/Frontends/VBoxHeadless/testcase/tstHeadless.cpp @@ -0,0 +1,218 @@ +/** @file + * + * VBox frontends: VBoxHeadless frontend: + * Testcases + */ + +/* + * Copyright (C) 2006-2023 Oracle and/or its affiliates. + * + * This file is part of VirtualBox base platform packages, as + * available from https://www.virtualbox.org. + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation, in version 3 of the + * License. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, see <https://www.gnu.org/licenses>. + * + * SPDX-License-Identifier: GPL-3.0-only + */ + +#include <VBox/com/array.h> +#include <VBox/com/com.h> +#include <VBox/com/string.h> +#include <VBox/com/Guid.h> +#include <VBox/com/ErrorInfo.h> +#include <VBox/com/errorprint.h> + +#include <VBox/com/VirtualBox.h> + +using namespace com; + +#include <VBox/log.h> +#include <iprt/initterm.h> +#include <iprt/stream.h> + + +//////////////////////////////////////////////////////////////////////////////// + +/** + * Entry point. + */ +int main(int argc, char **argv) +{ + // initialize VBox Runtime + RTR3InitExe(argc, &argv, 0); + + // the below cannot be Bstr because on Linux Bstr doesn't work + // until XPCOM (nsMemory) is initialized + const char *name = NULL; + const char *operation = NULL; + + // parse the command line + if (argc > 1) + name = argv [1]; + if (argc > 2) + operation = argv [2]; + + if (!name || !operation) + { + RTPrintf("\nUsage:\n\n" + "%s <machine_name> [on|off|pause|resume]\n\n", + argv [0]); + return 0; + } + + RTPrintf("\n"); + RTPrintf("tstHeadless STARTED.\n"); + + RTPrintf("VM name : {%s}\n" + "Operation : %s\n\n", + name, operation); + + HRESULT hrc; + + hrc = com::Initialize(); + if (FAILED(hrc)) + { + RTPrintf("ERROR: failed to initialize COM!\n"); + return hrc; + } + + do + { + ComPtr <IVirtualBoxClient> virtualBoxClient; + ComPtr <IVirtualBox> virtualBox; + ComPtr <ISession> session; + + RTPrintf("Creating VirtualBox object...\n"); + hrc = virtualBoxClient.createInprocObject(CLSID_VirtualBoxClient); + if (SUCCEEDED(hrc)) + hrc = virtualBoxClient->COMGETTER(VirtualBox)(virtualBox.asOutParam()); + if (FAILED(hrc)) + RTPrintf("ERROR: failed to create the VirtualBox object!\n"); + else + { + hrc = session.createInprocObject(CLSID_Session); + if (FAILED(hrc)) + RTPrintf("ERROR: failed to create a session object!\n"); + } + + if (FAILED(hrc)) + { + com::ErrorInfo info; + if (!info.isFullAvailable() && !info.isBasicAvailable()) + { + com::GluePrintRCMessage(hrc); + RTPrintf("Most likely, the VirtualBox COM server is not running or failed to start.\n"); + } + else + com::GluePrintErrorInfo(info); + break; + } + + ComPtr <IMachine> m; + + // find ID by name + CHECK_ERROR_BREAK(virtualBox, FindMachine(Bstr(name).raw(), + m.asOutParam())); + + if (!strcmp(operation, "on")) + { + ComPtr <IProgress> progress; + RTPrintf("Opening a new (remote) session...\n"); + CHECK_ERROR_BREAK(m, + LaunchVMProcess(session, Bstr("vrdp").raw(), + ComSafeArrayNullInParam(), progress.asOutParam())); + + RTPrintf("Waiting for the remote session to open...\n"); + CHECK_ERROR_BREAK(progress, WaitForCompletion(-1)); + + BOOL completed; + CHECK_ERROR_BREAK(progress, COMGETTER(Completed)(&completed)); + ASSERT(completed); + + LONG resultCode; + CHECK_ERROR_BREAK(progress, COMGETTER(ResultCode)(&resultCode)); + if (FAILED(resultCode)) + { + ProgressErrorInfo info(progress); + com::GluePrintErrorInfo(info); + } + else + { + RTPrintf("Remote session has been successfully opened.\n"); + } + } + else + { + RTPrintf("Opening an existing session...\n"); + CHECK_ERROR_BREAK(m, LockMachine(session, LockType_Shared)); + + ComPtr <IConsole> console; + CHECK_ERROR_BREAK(session, COMGETTER(Console)(console.asOutParam())); + + if (!strcmp(operation, "off")) + { + ComPtr <IProgress> progress; + RTPrintf("Powering the VM off...\n"); + CHECK_ERROR_BREAK(console, PowerDown(progress.asOutParam())); + + RTPrintf("Waiting for the VM to power down...\n"); + CHECK_ERROR_BREAK(progress, WaitForCompletion(-1)); + + BOOL completed; + CHECK_ERROR_BREAK(progress, COMGETTER(Completed)(&completed)); + ASSERT(completed); + + LONG resultCode; + CHECK_ERROR_BREAK(progress, COMGETTER(ResultCode)(&resultCode)); + if (FAILED(resultCode)) + { + ProgressErrorInfo info(progress); + com::GluePrintErrorInfo(info); + } + else + { + RTPrintf("VM is powered down.\n"); + } + } + else + if (!strcmp(operation, "pause")) + { + RTPrintf("Pausing the VM...\n"); + CHECK_ERROR_BREAK(console, Pause()); + } + else + if (!strcmp(operation, "resume")) + { + RTPrintf("Resuming the VM...\n"); + CHECK_ERROR_BREAK(console, Resume()); + } + else + { + RTPrintf("Invalid operation!\n"); + } + } + + RTPrintf("Closing the session (may fail after power off)...\n"); + CHECK_ERROR(session, UnlockMachine()); + } + while (0); + RTPrintf("\n"); + + com::Shutdown(); + + RTPrintf("tstHeadless FINISHED.\n"); + + return SUCCEEDED(hrc) ? RTEXITCODE_SUCCESS : RTEXITCODE_FAILURE; +} + |