dlls/msvcp100/tests/string.c

   1 /*
   2  * Copyright 2012 Piotr Caban for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <stdio.h>
  20 #include <limits.h>
  21
  22 #include <windef.h>
  23 #include <winbase.h>
  24 #include "wine/test.h"
  25
  26 typedef struct
  27 {
  28     char *str;
  29     char null_str;
  30 } _Yarn_char;
  31
  32 #undef __thiscall
  33 #ifdef __i386__
  34 #define __thiscall __stdcall
  35 #else
  36 #define __thiscall __cdecl
  37 #endif
  38
  39 /* Emulate a __thiscall */
  40 #ifdef __i386__
  41
  42 #include "pshpack1.h"
  43 struct thiscall_thunk
  44 {
  45     BYTE pop_eax;    /* popl  %eax (ret addr) */
  46     BYTE pop_edx;    /* popl  %edx (func) */
  47     BYTE pop_ecx;    /* popl  %ecx (this) */
  48     BYTE push_eax;   /* pushl %eax */
  49     WORD jmp_edx;    /* jmp  *%edx */
  50 };
  51 #include "poppack.h"
  52
  53 static void * (WINAPI *call_thiscall_func1)( void *func, void *this );
  54 static void * (WINAPI *call_thiscall_func2)( void *func, void *this, const void *a );
  55
  56 static void init_thiscall_thunk(void)
  57 {
  58     struct thiscall_thunk *thunk = VirtualAlloc( NULL, sizeof(*thunk),
  59             MEM_COMMIT, PAGE_EXECUTE_READWRITE );
  60     thunk->pop_eax  = 0x58;   /* popl  %eax */
  61     thunk->pop_edx  = 0x5a;   /* popl  %edx */
  62     thunk->pop_ecx  = 0x59;   /* popl  %ecx */
  63     thunk->push_eax = 0x50;   /* pushl %eax */
  64     thunk->jmp_edx  = 0xe2ff; /* jmp  *%edx */
  65     call_thiscall_func1 = (void *)thunk;
  66     call_thiscall_func2 = (void *)thunk;
  67 }
  68
  69 #define call_func1(func,_this) call_thiscall_func1(func,_this)
  70 #define call_func2(func,_this,a) call_thiscall_func2(func,_this,(const void*)(a))
  71
  72 #else
  73
  74 #define init_thiscall_thunk()
  75 #define call_func1(func,_this) func(_this)
  76 #define call_func2(func,_this,a) func(_this,a)
  77
  78 #endif /* __i386__ */
  79
  80 static _Yarn_char* (__thiscall *_Yarn_char_ctor_cstr)(_Yarn_char *this, const char *str);
  81 static _Yarn_char* (__thiscall *_Yarn_char_copy_ctor)(_Yarn_char *this, const _Yarn_char *copy);
  82 static void (__thiscall *_Yarn_char_dtor)(_Yarn_char *this);
  83 static const char* (__thiscall *_Yarn_char_c_str)(const _Yarn_char *this);
  84 static char (__thiscall *_Yarn_char_empty)(const _Yarn_char *this);
  85
  86 static HMODULE msvcp;
  87 #define SETNOFAIL(x,y) x = (void*)GetProcAddress(msvcp,y)
  88 #define SET(x,y) do { SETNOFAIL(x,y); ok(x != NULL, "Export '%s' not found\n", y); } while(0)
  89 static BOOL init(void)
  90 {
  91     msvcp = LoadLibraryA("msvcp100.dll");
  92     if(!msvcp) {
  93         win_skip("msvcp100.dll not installed\n");
  94         return FALSE;
  95     }
  96
  97     if(sizeof(void*) == 8) { /* 64-bit initialization */
  98         SET(_Yarn_char_ctor_cstr, "??0?$_Yarn@D@std@@QEAA@PEBD@Z");
  99         SET(_Yarn_char_copy_ctor, "??0?$_Yarn@D@std@@QEAA@AEBV01@@Z");
 100         SET(_Yarn_char_dtor, "??1?$_Yarn@D@std@@QEAA@XZ");
 101         SET(_Yarn_char_c_str, "?_C_str@?$_Yarn@D@std@@QEBAPEBDXZ");
 102         SET(_Yarn_char_empty, "?_Empty@?$_Yarn@D@std@@QEBA_NXZ");
 103     }else {
 104 #ifdef __arm__
 105         SET(_Yarn_char_ctor_cstr, "??0?$_Yarn@D@std@@QAA@PBD@Z");
 106         SET(_Yarn_char_copy_ctor, "??0?$_Yarn@D@std@@QAA@ABV01@@Z");
 107         SET(_Yarn_char_dtor, "??1?$_Yarn@D@std@@QAA@XZ");
 108         SET(_Yarn_char_c_str, "?_C_str@?$_Yarn@D@std@@QBAPBDXZ");
 109         SET(_Yarn_char_empty, "?_Empty@?$_Yarn@D@std@@QBA_NXZ");
 110 #else
 111         SET(_Yarn_char_ctor_cstr, "??0?$_Yarn@D@std@@QAE@PBD@Z");
 112         SET(_Yarn_char_copy_ctor, "??0?$_Yarn@D@std@@QAE@ABV01@@Z");
 113         SET(_Yarn_char_dtor, "??1?$_Yarn@D@std@@QAE@XZ");
 114         SET(_Yarn_char_c_str, "?_C_str@?$_Yarn@D@std@@QBEPBDXZ");
 115         SET(_Yarn_char_empty, "?_Empty@?$_Yarn@D@std@@QBE_NXZ");
 116 #endif /* __arm__ */
 117     }
 118
 119     init_thiscall_thunk();
 120     return TRUE;
 121 }
 122
 123 static void test__Yarn_char(void)
 124 {
 125     _Yarn_char c1, c2;
 126     char bool_ret;
 127     const char *str;
 128
 129     call_func2(_Yarn_char_ctor_cstr, &c1, NULL);
 130     ok(!c1.str, "c1.str = %s\n", c1.str);
 131     ok(!c1.null_str, "c1.null_str = %d\n", c1.null_str);
 132     bool_ret = (char)(INT_PTR)call_func1(_Yarn_char_empty, &c1);
 133     ok(bool_ret == 1, "ret = %d\n", bool_ret);
 134     str = call_func1(_Yarn_char_c_str, &c1);
 135     ok(str == &c1.null_str, "str = %p, expected %p\n", str, &c1.null_str);
 136     call_func1(_Yarn_char_dtor, &c1);
 137
 138     call_func2(_Yarn_char_ctor_cstr, &c1, "");
 139     ok(c1.str != NULL, "c1.str = NULL\n");
 140     ok(!strcmp(c1.str, ""), "c1.str = %s\n", c1.str);
 141     ok(!c1.null_str, "c1.null_str = %d\n", c1.null_str);
 142     bool_ret = (char)(INT_PTR)call_func1(_Yarn_char_empty, &c1);
 143     ok(bool_ret == 0, "ret = %d\n", bool_ret);
 144     str = call_func1(_Yarn_char_c_str, &c1);
 145     ok(str == c1.str, "str = %p, expected %p\n", str, c1.str);
 146     call_func1(_Yarn_char_dtor, &c1);
 147
 148     call_func2(_Yarn_char_ctor_cstr, &c1, "test");
 149     ok(!strcmp(c1.str, "test"), "c1.str = %s\n", c1.str);
 150     ok(!c1.null_str, "c1.null_str = %d\n", c1.null_str);
 151     bool_ret = (char)(INT_PTR)call_func1(_Yarn_char_empty, &c1);
 152     ok(bool_ret == 0, "ret = %d\n", bool_ret);
 153     str = call_func1(_Yarn_char_c_str, &c1);
 154     ok(str == c1.str, "str = %p, expected %p\n", str, c1.str);
 155
 156     call_func2(_Yarn_char_copy_ctor, &c2, &c1);
 157     ok(c1.str != c2.str, "c1.str = c2.str\n");
 158     ok(!strcmp(c2.str, "test"), "c2.str = %s\n", c2.str);
 159     ok(!c2.null_str, "c2.null_str = %d\n", c2.null_str);
 160     call_func1(_Yarn_char_dtor, &c2);
 161
 162     c1.null_str = 'a';
 163     call_func2(_Yarn_char_copy_ctor, &c2, &c1);
 164     ok(c1.str != c2.str, "c1.str = c2.str\n");
 165     ok(!strcmp(c2.str, "test"), "c2.str = %s\n", c2.str);
 166     ok(!c2.null_str, "c2.null_str = %d\n", c2.null_str);
 167     call_func1(_Yarn_char_dtor, &c2);
 168     call_func1(_Yarn_char_dtor, &c1);
 169 }
 170
 171 START_TEST(string)
 172 {
 173     if(!init())
 174         return;
 175
 176     test__Yarn_char();
 177
 178     FreeLibrary(msvcp);
 179 }