| 1 | /* $Id: VBoxServiceControl.h 76553 2019-01-01 01:45:53Z vboxsync $ */
|
|---|
| 2 | /** @file
|
|---|
| 3 | * VBoxServiceControl.h - Internal guest control definitions.
|
|---|
| 4 | */
|
|---|
| 5 |
|
|---|
| 6 | /*
|
|---|
| 7 | * Copyright (C) 2013-2019 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 | #ifndef ___VBoxServiceControl_h
|
|---|
| 19 | #define ___VBoxServiceControl_h
|
|---|
| 20 | #ifndef RT_WITHOUT_PRAGMA_ONCE
|
|---|
| 21 | # pragma once
|
|---|
| 22 | #endif
|
|---|
| 23 |
|
|---|
| 24 | #include <iprt/critsect.h>
|
|---|
| 25 | #include <iprt/list.h>
|
|---|
| 26 | #include <iprt/req.h>
|
|---|
| 27 |
|
|---|
| 28 | #include <VBox/VBoxGuestLib.h>
|
|---|
| 29 | #include <VBox/GuestHost/GuestControl.h>
|
|---|
| 30 | #include <VBox/HostServices/GuestControlSvc.h>
|
|---|
| 31 |
|
|---|
| 32 |
|
|---|
| 33 | /**
|
|---|
| 34 | * Pipe IDs for handling the guest process poll set.
|
|---|
| 35 | */
|
|---|
| 36 | typedef enum VBOXSERVICECTRLPIPEID
|
|---|
| 37 | {
|
|---|
| 38 | VBOXSERVICECTRLPIPEID_UNKNOWN = 0,
|
|---|
| 39 | VBOXSERVICECTRLPIPEID_STDIN = 10,
|
|---|
| 40 | VBOXSERVICECTRLPIPEID_STDIN_WRITABLE = 11,
|
|---|
| 41 | /** Pipe for reading from guest process' stdout. */
|
|---|
| 42 | VBOXSERVICECTRLPIPEID_STDOUT = 40,
|
|---|
| 43 | /** Pipe for reading from guest process' stderr. */
|
|---|
| 44 | VBOXSERVICECTRLPIPEID_STDERR = 50,
|
|---|
| 45 | /** Notification pipe for waking up the guest process
|
|---|
| 46 | * control thread. */
|
|---|
| 47 | VBOXSERVICECTRLPIPEID_IPC_NOTIFY = 100
|
|---|
| 48 | } VBOXSERVICECTRLPIPEID;
|
|---|
| 49 |
|
|---|
| 50 | /**
|
|---|
| 51 | * Structure for one (opened) guest file.
|
|---|
| 52 | */
|
|---|
| 53 | typedef struct VBOXSERVICECTRLFILE
|
|---|
| 54 | {
|
|---|
| 55 | /** Pointer to list archor of following
|
|---|
| 56 | * list node.
|
|---|
| 57 | * @todo Would be nice to have a RTListGetAnchor(). */
|
|---|
| 58 | PRTLISTANCHOR pAnchor;
|
|---|
| 59 | /** Node to global guest control file list. */
|
|---|
| 60 | /** @todo Use a map later? */
|
|---|
| 61 | RTLISTNODE Node;
|
|---|
| 62 | /** The file name. */
|
|---|
| 63 | char szName[RTPATH_MAX];
|
|---|
| 64 | /** The file handle on the guest. */
|
|---|
| 65 | RTFILE hFile;
|
|---|
| 66 | /** File handle to identify this file. */
|
|---|
| 67 | uint32_t uHandle;
|
|---|
| 68 | /** Context ID. */
|
|---|
| 69 | uint32_t uContextID;
|
|---|
| 70 | } VBOXSERVICECTRLFILE;
|
|---|
| 71 | /** Pointer to thread data. */
|
|---|
| 72 | typedef VBOXSERVICECTRLFILE *PVBOXSERVICECTRLFILE;
|
|---|
| 73 |
|
|---|
| 74 | typedef struct VBOXSERVICECTRLSESSIONSTARTUPINFO
|
|---|
| 75 | {
|
|---|
| 76 | /** The session's protocol version to use. */
|
|---|
| 77 | uint32_t uProtocol;
|
|---|
| 78 | /** The session's ID. */
|
|---|
| 79 | uint32_t uSessionID;
|
|---|
| 80 | /** User name (account) to start the guest session under. */
|
|---|
| 81 | char szUser[GUESTPROCESS_MAX_USER_LEN];
|
|---|
| 82 | /** Password of specified user name (account). */
|
|---|
| 83 | char szPassword[GUESTPROCESS_MAX_PASSWORD_LEN];
|
|---|
| 84 | /** Domain of the user account. */
|
|---|
| 85 | char szDomain[GUESTPROCESS_MAX_DOMAIN_LEN];
|
|---|
| 86 | /** Session creation flags.
|
|---|
| 87 | * @sa VBOXSERVICECTRLSESSIONSTARTUPFLAG_* flags. */
|
|---|
| 88 | uint32_t fFlags;
|
|---|
| 89 | } VBOXSERVICECTRLSESSIONSTARTUPINFO;
|
|---|
| 90 | /** Pointer to thread data. */
|
|---|
| 91 | typedef VBOXSERVICECTRLSESSIONSTARTUPINFO *PVBOXSERVICECTRLSESSIONSTARTUPINFO;
|
|---|
| 92 |
|
|---|
| 93 | /**
|
|---|
| 94 | * Structure for a guest session thread to
|
|---|
| 95 | * observe/control the forked session instance from
|
|---|
| 96 | * the VBoxService main executable.
|
|---|
| 97 | */
|
|---|
| 98 | typedef struct VBOXSERVICECTRLSESSIONTHREAD
|
|---|
| 99 | {
|
|---|
| 100 | /** Node to global guest control session list. */
|
|---|
| 101 | /** @todo Use a map later? */
|
|---|
| 102 | RTLISTNODE Node;
|
|---|
| 103 | /** The sessions's startup info. */
|
|---|
| 104 | VBOXSERVICECTRLSESSIONSTARTUPINFO StartupInfo;
|
|---|
| 105 | /** Critical section for thread-safe use. */
|
|---|
| 106 | RTCRITSECT CritSect;
|
|---|
| 107 | /** The worker thread. */
|
|---|
| 108 | RTTHREAD Thread;
|
|---|
| 109 | /** Process handle for forked child. */
|
|---|
| 110 | RTPROCESS hProcess;
|
|---|
| 111 | /** Shutdown indicator; will be set when the thread
|
|---|
| 112 | * needs (or is asked) to shutdown. */
|
|---|
| 113 | bool volatile fShutdown;
|
|---|
| 114 | /** Indicator set by the service thread exiting. */
|
|---|
| 115 | bool volatile fStopped;
|
|---|
| 116 | /** Whether the thread was started or not. */
|
|---|
| 117 | bool fStarted;
|
|---|
| 118 | #if 0 /* Pipe IPC not used yet. */
|
|---|
| 119 | /** Pollset containing all the pipes. */
|
|---|
| 120 | RTPOLLSET hPollSet;
|
|---|
| 121 | RTPIPE hStdInW;
|
|---|
| 122 | RTPIPE hStdOutR;
|
|---|
| 123 | RTPIPE hStdErrR;
|
|---|
| 124 | struct StdPipe
|
|---|
| 125 | {
|
|---|
| 126 | RTHANDLE hChild;
|
|---|
| 127 | PRTHANDLE phChild;
|
|---|
| 128 | } StdIn,
|
|---|
| 129 | StdOut,
|
|---|
| 130 | StdErr;
|
|---|
| 131 | /** The notification pipe associated with this guest session.
|
|---|
| 132 | * This is NIL_RTPIPE for output pipes. */
|
|---|
| 133 | RTPIPE hNotificationPipeW;
|
|---|
| 134 | /** The other end of hNotificationPipeW. */
|
|---|
| 135 | RTPIPE hNotificationPipeR;
|
|---|
| 136 | #endif
|
|---|
| 137 | /** Pipe for handing the secret key to the session process. */
|
|---|
| 138 | RTPIPE hKeyPipe;
|
|---|
| 139 | /** Secret key. */
|
|---|
| 140 | uint8_t abKey[_4K];
|
|---|
| 141 | } VBOXSERVICECTRLSESSIONTHREAD;
|
|---|
| 142 | /** Pointer to thread data. */
|
|---|
| 143 | typedef VBOXSERVICECTRLSESSIONTHREAD *PVBOXSERVICECTRLSESSIONTHREAD;
|
|---|
| 144 |
|
|---|
| 145 | /** Flag indicating that this session has been spawned from
|
|---|
| 146 | * the main executable. */
|
|---|
| 147 | #define VBOXSERVICECTRLSESSION_FLAG_SPAWN RT_BIT(0)
|
|---|
| 148 | /** Flag indicating that this session is anonymous, that is,
|
|---|
| 149 | * it will run start guest processes with the same credentials
|
|---|
| 150 | * as the main executable. */
|
|---|
| 151 | #define VBOXSERVICECTRLSESSION_FLAG_ANONYMOUS RT_BIT(1)
|
|---|
| 152 | /** Flag indicating that started guest processes will dump their
|
|---|
| 153 | * stdout output to a separate file on disk. For debugging. */
|
|---|
| 154 | #define VBOXSERVICECTRLSESSION_FLAG_DUMPSTDOUT RT_BIT(2)
|
|---|
| 155 | /** Flag indicating that started guest processes will dump their
|
|---|
| 156 | * stderr output to a separate file on disk. For debugging. */
|
|---|
| 157 | #define VBOXSERVICECTRLSESSION_FLAG_DUMPSTDERR RT_BIT(3)
|
|---|
| 158 |
|
|---|
| 159 | /**
|
|---|
| 160 | * Structure for maintaining a guest session. This also
|
|---|
| 161 | * contains all started threads (e.g. for guest processes).
|
|---|
| 162 | *
|
|---|
| 163 | * This structure can act in two different ways:
|
|---|
| 164 | * - For legacy guest control handling (protocol version < 2)
|
|---|
| 165 | * this acts as a per-guest process structure containing all
|
|---|
| 166 | * the information needed to get a guest process up and running.
|
|---|
| 167 | * - For newer guest control protocols (>= 2) this structure is
|
|---|
| 168 | * part of the forked session child, maintaining all guest
|
|---|
| 169 | * control objects under it.
|
|---|
| 170 | */
|
|---|
| 171 | typedef struct VBOXSERVICECTRLSESSION
|
|---|
| 172 | {
|
|---|
| 173 | /* The session's startup information. */
|
|---|
| 174 | VBOXSERVICECTRLSESSIONSTARTUPINFO
|
|---|
| 175 | StartupInfo;
|
|---|
| 176 | /** List of active guest process threads
|
|---|
| 177 | * (VBOXSERVICECTRLPROCESS). */
|
|---|
| 178 | RTLISTANCHOR lstProcesses;
|
|---|
| 179 | /** List of guest control files (VBOXSERVICECTRLFILE). */
|
|---|
| 180 | RTLISTANCHOR lstFiles;
|
|---|
| 181 | /** The session's critical section. */
|
|---|
| 182 | RTCRITSECT CritSect;
|
|---|
| 183 | /** Internal session flags, not related
|
|---|
| 184 | * to StartupInfo stuff.
|
|---|
| 185 | * @sa VBOXSERVICECTRLSESSION_FLAG_* flags. */
|
|---|
| 186 | uint32_t fFlags;
|
|---|
| 187 | /** How many processes do we allow keeping around at a time? */
|
|---|
| 188 | uint32_t uProcsMaxKept;
|
|---|
| 189 | } VBOXSERVICECTRLSESSION;
|
|---|
| 190 | /** Pointer to guest session. */
|
|---|
| 191 | typedef VBOXSERVICECTRLSESSION *PVBOXSERVICECTRLSESSION;
|
|---|
| 192 |
|
|---|
| 193 | /**
|
|---|
| 194 | * Structure holding information for starting a guest
|
|---|
| 195 | * process.
|
|---|
| 196 | */
|
|---|
| 197 | typedef struct VBOXSERVICECTRLPROCSTARTUPINFO
|
|---|
| 198 | {
|
|---|
| 199 | /** Full qualified path of process to start (without arguments). */
|
|---|
| 200 | char szCmd[GUESTPROCESS_MAX_CMD_LEN];
|
|---|
| 201 | /** Process execution flags. @sa */
|
|---|
| 202 | uint32_t uFlags;
|
|---|
| 203 | /** Command line arguments. */
|
|---|
| 204 | char szArgs[GUESTPROCESS_MAX_ARGS_LEN];
|
|---|
| 205 | /** Number of arguments specified in pszArgs. */
|
|---|
| 206 | uint32_t uNumArgs;
|
|---|
| 207 | /** String of environment variables ("FOO=BAR") to pass to the process
|
|---|
| 208 | * to start. */
|
|---|
| 209 | char szEnv[GUESTPROCESS_MAX_ENV_LEN];
|
|---|
| 210 | /** Size (in bytes) of environment variables block. */
|
|---|
| 211 | uint32_t cbEnv;
|
|---|
| 212 | /** Number of environment variables specified in pszEnv. */
|
|---|
| 213 | uint32_t uNumEnvVars;
|
|---|
| 214 | /** User name (account) to start the process under. */
|
|---|
| 215 | char szUser[GUESTPROCESS_MAX_USER_LEN];
|
|---|
| 216 | /** Password of specified user name (account). */
|
|---|
| 217 | char szPassword[GUESTPROCESS_MAX_PASSWORD_LEN];
|
|---|
| 218 | /** Domain to be used for authenticating the specified user name (account). */
|
|---|
| 219 | char szDomain[GUESTPROCESS_MAX_DOMAIN_LEN];
|
|---|
| 220 | /** Time limit (in ms) of the process' life time. */
|
|---|
| 221 | uint32_t uTimeLimitMS;
|
|---|
| 222 | /** Process priority. */
|
|---|
| 223 | uint32_t uPriority;
|
|---|
| 224 | /** Process affinity. At the moment we support
|
|---|
| 225 | * up to 4 * 64 = 256 CPUs. */
|
|---|
| 226 | uint64_t uAffinity[4];
|
|---|
| 227 | /** Number of used process affinity blocks. */
|
|---|
| 228 | uint32_t uNumAffinity;
|
|---|
| 229 | } VBOXSERVICECTRLPROCSTARTUPINFO;
|
|---|
| 230 | /** Pointer to a guest process block. */
|
|---|
| 231 | typedef VBOXSERVICECTRLPROCSTARTUPINFO *PVBOXSERVICECTRLPROCSTARTUPINFO;
|
|---|
| 232 |
|
|---|
| 233 | /**
|
|---|
| 234 | * Structure for holding data for one (started) guest process.
|
|---|
| 235 | */
|
|---|
| 236 | typedef struct VBOXSERVICECTRLPROCESS
|
|---|
| 237 | {
|
|---|
| 238 | /** Node. */
|
|---|
| 239 | RTLISTNODE Node;
|
|---|
| 240 | /** Process handle. */
|
|---|
| 241 | RTPROCESS hProcess;
|
|---|
| 242 | /** Number of references using this struct. */
|
|---|
| 243 | uint32_t cRefs;
|
|---|
| 244 | /** The worker thread. */
|
|---|
| 245 | RTTHREAD Thread;
|
|---|
| 246 | /** The session this guest process
|
|---|
| 247 | * is bound to. */
|
|---|
| 248 | PVBOXSERVICECTRLSESSION pSession;
|
|---|
| 249 | /** Shutdown indicator; will be set when the thread
|
|---|
| 250 | * needs (or is asked) to shutdown. */
|
|---|
| 251 | bool volatile fShutdown;
|
|---|
| 252 | /** Whether the guest process thread was stopped or not. */
|
|---|
| 253 | bool volatile fStopped;
|
|---|
| 254 | /** Whether the guest process thread was started or not. */
|
|---|
| 255 | bool fStarted;
|
|---|
| 256 | /** Context ID. */
|
|---|
| 257 | uint32_t uContextID;
|
|---|
| 258 | /** Critical section for thread-safe use. */
|
|---|
| 259 | RTCRITSECT CritSect;
|
|---|
| 260 | /** Process startup information. */
|
|---|
| 261 | VBOXSERVICECTRLPROCSTARTUPINFO
|
|---|
| 262 | StartupInfo;
|
|---|
| 263 | /** The process' PID assigned by the guest OS. */
|
|---|
| 264 | uint32_t uPID;
|
|---|
| 265 | /** The process' request queue to handle requests
|
|---|
| 266 | * from the outside, e.g. the session. */
|
|---|
| 267 | RTREQQUEUE hReqQueue;
|
|---|
| 268 | /** Our pollset, used for accessing the process'
|
|---|
| 269 | * std* pipes + the notification pipe. */
|
|---|
| 270 | RTPOLLSET hPollSet;
|
|---|
| 271 | /** StdIn pipe for addressing writes to the
|
|---|
| 272 | * guest process' stdin.*/
|
|---|
| 273 | RTPIPE hPipeStdInW;
|
|---|
| 274 | /** StdOut pipe for addressing reads from
|
|---|
| 275 | * guest process' stdout.*/
|
|---|
| 276 | RTPIPE hPipeStdOutR;
|
|---|
| 277 | /** StdOut pipe for addressing reads from
|
|---|
| 278 | * guest process' stdout.*/
|
|---|
| 279 | RTPIPE hPipeStdErrR;
|
|---|
| 280 |
|
|---|
| 281 | /** The write end of the notification pipe that is used to poke the thread
|
|---|
| 282 | * monitoring the process.
|
|---|
| 283 | * This is NIL_RTPIPE for output pipes. */
|
|---|
| 284 | RTPIPE hNotificationPipeW;
|
|---|
| 285 | /** The other end of hNotificationPipeW, read by vgsvcGstCtrlProcessProcLoop(). */
|
|---|
| 286 | RTPIPE hNotificationPipeR;
|
|---|
| 287 | } VBOXSERVICECTRLPROCESS;
|
|---|
| 288 | /** Pointer to thread data. */
|
|---|
| 289 | typedef VBOXSERVICECTRLPROCESS *PVBOXSERVICECTRLPROCESS;
|
|---|
| 290 |
|
|---|
| 291 | RT_C_DECLS_BEGIN
|
|---|
| 292 |
|
|---|
| 293 | extern RTLISTANCHOR g_lstControlSessionThreads;
|
|---|
| 294 | extern VBOXSERVICECTRLSESSION g_Session;
|
|---|
| 295 | extern uint32_t g_idControlSvcClient;
|
|---|
| 296 | extern bool g_fControlSupportsOptimizations;
|
|---|
| 297 |
|
|---|
| 298 |
|
|---|
| 299 | /** @name Guest session thread handling.
|
|---|
| 300 | * @{ */
|
|---|
| 301 | extern int VGSvcGstCtrlSessionThreadCreate(PRTLISTANCHOR pList, const PVBOXSERVICECTRLSESSIONSTARTUPINFO pSessionStartupInfo, PVBOXSERVICECTRLSESSIONTHREAD *ppSessionThread);
|
|---|
| 302 | extern int VGSvcGstCtrlSessionThreadDestroy(PVBOXSERVICECTRLSESSIONTHREAD pSession, uint32_t uFlags);
|
|---|
| 303 | extern int VGSvcGstCtrlSessionThreadDestroyAll(PRTLISTANCHOR pList, uint32_t uFlags);
|
|---|
| 304 | extern int VGSvcGstCtrlSessionThreadTerminate(PVBOXSERVICECTRLSESSIONTHREAD pSession);
|
|---|
| 305 | extern RTEXITCODE VGSvcGstCtrlSessionSpawnInit(int argc, char **argv);
|
|---|
| 306 | /** @} */
|
|---|
| 307 | /** @name Per-session functions.
|
|---|
| 308 | * @{ */
|
|---|
| 309 | extern PVBOXSERVICECTRLPROCESS VGSvcGstCtrlSessionRetainProcess(PVBOXSERVICECTRLSESSION pSession, uint32_t uPID);
|
|---|
| 310 | extern int VGSvcGstCtrlSessionClose(PVBOXSERVICECTRLSESSION pSession);
|
|---|
| 311 | extern int VGSvcGstCtrlSessionDestroy(PVBOXSERVICECTRLSESSION pSession);
|
|---|
| 312 | extern int VGSvcGstCtrlSessionInit(PVBOXSERVICECTRLSESSION pSession, uint32_t uFlags);
|
|---|
| 313 | extern int VGSvcGstCtrlSessionHandler(PVBOXSERVICECTRLSESSION pSession, uint32_t uMsg, PVBGLR3GUESTCTRLCMDCTX pHostCtx, void *pvScratchBuf, size_t cbScratchBuf, volatile bool *pfShutdown);
|
|---|
| 314 | extern int VGSvcGstCtrlSessionProcessAdd(PVBOXSERVICECTRLSESSION pSession, PVBOXSERVICECTRLPROCESS pProcess);
|
|---|
| 315 | extern int VGSvcGstCtrlSessionProcessRemove(PVBOXSERVICECTRLSESSION pSession, PVBOXSERVICECTRLPROCESS pProcess);
|
|---|
| 316 | extern int VGSvcGstCtrlSessionProcessStartAllowed(const PVBOXSERVICECTRLSESSION pSession, bool *pbAllowed);
|
|---|
| 317 | extern int VGSvcGstCtrlSessionReapProcesses(PVBOXSERVICECTRLSESSION pSession);
|
|---|
| 318 | /** @} */
|
|---|
| 319 | /** @name Per-guest process functions.
|
|---|
| 320 | * @{ */
|
|---|
| 321 | extern int VGSvcGstCtrlProcessFree(PVBOXSERVICECTRLPROCESS pProcess);
|
|---|
| 322 | extern int VGSvcGstCtrlProcessHandleInput(PVBOXSERVICECTRLPROCESS pProcess, PVBGLR3GUESTCTRLCMDCTX pHostCtx, bool fPendingClose, void *pvBuf, uint32_t cbBuf);
|
|---|
| 323 | extern int VGSvcGstCtrlProcessHandleOutput(PVBOXSERVICECTRLPROCESS pProcess, PVBGLR3GUESTCTRLCMDCTX pHostCtx, uint32_t uHandle, uint32_t cbToRead, uint32_t uFlags);
|
|---|
| 324 | extern int VGSvcGstCtrlProcessHandleTerm(PVBOXSERVICECTRLPROCESS pProcess);
|
|---|
| 325 | extern void VGSvcGstCtrlProcessRelease(PVBOXSERVICECTRLPROCESS pProcess);
|
|---|
| 326 | extern int VGSvcGstCtrlProcessStart(const PVBOXSERVICECTRLSESSION pSession, const PVBOXSERVICECTRLPROCSTARTUPINFO pStartupInfo, uint32_t uContext);
|
|---|
| 327 | extern int VGSvcGstCtrlProcessStop(PVBOXSERVICECTRLPROCESS pProcess);
|
|---|
| 328 | extern int VGSvcGstCtrlProcessWait(const PVBOXSERVICECTRLPROCESS pProcess, RTMSINTERVAL msTimeout, int *pRc);
|
|---|
| 329 | /** @} */
|
|---|
| 330 |
|
|---|
| 331 | RT_C_DECLS_END
|
|---|
| 332 |
|
|---|
| 333 | #endif
|
|---|
| 334 |
|
|---|
