/** * 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; }