dlls/msvcrt/tests/misc.c

   1 /*
   2  * Unit tests for miscellaneous msvcrt functions
   3  *
   4  * Copyright 2010 Andrew Nguyen
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "wine/test.h"
  22 #include <errno.h>
  23 #include "msvcrt.h"
  24
  25 static int (__cdecl *prand_s)(unsigned int *);
  26 static int (__cdecl *memcpy_s)(void *, MSVCRT_size_t, void*, MSVCRT_size_t);
  27
  28 static void init(void)
  29 {
  30     HMODULE hmod = GetModuleHandleA("msvcrt.dll");
  31
  32     prand_s = (void *)GetProcAddress(hmod, "rand_s");
  33     memcpy_s = (void*)GetProcAddress(hmod, "memcpy_s");
  34 }
  35
  36 static void test_rand_s(void)
  37 {
  38     int ret;
  39     unsigned int rand;
  40
  41     if (!prand_s)
  42     {
  43         win_skip("rand_s is not available\n");
  44         return;
  45     }
  46
  47     errno = EBADF;
  48     ret = prand_s(NULL);
  49     ok(ret == EINVAL, "Expected rand_s to return EINVAL, got %d\n", ret);
  50     ok(errno == EINVAL, "Expected errno to return EINVAL, got %d\n", errno);
  51
  52     ret = prand_s(&rand);
  53     ok(ret == 0, "Expected rand_s to return 0, got %d\n", ret);
  54 }
  55
  56 static void test_memcpy_s(void)
  57 {
  58     static char data[] = "data\0to\0be\0copied";
  59     static char dest[32];
  60     int ret;
  61
  62     if(!memcpy_s)
  63     {
  64         win_skip("memcpy_s in not available\n");
  65         return;
  66     }
  67
  68     errno = 0xdeadbeef;
  69     ret = memcpy_s(NULL, 0, NULL, 0);
  70     ok(ret == 0, "ret = %x\n", ret);
  71     ok(errno == 0xdeadbeef, "errno = %x\n", errno);
  72
  73     errno = 0xdeadbeef;
  74     dest[0] = 'x';
  75     ret = memcpy_s(dest, 10, NULL, 0);
  76     ok(ret == 0, "ret = %x\n", ret);
  77     ok(errno == 0xdeadbeef, "errno = %x\n", errno);
  78     ok(dest[0] == 'x', "dest[0] != \'x\'\n");
  79
  80     errno = 0xdeadbeef;
  81     ret = memcpy_s(NULL, 10, data, 10);
  82     ok(ret == EINVAL, "ret = %x\n", ret);
  83     ok(errno == EINVAL, "errno = %x\n", errno);
  84
  85     errno = 0xdeadbeef;
  86     dest[7] = 'x';
  87     ret = memcpy_s(dest, 10, data, 5);
  88     ok(ret == 0, "ret = %x\n", ret);
  89     ok(errno == 0xdeadbeef, "errno = %x\n", errno);
  90     ok(memcmp(dest, data, 10), "All data copied\n");
  91     ok(!memcmp(dest, data, 5), "First five bytes are different\n");
  92
  93     errno = 0xdeadbeef;
  94     ret = memcpy_s(data, 10, data, 10);
  95     ok(ret == 0, "ret = %x\n", ret);
  96     ok(errno == 0xdeadbeef, "errno = %x\n", errno);
  97     ok(!memcmp(dest, data, 5), "data was destroyed during overwritting\n");
  98
  99     errno = 0xdeadbeef;
 100     dest[0] = 'x';
 101     ret = memcpy_s(dest, 5, data, 10);
 102     ok(ret == ERANGE, "ret = %x\n", ret);
 103     ok(errno == ERANGE, "errno = %x\n", errno);
 104     ok(dest[0] == '\0', "dest[0] != \'\\0\'\n");
 105 }
 106
 107 START_TEST(misc)
 108 {
 109     init();
 110
 111     test_rand_s();
 112     test_memcpy_s();
 113 }