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
  24 #include <windef.h>
  25 #include <winbase.h>
  26 #include "wine/test.h"
  27
  28 #define DEFINE_EXPECT(func) \
  29     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
  30
  31 #define SET_EXPECT(func) \
  32     expect_ ## func = TRUE
  33
  34 #define CHECK_EXPECT2(func) \
  35     do { \
  36         ok(expect_ ##func, "unexpected call " #func "\n"); \
  37         called_ ## func = TRUE; \
  38     }while(0)
  39
  40 #define CHECK_EXPECT(func) \
  41     do { \
  42         CHECK_EXPECT2(func); \
  43         expect_ ## func = FALSE; \
  44     }while(0)
  45
  46 #define CHECK_CALLED(func) \
  47     do { \
  48         ok(called_ ## func, "expected " #func "\n"); \
  49         expect_ ## func = called_ ## func = FALSE; \
  50     }while(0)
  51
  52 DEFINE_EXPECT(invalid_parameter_handler);
  53
  54 static _invalid_parameter_handler (__cdecl *p_set_invalid_parameter_handler)(_invalid_parameter_handler);
  55
  56 static void __cdecl test_invalid_parameter_handler(const wchar_t *expression,
  57         const wchar_t *function, const wchar_t *file,
  58         unsigned line, uintptr_t arg)
  59 {
  60     CHECK_EXPECT(invalid_parameter_handler);
  61     ok(expression == NULL, "expression is not NULL\n");
  62     ok(function == NULL, "function is not NULL\n");
  63     ok(file == NULL, "file is not NULL\n");
  64     ok(line == 0, "line = %u\n", line);
  65     ok(arg == 0, "arg = %lx\n", (UINT_PTR)arg);
  66 }
  67
  68 static int* (__cdecl *p_errno)(void);
  69 static int (__cdecl *p_wmemcpy_s)(wchar_t *dest, size_t numberOfElements, const wchar_t *src, size_t count);
  70 static int (__cdecl *p_wmemmove_s)(wchar_t *dest, size_t numberOfElements, const wchar_t *src, size_t count);
  71
  72 /* make sure we use the correct errno */
  73 #undef errno
  74 #define errno (*p_errno())
  75
  76 #define SETNOFAIL(x,y) x = (void*)GetProcAddress(hcrt,y)
  77 #define SET(x,y) do { SETNOFAIL(x,y); ok(x != NULL, "Export '%s' not found\n", y); } while(0)
  78
  79 static BOOL init(void)
  80 {
  81     HMODULE hcrt;
  82
  83     SetLastError(0xdeadbeef);
  84     hcrt = LoadLibraryA("msvcr100.dll");
  85     if (!hcrt) {
  86         win_skip("msvcr100.dll not installed (got %d)\n", GetLastError());
  87         return FALSE;
  88     }
  89
  90     SET(p_errno, "_errno");
  91     SET(p_set_invalid_parameter_handler, "_set_invalid_parameter_handler");
  92     SET(p_wmemcpy_s, "wmemcpy_s");
  93     SET(p_wmemmove_s, "wmemmove_s");
  94
  95     return TRUE;
  96 }
  97
  98 #define NUMELMS(array) (sizeof(array)/sizeof((array)[0]))
  99
 100 #define okwchars(dst, b0, b1, b2, b3, b4, b5, b6, b7) \
 101     ok(dst[0] == b0 && dst[1] == b1 && dst[2] == b2 && dst[3] == b3 && \
 102        dst[4] == b4 && dst[5] == b5 && dst[6] == b6 && dst[7] == b7, \
 103        "Bad result: 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x\n",\
 104        dst[0], dst[1], dst[2], dst[3], dst[4], dst[5], dst[6], dst[7])
 105
 106 static void test_wmemcpy_s(void)
 107 {
 108     static wchar_t dest[8];
 109     static const wchar_t tiny[] = {'T',0,'I','N','Y',0};
 110     static const wchar_t big[] = {'a','t','o','o','l','o','n','g','s','t','r','i','n','g',0};
 111     const wchar_t XX = 0x5858;     /* two 'X' bytes */
 112     int ret;
 113
 114     ok(p_set_invalid_parameter_handler(test_invalid_parameter_handler) == NULL,
 115             "Invalid parameter handler was already set\n");
 116
 117     /* Normal */
 118     memset(dest, 'X', sizeof(dest));
 119     ret = p_wmemcpy_s(dest, NUMELMS(dest), tiny, NUMELMS(tiny));
 120     ok(ret == 0, "Copying a buffer into a big enough destination returned %d, expected 0\n", ret);
 121     okwchars(dest, tiny[0], tiny[1], tiny[2], tiny[3], tiny[4], tiny[5], XX, XX);
 122
 123     /* Vary source size */
 124     errno = 0xdeadbeef;
 125     SET_EXPECT(invalid_parameter_handler);
 126     memset(dest, 'X', sizeof(dest));
 127     ret = p_wmemcpy_s(dest, NUMELMS(dest), big, NUMELMS(big));
 128     ok(errno == ERANGE, "Copying a big buffer to a small destination errno %d, expected ERANGE\n", errno);
 129     ok(ret == ERANGE, "Copying a big buffer to a small destination returned %d, expected ERANGE\n", ret);
 130     okwchars(dest, 0, 0, 0, 0, 0, 0, 0, 0);
 131     CHECK_CALLED(invalid_parameter_handler);
 132
 133     /* Replace source with NULL */
 134     errno = 0xdeadbeef;
 135     SET_EXPECT(invalid_parameter_handler);
 136     memset(dest, 'X', sizeof(dest));
 137     ret = p_wmemcpy_s(dest, NUMELMS(dest), NULL, NUMELMS(tiny));
 138     ok(errno == EINVAL, "Copying a NULL source buffer errno %d, expected EINVAL\n", errno);
 139     ok(ret == EINVAL, "Copying a NULL source buffer returned %d, expected EINVAL\n", ret);
 140     okwchars(dest, 0, 0, 0, 0, 0, 0, 0, 0);
 141     CHECK_CALLED(invalid_parameter_handler);
 142
 143     /* Vary dest size */
 144     errno = 0xdeadbeef;
 145     SET_EXPECT(invalid_parameter_handler);
 146     memset(dest, 'X', sizeof(dest));
 147     ret = p_wmemcpy_s(dest, 0, tiny, NUMELMS(tiny));
 148     ok(errno == ERANGE, "Copying into a destination of size 0 errno %d, expected ERANGE\n", errno);
 149     ok(ret == ERANGE, "Copying into a destination of size 0 returned %d, expected ERANGE\n", ret);
 150     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 151     CHECK_CALLED(invalid_parameter_handler);
 152
 153     /* Replace dest with NULL */
 154     errno = 0xdeadbeef;
 155     SET_EXPECT(invalid_parameter_handler);
 156     ret = p_wmemcpy_s(NULL, NUMELMS(dest), tiny, NUMELMS(tiny));
 157     ok(errno == EINVAL, "Copying a tiny buffer to a big NULL destination errno %d, expected EINVAL\n", errno);
 158     ok(ret == EINVAL, "Copying a tiny buffer to a big NULL destination returned %d, expected EINVAL\n", ret);
 159     CHECK_CALLED(invalid_parameter_handler);
 160
 161     /* Combinations */
 162     errno = 0xdeadbeef;
 163     SET_EXPECT(invalid_parameter_handler);
 164     memset(dest, 'X', sizeof(dest));
 165     ret = p_wmemcpy_s(dest, 0, NULL, NUMELMS(tiny));
 166     ok(errno == EINVAL, "Copying a NULL buffer into a destination of size 0 errno %d, expected EINVAL\n", errno);
 167     ok(ret == EINVAL, "Copying a NULL buffer into a destination of size 0 returned %d, expected EINVAL\n", ret);
 168     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 169     CHECK_CALLED(invalid_parameter_handler);
 170
 171     ok(p_set_invalid_parameter_handler(NULL) == test_invalid_parameter_handler,
 172             "Cannot reset invalid parameter handler\n");
 173 }
 174
 175 static void test_wmemmove_s(void)
 176 {
 177     static wchar_t dest[8];
 178     static const wchar_t tiny[] = {'T',0,'I','N','Y',0};
 179     static const wchar_t big[] = {'a','t','o','o','l','o','n','g','s','t','r','i','n','g',0};
 180     const wchar_t XX = 0x5858;     /* two 'X' bytes */
 181     int ret;
 182
 183     ok(p_set_invalid_parameter_handler(test_invalid_parameter_handler) == NULL,
 184             "Invalid parameter handler was already set\n");
 185
 186     /* Normal */
 187     memset(dest, 'X', sizeof(dest));
 188     ret = p_wmemmove_s(dest, NUMELMS(dest), tiny, NUMELMS(tiny));
 189     ok(ret == 0, "Moving a buffer into a big enough destination returned %d, expected 0\n", ret);
 190     okwchars(dest, tiny[0], tiny[1], tiny[2], tiny[3], tiny[4], tiny[5], XX, XX);
 191
 192     /* Overlapping */
 193     memcpy(dest, big, sizeof(dest));
 194     ret = p_wmemmove_s(dest+1, NUMELMS(dest)-1, dest, NUMELMS(dest)-1);
 195     ok(ret == 0, "Moving a buffer up one char returned %d, expected 0\n", ret);
 196     okwchars(dest, big[0], big[0], big[1], big[2], big[3], big[4], big[5], big[6]);
 197
 198     /* Vary source size */
 199     errno = 0xdeadbeef;
 200     SET_EXPECT(invalid_parameter_handler);
 201     memset(dest, 'X', sizeof(dest));
 202     ret = p_wmemmove_s(dest, NUMELMS(dest), big, NUMELMS(big));
 203     ok(errno == ERANGE, "Moving a big buffer to a small destination errno %d, expected ERANGE\n", errno);
 204     ok(ret == ERANGE, "Moving a big buffer to a small destination returned %d, expected ERANGE\n", ret);
 205     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 206     CHECK_CALLED(invalid_parameter_handler);
 207
 208     /* Replace source with NULL */
 209     errno = 0xdeadbeef;
 210     SET_EXPECT(invalid_parameter_handler);
 211     memset(dest, 'X', sizeof(dest));
 212     ret = p_wmemmove_s(dest, NUMELMS(dest), NULL, NUMELMS(tiny));
 213     ok(errno == EINVAL, "Moving a NULL source buffer errno %d, expected EINVAL\n", errno);
 214     ok(ret == EINVAL, "Moving a NULL source buffer returned %d, expected EINVAL\n", ret);
 215     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 216     CHECK_CALLED(invalid_parameter_handler);
 217
 218     /* Vary dest size */
 219     errno = 0xdeadbeef;
 220     SET_EXPECT(invalid_parameter_handler);
 221     memset(dest, 'X', sizeof(dest));
 222     ret = p_wmemmove_s(dest, 0, tiny, NUMELMS(tiny));
 223     ok(errno == ERANGE, "Moving into a destination of size 0 errno %d, expected ERANGE\n", errno);
 224     ok(ret == ERANGE, "Moving into a destination of size 0 returned %d, expected ERANGE\n", ret);
 225     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 226     CHECK_CALLED(invalid_parameter_handler);
 227
 228     /* Replace dest with NULL */
 229     errno = 0xdeadbeef;
 230     SET_EXPECT(invalid_parameter_handler);
 231     ret = p_wmemmove_s(NULL, NUMELMS(dest), tiny, NUMELMS(tiny));
 232     ok(errno == EINVAL, "Moving a tiny buffer to a big NULL destination errno %d, expected EINVAL\n", errno);
 233     ok(ret == EINVAL, "Moving a tiny buffer to a big NULL destination returned %d, expected EINVAL\n", ret);
 234     CHECK_CALLED(invalid_parameter_handler);
 235
 236     /* Combinations */
 237     errno = 0xdeadbeef;
 238     SET_EXPECT(invalid_parameter_handler);
 239     memset(dest, 'X', sizeof(dest));
 240     ret = p_wmemmove_s(dest, 0, NULL, NUMELMS(tiny));
 241     ok(errno == EINVAL, "Moving a NULL buffer into a destination of size 0 errno %d, expected EINVAL\n", errno);
 242     ok(ret == EINVAL, "Moving a NULL buffer into a destination of size 0 returned %d, expected EINVAL\n", ret);
 243     okwchars(dest, XX, XX, XX, XX, XX, XX, XX, XX);
 244     CHECK_CALLED(invalid_parameter_handler);
 245
 246     ok(p_set_invalid_parameter_handler(NULL) == test_invalid_parameter_handler,
 247             "Cannot reset invalid parameter handler\n");
 248 }
 249
 250 START_TEST(msvcr100)
 251 {
 252     if (!init())
 253         return;
 254
 255     test_wmemcpy_s();
 256     test_wmemmove_s();
 257 }