dlls/userenv/tests/userenv.c - wine - Git at Google

 /*
  * Unit test suite for userenv functions
  *
  * Copyright 2008 Google (Lei Zhang)
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
  * version 2.1 of the License, or (at your option) any later version.
  *
  * This library is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
  * License along with this library; if not, write to the Free Software
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  */

 #include <stdio.h>
 #include <stdlib.h>
 #include <stdarg.h>

 #include "windef.h"
 #include "winbase.h"
 #include "winnls.h"
 #include "winreg.h"

 #include "userenv.h"

 #include "wine/test.h"

 #define expect(EXPECTED,GOT) ok((GOT)==(EXPECTED), "Expected %d, got %d\n", (EXPECTED), (GOT))
 #define expect_env(EXPECTED,GOT,VAR) ok((GOT)==(EXPECTED), "Expected %d, got %d for %s (%d)\n", (EXPECTED), (GOT), (VAR), j)
 #define expect_gle(EXPECTED) ok(GetLastError() == (EXPECTED), "Expected %d, got %d\n", (EXPECTED), GetLastError())

 static BOOL (WINAPI *pIsWow64Process)(HANDLE,PBOOL);

 struct profile_item
 {
     const char * name;
 };

 /* Helper function for retrieving environment variables */
 static BOOL get_env(const WCHAR * env, const char * var, char ** result)
 {
     const WCHAR * p = env;
     int envlen, varlen, buflen;
     char buf[256];

     if (!env || !var || !result) return FALSE;

     varlen = strlen(var);
     do
     {
         if (!WideCharToMultiByte( CP_ACP, 0, p, -1, buf, sizeof(buf), NULL, NULL )) buf[sizeof(buf)-1] = 0;
         envlen = strlen(buf);
         if (CompareStringA(GetThreadLocale(), NORM_IGNORECASE|LOCALE_USE_CP_ACP, buf, min(envlen, varlen), var, varlen) == CSTR_EQUAL)
         {
             if (buf[varlen] == '=')
             {
                 buflen = strlen(buf);
                 *result = HeapAlloc(GetProcessHeap(), 0, buflen + 1);
                 if (!*result) return FALSE;
                 memcpy(*result, buf, buflen + 1);
                 return TRUE;
             }
         }
         while (*p) p++;
         p++;
     } while (*p);
     return FALSE;
 }

 static void test_create_env(void)
 {
     BOOL r, is_wow64 = FALSE;
     HANDLE htok;
     WCHAR * env[4];
     char * st, systemroot[100];
     int i, j;

     static const struct profile_item common_vars[] = {
         { "ComSpec" },
         { "COMPUTERNAME" },
         { "NUMBER_OF_PROCESSORS" },
         { "OS" },
         { "PROCESSOR_ARCHITECTURE" },
         { "PROCESSOR_IDENTIFIER" },
         { "PROCESSOR_LEVEL" },
         { "PROCESSOR_REVISION" },
         { "SystemDrive" },
         { "SystemRoot" },
         { "windir" }
     };
     static const struct profile_item common_post_nt4_vars[] = {
         { "ALLUSERSPROFILE" },
         { "TEMP" },
         { "TMP" },
         { "CommonProgramFiles" },
         { "ProgramFiles" },
         { "PATH" },
         { "USERPROFILE" }
     };
     static const struct profile_item common_win64_vars[] = {
         { "ProgramW6432" },
         { "CommonProgramW6432" }
     };

     r = SetEnvironmentVariableA("WINE_XYZZY", "ZZYZX");
     expect(TRUE, r);

     r = GetEnvironmentVariableA("SystemRoot", systemroot, sizeof(systemroot));
     ok(r != 0, "GetEnvironmentVariable failed (%d)\n", GetLastError());

     r = SetEnvironmentVariableA("SystemRoot", "overwrite");
     expect(TRUE, r);

     if (0)
     {
         /* Crashes on NT4 */
         r = CreateEnvironmentBlock(NULL, NULL, FALSE);
         expect(FALSE, r);
     }

     r = OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY|TOKEN_DUPLICATE, &htok);
     expect(TRUE, r);

     if (0)
     {
         /* Crashes on NT4 */
         r = CreateEnvironmentBlock(NULL, htok, FALSE);
         expect(FALSE, r);
     }

     r = CreateEnvironmentBlock((LPVOID) &env[0], NULL, FALSE);
     expect(TRUE, r);

     r = CreateEnvironmentBlock((LPVOID) &env[1], htok, FALSE);
     expect(TRUE, r);

     r = CreateEnvironmentBlock((LPVOID) &env[2], NULL, TRUE);
     expect(TRUE, r);

     r = CreateEnvironmentBlock((LPVOID) &env[3], htok, TRUE);
     expect(TRUE, r);

     r = SetEnvironmentVariableA("SystemRoot", systemroot);
     expect(TRUE, r);

     for(i=0; i<4; i++)
     {
         r = get_env(env[i], "SystemRoot", &st);
         ok(!strcmp(st, "SystemRoot=overwrite"), "%s\n", st);
         expect(TRUE, r);
         HeapFree(GetProcessHeap(), 0, st);
     }

     /* Test for common environment variables (NT4 and higher) */
     for (i = 0; i < sizeof(common_vars)/sizeof(common_vars[0]); i++)
     {
         for (j = 0; j < 4; j++)
         {
             r = get_env(env[j], common_vars[i].name, &st);
             expect_env(TRUE, r, common_vars[i].name);
             if (r) HeapFree(GetProcessHeap(), 0, st);
         }
     }

     /* Test for common environment variables (post NT4) */
     if (!GetEnvironmentVariableA("ALLUSERSPROFILE", NULL, 0))
     {
         win_skip("Some environment variables are not present on NT4\n");
     }
     else
     {
         for (i = 0; i < sizeof(common_post_nt4_vars)/sizeof(common_post_nt4_vars[0]); i++)
         {
             for (j = 0; j < 4; j++)
             {
                 r = get_env(env[j], common_post_nt4_vars[i].name, &st);
                 expect_env(TRUE, r, common_post_nt4_vars[i].name);
                 if (r) HeapFree(GetProcessHeap(), 0, st);
             }
         }
     }

     if(pIsWow64Process)
         pIsWow64Process(GetCurrentProcess(), &is_wow64);
     if (sizeof(void*)==8 || is_wow64)
     {
         for (i = 0; i < sizeof(common_win64_vars)/sizeof(common_win64_vars[0]); i++)
         {
             for (j=0; j<4; j++)
             {
                 r = get_env(env[j], common_win64_vars[i].name, &st);
                 ok(r || broken(!r)/* Vista,2k3,XP */, "Expected 1, got 0 for %s\n", common_win64_vars[i].name);
                 if (r) HeapFree(GetProcessHeap(), 0, st);
             }
         }
     }

     r = get_env(env[0], "WINE_XYZZY", &st);
     expect(FALSE, r);

     r = get_env(env[1], "WINE_XYZZY", &st);
     expect(FALSE, r);

     r = get_env(env[2], "WINE_XYZZY", &st);
     expect(TRUE, r);
     if (r) HeapFree(GetProcessHeap(), 0, st);

     r = get_env(env[3], "WINE_XYZZY", &st);
     expect(TRUE, r);
     if (r) HeapFree(GetProcessHeap(), 0, st);

     for (i = 0; i < sizeof(env) / sizeof(env[0]); i++)
     {
         r = DestroyEnvironmentBlock(env[i]);
         expect(TRUE, r);
     }
 }

 static void test_get_profiles_dir(void)
 {
     static const char ProfileListA[] = "Software\\Microsoft\\Windows NT\\CurrentVersion\\ProfileList";
     static const char ProfilesDirectory[] = "ProfilesDirectory";
     BOOL r;
     DWORD cch, profiles_len;
     LONG l;
     HKEY key;
     char *profiles_dir, *buf, small_buf[1];

     l = RegOpenKeyExA(HKEY_LOCAL_MACHINE, ProfileListA, 0, KEY_READ, &key);
     ok(!l, "RegOpenKeyExA failed: %d\n", GetLastError());

     l = RegQueryValueExA(key, ProfilesDirectory, NULL, NULL, NULL, &cch);
     if (l)
     {
         win_skip("No ProfilesDirectory value (NT4), skipping tests\n");
         return;
     }
     buf = HeapAlloc(GetProcessHeap(), 0, cch);
     RegQueryValueExA(key, ProfilesDirectory, NULL, NULL, (BYTE *)buf, &cch);
     RegCloseKey(key);
     profiles_len = ExpandEnvironmentStringsA(buf, NULL, 0);
     profiles_dir = HeapAlloc(GetProcessHeap(), 0, profiles_len);
     ExpandEnvironmentStringsA(buf, profiles_dir, profiles_len);
     HeapFree(GetProcessHeap(), 0, buf);

     SetLastError(0xdeadbeef);
     r = GetProfilesDirectoryA(NULL, NULL);
     expect(FALSE, r);
     expect_gle(ERROR_INVALID_PARAMETER);
     SetLastError(0xdeadbeef);
     r = GetProfilesDirectoryA(NULL, &cch);
     expect(FALSE, r);
     expect_gle(ERROR_INVALID_PARAMETER);
     SetLastError(0xdeadbeef);
     cch = 1;
     r = GetProfilesDirectoryA(small_buf, &cch);
     expect(FALSE, r);
     expect_gle(ERROR_INSUFFICIENT_BUFFER);
     /* MSDN claims the returned character count includes the NULL terminator
      * when the buffer is too small, but that's not in fact what gets returned.
      */
     ok(cch == profiles_len - 1, "expected %d, got %d\n", profiles_len - 1, cch);
     /* Allocate one more character than the return value to prevent a buffer
      * overrun.
      */
     buf = HeapAlloc(GetProcessHeap(), 0, cch + 1);
     r = GetProfilesDirectoryA(buf, &cch);
     /* Rather than a BOOL, the return value is also the number of characters
      * stored in the buffer.
      */
     expect(profiles_len - 1, r);
     ok(!strcmp(buf, profiles_dir), "expected %s, got %s\n", profiles_dir, buf);

     HeapFree(GetProcessHeap(), 0, buf);
     HeapFree(GetProcessHeap(), 0, profiles_dir);

     SetLastError(0xdeadbeef);
     r = GetProfilesDirectoryW(NULL, NULL);
     expect(FALSE, r);
     expect_gle(ERROR_INVALID_PARAMETER);

     cch = 0;
     SetLastError(0xdeadbeef);
     r = GetProfilesDirectoryW(NULL, &cch);
     expect(FALSE, r);
     expect_gle(ERROR_INSUFFICIENT_BUFFER);
     ok(cch, "expected cch > 0\n");

     SetLastError(0xdeadbeef);
     r = GetProfilesDirectoryW(NULL, &cch);
     expect(FALSE, r);
     expect_gle(ERROR_INSUFFICIENT_BUFFER);
 }

 static void test_get_user_profile_dir(void)
 {
     BOOL ret;
     DWORD error, len;
     HANDLE token;
     char *dirA;
     WCHAR *dirW;

     if (!GetEnvironmentVariableA( "ALLUSERSPROFILE", NULL, 0 ))
     {
         win_skip("Skipping tests on NT4\n");
         return;
     }

     ret = OpenProcessToken( GetCurrentProcess(), TOKEN_QUERY, &token );
     ok(ret, "expected success %u\n", GetLastError());

     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryA( NULL, NULL, NULL );
     error = GetLastError();
     ok(!ret, "expected failure\n");
     ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);

     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryA( token, NULL, NULL );
     error = GetLastError();
     ok(!ret, "expected failure\n");
     ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);

     dirA = HeapAlloc( GetProcessHeap(), 0, 32 );
     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryA( token, dirA, NULL );
     error = GetLastError();
     ok(!ret, "expected failure\n");
     ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);
     HeapFree( GetProcessHeap(), 0, dirA );

     len = 0;
     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryA( token, NULL, &len );
     error = GetLastError();
     ok(!ret, "expected failure\n");
     ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);
     ok(!len, "expected 0, got %u\n", len);

     len = 0;
     dirA = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, 32 );
     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryA( token, dirA, &len );
     error = GetLastError();
     ok(!ret, "expected failure\n");
     ok(error == ERROR_INSUFFICIENT_BUFFER, "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", error);
     ok(len, "expected len > 0\n");
     HeapFree( GetProcessHeap(), 0, dirA );

     dirA = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, len );
     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryA( token, dirA, &len );
     ok(ret, "expected success %u\n", GetLastError());
     ok(len, "expected len > 0\n");
     ok(lstrlenA( dirA ) == len - 1, "length mismatch %d != %d - 1\n", lstrlenA( dirA ), len );
     trace("%s\n", dirA);
     HeapFree( GetProcessHeap(), 0, dirA );

     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryW( NULL, NULL, NULL );
     error = GetLastError();
     ok(!ret, "expected failure\n");
     todo_wine ok(error == ERROR_INVALID_HANDLE, "expected ERROR_INVALID_HANDLE, got %u\n", error);

     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryW( token, NULL, NULL );
     error = GetLastError();
     ok(!ret, "expected failure\n");
     ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);

     dirW = HeapAlloc( GetProcessHeap(), 0, 32 );
     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryW( token, dirW, NULL );
     error = GetLastError();
     ok(!ret, "expected failure\n");
     ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);
     HeapFree( GetProcessHeap(), 0, dirW );

     len = 0;
     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryW( token, NULL, &len );
     error = GetLastError();
     ok(!ret, "expected failure\n");
     ok(error == ERROR_INSUFFICIENT_BUFFER, "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", error);
     ok(len, "expected len > 0\n");

     dirW = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR) );
     SetLastError( 0xdeadbeef );
     ret = GetUserProfileDirectoryW( token, dirW, &len );
     ok(ret, "expected success %u\n", GetLastError());
     ok(len, "expected len > 0\n");
     ok(lstrlenW( dirW ) == len - 1, "length mismatch %d != %d - 1\n", lstrlenW( dirW ), len );
     HeapFree( GetProcessHeap(), 0, dirW );

     CloseHandle( token );
 }

 START_TEST(userenv)
 {
     pIsWow64Process = (void*)GetProcAddress(GetModuleHandleA("kernel32.dll"), "IsWow64Process");

     test_create_env();
     test_get_profiles_dir();
     test_get_user_profile_dir();
 }
	/*
	* Unit test suite for userenv functions
	*
	* Copyright 2008 Google (Lei Zhang)
	*
	* This library is free software; you can redistribute it and/or
	* modify it under the terms of the GNU Lesser General Public
	* License as published by the Free Software Foundation; either
	* version 2.1 of the License, or (at your option) any later version.
	*
	* This library is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	* Lesser General Public License for more details.
	*
	* You should have received a copy of the GNU Lesser General Public
	* License along with this library; if not, write to the Free Software
	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
	*/

	#include <stdio.h>
	#include <stdlib.h>
	#include <stdarg.h>

	#include "windef.h"
	#include "winbase.h"
	#include "winnls.h"
	#include "winreg.h"

	#include "userenv.h"

	#include "wine/test.h"

	#define expect(EXPECTED,GOT) ok((GOT)==(EXPECTED), "Expected %d, got %d\n", (EXPECTED), (GOT))
	#define expect_env(EXPECTED,GOT,VAR) ok((GOT)==(EXPECTED), "Expected %d, got %d for %s (%d)\n", (EXPECTED), (GOT), (VAR), j)
	#define expect_gle(EXPECTED) ok(GetLastError() == (EXPECTED), "Expected %d, got %d\n", (EXPECTED), GetLastError())

	static BOOL (WINAPI *pIsWow64Process)(HANDLE,PBOOL);

	struct profile_item
	{
	const char * name;
	};

	/* Helper function for retrieving environment variables */
	static BOOL get_env(const WCHAR * env, const char * var, char ** result)
	{
	const WCHAR * p = env;
	int envlen, varlen, buflen;
	char buf[256];

	if (!env \|\| !var \|\| !result) return FALSE;

	varlen = strlen(var);
	do
	{
	if (!WideCharToMultiByte( CP_ACP, 0, p, -1, buf, sizeof(buf), NULL, NULL )) buf[sizeof(buf)-1] = 0;
	envlen = strlen(buf);
	if (CompareStringA(GetThreadLocale(), NORM_IGNORECASE\|LOCALE_USE_CP_ACP, buf, min(envlen, varlen), var, varlen) == CSTR_EQUAL)
	{
	if (buf[varlen] == '=')
	{
	buflen = strlen(buf);
	*result = HeapAlloc(GetProcessHeap(), 0, buflen + 1);
	if (!*result) return FALSE;
	memcpy(*result, buf, buflen + 1);
	return TRUE;
	}
	}
	while (*p) p++;
	p++;
	} while (*p);
	return FALSE;
	}

	static void test_create_env(void)
	{
	BOOL r, is_wow64 = FALSE;
	HANDLE htok;
	WCHAR * env[4];
	char * st, systemroot[100];
	int i, j;

	static const struct profile_item common_vars[] = {
	{ "ComSpec" },
	{ "COMPUTERNAME" },
	{ "NUMBER_OF_PROCESSORS" },
	{ "OS" },
	{ "PROCESSOR_ARCHITECTURE" },
	{ "PROCESSOR_IDENTIFIER" },
	{ "PROCESSOR_LEVEL" },
	{ "PROCESSOR_REVISION" },
	{ "SystemDrive" },
	{ "SystemRoot" },
	{ "windir" }
	};
	static const struct profile_item common_post_nt4_vars[] = {
	{ "ALLUSERSPROFILE" },
	{ "TEMP" },
	{ "TMP" },
	{ "CommonProgramFiles" },
	{ "ProgramFiles" },
	{ "PATH" },
	{ "USERPROFILE" }
	};
	static const struct profile_item common_win64_vars[] = {
	{ "ProgramW6432" },
	{ "CommonProgramW6432" }
	};

	r = SetEnvironmentVariableA("WINE_XYZZY", "ZZYZX");
	expect(TRUE, r);

	r = GetEnvironmentVariableA("SystemRoot", systemroot, sizeof(systemroot));
	ok(r != 0, "GetEnvironmentVariable failed (%d)\n", GetLastError());

	r = SetEnvironmentVariableA("SystemRoot", "overwrite");
	expect(TRUE, r);

	if (0)
	{
	/* Crashes on NT4 */
	r = CreateEnvironmentBlock(NULL, NULL, FALSE);
	expect(FALSE, r);
	}

	r = OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY\|TOKEN_DUPLICATE, &htok);
	expect(TRUE, r);

	if (0)
	{
	/* Crashes on NT4 */
	r = CreateEnvironmentBlock(NULL, htok, FALSE);
	expect(FALSE, r);
	}

	r = CreateEnvironmentBlock((LPVOID) &env[0], NULL, FALSE);
	expect(TRUE, r);

	r = CreateEnvironmentBlock((LPVOID) &env[1], htok, FALSE);
	expect(TRUE, r);

	r = CreateEnvironmentBlock((LPVOID) &env[2], NULL, TRUE);
	expect(TRUE, r);

	r = CreateEnvironmentBlock((LPVOID) &env[3], htok, TRUE);
	expect(TRUE, r);

	r = SetEnvironmentVariableA("SystemRoot", systemroot);
	expect(TRUE, r);

	for(i=0; i<4; i++)
	{
	r = get_env(env[i], "SystemRoot", &st);
	ok(!strcmp(st, "SystemRoot=overwrite"), "%s\n", st);
	expect(TRUE, r);
	HeapFree(GetProcessHeap(), 0, st);
	}

	/* Test for common environment variables (NT4 and higher) */
	for (i = 0; i < sizeof(common_vars)/sizeof(common_vars[0]); i++)
	{
	for (j = 0; j < 4; j++)
	{
	r = get_env(env[j], common_vars[i].name, &st);
	expect_env(TRUE, r, common_vars[i].name);
	if (r) HeapFree(GetProcessHeap(), 0, st);
	}
	}

	/* Test for common environment variables (post NT4) */
	if (!GetEnvironmentVariableA("ALLUSERSPROFILE", NULL, 0))
	{
	win_skip("Some environment variables are not present on NT4\n");
	}
	else
	{
	for (i = 0; i < sizeof(common_post_nt4_vars)/sizeof(common_post_nt4_vars[0]); i++)
	{
	for (j = 0; j < 4; j++)
	{
	r = get_env(env[j], common_post_nt4_vars[i].name, &st);
	expect_env(TRUE, r, common_post_nt4_vars[i].name);
	if (r) HeapFree(GetProcessHeap(), 0, st);
	}
	}
	}

	if(pIsWow64Process)
	pIsWow64Process(GetCurrentProcess(), &is_wow64);
	if (sizeof(void*)==8 \|\| is_wow64)
	{
	for (i = 0; i < sizeof(common_win64_vars)/sizeof(common_win64_vars[0]); i++)
	{
	for (j=0; j<4; j++)
	{
	r = get_env(env[j], common_win64_vars[i].name, &st);
	ok(r \|\| broken(!r)/* Vista,2k3,XP */, "Expected 1, got 0 for %s\n", common_win64_vars[i].name);
	if (r) HeapFree(GetProcessHeap(), 0, st);
	}
	}
	}

	r = get_env(env[0], "WINE_XYZZY", &st);
	expect(FALSE, r);

	r = get_env(env[1], "WINE_XYZZY", &st);
	expect(FALSE, r);

	r = get_env(env[2], "WINE_XYZZY", &st);
	expect(TRUE, r);
	if (r) HeapFree(GetProcessHeap(), 0, st);

	r = get_env(env[3], "WINE_XYZZY", &st);
	expect(TRUE, r);
	if (r) HeapFree(GetProcessHeap(), 0, st);

	for (i = 0; i < sizeof(env) / sizeof(env[0]); i++)
	{
	r = DestroyEnvironmentBlock(env[i]);
	expect(TRUE, r);
	}
	}

	static void test_get_profiles_dir(void)
	{
	static const char ProfileListA[] = "Software\\Microsoft\\Windows NT\\CurrentVersion\\ProfileList";
	static const char ProfilesDirectory[] = "ProfilesDirectory";
	BOOL r;
	DWORD cch, profiles_len;
	LONG l;
	HKEY key;
	char profiles_dir, buf, small_buf[1];

	l = RegOpenKeyExA(HKEY_LOCAL_MACHINE, ProfileListA, 0, KEY_READ, &key);
	ok(!l, "RegOpenKeyExA failed: %d\n", GetLastError());

	l = RegQueryValueExA(key, ProfilesDirectory, NULL, NULL, NULL, &cch);
	if (l)
	{
	win_skip("No ProfilesDirectory value (NT4), skipping tests\n");
	return;
	}
	buf = HeapAlloc(GetProcessHeap(), 0, cch);
	RegQueryValueExA(key, ProfilesDirectory, NULL, NULL, (BYTE *)buf, &cch);
	RegCloseKey(key);
	profiles_len = ExpandEnvironmentStringsA(buf, NULL, 0);
	profiles_dir = HeapAlloc(GetProcessHeap(), 0, profiles_len);
	ExpandEnvironmentStringsA(buf, profiles_dir, profiles_len);
	HeapFree(GetProcessHeap(), 0, buf);

	SetLastError(0xdeadbeef);
	r = GetProfilesDirectoryA(NULL, NULL);
	expect(FALSE, r);
	expect_gle(ERROR_INVALID_PARAMETER);
	SetLastError(0xdeadbeef);
	r = GetProfilesDirectoryA(NULL, &cch);
	expect(FALSE, r);
	expect_gle(ERROR_INVALID_PARAMETER);
	SetLastError(0xdeadbeef);
	cch = 1;
	r = GetProfilesDirectoryA(small_buf, &cch);
	expect(FALSE, r);
	expect_gle(ERROR_INSUFFICIENT_BUFFER);
	/* MSDN claims the returned character count includes the NULL terminator
	* when the buffer is too small, but that's not in fact what gets returned.
	*/
	ok(cch == profiles_len - 1, "expected %d, got %d\n", profiles_len - 1, cch);
	/* Allocate one more character than the return value to prevent a buffer
	* overrun.
	*/
	buf = HeapAlloc(GetProcessHeap(), 0, cch + 1);
	r = GetProfilesDirectoryA(buf, &cch);
	/* Rather than a BOOL, the return value is also the number of characters
	* stored in the buffer.
	*/
	expect(profiles_len - 1, r);
	ok(!strcmp(buf, profiles_dir), "expected %s, got %s\n", profiles_dir, buf);

	HeapFree(GetProcessHeap(), 0, buf);
	HeapFree(GetProcessHeap(), 0, profiles_dir);

	SetLastError(0xdeadbeef);
	r = GetProfilesDirectoryW(NULL, NULL);
	expect(FALSE, r);
	expect_gle(ERROR_INVALID_PARAMETER);

	cch = 0;
	SetLastError(0xdeadbeef);
	r = GetProfilesDirectoryW(NULL, &cch);
	expect(FALSE, r);
	expect_gle(ERROR_INSUFFICIENT_BUFFER);
	ok(cch, "expected cch > 0\n");

	SetLastError(0xdeadbeef);
	r = GetProfilesDirectoryW(NULL, &cch);
	expect(FALSE, r);
	expect_gle(ERROR_INSUFFICIENT_BUFFER);
	}

	static void test_get_user_profile_dir(void)
	{
	BOOL ret;
	DWORD error, len;
	HANDLE token;
	char *dirA;
	WCHAR *dirW;

	if (!GetEnvironmentVariableA( "ALLUSERSPROFILE", NULL, 0 ))
	{
	win_skip("Skipping tests on NT4\n");
	return;
	}

	ret = OpenProcessToken( GetCurrentProcess(), TOKEN_QUERY, &token );
	ok(ret, "expected success %u\n", GetLastError());

	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryA( NULL, NULL, NULL );
	error = GetLastError();
	ok(!ret, "expected failure\n");
	ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);

	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryA( token, NULL, NULL );
	error = GetLastError();
	ok(!ret, "expected failure\n");
	ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);

	dirA = HeapAlloc( GetProcessHeap(), 0, 32 );
	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryA( token, dirA, NULL );
	error = GetLastError();
	ok(!ret, "expected failure\n");
	ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);
	HeapFree( GetProcessHeap(), 0, dirA );

	len = 0;
	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryA( token, NULL, &len );
	error = GetLastError();
	ok(!ret, "expected failure\n");
	ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);
	ok(!len, "expected 0, got %u\n", len);

	len = 0;
	dirA = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, 32 );
	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryA( token, dirA, &len );
	error = GetLastError();
	ok(!ret, "expected failure\n");
	ok(error == ERROR_INSUFFICIENT_BUFFER, "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", error);
	ok(len, "expected len > 0\n");
	HeapFree( GetProcessHeap(), 0, dirA );

	dirA = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, len );
	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryA( token, dirA, &len );
	ok(ret, "expected success %u\n", GetLastError());
	ok(len, "expected len > 0\n");
	ok(lstrlenA( dirA ) == len - 1, "length mismatch %d != %d - 1\n", lstrlenA( dirA ), len );
	trace("%s\n", dirA);
	HeapFree( GetProcessHeap(), 0, dirA );

	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryW( NULL, NULL, NULL );
	error = GetLastError();
	ok(!ret, "expected failure\n");
	todo_wine ok(error == ERROR_INVALID_HANDLE, "expected ERROR_INVALID_HANDLE, got %u\n", error);

	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryW( token, NULL, NULL );
	error = GetLastError();
	ok(!ret, "expected failure\n");
	ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);

	dirW = HeapAlloc( GetProcessHeap(), 0, 32 );
	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryW( token, dirW, NULL );
	error = GetLastError();
	ok(!ret, "expected failure\n");
	ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);
	HeapFree( GetProcessHeap(), 0, dirW );

	len = 0;
	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryW( token, NULL, &len );
	error = GetLastError();
	ok(!ret, "expected failure\n");
	ok(error == ERROR_INSUFFICIENT_BUFFER, "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", error);
	ok(len, "expected len > 0\n");

	dirW = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR) );
	SetLastError( 0xdeadbeef );
	ret = GetUserProfileDirectoryW( token, dirW, &len );
	ok(ret, "expected success %u\n", GetLastError());
	ok(len, "expected len > 0\n");
	ok(lstrlenW( dirW ) == len - 1, "length mismatch %d != %d - 1\n", lstrlenW( dirW ), len );
	HeapFree( GetProcessHeap(), 0, dirW );

	CloseHandle( token );
	}

	START_TEST(userenv)
	{
	pIsWow64Process = (void*)GetProcAddress(GetModuleHandleA("kernel32.dll"), "IsWow64Process");

	test_create_env();
	test_get_profiles_dir();
	test_get_user_profile_dir();
	}