Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522/* * Unit tests for SetupIterateCabinet * * Copyright 2007 Hans Leidekker * Copyright 2010 Andrew Nguyen * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */
#include <stdarg.h>
#include "windef.h"#include "winbase.h"#include "wingdi.h"#include "winuser.h"#include "winreg.h"#include "setupapi.h"#include "wine/test.h"
static const BYTE comp_cab_zip_multi[] = { 0x4d, 0x53, 0x43, 0x46, 0x00, 0x00, 0x00, 0x00, 0x9c, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x2c, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0x01, 0x01, 0x00, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x71, 0x00, 0x00, 0x00, 0x01, 0x00, 0x01, 0x00, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xd1, 0x38, 0xf0, 0x48, 0x20, 0x00, 0x74, 0x72, 0x69, 0x73, 0x74, 0x72, 0x61, 0x6d, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x00, 0xd1, 0x38, 0xf0, 0x48, 0x20, 0x00, 0x77, 0x69, 0x6e, 0x65, 0x00, 0x08, 0x00, 0x00, 0x00, 0x18, 0x00, 0x00, 0x00, 0x00, 0x00, 0xd1, 0x38, 0xf0, 0x48, 0x20, 0x00, 0x73, 0x68, 0x61, 0x6e, 0x64, 0x79, 0x00, 0x67, 0x2c, 0x03, 0x85, 0x23, 0x00, 0x20, 0x00, 0x43, 0x4b, 0xcb, 0x49, 0x2c, 0x2d, 0x4a, 0xcd, 0x4b, 0x4e, 0xe5, 0xe5, 0x2a, 0xcd, 0x4b, 0xce, 0xcf, 0x2d, 0x28, 0x4a, 0x2d, 0x2e, 0x4e, 0x4d, 0xe1, 0xe5, 0x2a, 0x2e, 0x49, 0x2d, 0xca, 0x03, 0x8a, 0x02, 0x00};
static const WCHAR docW[] = {'d','o','c',0};
static void create_source_fileA(LPSTR filename, const BYTE *data, DWORD size){ HANDLE handle; DWORD written;
handle = CreateFileA(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); WriteFile(handle, data, size, &written, NULL); CloseHandle(handle);}
static void create_source_fileW(LPWSTR filename, const BYTE *data, DWORD size){ HANDLE handle; DWORD written;
handle = CreateFileW(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); WriteFile(handle, data, size, &written, NULL); CloseHandle(handle);}
static UINT CALLBACK dummy_callbackA(PVOID Context, UINT Notification, UINT_PTR Param1, UINT_PTR Param2){ ok(0, "Received unexpected notification (%p, %u, %lu, %lu)\n", Context, Notification, Param1, Param2); return 0;}
static UINT CALLBACK dummy_callbackW(PVOID Context, UINT Notification, UINT_PTR Param1, UINT_PTR Param2){ ok(0, "Received unexpected notification (%p, %u, %lu, %lu)\n", Context, Notification, Param1, Param2); return 0;}
static void test_invalid_parametersA(void){ BOOL ret; char source[MAX_PATH], temp[MAX_PATH]; int i;
const struct { PCSTR CabinetFile; PSP_FILE_CALLBACK_A MsgHandler; DWORD expected_lasterror; int todo_lasterror; } invalid_parameters[] = { {NULL, NULL, ERROR_INVALID_PARAMETER}, {NULL, dummy_callbackA, ERROR_INVALID_PARAMETER}, {"c:\\nonexistent.cab", NULL, ERROR_FILE_NOT_FOUND}, {"c:\\nonexistent.cab", dummy_callbackA, ERROR_FILE_NOT_FOUND}, {source, NULL, ERROR_INVALID_DATA, 1}, {source, dummy_callbackA, ERROR_INVALID_DATA, 1}, };
GetTempPathA(sizeof(temp), temp); GetTempFileNameA(temp, "doc", 0, source);
create_source_fileA(source, NULL, 0);
for (i = 0; i < ARRAY_SIZE(invalid_parameters); i++) { SetLastError(0xdeadbeef); ret = SetupIterateCabinetA(invalid_parameters[i].CabinetFile, 0, invalid_parameters[i].MsgHandler, NULL); ok(!ret, "[%d] Expected SetupIterateCabinetA to return 0, got %d\n", i, ret); todo_wine_if (invalid_parameters[i].todo_lasterror) ok(GetLastError() == invalid_parameters[i].expected_lasterror, "[%d] Expected GetLastError() to return %u, got %u\n", i, invalid_parameters[i].expected_lasterror, GetLastError()); }
SetLastError(0xdeadbeef); ret = SetupIterateCabinetA("", 0, NULL, NULL); ok(!ret, "Expected SetupIterateCabinetA to return 0, got %d\n", ret); ok(GetLastError() == ERROR_NOT_ENOUGH_MEMORY || GetLastError() == ERROR_FILE_NOT_FOUND, /* Win9x/NT4/Win2k */ "Expected GetLastError() to return ERROR_NOT_ENOUGH_MEMORY, got %u\n", GetLastError());
SetLastError(0xdeadbeef); ret = SetupIterateCabinetA("", 0, dummy_callbackA, NULL); ok(!ret, "Expected SetupIterateCabinetA to return 0, got %d\n", ret); ok(GetLastError() == ERROR_NOT_ENOUGH_MEMORY || GetLastError() == ERROR_FILE_NOT_FOUND, /* Win9x/NT4/Win2k */ "Expected GetLastError() to return ERROR_NOT_ENOUGH_MEMORY, got %u\n", GetLastError());
DeleteFileA(source);}
static void test_invalid_parametersW(void){ static const WCHAR nonexistentW[] = {'c',':','\\','n','o','n','e','x','i','s','t','e','n','t','.','c','a','b',0}; static const WCHAR emptyW[] = {0};
BOOL ret; WCHAR source[MAX_PATH], temp[MAX_PATH]; int i;
const struct { PCWSTR CabinetFile; PSP_FILE_CALLBACK_W MsgHandler; DWORD expected_lasterror; int todo_lasterror; } invalid_parameters[] = { {nonexistentW, NULL, ERROR_FILE_NOT_FOUND}, {nonexistentW, dummy_callbackW, ERROR_FILE_NOT_FOUND}, {source, NULL, ERROR_INVALID_DATA, 1}, {source, dummy_callbackW, ERROR_INVALID_DATA, 1}, };
ret = SetupIterateCabinetW(NULL, 0, NULL, NULL); if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) { win_skip("SetupIterateCabinetW is not available\n"); return; }
GetTempPathW(ARRAY_SIZE(temp), temp); GetTempFileNameW(temp, docW, 0, source);
create_source_fileW(source, NULL, 0);
for (i = 0; i < ARRAY_SIZE(invalid_parameters); i++) { SetLastError(0xdeadbeef); ret = SetupIterateCabinetW(invalid_parameters[i].CabinetFile, 0, invalid_parameters[i].MsgHandler, NULL); ok(!ret, "[%d] Expected SetupIterateCabinetW to return 0, got %d\n", i, ret); todo_wine_if (invalid_parameters[i].todo_lasterror) ok(GetLastError() == invalid_parameters[i].expected_lasterror, "[%d] Expected GetLastError() to return %u, got %u\n", i, invalid_parameters[i].expected_lasterror, GetLastError()); }
SetLastError(0xdeadbeef); ret = SetupIterateCabinetW(NULL, 0, NULL, NULL); ok(!ret, "Expected SetupIterateCabinetW to return 0, got %d\n", ret); ok(GetLastError() == ERROR_INVALID_PARAMETER || GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Vista/Win2k8 */ "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
SetLastError(0xdeadbeef); ret = SetupIterateCabinetW(NULL, 0, dummy_callbackW, NULL); ok(!ret, "Expected SetupIterateCabinetW to return 0, got %d\n", ret); ok(GetLastError() == ERROR_INVALID_PARAMETER || GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Vista/Win2k8 */ "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
SetLastError(0xdeadbeef); ret = SetupIterateCabinetW(emptyW, 0, NULL, NULL); ok(!ret, "Expected SetupIterateCabinetW to return 0, got %d\n", ret); ok(GetLastError() == ERROR_NOT_ENOUGH_MEMORY || GetLastError() == ERROR_FILE_NOT_FOUND, /* NT4/Win2k */ "Expected GetLastError() to return ERROR_NOT_ENOUGH_MEMORY, got %u\n", GetLastError());
SetLastError(0xdeadbeef); ret = SetupIterateCabinetW(emptyW, 0, dummy_callbackW, NULL); ok(!ret, "Expected SetupIterateCabinetW to return 0, got %d\n", ret); ok(GetLastError() == ERROR_NOT_ENOUGH_MEMORY || GetLastError() == ERROR_FILE_NOT_FOUND, /* NT4/Win2k */ "Expected GetLastError() to return ERROR_NOT_ENOUGH_MEMORY, got %u\n", GetLastError());
DeleteFileW(source);}
static UINT CALLBACK crash_callbackA(PVOID Context, UINT Notification, UINT_PTR Param1, UINT_PTR Param2){ *(volatile char*)0 = 2; return 0;}
static UINT CALLBACK crash_callbackW(PVOID Context, UINT Notification, UINT_PTR Param1, UINT_PTR Param2){ *(volatile char*)0 = 2; return 0;}
static void test_invalid_callbackA(void){ BOOL ret; char source[MAX_PATH], temp[MAX_PATH];
GetTempPathA(sizeof(temp), temp); GetTempFileNameA(temp, "doc", 0, source);
create_source_fileA(source, comp_cab_zip_multi, sizeof(comp_cab_zip_multi));
SetLastError(0xdeadbeef); ret = SetupIterateCabinetA(source, 0, NULL, NULL); ok(!ret, "Expected SetupIterateCabinetA to return 0, got %d\n", ret); ok(GetLastError() == ERROR_INVALID_DATA, "Expected GetLastError() to return ERROR_INVALID_DATA, got %u\n", GetLastError());
SetLastError(0xdeadbeef); ret = SetupIterateCabinetA(source, 0, crash_callbackA, NULL); ok(!ret, "Expected SetupIterateCabinetA to return 0, got %d\n", ret); ok(GetLastError() == ERROR_INVALID_DATA, "Expected GetLastError() to return ERROR_INVALID_DATA, got %u\n", GetLastError());
DeleteFileA(source);}
static void test_invalid_callbackW(void){ BOOL ret; WCHAR source[MAX_PATH], temp[MAX_PATH];
ret = SetupIterateCabinetW(NULL, 0, NULL, NULL); if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) { win_skip("SetupIterateCabinetW is not available\n"); return; }
GetTempPathW(ARRAY_SIZE(temp), temp); GetTempFileNameW(temp, docW, 0, source);
create_source_fileW(source, comp_cab_zip_multi, sizeof(comp_cab_zip_multi));
SetLastError(0xdeadbeef); ret = SetupIterateCabinetW(source, 0, NULL, NULL); ok(!ret, "Expected SetupIterateCabinetW to return 0, got %d\n", ret); ok(GetLastError() == ERROR_INVALID_DATA, "Expected GetLastError() to return ERROR_INVALID_DATA, got %u\n", GetLastError());
SetLastError(0xdeadbeef); ret = SetupIterateCabinetW(source, 0, crash_callbackW, NULL); ok(!ret, "Expected SetupIterateCabinetW to return 0, got %d\n", ret); ok(GetLastError() == ERROR_INVALID_DATA, "Expected GetLastError() to return ERROR_INVALID_DATA, got %u\n", GetLastError());
DeleteFileW(source);}
static const char *expected_files[] = {"tristram", "wine", "shandy"};
struct contextA{ int count; const char *cabinet; const char *target;};
static UINT CALLBACK simple_callbackA(PVOID Context, UINT Notification, UINT_PTR Param1, UINT_PTR Param2){ static int index; struct contextA *ctx = Context;
switch (Notification) { case SPFILENOTIFY_CABINETINFO: { CABINET_INFO_A *info = (CABINET_INFO_A *)Param1;
ok(!strcmp(info->CabinetFile, ""), "Expected empty CabinetFile, got \"%s\"\n", info->CabinetFile);
index = 0; return NO_ERROR; } case SPFILENOTIFY_FILEINCABINET: { FILE_IN_CABINET_INFO_A *info = (FILE_IN_CABINET_INFO_A *)Param1; const char *cabinet_file = (const char *)Param2;
ctx->count++;
if (index < ARRAY_SIZE(expected_files)) { ok(!strcmp(expected_files[index], info->NameInCabinet), "[%d] Expected file \"%s\", got \"%s\"\n", index, expected_files[index], info->NameInCabinet); ok(!strcmp(ctx->cabinet, cabinet_file), "[%d] Expected cabinet \"%s\", got \"%s\"\n", index, ctx->cabinet, cabinet_file);
strcpy(info->FullTargetName, ctx->target); return FILEOP_DOIT; } else { ok(0, "Unexpectedly enumerated more than number of files in cabinet, index = %d\n", index); return FILEOP_ABORT; } } case SPFILENOTIFY_FILEEXTRACTED: { FILEPATHS_A *info = (FILEPATHS_A *)Param1;
ok(!strcmp(ctx->cabinet, info->Source), "[%d] Expected cabinet \"%s\", got \"%s\"\n", index, ctx->cabinet, info->Source); ok(!strcmp(ctx->target, info->Target), "[%d] Expected target \"%s\", got \"%s\"\n", index, ctx->target, info->Target); ok(info->Win32Error == 0, "[%d] Expected Win32Error 0, got %u\n", index, info->Win32Error);
index++; return NO_ERROR; } default: return NO_ERROR; }}
static void test_simple_enumerationA(void){ BOOL ret; char source[MAX_PATH], temp[MAX_PATH]; char target[MAX_PATH]; struct contextA ctx;
GetTempPathA(sizeof(temp), temp); GetTempFileNameA(temp, "doc", 0, source); GetTempFileNameA(temp, "doc", 0, target);
create_source_fileA(source, comp_cab_zip_multi, sizeof(comp_cab_zip_multi));
ctx.count = 0; ctx.cabinet = source; ctx.target = target; ret = SetupIterateCabinetA(source, 0, simple_callbackA, &ctx); ok(ret == 1, "Expected SetupIterateCabinetA to return 1, got %d\n", ret); ok(ctx.count == ARRAY_SIZE(expected_files), "Unexpectedly enumerated %d files\n", ctx.count);
DeleteFileA(source); DeleteFileA(target);}
static const WCHAR tristramW[] = {'t','r','i','s','t','r','a','m',0};static const WCHAR wineW[] = {'w','i','n','e',0};static const WCHAR shandyW[] = {'s','h','a','n','d','y',0};static const WCHAR *expected_filesW[] = {tristramW, wineW, shandyW};
struct contextW{ int count; const WCHAR *cabinet; const WCHAR *target;};
static UINT CALLBACK simple_callbackW(PVOID Context, UINT Notification, UINT_PTR Param1, UINT_PTR Param2){ static const WCHAR emptyW[] = {0}; static int index; struct contextW *ctx = Context;
switch (Notification) { case SPFILENOTIFY_CABINETINFO: { CABINET_INFO_W *info = (CABINET_INFO_W *)Param1;
ok(!lstrcmpW(info->CabinetFile, emptyW), "Expected empty CabinetFile, got %s\n", wine_dbgstr_w(info->CabinetFile));
index = 0; return NO_ERROR; } case SPFILENOTIFY_FILEINCABINET: { FILE_IN_CABINET_INFO_W *info = (FILE_IN_CABINET_INFO_W *)Param1; const WCHAR *cabinet_file = (const WCHAR *)Param2;
ctx->count++;
if (index < ARRAY_SIZE(expected_filesW)) { ok(!lstrcmpW(expected_filesW[index], info->NameInCabinet), "[%d] Expected file %s, got %s\n", index, wine_dbgstr_w(expected_filesW[index]), wine_dbgstr_w(info->NameInCabinet)); ok(!lstrcmpW(ctx->cabinet, cabinet_file), "[%d] Expected cabinet %s, got %s\n", index, wine_dbgstr_w(ctx->cabinet), wine_dbgstr_w(cabinet_file));
lstrcpyW(info->FullTargetName, ctx->target); return FILEOP_DOIT; } else { ok(0, "Unexpectedly enumerated more than number of files in cabinet, index = %d\n", index); return FILEOP_ABORT; } } case SPFILENOTIFY_FILEEXTRACTED: { FILEPATHS_W *info = (FILEPATHS_W *)Param1;
ok(!lstrcmpW(ctx->cabinet, info->Source), "[%d] Expected cabinet %s, got %s\n", index, wine_dbgstr_w(ctx->cabinet), wine_dbgstr_w(info->Source)); ok(!lstrcmpW(ctx->target, info->Target), "[%d] Expected target %s, got %s\n", index, wine_dbgstr_w(ctx->target), wine_dbgstr_w(info->Target)); ok(info->Win32Error == 0, "[%d] Expected Win32Error 0, got %u\n", index, info->Win32Error);
index++; return NO_ERROR; } default: return NO_ERROR; }}
static void test_simple_enumerationW(void){ BOOL ret; WCHAR source[MAX_PATH], temp[MAX_PATH]; WCHAR target[MAX_PATH]; struct contextW ctx;
ret = SetupIterateCabinetW(NULL, 0, NULL, NULL); if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) { win_skip("SetupIterateCabinetW is not available\n"); return; }
GetTempPathW(ARRAY_SIZE(temp), temp); GetTempFileNameW(temp, docW, 0, source); GetTempFileNameW(temp, docW, 0, target);
create_source_fileW(source, comp_cab_zip_multi, sizeof(comp_cab_zip_multi));
ctx.count = 0; ctx.cabinet = source; ctx.target = target; ret = SetupIterateCabinetW(source, 0, simple_callbackW, &ctx); ok(ret == 1, "Expected SetupIterateCabinetW to return 1, got %d\n", ret); ok(ctx.count == ARRAY_SIZE(expected_files), "Unexpectedly enumerated %d files\n", ctx.count);
DeleteFileW(source); DeleteFileW(target);}
START_TEST(setupcab){ test_invalid_parametersA(); test_invalid_parametersW();
/* Tests crash on NT4/Win9x/Win2k and Wine. */ if (0) { test_invalid_callbackA(); test_invalid_callbackW(); }
test_simple_enumerationA(); test_simple_enumerationW();}