ole32: Fix memory leaks in the storage test.
[wine] / dlls / mapi32 / tests / util.c
1 /*
2  * Unit test suite for MAPI utility functions
3  *
4  * Copyright 2004 Jon Griffiths
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 "windef.h"
23 #include "winbase.h"
24 #include "winuser.h"
25 #include "winerror.h"
26 #include "winnt.h"
27 #include "mapiutil.h"
28 #include "mapitags.h"
29 #include "mapi32_test.h"
30
31 static HMODULE hMapi32 = 0;
32
33 static SCODE (WINAPI *pScInitMapiUtil)(ULONG);
34 static void  (WINAPI *pSwapPword)(PUSHORT,ULONG);
35 static void  (WINAPI *pSwapPlong)(PULONG,ULONG);
36 static void  (WINAPI *pHexFromBin)(LPBYTE,int,LPWSTR);
37 static void  (WINAPI *pFBinFromHex)(LPWSTR,LPBYTE);
38 static UINT  (WINAPI *pUFromSz)(LPCSTR);
39 static ULONG (WINAPI *pUlFromSzHex)(LPCSTR);
40 static ULONG (WINAPI *pCbOfEncoded)(LPCSTR);
41 static BOOL  (WINAPI *pIsBadBoundedStringPtr)(LPCSTR,ULONG);
42
43 static void test_SwapPword(void)
44 {
45     USHORT shorts[3];
46
47     pSwapPword = (void*)GetProcAddress(hMapi32, "SwapPword@8");
48     if (!pSwapPword)
49         return;
50
51     shorts[0] = 0xff01;
52     shorts[1] = 0x10ff;
53     shorts[2] = 0x2001;
54     pSwapPword(shorts, 2);
55     ok((shorts[0] == 0x01ff && shorts[1] == 0xff10 && shorts[2] == 0x2001),
56        "Expected {0x01ff,0xff10,0x2001}, got {0x%04x,0x%04x,0x%04x}\n",
57        shorts[0], shorts[1], shorts[2]);
58 }
59
60 static void test_SwapPlong(void)
61 {
62     ULONG longs[3];
63
64     pSwapPlong = (void*)GetProcAddress(hMapi32, "SwapPlong@8");
65     if (!pSwapPlong)
66         return;
67
68     longs[0] = 0xffff0001;
69     longs[1] = 0x1000ffff;
70     longs[2] = 0x20000001;
71     pSwapPlong(longs, 2);
72     ok((longs[0] == 0x0100ffff && longs[1] == 0xffff0010 && longs[2] == 0x20000001),
73        "Expected {0x0100ffff,0xffff0010,0x20000001}, got {0x%08x,0x%08x,0x%08x}\n",
74        longs[0], longs[1], longs[2]);
75 }
76
77 static void test_HexFromBin(void)
78 {
79     static char res[] =       { "000102030405060708090A0B0C0D0E0F101112131415161"
80       "718191A1B1C1D1E1F202122232425262728292A2B2C2D2E2F303132333435363738393A3B"
81       "3C3D3E3F404142434445464748494A4B4C4D4E4F505152535455565758595A5B5C5D5E5F6"
82       "06162636465666768696A6B6C6D6E6F707172737475767778797A7B7C7D7E7F8081828384"
83       "85868788898A8B8C8D8E8F909192939495969798999A9B9C9D9E9FA0A1A2A3A4A5A6A7A8A"
84       "9AAABACADAEAFB0B1B2B3B4B5B6B7B8B9BABBBCBDBEBFC0C1C2C3C4C5C6C7C8C9CACBCCCD"
85       "CECFD0D1D2D3D4D5D6D7D8D9DADBDCDDDEDFE0E1E2E3E4E5E6E7E8E9EAEBECEDEEEFF0F1F"
86       "2F3F4F5F6F7F8F9FAFBFCFDFE\0X" };
87     BYTE data[255];
88     WCHAR strw[256];
89     BOOL bOk;
90     int i;
91
92     pHexFromBin = (void*)GetProcAddress(hMapi32, "HexFromBin@12");
93     pFBinFromHex = (void*)GetProcAddress(hMapi32, "FBinFromHex@8");
94     if (!pHexFromBin || !pFBinFromHex)
95         return;
96
97     for (i = 0; i < 255; i++)
98         data[i] = i;
99     memset(strw, 'X', sizeof(strw));
100     pHexFromBin(data, sizeof(data), strw);
101
102     ok(memcmp(strw, res, sizeof(res) - 1) == 0, "HexFromBin: Result differs\n");
103
104     memset(data, 0, sizeof(data));
105     pFBinFromHex((LPWSTR)res, data);
106     bOk = TRUE;
107     for (i = 0; i < 255; i++)
108         if (data[i] != i)
109             bOk = FALSE;
110     ok(bOk == TRUE, "FBinFromHex: Result differs\n");
111 }
112
113 static void test_UFromSz(void)
114 {
115     pUFromSz = (void*)GetProcAddress(hMapi32, "UFromSz@4");
116     if (!pUFromSz)
117         return;
118
119     ok(pUFromSz("105679") == 105679u,
120        "UFromSz: expected 105679, got %d\n", pUFromSz("105679"));
121
122     ok(pUFromSz(" 4") == 0, "UFromSz: exected 0. got %d\n",
123        pUFromSz(" 4"));
124 }
125
126 static void test_UlFromSzHex(void)
127 {
128     pUlFromSzHex = (void*)GetProcAddress(hMapi32, "UlFromSzHex@4");
129     if (!pUlFromSzHex)
130         return;
131
132     ok(pUlFromSzHex("fF") == 0xffu,
133        "UlFromSzHex: expected 0xff, got 0x%x\n", pUlFromSzHex("fF"));
134
135     ok(pUlFromSzHex(" c") == 0, "UlFromSzHex: exected 0x0. got 0x%x\n",
136        pUlFromSzHex(" c"));
137 }
138
139 static void test_CbOfEncoded(void)
140 {
141     char buff[129];
142     unsigned int i;
143
144     pCbOfEncoded = (void*)GetProcAddress(hMapi32, "CbOfEncoded@4");
145     if (!pCbOfEncoded)
146         return;
147
148     for (i = 0; i < sizeof(buff) - 1; i++)
149     {
150         ULONG ulRet, ulExpected = (((i | 3) >> 2) + 1) * 3;
151
152         memset(buff, '\0', sizeof(buff));
153         memset(buff, '?', i);
154         ulRet = pCbOfEncoded(buff);
155         ok(ulRet == ulExpected,
156            "CbOfEncoded(length %d): expected %d, got %d\n",
157            i, ulExpected, ulRet);
158     }
159 }
160
161 static void test_IsBadBoundedStringPtr(void)
162 {
163     pIsBadBoundedStringPtr = (void*)GetProcAddress(hMapi32, "IsBadBoundedStringPtr@8");
164     if (!pIsBadBoundedStringPtr)
165         return;
166
167     ok(pIsBadBoundedStringPtr(NULL, 0) == TRUE, "IsBadBoundedStringPtr: expected TRUE\n");
168     ok(pIsBadBoundedStringPtr("TEST", 4) == TRUE, "IsBadBoundedStringPtr: expected TRUE\n");
169     ok(pIsBadBoundedStringPtr("TEST", 5) == FALSE, "IsBadBoundedStringPtr: expected FALSE\n");
170 }
171
172 START_TEST(util)
173 {
174     SCODE ret;
175
176     if (!HaveDefaultMailClient())
177     {
178         win_skip("No default mail client installed\n");
179         return;
180     }
181
182     hMapi32 = LoadLibraryA("mapi32.dll");
183
184     pScInitMapiUtil = (void*)GetProcAddress(hMapi32, "ScInitMapiUtil@4");
185
186     if (!pScInitMapiUtil)
187     {
188         win_skip("ScInitMapiUtil is not available\n");
189         FreeLibrary(hMapi32);
190         return;
191     }
192
193     SetLastError(0xdeadbeef);
194     ret = pScInitMapiUtil(0);
195     if ((ret != S_OK) && (GetLastError() == ERROR_PROC_NOT_FOUND))
196     {
197         win_skip("ScInitMapiUtil is not implemented\n");
198         FreeLibrary(hMapi32);
199         return;
200     }
201     else if ((ret == E_FAIL) && (GetLastError() == ERROR_INVALID_HANDLE))
202     {
203         win_skip("ScInitMapiUtil doesn't work on some Win98 and WinME systems\n");
204         FreeLibrary(hMapi32);
205         return;
206     }
207
208     test_SwapPword();
209     test_SwapPlong();
210     test_HexFromBin();
211     test_UFromSz();
212     test_UlFromSzHex();
213     test_CbOfEncoded();
214     test_IsBadBoundedStringPtr();
215
216     FreeLibrary(hMapi32);
217 }