shcore.c File Reference

#include <stdarg.h>
#include <windows.h>
#include "initguid.h"
#include "objidl.h"
#include "shlwapi.h"
#include "wine/test.h"

Include dependency graph for shcore.c:

Go to the source code of this file.

Classes
struct	test_unk

Macros
#define	COBJMACROS
#define	REG_TEST_KEY   "Software\\Wine\\Test"
#define	REG_CURRENT_VERSION   "Software\\Microsoft\\Windows\\CurrentVersion\\explorer"
#define	X(f)   p##f = (void*)GetProcAddress(hshcore, #f)
#define	CHECK_FILE_SIZE(filename, exp_size)   _check_file_size(filename, exp_size, __LINE__)
#define	CHECK_STREAM_SIZE(obj, exp_size)   _check_stream_size(obj, exp_size, __LINE__)
#define	CHECK_STREAM_POS(obj, exp_pos)   _check_stream_pos(obj, exp_pos, __LINE__)

Functions
static	HRESULT (WINAPI *pGetProcessReference)(IUnknown **)
static	void (WINAPI *pSetProcessReference)(IUnknown *)
static	int (WINAPI *pSHUnicodeToAnsi)(const WCHAR *
static	HKEY (WINAPI *pSHRegDuplicateHKey)(HKEY)
static	DWORD (WINAPI *pSHDeleteKeyA)(HKEY
static const char *static const char const char DWORD void DWORD *static	LSTATUS (WINAPI *pSHRegGetValueA)(HKEY
static void	init (HMODULE hshcore)
static HRESULT WINAPI	unk_QI (IUnknown *iface, REFIID riid, void **obj)
static struct test_unk *	impl_from_IUnknown (IUnknown *iface)
static ULONG WINAPI	unk_AddRef (IUnknown *iface)
static ULONG WINAPI	unk_Release (IUnknown *iface)
static void	test_unk_init (struct test_unk *testunk)
static void	test_process_reference (void)
static void	test_SHUnicodeToAnsi (void)
static void	test_SHAnsiToUnicode (void)
static void	test_SHAnsiToAnsi (void)
static void	test_SHUnicodeToUnicode (void)
static void	test_SHRegDuplicateHKey (void)
static void	test_SHDeleteKey (void)
static HKEY	create_test_entries (void)
static DWORD	delete_key (HKEY hkey, LPCSTR parent, LPCSTR keyname)
static void	test_SHGetValue (void)
static void	test_SHRegGetValue (void)
static void	test_SHQueryValueEx (void)
static void	test_SHRegGetPath (void)
static void	test_SHCopyKey (void)
static void	_check_file_size (const CHAR *filename, LONG exp_size, int line)
static void	_check_stream_size (IStream *obj, LONG exp_size, int line)
static void	_check_stream_pos (IStream *obj, LONG exp_pos, int line)
static void	test_stream_size (void)
	START_TEST (shcore)

Variables
static char	int
static const char *static const char const char DWORD void DWORD *static const char const char	SRRF
static const char *static const char const char DWORD void DWORD *static const char const char DWORD void DWORD *static const char DWORD DWORD void *	buff
static const char *static const char const char DWORD void DWORD *static const char const char DWORD void DWORD *static const char DWORD DWORD void DWORD *	buff_len
static const char const char char	DWORD
static const char	HKEY
static DWORD	mode
static DWORD IStream **	stream
static ULARGE_INTEGER *	size
static const char	test_path1 [] = "%LONGSYSTEMVAR%\\subdir1"
static const char	test_path2 [] = "%FOO%\\subdir1"
static const char *	test_envvar1 = "bar"
static const char *	test_envvar2 = "ImARatherLongButIndeedNeededString"
static char	test_exp_path1 [MAX_PATH]
static char	test_exp_path2 [MAX_PATH]
static DWORD	exp_len1
static DWORD	exp_len2
static const char *	initial_buffer ="0123456789"
static const IUnknownVtbl	testunkvtbl

Macro Definition Documentation

CHECK_FILE_SIZE

#define CHECK_FILE_SIZE	(		filename,
			exp_size
	)		_check_file_size(filename, exp_size, __LINE__)

Definition at line 666 of file shcore.c.

CHECK_STREAM_POS

#define CHECK_STREAM_POS	(		obj,
			exp_pos
	)		_check_stream_pos(obj, exp_pos, __LINE__)

Definition at line 694 of file shcore.c.

CHECK_STREAM_SIZE

#define CHECK_STREAM_SIZE	(		obj,
			exp_size
	)		_check_stream_size(obj, exp_size, __LINE__)

Definition at line 677 of file shcore.c.

COBJMACROS

#define COBJMACROS

Definition at line 19 of file shcore.c.

REG_CURRENT_VERSION

#define REG_CURRENT_VERSION   "Software\\Microsoft\\Windows\\CurrentVersion\\explorer"

Definition at line 49 of file shcore.c.

REG_TEST_KEY

#define REG_TEST_KEY   "Software\\Wine\\Test"

Definition at line 48 of file shcore.c.

X

#define X

(

f

)

p##f = (void*)GetProcAddress(hshcore, #f)

Function Documentation

_check_file_size()

static void _check_file_size ( const CHAR *  filename, LONG  exp_size, int  line  )

static

Definition at line 667 of file shcore.c.

{
    HANDLE handle;
    DWORD file_size = 0xdeadbeef;
    handle = CreateFileA(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
    file_size = GetFileSize(handle, NULL);
    ok_(__FILE__,line)(file_size == exp_size, "got wrong file size: %ld.\n", file_size);
    CloseHandle(handle);
}

_check_stream_pos()

static void _check_stream_pos ( IStream *  obj, LONG  exp_pos, int  line  )

static

Definition at line 695 of file shcore.c.

{
    LARGE_INTEGER move;
    ULARGE_INTEGER pos;
    HRESULT hr;
    move.QuadPart = 0;
    pos.QuadPart = 0xdeadbeef;
    hr = IStream_Seek(obj, move, STREAM_SEEK_CUR, &pos);
    ok_(__FILE__,line)(hr == S_OK, "IStream_Seek failed: hr %#lx.\n", hr);
    ok_(__FILE__,line)(pos.QuadPart == exp_pos, "got wrong position: %s.\n",
                       wine_dbgstr_longlong(pos.QuadPart));
}

_check_stream_size()

static void _check_stream_size ( IStream *  obj, LONG  exp_size, int  line  )

static

Definition at line 678 of file shcore.c.

{
    ULARGE_INTEGER stream_size;
    STATSTG stat;
    HRESULT hr;
    stream_size.QuadPart = 0xdeadbeef;
    hr = pIStream_Size(obj, &stream_size);
    ok_(__FILE__,line)(hr == S_OK, "IStream_Size failed: hr %#lx.\n", hr);
    ok_(__FILE__,line)(stream_size.QuadPart == exp_size, "Size(): got wrong size of stream: %s.\n",
                       wine_dbgstr_longlong(stream_size.QuadPart));
    hr = IStream_Stat(obj, &stat, STATFLAG_NONAME);
    ok_(__FILE__,line)(hr == S_OK, "IStream_Stat failed: hr %#lx.\n", hr);
    ok_(__FILE__,line)(stat.cbSize.QuadPart == exp_size, "Stat(): got wrong size of stream: %s.\n",
                       wine_dbgstr_longlong(stat.cbSize.QuadPart));
}

create_test_entries()

static HKEY create_test_entries ( void  )

static

Definition at line 370 of file shcore.c.

{
    HKEY hKey;
    DWORD ret;
    DWORD nExpectedLen1, nExpectedLen2;
 
    SetEnvironmentVariableA("LONGSYSTEMVAR", test_envvar1);
    SetEnvironmentVariableA("FOO", test_envvar2);
 
    ret = RegCreateKeyA(HKEY_CURRENT_USER, REG_TEST_KEY, &hKey);
    ok(!ret, "Unexpected retval %lu.\n", ret);
 
    if (hKey)
    {
        ok(!RegSetValueExA(hKey, "Test1", 0, REG_EXPAND_SZ, (BYTE *)test_path1, strlen(test_path1)+1), "RegSetValueExA failed\n");
        ok(!RegSetValueExA(hKey, "Test2", 0, REG_SZ, (BYTE *)test_path1, strlen(test_path1)+1), "RegSetValueExA failed\n");
        ok(!RegSetValueExA(hKey, "Test3", 0, REG_EXPAND_SZ, (BYTE *)test_path2, strlen(test_path2)+1), "RegSetValueExA failed\n");
    }
 
    exp_len1 = ExpandEnvironmentStringsA(test_path1, test_exp_path1, sizeof(test_exp_path1));
    exp_len2 = ExpandEnvironmentStringsA(test_path2, test_exp_path2, sizeof(test_exp_path2));
 
    nExpectedLen1 = strlen(test_path1) - strlen("%LONGSYSTEMVAR%") + strlen(test_envvar1) + 1;
    nExpectedLen2 = strlen(test_path2) - strlen("%FOO%") + strlen(test_envvar2) + 1;
 
    /* Make sure we carry on with correct values */
    exp_len1 = nExpectedLen1;
    exp_len2 = nExpectedLen2;
 
    return hKey;
}

Referenced by test_SHCopyKey(), test_SHGetValue(), test_SHQueryValueEx(), test_SHRegGetPath(), and test_SHRegGetValue().

delete_key()

static DWORD delete_key ( HKEY  hkey, LPCSTR  parent, LPCSTR  keyname  )

static

Definition at line 403 of file shcore.c.

{
    HKEY parentKey;
    DWORD ret;
 
    RegCloseKey(hkey);
 
    /* open the parent of the key to close */
    ret = RegOpenKeyExA( HKEY_CURRENT_USER, parent, 0, KEY_ALL_ACCESS, &parentKey);
    if (ret != ERROR_SUCCESS)
        return ret;
 
    ret = pSHDeleteKeyA( parentKey, keyname );
    RegCloseKey(parentKey);
 
    return ret;
}

DWORD()

static DWORD ( WINAPI *  pSHDeleteKeyA )

static

HKEY()

static HKEY ( WINAPI *  pSHRegDuplicateHKey )

static

HRESULT()

static HRESULT ( WINAPI *  pGetProcessReference )

static

impl_from_IUnknown()

static struct test_unk * impl_from_IUnknown ( IUnknown *  iface )

static

Definition at line 102 of file shcore.c.

{
    return CONTAINING_RECORD(iface, struct test_unk, IUnknown_iface);
}

Referenced by unk_AddRef(), and unk_Release().

init()

static void init ( HMODULE  hshcore )

static

Definition at line 62 of file shcore.c.

{
#define X(f) p##f = (void*)GetProcAddress(hshcore, #f)
    X(GetProcessReference);
    X(SetProcessReference);
    X(SHUnicodeToAnsi);
    X(SHAnsiToUnicode);
    X(SHAnsiToAnsi);
    X(SHUnicodeToUnicode);
    X(SHRegDuplicateHKey);
    X(SHDeleteKeyA);
    X(SHGetValueA);
    X(SHRegGetValueA);
    X(SHQueryValueExA);
    X(SHRegGetPathA);
    X(SHCopyKeyA);
    X(SHCreateStreamOnFileA);
    X(IStream_Size);
#undef X
}

int()

static int ( WINAPI *  pSHUnicodeToAnsi ) const

static

LSTATUS()

static const char *static const char const char DWORD void DWORD *static LSTATUS ( WINAPI *  pSHRegGetValueA )

static

Referenced by test_SHRegGetValue().

START_TEST()

START_TEST

(

shcore

)

Definition at line 780 of file shcore.c.

{
    HMODULE hshcore = LoadLibraryA("shcore.dll");
 
    if (!hshcore)
    {
        win_skip("Shcore.dll is not available.\n");
        return;
    }
 
    init(hshcore);
 
    test_process_reference();
    test_SHUnicodeToAnsi();
    test_SHAnsiToUnicode();
    test_SHAnsiToAnsi();
    test_SHUnicodeToUnicode();
    test_SHRegDuplicateHKey();
    test_SHDeleteKey();
    test_SHGetValue();
    test_SHRegGetValue();
    test_SHQueryValueEx();
    test_SHRegGetPath();
    test_SHCopyKey();
    test_stream_size();
}

test_process_reference()

static void test_process_reference ( void  )

static

Definition at line 132 of file shcore.c.

{
    struct test_unk test_unk, test_unk2;
    IUnknown *obj;
    HMODULE hmod;
    HRESULT hr;
 
    obj = (void *)0xdeadbeef;
    hr = pGetProcessReference(&obj);
    ok(hr == E_FAIL, "Unexpected hr %#lx.\n", hr);
    ok(obj == NULL, "Unexpected pointer.\n");
 
    test_unk_init(&test_unk);
    test_unk_init(&test_unk2);
 
    pSetProcessReference(&test_unk.IUnknown_iface);
    ok(test_unk.refcount == 1, "Unexpected refcount %lu.\n", test_unk.refcount);
    pSetProcessReference(&test_unk2.IUnknown_iface);
    ok(test_unk.refcount == 1, "Unexpected refcount %lu.\n", test_unk.refcount);
    ok(test_unk2.refcount == 1, "Unexpected refcount %lu.\n", test_unk2.refcount);
 
    hr = pGetProcessReference(&obj);
    ok(hr == S_OK, "Failed to get reference, hr %#lx.\n", hr);
    ok(obj == &test_unk2.IUnknown_iface, "Unexpected pointer.\n");
    ok(test_unk2.refcount == 2, "Unexpected refcount %lu.\n", test_unk2.refcount);
 
    hmod = LoadLibraryA("shell32.dll");
 
    pSHGetInstanceExplorer = (void *)GetProcAddress(hmod, "SHGetInstanceExplorer");
    hr = pSHGetInstanceExplorer(&obj);
    ok(hr == S_OK, "Failed to get reference, hr %#lx.\n", hr);
    ok(obj == &test_unk2.IUnknown_iface, "Unexpected pointer.\n");
    ok(test_unk2.refcount == 3, "Unexpected refcount %lu.\n", test_unk2.refcount);
}

Referenced by START_TEST().

test_SHAnsiToAnsi()

static void test_SHAnsiToAnsi ( void  )

static

Definition at line 253 of file shcore.c.

{
    char buff[16];
    int ret;
 
    ret = pSHAnsiToAnsi(NULL, NULL, 0);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
 
    strcpy(buff, "abcdefghijklm");
    ret = pSHAnsiToAnsi("test", buff, 3);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
    ok(!strcmp(buff, "te"), "Unexpected buffer contents.\n");
    ok(buff[3] == 'd', "Unexpected buffer contents.\n");
 
    strcpy(buff, "abcdefghijklm");
    ret = pSHAnsiToAnsi("", buff, 3);
    ok(ret == 1, "Unexpected return value %d.\n", ret);
    ok(!*buff, "Unexpected buffer contents.\n");
    ok(buff[3] == 'd', "Unexpected buffer contents.\n");
 
    strcpy(buff, "abcdefghijklm");
    ret = pSHAnsiToAnsi("test", buff, 4);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
    ok(!strcmp(buff, "tes"), "Unexpected buffer contents.\n");
    ok(buff[4] == 'e', "Unexpected buffer contents.\n");
 
    strcpy(buff, "abcdefghijklm");
    ret = pSHAnsiToAnsi("test", buff, 5);
    ok(ret == 5, "Unexpected return value %d.\n", ret);
    ok(!strcmp(buff, "test"), "Unexpected buffer contents.\n");
    ok(buff[5] == 'f', "Unexpected buffer contents.\n");
 
    strcpy(buff, "abcdefghijklm");
    ret = pSHAnsiToAnsi("test", buff, 6);
    ok(ret == 5, "Unexpected return value %d.\n", ret);
    ok(!strcmp(buff, "test"), "Unexpected buffer contents.\n");
    ok(buff[5] == 'f', "Unexpected buffer contents.\n");
}

Referenced by START_TEST().

test_SHAnsiToUnicode()

static void test_SHAnsiToUnicode ( void  )

static

Definition at line 210 of file shcore.c.

{
    WCHAR buffW[16];
    int ret;
 
    ret = pSHAnsiToUnicode(NULL, NULL, 0);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
 
    buffW[0] = 1;
    buffW[1] = 2;
    ret = pSHAnsiToUnicode(NULL, buffW, 2);
    ok(ret == 1, "Unexpected return value %d.\n", ret);
    ok(buffW[0] == 0 && buffW[1] == 2, "Unexpected buffer contents.\n");
 
    buffW[0] = 1;
    ret = pSHAnsiToUnicode(NULL, buffW, 0);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
    ok(buffW[0] == 1, "Unexpected buffer contents.\n");
 
    buffW[0] = 1;
    ret = pSHAnsiToUnicode("", buffW, 1);
    ok(ret == 1, "Unexpected return value %d.\n", ret);
    ok(*buffW == 0, "Unexpected buffer contents.\n");
 
    buffW[0] = 1;
    ret = pSHAnsiToUnicode("test", buffW, 0);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
    ok(buffW[0] == 1, "Unexpected buffer contents.\n");
 
    buffW[0] = 1;
    ret = pSHAnsiToUnicode("test", buffW, 1);
    ok(ret == 1, "Unexpected return value %d.\n", ret);
    ok(*buffW == 0, "Unexpected buffer contents.\n");
 
    ret = pSHAnsiToUnicode("test", buffW, 16);
    ok(ret == 5, "Unexpected return value %d.\n", ret);
    ok(!lstrcmpW(buffW, L"test"), "Unexpected buffer contents.\n");
 
    ret = pSHAnsiToUnicode("test", buffW, 2);
    ok(ret == 2, "Unexpected return value %d.\n", ret);
    ok(buffW[0] == 't' && buffW[1] == 0, "Unexpected buffer contents.\n");
}

Referenced by START_TEST().

test_SHCopyKey()

static void test_SHCopyKey ( void  )

static

Definition at line 625 of file shcore.c.

{
    HKEY hKeySrc, hKeyDst;
    DWORD ret;
 
    HKEY hkey = create_test_entries();
 
    /* Delete existing destination sub keys */
    hKeyDst = NULL;
    if (!RegOpenKeyA(HKEY_CURRENT_USER, REG_TEST_KEY "\\CopyDestination", &hKeyDst) && hKeyDst)
    {
        pSHDeleteKeyA(hKeyDst, NULL);
        RegCloseKey(hKeyDst);
    }
 
    hKeyDst = NULL;
    ret = RegCreateKeyA(HKEY_CURRENT_USER, REG_TEST_KEY "\\CopyDestination", &hKeyDst);
    ok(!ret, "Failed to create a test key, ret %ld.\n", ret);
 
    hKeySrc = NULL;
    ret = RegOpenKeyA(HKEY_LOCAL_MACHINE, REG_CURRENT_VERSION, &hKeySrc);
    ok(!ret, "Failed to open a test key, ret %ld.\n", ret);
 
    ret = pSHCopyKeyA(hKeySrc, NULL, hKeyDst, 0);
    ok(!ret, "Copy failed, ret %lu.\n", ret);
 
    RegCloseKey(hKeySrc);
    RegCloseKey(hKeyDst);
 
    /* Check we copied the sub keys, i.e. something that's on every windows system (including Wine) */
    hKeyDst = NULL;
    ret = RegOpenKeyA(HKEY_CURRENT_USER, REG_TEST_KEY "\\CopyDestination\\Shell Folders", &hKeyDst);
    ok(!ret, "Failed to open a test key, ret %ld.\n", ret);
 
    /* And the we copied the values too */
    ok(!pSHQueryValueExA(hKeyDst, "Common AppData", NULL, NULL, NULL, NULL), "SHQueryValueExA failed\n");
 
    RegCloseKey(hKeyDst);
    delete_key( hkey, "Software\\Wine", "Test" );
}

Referenced by START_TEST().

test_SHDeleteKey()

static void test_SHDeleteKey ( void  )

static

Definition at line 348 of file shcore.c.

{
    HKEY hkey, hkey2;
    DWORD ret;
 
    ret = RegCreateKeyA(HKEY_CURRENT_USER, "Software\\Wine\\Test", &hkey);
    ok(!ret, "Failed to create test key, %ld.\n", ret);
 
    ret = RegCreateKeyA(hkey, "delete_key", &hkey2);
    ok(!ret, "Failed to create test key, %ld.\n", ret);
    RegCloseKey(hkey2);
 
    ret = RegDeleteKeyA(HKEY_CURRENT_USER, "Software\\Wine\\Test");
    ok(ret == ERROR_ACCESS_DENIED, "Unexpected return value %ld.\n", ret);
 
    ret = pSHDeleteKeyA(HKEY_CURRENT_USER, "Software\\Wine\\Test");
    ok(!ret, "Unexpected retval %lu.\n", ret);
 
    ret = RegCloseKey(hkey);
    ok(!ret, "Unexpected retval %lu.\n", ret);
}

Referenced by START_TEST().

test_SHGetValue()

static void test_SHGetValue ( void  )

static

Definition at line 421 of file shcore.c.

{
    DWORD size;
    DWORD type;
    DWORD ret;
    char buf[MAX_PATH];
 
    HKEY hkey = create_test_entries();
 
    strcpy(buf, initial_buffer);
    size = MAX_PATH;
    type = -1;
    ret = pSHGetValueA(HKEY_CURRENT_USER, REG_TEST_KEY, "Test1", &type, buf, &size);
    ok(!ret, "Failed to get value, ret %lu.\n", ret);
 
    ok(!strcmp(test_exp_path1, buf), "Unexpected value %s.\n", buf);
    ok(type == REG_SZ, "Unexpected type %ld.\n", type);
 
    strcpy(buf, initial_buffer);
    size = MAX_PATH;
    type = -1;
    ret = pSHGetValueA(HKEY_CURRENT_USER, REG_TEST_KEY, "Test2", &type, buf, &size);
    ok(!ret, "Failed to get value, ret %lu.\n", ret);
    ok(!strcmp(test_path1, buf), "Unexpected value %s.\n", buf);
    ok(type == REG_SZ, "Unexpected type %ld.\n", type);
 
    delete_key(hkey, "Software\\Wine", "Test");
}

Referenced by START_TEST().

test_SHQueryValueEx()

static void test_SHQueryValueEx ( void  )

static

Definition at line 489 of file shcore.c.

{
    DWORD buffer_len1,buffer_len2;
    DWORD ret, type, size;
    char buf[MAX_PATH];
    HKEY hKey, testkey;
 
    testkey = create_test_entries();
 
    ret = RegOpenKeyExA(HKEY_CURRENT_USER, REG_TEST_KEY, 0,  KEY_QUERY_VALUE, &hKey);
    ok(!ret, "Failed to open a key, ret %lu.\n", ret);
 
    /****** SHQueryValueExA ******/
 
    buffer_len1 = max(strlen(test_exp_path1)+1, strlen(test_path1)+1);
    buffer_len2 = max(strlen(test_exp_path2)+1, strlen(test_path2)+1);
 
    /*
     * Case 1.1 All arguments are NULL
     */
    ret = pSHQueryValueExA( hKey, "Test1", NULL, NULL, NULL, NULL);
    ok(!ret, "Failed to query value, ret %lu.\n", ret);
 
    /*
     * Case 1.2 dwType is set
     */
    type = -1;
    ret = pSHQueryValueExA( hKey, "Test1", NULL, &type, NULL, NULL);
    ok(!ret, "Failed to query value, ret %lu.\n", ret);
    ok(type == REG_SZ, "Unexpected type %ld.\n", type);
 
    size = 6;
    ret = pSHQueryValueExA( hKey, "Test1", NULL, NULL, NULL, &size);
    ok(!ret, "Failed to query value, ret %lu.\n", ret);
    ok(size == buffer_len1, "Buffer sizes %lu and %lu are not equal\n", size, buffer_len1);
 
    /*
     * Expanded > unexpanded
     */
    size = 6;
    ret = pSHQueryValueExA( hKey, "Test3", NULL, NULL, NULL, &size);
    ok(!ret, "Failed to query value, ret %lu.\n", ret);
    ok(size >= buffer_len2, "Buffer size %lu should be >= %lu.\n", size, buffer_len2);
 
    /*
     * Case 1 string shrinks during expanding
     */
    strcpy(buf, initial_buffer);
    size = 6;
    type = -1;
    ret = pSHQueryValueExA( hKey, "Test1", NULL, &type, buf, &size);
    ok(ret == ERROR_MORE_DATA, "Unexpected retval %ld.\n", ret);
    ok(!strcmp(initial_buffer, buf), "Comparing (%s) with (%s) failed\n", buf, initial_buffer);
    ok(size == buffer_len1, "Buffer sizes %lu and %lu are not equal\n", size, buffer_len1);
    ok(type == REG_SZ, "Unexpected type %ld.\n", type);
 
    /*
    * string grows during expanding
    * dwSize is smaller than the size of the unexpanded string
    */
    strcpy(buf, initial_buffer);
    size = 6;
    type = -1;
    ret = pSHQueryValueExA( hKey, "Test3", NULL, &type, buf, &size);
    ok(ret == ERROR_MORE_DATA, "Unexpected retval %ld.\n", ret);
    ok(!strcmp(initial_buffer, buf), "Comparing (%s) with (%s) failed\n", buf, initial_buffer);
    ok(size >= buffer_len2, "Buffer size %lu should be >= %lu.\n", size, buffer_len2);
    ok(type == REG_SZ, "Unexpected type %ld.\n", type);
 
    /*
    * string grows during expanding
    * dwSize is larger than the size of the unexpanded string, but
    * smaller than the part before the backslash. If the unexpanded
    * string fits into the buffer, it can get cut when expanded.
    */
    strcpy(buf, initial_buffer);
    size = strlen(test_envvar2) - 2;
    type = -1;
    ret = pSHQueryValueExA(hKey, "Test3", NULL, &type, buf, &size);
    ok(ret == ERROR_MORE_DATA, "Unexpected retval %ld.\n", ret);
 
    ok(!strcmp("", buf), "Unexpanded string %s.\n", buf);
 
    ok(size >= buffer_len2, "Buffer size %lu should be >= %lu.\n", size, buffer_len2);
    ok(type == REG_SZ, "Unexpected type %ld.\n", type);
 
    /*
    * string grows during expanding
    * dwSize is larger than the size of the part before the backslash,
    * but smaller than the expanded string. If the unexpanded string fits
    * into the buffer, it can get cut when expanded.
    */
    strcpy(buf, initial_buffer);
    size = exp_len2 - 4;
    type = -1;
    ret = pSHQueryValueExA( hKey, "Test3", NULL, &type, buf, &size);
    ok(ret == ERROR_MORE_DATA, "Unexpected retval %ld.\n", ret);
 
    ok( !strcmp("", buf) || !strcmp(test_envvar2, buf),
    "Expected empty or first part of the string \"%s\", got \"%s\"\n", test_envvar2, buf);
 
    ok(size >= buffer_len2, "Buffer size %lu should be >= %lu.\n", size, buffer_len2);
    ok(type == REG_SZ, "Unexpected type %ld.\n", type);
 
    /*
    * The buffer is NULL but the size is set
    */
    strcpy(buf, initial_buffer);
    size = 6;
    type = -1;
    ret = pSHQueryValueExA( hKey, "Test3", NULL, &type, NULL, &size);
    ok(!ret, "Failed to query value, ret %lu.\n", ret);
    ok(size >= buffer_len2, "Buffer size %lu should be >= %lu.\n", size, buffer_len2);
    ok(type == REG_SZ, "Unexpected type %ld.\n", type);
 
    RegCloseKey(hKey);
 
    delete_key(testkey, "Software\\Wine", "Test");
}

Referenced by START_TEST().

test_SHRegDuplicateHKey()

static void test_SHRegDuplicateHKey ( void  )

static

Definition at line 331 of file shcore.c.

{
    HKEY hkey, hkey2;
    DWORD ret;
 
    ret = RegCreateKeyA(HKEY_CURRENT_USER, "Software\\Wine\\Test", &hkey);
    ok(!ret, "Failed to create test key, ret %ld.\n", ret);
 
    hkey2 = pSHRegDuplicateHKey(hkey);
    ok(hkey2 != NULL && hkey2 != hkey, "Unexpected duplicate key.\n");
 
    RegCloseKey(hkey2);
    RegCloseKey(hkey);
 
    RegDeleteKeyA(HKEY_CURRENT_USER, "Software\\Wine\\Test");
}

Referenced by START_TEST().

test_SHRegGetPath()

static void test_SHRegGetPath ( void  )

static

Definition at line 609 of file shcore.c.

{
    char buf[MAX_PATH];
    DWORD ret;
    HKEY hkey;
 
    hkey = create_test_entries();
 
    strcpy(buf, initial_buffer);
    ret = pSHRegGetPathA(HKEY_CURRENT_USER, REG_TEST_KEY, "Test1", buf, 0);
    ok(!ret, "Failed to get path, ret %lu.\n", ret);
    ok(!strcmp(test_exp_path1, buf), "Unexpected path %s.\n", buf);
 
    delete_key(hkey, "Software\\Wine", "Test");
}

Referenced by START_TEST().

test_SHRegGetValue()

static void test_SHRegGetValue ( void  )

static

Definition at line 450 of file shcore.c.

{
    LSTATUS ret;
    DWORD size, type;
    char data[MAX_PATH];
 
    HKEY hkey = create_test_entries();
 
    size = MAX_PATH;
    ret = pSHRegGetValueA(HKEY_CURRENT_USER, REG_TEST_KEY, "Test1", SRRF_RT_REG_EXPAND_SZ, &type, data, &size);
    ok(ret == ERROR_INVALID_PARAMETER, "Unexpected retval %lu.\n", ret);
 
    size = MAX_PATH;
    ret = pSHRegGetValueA(HKEY_CURRENT_USER, REG_TEST_KEY, "Test1", SRRF_RT_REG_SZ, &type, data, &size);
    ok(!ret, "Unexpected retval %lu.\n", ret);
    ok(!strcmp(data, test_exp_path1), "data = %s, expected %s\n", data, test_exp_path1);
    ok(type == REG_SZ, "Unexpected type %ld.\n", type);
 
    size = MAX_PATH;
    ret = pSHRegGetValueA(HKEY_CURRENT_USER, REG_TEST_KEY, "Test1", SRRF_RT_REG_DWORD, &type, data, &size);
    ok(ret == ERROR_UNSUPPORTED_TYPE, "Unexpected retval %lu.\n", ret);
 
    size = MAX_PATH;
    ret = pSHRegGetValueA(HKEY_CURRENT_USER, REG_TEST_KEY, "Test2", SRRF_RT_REG_EXPAND_SZ, &type, data, &size);
    ok(ret == ERROR_INVALID_PARAMETER, "Unexpected retval %lu.\n", ret);
 
    size = MAX_PATH;
    ret = pSHRegGetValueA(HKEY_CURRENT_USER, REG_TEST_KEY, "Test2", SRRF_RT_REG_SZ, &type, data, &size);
    ok(!ret, "Unexpected retval %lu.\n", ret);
    ok(!strcmp(data, test_path1), "data = %s, expected %s\n", data, test_path1);
    ok(type == REG_SZ, "Unexpected type %ld.\n", type);
 
    size = MAX_PATH;
    ret = pSHRegGetValueA(HKEY_CURRENT_USER, REG_TEST_KEY, "Test2", SRRF_RT_REG_QWORD, &type, data, &size);
    ok(ret == ERROR_UNSUPPORTED_TYPE, "Unexpected retval %lu.\n", ret);
 
    delete_key(hkey, "Software\\Wine", "Test");
}

Referenced by START_TEST().

test_SHUnicodeToAnsi()

static void test_SHUnicodeToAnsi ( void  )

static

Definition at line 167 of file shcore.c.

{
    char buff[16];
    int ret;
 
    ret = pSHUnicodeToAnsi(NULL, NULL, 0);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
 
    strcpy(buff, "abc");
    ret = pSHUnicodeToAnsi(NULL, buff, 2);
    ok(ret == 1, "Unexpected return value %d.\n", ret);
    ok(buff[0] == 0 && buff[1] == 'b', "Unexpected buffer contents.\n");
 
    buff[0] = 1;
    ret = pSHUnicodeToAnsi(NULL, buff, 0);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
    ok(buff[0] == 1, "Unexpected buffer contents.\n");
 
    buff[0] = 1;
    strcpy(buff, "test");
    ret = pSHUnicodeToAnsi(L"", buff, 1);
    ok(ret == 1, "Unexpected return value %d.\n", ret);
    ok(*buff == 0, "Unexpected buffer contents.\n");
 
    buff[0] = 1;
    ret = pSHUnicodeToAnsi(L"test", buff, 0);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
    ok(buff[0] == 1, "Unexpected buffer contents.\n");
 
    buff[0] = 1;
    ret = pSHUnicodeToAnsi(L"test", buff, 1);
    ok(ret == 1, "Unexpected return value %d.\n", ret);
    ok(*buff == 0, "Unexpected buffer contents.\n");
 
    ret = pSHUnicodeToAnsi(L"test", buff, 16);
    ok(ret == 5, "Unexpected return value %d.\n", ret);
    ok(!strcmp(buff, "test"), "Unexpected buffer contents.\n");
 
    ret = pSHUnicodeToAnsi(L"test", buff, 2);
    ok(ret == 2, "Unexpected return value %d.\n", ret);
    ok(!strcmp(buff, "t"), "Unexpected buffer contents.\n");
}

Referenced by START_TEST().

test_SHUnicodeToUnicode()

static void test_SHUnicodeToUnicode ( void  )

static

Definition at line 292 of file shcore.c.

{
    WCHAR buff[16];
    int ret;
 
    ret = pSHUnicodeToUnicode(NULL, NULL, 0);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
 
    lstrcpyW(buff, L"abcdefghiklm");
    ret = pSHUnicodeToUnicode(L"test", buff, 3);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
    ok(!memcmp(buff, L"test", 2 * sizeof(WCHAR)) && !buff[2], "Unexpected buffer contents.\n");
    ok(buff[3] == 'd', "Unexpected buffer contents.\n");
 
    lstrcpyW(buff, L"abcdefghiklm");
    ret = pSHUnicodeToUnicode(L"", buff, 3);
    ok(ret == 1, "Unexpected return value %d.\n", ret);
    ok(!*buff, "Unexpected buffer contents.\n");
    ok(buff[3] == 'd', "Unexpected buffer contents.\n");
 
    lstrcpyW(buff, L"abcdefghiklm");
    ret = pSHUnicodeToUnicode(L"test", buff, 4);
    ok(ret == 0, "Unexpected return value %d.\n", ret);
    ok(!memcmp(buff, L"test", 3 * sizeof(WCHAR)) && !buff[3], "Unexpected buffer contents.\n");
    ok(buff[4] == 'e', "Unexpected buffer contents.\n");
 
    lstrcpyW(buff, L"abcdefghiklm");
    ret = pSHUnicodeToUnicode(L"test", buff, 5);
    ok(ret == 5, "Unexpected return value %d.\n", ret);
    ok(!lstrcmpW(buff, L"test"), "Unexpected buffer contents.\n");
    ok(buff[5] == 'f', "Unexpected buffer contents.\n");
 
    lstrcpyW(buff, L"abcdefghiklm");
    ret = pSHUnicodeToUnicode(L"test", buff, 6);
    ok(ret == 5, "Unexpected return value %d.\n", ret);
    ok(!lstrcmpW(buff, L"test"), "Unexpected buffer contents.\n");
    ok(buff[5] == 'f', "Unexpected buffer contents.\n");
}

Referenced by START_TEST().

test_stream_size()

static void test_stream_size ( void  )

static

Definition at line 708 of file shcore.c.

{
    static const byte test_data[] = {0x1,0x2,0x3,0x4,0x5,0x6};
    static const CHAR filename[] = "test_file";
    IStream *stream, *stream2;
    HANDLE handle;
    DWORD written = 0;
    ULARGE_INTEGER stream_size;
    HRESULT hr;
 
    handle = CreateFileA(filename, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
    ok(handle != INVALID_HANDLE_VALUE, "File creation failed: %lu.\n", GetLastError());
    WriteFile(handle, test_data, sizeof(test_data), &written, NULL);
    ok(written == sizeof(test_data), "Failed to write data into file.\n");
    CloseHandle(handle);
 
    /* in read-only mode, SetSize() will success but it has no effect on Size() and the file */
    hr = pSHCreateStreamOnFileA(filename, STGM_FAILIFTHERE|STGM_READ, &stream);
    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
    CHECK_STREAM_SIZE(stream, sizeof(test_data));
    stream_size.QuadPart = 0;
    hr = IStream_SetSize(stream, stream_size);
    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
    CHECK_STREAM_SIZE(stream, sizeof(test_data));
    CHECK_STREAM_POS(stream, 0);
    stream_size.QuadPart = 100;
    hr = IStream_SetSize(stream, stream_size);
    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
    CHECK_STREAM_SIZE(stream, sizeof(test_data));
    CHECK_STREAM_POS(stream, 100);
    IStream_Release(stream);
    CHECK_FILE_SIZE(filename, sizeof(test_data));
 
    hr = pSHCreateStreamOnFileA(filename, STGM_FAILIFTHERE|STGM_WRITE, &stream);
    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
    hr = pSHCreateStreamOnFileA(filename, STGM_FAILIFTHERE|STGM_READ, &stream2);
    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
    CHECK_STREAM_SIZE(stream, sizeof(test_data));
    CHECK_STREAM_SIZE(stream2, sizeof(test_data));
    CHECK_STREAM_POS(stream, 0);
    CHECK_STREAM_POS(stream2, 0);
 
    stream_size.QuadPart = 0;
    hr = IStream_SetSize(stream, stream_size);
    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
    CHECK_STREAM_SIZE(stream, 0);
    CHECK_STREAM_SIZE(stream2, 0);
    CHECK_STREAM_POS(stream, 0);
    CHECK_STREAM_POS(stream2, 0);
 
    stream_size.QuadPart = 100;
    hr = IStream_SetSize(stream, stream_size);
    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
    CHECK_STREAM_SIZE(stream, 100);
    CHECK_STREAM_SIZE(stream2, 100);
    CHECK_STREAM_POS(stream, 0);
    CHECK_STREAM_POS(stream2, 0);
 
    stream_size.QuadPart = 90;
    hr = IStream_SetSize(stream2, stream_size);
    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
    CHECK_STREAM_SIZE(stream, 100);
    CHECK_STREAM_SIZE(stream2, 100);
    CHECK_STREAM_POS(stream, 0);
    CHECK_STREAM_POS(stream2, 90);
    IStream_Release(stream);
    IStream_Release(stream2);
    CHECK_FILE_SIZE(filename, 100);
 
    DeleteFileA(filename);
}

Referenced by START_TEST().

test_unk_init()

static void test_unk_init ( struct test_unk *  testunk )

static

Definition at line 126 of file shcore.c.

{
    testunk->IUnknown_iface.lpVtbl = &testunkvtbl;
    testunk->refcount = 1;
}

Referenced by test_process_reference().

unk_AddRef()

static ULONG WINAPI unk_AddRef ( IUnknown *  iface )

static

Definition at line 107 of file shcore.c.

{
    struct test_unk *obj = impl_from_IUnknown(iface);
    return InterlockedIncrement(&obj->refcount);
}

unk_QI()

static HRESULT WINAPI unk_QI ( IUnknown *  iface, REFIID  riid, void **  obj  )

static

Definition at line 83 of file shcore.c.

{
    if (IsEqualIID(riid, &IID_IUnknown))
    {
        *obj = iface;
        IUnknown_AddRef(iface);
        return S_OK;
    }
 
    *obj = NULL;
    return E_NOINTERFACE;
}

unk_Release()

static ULONG WINAPI unk_Release ( IUnknown *  iface )

static

Definition at line 113 of file shcore.c.

{
    struct test_unk *obj = impl_from_IUnknown(iface);
    return InterlockedDecrement(&obj->refcount);
}

void()

static void ( WINAPI *  pSetProcessReference )

static

Variable Documentation

buff

const char *static const char const char DWORD void DWORD *static const char const char DWORD void DWORD *static const char DWORD DWORD void* buff

Definition at line 41 of file shcore.c.

Referenced by _BSTR_CY(), _BSTR_DATE(), _BSTR_DEC(), _gcvt(), _gcvt_s(), _ILCreateFromFindDataW(), ata_check_unit(), BSCHttpNegotiate_BeginningTransaction(), check_class(), check_reg_entries(), compare_info(), convert_str(), DIALOG_StatusBarUpdateCaretPos(), disk_ioctl(), disk_read(), disk_write(), doparse(), DoTypeFile(), DP_LoadSP(), dumpMenu(), f_read(), f_write(), filestream_CopyTo(), filestream_Read(), filestream_Write(), find_best_locale_proc(), get_assembly_version(), get_temp_buffer(), get_vtbl_entry_name(), GetBuff(), GetContentsOfDialog(), GetContentsOfMenu(), GetExportFunctionNames(), GetImportFunctionNamesByModule(), GetListOfResourceTypes(), getSection(), IDirectMusic8Impl_GetDefaultPort(), IDirectPlay4AImpl_EnumConnections(), IDirectPlayLobby3AImpl_EnumAddressTypes(), IDirectPlayLobby3AImpl_EnumLocalApplications(), IStreamIteratorTest::istmit1(), main(), memstream_Read(), memstream_Write(), MONTHCAL_UpdateSize(), mxwriter_get_output(), myGetToken(), NLS_GetDateTimeFormatW(), openSection(), PathCompactPathW(), PathFindOnPathExA(), PathFindOnPathExW(), PathMakeSystemFolderW(), pfr_lookup_bitmap_data(), PrintMenu(), query_file_path(), Read(), readSection(), REBAR_DumpBand(), REBAR_DumpBandInfo(), SetupQueryInfVersionInformationA(), SetupQueryInfVersionInformationW(), SHAboutInfoA(), SHAboutInfoW(), SHCopyKeyW(), SHLWAPI_PathFindInOtherDirs(), SHOpenRegStream2W(), SHQueryValueExA(), SHQueryValueExW(), SHRegGetIntW(), SlistTest::slist1(), SQLGetPrivateProfileString(), SQLGetPrivateProfileStringW(), test__wfopen_s(), test_actctx_classes(), test_aw_conversion_dlgproc(), test_aw_conversion_dlgproc2(), test_bufferrawformat(), test_canceleditlabel(), test_CbOfEncoded(), test_CList(), test_columns(), test_comboex_CB_GETLBTEXT(), test_cp932(), test_create(), test_createProcessingInstruction(), test_dispinfo(), test_domdoc(), test_drawimage(), test_EM_GETLINE(), test_find_window_class(), test_fopen_s(), test_FromGdiDib(), test_gdi_objects(), test_GdipCreateBitmapFromHBITMAP(), test_GdipDrawImageFX(), test_GdipDrawImagePointsRect(), test_getElementsByTagName(), test_getroletext(), test_IStream_Read(), test_IStream_Write(), test_keynames(), test_mxwriter_flush(), test_norecompute(), test_OpenFile(), test_PathMakePretty(), test_PathUnExpandEnvStrings(), test_put_data(), test_Scan0(), test_SHAnsiToAnsi(), test_SHCreateMemStream(), test_SHFormatDateTimeA(), test_SHFormatDateTimeW(), test_SHUnicodeToAnsi(), test_SHUnicodeToUnicode(), test_sorting(), test_state_image(), test_StreamOnIStream(), test_StrRetToStringNW(), test_thread_description(), test_TVM_SORTCHILDREN(), test_VarAbs(), test_VarBoolFromStr(), test_VarDateFromStr(), test_VarDecFromStr(), test_VarFormat(), test_VarFormatFromTokens(), test_VarFormatNumber(), test_VarI1FromStr(), test_VarI2FromStr(), test_VarI4FromStr(), test_VarI8FromStr(), test_VariantToString(), test_VarR4FromStr(), test_VarR8FromStr(), test_VarRound(), test_VarUI1FromStr(), test_VarUI2FromStr(), test_VarUI4FromStr(), test_VarUI8FromStr(), test_VarWeekdayName(), test_wm_set_get_text(), textstream_read(), trace_cache(), TranslateFunctionName(), UDFCanNameBeA8dot3(), UDFCheckArea(), UDFCompleteMount(), UDFDecompressUnicode(), UDFIndexDirectory(), UDFReadDiscTrackInfo(), UpdateStatusBar(), VarBstrFromCy(), VarBstrFromDec(), VarDateFromStr(), VarFormat(), VarFormatCurrency(), VarFormatNumber(), VarFormatPercent(), VARIANT_BstrFromReal(), VARIANT_BstrReplaceDecimal(), VARIANT_FormatDate(), VARIANT_FormatNumber(), VARIANT_FormatString(), VARIANT_GetLocalisedNumberChars(), wineXmlCallbackLog(), WINHELP_RtfStreamIn(), write(), write_output_buffer(), and wtoascii().

buff_len

const char *static const char const char DWORD void DWORD *static const char const char DWORD void DWORD *static const char DWORD DWORD void DWORD* buff_len

Definition at line 41 of file shcore.c.

Referenced by SHQueryValueExA(), SHQueryValueExW(), SHRegGetIntW(), SQLGetPrivateProfileString(), and SQLGetPrivateProfileStringW().

DWORD

const char DWORD

Definition at line 42 of file shcore.c.

exp_len1

DWORD exp_len1

static

Definition at line 58 of file shcore.c.

Referenced by create_test_entries().

exp_len2

DWORD exp_len2

static

Definition at line 59 of file shcore.c.

Referenced by create_test_entries(), and test_SHQueryValueEx().

HKEY

const char HKEY

Definition at line 43 of file shcore.c.

initial_buffer

const char* initial_buffer ="0123456789"

static

Definition at line 60 of file shcore.c.

Referenced by test_SHGetValue(), test_SHQueryValueEx(), and test_SHRegGetPath().

int

WCHAR int

Definition at line 33 of file shcore.c.

mode

DWORD mode

Definition at line 44 of file shcore.c.

size

ULARGE_INTEGER* size

Definition at line 45 of file shcore.c.

Referenced by test_SHGetValue(), test_SHQueryValueEx(), and test_SHRegGetValue().

SRRF

const char *static const char const char DWORD void DWORD *static const char const char SRRF

Definition at line 40 of file shcore.c.

stream

DWORD IStream** stream

Definition at line 44 of file shcore.c.

Referenced by test_stream_size().

test_envvar1

const char* test_envvar1 = "bar"

static

Definition at line 54 of file shcore.c.

Referenced by create_test_entries().

test_envvar2

const char* test_envvar2 = "ImARatherLongButIndeedNeededString"

static

Definition at line 55 of file shcore.c.

Referenced by create_test_entries(), and test_SHQueryValueEx().

test_exp_path1

char test_exp_path1[MAX_PATH]

static

Definition at line 56 of file shcore.c.

Referenced by create_test_entries(), test_SHGetValue(), test_SHQueryValueEx(), test_SHRegGetPath(), and test_SHRegGetValue().

test_exp_path2

char test_exp_path2[MAX_PATH]

static

Definition at line 57 of file shcore.c.

Referenced by create_test_entries(), and test_SHQueryValueEx().

test_path1

const char test_path1[] = "%LONGSYSTEMVAR%\\subdir1"

static

Definition at line 51 of file shcore.c.

Referenced by create_test_entries(), test_SHGetValue(), test_SHQueryValueEx(), and test_SHRegGetValue().

test_path2

const char test_path2[] = "%FOO%\\subdir1"

static

Definition at line 52 of file shcore.c.

Referenced by create_test_entries(), and test_SHQueryValueEx().

testunkvtbl

const IUnknownVtbl testunkvtbl

static

Initial value:

=
{
    unk_QI,
    unk_AddRef,
    unk_Release,
}

Definition at line 119 of file shcore.c.

Referenced by test_unk_init().