}
}
}
-#ifdef __REACTOS__
- if (dwFlags & CONNECT_UPDATE_PROFILE)
+ if (ret == WN_SUCCESS && dwFlags & CONNECT_UPDATE_PROFILE)
{
HKEY user_profile;
- WCHAR *coma = strchrW(lpName, ':');
- if (coma && RegOpenCurrentUser(KEY_ALL_ACCESS, &user_profile) == ERROR_SUCCESS)
+ /* FIXME: Only remove it if that's a drive letter */
+ if (isalphaW(lpName[0]) && lpName[1] == ':' &&
+ RegOpenCurrentUser(KEY_ALL_ACCESS, &user_profile) == ERROR_SUCCESS)
{
- WCHAR *subkey;
- DWORD len;
-
- len = (ULONG_PTR)coma - (ULONG_PTR)lpName + sizeof(L"Network\\");
- subkey = HeapAlloc(GetProcessHeap(), 0, len);
- if (subkey)
- {
- strcpyW(subkey, L"Network\\");
- memcpy(subkey + (sizeof(L"Network\\") / sizeof(WCHAR)) - 1, lpName, (ULONG_PTR)coma - (ULONG_PTR)lpName);
- subkey[len / sizeof(WCHAR) - 1] = 0;
+ WCHAR subkey[10] = {'N', 'e', 't', 'w', 'o', 'r', 'k', '\\', lpName[0], 0};
- TRACE("Removing: %S\n", subkey);
-
- RegDeleteKeyW(user_profile, subkey);
- HeapFree(GetProcessHeap(), 0, subkey);
- }
+ RegDeleteKeyW(user_profile, subkey);
RegCloseKey(user_profile);
}
}
-#endif
return ret;
}