GuestCtrlImplPrivate.h@ 98123

Last change on this file since 98123 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: 41.1 KB

Line
1	/* $Id: GuestCtrlImplPrivate.h 98103 2023-01-17 14:15:46Z vboxsync $ */
2	/** @file
3	* Internal helpers/structures for guest control functionality.
4	*/
5
6	/*
7	* Copyright (C) 2011-2023 Oracle and/or its affiliates.
8	*
9	* This file is part of VirtualBox base platform packages, as
10	* available from https://www.virtualbox.org.
11	*
12	* This program is free software; you can redistribute it and/or
13	* modify it under the terms of the GNU General Public License
14	* as published by the Free Software Foundation, in version 3 of the
15	* License.
16	*
17	* This program is distributed in the hope that it will be useful, but
18	* WITHOUT ANY WARRANTY; without even the implied warranty of
19	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20	* General Public License for more details.
21	*
22	* You should have received a copy of the GNU General Public License
23	* along with this program; if not, see <https://www.gnu.org/licenses>.
24	*
25	* SPDX-License-Identifier: GPL-3.0-only
26	*/
27
28	#ifndef MAIN_INCLUDED_GuestCtrlImplPrivate_h
29	#define MAIN_INCLUDED_GuestCtrlImplPrivate_h
30	#ifndef RT_WITHOUT_PRAGMA_ONCE
31	# pragma once
32	#endif
33
34	#include "ConsoleImpl.h"
35	#include "Global.h"
36
37	#include <iprt/asm.h>
38	#include <iprt/env.h>
39	#include <iprt/semaphore.h>
40	#include <iprt/cpp/utils.h>
41
42	#include <VBox/com/com.h>
43	#include <VBox/com/ErrorInfo.h>
44	#include <VBox/com/string.h>
45	#include <VBox/com/VirtualBox.h>
46	#include <VBox/err.h> /* VERR_GSTCTL_GUEST_ERROR */
47
48	#include <map>
49	#include <vector>
50
51	using namespace com;
52
53	#ifdef VBOX_WITH_GUEST_CONTROL
54	# include <VBox/GuestHost/GuestControl.h>
55	# include <VBox/HostServices/GuestControlSvc.h>
56	using namespace guestControl;
57	#endif
58
59	/** Vector holding a process' CPU affinity. */
60	typedef std::vector <LONG> ProcessAffinity;
61	/** Vector holding process startup arguments. */
62	typedef std::vector <Utf8Str> ProcessArguments;
63
64	class GuestProcessStreamBlock;
65	class GuestSession;
66
67
68	/**
69	* Simple structure mantaining guest credentials.
70	*/
71	struct GuestCredentials
72	{
73	Utf8Str mUser;
74	Utf8Str mPassword;
75	Utf8Str mDomain;
76	};
77
78
79	/**
80	* Wrapper around the RTEnv API, unusable base class.
81	*
82	* @remarks Feel free to elevate this class to iprt/cpp/env.h as RTCEnv.
83	*/
84	class GuestEnvironmentBase
85	{
86	public:
87	/**
88	* Default constructor.
89	*
90	* The user must invoke one of the init methods before using the object.
91	*/
92	GuestEnvironmentBase(void)
93	: m_hEnv(NIL_RTENV)
94	, m_cRefs(1)
95	, m_fFlags(0)
96	{ }
97
98	/**
99	* Destructor.
100	*/
101	virtual ~GuestEnvironmentBase(void)
102	{
103	Assert(m_cRefs <= 1);
104	int rc = RTEnvDestroy(m_hEnv); AssertRC(rc);
105	m_hEnv = NIL_RTENV;
106	}
107
108	/**
109	* Retains a reference to this object.
110	* @returns New reference count.
111	* @remarks Sharing an object is currently only safe if no changes are made to
112	* it because RTENV does not yet implement any locking. For the only
113	* purpose we need this, implementing IGuestProcess::environment by
114	* using IGuestSession::environmentBase, that's fine as the session
115	* base environment is immutable.
116	*/
117	uint32_t retain(void)
118	{
119	uint32_t cRefs = ASMAtomicIncU32(&m_cRefs);
120	Assert(cRefs > 1); Assert(cRefs < _1M);
121	return cRefs;
122
123	}
124	/** Useful shortcut. */
125	uint32_t retainConst(void) const { return unconst(this)->retain(); }
126
127	/**
128	* Releases a reference to this object, deleting the object when reaching zero.
129	* @returns New reference count.
130	*/
131	uint32_t release(void)
132	{
133	uint32_t cRefs = ASMAtomicDecU32(&m_cRefs);
134	Assert(cRefs < _1M);
135	if (cRefs == 0)
136	delete this;
137	return cRefs;
138	}
139
140	/** Useful shortcut. */
141	uint32_t releaseConst(void) const { return unconst(this)->retain(); }
142
143	/**
144	* Checks if the environment has been successfully initialized or not.
145	*
146	* @returns @c true if initialized, @c false if not.
147	*/
148	bool isInitialized(void) const
149	{
150	return m_hEnv != NIL_RTENV;
151	}
152
153	/**
154	* Returns the variable count.
155	* @return Number of variables.
156	* @sa RTEnvCountEx
157	*/
158	uint32_t count(void) const
159	{
160	return RTEnvCountEx(m_hEnv);
161	}
162
163	/**
164	* Deletes the environment change record entirely.
165	*
166	* The count() method will return zero after this call.
167	*
168	* @sa RTEnvReset
169	*/
170	void reset(void)
171	{
172	int rc = RTEnvReset(m_hEnv);
173	AssertRC(rc);
174	}
175
176	/**
177	* Exports the environment change block as an array of putenv style strings.
178	*
179	*
180	* @returns VINF_SUCCESS or VERR_NO_MEMORY.
181	* @param pArray The output array.
182	*/
183	int queryPutEnvArray(std::vector<com::Utf8Str> *pArray) const
184	{
185	uint32_t cVars = RTEnvCountEx(m_hEnv);
186	try
187	{
188	pArray->resize(cVars);
189	for (uint32_t iVar = 0; iVar < cVars; iVar++)
190	{
191	const char *psz = RTEnvGetByIndexRawEx(m_hEnv, iVar);
192	AssertReturn(psz, VERR_INTERNAL_ERROR_3); /* someone is racing us! */
193	(*pArray)[iVar] = psz;
194	}
195	return VINF_SUCCESS;
196	}
197	catch (std::bad_alloc &)
198	{
199	return VERR_NO_MEMORY;
200	}
201	}
202
203	/**
204	* Applies an array of putenv style strings.
205	*
206	* @returns IPRT status code.
207	* @param rArray The array with the putenv style strings.
208	* @param pidxError Where to return the index causing trouble on
209	* failure. Optional.
210	* @sa RTEnvPutEx
211	*/
212	int applyPutEnvArray(const std::vector<com::Utf8Str> &rArray, size_t *pidxError = NULL)
213	{
214	size_t const cArray = rArray.size();
215	for (size_t i = 0; i < cArray; i++)
216	{
217	int rc = RTEnvPutEx(m_hEnv, rArray[i].c_str());
218	if (RT_FAILURE(rc))
219	{
220	if (pidxError)
221	*pidxError = i;
222	return rc;
223	}
224	}
225	return VINF_SUCCESS;
226	}
227
228	/**
229	* Applies the changes from another environment to this.
230	*
231	* @returns IPRT status code.
232	* @param rChanges Reference to an environment which variables will be
233	* imported and, if it's a change record, schedule
234	* variable unsets will be applied.
235	* @sa RTEnvApplyChanges
236	*/
237	int applyChanges(const GuestEnvironmentBase &rChanges)
238	{
239	return RTEnvApplyChanges(m_hEnv, rChanges.m_hEnv);
240	}
241
242	/**
243	* See RTEnvQueryUtf8Block for details.
244	* @returns IPRT status code.
245	* @param ppszzBlock Where to return the block pointer.
246	* @param pcbBlock Where to optionally return the block size.
247	* @sa RTEnvQueryUtf8Block
248	*/
249	int queryUtf8Block(char *ppszzBlock, size_t pcbBlock)
250	{
251	return RTEnvQueryUtf8Block(m_hEnv, true /fSorted/, ppszzBlock, pcbBlock);
252	}
253
254	/**
255	* Frees what queryUtf8Block returned, NULL ignored.
256	* @sa RTEnvFreeUtf8Block
257	*/
258	static void freeUtf8Block(char *pszzBlock)
259	{
260	return RTEnvFreeUtf8Block(pszzBlock);
261	}
262
263	/**
264	* Applies a block on the format returned by queryUtf8Block.
265	*
266	* @returns IPRT status code.
267	* @param pszzBlock Pointer to the block.
268	* @param cbBlock The size of the block.
269	* @param fNoEqualMeansUnset Whether the lack of a '=' (equal) sign in a
270	* string means it should be unset (@c true), or if
271	* it means the variable should be defined with an
272	* empty value (@c false, the default).
273	* @todo move this to RTEnv!
274	*/
275	int copyUtf8Block(const char *pszzBlock, size_t cbBlock, bool fNoEqualMeansUnset = false)
276	{
277	int rc = VINF_SUCCESS;
278	while (cbBlock > 0 && *pszzBlock != '\0')
279	{
280	const char pszEnd = (const char )memchr(pszzBlock, '\0', cbBlock);
281	if (!pszEnd)
282	return VERR_BUFFER_UNDERFLOW;
283	int rc2;
284	if (fNoEqualMeansUnset \|\| strchr(pszzBlock, '='))
285	rc2 = RTEnvPutEx(m_hEnv, pszzBlock);
286	else
287	rc2 = RTEnvSetEx(m_hEnv, pszzBlock, "");
288	if (RT_FAILURE(rc2) && RT_SUCCESS(rc))
289	rc = rc2;
290
291	/* Advance. */
292	cbBlock -= pszEnd - pszzBlock;
293	if (cbBlock < 2)
294	return VERR_BUFFER_UNDERFLOW;
295	cbBlock--;
296	pszzBlock = pszEnd + 1;
297	}
298
299	/* The remainder must be zero padded. */
300	if (RT_SUCCESS(rc))
301	{
302	if (ASMMemIsZero(pszzBlock, cbBlock))
303	return VINF_SUCCESS;
304	return VERR_TOO_MUCH_DATA;
305	}
306	return rc;
307	}
308
309	/**
310	* Get an environment variable.
311	*
312	* @returns IPRT status code.
313	* @param rName The variable name.
314	* @param pValue Where to return the value.
315	* @sa RTEnvGetEx
316	*/
317	int getVariable(const com::Utf8Str &rName, com::Utf8Str *pValue) const
318	{
319	size_t cchNeeded;
320	int rc = RTEnvGetEx(m_hEnv, rName.c_str(), NULL, 0, &cchNeeded);
321	if ( RT_SUCCESS(rc)
322	\|\| rc == VERR_BUFFER_OVERFLOW)
323	{
324	try
325	{
326	pValue->reserve(cchNeeded + 1);
327	rc = RTEnvGetEx(m_hEnv, rName.c_str(), pValue->mutableRaw(), pValue->capacity(), NULL);
328	pValue->jolt();
329	}
330	catch (std::bad_alloc &)
331	{
332	rc = VERR_NO_STR_MEMORY;
333	}
334	}
335	return rc;
336	}
337
338	/**
339	* Checks if the given variable exists.
340	*
341	* @returns @c true if it exists, @c false if not or if it's an scheduled unset
342	* in a environment change record.
343	* @param rName The variable name.
344	* @sa RTEnvExistEx
345	*/
346	bool doesVariableExist(const com::Utf8Str &rName) const
347	{
348	return RTEnvExistEx(m_hEnv, rName.c_str());
349	}
350
351	/**
352	* Set an environment variable.
353	*
354	* @returns IPRT status code.
355	* @param rName The variable name.
356	* @param rValue The value of the variable.
357	* @sa RTEnvSetEx
358	*/
359	int setVariable(const com::Utf8Str &rName, const com::Utf8Str &rValue)
360	{
361	return RTEnvSetEx(m_hEnv, rName.c_str(), rValue.c_str());
362	}
363
364	/**
365	* Unset an environment variable.
366	*
367	* @returns IPRT status code.
368	* @param rName The variable name.
369	* @sa RTEnvUnsetEx
370	*/
371	int unsetVariable(const com::Utf8Str &rName)
372	{
373	return RTEnvUnsetEx(m_hEnv, rName.c_str());
374	}
375
376	protected:
377	/**
378	* Copy constructor.
379	* @throws HRESULT
380	*/
381	GuestEnvironmentBase(const GuestEnvironmentBase &rThat, bool fChangeRecord, uint32_t fFlags = 0)
382	: m_hEnv(NIL_RTENV)
383	, m_cRefs(1)
384	, m_fFlags(fFlags)
385	{
386	int rc = cloneCommon(rThat, fChangeRecord);
387	if (RT_FAILURE(rc))
388	throw (Global::vboxStatusCodeToCOM(rc));
389	}
390
391	/**
392	* Common clone/copy method with type conversion abilities.
393	*
394	* @returns IPRT status code.
395	* @param rThat The object to clone.
396	* @param fChangeRecord Whether the this instance is a change record (true)
397	* or normal (false) environment.
398	*/
399	int cloneCommon(const GuestEnvironmentBase &rThat, bool fChangeRecord)
400	{
401	int rc = VINF_SUCCESS;
402	RTENV hNewEnv = NIL_RTENV;
403	if (rThat.m_hEnv != NIL_RTENV)
404	{
405	/*
406	* Clone it.
407	*/
408	if (RTEnvIsChangeRecord(rThat.m_hEnv) == fChangeRecord)
409	rc = RTEnvClone(&hNewEnv, rThat.m_hEnv);
410	else
411	{
412	/* Need to type convert it. */
413	if (fChangeRecord)
414	rc = RTEnvCreateChangeRecordEx(&hNewEnv, rThat.m_fFlags);
415	else
416	rc = RTEnvCreateEx(&hNewEnv, rThat.m_fFlags);
417	if (RT_SUCCESS(rc))
418	{
419	rc = RTEnvApplyChanges(hNewEnv, rThat.m_hEnv);
420	if (RT_FAILURE(rc))
421	RTEnvDestroy(hNewEnv);
422	}
423	}
424	}
425	else
426	{
427	/*
428	* Create an empty one so the object works smoothly.
429	* (Relevant for GuestProcessStartupInfo and internal commands.)
430	*/
431	if (fChangeRecord)
432	rc = RTEnvCreateChangeRecordEx(&hNewEnv, rThat.m_fFlags);
433	else
434	rc = RTEnvCreateEx(&hNewEnv, rThat.m_fFlags);
435	}
436	if (RT_SUCCESS(rc))
437	{
438	RTEnvDestroy(m_hEnv);
439	m_hEnv = hNewEnv;
440	m_fFlags = rThat.m_fFlags;
441	}
442	return rc;
443	}
444
445
446	/** The environment change record. */
447	RTENV m_hEnv;
448	/** Reference counter. */
449	uint32_t volatile m_cRefs;
450	/** RTENV_CREATE_F_XXX. */
451	uint32_t m_fFlags;
452	};
453
454	class GuestEnvironmentChanges;
455
456
457	/**
458	* Wrapper around the RTEnv API for a normal environment.
459	*/
460	class GuestEnvironment : public GuestEnvironmentBase
461	{
462	public:
463	/**
464	* Default constructor.
465	*
466	* The user must invoke one of the init methods before using the object.
467	*/
468	GuestEnvironment(void)
469	: GuestEnvironmentBase()
470	{ }
471
472	/**
473	* Copy operator.
474	* @param rThat The object to copy.
475	* @throws HRESULT
476	*/
477	GuestEnvironment(const GuestEnvironment &rThat)
478	: GuestEnvironmentBase(rThat, false /fChangeRecord/)
479	{ }
480
481	/**
482	* Copy operator.
483	* @param rThat The object to copy.
484	* @throws HRESULT
485	*/
486	GuestEnvironment(const GuestEnvironmentBase &rThat)
487	: GuestEnvironmentBase(rThat, false /fChangeRecord/)
488	{ }
489
490	/**
491	* Initialize this as a normal environment block.
492	* @returns IPRT status code.
493	* @param fFlags RTENV_CREATE_F_XXX
494	*/
495	int initNormal(uint32_t fFlags)
496	{
497	AssertReturn(m_hEnv == NIL_RTENV, VERR_WRONG_ORDER);
498	m_fFlags = fFlags;
499	return RTEnvCreateEx(&m_hEnv, fFlags);
500	}
501
502	/**
503	* Replaces this environemnt with that in @a rThat.
504	*
505	* @returns IPRT status code
506	* @param rThat The environment to copy. If it's a different type
507	* we'll convert the data to a normal environment block.
508	*/
509	int copy(const GuestEnvironmentBase &rThat)
510	{
511	return cloneCommon(rThat, false /fChangeRecord/);
512	}
513
514	/**
515	* @copydoc copy()
516	*/
517	GuestEnvironment &operator=(const GuestEnvironmentBase &rThat)
518	{
519	int rc = copy(rThat);
520	if (RT_FAILURE(rc))
521	throw (Global::vboxStatusCodeToCOM(rc));
522	return *this;
523	}
524
525	/** @copydoc copy() */
526	GuestEnvironment &operator=(const GuestEnvironment &rThat)
527	{ return operator=((const GuestEnvironmentBase &)rThat); }
528
529	/** @copydoc copy() */
530	GuestEnvironment &operator=(const GuestEnvironmentChanges &rThat)
531	{ return operator=((const GuestEnvironmentBase &)rThat); }
532
533	};
534
535
536	/**
537	* Wrapper around the RTEnv API for a environment change record.
538	*
539	* This class is used as a record of changes to be applied to a different
540	* environment block (in VBoxService before launching a new process).
541	*/
542	class GuestEnvironmentChanges : public GuestEnvironmentBase
543	{
544	public:
545	/**
546	* Default constructor.
547	*
548	* The user must invoke one of the init methods before using the object.
549	*/
550	GuestEnvironmentChanges(void)
551	: GuestEnvironmentBase()
552	{ }
553
554	/**
555	* Copy operator.
556	* @param rThat The object to copy.
557	* @throws HRESULT
558	*/
559	GuestEnvironmentChanges(const GuestEnvironmentChanges &rThat)
560	: GuestEnvironmentBase(rThat, true /fChangeRecord/)
561	{ }
562
563	/**
564	* Copy operator.
565	* @param rThat The object to copy.
566	* @throws HRESULT
567	*/
568	GuestEnvironmentChanges(const GuestEnvironmentBase &rThat)
569	: GuestEnvironmentBase(rThat, true /fChangeRecord/)
570	{ }
571
572	/**
573	* Initialize this as a environment change record.
574	* @returns IPRT status code.
575	* @param fFlags RTENV_CREATE_F_XXX
576	*/
577	int initChangeRecord(uint32_t fFlags)
578	{
579	AssertReturn(m_hEnv == NIL_RTENV, VERR_WRONG_ORDER);
580	m_fFlags = fFlags;
581	return RTEnvCreateChangeRecordEx(&m_hEnv, fFlags);
582	}
583
584	/**
585	* Replaces this environemnt with that in @a rThat.
586	*
587	* @returns IPRT status code
588	* @param rThat The environment to copy. If it's a different type
589	* we'll convert the data to a set of changes.
590	*/
591	int copy(const GuestEnvironmentBase &rThat)
592	{
593	return cloneCommon(rThat, true /fChangeRecord/);
594	}
595
596	/**
597	* @copydoc copy()
598	*/
599	GuestEnvironmentChanges &operator=(const GuestEnvironmentBase &rThat)
600	{
601	int rc = copy(rThat);
602	if (RT_FAILURE(rc))
603	throw (Global::vboxStatusCodeToCOM(rc));
604	return *this;
605	}
606
607	/** @copydoc copy() */
608	GuestEnvironmentChanges &operator=(const GuestEnvironmentChanges &rThat)
609	{ return operator=((const GuestEnvironmentBase &)rThat); }
610
611	/** @copydoc copy() */
612	GuestEnvironmentChanges &operator=(const GuestEnvironment &rThat)
613	{ return operator=((const GuestEnvironmentBase &)rThat); }
614	};
615
616	/**
617	* Class for keeping guest error information.
618	*/
619	class GuestErrorInfo
620	{
621	public:
622
623	/**
624	* Enumeration for specifying the guest error type.
625	*/
626	enum Type
627	{
628	/** Guest error is anonymous. Avoid this. */
629	Type_Anonymous = 0,
630	/** Guest error is from a guest session. */
631	Type_Session,
632	/** Guest error is from a guest process. */
633	Type_Process,
634	/** Guest error is from a guest file object. */
635	Type_File,
636	/** Guest error is from a guest directory object. */
637	Type_Directory,
638	/** Guest error is from a the built-in toolbox "vbox_cat" command. */
639	Type_ToolCat,
640	/** Guest error is from a the built-in toolbox "vbox_ls" command. */
641	Type_ToolLs,
642	/** Guest error is from a the built-in toolbox "vbox_rm" command. */
643	Type_ToolRm,
644	/** Guest error is from a the built-in toolbox "vbox_mkdir" command. */
645	Type_ToolMkDir,
646	/** Guest error is from a the built-in toolbox "vbox_mktemp" command. */
647	Type_ToolMkTemp,
648	/** Guest error is from a the built-in toolbox "vbox_stat" command. */
649	Type_ToolStat,
650	/** The usual 32-bit hack. */
651	Type_32BIT_HACK = 0x7fffffff
652	};
653
654	/**
655	* Initialization constructor.
656	*
657	* @param eType Error type to use.
658	* @param rc IPRT-style rc to use.
659	* @param pcszWhat Subject to use.
660	*/
661	GuestErrorInfo(GuestErrorInfo::Type eType, int rc, const char *pcszWhat)
662	{
663	int rc2 = setV(eType, rc, pcszWhat);
664	if (RT_FAILURE(rc2))
665	throw rc2;
666	}
667
668	/**
669	* Returns the (IPRT-style) rc of this error.
670	*
671	* @returns VBox status code.
672	*/
673	int getRc(void) const { return mRc; }
674
675	/**
676	* Returns the type of this error.
677	*
678	* @returns Error type.
679	*/
680	Type getType(void) const { return mType; }
681
682	/**
683	* Returns the subject of this error.
684	*
685	* @returns Subject as a string.
686	*/
687	Utf8Str getWhat(void) const { return mWhat; }
688
689	/**
690	* Sets the error information using a variable arguments list (va_list).
691	*
692	* @returns VBox status code.
693	* @param eType Error type to use.
694	* @param rc IPRT-style rc to use.
695	* @param pcszWhat Subject to use.
696	*/
697	int setV(GuestErrorInfo::Type eType, int rc, const char *pcszWhat)
698	{
699	mType = eType;
700	mRc = rc;
701	mWhat = pcszWhat;
702
703	return VINF_SUCCESS;
704	}
705
706	protected:
707
708	/** Error type. */
709	Type mType;
710	/** IPRT-style error code. */
711	int mRc;
712	/** Subject string related to this error. */
713	Utf8Str mWhat;
714	};
715
716	/**
717	* Structure for keeping all the relevant guest directory
718	* information around.
719	*/
720	struct GuestDirectoryOpenInfo
721	{
722	GuestDirectoryOpenInfo(void)
723	: mFlags(0) { }
724
725	/** The directory path. */
726	Utf8Str mPath;
727	/** Then open filter. */
728	Utf8Str mFilter;
729	/** Opening flags. */
730	uint32_t mFlags;
731	};
732
733
734	/**
735	* Structure for keeping all the relevant guest file
736	* information around.
737	*/
738	struct GuestFileOpenInfo
739	{
740	GuestFileOpenInfo(void)
741	: mAccessMode((FileAccessMode_T)0)
742	, mOpenAction((FileOpenAction_T)0)
743	, mSharingMode((FileSharingMode_T)0)
744	, mCreationMode(0)
745	, mfOpenEx(0) { }
746
747	/**
748	* Validates a file open info.
749	*
750	* @returns \c true if valid, \c false if not.
751	*/
752	bool IsValid(void) const
753	{
754	if (mfOpenEx) /** @todo Open flags not implemented yet. */
755	return false;
756
757	switch (mOpenAction)
758	{
759	case FileOpenAction_OpenExisting:
760	break;
761	case FileOpenAction_OpenOrCreate:
762	break;
763	case FileOpenAction_CreateNew:
764	break;
765	case FileOpenAction_CreateOrReplace:
766	break;
767	case FileOpenAction_OpenExistingTruncated:
768	{
769	if ( mAccessMode == FileAccessMode_ReadOnly
770	\|\| mAccessMode == FileAccessMode_AppendOnly
771	\|\| mAccessMode == FileAccessMode_AppendRead)
772	return false;
773	break;
774	}
775	case FileOpenAction_AppendOrCreate: /* Deprecated, do not use. */
776	break;
777	default:
778	AssertFailedReturn(false);
779	break;
780	}
781
782	return true; /** @todo Do we need more checks here? */
783	}
784
785	/** The filename. */
786	Utf8Str mFilename;
787	/** The file access mode. */
788	FileAccessMode_T mAccessMode;
789	/** The file open action. */
790	FileOpenAction_T mOpenAction;
791	/** The file sharing mode. */
792	FileSharingMode_T mSharingMode;
793	/** Octal creation mode. */
794	uint32_t mCreationMode;
795	/** Extended open flags (currently none defined). */
796	uint32_t mfOpenEx;
797	};
798
799
800	/**
801	* Structure representing information of a
802	* file system object.
803	*/
804	struct GuestFsObjData
805	{
806	GuestFsObjData(void)
807	: mType(FsObjType_Unknown)
808	, mObjectSize(0)
809	, mAllocatedSize(0)
810	, mAccessTime(0)
811	, mBirthTime(0)
812	, mChangeTime(0)
813	, mModificationTime(0)
814	, mUID(0)
815	, mGID(0)
816	, mNodeID(0)
817	, mNodeIDDevice(0)
818	, mNumHardLinks(0)
819	, mDeviceNumber(0)
820	, mGenerationID(0)
821	, mUserFlags(0) { }
822
823	/** @name Helper functions to extract the data from a certin VBoxService tool's guest stream block.
824	* @{ */
825	int FromLs(const GuestProcessStreamBlock &strmBlk, bool fLong);
826	int FromRm(const GuestProcessStreamBlock &strmBlk);
827	int FromStat(const GuestProcessStreamBlock &strmBlk);
828	int FromMkTemp(const GuestProcessStreamBlock &strmBlk);
829	/** @} */
830
831	/** @name Static helper functions to work with time from stream block keys.
832	* @{ */
833	static PRTTIMESPEC TimeSpecFromKey(const GuestProcessStreamBlock &strmBlk, const Utf8Str &strKey, PRTTIMESPEC pTimeSpec);
834	static int64_t UnixEpochNsFromKey(const GuestProcessStreamBlock &strmBlk, const Utf8Str &strKey);
835	/** @} */
836
837	/** @name helper functions to work with IPRT stuff.
838	* @{ */
839	RTFMODE GetFileMode(void) const;
840	/** @} */
841
842	Utf8Str mName;
843	FsObjType_T mType;
844	Utf8Str mFileAttrs;
845	int64_t mObjectSize;
846	int64_t mAllocatedSize;
847	int64_t mAccessTime;
848	int64_t mBirthTime;
849	int64_t mChangeTime;
850	int64_t mModificationTime;
851	Utf8Str mUserName;
852	int32_t mUID;
853	int32_t mGID;
854	Utf8Str mGroupName;
855	Utf8Str mACL;
856	int64_t mNodeID;
857	uint32_t mNodeIDDevice;
858	uint32_t mNumHardLinks;
859	uint32_t mDeviceNumber;
860	uint32_t mGenerationID;
861	uint32_t mUserFlags;
862	};
863
864
865	/**
866	* Structure for keeping all the relevant guest session
867	* startup parameters around.
868	*/
869	class GuestSessionStartupInfo
870	{
871	public:
872
873	GuestSessionStartupInfo(void)
874	: mID(UINT32_MAX)
875	, mIsInternal(false /* Non-internal session */)
876	, mOpenTimeoutMS(30 * 1000 /* 30s opening timeout */)
877	, mOpenFlags(0 /* No opening flags set */) { }
878
879	/** The session's friendly name. Optional. */
880	Utf8Str mName;
881	/** The session's unique ID. Used to encode a context ID.
882	* UINT32_MAX if not initialized. */
883	uint32_t mID;
884	/** Flag indicating if this is an internal session
885	* or not. Internal session are not accessible by
886	* public API clients. */
887	bool mIsInternal;
888	/** Timeout (in ms) used for opening the session. */
889	uint32_t mOpenTimeoutMS;
890	/** Session opening flags. */
891	uint32_t mOpenFlags;
892	};
893
894
895	/**
896	* Structure for keeping all the relevant guest process
897	* startup parameters around.
898	*/
899	class GuestProcessStartupInfo
900	{
901	public:
902
903	GuestProcessStartupInfo(void)
904	: mFlags(ProcessCreateFlag_None)
905	, mTimeoutMS(UINT32_MAX /* No timeout by default */)
906	, mPriority(ProcessPriority_Default)
907	, mAffinity(0) { }
908
909	/** The process' friendly name. */
910	Utf8Str mName;
911	/** The executable. */
912	Utf8Str mExecutable;
913	/** Arguments vector (starting with argument \#0). */
914	ProcessArguments mArguments;
915	/** The process environment change record. */
916	GuestEnvironmentChanges mEnvironmentChanges;
917	/** Process creation flags. */
918	uint32_t mFlags;
919	/** Timeout (in ms) the process is allowed to run.
920	* Specify UINT32_MAX if no timeout (unlimited run time) is given. */
921	ULONG mTimeoutMS;
922	/** Process priority. */
923	ProcessPriority_T mPriority;
924	/** Process affinity. At the moment we
925	* only support 64 VCPUs. API and
926	* guest can do more already! */
927	uint64_t mAffinity;
928	};
929
930
931	/**
932	* Class representing the "value" side of a "key=value" pair.
933	*/
934	class GuestProcessStreamValue
935	{
936	public:
937
938	GuestProcessStreamValue(void) { }
939	GuestProcessStreamValue(const char *pszValue)
940	: mValue(pszValue) {}
941
942	GuestProcessStreamValue(const GuestProcessStreamValue& aThat)
943	: mValue(aThat.mValue) { }
944
945	/** Copy assignment operator. */
946	GuestProcessStreamValue &operator=(GuestProcessStreamValue const &a_rThat) RT_NOEXCEPT
947	{
948	mValue = a_rThat.mValue;
949
950	return *this;
951	}
952
953	Utf8Str mValue;
954	};
955
956	/** Map containing "key=value" pairs of a guest process stream. */
957	typedef std::pair< Utf8Str, GuestProcessStreamValue > GuestCtrlStreamPair;
958	typedef std::map < Utf8Str, GuestProcessStreamValue > GuestCtrlStreamPairMap;
959	typedef std::map < Utf8Str, GuestProcessStreamValue >::iterator GuestCtrlStreamPairMapIter;
960	typedef std::map < Utf8Str, GuestProcessStreamValue >::const_iterator GuestCtrlStreamPairMapIterConst;
961
962	/**
963	* Class representing a block of stream pairs (key=value). Each block in a raw guest
964	* output stream is separated by "\0\0", each pair is separated by "\0". The overall
965	* end of a guest stream is marked by "\0\0\0\0".
966	*/
967	class GuestProcessStreamBlock
968	{
969	public:
970
971	GuestProcessStreamBlock(void);
972
973	virtual ~GuestProcessStreamBlock(void);
974
975	public:
976
977	void Clear(void);
978
979	#ifdef DEBUG
980	void DumpToLog(void) const;
981	#endif
982
983	const char GetString(const char pszKey) const;
984	size_t GetCount(void) const;
985	int GetRc(void) const;
986	int GetInt64Ex(const char pszKey, int64_t piVal) const;
987	int64_t GetInt64(const char *pszKey) const;
988	int GetUInt32Ex(const char pszKey, uint32_t puVal) const;
989	uint32_t GetUInt32(const char *pszKey, uint32_t uDefault = 0) const;
990	int32_t GetInt32(const char *pszKey, int32_t iDefault = 0) const;
991
992	bool IsEmpty(void) { return mPairs.empty(); }
993
994	int SetValue(const char pszKey, const char pszValue);
995
996	protected:
997
998	GuestCtrlStreamPairMap mPairs;
999	};
1000
1001	/** Vector containing multiple allocated stream pair objects. */
1002	typedef std::vector< GuestProcessStreamBlock > GuestCtrlStreamObjects;
1003	typedef std::vector< GuestProcessStreamBlock >::iterator GuestCtrlStreamObjectsIter;
1004	typedef std::vector< GuestProcessStreamBlock >::const_iterator GuestCtrlStreamObjectsIterConst;
1005
1006	/**
1007	* Class for parsing machine-readable guest process output by VBoxService'
1008	* toolbox commands ("vbox_ls", "vbox_stat" etc), aka "guest stream".
1009	*/
1010	class GuestProcessStream
1011	{
1012
1013	public:
1014
1015	GuestProcessStream();
1016
1017	virtual ~GuestProcessStream();
1018
1019	public:
1020
1021	int AddData(const BYTE *pbData, size_t cbData);
1022
1023	void Destroy();
1024
1025	#ifdef DEBUG
1026	void Dump(const char *pszFile);
1027	#endif
1028
1029	size_t GetOffset() { return m_offBuffer; }
1030
1031	size_t GetSize() { return m_cbUsed; }
1032
1033	int ParseBlock(GuestProcessStreamBlock &streamBlock);
1034
1035	protected:
1036
1037	/** Maximum allowed size the stream buffer can grow to.
1038	* Defaults to 32 MB. */
1039	size_t m_cbMax;
1040	/** Currently allocated size of internal stream buffer. */
1041	size_t m_cbAllocated;
1042	/** Currently used size at m_offBuffer. */
1043	size_t m_cbUsed;
1044	/** Current byte offset within the internal stream buffer. */
1045	size_t m_offBuffer;
1046	/** Internal stream buffer. */
1047	BYTE *m_pbBuffer;
1048	};
1049
1050	class Guest;
1051	class Progress;
1052
1053	class GuestWaitEventPayload
1054	{
1055
1056	public:
1057
1058	GuestWaitEventPayload(void)
1059	: uType(0),
1060	cbData(0),
1061	pvData(NULL) { }
1062
1063	/**
1064	* Initialization constructor. Will throw() VBox status code (rc).
1065	*
1066	* @param uTypePayload Payload type to set.
1067	* @param pvPayload Pointer to payload data to set (deep copy).
1068	* @param cbPayload Size (in bytes) of payload data to set.
1069	*/
1070	GuestWaitEventPayload(uint32_t uTypePayload,
1071	const void *pvPayload, uint32_t cbPayload)
1072	: uType(0),
1073	cbData(0),
1074	pvData(NULL)
1075	{
1076	int rc = copyFrom(uTypePayload, pvPayload, cbPayload);
1077	if (RT_FAILURE(rc))
1078	throw rc;
1079	}
1080
1081	virtual ~GuestWaitEventPayload(void)
1082	{
1083	Clear();
1084	}
1085
1086	GuestWaitEventPayload& operator=(const GuestWaitEventPayload &that)
1087	{
1088	CopyFromDeep(that);
1089	return *this;
1090	}
1091
1092	public:
1093
1094	void Clear(void)
1095	{
1096	if (pvData)
1097	{
1098	Assert(cbData);
1099	RTMemFree(pvData);
1100	cbData = 0;
1101	pvData = NULL;
1102	}
1103	uType = 0;
1104	}
1105
1106	int CopyFromDeep(const GuestWaitEventPayload &payload)
1107	{
1108	return copyFrom(payload.uType, payload.pvData, payload.cbData);
1109	}
1110
1111	const void* Raw(void) const { return pvData; }
1112
1113	size_t Size(void) const { return cbData; }
1114
1115	uint32_t Type(void) const { return uType; }
1116
1117	void* MutableRaw(void) { return pvData; }
1118
1119	Utf8Str ToString(void)
1120	{
1121	const char pszStr = (const char )pvData;
1122	size_t cbStr = cbData;
1123
1124	if (RT_FAILURE(RTStrValidateEncodingEx(pszStr, cbStr,
1125	RTSTR_VALIDATE_ENCODING_ZERO_TERMINATED \| RTSTR_VALIDATE_ENCODING_EXACT_LENGTH)))
1126	{
1127	AssertFailed();
1128	return "";
1129	}
1130
1131	return Utf8Str(pszStr, cbStr);
1132	}
1133
1134	protected:
1135
1136	int copyFrom(uint32_t uTypePayload, const void *pvPayload, uint32_t cbPayload)
1137	{
1138	if (cbPayload > _64K) /* Paranoia. */
1139	return VERR_TOO_MUCH_DATA;
1140
1141	Clear();
1142
1143	int rc = VINF_SUCCESS;
1144
1145	if (cbPayload)
1146	{
1147	pvData = RTMemAlloc(cbPayload);
1148	if (pvData)
1149	{
1150	uType = uTypePayload;
1151
1152	memcpy(pvData, pvPayload, cbPayload);
1153	cbData = cbPayload;
1154	}
1155	else
1156	rc = VERR_NO_MEMORY;
1157	}
1158	else
1159	{
1160	uType = uTypePayload;
1161
1162	pvData = NULL;
1163	cbData = 0;
1164	}
1165
1166	return rc;
1167	}
1168
1169	protected:
1170
1171	/** Type of payload. */
1172	uint32_t uType;
1173	/** Size (in bytes) of payload. */
1174	uint32_t cbData;
1175	/** Pointer to actual payload data. */
1176	void *pvData;
1177	};
1178
1179	class GuestWaitEventBase
1180	{
1181
1182	protected:
1183
1184	GuestWaitEventBase(void);
1185	virtual ~GuestWaitEventBase(void);
1186
1187	public:
1188
1189	uint32_t ContextID(void) { return mCID; };
1190	int GuestResult(void) { return mGuestRc; }
1191	int Result(void) { return mRc; }
1192	GuestWaitEventPayload & Payload(void) { return mPayload; }
1193	int SignalInternal(int rc, int guestRc, const GuestWaitEventPayload *pPayload);
1194	int Wait(RTMSINTERVAL uTimeoutMS);
1195
1196	protected:
1197
1198	int Init(uint32_t uCID);
1199
1200	protected:
1201
1202	/* Shutdown indicator. */
1203	bool mfAborted;
1204	/* Associated context ID (CID). */
1205	uint32_t mCID;
1206	/** The event semaphore for triggering
1207	* the actual event. */
1208	RTSEMEVENT mEventSem;
1209	/** The event's overall result. If
1210	* set to VERR_GSTCTL_GUEST_ERROR,
1211	* mGuestRc will contain the actual
1212	* error code from the guest side. */
1213	int mRc;
1214	/** The event'S overall result from the
1215	* guest side. If used, mRc must be
1216	* set to VERR_GSTCTL_GUEST_ERROR. */
1217	int mGuestRc;
1218	/** The event's payload data. Optional. */
1219	GuestWaitEventPayload mPayload;
1220	};
1221
1222	/** List of public guest event types. */
1223	typedef std::list < VBoxEventType_T > GuestEventTypes;
1224
1225	class GuestWaitEvent : public GuestWaitEventBase
1226	{
1227
1228	public:
1229
1230	GuestWaitEvent(void);
1231	virtual ~GuestWaitEvent(void);
1232
1233	public:
1234
1235	int Init(uint32_t uCID);
1236	int Init(uint32_t uCID, const GuestEventTypes &lstEvents);
1237	int Cancel(void);
1238	const ComPtr<IEvent> Event(void) { return mEvent; }
1239	bool HasGuestError(void) const { return mRc == VERR_GSTCTL_GUEST_ERROR; }
1240	int GetGuestError(void) const { return mGuestRc; }
1241	int SignalExternal(IEvent *pEvent);
1242	const GuestEventTypes &Types(void) { return mEventTypes; }
1243	size_t TypeCount(void) { return mEventTypes.size(); }
1244
1245	protected:
1246
1247	/** List of public event types this event should
1248	* be signalled on. Optional. */
1249	GuestEventTypes mEventTypes;
1250	/** Pointer to the actual public event, if any. */
1251	ComPtr<IEvent> mEvent;
1252	};
1253	/** Map of pointers to guest events. The primary key
1254	* contains the context ID. */
1255	typedef std::map < uint32_t, GuestWaitEvent* > GuestWaitEvents;
1256	/** Map of wait events per public guest event. Nice for
1257	* faster lookups when signalling a whole event group. */
1258	typedef std::map < VBoxEventType_T, GuestWaitEvents > GuestEventGroup;
1259
1260	class GuestBase
1261	{
1262
1263	public:
1264
1265	GuestBase(void);
1266	virtual ~GuestBase(void);
1267
1268	public:
1269
1270	/** Signals a wait event using a public guest event; also used for
1271	* for external event listeners. */
1272	int signalWaitEvent(VBoxEventType_T aType, IEvent *aEvent);
1273	/** Signals a wait event using a guest rc. */
1274	int signalWaitEventInternal(PVBOXGUESTCTRLHOSTCBCTX pCbCtx, int guestRc, const GuestWaitEventPayload *pPayload);
1275	/** Signals a wait event without letting public guest events know,
1276	* extended director's cut version. */
1277	int signalWaitEventInternalEx(PVBOXGUESTCTRLHOSTCBCTX pCbCtx, int rc, int guestRc, const GuestWaitEventPayload *pPayload);
1278
1279	public:
1280
1281	int baseInit(void);
1282	void baseUninit(void);
1283	int cancelWaitEvents(void);
1284	int dispatchGeneric(PVBOXGUESTCTRLHOSTCBCTX pCtxCb, PVBOXGUESTCTRLHOSTCALLBACK pSvcCb);
1285	int generateContextID(uint32_t uSessionID, uint32_t uObjectID, uint32_t *puContextID);
1286	int registerWaitEvent(uint32_t uSessionID, uint32_t uObjectID, GuestWaitEvent **ppEvent);
1287	int registerWaitEventEx(uint32_t uSessionID, uint32_t uObjectID, const GuestEventTypes &lstEvents, GuestWaitEvent **ppEvent);
1288	int unregisterWaitEvent(GuestWaitEvent *pEvent);
1289	int waitForEvent(GuestWaitEvent pEvent, uint32_t uTimeoutMS, VBoxEventType_T pType, IEvent **ppEvent);
1290
1291	public:
1292
1293	static FsObjType_T fileModeToFsObjType(RTFMODE fMode);
1294	static const char *fsObjTypeToStr(FsObjType_T enmType);
1295	static const char *pathStyleToStr(PathStyle_T enmPathStyle);
1296	static Utf8Str getErrorAsString(const Utf8Str &strAction, const GuestErrorInfo& guestErrorInfo);
1297	static Utf8Str getErrorAsString(const GuestErrorInfo &guestErrorInfo);
1298
1299	protected:
1300
1301	/** Pointer to the console object. Needed
1302	* for HGCM (VMMDev) communication. */
1303	Console *mConsole;
1304	/** The next context ID counter component for this object. */
1305	uint32_t mNextContextID;
1306	/** Local listener for handling the waiting events
1307	* internally. */
1308	ComPtr<IEventListener> mLocalListener;
1309	/** Critical section for wait events access. */
1310	RTCRITSECT mWaitEventCritSect;
1311	/** Map of registered wait events per event group. */
1312	GuestEventGroup mWaitEventGroups;
1313	/** Map of registered wait events. */
1314	GuestWaitEvents mWaitEvents;
1315	};
1316
1317	/**
1318	* Virtual class (interface) for guest objects (processes, files, ...) --
1319	* contains all per-object callback management.
1320	*/
1321	class GuestObject : public GuestBase
1322	{
1323	friend class GuestSession;
1324
1325	public:
1326
1327	GuestObject(void);
1328	virtual ~GuestObject(void);
1329
1330	public:
1331
1332	ULONG getObjectID(void) { return mObjectID; }
1333
1334	protected:
1335
1336	/**
1337	* Called by IGuestSession when the session status has been changed.
1338	*
1339	* @returns VBox status code.
1340	* @param enmSessionStatus New session status.
1341	*/
1342	virtual int i_onSessionStatusChange(GuestSessionStatus_T enmSessionStatus) = 0;
1343
1344	/**
1345	* Called by IGuestSession right before this object gets
1346	* unregistered (removed) from the public object list.
1347	*/
1348	virtual int i_onUnregister(void) = 0;
1349
1350	/** Callback dispatcher -- must be implemented by the actual object. */
1351	virtual int i_callbackDispatcher(PVBOXGUESTCTRLHOSTCBCTX pCbCtx, PVBOXGUESTCTRLHOSTCALLBACK pSvcCb) = 0;
1352
1353	protected:
1354
1355	int bindToSession(Console pConsole, GuestSession pSession, uint32_t uObjectID);
1356	int registerWaitEvent(const GuestEventTypes &lstEvents, GuestWaitEvent **ppEvent);
1357	int sendMessage(uint32_t uFunction, uint32_t cParms, PVBOXHGCMSVCPARM paParms);
1358
1359	protected:
1360
1361	/** @name Common parameters for all derived objects. They have their own
1362	* mData structure to keep their specific data around.
1363	* @{ */
1364	/** Pointer to parent session. Per definition
1365	* this objects always lives shorter than the
1366	* parent.
1367	* @todo r=bird: When wanting to use mSession in the
1368	* IGuestProcess::getEnvironment() implementation I wanted to access
1369	* GuestSession::mData::mpBaseEnvironment. Seeing the comment in
1370	* GuestProcess::terminate() saying:
1371	* "Now only API clients still can hold references to it."
1372	* and recalling seeing similar things in VirtualBox.xidl or some such place,
1373	* I'm wondering how this "per definition" behavior is enforced. Is there any
1374	* GuestProcess:uninit() call or similar magic that invalidates objects that
1375	* GuestSession loses track of in place like GuestProcess::terminate() that I've
1376	* failed to spot?
1377	*
1378	* Please enlighten me.
1379	*/
1380	GuestSession *mSession;
1381	/** The object ID -- must be unique for each guest
1382	* object and is encoded into the context ID. Must
1383	* be set manually when initializing the object.
1384	*
1385	* For guest processes this is the internal PID,
1386	* for guest files this is the internal file ID. */
1387	uint32_t mObjectID;
1388	/** @} */
1389	};
1390
1391	/** Returns the path separator based on \a a_enmPathStyle as a C-string. */
1392	#define PATH_STYLE_SEP_STR(a_enmPathStyle) a_enmPathStyle == PathStyle_DOS ? "\\" : "/"
1393	#if defined(RT_OS_WINDOWS) \|\| defined(RT_OS_OS2)
1394	# define PATH_STYLE_NATIVE PathStyle_DOS
1395	#else
1396	# define PATH_STYLE_NATIVE PathStyle_UNIX
1397	#endif
1398
1399	/**
1400	* Class for handling guest / host path functions.
1401	*/
1402	class GuestPath
1403	{
1404	private:
1405
1406	/**
1407	* Default constructor.
1408	*
1409	* Not directly instantiable (yet).
1410	*/
1411	GuestPath(void) { }
1412
1413	public:
1414
1415	/** @name Static helper functions.
1416	* @{ */
1417	static int BuildDestinationPath(const Utf8Str &strSrcPath, PathStyle_T enmSrcPathStyle, Utf8Str &strDstPath, PathStyle_T enmDstPathStyle);
1418	static int Translate(Utf8Str &strPath, PathStyle_T enmSrcPathStyle, PathStyle_T enmDstPathStyle, bool fForce = false);
1419	/** @} */
1420	};
1421	#endif /* !MAIN_INCLUDED_GuestCtrlImplPrivate_h */
1422

Note: See TracBrowser for help on using the repository browser.

source: vbox/trunk/src/VBox/Main/include/GuestCtrlImplPrivate.h@ 98123

Download in other formats: