6ef855ecf2121f708685307839f1262e0db1a024vboxsync * VirtualBox Main - Extension Pack Helper Application, usually set-uid-to-root.
441579693f771e49eb05f2bd20c316232155675bvboxsync * Copyright (C) 2010-2012 Oracle Corporation
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * available from http://www.virtualbox.org. This file is free software;
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * you can redistribute it and/or modify it under the terms of the GNU
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * General Public License (GPL) as published by the Free Software
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync/*******************************************************************************
6ef855ecf2121f708685307839f1262e0db1a024vboxsync* Header Files *
6ef855ecf2121f708685307839f1262e0db1a024vboxsync*******************************************************************************/
6ef855ecf2121f708685307839f1262e0db1a024vboxsync/*******************************************************************************
6ef855ecf2121f708685307839f1262e0db1a024vboxsync* Defined Constants And Macros *
6ef855ecf2121f708685307839f1262e0db1a024vboxsync*******************************************************************************/
6ef855ecf2121f708685307839f1262e0db1a024vboxsync/** Enable elevation on Windows and Darwin. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync/** @name Command and option names
6ef855ecf2121f708685307839f1262e0db1a024vboxsync/*******************************************************************************
6ef855ecf2121f708685307839f1262e0db1a024vboxsync* Global Variables *
6ef855ecf2121f708685307839f1262e0db1a024vboxsync*******************************************************************************/
6ef855ecf2121f708685307839f1262e0db1a024vboxsync/* Override RTAssertShouldPanic to prevent gdb process creation. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return true;
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Handle the special standard options when these are specified after the
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param ch The option character.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgInfo(VBOX_PRODUCT " Extension Pack Helper App\n"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "All rights reserved.\n"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "This NOT intended for general use, please use VBoxManage instead\n"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "or call the IExtPackManager API directly.\n"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "Usage: %s <command> [options]\n"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "Commands:\n"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync " install --base-dir <dir> --cert-dir <dir> --name <name> \\\n"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync " --tarball <tarball> --tarball-fd <fd>\n"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync " uninstall --base-dir <dir> --name <name>\n"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync " cleanup --base-dir <dir>\n"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTPrintf("%sr%d\n", VBOX_VERSION_STRING, RTBldCfgRevision());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Checks if the cerficiate directory is valid.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns true if it is valid, false if it isn't.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszCertDir The certificate directory to validate.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic bool IsValidCertificateDir(const char *pszCertDir)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Just be darn strict for now.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTPathAppPrivateNoArch(szCorrect, sizeof(szCorrect));
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return false;
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathAppend(szCorrect, sizeof(szCorrect), VBOX_EXTPACK_CERT_DIR);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return false;
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Checks if the base directory is valid.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns true if it is valid, false if it isn't.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszBaesDir The base directory to validate.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Just be darn strict for now.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTPathAppPrivateArchTop(szCorrect, sizeof(szCorrect));
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return false;
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathAppend(szCorrect, sizeof(szCorrect), VBOX_EXTPACK_INSTALL_DIR);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return false;
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Cleans up a temporary extension pack directory.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * This is used by 'uninstall', 'cleanup' and in the failure path of 'install'.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns The program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszDir The directory to clean up. The caller is
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * responsible for making sure this is valid.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param fTemporary Whether this is a temporary install directory or
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE RemoveExtPackDir(const char *pszDir, bool fTemporary)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /** @todo May have to undo 555 modes here later. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTDirRemoveRecursive(pszDir, RTDIRRMREC_F_CONTENT_AND_DIR);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "Failed to delete the %sextension pack directory: %Rrc ('%s')",
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Common uninstall worker used by both uninstall and install --replace.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns success or failure, message displayed on failure.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszExtPackDir The extension pack directory name.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE CommonUninstallWorker(const char *pszExtPackDir)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /* Rename the extension pack directory before deleting it to prevent new
6ef855ecf2121f708685307839f1262e0db1a024vboxsync VM processes from picking it up. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTStrCopy(szExtPackUnInstDir, sizeof(szExtPackUnInstDir), pszExtPackDir);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTStrCat(szExtPackUnInstDir, sizeof(szExtPackUnInstDir), "-_-uninst");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to construct temporary extension pack path: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTDirRename(pszExtPackDir, szExtPackUnInstDir, RTPATHRENAME_FLAGS_NO_REPLACE);
89054af893e7d71b5cbe100afa0e034b77f95aadvboxsync /* Automatic cleanup and try again. It's in theory possible that we're
89054af893e7d71b5cbe100afa0e034b77f95aadvboxsync racing another cleanup operation here, so just ignore errors and try
89054af893e7d71b5cbe100afa0e034b77f95aadvboxsync again. (There is no installation race due to the exclusive temporary
89054af893e7d71b5cbe100afa0e034b77f95aadvboxsync installation directory.) */
89054af893e7d71b5cbe100afa0e034b77f95aadvboxsync RemoveExtPackDir(szExtPackUnInstDir, false /*fTemporary*/);
89054af893e7d71b5cbe100afa0e034b77f95aadvboxsync rc = RTDirRename(pszExtPackDir, szExtPackUnInstDir, RTPATHRENAME_FLAGS_NO_REPLACE);
89054af893e7d71b5cbe100afa0e034b77f95aadvboxsync "Failed to rename the extension pack directory: %Rrc\n"
89054af893e7d71b5cbe100afa0e034b77f95aadvboxsync "If the problem persists, try running the command: VBoxManage extpack cleanup", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /* Recursively delete the directory content. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RemoveExtPackDir(szExtPackUnInstDir, false /*fTemporary*/);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Wrapper around VBoxExtPackOpenTarFss.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns success or failure, message displayed on failure.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hTarballFile The handle to the tarball file.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param phTarFss Where to return the filesystem stream handle.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE OpenTarFss(RTFILE hTarballFile, PRTVFSFSSTREAM phTarFss)
441579693f771e49eb05f2bd20c316232155675bvboxsync int rc = VBoxExtPackOpenTarFss(hTarballFile, szError, sizeof(szError), phTarFss, NULL);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "%s", szError);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Sets the permissions of the temporary extension pack directory just before
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * renaming it.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * By default the temporary directory is only accessible by root, this function
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * will make it world readable and browseable.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns The program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszDir The temporary extension pack directory.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE SetExtPackPermissions(const char *pszDir)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to set directory permissions: %Rrc ('%s')", rc, pszDir);
1d86b678ed3df7ee8ab1f43edc2660738b0ceda8vboxsync /** @todo TrustedInstaller? */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Wrapper around VBoxExtPackValidateMember.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns Program exit code, failure with message.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszName The name of the directory.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param enmType The object type.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hVfsObj The VFS object.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE ValidateMemberOfExtPack(const char *pszName, RTVFSOBJTYPE enmType, RTVFSOBJ hVfsObj)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = VBoxExtPackValidateMember(pszName, enmType, hVfsObj, szError, sizeof(szError));
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "%s", szError);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Validates the extension pack tarball prior to unpacking.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Operations performed:
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * - Hardening checks.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns The program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszDir The directory where the extension pack has been
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * unpacked.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszExtPackName The expected extension pack name.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszTarball The name of the tarball in case we have to
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * complain about something.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE ValidateUnpackedExtPack(const char *pszDir, const char *pszTarball, const char *pszExtPackName)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgInfo("Validating unpacked extension pack...");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = SUPR3HardenedVerifyDir(pszDir, true /*fRecursive*/, true /*fCheckFiles*/, &ErrInfo.Core);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Hardening check failed with %Rrc: %s", rc, ErrInfo.Core.pszMsg);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Unpacks a directory from an extension pack tarball.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns Program exit code, failure with message.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszDstDirName The name of the unpacked directory.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hVfsObj The source object for the directory.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE UnpackExtPackDir(const char *pszDstDirName, RTVFSOBJ hVfsObj)
adbb0da2a65fc315cb37869fd2c6c80c7d8d5b8bvboxsync * Get the mode mask before creating the directory.
adbb0da2a65fc315cb37869fd2c6c80c7d8d5b8bvboxsync int rc = RTVfsObjQueryInfo(hVfsObj, &ObjInfo, RTFSOBJATTRADD_NOTHING);
adbb0da2a65fc315cb37869fd2c6c80c7d8d5b8bvboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTVfsObjQueryInfo failed on '%s': %Rrc", pszDstDirName, rc);
adbb0da2a65fc315cb37869fd2c6c80c7d8d5b8bvboxsync ObjInfo.Attr.fMode &= ~(RTFS_UNIX_IWOTH | RTFS_UNIX_IWGRP);
939e2ecb812c6402abcc63e7d615c5444acfd02evboxsync rc = RTDirCreate(pszDstDirName, ObjInfo.Attr.fMode, 0);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to create directory '%s': %Rrc", pszDstDirName, rc);
adbb0da2a65fc315cb37869fd2c6c80c7d8d5b8bvboxsync * Because of umask, we have to apply the mode again.
adbb0da2a65fc315cb37869fd2c6c80c7d8d5b8bvboxsync rc = RTPathSetMode(pszDstDirName, ObjInfo.Attr.fMode);
adbb0da2a65fc315cb37869fd2c6c80c7d8d5b8bvboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to set directory permissions on '%s': %Rrc", pszDstDirName, rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /** @todo Ownership tricks on windows? */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Unpacks a file from an extension pack tarball.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns Program exit code, failure with message.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszName The name in the tarball.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszDstFilename The name of the unpacked file.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hVfsIosSrc The source stream for the file.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hUnpackManifest The manifest to add the file digest to.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE UnpackExtPackFile(const char *pszName, const char *pszDstFilename,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTVFSIOSTREAM hVfsIosSrc, RTMANIFEST hUnpackManifest)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Query the object info, we'll need it for buffer sizing as well as
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * setting the file mode.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTVfsIoStrmQueryInfo(hVfsIosSrc, &ObjInfo, RTFSOBJATTRADD_NOTHING);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTVfsIoStrmQueryInfo failed with %Rrc on '%s'", rc, pszDstFilename);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Create the file.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync uint32_t fFlags = RTFILE_O_WRITE | RTFILE_O_DENY_ALL | RTFILE_O_CREATE | (0600 << RTFILE_O_CREATE_MODE_SHIFT);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to create '%s': %Rrc", pszDstFilename, rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Create a I/O stream for the destination file, stack a manifest entry
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * creator on top of it.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTVfsIoStrmFromRTFile(hFile, fFlags, true /*fLeaveOpen*/, &hVfsIosDst2);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTManifestEntryAddPassthruIoStream(hUnpackManifest, hVfsIosDst2, pszName,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Pump the data thru.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTVfsUtilPumpIoStreams(hVfsIosSrc, hVfsIosDst, (uint32_t)RT_MIN(ObjInfo.cbObject, _1G));
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Set the mode mask.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync ObjInfo.Attr.fMode &= ~(RTFS_UNIX_IWOTH | RTFS_UNIX_IWGRP);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /** @todo Windows needs to do more here, I think. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("Failed to set the mode of '%s' to %RTfmode: %Rrc", pszDstFilename, ObjInfo.Attr.fMode, rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("RTManifestPtIosAddEntryNow failed for '%s': %Rrc", pszDstFilename, rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("RTVfsUtilPumpIoStreams failed for '%s': %Rrc", pszDstFilename, rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("RTManifestEntryAddPassthruIoStream failed: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("RTVfsIoStrmFromRTFile failed: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Unpacks the extension pack into the specified directory.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * This will apply ownership and permission changes to all the content, the
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * exception is @a pszDirDst which will be handled by SetExtPackPermissions.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns The program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hTarballFile The tarball to unpack.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszDirDst Where to unpack it.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hValidManifest The manifest we've validated.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszTarball The name of the tarball in case we have to
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * complain about something.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE UnpackExtPack(RTFILE hTarballFile, const char *pszDirDst, RTMANIFEST hValidManifest,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgInfo("Unpacking extension pack into '%s'...", pszDirDst);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Set up the destination path.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTPathAbs(pszDirDst, szDstPath, sizeof(szDstPath) - VBOX_EXTPACK_MAX_MEMBER_NAME_LENGTH - 2);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPathAbs('%s',,) failed: %Rrc", pszDirDst, rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync size_t offDstPath = RTPathStripTrailingSlash(szDstPath);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Open the tar.gz filesystem stream and set up an manifest in-memory file.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTEXITCODE rcExit = OpenTarFss(hTarballFile, &hTarFss);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTManifestCreate(0 /*fFlags*/, &hUnpackManifest);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Process the tarball (would be nice to move this to a function).
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Get the next stream object.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTVfsFsStrmNext(hTarFss, &pszName, &enmType, &hVfsObj);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "RTVfsFsStrmNext failed: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync const char *pszAdjName = pszName[0] == '.' && pszName[1] == '/' ? &pszName[2] : pszName;
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Check the type & name validity then unpack it.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = ValidateMemberOfExtPack(pszName, enmType, hVfsObj);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTStrCopy(&szDstPath[offDstPath], sizeof(szDstPath) - offDstPath, pszAdjName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTVFSIOSTREAM hVfsIos = RTVfsObjToIoStream(hVfsObj);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = UnpackExtPackFile(pszAdjName, szDstPath, hVfsIos, hUnpackManifest);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "Name is too long: '%s' (%Rrc)", pszAdjName, rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Clean up and break out on failure.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Check that what we just extracted matches the already verified
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * manifest.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTManifestEqualsEx(hUnpackManifest, hValidManifest, NULL /*papszIgnoreEntries*/, NULL /*papszIgnoreAttr*/,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "Manifest mismatch: %s", szError);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "RTManifestEqualsEx failed: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTVfsIoStrmFromStdHandle(RTHANDLESTD_OUTPUT, RTFILE_O_WRITE, true, &hVfsIosStdOut);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTVfsIoStrmWrite(hVfsIosStdOut, "Unpack:\n", sizeof("Unpack:\n") - 1, true, NULL);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTManifestWriteStandard(hUnpackManifest, hVfsIosStdOut);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTVfsIoStrmWrite(hVfsIosStdOut, "Valid:\n", sizeof("Valid:\n") - 1, true, NULL);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTManifestWriteStandard(hValidManifest, hVfsIosStdOut);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Wrapper around VBoxExtPackValidateTarball.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns The program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hTarballFile The handle to open the @a pszTarball file.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszExtPackName The name of the extension pack name.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszTarball The name of the tarball in case we have to
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * complain about something.
441579693f771e49eb05f2bd20c316232155675bvboxsync * @param pszTarballDigest The SHA-256 digest of the tarball.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param phValidManifest Where to return the handle to fully validated
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * the manifest for the extension pack. This
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * includes all files.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE ValidateExtPackTarball(RTFILE hTarballFile, const char *pszExtPackName, const char *pszTarball,
441579693f771e49eb05f2bd20c316232155675bvboxsync const char *pszTarballDigest, PRTMANIFEST phValidManifest)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgInfo("Validating extension pack '%s' ('%s')...", pszTarball, pszExtPackName);
441579693f771e49eb05f2bd20c316232155675bvboxsync int rc = VBoxExtPackValidateTarball(hTarballFile, pszExtPackName, pszTarball, pszTarballDigest,
441579693f771e49eb05f2bd20c316232155675bvboxsync szError, sizeof(szError), phValidManifest, NULL /*phXmlFile*/, NULL /*pStrDigest*/);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "%s", szError);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * The 2nd part of the installation process.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns The program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszBaseDir The base directory.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszCertDir The certificat directory.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszTarball The tarball name.
441579693f771e49eb05f2bd20c316232155675bvboxsync * @param pszTarballDigest The SHA-256 digest of the tarball. Empty string
441579693f771e49eb05f2bd20c316232155675bvboxsync * if no digest available.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hTarballFile The handle to open the @a pszTarball file.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hTarballFileOpt The tarball file handle (optional).
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszName The extension pack name.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszMangledName The mangled extension pack name.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param fReplace Whether to replace any existing ext pack.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE DoInstall2(const char *pszBaseDir, const char *pszCertDir, const char *pszTarball,
441579693f771e49eb05f2bd20c316232155675bvboxsync const char *pszTarballDigest, RTFILE hTarballFile, RTFILE hTarballFileOpt,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync const char *pszName, const char *pszMangledName, bool fReplace)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Do some basic validation of the tarball file.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTFileQueryInfo(hTarballFile, &ObjInfo, RTFSOBJATTRADD_UNIX);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTFileQueryInfo failed with %Rrc on '%s'", rc, pszTarball);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Not a regular file: %s", pszTarball);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTFileQueryInfo(hTarballFileOpt, &ObjInfo2, RTFSOBJATTRADD_UNIX);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTFileQueryInfo failed with %Rrc on --tarball-fd", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if ( ObjInfo.Attr.u.Unix.INodeIdDevice != ObjInfo2.Attr.u.Unix.INodeIdDevice
6ef855ecf2121f708685307839f1262e0db1a024vboxsync || ObjInfo.Attr.u.Unix.INodeId != ObjInfo2.Attr.u.Unix.INodeId)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "--tarball and --tarball-fd does not match");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Construct the paths to the two directories we'll be using.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathJoin(szFinalPath, sizeof(szFinalPath), pszBaseDir, pszMangledName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "Failed to construct the path to the final extension pack directory: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathJoin(szTmpPath, sizeof(szTmpPath) - 64, pszBaseDir, pszMangledName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTStrPrintf(&szTmpPath[cchTmpPath], sizeof(szTmpPath) - cchTmpPath, "-_-inst-%u", (uint32_t)RTProcSelf());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "Failed to construct the path to the temporary extension pack directory: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Check that they don't exist at this point in time, unless fReplace=true.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathQueryInfoEx(szFinalPath, &ObjInfo, RTFSOBJATTRADD_NOTHING, RTPATH_F_ON_LINK);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (RT_SUCCESS(rc) && RTFS_IS_DIRECTORY(ObjInfo.Attr.fMode))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "The extension pack is already installed. You must uninstall the old one first.");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "Found non-directory file system object where the extension pack would be installed ('%s')",
6ef855ecf2121f708685307839f1262e0db1a024vboxsync else if (rc != VERR_FILE_NOT_FOUND && rc != VERR_PATH_NOT_FOUND)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Unexpected RTPathQueryInfoEx status code %Rrc for '%s'", rc, szFinalPath);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathQueryInfoEx(szTmpPath, &ObjInfo, RTFSOBJATTRADD_NOTHING, RTPATH_F_ON_LINK);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (rc != VERR_FILE_NOT_FOUND && rc != VERR_PATH_NOT_FOUND)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Unexpected RTPathQueryInfoEx status code %Rrc for '%s'", rc, szFinalPath);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Create the temporary directory and prepare the extension pack within it.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * If all checks out correctly, rename it to the final directory.
383d5bd7b4b12176fbba2defc6c162e5b84e8ac0vboxsync * Because of umask, we have to apply the mode again.
383d5bd7b4b12176fbba2defc6c162e5b84e8ac0vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to set directory permissions on '%s': %Rrc", pszBaseDir, rc);
383d5bd7b4b12176fbba2defc6c162e5b84e8ac0vboxsync /** @todo Ownership tricks on windows? */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to create temporary directory: %Rrc ('%s')", rc, szTmpPath);
441579693f771e49eb05f2bd20c316232155675bvboxsync RTEXITCODE rcExit = ValidateExtPackTarball(hTarballFile, pszName, pszTarball, pszTarballDigest, &hValidManifest);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = UnpackExtPack(hTarballFile, szTmpPath, hValidManifest, pszTarball);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = ValidateUnpackedExtPack(szTmpPath, pszTarball, pszName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTDirRename(szTmpPath, szFinalPath, RTPATHRENAME_FLAGS_NO_REPLACE);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /* Automatic uninstall if --replace was given. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTDirRename(szTmpPath, szFinalPath, RTPATHRENAME_FLAGS_NO_REPLACE);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgInfo("Successfully installed '%s' (%s)", pszName, pszTarball);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync "Failed to rename the temporary directory to the final one: %Rrc ('%s' -> '%s')",
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Clean up the temporary directory on failure.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Implements the 'install' command.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns The program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param argc The number of program arguments.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param argv The program arguments.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Parse the parameters.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Note! The --base-dir and --cert-dir are only for checking that the
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * caller and this help applications have the same idea of where
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * things are. Likewise, the --name is for verifying assumptions
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * the caller made about the name. The optional --tarball-fd option
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * is just for easing the paranoia on the user side.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTGetOptInit(&GetState, argc, argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /*fFlags*/);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTGetOptInit failed: %Rrc\n", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Too many --base-dir options");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Invalid base directory: '%s'", pszBaseDir);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Too many --cert-dir options");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Invalid certificate directory: '%s'", pszCertDir);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Too many --name options");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Invalid extension pack name: '%s'", pszName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Too many --tarball options");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Too many --tarball-fd options");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "The --tarball-fd value is out of range: %#RX64", ValueUnion.u64);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "RTFileFromNative failed on --target-fd value: %Rrc", rc);
441579693f771e49eb05f2bd20c316232155675bvboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Too many --sha-256 options");
441579693f771e49eb05f2bd20c316232155675bvboxsync rc = RTSha256FromString(pszTarballDigest, abDigest);
441579693f771e49eb05f2bd20c316232155675bvboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Bad SHA-256 string: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Missing --name option");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Missing --base-dir option");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Missing --cert-dir option");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Missing --tarball option");
441579693f771e49eb05f2bd20c316232155675bvboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Missing --sha-256 option");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Ok, down to business.
6c6531128b39093daeac902a8705c0bdf15b31ccvboxsync RTCString *pstrMangledName = VBoxExtPackMangleName(pszName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to mangle name ('%s)", pszName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTFileOpen(&hTarballFile, pszTarball, RTFILE_O_READ | RTFILE_O_OPEN | RTFILE_O_DENY_WRITE);
441579693f771e49eb05f2bd20c316232155675bvboxsync rcExit = DoInstall2(pszBaseDir, pszCertDir, pszTarball, pszTarballDigest, hTarballFile, hTarballFileOpt,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to open the extension pack tarball: %Rrc ('%s')", rc, pszTarball);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Implements the 'uninstall' command.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns The program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param argc The number of program arguments.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param argv The program arguments.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE DoUninstall(int argc, char **argv)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Parse the parameters.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Note! The --base-dir is only for checking that the caller and this help
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * applications have the same idea of where things are.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTGetOptInit(&GetState, argc, argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /*fFlags*/);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTGetOptInit failed: %Rrc\n", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Too many --base-dir options");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Invalid base directory: '%s'", pszBaseDir);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Too many --name options");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Invalid extension pack name: '%s'", pszName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Missing --name option");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Missing --base-dir option");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Mangle the name so we can construct the directory names.
6c6531128b39093daeac902a8705c0bdf15b31ccvboxsync RTCString *pstrMangledName = VBoxExtPackMangleName(pszName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to mangle name ('%s)", pszName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Ok, down to business.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /* Check that it exists. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathJoin(szExtPackDir, sizeof(szExtPackDir), pszBaseDir, strMangledName.c_str());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to construct extension pack path: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgInfo("Extension pack not installed. Nothing to do.");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTEXITCODE rcExit = CommonUninstallWorker(szExtPackDir);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgInfo("Successfully removed extension pack '%s'\n", pszName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Implements the 'cleanup' command.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns The program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param argc The number of program arguments.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param argv The program arguments.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Parse the parameters.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Note! The --base-dir is only for checking that the caller and this help
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * applications have the same idea of where things are.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTGetOptInit(&GetState, argc, argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /*fFlags*/);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTGetOptInit failed: %Rrc\n", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Too many --base-dir options");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Invalid base directory: '%s'", pszBaseDir);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Missing --base-dir option");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Ok, down to business.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed open the base directory: %Rrc ('%s')", rc, pszBaseDir);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTDirReadEx(pDir, &Entry, NULL /*pcbDirEntry*/, RTFSOBJATTRADD_NOTHING, RTPATH_F_ON_LINK);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "RTDirReadEx returns %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Only directories which conform with our temporary install/uninstall
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * naming scheme are candidates for cleaning.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync fCandidate = VBoxExtPackIsValidMangledName(Entry.szName, pszMarker - &Entry.szName[0]);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Recursive delete, safe.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathJoin(szPath, sizeof(szPath), pszBaseDir, Entry.szName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTEXITCODE rcExit2 = RemoveExtPackDir(szPath, true /*fTemporary*/);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgInfo("Successfully removed '%s'.", Entry.szName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPathJoin failed with %Rrc for '%s'", rc, Entry.szName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync#if !defined(RT_OS_WINDOWS) && !defined(RT_OS_DARWIN)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Looks in standard locations for a suitable exec tool.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns true if found, false if not.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszPath Where to store the path to the tool on
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * successs.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param cbPath The size of the buffer @a pszPath points to.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszName The name of the tool we're looking for.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic bool FindExecTool(char *pszPath, size_t cbPath, const char *pszName)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync for (unsigned i = 0; i < RT_ELEMENTS(s_apszPaths); i++)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTPathJoin(pszPath, cbPath, s_apszPaths[i], pszName);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathQueryInfoEx(pszPath, &ObjInfo, RTFSOBJATTRADD_UNIX, RTPATH_F_FOLLOW_LINK);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return true;
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return false;
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Copies the content of a file to a stream.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param hSrc The source file.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pDst The destination stream.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param fComplain Whether to complain about errors (i.e. is this
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * stderr, if not keep the trap shut because it
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * may be missing when running under VBoxSVC.)
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic void CopyFileToStdXxx(RTFILE hSrc, PRTSTREAM pDst, bool fComplain)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTFileRead(hSrc, abBuf, sizeof(abBuf), &cbRead);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Relaunches ourselves as a elevated process using platform specific facilities.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns Program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszExecPath The executable path.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param papszArgs The arguments.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param cSuArgs The number of argument entries reserved for the
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * 'su' like programs at the start of papszArgs.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param cMyArgs The number of arguments following @a cSuArgs.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param iCmd The command that is being executed. (For
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * selecting messages.)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszDisplayInfoHack Display information hack. Platform specific++.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE RelaunchElevatedNative(const char *pszExecPath, const char **papszArgs, int cSuArgs, int cMyArgs,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync CoInitializeEx(NULL, COINIT_APARTMENTTHREADED | COINIT_DISABLE_OLE1DDE);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTStrToUtf16(pszExecPath, (PRTUTF16 *)&Info.lpFile);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTGetOptArgvToString(&pszCmdLine, &papszArgs[cSuArgs + 1], RTGETOPTARGV_CNV_QUOTE_MS_CRT);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTStrToUtf16(pszCmdLine, (PRTUTF16 *)&Info.lpParameters);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /* Apply display hacks. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync const char *pszArg = strstr(pszDisplayInfoHack, "hwnd=");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTStrToUInt64Ex(pszArg + sizeof("hwnd=") - 1, NULL, 0, &u64Hwnd);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync Info.hMonitor = MonitorFromWindow(hwnd, MONITOR_DEFAULTTOPRIMARY);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync Info.hMonitor = MonitorFromPoint(Pt, MONITOR_DEFAULTTOPRIMARY);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Wait for the process, make sure the deal with messages.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync DWORD dwRc = MsgWaitForMultipleObjects(1, &Info.hProcess, FALSE, 5000/*ms*/, QS_ALLEVENTS);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("MsgWaitForMultipleObjects returned: %#x (%d), err=%u", dwRc, dwRc, GetLastError());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (GetExitCodeProcess(Info.hProcess, &dwExitCode))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("ShellExecuteExW return INVALID_HANDLE_VALUE as Info.hProcess");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("ShellExecuteExW failed: %u (%#x)", GetLastError(), GetLastError());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTPathAppPrivateArch(szIconName, sizeof(szIconName));
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathAppend(szIconName, sizeof(szIconName), "../Resources/virtualbox.png");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to construct icon path: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync OSStatus orc = AuthorizationCreate(NULL, 0, kAuthorizationFlagDefaults, &AuthRef);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Preautorize the privileged execution of ourselves.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync AuthorizationItem AuthItem = { kAuthorizationRightExecute, 0, NULL, 0 };
6ef855ecf2121f708685307839f1262e0db1a024vboxsync static char s_szPrompt[] = "VirtualBox needs further rights to make changes to your installation.\n\n";
6ef855ecf2121f708685307839f1262e0db1a024vboxsync { kAuthorizationEnvironmentPrompt, strlen(s_szPrompt), s_szPrompt, 0 },
6ef855ecf2121f708685307839f1262e0db1a024vboxsync { kAuthorizationEnvironmentIcon, strlen(szIconName), szIconName, 0 }
6ef855ecf2121f708685307839f1262e0db1a024vboxsync AuthorizationEnvironment AuthEnv = { RT_ELEMENTS(aAuthEnvItems), aAuthEnvItems };
6ef855ecf2121f708685307839f1262e0db1a024vboxsync orc = AuthorizationCopyRights(AuthRef, &AuthRights, &AuthEnv,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync kAuthorizationFlagPreAuthorize | kAuthorizationFlagInteractionAllowed
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Execute with extra permissions
6ef855ecf2121f708685307839f1262e0db1a024vboxsync orc = AuthorizationExecuteWithPrivileges(AuthRef, pszExecPath, kAuthorizationFlagDefaults,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Read the output of the tool, the read will fail when it quits.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync size_t cbRead = fread(achBuf, 1, sizeof(achBuf), pSocketStrm);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("AuthorizationExecuteWithPrivileges failed: %d", orc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("AuthorizationCopyRights failed: %d", orc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync AuthorizationFree(AuthRef, kAuthorizationFlagDefaults);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Several of the alternatives below will require a command line.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTGetOptArgvToString(&pszCmdLine, &papszArgs[cSuArgs], RTGETOPTARGV_CNV_QUOTE_BOURNE_SH);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTGetOptArgvToString failed: %Rrc");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Look for various standard stuff for executing a program as root.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * N.B. When adding new arguments, please make 100% sure RelaunchElevated
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * allocates enough array entries.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * TODO: Feel free to contribute code for using PolicyKit directly.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * kdesudo is available on KDE3/KDE4
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (fHaveDisplayVar && FindExecTool(szExecTool, sizeof(szExecTool), "kdesudo"))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync ? "VirtualBox extension pack installer"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync ? "VirtualBox extension pack uninstaller"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync : "VirtualBox extension pack maintainer";
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * gksu is our favorite as it is very well integrated.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync else if (fHaveDisplayVar && FindExecTool(szExecTool, sizeof(szExecTool), "gksu"))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync#if 0 /* older gksu does not grok --description nor '--' and multiple args. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync ? "VirtualBox extension pack installer"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync ? "VirtualBox extension pack uninstaller"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync : "VirtualBox extension pack maintainer";
6ef855ecf2121f708685307839f1262e0db1a024vboxsync#elif defined(RT_OS_SOLARIS) /* Force it not to use pfexec as it won't wait then. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * pkexec may work for ssh console sessions as well if the right agents
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * are installed. However it is very generic and does not allow for any
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * custom messages. Thus it comes after gksu.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync else if (FindExecTool(szExecTool, sizeof(szExecTool), "pkexec"))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * The ultimate fallback is running 'su -' within an xterm. We use the
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * title of the xterm to tell what is going on.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync && FindExecTool(szExecTool, sizeof(szExecTool), "su")
6ef855ecf2121f708685307839f1262e0db1a024vboxsync && FindExecTool(szXterm, sizeof(szXterm), "xterm"))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync ? "VirtualBox extension pack installer - su"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync ? "VirtualBox extension pack uninstaller - su"
6ef855ecf2121f708685307839f1262e0db1a024vboxsync : "VirtualBox extension pack maintainer - su";
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("Unable to locate 'pkexec', 'gksu' or 'su+xterm'. Try perform the operation using VBoxManage running as root");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("Unable to locate 'pkexec'. Try perform the operation using VBoxManage running as root");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Argument list constructed, execute it and wait for the exec
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * program to complete.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTProcCreateEx(papszArgs[iSuArg], &papszArgs[iSuArg], RTENV_DEFAULT, 0 /*fFlags*/,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync NULL /*phStdIn*/, NULL /*phStdOut*/, NULL /*phStdErr*/, NULL /*pszAsUser*/, NULL /*pszPassword*/,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTProcWait(hProcess, RTPROCWAIT_FLAGS_BLOCK, &Status);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("Error while waiting for '%s': %Rrc", papszArgs[iSuArg], rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTMsgError("Failed to execute '%s': %Rrc", papszArgs[iSuArg], rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Relaunches ourselves as a elevated process using platform specific facilities.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns Program exit code.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param argc The number of arguments.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param argv The arguments.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param iCmd The command that is being executed.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pszDisplayInfoHack Display information hack. Platform specific++.
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncstatic RTEXITCODE RelaunchElevated(int argc, char **argv, int iCmd, const char *pszDisplayInfoHack)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * We need the executable name later, so get it now when it's easy to quit.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (!RTProcGetExecutablePath(szExecPath,sizeof(szExecPath)))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTProcGetExecutablePath failed");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Create a couple of temporary files for stderr and stdout.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPathTemp failed: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathAppend(szTempDir, sizeof(szTempDir), "VBoxExtPackHelper-XXXXXX");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPathAppend failed: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTDirCreateTemp failed: %Rrc", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathJoin(szStdOut, sizeof(szStdOut), szTempDir, "stdout");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTPathJoin(szStdErr, sizeof(szStdErr), szTempDir, "stderr");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTFileOpen(&hStdOut, szStdOut, RTFILE_O_READWRITE | RTFILE_O_CREATE | RTFILE_O_DENY_NONE
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTFileOpen(&hStdErr, szStdErr, RTFILE_O_READWRITE | RTFILE_O_CREATE | RTFILE_O_DENY_NONE
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Insert the --elevated and stdout/err names into the argument
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * list. Note that darwin skips the --stdout bit, so don't
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * change the order here.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync char const **papszArgs = (char const **)RTMemTmpAllocZ((cSuArgs + cArgs + 1) * sizeof(const char *));
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Do the platform specific process execution (waiting included).
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RelaunchElevatedNative(szExecPath, papszArgs, cSuArgs, cArgs, iCmd, pszDisplayInfoHack);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Copy the standard files to our standard handles.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync CopyFileToStdXxx(hStdErr, g_pStdErr, true /*fComplain*/);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Checks if the process is elevated or not.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @returns RTEXITCODE_SUCCESS if preconditions are fine,
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * otherwise error message + RTEXITCODE_FAILURE.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * @param pfElevated Where to store the elevation indicator.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /** @todo This should probably check if UAC is diabled and if we are
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Administrator first. Also needs to check for Vista+ first, probably.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "OpenProcessToken failed: %u (%#x)", GetLastError(), GetLastError());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Check if we're member of the Administrators group. If we aren't, there
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * is no way to elevate ourselves to system admin.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * N.B. CheckTokenMembership does not do the job here (due to attributes?).
6ef855ecf2121f708685307839f1262e0db1a024vboxsync SID_IDENTIFIER_AUTHORITY NtAuthority = SECURITY_NT_AUTHORITY;
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (AllocateAndInitializeSid(&NtAuthority, 2, SECURITY_BUILTIN_DOMAIN_RID, DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &pAdminGrpSid))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync ConvertSidToStringSid(pAdminGrpSid, &pszAdminGrpSid);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if ( !GetTokenInformation(hToken, TokenGroups, NULL, 0, &cb)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync PTOKEN_GROUPS pTokenGroups = (PTOKEN_GROUPS)RTMemAllocZ(cb);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (GetTokenInformation(hToken, TokenGroups, pTokenGroups, cb, &cb))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync for (DWORD iGrp = 0; iGrp < pTokenGroups->GroupCount; iGrp++)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync ConvertSidToStringSid(pTokenGroups->Groups[iGrp].Sid, &pszGrpSid);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (EqualSid(pAdminGrpSid, pTokenGroups->Groups[iGrp].Sid))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /* That it's listed is enough I think, ignore attributes. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "GetTokenInformation(TokenGroups,cb) failed: %u (%#x)", GetLastError(), GetLastError());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "GetTokenInformation(TokenGroups,0) failed: %u (%#x)", GetLastError(), GetLastError());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "AllocateAndInitializeSid failed: %u (%#x)", GetLastError(), GetLastError());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Check the integrity level (Vista / UAC).
6ef855ecf2121f708685307839f1262e0db1a024vboxsync# define MY_TokenIntegrityLevel ((TOKEN_INFORMATION_CLASS)25)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if ( !GetTokenInformation(hToken, MY_TokenIntegrityLevel, NULL, 0, &cb)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync PSID_AND_ATTRIBUTES pSidAndAttr = (PSID_AND_ATTRIBUTES)RTMemAlloc(cb);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (GetTokenInformation(hToken, MY_TokenIntegrityLevel, pSidAndAttr, cb, &cb))
6ef855ecf2121f708685307839f1262e0db1a024vboxsync DWORD dwIntegrityLevel = *GetSidSubAuthority(pSidAndAttr->Sid, *GetSidSubAuthorityCount(pSidAndAttr->Sid) - 1U);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (dwIntegrityLevel >= MY_SECURITY_MANDATORY_HIGH_RID)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "GetTokenInformation failed: %u (%#x)", GetLastError(), GetLastError());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync else if ( GetLastError() == ERROR_INVALID_PARAMETER
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "GetTokenInformation failed: %u (%#x)", GetLastError(), GetLastError());
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "Membership in the Administrators group is required to perform this action");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * On Unixy systems, we check if the executable and the current user is
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * the same. This heuristic works fine for both hardened and development
6ef855ecf2121f708685307839f1262e0db1a024vboxsync if (RTProcGetExecutablePath(szExecPath, sizeof(szExecPath)) == NULL)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTProcGetExecutablePath failed");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync int rc = RTPathQueryInfoEx(szExecPath, &ObjInfo, RTFSOBJATTRADD_UNIX, RTPATH_F_ON_LINK);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPathQueryInfoEx failed");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync#endif /* WITH_ELEVATION */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Initialize IPRT and check that we're correctly installed.
7d172bb4ed1217e28e28829193159954f6b40ef7vboxsync int rc = RTR3InitExe(argc, &argv, RTR3INIT_FLAGS_UTF8_ARGV); /* WinMain gives us UTF-8, see below. */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = SUPR3HardenedVerifySelf(argv[0], true /*fInternal*/, &ErrInfo.Core);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "%s", ErrInfo.Core.pszMsg);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Elevation check.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Parse the top level arguments until we find a command.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync { "uninstall", CMD_UNINSTALL, RTGETOPT_REQ_NOTHING },
6ef855ecf2121f708685307839f1262e0db1a024vboxsync { "--elevated", OPT_ELEVATED, RTGETOPT_REQ_NOTHING },
6ef855ecf2121f708685307839f1262e0db1a024vboxsync { "--display-info-hack", OPT_DISP_INFO_HACK, RTGETOPT_REQ_STRING },
6ef855ecf2121f708685307839f1262e0db1a024vboxsync rc = RTGetOptInit(&GetState, argc, argv, s_aOptions, RT_ELEMENTS(s_aOptions), 1, 0 /*fFlags*/);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTGetOptInit failed: %Rrc\n", rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "No command specified");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RelaunchElevated(argc, argv, ch, pszDisplayInfoHack);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * Standard error should end with rcExit=RTEXITCODE_SUCCESS on
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * success since the exit code may otherwise get lost in the
6ef855ecf2121f708685307839f1262e0db1a024vboxsync * process elevation fun.
6ef855ecf2121f708685307839f1262e0db1a024vboxsync RTStrmPrintf(g_pStdErr, "rcExit=RTEXITCODE_SUCCESS\n");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync FILE *pFile = freopen(ValueUnion.psz, "r+", ch == OPT_STDOUT ? stdout : stderr);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "freopen on '%s': %Rrc", ValueUnion.psz, rc);
6ef855ecf2121f708685307839f1262e0db1a024vboxsync return RTMsgErrorExit(RTEXITCODE_SYNTAX, "--display-info-hack shall only occur once");
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /* not currently reached */
6ef855ecf2121f708685307839f1262e0db1a024vboxsync /* not reached */
6ef855ecf2121f708685307839f1262e0db1a024vboxsyncextern "C" int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nShowCmd)
6ef855ecf2121f708685307839f1262e0db1a024vboxsync NOREF(hPrevInstance); NOREF(nShowCmd); NOREF(lpCmdLine);
152e1e373e38dea3d8cfe8bc06c7d3050cb130e2vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "GetCommandLineW failed");
152e1e373e38dea3d8cfe8bc06c7d3050cb130e2vboxsync rc = RTUtf16ToUtf8(pwszCmdLine, &pszCmdLine); /* leaked */
152e1e373e38dea3d8cfe8bc06c7d3050cb130e2vboxsync return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to convert the command line: %Rrc", rc);
152e1e373e38dea3d8cfe8bc06c7d3050cb130e2vboxsync rc = RTGetOptArgvFromString(&papszArgs, &cArgs, pszCmdLine, NULL);
152e1e373e38dea3d8cfe8bc06c7d3050cb130e2vboxsync rc = RTMsgErrorExit(RTEXITCODE_FAILURE, "RTGetOptArgvFromString failed: %Rrc", rc);