| /* |
| * Copyright 2010 Piotr Caban for CodeWeavers |
| * |
| * 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 <locale.h> |
| |
| #include <windef.h> |
| #include <winbase.h> |
| #include "wine/test.h" |
| |
| typedef struct { |
| LCID handle; |
| unsigned page; |
| short *table; |
| int delfl; |
| } MSVCP__Ctypevec; |
| |
| /* basic_string<char, char_traits<char>, allocator<char>> */ |
| #define BUF_SIZE_CHAR 16 |
| typedef struct |
| { |
| void *allocator; |
| union { |
| char buf[BUF_SIZE_CHAR]; |
| char *ptr; |
| } data; |
| size_t size; |
| size_t res; |
| } basic_string_char; |
| |
| static void* (__cdecl *p_set_invalid_parameter_handler)(void*); |
| static _locale_t (__cdecl *p__get_current_locale)(void); |
| static void (__cdecl *p_free)(void*); |
| |
| static void (__cdecl *p_char_assign)(void*, const void*); |
| static void (__cdecl *p_wchar_assign)(void*, const void*); |
| static void (__cdecl *p_short_assign)(void*, const void*); |
| |
| static BYTE (__cdecl *p_char_eq)(const void*, const void*); |
| static BYTE (__cdecl *p_wchar_eq)(const void*, const void*); |
| static BYTE (__cdecl *p_short_eq)(const void*, const void*); |
| |
| static char* (__cdecl *p_Copy_s)(char*, size_t, const char*, size_t); |
| |
| static unsigned short (__cdecl *p_wctype)(const char*); |
| static MSVCP__Ctypevec (__cdecl *p__Getctype)(void); |
| |
| #undef __thiscall |
| #ifdef __i386__ |
| #define __thiscall __stdcall |
| #else |
| #define __thiscall __cdecl |
| #endif |
| |
| static char* (__thiscall *p_char_address)(void*, char*); |
| static void* (__thiscall *p_char_ctor)(void*); |
| static void (__thiscall *p_char_deallocate)(void*, char*, size_t); |
| static char* (__thiscall *p_char_allocate)(void*, size_t); |
| static void (__thiscall *p_char_construct)(void*, char*, const char*); |
| static size_t (__thiscall *p_char_max_size)(void*); |
| |
| void* (__thiscall *p_collate_char_ctor_refs)(void*, size_t); |
| int (__thiscall *p_collate_char_compare)(const void*, const char*, |
| const char*, const char*, const char*); |
| void (__thiscall *p_collate_char_dtor)(void*); |
| void* (__thiscall *p_numpunct_char_ctor)(void*); |
| basic_string_char* (__thiscall *p_numpunct_char_falsename)(void*,basic_string_char*); |
| void (__thiscall *p_numpunct_char_dtor)(void*); |
| static void (__thiscall *p_basic_string_char_dtor)(basic_string_char*); |
| static const char* (__thiscall *p_basic_string_char_cstr)(basic_string_char*); |
| |
| static int invalid_parameter = 0; |
| static void __cdecl test_invalid_parameter_handler(const wchar_t *expression, |
| const wchar_t *function, const wchar_t *file, |
| unsigned line, uintptr_t arg) |
| { |
| ok(expression == NULL, "expression is not NULL\n"); |
| ok(function == NULL, "function is not NULL\n"); |
| ok(file == NULL, "file is not NULL\n"); |
| ok(line == 0, "line = %u\n", line); |
| ok(arg == 0, "arg = %lx\n", (UINT_PTR)arg); |
| invalid_parameter++; |
| } |
| |
| /* Emulate a __thiscall */ |
| #ifdef __i386__ |
| |
| #include "pshpack1.h" |
| struct thiscall_thunk |
| { |
| BYTE pop_eax; /* popl %eax (ret addr) */ |
| BYTE pop_edx; /* popl %edx (func) */ |
| BYTE pop_ecx; /* popl %ecx (this) */ |
| BYTE push_eax; /* pushl %eax */ |
| WORD jmp_edx; /* jmp *%edx */ |
| }; |
| #include "poppack.h" |
| |
| static void * (WINAPI *call_thiscall_func1)( void *func, void *this ); |
| static void * (WINAPI *call_thiscall_func2)( void *func, void *this, const void *a ); |
| static void * (WINAPI *call_thiscall_func3)( void *func, void *this, const void *a, const void *b ); |
| static void * (WINAPI *call_thiscall_func5)( void *func, void *this, const void *a, const void *b, |
| const void *c, const void *d ); |
| struct thiscall_thunk_retptr *thunk_retptr; |
| |
| static void init_thiscall_thunk(void) |
| { |
| struct thiscall_thunk *thunk = VirtualAlloc( NULL, sizeof(*thunk), |
| MEM_COMMIT, PAGE_EXECUTE_READWRITE ); |
| thunk->pop_eax = 0x58; /* popl %eax */ |
| thunk->pop_edx = 0x5a; /* popl %edx */ |
| thunk->pop_ecx = 0x59; /* popl %ecx */ |
| thunk->push_eax = 0x50; /* pushl %eax */ |
| thunk->jmp_edx = 0xe2ff; /* jmp *%edx */ |
| call_thiscall_func1 = (void *)thunk; |
| call_thiscall_func2 = (void *)thunk; |
| call_thiscall_func3 = (void *)thunk; |
| call_thiscall_func5 = (void *)thunk; |
| } |
| |
| #define call_func1(func,_this) call_thiscall_func1(func,_this) |
| #define call_func2(func,_this,a) call_thiscall_func2(func,_this,(const void*)(a)) |
| #define call_func3(func,_this,a,b) call_thiscall_func3(func,_this,(const void*)(a),(const void*)(b)) |
| #define call_func5(func,_this,a,b,c,d) call_thiscall_func5(func,_this,(const void*)(a),(const void*)(b), \ |
| (const void*)(c), (const void *)(d)) |
| |
| #else |
| |
| #define init_thiscall_thunk() |
| #define call_func1(func,_this) func(_this) |
| #define call_func2(func,_this,a) func(_this,a) |
| #define call_func3(func,_this,a,b) func(_this,a,b) |
| #define call_func5(func,_this,a,b,c,d) func(_this,a,b,c,d) |
| |
| #endif /* __i386__ */ |
| |
| #define SETNOFAIL(x,y) x = (void*)GetProcAddress(msvcp,y) |
| #define SET(x,y) do { SETNOFAIL(x,y); ok(x != NULL, "Export '%s' not found\n", y); } while(0) |
| static BOOL init(void) |
| { |
| HMODULE msvcr = LoadLibraryA("msvcr90.dll"); |
| HMODULE msvcp = LoadLibraryA("msvcp90.dll"); |
| if(!msvcr || !msvcp) { |
| win_skip("msvcp90.dll or msvcrt90.dll not installed\n"); |
| return FALSE; |
| } |
| |
| p_set_invalid_parameter_handler = (void*)GetProcAddress(msvcr, "_set_invalid_parameter_handler"); |
| p__get_current_locale = (void*)GetProcAddress(msvcr, "_get_current_locale"); |
| p_free = (void*)GetProcAddress(msvcr, "free"); |
| if(!p_set_invalid_parameter_handler || !p__get_current_locale || !p_free) { |
| win_skip("Error setting tests environment\n"); |
| return FALSE; |
| } |
| |
| p_set_invalid_parameter_handler(test_invalid_parameter_handler); |
| |
| SET(p_wctype, "wctype"); |
| SET(p__Getctype, "_Getctype"); |
| if(sizeof(void*) == 8) { /* 64-bit initialization */ |
| SET(p_char_assign, "?assign@?$char_traits@D@std@@SAXAEADAEBD@Z"); |
| SET(p_wchar_assign, "?assign@?$char_traits@_W@std@@SAXAEA_WAEB_W@Z"); |
| SET(p_short_assign, "?assign@?$char_traits@G@std@@SAXAEAGAEBG@Z"); |
| |
| SET(p_char_eq, "?eq@?$char_traits@D@std@@SA_NAEBD0@Z"); |
| SET(p_wchar_eq, "?eq@?$char_traits@_W@std@@SA_NAEB_W0@Z"); |
| SET(p_short_eq, "?eq@?$char_traits@G@std@@SA_NAEBG0@Z"); |
| |
| SET(p_Copy_s, "?_Copy_s@?$char_traits@D@std@@SAPEADPEAD_KPEBD1@Z"); |
| |
| SET(p_char_address, "?address@?$allocator@D@std@@QEBAPEADAEAD@Z"); |
| SET(p_char_ctor, "??0?$allocator@D@std@@QEAA@XZ"); |
| SET(p_char_deallocate, "?deallocate@?$allocator@D@std@@QEAAXPEAD_K@Z"); |
| SET(p_char_allocate, "?allocate@?$allocator@D@std@@QEAAPEAD_K@Z"); |
| SET(p_char_construct, "?construct@?$allocator@D@std@@QEAAXPEADAEBD@Z"); |
| SET(p_char_max_size, "?max_size@?$allocator@D@std@@QEBA_KXZ"); |
| |
| SET(p_collate_char_ctor_refs, "??0?$collate@D@std@@QEAA@_K@Z"); |
| SET(p_collate_char_compare, "?compare@?$collate@D@std@@QEBAHPEBD000@Z"); |
| SET(p_collate_char_dtor, "??1?$collate@D@std@@MEAA@XZ"); |
| SET(p_numpunct_char_ctor, "??_F?$numpunct@D@std@@QEAAXXZ"); |
| SET(p_numpunct_char_falsename, "?falsename@?$numpunct@D@std@@QEBA?AV?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@2@XZ"); |
| SET(p_numpunct_char_dtor, "??1?$numpunct@D@std@@MEAA@XZ"); |
| SET(p_basic_string_char_dtor, |
| "??1?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QEAA@XZ"); |
| SET(p_basic_string_char_cstr, |
| "?c_str@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QEBAPEBDXZ"); |
| } else { |
| SET(p_char_assign, "?assign@?$char_traits@D@std@@SAXAADABD@Z"); |
| SET(p_wchar_assign, "?assign@?$char_traits@_W@std@@SAXAA_WAB_W@Z"); |
| SET(p_short_assign, "?assign@?$char_traits@G@std@@SAXAAGABG@Z"); |
| |
| SET(p_char_eq, "?eq@?$char_traits@D@std@@SA_NABD0@Z"); |
| SET(p_wchar_eq, "?eq@?$char_traits@_W@std@@SA_NAB_W0@Z"); |
| SET(p_short_eq, "?eq@?$char_traits@G@std@@SA_NABG0@Z"); |
| |
| SET(p_Copy_s, "?_Copy_s@?$char_traits@D@std@@SAPADPADIPBDI@Z"); |
| |
| SET(p_char_address, "?address@?$allocator@D@std@@QBEPADAAD@Z"); |
| SET(p_char_ctor, "??0?$allocator@D@std@@QAE@XZ"); |
| SET(p_char_deallocate, "?deallocate@?$allocator@D@std@@QAEXPADI@Z"); |
| SET(p_char_allocate, "?allocate@?$allocator@D@std@@QAEPADI@Z"); |
| SET(p_char_construct, "?construct@?$allocator@D@std@@QAEXPADABD@Z"); |
| SET(p_char_max_size, "?max_size@?$allocator@D@std@@QBEIXZ"); |
| |
| SET(p_collate_char_ctor_refs, "??0?$collate@D@std@@QAE@I@Z"); |
| SET(p_collate_char_compare, "?compare@?$collate@D@std@@QBEHPBD000@Z"); |
| SET(p_collate_char_dtor, "??1?$collate@D@std@@MAE@XZ"); |
| SET(p_numpunct_char_ctor, "??_F?$numpunct@D@std@@QAEXXZ"); |
| SET(p_numpunct_char_falsename, "?falsename@?$numpunct@D@std@@QBE?AV?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@2@XZ"); |
| SET(p_numpunct_char_dtor, "??1?$numpunct@D@std@@MAE@XZ"); |
| SET(p_basic_string_char_dtor, |
| "??1?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QAE@XZ"); |
| SET(p_basic_string_char_cstr, |
| "?c_str@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QBEPBDXZ"); |
| } |
| |
| init_thiscall_thunk(); |
| return TRUE; |
| } |
| |
| static void test_assign(void) |
| { |
| const char in[] = "abc"; |
| char out[4]; |
| |
| out[1] = '#'; |
| p_char_assign(out, in); |
| ok(out[0] == in[0], "out[0] = %c\n", out[0]); |
| ok(out[1] == '#', "out[1] = %c\n", out[1]); |
| |
| out[2] = '#'; |
| p_wchar_assign(out, in); |
| ok(*((char*)out)==in[0] && *((char*)out+1)==in[1], |
| "out[0] = %d, out[1] = %d\n", (int)out[0], (int)out[1]); |
| ok(out[2] == '#', "out[2] = %c\n", out[2]); |
| |
| out[2] = '#'; |
| p_short_assign(out, in); |
| ok(*((char*)out)==in[0] && *((char*)out+1)==in[1], |
| "out[0] = %d, out[1] = %d\n", (int)out[0], (int)out[1]); |
| ok(out[2] == '#', "out[2] = %c\n", out[2]); |
| } |
| |
| static void test_equal(void) |
| { |
| static const char in1[] = "abc"; |
| static const char in2[] = "ab"; |
| static const char in3[] = "a"; |
| static const char in4[] = "b"; |
| BYTE ret; |
| |
| ret = p_char_eq(in1, in2); |
| ok(ret == TRUE, "ret = %d\n", (int)ret); |
| ret = p_char_eq(in1, in3); |
| ok(ret == TRUE, "ret = %d\n", (int)ret); |
| ret = p_char_eq(in1, in4); |
| ok(ret == FALSE, "ret = %d\n", (int)ret); |
| |
| ret = p_wchar_eq(in1, in2); |
| ok(ret == TRUE, "ret = %d\n", (int)ret); |
| ret = p_wchar_eq(in1, in3); |
| ok(ret == FALSE, "ret = %d\n", (int)ret); |
| ret = p_wchar_eq(in1, in4); |
| ok(ret == FALSE, "ret = %d\n", (int)ret); |
| |
| ret = p_short_eq(in1, in2); |
| ok(ret == TRUE, "ret = %d\n", (int)ret); |
| ret = p_short_eq(in1, in3); |
| ok(ret == FALSE, "ret = %d\n", (int)ret); |
| ret = p_short_eq(in1, in4); |
| ok(ret == FALSE, "ret = %d\n", (int)ret); |
| } |
| |
| static void test_Copy_s(void) |
| { |
| static const char src[] = "abcd"; |
| char dest[32], *ret; |
| |
| dest[4] = '#'; |
| dest[5] = '\0'; |
| ret = p_Copy_s(dest, 4, src, 4); |
| ok(ret == dest, "ret != dest\n"); |
| ok(dest[4] == '#', "dest[4] != '#'\n"); |
| ok(!memcmp(dest, src, sizeof(char[4])), "dest = %s\n", dest); |
| |
| ret = p_Copy_s(dest, 32, src, 4); |
| ok(ret == dest, "ret != dest\n"); |
| ok(dest[4] == '#', "dest[4] != '#'\n"); |
| ok(!memcmp(dest, src, sizeof(char[4])), "dest = %s\n", dest); |
| |
| errno = 0xdeadbeef; |
| dest[0] = '#'; |
| ret = p_Copy_s(dest, 3, src, 4); |
| ok(ret == dest, "ret != dest\n"); |
| ok(dest[0] == '\0', "dest[0] != 0\n"); |
| ok(invalid_parameter==1, "invalid_parameter = %d\n", |
| invalid_parameter); |
| invalid_parameter = 0; |
| ok(errno == 0xdeadbeef, "errno = %d\n", errno); |
| |
| errno = 0xdeadbeef; |
| p_Copy_s(NULL, 32, src, 4); |
| ok(invalid_parameter==1, "invalid_parameter = %d\n", |
| invalid_parameter); |
| invalid_parameter = 0; |
| ok(errno == 0xdeadbeef, "errno = %d\n", errno); |
| |
| errno = 0xdeadbeef; |
| p_Copy_s(dest, 32, NULL, 4); |
| ok(invalid_parameter==1, "invalid_parameter = %d\n", |
| invalid_parameter); |
| invalid_parameter = 0; |
| ok(errno == 0xdeadbeef, "errno = %d\n", errno); |
| } |
| |
| static void test_wctype(void) |
| { |
| static const struct { |
| const char *name; |
| unsigned short mask; |
| } properties[] = { |
| { "alnum", 0x107 }, |
| { "alpha", 0x103 }, |
| { "cntrl", 0x020 }, |
| { "digit", 0x004 }, |
| { "graph", 0x117 }, |
| { "lower", 0x002 }, |
| { "print", 0x157 }, |
| { "punct", 0x010 }, |
| { "space", 0x008 }, |
| { "upper", 0x001 }, |
| { "xdigit", 0x080 }, |
| { "ALNUM", 0x000 }, |
| { "Alnum", 0x000 }, |
| { "", 0x000 } |
| }; |
| int i, ret; |
| |
| for(i=0; i<sizeof(properties)/sizeof(properties[0]); i++) { |
| ret = p_wctype(properties[i].name); |
| ok(properties[i].mask == ret, "%d - Expected %x, got %x\n", i, properties[i].mask, ret); |
| } |
| } |
| |
| static void test__Getctype(void) |
| { |
| MSVCP__Ctypevec ret; |
| |
| ret = p__Getctype(); |
| ok(ret.handle == 0, "ret.handle = %d\n", ret.handle); |
| ok(ret.page == 0, "ret.page = %d\n", ret.page); |
| ok(ret.delfl == 1, "ret.delfl = %d\n", ret.delfl); |
| ok(ret.table[0] == 32, "ret.table[0] = %d\n", ret.table[0]); |
| p_free(ret.table); |
| |
| p__get_current_locale()->locinfo->lc_handle[LC_COLLATE] = 1; |
| ret = p__Getctype(); |
| ok(ret.handle == 1, "ret.handle = %d\n", ret.handle); |
| ok(ret.page == 0, "ret.page = %d\n", ret.page); |
| ok(ret.delfl == 1, "ret.delfl = %d\n", ret.delfl); |
| ok(ret.table[0] == 32, "ret.table[0] = %d\n", ret.table[0]); |
| p_free(ret.table); |
| } |
| |
| static void test_allocator_char(void) |
| { |
| void *allocator = (void*)0xdeadbeef; |
| char *ptr; |
| char val; |
| unsigned int size; |
| |
| allocator = call_func1(p_char_ctor, allocator); |
| ok(allocator == (void*)0xdeadbeef, "allocator = %p\n", allocator); |
| |
| ptr = call_func2(p_char_address, NULL, (void*)0xdeadbeef); |
| ok(ptr == (void*)0xdeadbeef, "incorrect address (%p)\n", ptr); |
| |
| ptr = NULL; |
| ptr = call_func2(p_char_allocate, allocator, 10); |
| ok(ptr != NULL, "Memory allocation failed\n"); |
| |
| ptr[0] = ptr[1] = '#'; |
| val = 'a'; |
| call_func3(p_char_construct, allocator, ptr, &val); |
| val = 'b'; |
| call_func3(p_char_construct, allocator, (ptr+1), &val); |
| ok(ptr[0] == 'a', "ptr[0] = %c\n", ptr[0]); |
| ok(ptr[1] == 'b', "ptr[1] = %c\n", ptr[1]); |
| |
| call_func3(p_char_deallocate, allocator, ptr, -1); |
| |
| size = (unsigned int)call_func1(p_char_max_size, allocator); |
| ok(size == (unsigned int)0xffffffff, "size = %x\n", size); |
| } |
| |
| static void test_virtual_call(void) |
| { |
| BYTE this[256]; |
| basic_string_char bstr; |
| const char *p; |
| char str1[] = "test"; |
| char str2[] = "TEST"; |
| int ret; |
| |
| call_func2(p_collate_char_ctor_refs, this, 0); |
| ret = (int)call_func5(p_collate_char_compare, this, str1, str1+4, str1, str1+4); |
| ok(ret == 0, "collate<char>::compare returned %d\n", ret); |
| ret = (int)call_func5(p_collate_char_compare, this, str2, str2+4, str1, str1+4); |
| ok(ret == 1, "collate<char>::compare returned %d\n", ret); |
| ret = (int)call_func5(p_collate_char_compare, this, str1, str1+3, str1, str1+4); |
| ok(ret == -1, "collate<char>::compare returned %d\n", ret); |
| call_func1(p_collate_char_dtor, this); |
| |
| call_func1(p_numpunct_char_ctor, this); |
| call_func2(p_numpunct_char_falsename, this, &bstr); |
| p = call_func1(p_basic_string_char_cstr, &bstr); |
| ok(!strcmp(p, "false"), "numpunct<char>::falsename returned %s\n", p); |
| call_func1(p_basic_string_char_dtor, &bstr); |
| call_func1(p_numpunct_char_dtor, this); |
| } |
| |
| START_TEST(misc) |
| { |
| if(!init()) |
| return; |
| |
| test_assign(); |
| test_equal(); |
| test_Copy_s(); |
| test_wctype(); |
| test__Getctype(); |
| test_virtual_call(); |
| |
| test_allocator_char(); |
| |
| ok(!invalid_parameter, "invalid_parameter_handler was invoked too many times\n"); |
| } |