dlls/concrt140/tests/concrt140.c

   1 /*
   2  * Copyright 2020 Daniel Lehman
   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 "wine/test.h"
  20
  21 #ifdef __i386__
  22 #include "pshpack1.h"
  23 struct thiscall_thunk
  24 {
  25     BYTE pop_eax;    /* popl  %eax (ret addr) */
  26     BYTE pop_edx;    /* popl  %edx (func) */
  27     BYTE pop_ecx;    /* popl  %ecx (this) */
  28     BYTE push_eax;   /* pushl %eax */
  29     WORD jmp_edx;    /* jmp  *%edx */
  30 };
  31 #include "poppack.h"
  32
  33 static ULONG_PTR (WINAPI *call_thiscall_func1)( void *func, void *this );
  34 static ULONG_PTR (WINAPI *call_thiscall_func3)( void *func,
  35         void *this, const void *a, const void *b );
  36
  37 static void init_thiscall_thunk(void)
  38 {
  39     struct thiscall_thunk *thunk = VirtualAlloc( NULL, sizeof(*thunk),
  40             MEM_COMMIT, PAGE_EXECUTE_READWRITE );
  41     thunk->pop_eax  = 0x58;   /* popl  %eax */
  42     thunk->pop_edx  = 0x5a;   /* popl  %edx */
  43     thunk->pop_ecx  = 0x59;   /* popl  %ecx */
  44     thunk->push_eax = 0x50;   /* pushl %eax */
  45     thunk->jmp_edx  = 0xe2ff; /* jmp  *%edx */
  46     call_thiscall_func1 = (void *)thunk;
  47     call_thiscall_func3 = (void *)thunk;
  48 }
  49
  50 #define call_func1(func,_this) call_thiscall_func1(func,_this)
  51 #define call_func3(func,_this,a,b) call_thiscall_func3(func,_this,(const void*)(a),(const void*)(b))
  52
  53 #else
  54
  55 #define init_thiscall_thunk()
  56 #define call_func1(func,_this) func(_this)
  57 #define call_func3(func,_this,a,b) func(_this,a,b)
  58
  59 #endif /* __i386__ */
  60
  61 #undef __thiscall
  62 #ifdef __i386__
  63 #define __thiscall __stdcall
  64 #else
  65 #define __thiscall __cdecl
  66 #endif
  67
  68 typedef void (*vtable_ptr)(void);
  69
  70 typedef struct {
  71     const vtable_ptr *vtable;
  72 } Context;
  73
  74 typedef struct {
  75     Context *ctx;
  76 } _Context;
  77
  78 typedef struct {
  79     const vtable_ptr *vtable;
  80     void *timer;
  81     unsigned int elapse;
  82     unsigned char repeat;
  83 } _Timer;
  84
  85 static Context* (__cdecl *p_Context_CurrentContext)(void);
  86 static _Context* (__cdecl *p__Context__CurrentContext)(_Context*);
  87
  88 static _Timer* (__thiscall *p__Timer_ctor)(_Timer*,unsigned int,unsigned char);
  89 static void (__thiscall *p__Timer_dtor)(_Timer*);
  90 static void (__thiscall *p__Timer__Start)(_Timer*);
  91 static void (__thiscall *p__Timer__Stop)(_Timer*);
  92
  93 static const BYTE *p_byte_reverse_table;
  94
  95 #define SETNOFAIL(x,y) x = (void*)GetProcAddress(module,y)
  96 #define SET(x,y) do { SETNOFAIL(x,y); ok(x != NULL, "Export '%s' not found\n", y); } while(0)
  97
  98 static BOOL init(void)
  99 {
 100     HMODULE module;
 101
 102     module = LoadLibraryA("concrt140.dll");
 103     if (!module)
 104     {
 105         win_skip("concrt140.dll not installed\n");
 106         return FALSE;
 107     }
 108
 109     SET(p__Context__CurrentContext,
 110         "?_CurrentContext@_Context@details@Concurrency@@SA?AV123@XZ");
 111     if(sizeof(void*) == 8) { /* 64-bit initialization */
 112         SET(p_Context_CurrentContext,
 113                 "?CurrentContext@Context@Concurrency@@SAPEAV12@XZ");
 114         SET(p__Timer_ctor,
 115                 "??0_Timer@details@Concurrency@@IEAA@I_N@Z");
 116         SET(p__Timer_dtor,
 117                 "??1_Timer@details@Concurrency@@MEAA@XZ");
 118         SET(p__Timer__Start,
 119                 "?_Start@_Timer@details@Concurrency@@IEAAXXZ");
 120         SET(p__Timer__Stop,
 121                 "?_Stop@_Timer@details@Concurrency@@IEAAXXZ");
 122     } else {
 123         SET(p_Context_CurrentContext,
 124                 "?CurrentContext@Context@Concurrency@@SAPAV12@XZ");
 125 #ifdef __arm__
 126         SET(p__Timer_ctor,
 127                 "??0_Timer@details@Concurrency@@IAA@I_N@Z");
 128         SET(p__Timer_dtor,
 129                 "??1_Timer@details@Concurrency@@MAA@XZ");
 130         SET(p__Timer__Start,
 131                 "?_Start@_Timer@details@Concurrency@@IAAXXZ");
 132         SET(p__Timer__Stop,
 133                 "?_Stop@_Timer@details@Concurrency@@IAAXXZ");
 134 #else
 135         SET(p__Timer_ctor,
 136                 "??0_Timer@details@Concurrency@@IAE@I_N@Z");
 137         SET(p__Timer_dtor,
 138                 "??1_Timer@details@Concurrency@@MAE@XZ");
 139         SET(p__Timer__Start,
 140                 "?_Start@_Timer@details@Concurrency@@IAEXXZ");
 141         SET(p__Timer__Stop,
 142                 "?_Stop@_Timer@details@Concurrency@@IAEXXZ");
 143 #endif
 144     }
 145
 146     init_thiscall_thunk();
 147     SET(p_byte_reverse_table, "?_Byte_reverse_table@details@Concurrency@@3QBEB");
 148     return TRUE;
 149 }
 150
 151 static void test_CurrentContext(void)
 152 {
 153     _Context _ctx, *ret;
 154     Context *ctx;
 155
 156     ctx = p_Context_CurrentContext();
 157     ok(!!ctx, "got NULL\n");
 158
 159     memset(&_ctx, 0xcc, sizeof(_ctx));
 160     ret = p__Context__CurrentContext(&_ctx);
 161     ok(_ctx.ctx == ctx, "expected %p, got %p\n", ctx, _ctx.ctx);
 162     ok(ret == &_ctx, "expected %p, got %p\n", &_ctx, ret);
 163 }
 164
 165 static HANDLE callback_called;
 166 static void __cdecl timer_callback(_Timer *this)
 167 {
 168     SetEvent(callback_called);
 169 }
 170
 171 static void test_Timer(void)
 172 {
 173     vtable_ptr vtable[2];
 174     _Timer timer;
 175     DWORD ret;
 176
 177     callback_called = CreateEventW(NULL, FALSE, FALSE, NULL);
 178     ok(callback_called != NULL, "CreateEvent failed\n");
 179
 180     call_func3(p__Timer_ctor, &timer, 1, TRUE);
 181     ok(!timer.timer, "timer already set to %p\n", timer.timer);
 182     ok(timer.elapse == 1, "elapse = %u, expected 0\n", timer.elapse);
 183     ok(timer.repeat, "timer.repeat = FALSE\n");
 184     vtable[0] = timer.vtable[0];
 185     vtable[1] = (vtable_ptr)timer_callback;
 186     timer.vtable = vtable;
 187     call_func1(p__Timer__Start, &timer);
 188     ok(timer.timer != NULL, "timer = NULL\n");
 189     ret = WaitForSingleObject(callback_called, 1000);
 190     ok(ret == WAIT_OBJECT_0, "WaitForSingleObject returned %lu\n", ret);
 191     ret = WaitForSingleObject(callback_called, 1000);
 192     ok(ret == WAIT_OBJECT_0, "WaitForSingleObject returned %lu\n", ret);
 193     call_func1(p__Timer__Stop, &timer);
 194     ok(!timer.timer, "timer != NULL\n");
 195     call_func1(p__Timer_dtor, &timer);
 196     ResetEvent(callback_called);
 197
 198     call_func3(p__Timer_ctor, &timer, 1, FALSE);
 199     timer.vtable = vtable;
 200     call_func1(p__Timer__Start, &timer);
 201     ok(timer.timer != NULL, "timer = NULL\n");
 202     ret = WaitForSingleObject(callback_called, 1000);
 203     ok(ret == WAIT_OBJECT_0, "WaitForSingleObject returned %lu\n", ret);
 204     ret = WaitForSingleObject(callback_called, 100);
 205     ok(ret == WAIT_TIMEOUT, "WaitForSingleObject returned %lu\n", ret);
 206     call_func1(p__Timer_dtor, &timer);
 207
 208     call_func3(p__Timer_ctor, &timer, 0, TRUE);
 209     timer.vtable = vtable;
 210     call_func1(p__Timer__Start, &timer);
 211     ok(timer.timer != NULL, "timer = NULL\n");
 212     ret = WaitForSingleObject(callback_called, 1000);
 213     ok(ret == WAIT_OBJECT_0, "WaitForSingleObject returned %lu\n", ret);
 214     ret = WaitForSingleObject(callback_called, 100);
 215     ok(ret == WAIT_TIMEOUT, "WaitForSingleObject returned %lu\n", ret);
 216     call_func1(p__Timer__Stop, &timer);
 217     ok(!timer.timer, "timer != NULL\n");
 218     call_func1(p__Timer_dtor, &timer);
 219
 220     CloseHandle(callback_called);
 221 }
 222
 223 static BYTE byte_reverse(BYTE b)
 224 {
 225     b = ((b & 0xf0) >> 4) | ((b & 0x0f) << 4);
 226     b = ((b & 0xcc) >> 2) | ((b & 0x33) << 2);
 227     b = ((b & 0xaa) >> 1) | ((b & 0x55) << 1);
 228     return b;
 229 }
 230
 231 static void test_data_exports(void)
 232 {
 233     unsigned int i;
 234
 235     ok(IsBadWritePtr((BYTE *)p_byte_reverse_table, 256), "byte_reverse_table is writeable.\n");
 236     for (i = 0; i < 256; ++i)
 237     {
 238         ok(p_byte_reverse_table[i] == byte_reverse(i), "Got unexpected byte %#x, expected %#x.\n",
 239                 p_byte_reverse_table[i], byte_reverse(i));
 240     }
 241 }
 242
 243 START_TEST(concrt140)
 244 {
 245     if (!init()) return;
 246     test_CurrentContext();
 247     test_Timer();
 248     test_data_exports();
 249 }