wchar_t *resolveUNC(const wchar_t *filename)
{
UNIVERSAL_NAME_INFOW uniwTemp;
DWORD uncSize = sizeof(UNIVERSAL_NAME_INFOW);
DWORD res = WNetGetUniversalNameW(filename, UNIVERSAL_NAME_INFO_LEVEL,
&uniwTemp, &uncSize);
if( res != ERROR_MORE_DATA ) {
return 0;
}
uncSize += sizeof(wchar_t);
uint8_t *uncData = new uint8_t[uncSize];
if( uncData == 0 ) {
return 0;
}
res = WNetGetUniversalNameW(filename, UNIVERSAL_NAME_INFO_LEVEL,
uncData, &uncSize);
if( res != NO_ERROR ) {
delete[] uncData;
return 0;
}
UNIVERSAL_NAME_INFOW *uniw = (UNIVERSAL_NAME_INFOW*)uncData;
const size_t len = csStringLen(uniw->lpUniversalName);
if( len < 1 ) {
delete[] uncData;
return 0;
}
wchar_t *uncName = new wchar_t[len+1];
if( uncName == 0 ) {
delete[] uncData;
return 0;
}
csStringNCpy(uncName, uniw->lpUniversalName, len);
uncName[len] = L'\0';
delete[] uncData;
return uncName;
}
HRESULT PathConvertMappedDriveToUNC(PCWSTR pszPath, PWSTR *ppszUNC)
{
*ppszUNC = NULL;
HRESULT hr = E_FAIL;
#if 0
// build simple root "C:" (with no slash) since WNetGetConnection does not deal with these
WCHAR szDrive[3] = {pszPath[0], pszPath[1], 0};
if (DRIVE_REMOTE == GetDriveType(szDrive))
{
WCHAR szRemoteName[MAX_PATH];
DWORD dwLen = ARRAYSIZE(szRemoteName);
if (NO_ERROR == WNetGetConnection(szDrive, szRemoteName, &dwLen))
{
// +3 to skip to "\foo" in "C:\foo"
WCHAR szUNC[MAX_PATH];
PathCombine(szUNC, szRemoteName, pszPath + 3);
hr = SHStrDup(szUNC, ppszUNC);
}
}
#else
// alternate implementation
struct
{
UNIVERSAL_NAME_INFO uni; // a pointer
WCHAR szBuf[MAX_PATH]; // buffer for path, make bigger in the future
} uniBuffer;
DWORD cbBuffer = sizeof(uniBuffer);
// test this with disconnected drive letters, might fail for them
hr = HRESULT_FROM_WIN32(WNetGetUniversalNameW(pszPath, UNIVERSAL_NAME_INFO_LEVEL, &uniBuffer, &cbBuffer));
if (SUCCEEDED(hr))
{
PathRemoveBackslash(uniBuffer.uni.lpUniversalName); // some cases return trailing slash
hr = SHStrDup(uniBuffer.uni.lpUniversalName, ppszUNC);
}
#endif
return hr;
}
static void test_WNetGetRemoteName(void)
{
DWORD ret;
char buffer[1024];
DWORD drive_type, info_size, fail_size;
char driveA[] = "A:\\";
char driveandpathA[] = "A:\\file.txt";
WCHAR driveW[] = {'A',':','\\',0};
for(; *driveA <= 'Z'; ++*driveA, ++*driveandpathA, ++*driveW){
drive_type = GetDriveTypeW(driveW);
info_size = sizeof(buffer);
ret = WNetGetUniversalNameA(driveA, REMOTE_NAME_INFO_LEVEL,
buffer, &info_size);
if(drive_type == DRIVE_REMOTE)
todo_wine
ok(ret == WN_NO_ERROR, "WNetGetUniversalNameA failed: %08x\n", ret);
else
ok(ret == WN_NOT_CONNECTED || ret == WN_NO_NET_OR_BAD_PATH,
"(%s) WNetGetUniversalNameA gave wrong error: %u\n", driveA, ret);
ok(info_size == sizeof(buffer), "Got wrong size: %u\n", info_size);
fail_size = 0;
ret = WNetGetUniversalNameA(driveA, REMOTE_NAME_INFO_LEVEL,
buffer, &fail_size);
if(drive_type == DRIVE_REMOTE)
todo_wine
ok(ret == WN_BAD_VALUE || ret == WN_MORE_DATA, "WNetGetUniversalNameA failed: %08x\n", ret);
else
ok(ret == WN_NOT_CONNECTED || ret == WN_NO_NET_OR_BAD_PATH,
"(%s) WNetGetUniversalNameA gave wrong error: %u\n", driveA, ret);
ret = WNetGetUniversalNameA(driveA, REMOTE_NAME_INFO_LEVEL,
buffer, NULL);
todo_wine ok(ret == WN_BAD_POINTER, "WNetGetUniversalNameA failed: %08x\n", ret);
ret = WNetGetUniversalNameA(driveA, REMOTE_NAME_INFO_LEVEL,
NULL, &info_size);
if(drive_type == DRIVE_REMOTE)
todo_wine
ok(ret == WN_BAD_POINTER || ret == WN_BAD_VALUE, "WNetGetUniversalNameA failed: %08x\n", ret);
else
ok(ret == WN_NOT_CONNECTED || ret == WN_BAD_VALUE,
"(%s) WNetGetUniversalNameA gave wrong error: %u\n", driveA, ret);
fail_size = ARRAY_SIZE(driveA) - 1;
ret = WNetGetUniversalNameA(driveA, REMOTE_NAME_INFO_LEVEL,
buffer, &fail_size);
if(drive_type == DRIVE_REMOTE)
todo_wine ok(ret == WN_MORE_DATA, "WNetGetUniversalNameA failed: %08x\n", ret);
ret = WNetGetUniversalNameA(driveandpathA, REMOTE_NAME_INFO_LEVEL,
buffer, &info_size);
if(drive_type == DRIVE_REMOTE)
todo_wine ok(ret == WN_NO_ERROR, "WNetGetUniversalNameA failed: %08x\n", ret);
info_size = sizeof(buffer);
ret = WNetGetUniversalNameW(driveW, REMOTE_NAME_INFO_LEVEL,
buffer, &info_size);
todo_wine{
if(drive_type == DRIVE_REMOTE)
ok(ret == WN_NO_ERROR, "WNetGetUniversalNameW failed: %08x\n", ret);
else
ok(ret == WN_NOT_CONNECTED || ret == WN_NO_NET_OR_BAD_PATH,
"(%s) WNetGetUniversalNameW gave wrong error: %u\n", driveA, ret);
}
ok(info_size == sizeof(buffer), "Got wrong size: %u\n", info_size);
}
}
static void test_WNetGetUniversalName(void)
{
DWORD ret;
char buffer[1024];
DWORD drive_type, info_size, fail_size;
char driveA[] = "A:\\";
char driveandpathA[] = "A:\\file.txt";
WCHAR driveW[] = {'A',':','\\',0};
for(; *driveA <= 'Z'; ++*driveA, ++*driveandpathA, ++*driveW){
drive_type = GetDriveTypeW(driveW);
info_size = sizeof(buffer);
ret = WNetGetUniversalNameA(driveA, UNIVERSAL_NAME_INFO_LEVEL,
buffer, &info_size);
if(drive_type == DRIVE_REMOTE)
ok(ret == WN_NO_ERROR, "WNetGetUniversalNameA failed: %08x\n", ret);
else
/* WN_NO_NET_OR_BAD_PATH (DRIVE_FIXED) returned from the virtual drive (usual Q:)
created by the microsoft application virtualization client */
ok((ret == WN_NOT_CONNECTED) || (ret == WN_NO_NET_OR_BAD_PATH),
"WNetGetUniversalNameA(%s, ...) returned %u (drive_type: %u)\n",
driveA, ret, drive_type);
ok(info_size == sizeof(buffer), "Got wrong size: %u\n", info_size);
fail_size = 0;
ret = WNetGetUniversalNameA(driveA, UNIVERSAL_NAME_INFO_LEVEL,
buffer, &fail_size);
if(drive_type == DRIVE_REMOTE)
todo_wine ok(ret == WN_BAD_VALUE || ret == WN_MORE_DATA, "WNetGetUniversalNameA failed: %08x\n", ret);
else
ok(ret == WN_NOT_CONNECTED || ret == WN_NO_NET_OR_BAD_PATH,
"(%s) WNetGetUniversalNameW gave wrong error: %u\n", driveA, ret);
fail_size = ARRAY_SIZE(driveA) - 1;
ret = WNetGetUniversalNameA(driveA, UNIVERSAL_NAME_INFO_LEVEL,
buffer, &fail_size);
if(drive_type == DRIVE_REMOTE)
ok(ret == WN_MORE_DATA, "WNetGetUniversalNameA failed: %08x\n", ret);
ret = WNetGetUniversalNameA(driveandpathA, UNIVERSAL_NAME_INFO_LEVEL,
buffer, &info_size);
if(drive_type == DRIVE_REMOTE)
todo_wine ok(ret == WN_NO_ERROR, "WNetGetUniversalNameA failed: %08x\n", ret);
info_size = sizeof(buffer);
ret = WNetGetUniversalNameW(driveW, UNIVERSAL_NAME_INFO_LEVEL,
buffer, &info_size);
if(drive_type == DRIVE_REMOTE)
ok(ret == WN_NO_ERROR, "WNetGetUniversalNameW failed: %08x\n", ret);
else
ok((ret == WN_NOT_CONNECTED) || (ret == WN_NO_NET_OR_BAD_PATH),
"WNetGetUniversalNameW(%s, ...) returned %u (drive_type: %u)\n",
wine_dbgstr_w(driveW), ret, drive_type);
if(drive_type != DRIVE_REMOTE)
ok(info_size == sizeof(buffer), "Got wrong size: %u\n", info_size);
}
}
