/**
* Wrapper around RTNtPathExpand8dot3Path that allocates a buffer instead of
* working on the input buffer.
*
* @returns IPRT status code, see RTNtPathExpand8dot3Path().
* @param pUniStrSrc The path to fix up. MaximumLength is the max buffer
* length.
* @param fPathOnly Whether to only process the path and leave the filename
* as passed in.
* @param pUniStrDst Output string. On success, the caller must use
* RTUtf16Free to free what the Buffer member points to.
* This is all zeros and NULL on failure.
*/
RTDECL(int) RTNtPathExpand8dot3PathA(PCUNICODE_STRING pUniStrSrc, bool fPathOnly, PUNICODE_STRING pUniStrDst)
{
/* Guess a reasonable size for the final version. */
size_t const cbShort = pUniStrSrc->Length;
size_t cbLong = RT_MIN(_64K - 1, cbShort * 8);
if (cbLong < RTPATH_MAX)
cbLong = RTPATH_MAX * 2;
AssertCompile(RTPATH_MAX * 2 < _64K);
pUniStrDst->Buffer = (WCHAR *)RTUtf16Alloc(cbLong);
if (pUniStrDst->Buffer != NULL)
{
/* Copy over the short name and fix it up. */
pUniStrDst->MaximumLength = (uint16_t)cbLong;
pUniStrDst->Length = (uint16_t)cbShort;
memcpy(pUniStrDst->Buffer, pUniStrSrc->Buffer, cbShort);
pUniStrDst->Buffer[cbShort / sizeof(WCHAR)] = '\0';
int rc = RTNtPathExpand8dot3Path(pUniStrDst, fPathOnly);
if (RT_SUCCESS(rc))
return rc;
/* We failed, bail. */
RTUtf16Free(pUniStrDst->Buffer);
pUniStrDst->Buffer = NULL;
}
pUniStrDst->Length = 0;
pUniStrDst->MaximumLength = 0;
return VERR_NO_UTF16_MEMORY;
}
int rtldrNativeLoad(const char *pszFilename, uintptr_t *phHandle, uint32_t fFlags, PRTERRINFO pErrInfo)
{
Assert(sizeof(*phHandle) >= sizeof(HMODULE));
AssertReturn(!(fFlags & RTLDRLOAD_FLAGS_GLOBAL), VERR_INVALID_FLAGS);
AssertLogRelMsgReturn(RTPathStartsWithRoot(pszFilename), /* Relative names will still be applied to the search path. */
("pszFilename='%s'\n", pszFilename),
VERR_INTERNAL_ERROR_2);
/*
* Convert to UTF-16 and make sure it got a .DLL suffix.
*/
int rc;
RTUTF16 *pwszNative = NULL;
if (RTPathHasSuffix(pszFilename))
rc = RTStrToUtf16(pszFilename, &pwszNative);
else
{
size_t cwcAlloc;
rc = RTStrCalcUtf16LenEx(pszFilename, RTSTR_MAX, &cwcAlloc);
if (RT_SUCCESS(rc))
{
cwcAlloc += sizeof(".DLL");
pwszNative = RTUtf16Alloc(cwcAlloc * sizeof(RTUTF16));
if (pwszNative)
{
size_t cwcNative;
rc = RTStrToUtf16Ex(pszFilename, RTSTR_MAX, &pwszNative, cwcAlloc, &cwcNative);
if (RT_SUCCESS(rc))
rc = RTUtf16CopyAscii(&pwszNative[cwcNative], cwcAlloc - cwcNative, ".DLL");
}
else
rc = VERR_NO_UTF16_MEMORY;
}
}
if (RT_SUCCESS(rc))
{
/*
* Attempt load.
*/
HMODULE hmod;
static int s_iSearchDllLoadDirSupported = 0;
if ( !(fFlags & RTLDRLOAD_FLAGS_NT_SEARCH_DLL_LOAD_DIR)
|| s_iSearchDllLoadDirSupported < 0)
hmod = LoadLibraryExW(pwszNative, NULL, 0);
else
{
hmod = LoadLibraryExW(pwszNative, NULL, LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR | LOAD_LIBRARY_SEARCH_SYSTEM32
| LOAD_LIBRARY_SEARCH_APPLICATION_DIR);
if (s_iSearchDllLoadDirSupported == 0)
{
if (hmod != NULL || GetLastError() != ERROR_INVALID_PARAMETER)
s_iSearchDllLoadDirSupported = 1;
else
{
s_iSearchDllLoadDirSupported = -1;
hmod = LoadLibraryExW(pwszNative, NULL, 0);
}
}
}
if (hmod)
{
*phHandle = (uintptr_t)hmod;
RTUtf16Free(pwszNative);
return VINF_SUCCESS;
}
/*
* Try figure why it failed to load.
*/
DWORD dwErr = GetLastError();
rc = RTErrConvertFromWin32(dwErr);
rc = RTErrInfoSetF(pErrInfo, rc, "GetLastError=%u", dwErr);
}
else
rc = RTErrInfoSetF(pErrInfo, rc, "Error converting UTF-8 to UTF-16 string.");
RTUtf16Free(pwszNative);
return rc;
}
