int GetVBoxUserHomeDirectory(char *aDir, size_t aDirLen)
{
AssertReturn(aDir, VERR_INVALID_POINTER);
AssertReturn(aDirLen > 0, VERR_BUFFER_OVERFLOW);
/* start with null */
*aDir = 0;
char szTmp[RTPATH_MAX];
int vrc = RTEnvGetEx(RTENV_DEFAULT, "VBOX_USER_HOME", szTmp, sizeof(szTmp), NULL);
if (RT_SUCCESS(vrc) || vrc == VERR_ENV_VAR_NOT_FOUND)
{
if (RT_SUCCESS(vrc))
{
/* get the full path name */
vrc = RTPathAbs(szTmp, aDir, aDirLen);
}
else
{
/* compose the config directory (full path) */
/** @todo r=bird: RTPathUserHome doesn't necessarily return a full (abs) path
* like the comment above seems to indicate. */
vrc = RTPathUserHome(aDir, aDirLen);
if (RT_SUCCESS(vrc))
vrc = RTPathAppend(aDir, aDirLen, VBOX_USER_HOME_SUFFIX);
}
/* ensure the home directory exists */
if (RT_SUCCESS(vrc))
if (!RTDirExists(aDir))
vrc = RTDirCreateFullPath(aDir, 0700);
}
return vrc;
}
/**
* Returns the user's home directory. Wrapper around RTPathUserHome().
* @param strPath
* @return
*/
HRESULT SystemProperties::getUserHomeDirectory(Utf8Str &strPath)
{
char szHome[RTPATH_MAX];
int vrc = RTPathUserHome(szHome, sizeof(szHome));
if (RT_FAILURE(vrc))
return setError(E_FAIL,
tr("Cannot determine user home directory (%Rrc)"),
vrc);
strPath = szHome;
return S_OK;
}
RTDECL(int) RTPathUserDocuments(char *pszPath, size_t cchPath)
{
/*
* Validate input
*/
AssertPtrReturn(pszPath, VERR_INVALID_POINTER);
AssertReturn(cchPath, VERR_INVALID_PARAMETER);
/*
* Try NSSystemDirectories first since that works for directories that doesn't exist.
*/
int rc = VERR_PATH_NOT_FOUND;
NSSearchPathEnumerationState EnmState = NSStartSearchPathEnumeration(NSDocumentDirectory, NSUserDomainMask);
if (EnmState != 0)
{
char szTmp[PATH_MAX];
szTmp[0] = szTmp[PATH_MAX - 1] = '\0';
EnmState = NSGetNextSearchPathEnumeration(EnmState, szTmp);
if (EnmState != 0)
{
size_t cchTmp = strlen(szTmp);
if (cchTmp >= cchPath)
return VERR_BUFFER_OVERFLOW;
if (szTmp[0] == '~' && szTmp[1] == '/')
{
/* Expand tilde. */
rc = RTPathUserHome(pszPath, cchPath - cchTmp + 2);
if (RT_FAILURE(rc))
return rc;
rc = RTPathAppend(pszPath, cchPath, &szTmp[2]);
}
else
rc = RTStrCopy(pszPath, cchPath, szTmp);
return rc;
}
}
#ifdef IPRT_USE_CORE_SERVICE_FOR_USER_DOCUMENTS
/*
* Fall back on FSFindFolder in case the above should fail...
*/
FSRef ref;
OSErr err = FSFindFolder(kOnAppropriateDisk, kDocumentsFolderType, false /* createFolder */, &ref);
if (err == noErr)
{
err = FSRefMakePath(&ref, (UInt8*)pszPath, cchPath);
if (err == noErr)
return VINF_SUCCESS;
}
#endif
Assert(RT_FAILURE_NP(rc));
return rc;
}
HRESULT SystemProperties::i_setDefaultAdditionsISO(const com::Utf8Str &aPath)
{
com::Utf8Str path(aPath);
if (path.isEmpty())
{
char strTemp[RTPATH_MAX];
int vrc = RTPathAppPrivateNoArch(strTemp, sizeof(strTemp));
AssertRC(vrc);
Utf8Str strSrc1 = Utf8Str(strTemp).append("/VBoxGuestAdditions.iso");
vrc = RTPathExecDir(strTemp, sizeof(strTemp));
AssertRC(vrc);
Utf8Str strSrc2 = Utf8Str(strTemp).append("/additions/VBoxGuestAdditions.iso");
vrc = RTPathUserHome(strTemp, sizeof(strTemp));
AssertRC(vrc);
Utf8Str strSrc3 = Utf8StrFmt("%s/VBoxGuestAdditions_%s.iso", strTemp, VirtualBox::i_getVersionNormalized().c_str());
/* Check the standard image locations */
if (RTFileExists(strSrc1.c_str()))
path = strSrc1;
else if (RTFileExists(strSrc2.c_str()))
path = strSrc2;
else if (RTFileExists(strSrc3.c_str()))
path = strSrc3;
else
return setError(E_FAIL,
tr("Cannot determine default Guest Additions ISO location. Most likely they are not available"));
}
if (!RTPathStartsWithRoot(path.c_str()))
return setError(E_INVALIDARG,
tr("Given default machine Guest Additions ISO file '%s' is not fully qualified"),
path.c_str());
if (!RTFileExists(path.c_str()))
return setError(E_INVALIDARG,
tr("Given default machine Guest Additions ISO file '%s' does not exist"),
path.c_str());
m->strDefaultAdditionsISO = path;
return S_OK;
}
/**
* Reads the CFGM configuration of the DBGC.
*
* Popuplates the PDBGC::pszHistoryFile, PDBGC::pszGlobalInitScript and
* PDBGC::pszLocalInitScript members.
*
* @returns VBox status code.
* @param pDbgc The console instance.
* @param pUVM The user mode VM handle.
*/
static int dbgcReadConfig(PDBGC pDbgc, PUVM pUVM)
{
/*
* Get and validate the configuration node.
*/
PCFGMNODE pNode = CFGMR3GetChild(CFGMR3GetRootU(pUVM), "DBGC");
int rc = CFGMR3ValidateConfig(pNode, "/DBGC/",
"Enabled|"
"HistoryFile|"
"LocalInitScript|"
"GlobalInitScript",
"", "DBGC", 0);
AssertRCReturn(rc, rc);
/*
* Query the values.
*/
char szHomeDefault[RTPATH_MAX];
rc = RTPathUserHome(szHomeDefault, sizeof(szHomeDefault) - 32);
AssertLogRelRCReturn(rc, rc);
size_t cchHome = strlen(szHomeDefault);
/** @cfgm{/DBGC/HistoryFile, string, ${HOME}/.vboxdbgc-history}
* The command history file of the VBox debugger. */
rc = RTPathAppend(szHomeDefault, sizeof(szHomeDefault), ".vboxdbgc-history");
AssertLogRelRCReturn(rc, rc);
char szPath[RTPATH_MAX];
rc = CFGMR3QueryStringDef(pNode, "HistoryFile", szPath, sizeof(szPath), szHomeDefault);
AssertLogRelRCReturn(rc, rc);
pDbgc->pszHistoryFile = RTStrDup(szPath);
AssertReturn(pDbgc->pszHistoryFile, VERR_NO_STR_MEMORY);
/** @cfgm{/DBGC/GlobalInitFile, string, ${HOME}/.vboxdbgc-init}
* The global init script of the VBox debugger. */
szHomeDefault[cchHome] = '\0';
rc = RTPathAppend(szHomeDefault, sizeof(szHomeDefault), ".vboxdbgc-init");
AssertLogRelRCReturn(rc, rc);
rc = CFGMR3QueryStringDef(pNode, "GlobalInitScript", szPath, sizeof(szPath), szHomeDefault);
AssertLogRelRCReturn(rc, rc);
pDbgc->pszGlobalInitScript = RTStrDup(szPath);
AssertReturn(pDbgc->pszGlobalInitScript, VERR_NO_STR_MEMORY);
/** @cfgm{/DBGC/LocalInitFile, string, none}
* The VM local init script of the VBox debugger. */
rc = CFGMR3QueryString(pNode, "LocalInitScript", szPath, sizeof(szPath));
if (RT_SUCCESS(rc))
{
pDbgc->pszLocalInitScript = RTStrDup(szPath);
AssertReturn(pDbgc->pszLocalInitScript, VERR_NO_STR_MEMORY);
}
else
{
AssertLogRelReturn(rc == VERR_CFGM_VALUE_NOT_FOUND || rc == VERR_CFGM_NO_PARENT, rc);
pDbgc->pszLocalInitScript = NULL;
}
return VINF_SUCCESS;
}
int main()
{
char szPath[RTPATH_MAX];
/*
* Init RT+Test.
*/
RTTEST hTest;
int rc = RTTestInitAndCreate("tstRTPath", &hTest);
if (rc)
return rc;
RTTestBanner(hTest);
/*
* RTPathExecDir, RTPathUserHome and RTProcGetExecutablePath.
*/
RTTestSub(hTest, "RTPathExecDir");
RTTESTI_CHECK_RC(rc = RTPathExecDir(szPath, sizeof(szPath)), VINF_SUCCESS);
if (RT_SUCCESS(rc))
RTTestIPrintf(RTTESTLVL_INFO, "ExecDir={%s}\n", szPath);
RTTestSub(hTest, "RTProcGetExecutablePath");
if (RTProcGetExecutablePath(szPath, sizeof(szPath)) == szPath)
RTTestIPrintf(RTTESTLVL_INFO, "ExecutableName={%s}\n", szPath);
else
RTTestIFailed("RTProcGetExecutablePath -> NULL");
RTTestSub(hTest, "RTPathUserHome");
RTTESTI_CHECK_RC(rc = RTPathUserHome(szPath, sizeof(szPath)), VINF_SUCCESS);
if (RT_SUCCESS(rc))
RTTestIPrintf(RTTESTLVL_INFO, "UserHome={%s}\n", szPath);
RTTestSub(hTest, "RTPathUserDocuments");
RTTESTI_CHECK_RC(rc = RTPathUserDocuments(szPath, sizeof(szPath)), VINF_SUCCESS);
if (RT_SUCCESS(rc))
RTTestIPrintf(RTTESTLVL_INFO, "UserDocuments={%s}\n", szPath);
RTTestSub(hTest, "RTPathTemp");
RTTESTI_CHECK_RC(rc = RTPathTemp(szPath, sizeof(szPath)), VINF_SUCCESS);
if (RT_SUCCESS(rc))
RTTestIPrintf(RTTESTLVL_INFO, "PathTemp={%s}\n", szPath);
size_t cch = strlen(szPath);
RTTESTI_CHECK_RC(RTPathTemp(szPath, cch), VERR_BUFFER_OVERFLOW);
RTTESTI_CHECK_RC(RTPathTemp(szPath, cch+1), VINF_SUCCESS);
RTTESTI_CHECK_RC(RTPathTemp(szPath, cch+2), VINF_SUCCESS);
/*
* RTPathAbsEx
*/
RTTestSub(hTest, "RTPathAbsEx");
static const struct
{
const char *pcszInputBase;
const char *pcszInputPath;
int rc;
const char *pcszOutput;
}
s_aRTPathAbsExTests[] =
{
#if defined (RT_OS_OS2) || defined (RT_OS_WINDOWS)
{ NULL, "", VERR_INVALID_PARAMETER, NULL },
{ NULL, ".", VINF_SUCCESS, "%p" },
{ NULL, "\\", VINF_SUCCESS, "%d\\" },
{ NULL, "\\..", VINF_SUCCESS, "%d\\" },
{ NULL, "/absolute/..", VINF_SUCCESS, "%d\\" },
{ NULL, "/absolute\\\\../..", VINF_SUCCESS, "%d\\" },
{ NULL, "/absolute//../path\\", VINF_SUCCESS, "%d\\path" },
{ NULL, "/absolute/../../path", VINF_SUCCESS, "%d\\path" },
{ NULL, "relative/../dir\\.\\.\\.\\file.txt", VINF_SUCCESS, "%p\\dir\\file.txt" },
{ NULL, "\\data\\", VINF_SUCCESS, "%d\\data" },
{ "relative_base/dir\\", "\\from_root", VINF_SUCCESS, "%d\\from_root" },
{ "relative_base/dir/", "relative_also", VINF_SUCCESS, "%p\\relative_base\\dir\\relative_also" },
#else
{ NULL, "", VERR_INVALID_PARAMETER, NULL },
{ NULL, ".", VINF_SUCCESS, "%p" },
{ NULL, "/", VINF_SUCCESS, "/" },
{ NULL, "/..", VINF_SUCCESS, "/" },
{ NULL, "/absolute/..", VINF_SUCCESS, "/" },
{ NULL, "/absolute\\\\../..", VINF_SUCCESS, "/" },
{ NULL, "/absolute//../path/", VINF_SUCCESS, "/path" },
{ NULL, "/absolute/../../path", VINF_SUCCESS, "/path" },
{ NULL, "relative/../dir/./././file.txt", VINF_SUCCESS, "%p/dir/file.txt" },
{ NULL, "relative/../dir\\.\\.\\.\\file.txt", VINF_SUCCESS, "%p/dir\\.\\.\\.\\file.txt" }, /* linux-specific */
{ NULL, "/data/", VINF_SUCCESS, "/data" },
{ "relative_base/dir/", "/from_root", VINF_SUCCESS, "/from_root" },
{ "relative_base/dir/", "relative_also", VINF_SUCCESS, "%p/relative_base/dir/relative_also" },
#endif
#if defined (RT_OS_OS2) || defined (RT_OS_WINDOWS)
{ NULL, "C:\\", VINF_SUCCESS, "C:\\" },
{ "C:\\", "..", VINF_SUCCESS, "C:\\" },
{ "C:\\temp", "..", VINF_SUCCESS, "C:\\" },
{ "C:\\VirtualBox/Machines", "..\\VirtualBox.xml", VINF_SUCCESS, "C:\\VirtualBox\\VirtualBox.xml" },
{ "C:\\MustDie", "\\from_root/dir/..", VINF_SUCCESS, "C:\\from_root" },
{ "C:\\temp", "D:\\data", VINF_SUCCESS, "D:\\data" },
{ NULL, "\\\\server\\..\\share", VINF_SUCCESS, "\\\\server\\..\\share" /* kind of strange */ },
{ NULL, "\\\\server/", VINF_SUCCESS, "\\\\server" },
{ NULL, "\\\\", VINF_SUCCESS, "\\\\" },
{ NULL, "\\\\\\something", VINF_SUCCESS, "\\\\\\something" /* kind of strange */ },
{ "\\\\server\\share_as_base", "/from_root", VINF_SUCCESS, "\\\\server\\from_root" },
{ "\\\\just_server", "/from_root", VINF_SUCCESS, "\\\\just_server\\from_root" },
{ "\\\\server\\share_as_base", "relative\\data", VINF_SUCCESS, "\\\\server\\share_as_base\\relative\\data" },
{ "base", "\\\\?\\UNC\\relative/edwef/..", VINF_SUCCESS, "\\\\?\\UNC\\relative" },
{ "\\\\?\\UNC\\base", "/from_root", VERR_INVALID_NAME, NULL },
#else
{ "/temp", "..", VINF_SUCCESS, "/" },
{ "/VirtualBox/Machines", "../VirtualBox.xml", VINF_SUCCESS, "/VirtualBox/VirtualBox.xml" },
{ "/MustDie", "/from_root/dir/..", VINF_SUCCESS, "/from_root" },
{ "\\temp", "\\data", VINF_SUCCESS, "%p/\\temp/\\data" },
#endif
};
for (unsigned i = 0; i < RT_ELEMENTS(s_aRTPathAbsExTests); ++ i)
{
rc = RTPathAbsEx(s_aRTPathAbsExTests[i].pcszInputBase,
s_aRTPathAbsExTests[i].pcszInputPath,
szPath, sizeof(szPath));
if (rc != s_aRTPathAbsExTests[i].rc)
{
RTTestIFailed("unexpected result code!\n"
" input base: '%s'\n"
" input path: '%s'\n"
" output: '%s'\n"
" rc: %Rrc\n"
" expected rc: %Rrc",
s_aRTPathAbsExTests[i].pcszInputBase,
s_aRTPathAbsExTests[i].pcszInputPath,
szPath, rc,
s_aRTPathAbsExTests[i].rc);
continue;
}
char szTmp[RTPATH_MAX];
char *pszExpected = NULL;
if (s_aRTPathAbsExTests[i].pcszOutput != NULL)
{
if (s_aRTPathAbsExTests[i].pcszOutput[0] == '%')
{
RTTESTI_CHECK_RC(rc = RTPathGetCurrent(szTmp, sizeof(szTmp)), VINF_SUCCESS);
if (RT_FAILURE(rc))
break;
pszExpected = szTmp;
if (s_aRTPathAbsExTests[i].pcszOutput[1] == 'p')
{
cch = strlen(szTmp);
if (cch + strlen(s_aRTPathAbsExTests[i].pcszOutput) - 2 <= sizeof(szTmp))
strcpy(szTmp + cch, s_aRTPathAbsExTests[i].pcszOutput + 2);
}
#if defined(RT_OS_OS2) || defined(RT_OS_WINDOWS)
else if (s_aRTPathAbsExTests[i].pcszOutput[1] == 'd')
{
if (2 + strlen(s_aRTPathAbsExTests[i].pcszOutput) - 2 <= sizeof(szTmp))
strcpy(szTmp + 2, s_aRTPathAbsExTests[i].pcszOutput + 2);
}
#endif
}
else
{
strcpy(szTmp, s_aRTPathAbsExTests[i].pcszOutput);
pszExpected = szTmp;
}
if (strcmp(szPath, pszExpected))
{
RTTestIFailed("Unexpected result\n"
" input base: '%s'\n"
" input path: '%s'\n"
" output: '%s'\n"
" expected: '%s'",
s_aRTPathAbsExTests[i].pcszInputBase,
s_aRTPathAbsExTests[i].pcszInputPath,
szPath,
s_aRTPathAbsExTests[i].pcszOutput);
}
}
}
/*
* RTPathStripFilename
*/
RTTestSub(hTest, "RTPathStripFilename");
static const char *s_apszStripFilenameTests[] =
{
"/usr/include///", "/usr/include//",
"/usr/include/", "/usr/include",
"/usr/include", "/usr",
"/usr", "/",
"usr", ".",
#if defined (RT_OS_OS2) || defined (RT_OS_WINDOWS)
"c:/windows", "c:/",
"c:/", "c:/",
"D:", "D:",
"C:\\OS2\\DLLS", "C:\\OS2",
#endif
};
for (unsigned i = 0; i < RT_ELEMENTS(s_apszStripFilenameTests); i += 2)
{
const char *pszInput = s_apszStripFilenameTests[i];
const char *pszExpect = s_apszStripFilenameTests[i + 1];
strcpy(szPath, pszInput);
RTPathStripFilename(szPath);
if (strcmp(szPath, pszExpect))
{
RTTestIFailed("Unexpected result\n"
" input: '%s'\n"
" output: '%s'\n"
"expected: '%s'",
pszInput, szPath, pszExpect);
}
}
/*
* RTPathAppend.
*/
RTTestSub(hTest, "RTPathAppend");
static const char *s_apszAppendTests[] =
{
/* base append result */
"/", "", "/",
"", "/", "/",
"/", "/", "/",
"/x", "", "/x",
"/x", "/", "/x/",
"/", "x", "/x",
"dir", "file", "dir/file",
"dir", "/file", "dir/file",
"dir", "//file", "dir/file",
"dir", "///file", "dir/file",
"dir/", "/file", "dir/file",
"dir/", "//file", "dir/file",
"dir/", "///file", "dir/file",
"dir//", "file", "dir/file",
"dir//", "/file", "dir/file",
"dir//", "//file", "dir/file",
"dir///", "///file", "dir/file",
"/bin/testcase", "foo.r0", "/bin/testcase/foo.r0",
#if defined (RT_OS_OS2) || defined (RT_OS_WINDOWS)
"/", "\\", "/",
"\\", "/", "\\",
"\\\\srv\\shr", "dir//", "\\\\srv\\shr/dir//",
"\\\\srv\\shr", "dir//file", "\\\\srv\\shr/dir//file",
"\\\\srv\\shr", "//dir//", "\\\\srv\\shr/dir//",
"\\\\srv\\shr", "/\\dir//", "\\\\srv\\shr\\dir//",
"\\\\", "not-srv/not-shr/file", "\\not-srv/not-shr/file",
"C:", "autoexec.bat", "C:autoexec.bat",
"C:", "/autoexec.bat", "C:/autoexec.bat",
"C:", "\\autoexec.bat", "C:\\autoexec.bat",
"C:\\", "/autoexec.bat", "C:\\autoexec.bat",
"C:\\\\", "autoexec.bat", "C:\\autoexec.bat",
"E:\\bin\\testcase", "foo.r0", "E:\\bin\\testcase/foo.r0",
#endif
};
for (unsigned i = 0; i < RT_ELEMENTS(s_apszAppendTests); i += 3)
{
const char *pszInput = s_apszAppendTests[i];
const char *pszAppend = s_apszAppendTests[i + 1];
const char *pszExpect = s_apszAppendTests[i + 2];
strcpy(szPath, pszInput);
RTTESTI_CHECK_RC(rc = RTPathAppend(szPath, sizeof(szPath), pszAppend), VINF_SUCCESS);
if (RT_FAILURE(rc))
continue;
if (strcmp(szPath, pszExpect))
{
RTTestIFailed("Unexpected result\n"
" input: '%s'\n"
" append: '%s'\n"
" output: '%s'\n"
"expected: '%s'",
pszInput, pszAppend, szPath, pszExpect);
}
else
{
size_t const cchResult = strlen(szPath);
strcpy(szPath, pszInput);
RTTESTI_CHECK_RC(rc = RTPathAppend(szPath, cchResult + 2, pszAppend), VINF_SUCCESS);
RTTESTI_CHECK(RT_FAILURE(rc) || !strcmp(szPath, pszExpect));
strcpy(szPath, pszInput);
RTTESTI_CHECK_RC(rc = RTPathAppend(szPath, cchResult + 1, pszAppend), VINF_SUCCESS);
RTTESTI_CHECK(RT_FAILURE(rc) || !strcmp(szPath, pszExpect));
if (strlen(pszInput) < cchResult)
{
strcpy(szPath, pszInput);
RTTESTI_CHECK_RC(RTPathAppend(szPath, cchResult, pszAppend), VERR_BUFFER_OVERFLOW);
}
}
}
/*
* RTPathJoin - reuse the append tests.
*/
RTTestSub(hTest, "RTPathJoin");
for (unsigned i = 0; i < RT_ELEMENTS(s_apszAppendTests); i += 3)
{
const char *pszInput = s_apszAppendTests[i];
const char *pszAppend = s_apszAppendTests[i + 1];
const char *pszExpect = s_apszAppendTests[i + 2];
memset(szPath, 'a', sizeof(szPath)); szPath[sizeof(szPath) - 1] = '\0';
RTTESTI_CHECK_RC(rc = RTPathJoin(szPath, sizeof(szPath), pszInput, pszAppend), VINF_SUCCESS);
if (RT_FAILURE(rc))
continue;
if (strcmp(szPath, pszExpect))
{
RTTestIFailed("Unexpected result\n"
" input: '%s'\n"
" append: '%s'\n"
" output: '%s'\n"
"expected: '%s'",
pszInput, pszAppend, szPath, pszExpect);
}
else
{
size_t const cchResult = strlen(szPath);
memset(szPath, 'a', sizeof(szPath)); szPath[sizeof(szPath) - 1] = '\0';
RTTESTI_CHECK_RC(rc = RTPathJoin(szPath, cchResult + 2, pszInput, pszAppend), VINF_SUCCESS);
RTTESTI_CHECK(RT_FAILURE(rc) || !strcmp(szPath, pszExpect));
memset(szPath, 'a', sizeof(szPath)); szPath[sizeof(szPath) - 1] = '\0';
RTTESTI_CHECK_RC(rc = RTPathJoin(szPath, cchResult + 1, pszInput, pszAppend), VINF_SUCCESS);
RTTESTI_CHECK(RT_FAILURE(rc) || !strcmp(szPath, pszExpect));
RTTESTI_CHECK_RC(rc = RTPathJoin(szPath, cchResult, pszInput, pszAppend), VERR_BUFFER_OVERFLOW);
}
}
/*
* RTPathJoinA - reuse the append tests.
*/
RTTestSub(hTest, "RTPathJoinA");
for (unsigned i = 0; i < RT_ELEMENTS(s_apszAppendTests); i += 3)
{
const char *pszInput = s_apszAppendTests[i];
const char *pszAppend = s_apszAppendTests[i + 1];
const char *pszExpect = s_apszAppendTests[i + 2];
char *pszPathDst;
RTTESTI_CHECK(pszPathDst = RTPathJoinA(pszInput, pszAppend));
if (!pszPathDst)
continue;
if (strcmp(pszPathDst, pszExpect))
{
RTTestIFailed("Unexpected result\n"
" input: '%s'\n"
" append: '%s'\n"
" output: '%s'\n"
"expected: '%s'",
pszInput, pszAppend, pszPathDst, pszExpect);
}
RTStrFree(pszPathDst);
}
/*
* RTPathStripTrailingSlash
*/
static const char *s_apszStripTrailingSlash[] =
{
/* input result */
"/", "/",
"//", "/",
"////////////////////", "/",
"/tmp", "/tmp",
"/tmp////////////////", "/tmp",
"tmp", "tmp",
"tmp////////////////", "tmp",
"./", ".",
#if defined (RT_OS_OS2) || defined (RT_OS_WINDOWS)
"////////////////////", "/",
"D:", "D:",
"D:/", "D:/",
"D:\\", "D:\\",
"D:\\/\\", "D:\\",
"D:/\\/\\", "D:/",
"C:/Temp", "D:/Temp",
"C:/Temp/", "D:/Temp/",
"C:/Temp\\/", "D:/Temp",
#endif
};
for (unsigned i = 0; i < RT_ELEMENTS(s_apszStripTrailingSlash); i += 2)
{
const char *pszInput = s_apszStripTrailingSlash[i];
const char *pszExpect = s_apszStripTrailingSlash[i + 1];
strcpy(szPath, pszInput);
cch = RTPathStripTrailingSlash(szPath);
if (strcmp(szPath, pszExpect))
RTTestIFailed("Unexpected result\n"
" input: '%s'\n"
" output: '%s'\n"
"expected: '%s'",
pszInput, szPath, pszExpect);
else
RTTESTI_CHECK(cch == strlen(szPath));
}
/*
* RTPathCountComponents
*/
RTTestSub(hTest, "RTPathCountComponents");
RTTESTI_CHECK(RTPathCountComponents("") == 0);
RTTESTI_CHECK(RTPathCountComponents("/") == 1);
RTTESTI_CHECK(RTPathCountComponents("//") == 1);
RTTESTI_CHECK(RTPathCountComponents("//////////////") == 1);
RTTESTI_CHECK(RTPathCountComponents("//////////////bin") == 2);
RTTESTI_CHECK(RTPathCountComponents("//////////////bin/") == 2);
RTTESTI_CHECK(RTPathCountComponents("//////////////bin/////") == 2);
RTTESTI_CHECK(RTPathCountComponents("..") == 1);
RTTESTI_CHECK(RTPathCountComponents("../") == 1);
RTTESTI_CHECK(RTPathCountComponents("../..") == 2);
RTTESTI_CHECK(RTPathCountComponents("../../") == 2);
#if defined (RT_OS_OS2) || defined (RT_OS_WINDOWS)
RTTESTI_CHECK(RTPathCountComponents("d:") == 1);
RTTESTI_CHECK(RTPathCountComponents("d:/") == 1);
RTTESTI_CHECK(RTPathCountComponents("d:/\\") == 1);
RTTESTI_CHECK(RTPathCountComponents("d:\\") == 1);
RTTESTI_CHECK(RTPathCountComponents("c:\\config.sys") == 2);
RTTESTI_CHECK(RTPathCountComponents("c:\\windows") == 2);
RTTESTI_CHECK(RTPathCountComponents("c:\\windows\\") == 2);
RTTESTI_CHECK(RTPathCountComponents("c:\\windows\\system32") == 3);
RTTESTI_CHECK(RTPathCountComponents("//./C$") == 1);
RTTESTI_CHECK(RTPathCountComponents("\\\\.\\C$") == 1);
RTTESTI_CHECK(RTPathCountComponents("/\\.\\C$") == 1);
RTTESTI_CHECK(RTPathCountComponents("//myserver") == 1);
RTTESTI_CHECK(RTPathCountComponents("//myserver/") == 1);
RTTESTI_CHECK(RTPathCountComponents("//myserver/share") == 1);
RTTESTI_CHECK(RTPathCountComponents("//myserver/share/") == 1);
RTTESTI_CHECK(RTPathCountComponents("//myserver/share\\") == 1);
RTTESTI_CHECK(RTPathCountComponents("//myserver/share\\x") == 2);
RTTESTI_CHECK(RTPathCountComponents("//myserver/share\\x\\y") == 3);
RTTESTI_CHECK(RTPathCountComponents("//myserver/share\\x\\y\\") == 3);
#endif
/*
* RTPathCopyComponents
*/
struct
{
const char *pszSrc;
size_t cComponents;
const char *pszResult;
} s_aCopyComponents[] =
{
{ "", 0, "" },
{ "", 5, "" },
{ "/", 0, "" },
{ "/", 1, "/" },
{ "/", 2, "/" },
{ "/usr/bin/sed", 0, "" },
{ "/usr/bin/sed", 1, "/" },
{ "/usr/bin/sed", 2, "/usr/" },
{ "/usr/bin/sed", 3, "/usr/bin/" },
{ "/usr/bin/sed", 4, "/usr/bin/sed" },
{ "/usr/bin/sed", 5, "/usr/bin/sed" },
{ "/usr/bin/sed", 6, "/usr/bin/sed" },
{ "/usr///bin/sed", 2, "/usr///" },
};
for (unsigned i = 0; i < RT_ELEMENTS(s_aCopyComponents); i++)
{
const char *pszInput = s_aCopyComponents[i].pszSrc;
size_t cComponents = s_aCopyComponents[i].cComponents;
const char *pszResult = s_aCopyComponents[i].pszResult;
memset(szPath, 'a', sizeof(szPath));
rc = RTPathCopyComponents(szPath, sizeof(szPath), pszInput, cComponents);
RTTESTI_CHECK_RC(rc, VINF_SUCCESS);
if (RT_SUCCESS(rc) && strcmp(szPath, pszResult))
RTTestIFailed("Unexpected result\n"
" input: '%s' cComponents=%u\n"
" output: '%s'\n"
"expected: '%s'",
pszInput, cComponents, szPath, pszResult);
else if (RT_SUCCESS(rc))
{
RTTESTI_CHECK_RC(RTPathCopyComponents(szPath, strlen(pszResult) + 1, pszInput, cComponents), VINF_SUCCESS);
RTTESTI_CHECK_RC(RTPathCopyComponents(szPath, strlen(pszResult), pszInput, cComponents), VERR_BUFFER_OVERFLOW);
}
}
/*
* RTPathStripExt
*/
RTTestSub(hTest, "RTPathStripExt");
struct
{
const char *pszSrc;
const char *pszResult;
} s_aStripExt[] =
{
{ "filename.ext", "filename" },
{ "filename.ext1.ext2.ext3", "filename.ext1.ext2" },
{ "filename..ext", "filename." },
{ "filename.ext.", "filename.ext" }, /** @todo This is a bit weird/wrong, but not half as weird as the way Windows+OS/2 deals with a trailing dots. */
};
for (unsigned i = 0; i < RT_ELEMENTS(s_aStripExt); i++)
{
const char *pszInput = s_aStripExt[i].pszSrc;
const char *pszResult = s_aStripExt[i].pszResult;
strcpy(szPath, pszInput);
RTPathStripExt(szPath);
if (strcmp(szPath, pszResult))
RTTestIFailed("Unexpected result\n"
" input: '%s'\n"
" output: '%s'\n"
"expected: '%s'",
pszInput, szPath, pszResult);
}
/*
* RTPathCalcRelative
*/
RTTestSub(hTest, "RTPathCalcRelative");
struct
{
const char *pszFrom;
const char *pszTo;
int rc;
const char *pszExpected;
} s_aRelPath[] =
{
{ "/home/test.ext", "/home/test2.ext", VINF_SUCCESS, "test2.ext"},
{ "/dir/test.ext", "/dir/dir2/test2.ext", VINF_SUCCESS, "dir2/test2.ext"},
{ "/dir/dir2/test.ext", "/dir/test2.ext", VINF_SUCCESS, "../test2.ext"},
{ "/dir/dir2/test.ext", "/dir/dir3/test2.ext", VINF_SUCCESS, "../dir3/test2.ext"},
#if defined (RT_OS_OS2) || defined (RT_OS_WINDOWS)
{ "\\\\server\\share\\test.ext", "\\\\server\\share2\\test2.ext", VERR_NOT_SUPPORTED, ""},
{ "c:\\dir\\test.ext", "f:\\dir\\test.ext", VERR_NOT_SUPPORTED, ""}
#endif
};
for (unsigned i = 0; i < RT_ELEMENTS(s_aRelPath); i++)
{
const char *pszFrom = s_aRelPath[i].pszFrom;
const char *pszTo = s_aRelPath[i].pszTo;
rc = RTPathCalcRelative(szPath, sizeof(szPath), pszFrom, pszTo);
if (rc != s_aRelPath[i].rc)
RTTestIFailed("Unexpected return code\n"
" got: %Rrc\n"
"expected: %Rrc",
rc, s_aRelPath[i].rc);
else if ( RT_SUCCESS(rc)
&& strcmp(szPath, s_aRelPath[i].pszExpected))
RTTestIFailed("Unexpected result\n"
" from: '%s'\n"
" to: '%s'\n"
" output: '%s'\n"
"expected: '%s'",
pszFrom, pszTo, szPath, s_aRelPath[i].pszExpected);
}
/*
* Summary.
*/
return RTTestSummaryAndDestroy(hTest);
}
/**
* The main loop for the VBoxClient daemon.
*/
int main(int argc, char *argv[])
{
/* Initialise our runtime before all else. */
int rc = RTR3InitExe(argc, &argv, 0);
if (RT_FAILURE(rc))
return RTMsgInitFailure(rc);
int rcClipboard;
const char *pszFileName = RTPathFilename(argv[0]);
bool fDaemonise = true;
/* Have any fatal errors occurred yet? */
bool fSuccess = true;
/* Do we know which service we wish to run? */
bool fHaveService = false;
if (NULL == pszFileName)
pszFileName = "VBoxClient";
/* Initialise our global clean-up critical section */
rc = RTCritSectInit(&g_critSect);
if (RT_FAILURE(rc))
{
/* Of course, this should never happen. */
RTPrintf("%s: Failed to initialise the global critical section, rc=%Rrc\n", pszFileName, rc);
return 1;
}
/* Parse our option(s) */
/** @todo Use RTGetOpt() if the arguments become more complex. */
for (int i = 1; i < argc; ++i)
{
if (!strcmp(argv[i], "-d") || !strcmp(argv[i], "--nodaemon"))
fDaemonise = false;
else if (!strcmp(argv[i], "--clipboard"))
{
if (g_pService == NULL)
g_pService = VBoxClient::GetClipboardService();
else
fSuccess = false;
}
else if (!strcmp(argv[i], "--display"))
{
if (g_pService == NULL)
g_pService = VBoxClient::GetDisplayService();
else
fSuccess = false;
}
else if (!strcmp(argv[i], "--seamless"))
{
if (g_pService == NULL)
g_pService = VBoxClient::GetSeamlessService();
else
fSuccess = false;
}
else if (!strcmp(argv[i], "--checkhostversion"))
{
if (g_pService == NULL)
g_pService = VBoxClient::GetHostVersionService();
else
fSuccess = false;
}
#ifdef VBOX_WITH_DRAG_AND_DROP
else if (!strcmp(argv[i], "--draganddrop"))
{
if (g_pService == NULL)
g_pService = VBoxClient::GetDragAndDropService();
else
fSuccess = false;
}
#endif /* VBOX_WITH_DRAG_AND_DROP */
else if (!strcmp(argv[i], "-h") || !strcmp(argv[i], "--help"))
{
vboxClientUsage(pszFileName);
return 0;
}
else
{
RTPrintf("%s: unrecognized option `%s'\n", pszFileName, argv[i]);
RTPrintf("Try `%s --help' for more information\n", pszFileName);
return 1;
}
}
if (!fSuccess || !g_pService)
{
vboxClientUsage(pszFileName);
return 1;
}
/* Get the path for the pidfiles */
rc = RTPathUserHome(g_szPidFile, sizeof(g_szPidFile));
if (RT_FAILURE(rc))
{
RTPrintf("VBoxClient: failed to get home directory, rc=%Rrc. Exiting.\n", rc);
LogRel(("VBoxClient: failed to get home directory, rc=%Rrc. Exiting.\n", rc));
return 1;
}
rc = RTPathAppend(g_szPidFile, sizeof(g_szPidFile), g_pService->getPidFilePath());
if (RT_FAILURE(rc))
{
RTPrintf("VBoxClient: RTPathAppend failed with rc=%Rrc. Exiting.\n", rc);
LogRel(("VBoxClient: RTPathAppend failed with rc=%Rrc. Exiting.\n", rc));
return 1;
}
/* Initialise the guest library. */
if (RT_FAILURE(VbglR3InitUser()))
{
RTPrintf("Failed to connect to the VirtualBox kernel service\n");
LogRel(("Failed to connect to the VirtualBox kernel service\n"));
return 1;
}
if (fDaemonise)
{
rc = VbglR3Daemonize(false /* fNoChDir */, false /* fNoClose */);
if (RT_FAILURE(rc))
{
RTPrintf("VBoxClient: failed to daemonize. Exiting.\n");
LogRel(("VBoxClient: failed to daemonize. Exiting.\n"));
# ifdef DEBUG
RTPrintf("Error %Rrc\n", rc);
# endif
return 1;
}
}
if (g_szPidFile[0] && RT_FAILURE(VbglR3PidFile(g_szPidFile, &g_hPidFile)))
{
RTPrintf("Failed to create a pidfile. Exiting.\n");
LogRel(("Failed to create a pidfile. Exiting.\n"));
VbglR3Term();
return 1;
}
/* Set signal handlers to clean up on exit. */
vboxClientSetSignalHandlers();
/* Set an X11 error handler, so that we don't die when we get unavoidable errors. */
XSetErrorHandler(vboxClientXLibErrorHandler);
/* Set an X11 I/O error handler, so that we can shutdown properly on fatal errors. */
XSetIOErrorHandler(vboxClientXLibIOErrorHandler);
g_pService->run(fDaemonise);
VBoxClient::CleanUp();
return 1; /* We should never get here. */
}
/**
* The main loop for the VBoxClient daemon.
* @todo Clean up for readability.
*/
int main(int argc, char *argv[])
{
bool fDaemonise = true, fRespawn = true;
int rc;
const char *pcszFileName, *pcszStage;
/* Initialise our runtime before all else. */
rc = RTR3InitExe(argc, &argv, 0);
if (RT_FAILURE(rc))
return RTMsgInitFailure(rc);
/* This should never be called twice in one process - in fact one Display
* object should probably never be used from multiple threads anyway. */
if (!XInitThreads())
VBClFatalError(("Failed to initialize X11 threads\n"));
/* Get our file name for error output. */
pcszFileName = RTPathFilename(argv[0]);
if (!pcszFileName)
pcszFileName = "VBoxClient";
/* Parse our option(s) */
/** @todo Use RTGetOpt() if the arguments become more complex. */
for (int i = 1; i < argc; ++i)
{
rc = VERR_INVALID_PARAMETER;
if (!strcmp(argv[i], "-d") || !strcmp(argv[i], "--nodaemon"))
{
/* If the user is running in "no daemon" mode anyway, send critical
* logging to stdout as well. */
PRTLOGGER pReleaseLog = RTLogRelDefaultInstance();
if (pReleaseLog)
rc = RTLogDestinations(pReleaseLog, "stdout");
if (pReleaseLog && RT_FAILURE(rc))
RTPrintf("%s: failed to redivert error output, rc=%Rrc\n",
pcszFileName, rc);
fDaemonise = false;
}
else if (!strcmp(argv[i], "--no-respawn"))
{
fRespawn = false;
}
else if (!strcmp(argv[i], "--clipboard"))
{
if (g_pService)
break;
g_pService = VBClGetClipboardService();
}
else if (!strcmp(argv[i], "--display"))
{
if (g_pService)
break;
g_pService = VBClGetDisplayService();
}
else if (!strcmp(argv[i], "--seamless"))
{
if (g_pService)
break;
g_pService = VBClGetSeamlessService();
}
else if (!strcmp(argv[i], "--checkhostversion"))
{
if (g_pService)
break;
g_pService = VBClGetHostVersionService();
}
#ifdef VBOX_WITH_DRAG_AND_DROP
else if (!strcmp(argv[i], "--draganddrop"))
{
if (g_pService)
break;
g_pService = VBClGetDragAndDropService();
}
#endif /* VBOX_WITH_DRAG_AND_DROP */
else if (!strcmp(argv[i], "-h") || !strcmp(argv[i], "--help"))
{
vboxClientUsage(pcszFileName);
return 0;
}
else
{
RTPrintf("%s: unrecognized option `%s'\n", pcszFileName, argv[i]);
RTPrintf("Try `%s --help' for more information\n", pcszFileName);
return 1;
}
rc = VINF_SUCCESS;
}
if (RT_FAILURE(rc) || !g_pService)
{
vboxClientUsage(pcszFileName);
return 1;
}
rc = RTCritSectInit(&g_critSect);
if (RT_FAILURE(rc))
VBClFatalError(("Initialising critical section: %Rrc\n", rc));
rc = RTPathUserHome(g_szPidFile, sizeof(g_szPidFile));
if (RT_FAILURE(rc))
VBClFatalError(("Getting home directory for pid-file: %Rrc\n", rc));
rc = RTPathAppend(g_szPidFile, sizeof(g_szPidFile),
(*g_pService)->getPidFilePath());
if (RT_FAILURE(rc))
VBClFatalError(("Creating pid-file path: %Rrc\n", rc));
if (fDaemonise)
rc = VbglR3Daemonize(false /* fNoChDir */, false /* fNoClose */, fRespawn, &cRespawn);
if (RT_FAILURE(rc))
VBClFatalError(("Daemonizing: %Rrc\n", rc));
if (g_szPidFile[0])
rc = VbglR3PidFile(g_szPidFile, &g_hPidFile);
if (rc == VERR_FILE_LOCK_VIOLATION) /* Already running. */
return 0;
if (RT_FAILURE(rc))
VBClFatalError(("Creating pid-file: %Rrc\n", rc));
/* Set signal handlers to clean up on exit. */
vboxClientSetSignalHandlers();
#ifndef VBOXCLIENT_WITHOUT_X11
/* Set an X11 error handler, so that we don't die when we get unavoidable
* errors. */
XSetErrorHandler(vboxClientXLibErrorHandler);
/* Set an X11 I/O error handler, so that we can shutdown properly on
* fatal errors. */
XSetIOErrorHandler(vboxClientXLibIOErrorHandler);
#endif
rc = (*g_pService)->init(g_pService);
if (RT_FAILURE(rc))
VBClFatalError(("Initialising service: %Rrc\n", rc));
rc = (*g_pService)->run(g_pService, fDaemonise);
if (RT_FAILURE(rc))
VBClFatalError(("Service main loop failed: %Rrc\n", rc));
VBClCleanUp();
return 0;
}