msvcp90/tests: Added char_traits<char>::_Copy_s tests.
[wine] / dlls / msvcp90 / tests / misc.c
1 /*
2  * Copyright 2010 Piotr Caban for CodeWeavers
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 <stdio.h>
20
21 #include <windef.h>
22 #include <winbase.h>
23 #include "wine/test.h"
24
25 static void* (__cdecl *p_set_invalid_parameter_handler)(void*);
26
27 static void (__cdecl *p_char_assign)(void*, const void*);
28 static void (__cdecl *p_wchar_assign)(void*, const void*);
29 static void (__cdecl *p_short_assign)(void*, const void*);
30
31 static BYTE (__cdecl *p_char_eq)(const void*, const void*);
32 static BYTE (__cdecl *p_wchar_eq)(const void*, const void*);
33 static BYTE (__cdecl *p_short_eq)(const void*, const void*);
34
35 static char* (__cdecl *p_Copy_s)(char*, unsigned int, const char*, unsigned int);
36
37 static int invalid_parameter = 0;
38 static void __cdecl test_invalid_parameter_handler(const wchar_t *expression,
39         const wchar_t *function, const wchar_t *file,
40         unsigned line, uintptr_t arg)
41 {
42     ok(expression == NULL, "expression is not NULL\n");
43     ok(function == NULL, "function is not NULL\n");
44     ok(file == NULL, "file is not NULL\n");
45     ok(line == 0, "line = %u\n", line);
46     ok(arg == 0, "arg = %lx\n", (UINT_PTR)arg);
47     invalid_parameter++;
48 }
49
50 static BOOL init(void)
51 {
52     HMODULE msvcr = LoadLibraryA("msvcr90.dll");
53     HMODULE msvcp = LoadLibraryA("msvcp90.dll");
54     if(!msvcr || !msvcp) {
55         win_skip("msvcp90.dll or msvcrt90.dll not installed\n");
56         return FALSE;
57     }
58
59     p_set_invalid_parameter_handler = (void*)GetProcAddress(msvcr, "_set_invalid_parameter_handler");
60     if(!p_set_invalid_parameter_handler) {
61         win_skip("Error setting tests environment\n");
62         return FALSE;
63     }
64
65     p_set_invalid_parameter_handler(test_invalid_parameter_handler);
66
67     p_char_assign = (void*)GetProcAddress(msvcp, "?assign@?$char_traits@D@std@@SAXAADABD@Z");
68     p_wchar_assign = (void*)GetProcAddress(msvcp, "?assign@?$char_traits@_W@std@@SAXAA_WAB_W@Z");
69     p_short_assign = (void*)GetProcAddress(msvcp, "?assign@?$char_traits@G@std@@SAXAAGABG@Z");
70
71     p_char_eq = (void*)GetProcAddress(msvcp, "?eq@?$char_traits@D@std@@SA_NABD0@Z");
72     p_wchar_eq = (void*)GetProcAddress(msvcp, "?eq@?$char_traits@_W@std@@SA_NAB_W0@Z");
73     p_short_eq = (void*)GetProcAddress(msvcp, "?eq@?$char_traits@G@std@@SA_NABG0@Z");
74
75     p_Copy_s = (void*)GetProcAddress(msvcp, "?_Copy_s@?$char_traits@D@std@@SAPADPADIPBDI@Z");
76
77     return TRUE;
78 }
79
80 static void test_assign(void)
81 {
82     const char in[] = "abc";
83     char out[4];
84
85     if(!p_char_assign || !p_wchar_assign || !p_short_assign) {
86         win_skip("assign tests skipped\n");
87         return;
88     }
89
90     out[1] = '#';
91     p_char_assign(out, in);
92     ok(out[0] == in[0], "out[0] = %c\n", out[0]);
93     ok(out[1] == '#', "out[1] = %c\n", out[1]);
94
95     out[2] = '#';
96     p_wchar_assign(out, in);
97     ok(*((char*)out)==in[0] && *((char*)out+1)==in[1],
98             "out[0] = %d, out[1] = %d\n", (int)out[0], (int)out[1]);
99     ok(out[2] == '#', "out[2] = %c\n", out[2]);
100
101     out[2] = '#';
102     p_short_assign(out, in);
103     ok(*((char*)out)==in[0] && *((char*)out+1)==in[1],
104             "out[0] = %d, out[1] = %d\n", (int)out[0], (int)out[1]);
105     ok(out[2] == '#', "out[2] = %c\n", out[2]);
106 }
107
108 static void test_equal(void)
109 {
110     static const char in1[] = "abc";
111     static const char in2[] = "ab";
112     static const char in3[] = "a";
113     static const char in4[] = "b";
114     BYTE ret;
115
116     if(!p_char_eq || !p_wchar_eq || !p_short_eq) {
117         win_skip("equal tests skipped\n");
118         return;
119     }
120
121     ret = p_char_eq(in1, in2);
122     ok(ret == TRUE, "ret = %d\n", (int)ret);
123     ret = p_char_eq(in1, in3);
124     ok(ret == TRUE, "ret = %d\n", (int)ret);
125     ret = p_char_eq(in1, in4);
126     ok(ret == FALSE, "ret = %d\n", (int)ret);
127
128     ret = p_wchar_eq(in1, in2);
129     ok(ret == TRUE, "ret = %d\n", (int)ret);
130     ret = p_wchar_eq(in1, in3);
131     ok(ret == FALSE, "ret = %d\n", (int)ret);
132     ret = p_wchar_eq(in1, in4);
133     ok(ret == FALSE, "ret = %d\n", (int)ret);
134
135     ret = p_short_eq(in1, in2);
136     ok(ret == TRUE, "ret = %d\n", (int)ret);
137     ret = p_short_eq(in1, in3);
138     ok(ret == FALSE, "ret = %d\n", (int)ret);
139     ret = p_short_eq(in1, in4);
140     ok(ret == FALSE, "ret = %d\n", (int)ret);
141 }
142
143 static void test_Copy_s(void)
144 {
145     static const char src[] = "abcd";
146     char dest[32], *ret;
147
148     if(!p_Copy_s) {
149         win_skip("Copy_s tests skipped\n");
150         return;
151     }
152
153     dest[4] = '#';
154     dest[5] = '\0';
155     ret = p_Copy_s(dest, 4, src, 4);
156     ok(ret == dest, "ret != dest\n");
157     ok(dest[4] == '#', "dest[4] != '#'\n");
158     ok(!memcmp(dest, src, sizeof(char[4])), "dest = %s\n", dest);
159
160     ret = p_Copy_s(dest, 32, src, 4);
161     ok(ret == dest, "ret != dest\n");
162     ok(dest[4] == '#', "dest[4] != '#'\n");
163     ok(!memcmp(dest, src, sizeof(char[4])), "dest = %s\n", dest);
164
165     errno = 0xdeadbeef;
166     dest[0] = '#';
167     ret = p_Copy_s(dest, 3, src, 4);
168     ok(ret == dest, "ret != dest\n");
169     ok(dest[0] == '\0', "dest[0] != 0\n");
170     ok(invalid_parameter==1, "invalid_parameter = %d\n",
171             invalid_parameter);
172     invalid_parameter = 0;
173     ok(errno == 0xdeadbeef, "errno = %d\n", errno);
174
175     errno = 0xdeadbeef;
176     p_Copy_s(NULL, 32, src, 4);
177     ok(invalid_parameter==1, "invalid_parameter = %d\n",
178             invalid_parameter);
179     invalid_parameter = 0;
180     ok(errno == 0xdeadbeef, "errno = %d\n", errno);
181
182     errno = 0xdeadbeef;
183     p_Copy_s(dest, 32, NULL, 4);
184     ok(invalid_parameter==1, "invalid_parameter = %d\n",
185             invalid_parameter);
186     invalid_parameter = 0;
187     ok(errno == 0xdeadbeef, "errno = %d\n", errno);
188 }
189
190 START_TEST(misc)
191 {
192     if(!init())
193         return;
194
195     test_assign();
196     test_equal();
197     test_Copy_s();
198
199     ok(!invalid_parameter, "invalid_parameter_handler was invoked too many times\n");
200 }