RTDECL(int) RTPathGetCurrentOnDrive(char chDrive, char *pszPath, size_t cbPath)
{
#ifdef HAVE_DRIVE
/*
* Check if it's the same drive as the current directory.
*/
int rc = RTPathGetCurrent(pszPath, cbPath);
if (RT_SUCCESS(rc))
{
if ( ( chDrive == *pszPath
|| RT_C_TO_LOWER(chDrive) == RT_C_TO_LOWER(*pszPath))
&& RTPATH_IS_VOLSEP(pszPath[1]))
return rc;
/*
* Different drive, indicate root.
*/
if (cbPath >= 4)
{
pszPath[0] = RT_C_TO_UPPER(chDrive);
pszPath[1] = ':';
pszPath[2] = RTPATH_SLASH;
pszPath[3] = '\0';
return VINF_SUCCESS;
}
}
return rc;
#else
/*
* No driver letters, just return root slash on whatever we're asked.
*/
NOREF(chDrive);
if (cbPath >= 2)
{
pszPath[0] = RTPATH_SLASH;
pszPath[1] = '\0';
return VINF_SUCCESS;
}
return VERR_BUFFER_OVERFLOW;
#endif
}
RTDECL(int) RTPathAbs(const char *pszPath, char *pszAbsPath, size_t cchAbsPath)
{
int rc;
/*
* Validation.
*/
AssertPtr(pszAbsPath);
AssertPtr(pszPath);
if (RT_UNLIKELY(!*pszPath))
return VERR_INVALID_PARAMETER;
/*
* Make a clean working copy of the input.
*/
size_t cchPath = strlen(pszPath);
if (cchPath > PATH_MAX)
{
LogFlow(("RTPathAbs(%p:{%s}, %p, %d): returns %Rrc\n", pszPath, pszPath, pszAbsPath, cchAbsPath, VERR_FILENAME_TOO_LONG));
return VERR_FILENAME_TOO_LONG;
}
char szTmpPath[PATH_MAX + 1];
memcpy(szTmpPath, pszPath, cchPath + 1);
size_t cchTmpPath = fsCleanPath(szTmpPath);
/*
* Handle "." specially (fsCleanPath does).
*/
if (szTmpPath[0] == '.' && !szTmpPath[1])
return RTPathGetCurrent(pszAbsPath, cchAbsPath);
/*
* Do we have a root slash?
*/
char *pszCur = szTmpPath;
#ifdef HAVE_DRIVE
if (pszCur[0] && RTPATH_IS_VOLSEP(pszCur[1]) && pszCur[2] == '/')
pszCur += 3;
# ifdef HAVE_UNC
else if (pszCur[0] == '/' && pszCur[1] == '/')
pszCur += 2;
# endif
#else /* !HAVE_DRIVE */
if (pszCur[0] == '/')
pszCur += 1;
#endif /* !HAVE_DRIVE */
else
{
/*
* No, prepend the current directory to the relative path.
*/
char szCurDir[RTPATH_MAX];
rc = RTPathGetCurrent(szCurDir, sizeof(szCurDir));
AssertRCReturn(rc, rc);
size_t cchCurDir = fsCleanPath(szCurDir); /* paranoia */
if (cchCurDir + cchTmpPath + 1 > PATH_MAX)
{
LogFlow(("RTPathAbs(%p:{%s}, %p, %d): returns %Rrc\n", pszPath, pszPath, pszAbsPath, cchAbsPath, VERR_FILENAME_TOO_LONG));
return VERR_FILENAME_TOO_LONG;
}
memmove(szTmpPath + cchCurDir + 1, szTmpPath, cchTmpPath + 1);
memcpy(szTmpPath, szCurDir, cchCurDir);
szTmpPath[cchCurDir] = '/';
#ifdef HAVE_DRIVE
if (pszCur[0] && RTPATH_IS_VOLSEP(pszCur[1]) && pszCur[2] == '/')
pszCur += 3;
# ifdef HAVE_UNC
else if (pszCur[0] == '/' && pszCur[1] == '/')
pszCur += 2;
# endif
#else
if (pszCur[0] == '/')
pszCur += 1;
#endif
else
AssertMsgFailedReturn(("pszCur=%s\n", pszCur), VERR_INTERNAL_ERROR);
}
char *pszTop = pszCur;
/*
* Get rid of double dot path components by evaluating them.
*/
for (;;)
{
if ( pszCur[0] == '.'
&& pszCur[1] == '.'
&& (!pszCur[2] || pszCur[2] == '/'))
{
/* rewind to the previous component if any */
char *pszPrev = pszCur - 1;
if (pszPrev > pszTop)
while (*--pszPrev != '/')
;
AssertMsg(*pszPrev == '/', ("szTmpPath={%s}, pszPrev=+%u\n", szTmpPath, pszPrev - szTmpPath));
memmove(pszPrev, pszCur + 2, strlen(pszCur + 2) + 1);
pszCur = pszPrev;
}
else
{
/* advance to end of component. */
while (*pszCur && *pszCur != '/')
pszCur++;
}
if (!*pszCur)
break;
/* skip the slash */
++pszCur;
}
if (pszCur < pszTop)
{
/*
* We overwrote the root slash with '\0', restore it.
*/
*pszCur++ = '/';
*pszCur = '\0';
}
else if (pszCur > pszTop && pszCur[-1] == '/')
{
/*
* Extra trailing slash in a non-root path, remove it.
* (A bit questionable...)
*/
*--pszCur = '\0';
}
/*
* Copy the result to the user buffer.
*/
cchTmpPath = pszCur - szTmpPath;
if (cchTmpPath < cchAbsPath)
{
memcpy(pszAbsPath, szTmpPath, cchTmpPath + 1);
rc = VINF_SUCCESS;
}
else
rc = VERR_BUFFER_OVERFLOW;
LogFlow(("RTPathAbs(%p:{%s}, %p:{%s}, %d): returns %Rrc\n", pszPath, pszPath, pszAbsPath,
RT_SUCCESS(rc) ? pszAbsPath : "<failed>", cchAbsPath, rc));
return rc;
}
RTDECL(int) RTPathAbs(const char *pszPath, char *pszAbsPath, size_t cchAbsPath)
{
int rc;
/*
* Validation.
*/
AssertPtr(pszAbsPath);
AssertPtr(pszPath);
if (RT_UNLIKELY(!*pszPath))
return VERR_INVALID_PARAMETER; //VERR_INVALID_NAME;
/*
* Make a clean working copy of the input.
*/
size_t cchPath = strlen(pszPath);
if (cchPath >= RTPATH_MAX)
{
LogFlow(("RTPathAbs(%p:{%s}, %p, %d): returns %Rrc\n", pszPath, pszPath, pszAbsPath, cchAbsPath, VERR_FILENAME_TOO_LONG));
return VERR_FILENAME_TOO_LONG;
}
char szTmpPath[RTPATH_MAX];
memcpy(szTmpPath, pszPath, cchPath + 1);
size_t cchTmpPath = fsCleanPath(szTmpPath);
/*
* Handle "." specially (fsCleanPath does).
*/
if (szTmpPath[0] == '.')
{
if ( cchTmpPath == 1
|| (cchTmpPath == 2 && szTmpPath[1] == RTPATH_SLASH))
{
rc = RTPathGetCurrent(pszAbsPath, cchAbsPath);
if (RT_SUCCESS(rc))
{
size_t cch = fsCleanPath(pszAbsPath);
char *pszTop = rtPathSkipRootSpec(pszAbsPath);
#if 1
if ((uintptr_t)&pszAbsPath[cch] > (uintptr_t)pszTop && pszAbsPath[cch - 1] == RTPATH_SLASH)
pszAbsPath[cch - 1] = '\0';
#else
if ( cchTmpPath == 2
&& (uintptr_t)&pszAbsPath[cch - 1] > (uintptr_t)pszTop && pszAbsPath[cch - 1] != RTPATH_SLASH)
{
if (cch + 1 < cchAbsPath)
{
pszAbsPath[cch++] = RTPATH_SLASH;
pszAbsPath[cch] = '\0';
}
else
rc = VERR_BUFFER_OVERFLOW;
}
#endif
}
return rc;
}
}
/*
* Do we have an incomplete root spec? Supply the missing bits.
*/
#ifdef HAVE_DRIVE
if ( !(szTmpPath[0] && RTPATH_IS_VOLSEP(szTmpPath[1]) && szTmpPath[2] == RTPATH_SLASH)
# ifdef HAVE_UNC
&& !(szTmpPath[0] == RTPATH_SLASH && szTmpPath[1] == RTPATH_SLASH)
# endif
)
#else
if (szTmpPath[0] != RTPATH_SLASH)
#endif
{
char szCurDir[RTPATH_MAX];
size_t cchCurDir;
int offApplyAt;
bool fNeedSlash;
#ifdef HAVE_DRIVE
if (szTmpPath[0] && RTPATH_IS_VOLSEP(szTmpPath[1]) && szTmpPath[2] != RTPATH_SLASH)
{
/*
* Relative to drive specific current directory.
*/
rc = RTPathGetCurrentOnDrive(szTmpPath[0], szCurDir, sizeof(szCurDir));
fNeedSlash = true;
offApplyAt = 2;
}
# ifdef HAVE_UNC
else if (szTmpPath[0] == RTPATH_SLASH && szTmpPath[1] != RTPATH_SLASH)
# else
else if (szTmpPath[0] == RTPATH_SLASH)
# endif
{
/*
* Root of current drive. This may return a UNC root if we're not
* standing on a drive but on a UNC share.
*/
rc = RTPathGetCurrentDrive(szCurDir, sizeof(szCurDir));
fNeedSlash = false;
offApplyAt = 0;
}
else
#endif
{
/*
* Relative to current directory.
*/
rc = RTPathGetCurrent(szCurDir, sizeof(szCurDir));
fNeedSlash = true;
offApplyAt = 0;
}
if (RT_SUCCESS(rc))
{
cchCurDir = fsCleanPath(szCurDir);
if (fNeedSlash && cchCurDir > 0 && szCurDir[cchCurDir - 1] == RTPATH_SLASH)
fNeedSlash = false;
if (cchCurDir + fNeedSlash + cchTmpPath - offApplyAt <= RTPATH_MAX)
{
memmove(szTmpPath + cchCurDir + fNeedSlash, szTmpPath + offApplyAt, cchTmpPath + 1 - offApplyAt);
memcpy(szTmpPath, szCurDir, cchCurDir);
if (fNeedSlash)
szTmpPath[cchCurDir] = RTPATH_SLASH;
}
else
rc = VERR_FILENAME_TOO_LONG;
}
if (RT_FAILURE(rc))
{
LogFlow(("RTPathAbs(%p:{%s}, %p, %d): returns %Rrc\n", pszPath, pszPath, pszAbsPath, cchAbsPath, rc));
return rc;
}
}
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);
}
RTDECL(int) RTPathGetCurrentDrive(char *pszPath, size_t cbPath)
{
#ifdef HAVE_DRIVE
/*
* Query the current directroy and extract the wanted information from it.
*/
int rc = RTPathGetCurrent(pszPath, cbPath);
if (RT_SUCCESS(rc))
{
/*
* Drive letter? Chop off at root slash.
*/
if (pszPath[0] && RTPATH_IS_VOLSEP(pszPath[1]))
{
pszPath[2] = '\0';
return rc;
}
/*
* UNC? Chop off after share.
*/
if ( RTPATH_IS_SLASH(pszPath[0])
&& RTPATH_IS_SLASH(pszPath[1])
&& !RTPATH_IS_SLASH(pszPath[2])
&& pszPath[2])
{
/* Work thru the server name. */
size_t off = 3;
while (!RTPATH_IS_SLASH(pszPath[off]) && pszPath[off])
off++;
size_t offServerSlash = off;
/* Is there a share name? */
if (RTPATH_IS_SLASH(pszPath[off]))
{
while (RTPATH_IS_SLASH(pszPath[off]))
off++;
if (pszPath[off])
{
/* Work thru the share name. */
while (!RTPATH_IS_SLASH(pszPath[off]) && pszPath[off])
off++;
}
/* No share name, chop at server name. */
else
off = offServerSlash;
}
}
return VERR_INTERNAL_ERROR_4;
}
return rc;
#else /* !HAVE_DRIVE */
/*
* No drive letters on this system, return empty string.
*/
if (cbPath > 0)
{
*pszPath = '\0';
return VINF_SUCCESS;
}
return VERR_BUFFER_OVERFLOW;
#endif /* !HAVE_DRIVE */
}
