msvcrt: Fix wcsncpy_s implementation.
[wine] / dlls / msvcrt / tests / data.c
1 /*
2  * Tests msvcrt/data.c
3  *
4  * Copyright 2006 Andrew Ziem
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 <stdarg.h>
23 #include <stdlib.h>
24 #include <stdio.h>
25 #include <fcntl.h>
26 #include <sys/stat.h>
27 #include <io.h>
28 #include <windef.h>
29 #include <winbase.h>
30 #include <winnls.h>
31 #include <process.h>
32 #include <errno.h>
33
34 typedef void (__cdecl *_INITTERMFUN)(void);
35 static void (__cdecl *p_initterm)(_INITTERMFUN *start, _INITTERMFUN *end);
36
37 static int (__cdecl *p_get_pgmptr)(char **p);
38
39 static int callbacked;
40
41 static void __cdecl initcallback(void)
42 {
43    callbacked++;
44 }
45
46 #define initterm_test(start, end, expected) \
47     callbacked = 0; \
48     p_initterm(start, end); \
49     ok(expected == callbacked,"_initterm: callbacks count mismatch: got %i, expected %i\n", callbacked, expected);
50
51 static void test_initterm(void)
52 {
53     int i;
54     static _INITTERMFUN callbacks[4];
55
56     if (!p_initterm)
57         return;
58
59     for (i = 0; i < 4; i++)
60     {
61         callbacks[i] = initcallback;
62     }
63
64     initterm_test(&callbacks[0], &callbacks[1], 1);
65     initterm_test(&callbacks[0], &callbacks[2], 2);
66     initterm_test(&callbacks[0], &callbacks[3], 3);
67
68     callbacks[1] = NULL;
69     initterm_test(&callbacks[0], &callbacks[3], 2);
70 }
71
72 static void test_initvar( HMODULE hmsvcrt )
73 {
74     OSVERSIONINFO osvi = { sizeof(OSVERSIONINFO) };
75     int *pp_winver   = (int*)GetProcAddress(hmsvcrt, "_winver");
76     int *pp_winmajor = (int*)GetProcAddress(hmsvcrt, "_winmajor");
77     int *pp_winminor = (int*)GetProcAddress(hmsvcrt, "_winminor");
78     int *pp_osver    = (int*)GetProcAddress(hmsvcrt, "_osver");
79     int *pp_osplatform = (int*)GetProcAddress(hmsvcrt, "_osplatform");
80     unsigned int winver, winmajor, winminor, osver, osplatform;
81
82     if( !( pp_winmajor && pp_winminor && pp_winver)) {
83         win_skip("_winver variables are not available\n");
84         return;
85     }
86     winver = *pp_winver;
87     winminor = *pp_winminor;
88     winmajor = *pp_winmajor;
89     GetVersionEx( &osvi);
90     ok( winminor == osvi.dwMinorVersion, "Wrong value for _winminor %02x expected %02x\n",
91             winminor, osvi.dwMinorVersion);
92     ok( winmajor == osvi.dwMajorVersion, "Wrong value for _winmajor %02x expected %02x\n",
93             winmajor, osvi.dwMajorVersion);
94     ok( winver == ((osvi.dwMajorVersion << 8) | osvi.dwMinorVersion),
95             "Wrong value for _winver %02x expected %02x\n",
96             winver, ((osvi.dwMajorVersion << 8) | osvi.dwMinorVersion));
97     if( !pp_osver || !pp_osplatform ) {
98         win_skip("_osver variables are not available\n");
99         return;
100     }
101     osver = *pp_osver;
102     osplatform = *pp_osplatform;
103     ok( osver == (osvi.dwBuildNumber & 0xffff) ||
104             ((osvi.dwBuildNumber >> 24) == osvi.dwMajorVersion &&
105                  ((osvi.dwBuildNumber >> 16) & 0xff) == osvi.dwMinorVersion), /* 95/98/ME */
106             "Wrong value for _osver %04x expected %04x\n",
107             osver, osvi.dwBuildNumber);
108     ok(osplatform == osvi.dwPlatformId,
109             "Wrong value for _osplatform %x exprected %x\n",
110             osplatform, osvi.dwPlatformId);
111 }
112
113 static void test_get_pgmptr(void)
114 {
115     char *pgm = NULL;
116     int res;
117
118     if (!p_get_pgmptr)
119         return;
120
121     res = p_get_pgmptr(&pgm);
122
123     ok( res == 0, "Wrong _get_pgmptr return value %d expected 0\n", res);
124     ok( pgm != NULL, "_get_pgmptr returned a NULL pointer\n" );
125 }
126
127 START_TEST(data)
128 {
129     HMODULE hmsvcrt;
130
131     hmsvcrt = GetModuleHandleA("msvcrt.dll");
132     if (!hmsvcrt)
133         hmsvcrt = GetModuleHandleA("msvcrtd.dll");
134     if (hmsvcrt)
135     {
136         p_initterm=(void*)GetProcAddress(hmsvcrt, "_initterm");
137         p_get_pgmptr=(void*)GetProcAddress(hmsvcrt, "_get_pgmptr");
138     }
139     test_initterm();
140     test_initvar(hmsvcrt);
141     test_get_pgmptr();
142 }