git.oblomov.eu Git - wine/blob - dlls/msvcr100/tests/msvcr100.c

   1 /*
   2  * Copyright 2012 Dan Kegel
   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 <errno.h>
  20 #include <stdarg.h>
  21 #include <stdlib.h>
  22 #include <wchar.h>
  23 #include <stdio.h>
  24
  25 #include <windef.h>
  26 #include <winbase.h>
  27 #include "wine/test.h"
  28
  29 #define DEFINE_EXPECT(func) \
  30     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
  31
  32 #define SET_EXPECT(func) \
  33     expect_ ## func = TRUE
  34
  35 #define CHECK_EXPECT2(func) \
  36     do { \
  37         ok(expect_ ##func, "unexpected call " #func "\n"); \
  38         called_ ## func = TRUE; \
  39     }while(0)
  40
  41 #define CHECK_EXPECT(func) \
  42     do { \
  43         CHECK_EXPECT2(func); \
  44         expect_ ## func = FALSE; \
  45     }while(0)
  46
  47 #define CHECK_CALLED(func) \
  48     do { \
  49         ok(called_ ## func, "expected " #func "\n"); \
  50         expect_ ## func = called_ ## func = FALSE; \
  51     }while(0)
  52
  53 DEFINE_EXPECT(invalid_parameter_handler);
  54
  55 static _invalid_parameter_handler (__cdecl *p_set_invalid_parameter_handler)(_invalid_parameter_handler);
  56
  57 static void __cdecl test_invalid_parameter_handler(const wchar_t *expression,
  58         const wchar_t *function, const wchar_t *file,
  59         unsigned line, uintptr_t arg)
  60 {
  61     CHECK_EXPECT(invalid_parameter_handler);
  62     ok(expression == NULL, "expression is not NULL\n");
  63     ok(function == NULL, "function is not NULL\n");
  64     ok(file == NULL, "file is not NULL\n");
  65     ok(line == 0, "line = %u\n", line);
  66     ok(arg == 0, "arg = %lx\n", (UINT_PTR)arg);
  67 }
  68
  69 static int* (__cdecl *p_errno)(void);
  70 static int (__cdecl *p_wmemcpy_s)(wchar_t *dest, size_t numberOfElements, const wchar_t *src, size_t count);
  71 static int (__cdecl *p_wmemmove_s)(wchar_t *dest, size_t numberOfElements, const wchar_t *src, size_t count);
  72 static FILE* (__cdecl *p_fopen)(const char*,const char*);
  73 static int (__cdecl *p_fclose)(FILE*);
  74 static size_t (__cdecl *p_fread_s)(void*,size_t,size_t,size_t,FILE*);
  75 static void* (__cdecl *p__aligned_offset_malloc)(size_t, size_t, size_t);
  76 static void (__cdecl *p__aligned_free)(void*);
  77 static size_t (__cdecl *p__aligned_msize)(void*, size_t, size_t);
  78
  79 /* make sure we use the correct errno */
  80 #undef errno
  81 #define errno (*p_errno())
  82
  83 #define SETNOFAIL(x,y) x = (void*)GetProcAddress(hcrt,y)
  84 #define SET(x,y) do { SETNOFAIL(x,y); ok(x != NULL, "Export '%s' not found\n", y); } while(0)
  85
  86 static BOOL init(void)
  87 {
  88     HMODULE hcrt;
  89
  90     SetLastError(0xdeadbeef);
  91     hcrt = LoadLibraryA("msvcr100.dll");
  92     if (!hcrt) {
  93         win_skip("msvcr100.dll not installed (got %d)\n", GetLastError());
  94         return FALSE;
  95     }
  96
  97     SET(p_errno, "_errno");
  98     SET(p_set_invalid_parameter_handler, "_set_invalid_parameter_handler");
  99     SET(p_wmemcpy_s, "wmemcpy_s");
 100     SET(p_wmemmove_s, "wmemmove_s");
 101     SET(p_fopen, "fopen");
 102     SET(p_fclose, "fclose");
 103     SET(p_fread_s, "fread_s");
 104     SET(p__aligned_offset_malloc, "_aligned_offset_malloc");
 105     SET(p__aligned_free, "_aligned_free");
 106     SET(p__aligned_msize, "_aligned_msize");
 107
 108     return TRUE;
 109 }
 110
 111 #define NUMELMS(array) (sizeof(array)/sizeof((array)[0]))
 112
 113 #define okwchars(dst, b0, b1, b2, b3, b4, b5, b6, b7) \
 114     ok(dst[0] == b0 && dst[1] == b1 && dst[2] == b2 && dst[3] == b3 && \
 115        dst[4] == b4 && dst[5] == b5 && dst[6] == b6 && dst[7] == b7, \
 116        "Bad result: 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x\n",\
 117        dst[0], dst[1], dst[2], dst[3], dst[4], dst[5], dst[6], dst[7])
 118
 119 static void test_wmemcpy_s(void)
 120 {
 121     static wchar_t dest[8];
 122     static const wchar_t tiny[] = {'T',0,'I','N','Y',0};
 123     static const wchar_t big[] = {'a','t','o','o','l','o','n','g','s','t','r','i','n','g',0};
 124     const wchar_t XX = 0x5858;     /* two 'X' bytes */
 125     int ret;
 126
 127     ok(p_set_invalid_parameter_handler(test_invalid_parameter_handler) == NULL,
 128             "Invalid parameter handler was already set\n");
 129
 130     /* Normal */
 131     memset(dest, 'X', sizeof(dest));
 132     ret = p_wmemcpy_s(dest, NUMELMS(dest), tiny, NUMELMS(tiny));
 133     ok(ret == 0, "Copying a buffer into a big enough destination returned %d, expected 0\n", ret);
 134     okwchars(dest, tiny[0], tiny[1], tiny[2], tiny[3], tiny[4], tiny[5], XX, XX);
 135
 136     /* Vary source size */
 137     errno = 0xdeadbeef;
 138     SET_EXPECT(invalid_parameter_handler);
 139     memset(dest, 'X', sizeof(dest));
 140     ret = p_wmemcpy_s(dest, NUMELMS(dest), big, NUMELMS(big));
 141     ok(errno == ERANGE, "Copying a big buffer to a small destination errno %d, expected ERANGE\n", errno);
 142     ok(ret == ERANGE, "Copying a big buffer to a small destination returned %d, expected ERANGE\n", ret);
 143     okwchars(dest, 0, 0, 0, 0, 0, 0, 0, 0);
 144     CHECK_CALLED(invalid_parameter_handler);
 145
 146     /* Replace source with NULL */
 147     errno = 0xdeadbeef;
 148     SET_EXPECT(invalid_parameter_handler);
 149     memset(dest, 'X', sizeof(dest));
 150     ret = p_wmemcpy_s(dest, NUMELMS(dest), NULL, NUMELMS(tiny));
 151     ok(errno == EINVAL, "Copying a NULL source buffer errno %d, expected EINVAL\n", errno);
 152     ok(ret == EINVAL, "Copying a NULL source buffer returned %d, expected EINVAL\n", ret);
 153     okwchars(dest, 0, 0, 0, 0, 0, 0, 0, 0);
 154     CHECK_CALLED(invalid_parameter_handler);
 155
 156     /* Vary dest size */
 157     errno = 0xdeadbeef;
 158     SET_EXPECT(invalid_parameter_handler);
 159     memset(dest, 'X', sizeof(dest));
 160     ret = p_wmemcpy_s(dest, 0, tiny, NUMELMS(tiny));
 161     ok(errno == ERANGE, "Copying into a destination of size 0 errno %d, expected ERANGE\n", errno);
 162     ok(ret == ERANGE, "Copying into a destination of size 0 returned %d, expected ERANGE\n", ret);
 163     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 164     CHECK_CALLED(invalid_parameter_handler);
 165
 166     /* Replace dest with NULL */
 167     errno = 0xdeadbeef;
 168     SET_EXPECT(invalid_parameter_handler);
 169     ret = p_wmemcpy_s(NULL, NUMELMS(dest), tiny, NUMELMS(tiny));
 170     ok(errno == EINVAL, "Copying a tiny buffer to a big NULL destination errno %d, expected EINVAL\n", errno);
 171     ok(ret == EINVAL, "Copying a tiny buffer to a big NULL destination returned %d, expected EINVAL\n", ret);
 172     CHECK_CALLED(invalid_parameter_handler);
 173
 174     /* Combinations */
 175     errno = 0xdeadbeef;
 176     SET_EXPECT(invalid_parameter_handler);
 177     memset(dest, 'X', sizeof(dest));
 178     ret = p_wmemcpy_s(dest, 0, NULL, NUMELMS(tiny));
 179     ok(errno == EINVAL, "Copying a NULL buffer into a destination of size 0 errno %d, expected EINVAL\n", errno);
 180     ok(ret == EINVAL, "Copying a NULL buffer into a destination of size 0 returned %d, expected EINVAL\n", ret);
 181     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 182     CHECK_CALLED(invalid_parameter_handler);
 183
 184     ok(p_set_invalid_parameter_handler(NULL) == test_invalid_parameter_handler,
 185             "Cannot reset invalid parameter handler\n");
 186 }
 187
 188 static void test_wmemmove_s(void)
 189 {
 190     static wchar_t dest[8];
 191     static const wchar_t tiny[] = {'T',0,'I','N','Y',0};
 192     static const wchar_t big[] = {'a','t','o','o','l','o','n','g','s','t','r','i','n','g',0};
 193     const wchar_t XX = 0x5858;     /* two 'X' bytes */
 194     int ret;
 195
 196     ok(p_set_invalid_parameter_handler(test_invalid_parameter_handler) == NULL,
 197             "Invalid parameter handler was already set\n");
 198
 199     /* Normal */
 200     memset(dest, 'X', sizeof(dest));
 201     ret = p_wmemmove_s(dest, NUMELMS(dest), tiny, NUMELMS(tiny));
 202     ok(ret == 0, "Moving a buffer into a big enough destination returned %d, expected 0\n", ret);
 203     okwchars(dest, tiny[0], tiny[1], tiny[2], tiny[3], tiny[4], tiny[5], XX, XX);
 204
 205     /* Overlapping */
 206     memcpy(dest, big, sizeof(dest));
 207     ret = p_wmemmove_s(dest+1, NUMELMS(dest)-1, dest, NUMELMS(dest)-1);
 208     ok(ret == 0, "Moving a buffer up one char returned %d, expected 0\n", ret);
 209     okwchars(dest, big[0], big[0], big[1], big[2], big[3], big[4], big[5], big[6]);
 210
 211     /* Vary source size */
 212     errno = 0xdeadbeef;
 213     SET_EXPECT(invalid_parameter_handler);
 214     memset(dest, 'X', sizeof(dest));
 215     ret = p_wmemmove_s(dest, NUMELMS(dest), big, NUMELMS(big));
 216     ok(errno == ERANGE, "Moving a big buffer to a small destination errno %d, expected ERANGE\n", errno);
 217     ok(ret == ERANGE, "Moving a big buffer to a small destination returned %d, expected ERANGE\n", ret);
 218     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 219     CHECK_CALLED(invalid_parameter_handler);
 220
 221     /* Replace source with NULL */
 222     errno = 0xdeadbeef;
 223     SET_EXPECT(invalid_parameter_handler);
 224     memset(dest, 'X', sizeof(dest));
 225     ret = p_wmemmove_s(dest, NUMELMS(dest), NULL, NUMELMS(tiny));
 226     ok(errno == EINVAL, "Moving a NULL source buffer errno %d, expected EINVAL\n", errno);
 227     ok(ret == EINVAL, "Moving a NULL source buffer returned %d, expected EINVAL\n", ret);
 228     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 229     CHECK_CALLED(invalid_parameter_handler);
 230
 231     /* Vary dest size */
 232     errno = 0xdeadbeef;
 233     SET_EXPECT(invalid_parameter_handler);
 234     memset(dest, 'X', sizeof(dest));
 235     ret = p_wmemmove_s(dest, 0, tiny, NUMELMS(tiny));
 236     ok(errno == ERANGE, "Moving into a destination of size 0 errno %d, expected ERANGE\n", errno);
 237     ok(ret == ERANGE, "Moving into a destination of size 0 returned %d, expected ERANGE\n", ret);
 238     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 239     CHECK_CALLED(invalid_parameter_handler);
 240
 241     /* Replace dest with NULL */
 242     errno = 0xdeadbeef;
 243     SET_EXPECT(invalid_parameter_handler);
 244     ret = p_wmemmove_s(NULL, NUMELMS(dest), tiny, NUMELMS(tiny));
 245     ok(errno == EINVAL, "Moving a tiny buffer to a big NULL destination errno %d, expected EINVAL\n", errno);
 246     ok(ret == EINVAL, "Moving a tiny buffer to a big NULL destination returned %d, expected EINVAL\n", ret);
 247     CHECK_CALLED(invalid_parameter_handler);
 248
 249     /* Combinations */
 250     errno = 0xdeadbeef;
 251     SET_EXPECT(invalid_parameter_handler);
 252     memset(dest, 'X', sizeof(dest));
 253     ret = p_wmemmove_s(dest, 0, NULL, NUMELMS(tiny));
 254     ok(errno == EINVAL, "Moving a NULL buffer into a destination of size 0 errno %d, expected EINVAL\n", errno);
 255     ok(ret == EINVAL, "Moving a NULL buffer into a destination of size 0 returned %d, expected EINVAL\n", ret);
 256     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 257     CHECK_CALLED(invalid_parameter_handler);
 258
 259     ok(p_set_invalid_parameter_handler(NULL) == test_invalid_parameter_handler,
 260             "Cannot reset invalid parameter handler\n");
 261 }
 262
 263 static void test_fread_s(void)
 264 {
 265     static const char test_file[] = "fread_s.tst";
 266     int ret;
 267     char buf[10];
 268
 269     FILE *f = fopen(test_file, "w");
 270     if(!f) {
 271         skip("Error creating test file\n");
 272         return;
 273     }
 274     fwrite("test", 1, 4, f);
 275     fclose(f);
 276
 277     ok(p_set_invalid_parameter_handler(test_invalid_parameter_handler) == NULL,
 278             "Invalid parameter handler was already set\n");
 279
 280     SET_EXPECT(invalid_parameter_handler);
 281     errno = 0xdeadbeef;
 282     ret = p_fread_s(buf, sizeof(buf), 1, 1, NULL);
 283     ok(ret == 0, "fread_s returned %d, expected 0\n", ret);
 284     ok(errno == EINVAL, "errno = %d, expected EINVAL\n", errno);
 285     CHECK_CALLED(invalid_parameter_handler);
 286
 287     f = p_fopen(test_file, "r");
 288     errno = 0xdeadbeef;
 289     ret = p_fread_s(NULL, sizeof(buf), 0, 1, f);
 290     ok(ret == 0, "fread_s returned %d, expected 0\n", ret);
 291     ok(errno == 0xdeadbeef, "errno = %d, expected 0xdeadbeef\n", errno);
 292     ret = p_fread_s(NULL, sizeof(buf), 1, 0, f);
 293     ok(ret == 0, "fread_s returned %d, expected 0\n", ret);
 294     ok(errno == 0xdeadbeef, "errno = %d, expected 0xdeadbeef\n", errno);
 295
 296     SET_EXPECT(invalid_parameter_handler);
 297     errno = 0xdeadbeef;
 298     ret = p_fread_s(NULL, sizeof(buf), 1, 1, f);
 299     ok(ret == 0, "fread_s returned %d, expected 0\n", ret);
 300     ok(errno == EINVAL, "errno = %d, expected EINVAL\n", errno);
 301     CHECK_CALLED(invalid_parameter_handler);
 302
 303     SET_EXPECT(invalid_parameter_handler);
 304     errno = 0xdeadbeef;
 305     buf[1] = 'a';
 306     ret = p_fread_s(buf, 3, 1, 10, f);
 307     ok(ret == 0, "fread_s returned %d, expected 0\n", ret);
 308     ok(buf[0] == 0, "buf[0] = '%c', expected 0\n", buf[0]);
 309     ok(buf[1] == 0, "buf[1] = '%c', expected 0\n", buf[1]);
 310     ok(errno == ERANGE, "errno = %d, expected ERANGE\n", errno);
 311     CHECK_CALLED(invalid_parameter_handler);
 312
 313     SET_EXPECT(invalid_parameter_handler);
 314     errno = 0xdeadbeef;
 315     ret = p_fread_s(buf, 2, 1, 10, f);
 316     ok(ret == 0, "fread_s returned %d, expected 0\n", ret);
 317     ok(buf[0] == 0, "buf[0] = '%c', expected 0\n", buf[0]);
 318     ok(errno == ERANGE, "errno = %d, expected ERANGE\n", errno);
 319     CHECK_CALLED(invalid_parameter_handler);
 320
 321     memset(buf, 'a', sizeof(buf));
 322     ret = p_fread_s(buf, sizeof(buf), 3, 10, f);
 323     ok(ret==1, "fread_s returned %d, expected 1\n", ret);
 324     ok(buf[0] == 'e', "buf[0] = '%c', expected 'e'\n", buf[0]);
 325     ok(buf[1] == 's', "buf[1] = '%c', expected 's'\n", buf[1]);
 326     ok(buf[2] == 't', "buf[2] = '%c', expected 't'\n", buf[2]);
 327     ok(buf[3] == 'a', "buf[3] = '%c', expected 'a'\n", buf[3]);
 328     p_fclose(f);
 329
 330     ok(p_set_invalid_parameter_handler(NULL) == test_invalid_parameter_handler,
 331             "Cannot reset invalid parameter handler\n");
 332     unlink(test_file);
 333 }
 334
 335 static void test__aligned_msize(void)
 336 {
 337     void *mem;
 338     int ret;
 339
 340     mem = p__aligned_offset_malloc(23, 16, 7);
 341     ret = p__aligned_msize(mem, 16, 7);
 342     ok(ret == 23, "_aligned_msize returned %d\n", ret);
 343     ret = p__aligned_msize(mem, 15, 7);
 344     ok(ret == 24, "_aligned_msize returned %d\n", ret);
 345     ret = p__aligned_msize(mem, 11, 7);
 346     ok(ret == 28, "_aligned_msize returned %d\n", ret);
 347     ret = p__aligned_msize(mem, 1, 7);
 348     ok(ret == 39-sizeof(void*), "_aligned_msize returned %d\n", ret);
 349     ret = p__aligned_msize(mem, 8, 0);
 350     todo_wine ok(ret == 32, "_aligned_msize returned %d\n", ret);
 351     p__aligned_free(mem);
 352
 353     mem = p__aligned_offset_malloc(3, 16, 0);
 354     ret = p__aligned_msize(mem, 16, 0);
 355     ok(ret == 3, "_aligned_msize returned %d\n", ret);
 356     ret = p__aligned_msize(mem, 11, 0);
 357     ok(ret == 8, "_aligned_msize returned %d\n", ret);
 358     p__aligned_free(mem);
 359 }
 360
 361 START_TEST(msvcr100)
 362 {
 363     if (!init())
 364         return;
 365
 366     test_wmemcpy_s();
 367     test_wmemmove_s();
 368     test_fread_s();
 369     test__aligned_msize();
 370 }