2 * Unit test suite for userenv functions
4 * Copyright 2008 Google (Lei Zhang)
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
32 #include "wine/test.h"
34 #define expect(EXPECTED,GOT) ok((GOT)==(EXPECTED), "Expected %d, got %d\n", (EXPECTED), (GOT))
35 #define expect_env(EXPECTED,GOT,VAR) ok((GOT)==(EXPECTED), "Expected %d, got %d for %s (%d)\n", (EXPECTED), (GOT), (VAR), j)
36 #define expect_gle(EXPECTED) ok(GetLastError() == (EXPECTED), "Expected %d, got %d\n", (EXPECTED), GetLastError())
38 static BOOL (WINAPI
*pIsWow64Process
)(HANDLE
,PBOOL
);
45 /* Helper function for retrieving environment variables */
46 static BOOL
get_env(const WCHAR
* env
, const char * var
, char ** result
)
48 const WCHAR
* p
= env
;
49 int envlen
, varlen
, buflen
;
52 if (!env
|| !var
|| !result
) return FALSE
;
57 if (!WideCharToMultiByte( CP_ACP
, 0, p
, -1, buf
, sizeof(buf
), NULL
, NULL
)) buf
[sizeof(buf
)-1] = 0;
59 if (CompareStringA(GetThreadLocale(), NORM_IGNORECASE
|LOCALE_USE_CP_ACP
, buf
, min(envlen
, varlen
), var
, varlen
) == CSTR_EQUAL
)
61 if (buf
[varlen
] == '=')
64 *result
= HeapAlloc(GetProcessHeap(), 0, buflen
+ 1);
65 if (!*result
) return FALSE
;
66 memcpy(*result
, buf
, buflen
+ 1);
76 static void test_create_env(void)
78 BOOL r
, is_wow64
= FALSE
;
81 char * st
, systemroot
[100];
84 static const struct profile_item common_vars
[] = {
87 { "NUMBER_OF_PROCESSORS" },
89 { "PROCESSOR_ARCHITECTURE" },
90 { "PROCESSOR_IDENTIFIER" },
91 { "PROCESSOR_LEVEL" },
92 { "PROCESSOR_REVISION" },
97 static const struct profile_item common_post_nt4_vars
[] = {
98 { "ALLUSERSPROFILE" },
101 { "CommonProgramFiles" },
106 static const struct profile_item common_win64_vars
[] = {
108 { "CommonProgramW6432" }
111 r
= SetEnvironmentVariableA("WINE_XYZZY", "ZZYZX");
114 r
= GetEnvironmentVariableA("SystemRoot", systemroot
, sizeof(systemroot
));
115 ok(r
!= 0, "GetEnvironmentVariable failed (%d)\n", GetLastError());
117 r
= SetEnvironmentVariableA("SystemRoot", "overwrite");
123 r
= CreateEnvironmentBlock(NULL
, NULL
, FALSE
);
127 r
= OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY
|TOKEN_DUPLICATE
, &htok
);
133 r
= CreateEnvironmentBlock(NULL
, htok
, FALSE
);
137 r
= CreateEnvironmentBlock((LPVOID
) &env
[0], NULL
, FALSE
);
140 r
= CreateEnvironmentBlock((LPVOID
) &env
[1], htok
, FALSE
);
143 r
= CreateEnvironmentBlock((LPVOID
) &env
[2], NULL
, TRUE
);
146 r
= CreateEnvironmentBlock((LPVOID
) &env
[3], htok
, TRUE
);
149 r
= SetEnvironmentVariableA("SystemRoot", systemroot
);
154 r
= get_env(env
[i
], "SystemRoot", &st
);
155 ok(!strcmp(st
, "SystemRoot=overwrite"), "%s\n", st
);
159 /* Test for common environment variables (NT4 and higher) */
160 for (i
= 0; i
< sizeof(common_vars
)/sizeof(common_vars
[0]); i
++)
162 for (j
= 0; j
< 4; j
++)
164 r
= get_env(env
[j
], common_vars
[i
].name
, &st
);
165 expect_env(TRUE
, r
, common_vars
[i
].name
);
166 if (r
) HeapFree(GetProcessHeap(), 0, st
);
170 /* Test for common environment variables (post NT4) */
171 if (!GetEnvironmentVariableA("ALLUSERSPROFILE", NULL
, 0))
173 win_skip("Some environment variables are not present on NT4\n");
177 for (i
= 0; i
< sizeof(common_post_nt4_vars
)/sizeof(common_post_nt4_vars
[0]); i
++)
179 for (j
= 0; j
< 4; j
++)
181 r
= get_env(env
[j
], common_post_nt4_vars
[i
].name
, &st
);
182 expect_env(TRUE
, r
, common_post_nt4_vars
[i
].name
);
183 if (r
) HeapFree(GetProcessHeap(), 0, st
);
189 pIsWow64Process(GetCurrentProcess(), &is_wow64
);
190 if (sizeof(void*)==8 || is_wow64
)
192 for (i
= 0; i
< sizeof(common_win64_vars
)/sizeof(common_win64_vars
[0]); i
++)
196 r
= get_env(env
[j
], common_win64_vars
[i
].name
, &st
);
197 ok(r
|| broken(!r
)/* Vista,2k3,XP */, "Expected 1, got 0 for %s\n", common_win64_vars
[i
].name
);
198 if (r
) HeapFree(GetProcessHeap(), 0, st
);
203 r
= get_env(env
[0], "WINE_XYZZY", &st
);
206 r
= get_env(env
[1], "WINE_XYZZY", &st
);
209 r
= get_env(env
[2], "WINE_XYZZY", &st
);
211 if (r
) HeapFree(GetProcessHeap(), 0, st
);
213 r
= get_env(env
[3], "WINE_XYZZY", &st
);
215 if (r
) HeapFree(GetProcessHeap(), 0, st
);
217 for (i
= 0; i
< sizeof(env
) / sizeof(env
[0]); i
++)
219 r
= DestroyEnvironmentBlock(env
[i
]);
224 static void test_get_profiles_dir(void)
226 static const char ProfileListA
[] = "Software\\Microsoft\\Windows NT\\CurrentVersion\\ProfileList";
227 static const char ProfilesDirectory
[] = "ProfilesDirectory";
229 DWORD cch
, profiles_len
;
232 char *profiles_dir
, *buf
, small_buf
[1];
234 l
= RegOpenKeyExA(HKEY_LOCAL_MACHINE
, ProfileListA
, 0, KEY_READ
, &key
);
235 ok(!l
, "RegOpenKeyExA failed: %d\n", GetLastError());
237 l
= RegQueryValueExA(key
, ProfilesDirectory
, NULL
, NULL
, NULL
, &cch
);
240 win_skip("No ProfilesDirectory value (NT4), skipping tests\n");
243 buf
= HeapAlloc(GetProcessHeap(), 0, cch
);
244 RegQueryValueExA(key
, ProfilesDirectory
, NULL
, NULL
, (BYTE
*)buf
, &cch
);
246 profiles_len
= ExpandEnvironmentStringsA(buf
, NULL
, 0);
247 profiles_dir
= HeapAlloc(GetProcessHeap(), 0, profiles_len
);
248 ExpandEnvironmentStringsA(buf
, profiles_dir
, profiles_len
);
249 HeapFree(GetProcessHeap(), 0, buf
);
251 SetLastError(0xdeadbeef);
252 r
= GetProfilesDirectoryA(NULL
, NULL
);
254 expect_gle(ERROR_INVALID_PARAMETER
);
255 SetLastError(0xdeadbeef);
256 r
= GetProfilesDirectoryA(NULL
, &cch
);
258 expect_gle(ERROR_INVALID_PARAMETER
);
259 SetLastError(0xdeadbeef);
261 r
= GetProfilesDirectoryA(small_buf
, &cch
);
263 expect_gle(ERROR_INSUFFICIENT_BUFFER
);
264 /* MSDN claims the returned character count includes the NULL terminator
265 * when the buffer is too small, but that's not in fact what gets returned.
267 ok(cch
== profiles_len
- 1, "expected %d, got %d\n", profiles_len
- 1, cch
);
268 /* Allocate one more character than the return value to prevent a buffer
271 buf
= HeapAlloc(GetProcessHeap(), 0, cch
+ 1);
272 r
= GetProfilesDirectoryA(buf
, &cch
);
273 /* Rather than a BOOL, the return value is also the number of characters
274 * stored in the buffer.
276 expect(profiles_len
- 1, r
);
277 ok(!strcmp(buf
, profiles_dir
), "expected %s, got %s\n", profiles_dir
, buf
);
279 HeapFree(GetProcessHeap(), 0, buf
);
280 HeapFree(GetProcessHeap(), 0, profiles_dir
);
282 SetLastError(0xdeadbeef);
283 r
= GetProfilesDirectoryW(NULL
, NULL
);
285 expect_gle(ERROR_INVALID_PARAMETER
);
288 SetLastError(0xdeadbeef);
289 r
= GetProfilesDirectoryW(NULL
, &cch
);
291 expect_gle(ERROR_INSUFFICIENT_BUFFER
);
292 ok(cch
, "expected cch > 0\n");
294 SetLastError(0xdeadbeef);
295 r
= GetProfilesDirectoryW(NULL
, &cch
);
297 expect_gle(ERROR_INSUFFICIENT_BUFFER
);
300 static void test_get_user_profile_dir(void)
308 if (!GetEnvironmentVariableA( "ALLUSERSPROFILE", NULL
, 0 ))
310 win_skip("Skipping tests on NT4\n");
314 ret
= OpenProcessToken( GetCurrentProcess(), TOKEN_QUERY
, &token
);
315 ok(ret
, "expected success %u\n", GetLastError());
317 SetLastError( 0xdeadbeef );
318 ret
= GetUserProfileDirectoryA( NULL
, NULL
, NULL
);
319 error
= GetLastError();
320 ok(!ret
, "expected failure\n");
321 ok(error
== ERROR_INVALID_PARAMETER
, "expected ERROR_INVALID_PARAMETER, got %u\n", error
);
323 SetLastError( 0xdeadbeef );
324 ret
= GetUserProfileDirectoryA( token
, NULL
, NULL
);
325 error
= GetLastError();
326 ok(!ret
, "expected failure\n");
327 ok(error
== ERROR_INVALID_PARAMETER
, "expected ERROR_INVALID_PARAMETER, got %u\n", error
);
329 dirA
= HeapAlloc( GetProcessHeap(), 0, 32 );
330 SetLastError( 0xdeadbeef );
331 ret
= GetUserProfileDirectoryA( token
, dirA
, NULL
);
332 error
= GetLastError();
333 ok(!ret
, "expected failure\n");
334 ok(error
== ERROR_INVALID_PARAMETER
, "expected ERROR_INVALID_PARAMETER, got %u\n", error
);
335 HeapFree( GetProcessHeap(), 0, dirA
);
338 SetLastError( 0xdeadbeef );
339 ret
= GetUserProfileDirectoryA( token
, NULL
, &len
);
340 error
= GetLastError();
341 ok(!ret
, "expected failure\n");
342 ok(error
== ERROR_INVALID_PARAMETER
, "expected ERROR_INVALID_PARAMETER, got %u\n", error
);
343 ok(!len
, "expected 0, got %u\n", len
);
346 dirA
= HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY
, 32 );
347 SetLastError( 0xdeadbeef );
348 ret
= GetUserProfileDirectoryA( token
, dirA
, &len
);
349 error
= GetLastError();
350 ok(!ret
, "expected failure\n");
351 ok(error
== ERROR_INSUFFICIENT_BUFFER
, "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", error
);
352 ok(len
, "expected len > 0\n");
353 HeapFree( GetProcessHeap(), 0, dirA
);
355 dirA
= HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY
, len
);
356 SetLastError( 0xdeadbeef );
357 ret
= GetUserProfileDirectoryA( token
, dirA
, &len
);
358 ok(ret
, "expected success %u\n", GetLastError());
359 ok(len
, "expected len > 0\n");
360 ok(lstrlenA( dirA
) == len
- 1, "length mismatch %d != %d - 1\n", lstrlenA( dirA
), len
);
362 HeapFree( GetProcessHeap(), 0, dirA
);
364 SetLastError( 0xdeadbeef );
365 ret
= GetUserProfileDirectoryW( NULL
, NULL
, NULL
);
366 error
= GetLastError();
367 ok(!ret
, "expected failure\n");
368 todo_wine
ok(error
== ERROR_INVALID_HANDLE
, "expected ERROR_INVALID_HANDLE, got %u\n", error
);
370 SetLastError( 0xdeadbeef );
371 ret
= GetUserProfileDirectoryW( token
, NULL
, NULL
);
372 error
= GetLastError();
373 ok(!ret
, "expected failure\n");
374 ok(error
== ERROR_INVALID_PARAMETER
, "expected ERROR_INVALID_PARAMETER, got %u\n", error
);
376 dirW
= HeapAlloc( GetProcessHeap(), 0, 32 );
377 SetLastError( 0xdeadbeef );
378 ret
= GetUserProfileDirectoryW( token
, dirW
, NULL
);
379 error
= GetLastError();
380 ok(!ret
, "expected failure\n");
381 ok(error
== ERROR_INVALID_PARAMETER
, "expected ERROR_INVALID_PARAMETER, got %u\n", error
);
382 HeapFree( GetProcessHeap(), 0, dirW
);
385 SetLastError( 0xdeadbeef );
386 ret
= GetUserProfileDirectoryW( token
, NULL
, &len
);
387 error
= GetLastError();
388 ok(!ret
, "expected failure\n");
389 ok(error
== ERROR_INSUFFICIENT_BUFFER
, "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", error
);
390 ok(len
, "expected len > 0\n");
392 dirW
= HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY
, len
* sizeof(WCHAR
) );
393 SetLastError( 0xdeadbeef );
394 ret
= GetUserProfileDirectoryW( token
, dirW
, &len
);
395 ok(ret
, "expected success %u\n", GetLastError());
396 ok(len
, "expected len > 0\n");
397 ok(lstrlenW( dirW
) == len
- 1, "length mismatch %d != %d - 1\n", lstrlenW( dirW
), len
);
398 HeapFree( GetProcessHeap(), 0, dirW
);
400 CloseHandle( token
);
405 pIsWow64Process
= (void*)GetProcAddress(GetModuleHandleA("kernel32.dll"), "IsWow64Process");
408 test_get_profiles_dir();
409 test_get_user_profile_dir();