Fixed header dependencies to be fully compatible with the Windows
[wine/multimedia.git] / dlls / shlwapi / tests / clist.c
blob9617b5567751a214dc88b13053d4f7bb012dbeb9
1 /* Unit test suite for SHLWAPI Compact List and IStream ordinal functions
3 * Copyright 2002 Jon Griffiths
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2.1 of the License, or (at your option) any later version.
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 #include <stdarg.h>
22 #include "wine/test.h"
23 #include "windef.h"
24 #include "winbase.h"
25 #include "objbase.h"
27 typedef struct tagSHLWAPI_CLIST
29 ULONG ulSize;
30 ULONG ulId;
31 } SHLWAPI_CLIST, *LPSHLWAPI_CLIST;
33 typedef const SHLWAPI_CLIST* LPCSHLWAPI_CLIST;
35 /* Items to add */
36 static const SHLWAPI_CLIST SHLWAPI_CLIST_items[] =
38 {4, 1},
39 {8, 3},
40 {12, 2},
41 {16, 8},
42 {20, 9},
43 {3, 11},
44 {9, 82},
45 {33, 16},
46 {32, 55},
47 {24, 100},
48 {39, 116},
49 { 0, 0}
52 /* Dummy IStream object for testing calls */
53 typedef struct
55 void* lpVtbl;
56 ULONG ref;
57 int readcalls;
58 BOOL failreadcall;
59 BOOL failreadsize;
60 BOOL readbeyondend;
61 BOOL readreturnlarge;
62 int writecalls;
63 BOOL failwritecall;
64 BOOL failwritesize;
65 int seekcalls;
66 int statcalls;
67 BOOL failstatcall;
68 LPCSHLWAPI_CLIST item;
69 ULARGE_INTEGER pos;
70 } _IDummyStream;
72 static
73 HRESULT WINAPI QueryInterface(_IDummyStream *This,REFIID riid, LPVOID *ppvObj)
75 return S_OK;
78 static ULONG WINAPI AddRef(_IDummyStream *This)
80 return ++This->ref;
83 static ULONG WINAPI Release(_IDummyStream *This)
85 return --This->ref;
88 static HRESULT WINAPI Read(_IDummyStream* This, LPVOID lpMem, ULONG ulSize,
89 PULONG lpRead)
91 HRESULT hRet = S_OK;
92 ++This->readcalls;
94 if (This->failreadcall)
96 return STG_E_ACCESSDENIED;
98 else if (This->failreadsize)
100 *lpRead = ulSize + 8;
101 return S_OK;
103 else if (This->readreturnlarge)
105 *((ULONG*)lpMem) = 0xffff01;
106 *lpRead = ulSize;
107 This->readreturnlarge = FALSE;
108 return S_OK;
110 if (ulSize == sizeof(ULONG))
112 /* Read size of item */
113 *((ULONG*)lpMem) = This->item->ulSize ? This->item->ulSize + sizeof(SHLWAPI_CLIST) : 0;
114 *lpRead = ulSize;
116 else
118 unsigned int i;
119 char* buff = (char*)lpMem;
121 /* Read item data */
122 if (!This->item->ulSize)
124 This->readbeyondend = TRUE;
125 *lpRead = 0;
126 return E_FAIL; /* Should never happen */
128 *((ULONG*)lpMem) = This->item->ulId;
129 *lpRead = ulSize;
131 for (i = 0; i < This->item->ulSize; i++)
132 buff[4+i] = i*2;
134 This->item++;
136 return hRet;
139 static HRESULT WINAPI Write(_IDummyStream* This, LPVOID lpMem, ULONG ulSize,
140 PULONG lpWritten)
142 HRESULT hRet = S_OK;
144 ++This->writecalls;
145 if (This->failwritecall)
147 return STG_E_ACCESSDENIED;
149 else if (This->failwritesize)
151 *lpWritten = 0;
153 else
154 *lpWritten = ulSize;
155 return hRet;
158 static HRESULT WINAPI Seek(_IDummyStream* This, LARGE_INTEGER dlibMove,
159 DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition)
161 ++This->seekcalls;
162 This->pos.QuadPart = dlibMove.QuadPart;
163 if (plibNewPosition)
164 plibNewPosition->QuadPart = dlibMove.QuadPart;
165 return S_OK;
168 static HRESULT WINAPI Stat(_IDummyStream* This, STATSTG* pstatstg,
169 DWORD grfStatFlag)
171 ++This->statcalls;
172 if (This->failstatcall)
173 return E_FAIL;
174 if (pstatstg)
175 pstatstg->cbSize.QuadPart = This->pos.QuadPart;
176 return S_OK;
179 /* VTable */
180 static void* iclvt[] =
182 QueryInterface,
183 AddRef,
184 Release,
185 Read,
186 Write,
187 Seek,
188 NULL, /* SetSize */
189 NULL, /* CopyTo */
190 NULL, /* Commit */
191 NULL, /* Revert */
192 NULL, /* LockRegion */
193 NULL, /* UnlockRegion */
194 Stat,
195 NULL /* Clone */
198 /* Function ptrs for ordinal calls */
199 static HMODULE SHLWAPI_hshlwapi = 0;
201 static VOID (WINAPI *pSHLWAPI_19)(LPSHLWAPI_CLIST);
202 static HRESULT (WINAPI *pSHLWAPI_20)(LPSHLWAPI_CLIST*,LPCSHLWAPI_CLIST);
203 static BOOL (WINAPI *pSHLWAPI_21)(LPSHLWAPI_CLIST*,ULONG);
204 static LPSHLWAPI_CLIST (WINAPI *pSHLWAPI_22)(LPSHLWAPI_CLIST,ULONG);
205 static HRESULT (WINAPI *pSHLWAPI_17)(_IDummyStream*,LPSHLWAPI_CLIST);
206 static HRESULT (WINAPI *pSHLWAPI_18)(_IDummyStream*,LPSHLWAPI_CLIST*);
208 static BOOL (WINAPI *pSHLWAPI_166)(_IDummyStream*);
209 static HRESULT (WINAPI *pSHLWAPI_184)(_IDummyStream*,LPVOID,ULONG);
210 static HRESULT (WINAPI *pSHLWAPI_212)(_IDummyStream*,LPCVOID,ULONG);
211 static HRESULT (WINAPI *pSHLWAPI_213)(_IDummyStream*);
212 static HRESULT (WINAPI *pSHLWAPI_214)(_IDummyStream*,ULARGE_INTEGER*);
215 static void InitFunctionPtrs()
217 SHLWAPI_hshlwapi = LoadLibraryA("shlwapi.dll");
218 ok(SHLWAPI_hshlwapi != 0, "LoadLibrary failed");
219 if (SHLWAPI_hshlwapi)
221 pSHLWAPI_17 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)17);
222 ok(pSHLWAPI_17 != 0, "No Ordinal 17");
223 pSHLWAPI_18 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)18);
224 ok(pSHLWAPI_18 != 0, "No Ordinal 18");
225 pSHLWAPI_19 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)19);
226 ok(pSHLWAPI_19 != 0, "No Ordinal 19");
227 pSHLWAPI_20 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)20);
228 ok(pSHLWAPI_20 != 0, "No Ordinal 20");
229 pSHLWAPI_21 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)21);
230 ok(pSHLWAPI_21 != 0, "No Ordinal 21");
231 pSHLWAPI_22 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)22);
232 ok(pSHLWAPI_22 != 0, "No Ordinal 22");
233 pSHLWAPI_166 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)166);
234 ok(pSHLWAPI_166 != 0, "No Ordinal 166");
235 pSHLWAPI_184 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)184);
236 ok(pSHLWAPI_184 != 0, "No Ordinal 184");
237 pSHLWAPI_212 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)212);
238 ok(pSHLWAPI_212 != 0, "No Ordinal 212");
239 pSHLWAPI_213 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)213);
240 ok(pSHLWAPI_213 != 0, "No Ordinal 213");
241 pSHLWAPI_214 = (void *)GetProcAddress( SHLWAPI_hshlwapi, (LPSTR)214);
242 ok(pSHLWAPI_214 != 0, "No Ordinal 214");
246 static void InitDummyStream(_IDummyStream* iface)
248 iface->lpVtbl = (void*)iclvt;
249 iface->ref = 1;
250 iface->readcalls = 0;
251 iface->failreadcall = FALSE;
252 iface->failreadsize = FALSE;
253 iface->readbeyondend = FALSE;
254 iface->readreturnlarge = FALSE;
255 iface->writecalls = 0;
256 iface->failwritecall = FALSE;
257 iface->failwritesize = FALSE;
258 iface->seekcalls = 0;
259 iface->statcalls = 0;
260 iface->failstatcall = FALSE;
261 iface->item = SHLWAPI_CLIST_items;
262 iface->pos.QuadPart = 0;
266 static void test_CList(void)
268 _IDummyStream streamobj;
269 LPSHLWAPI_CLIST list = NULL;
270 LPCSHLWAPI_CLIST item = SHLWAPI_CLIST_items;
271 HRESULT hRet;
272 LPSHLWAPI_CLIST inserted;
273 BYTE buff[64];
274 unsigned int i;
276 if (!pSHLWAPI_17 || !pSHLWAPI_18 || !pSHLWAPI_19 || !pSHLWAPI_20 ||
277 !pSHLWAPI_21 || !pSHLWAPI_22)
278 return;
280 /* Populate a list and test the items are added correctly */
281 while (item->ulSize)
283 /* Create item and fill with data */
284 inserted = (LPSHLWAPI_CLIST)buff;
285 inserted->ulSize = item->ulSize + sizeof(SHLWAPI_CLIST);
286 inserted->ulId = item->ulId;
287 for (i = 0; i < item->ulSize; i++)
288 buff[sizeof(SHLWAPI_CLIST)+i] = i*2;
290 /* Add it */
291 hRet = pSHLWAPI_20(&list, inserted);
292 ok(hRet > S_OK, "failed list add");
294 if (hRet > S_OK)
296 ok(list && list->ulSize, "item not added");
298 /* Find it */
299 inserted = pSHLWAPI_22(list, item->ulId);
300 ok(inserted != NULL, "lost after adding");
302 ok(!inserted || inserted->ulId != -1, "find returned a container");
304 /* Check size */
305 if (inserted && inserted->ulSize & 0x3)
307 /* Contained */
308 ok(inserted[-1].ulId == -1, "invalid size is not countained");
309 ok(inserted[-1].ulSize > inserted->ulSize+sizeof(SHLWAPI_CLIST),
310 "container too small");
312 else if (inserted)
314 ok(inserted->ulSize==item->ulSize+sizeof(SHLWAPI_CLIST),
315 "id %ld size wrong (%ld!=%ld)", inserted->ulId, inserted->ulSize,
316 item->ulSize+sizeof(SHLWAPI_CLIST));
318 if (inserted)
320 BOOL bDataOK = TRUE;
321 LPBYTE bufftest = (LPBYTE)inserted;
323 for (i = 0; i < inserted->ulSize - sizeof(SHLWAPI_CLIST); i++)
324 if (bufftest[sizeof(SHLWAPI_CLIST)+i] != i*2)
325 bDataOK = FALSE;
327 ok(bDataOK == TRUE, "data corrupted on insert");
329 ok(!inserted || inserted->ulId==item->ulId, "find got wrong item");
331 item++;
334 /* Write the list */
335 InitDummyStream(&streamobj);
337 hRet = pSHLWAPI_17(&streamobj, list);
338 ok(hRet == S_OK, "write failed");
339 if (hRet == S_OK)
341 /* 1 call for each element, + 1 for OK (use our null element for this) */
342 ok(streamobj.writecalls == sizeof(SHLWAPI_CLIST_items)/sizeof(SHLWAPI_CLIST),
343 "wrong call count");
344 ok(streamobj.readcalls == 0,"called Read() in write");
345 ok(streamobj.seekcalls == 0,"called Seek() in write");
348 /* Failure cases for writing */
349 InitDummyStream(&streamobj);
350 streamobj.failwritecall = TRUE;
351 hRet = pSHLWAPI_17(&streamobj, list);
352 ok(hRet == STG_E_ACCESSDENIED, "changed object failure return");
353 ok(streamobj.writecalls == 1, "called object after failure");
354 ok(streamobj.readcalls == 0,"called Read() after failure");
355 ok(streamobj.seekcalls == 0,"called Seek() after failure");
357 InitDummyStream(&streamobj);
358 streamobj.failwritesize = TRUE;
359 hRet = pSHLWAPI_17(&streamobj, list);
360 ok(hRet == STG_E_MEDIUMFULL, "changed size failure return");
361 ok(streamobj.writecalls == 1, "called object after size failure");
362 ok(streamobj.readcalls == 0,"called Read() after failure");
363 ok(streamobj.seekcalls == 0,"called Seek() after failure");
365 /* Invalid inputs for adding */
366 inserted = (LPSHLWAPI_CLIST)buff;
367 inserted->ulSize = sizeof(SHLWAPI_CLIST) -1;
368 inserted->ulId = 33;
369 hRet = pSHLWAPI_20(&list, inserted);
370 /* The call succeeds but the item is not inserted */
371 ok(hRet == S_OK, "failed bad element size");
372 inserted = pSHLWAPI_22(list, 33);
373 ok(inserted == NULL, "inserted bad element size");
375 inserted = (LPSHLWAPI_CLIST)buff;
376 inserted->ulSize = 44;
377 inserted->ulId = -1;
378 hRet = pSHLWAPI_20(&list, inserted);
379 /* The call succeeds but the item is not inserted */
380 ok(hRet == S_OK, "failed adding a container");
382 item = SHLWAPI_CLIST_items;
384 /* Look for non-existing item in populated list */
385 inserted = pSHLWAPI_22(list, 99999999);
386 ok(inserted == NULL, "found a non-existing item");
388 while (item->ulSize)
390 /* Delete items */
391 BOOL bRet = pSHLWAPI_21(&list, item->ulId);
392 ok(bRet == TRUE, "couldn't find item to delete");
393 item++;
396 /* Look for non-existing item in empty list */
397 inserted = pSHLWAPI_22(list, 99999999);
398 ok(inserted == NULL, "found an item in empty list");
400 /* Create a list by reading in data */
401 InitDummyStream(&streamobj);
403 hRet = pSHLWAPI_18(&streamobj, &list);
404 ok(hRet == S_OK, "failed create from Read()");
405 if (hRet == S_OK)
407 ok(streamobj.readbeyondend == FALSE, "read beyond end");
408 /* 2 calls per item, but only 1 for the terminator */
409 ok(streamobj.readcalls == sizeof(SHLWAPI_CLIST_items)/sizeof(SHLWAPI_CLIST)*2-1,
410 "wrong call count");
411 ok(streamobj.writecalls == 0, "called Write() from create");
412 ok(streamobj.seekcalls == 0,"called Seek() from create");
414 item = SHLWAPI_CLIST_items;
416 /* Check the items were added correctly */
417 while (item->ulSize)
419 inserted = pSHLWAPI_22(list, item->ulId);
420 ok(inserted != NULL, "lost after adding");
422 ok(!inserted || inserted->ulId != -1, "find returned a container");
424 /* Check size */
425 if (inserted && inserted->ulSize & 0x3)
427 /* Contained */
428 ok(inserted[-1].ulId == -1, "invalid size is not countained");
429 ok(inserted[-1].ulSize > inserted->ulSize+sizeof(SHLWAPI_CLIST),
430 "container too small");
432 else if (inserted)
434 ok(inserted->ulSize==item->ulSize+sizeof(SHLWAPI_CLIST),
435 "id %ld size wrong (%ld!=%ld)", inserted->ulId, inserted->ulSize,
436 item->ulSize+sizeof(SHLWAPI_CLIST));
438 ok(!inserted || inserted->ulId==item->ulId, "find got wrong item");
439 if (inserted)
441 BOOL bDataOK = TRUE;
442 LPBYTE bufftest = (LPBYTE)inserted;
444 for (i = 0; i < inserted->ulSize - sizeof(SHLWAPI_CLIST); i++)
445 if (bufftest[sizeof(SHLWAPI_CLIST)+i] != i*2)
446 bDataOK = FALSE;
448 ok(bDataOK == TRUE, "data corrupted on insert");
450 item++;
454 /* Failure cases for reading */
455 InitDummyStream(&streamobj);
456 streamobj.failreadcall = TRUE;
457 hRet = pSHLWAPI_18(&streamobj, &list);
458 ok(hRet == STG_E_ACCESSDENIED, "changed object failure return");
459 ok(streamobj.readbeyondend == FALSE, "read beyond end");
460 ok(streamobj.readcalls == 1, "called object after read failure");
461 ok(streamobj.writecalls == 0,"called Write() after read failure");
462 ok(streamobj.seekcalls == 0,"called Seek() after read failure");
464 /* Read returns large object */
465 InitDummyStream(&streamobj);
466 streamobj.readreturnlarge = TRUE;
467 hRet = pSHLWAPI_18(&streamobj, &list);
468 ok(hRet == S_OK, "failed create from Read() with large item");
469 ok(streamobj.readbeyondend == FALSE, "read beyond end");
470 ok(streamobj.readcalls == 1,"wrong call count");
471 ok(streamobj.writecalls == 0,"called Write() after read failure");
472 ok(streamobj.seekcalls == 2,"wrong Seek() call count (%d)", streamobj.seekcalls);
474 pSHLWAPI_19(list);
477 static void test_SHLWAPI_166(void)
479 _IDummyStream streamobj;
480 BOOL bRet;
482 if (!pSHLWAPI_166)
483 return;
485 InitDummyStream(&streamobj);
486 bRet = pSHLWAPI_166(&streamobj);
488 ok(bRet == TRUE, "failed before seek adjusted");
489 ok(streamobj.readcalls == 0, "called Read()");
490 ok(streamobj.writecalls == 0, "called Write()");
491 ok(streamobj.seekcalls == 0, "called Seek()");
492 ok(streamobj.statcalls == 1, "wrong call count");
494 streamobj.statcalls = 0;
495 streamobj.pos.QuadPart = 50001;
497 bRet = pSHLWAPI_166(&streamobj);
499 ok(bRet == FALSE, "failed after seek adjusted");
500 ok(streamobj.readcalls == 0, "called Read()");
501 ok(streamobj.writecalls == 0, "called Write()");
502 ok(streamobj.seekcalls == 0, "called Seek()");
503 ok(streamobj.statcalls == 1, "wrong call count");
505 /* Failure cases */
506 InitDummyStream(&streamobj);
507 streamobj.pos.QuadPart = 50001;
508 streamobj.failstatcall = TRUE; /* 1: Stat() Bad, Read() OK */
509 bRet = pSHLWAPI_166(&streamobj);
510 ok(bRet == FALSE, "should be FALSE after read is OK");
511 ok(streamobj.readcalls == 1, "wrong call count");
512 ok(streamobj.writecalls == 0, "called Write()");
513 ok(streamobj.seekcalls == 1, "wrong call count");
514 ok(streamobj.statcalls == 1, "wrong call count");
515 ok(streamobj.pos.QuadPart == 0, "Didn't seek to start");
517 InitDummyStream(&streamobj);
518 streamobj.pos.QuadPart = 50001;
519 streamobj.failstatcall = TRUE;
520 streamobj.failreadcall = TRUE; /* 2: Stat() Bad, Read() Bad Also */
521 bRet = pSHLWAPI_166(&streamobj);
522 ok(bRet == TRUE, "Should be true after read fails");
523 ok(streamobj.readcalls == 1, "wrong call count");
524 ok(streamobj.writecalls == 0, "called Write()");
525 ok(streamobj.seekcalls == 0, "Called Seek()");
526 ok(streamobj.statcalls == 1, "wrong call count");
527 ok(streamobj.pos.QuadPart == 50001, "called Seek() after read failed");
530 static void test_SHLWAPI_184(void)
532 _IDummyStream streamobj;
533 char buff[256];
534 HRESULT hRet;
536 if (!pSHLWAPI_184)
537 return;
539 InitDummyStream(&streamobj);
540 hRet = pSHLWAPI_184(&streamobj, buff, sizeof(buff));
542 ok(hRet == S_OK, "failed Read()");
543 ok(streamobj.readcalls == 1, "wrong call count");
544 ok(streamobj.writecalls == 0, "called Write()");
545 ok(streamobj.seekcalls == 0, "called Seek()");
548 static void test_SHLWAPI_212(void)
550 _IDummyStream streamobj;
551 char buff[256];
552 HRESULT hRet;
554 if (!pSHLWAPI_212)
555 return;
557 InitDummyStream(&streamobj);
558 hRet = pSHLWAPI_212(&streamobj, buff, sizeof(buff));
560 ok(hRet == S_OK, "failed Write()");
561 ok(streamobj.readcalls == 0, "called Read()");
562 ok(streamobj.writecalls == 1, "wrong call count");
563 ok(streamobj.seekcalls == 0, "called Seek()");
566 static void test_SHLWAPI_213(void)
568 _IDummyStream streamobj;
569 ULARGE_INTEGER ul;
570 LARGE_INTEGER ll;
571 HRESULT hRet;
573 if (!pSHLWAPI_213 || !pSHLWAPI_214)
574 return;
576 InitDummyStream(&streamobj);
577 ll.QuadPart = 5000l;
578 Seek(&streamobj, ll, 0, NULL); /* Seek to 5000l */
580 streamobj.seekcalls = 0;
581 pSHLWAPI_213(&streamobj); /* Should rewind */
582 ok(streamobj.statcalls == 0, "called Stat()");
583 ok(streamobj.readcalls == 0, "called Read()");
584 ok(streamobj.writecalls == 0, "called Write()");
585 ok(streamobj.seekcalls == 1, "wrong call count");
587 ul.QuadPart = 50001;
588 hRet = pSHLWAPI_214(&streamobj, &ul);
589 ok(hRet == S_OK, "failed Stat()");
590 ok(ul.QuadPart == 0, "213 didn't rewind stream");
593 static void test_SHLWAPI_214(void)
595 _IDummyStream streamobj;
596 ULARGE_INTEGER ul;
597 LARGE_INTEGER ll;
598 HRESULT hRet;
600 if (!pSHLWAPI_214)
601 return;
603 InitDummyStream(&streamobj);
604 ll.QuadPart = 5000l;
605 Seek(&streamobj, ll, 0, NULL);
606 ul.QuadPart = 0;
607 streamobj.seekcalls = 0;
608 hRet = pSHLWAPI_214(&streamobj, &ul);
610 ok(hRet == S_OK, "failed Stat()");
611 ok(streamobj.statcalls == 1, "wrong call count");
612 ok(streamobj.readcalls == 0, "called Read()");
613 ok(streamobj.writecalls == 0, "called Write()");
614 ok(streamobj.seekcalls == 0, "called Seek()");
615 ok(ul.QuadPart == 5000l, "Stat gave wrong size");
618 START_TEST(clist)
620 InitFunctionPtrs();
622 test_CList();
624 test_SHLWAPI_166();
625 test_SHLWAPI_184();
626 test_SHLWAPI_212();
627 test_SHLWAPI_213();
628 test_SHLWAPI_214();
630 if (SHLWAPI_hshlwapi)
631 FreeLibrary(SHLWAPI_hshlwapi);