diskspace.c File Reference

#include <stdarg.h>
#include <stdio.h>
#include "windef.h"
#include "winbase.h"
#include "winnls.h"
#include "winuser.h"
#include "winreg.h"
#include "setupapi.h"
#include "wine/test.h"

Include dependency graph for diskspace.c:

Go to the source code of this file.

Classes
struct	device_usage
struct	section
struct	section_i

Macros
#define	STD_HEADER   "[Version]\r\nSignature=\"$CHICAGO$\"\r\n"

Functions
static const char *	debugstr_longlong (ULONGLONG ll)
static HINF	inf_open_file_content (const char *tmpfilename, const char *data, UINT *err_line)
static void	test_SetupCreateDiskSpaceListA (void)
static void	test_SetupCreateDiskSpaceListW (void)
static void	test_SetupDuplicateDiskSpaceListA (void)
static void	test_SetupDuplicateDiskSpaceListW (void)
static LONGLONG	get_file_size (char *path)
static void	test_SetupQuerySpaceRequiredOnDriveA (void)
static void	test_SetupQuerySpaceRequiredOnDriveW (void)
static void	test_SetupAddToDiskSpaceListA (void)
static void	test_SetupQueryDrivesInDiskSpaceListA (void)
static void	test_SetupAddSectionToDiskSpaceListA (void)
static void	test_SetupAddInstallSectionToDiskSpaceListA (void)
	START_TEST (diskspace)

Variables
static BOOL	is_win9x
struct {
const char *   data
struct section   sections [2]
const char *   devices
int   device_length
struct device_usage   usage [2]
}	section_test []
struct {
const char *   data
struct section_i   sections [2]
const char *   devices
int   device_length
struct device_usage   usage [2]
}	section_test_i []

Macro Definition Documentation

STD_HEADER

#define STD_HEADER   "[Version]\r\nSignature=\"$CHICAGO$\"\r\n"

Definition at line 35 of file diskspace.c.

Function Documentation

debugstr_longlong()

static const char * debugstr_longlong ( ULONGLONG  ll )

inlinestatic

Definition at line 37 of file diskspace.c.

{
    static char string[17];
    if (sizeof(ll) > sizeof(unsigned long) && ll >> 32)
        sprintf(string, "%lx%08lx", (unsigned long)(ll >> 32), (unsigned long)ll);
    else
        sprintf(string, "%lx", (unsigned long)ll);
    return string;
}

Referenced by test_SetupQuerySpaceRequiredOnDriveA().

get_file_size()

static LONGLONG get_file_size ( char *  path )

static

Definition at line 328 of file diskspace.c.

{
    HANDLE file;
    LARGE_INTEGER size;
 
    file = CreateFileA(path, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE,
                       NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    if (file == INVALID_HANDLE_VALUE) return 0;
 
    if (!GetFileSizeEx(file, &size))
        size.QuadPart = 0;
 
    CloseHandle(file);
    return size.QuadPart;
}

inf_open_file_content()

static HINF inf_open_file_content ( const char *  tmpfilename, const char *  data, UINT *  err_line  )

static

Definition at line 48 of file diskspace.c.

{
    DWORD res;
    HANDLE handle = CreateFileA(tmpfilename, GENERIC_READ|GENERIC_WRITE,
                                FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, 0, 0);
    if (handle == INVALID_HANDLE_VALUE) return 0;
    if (!WriteFile( handle, data, strlen(data), &res, NULL )) trace( "write error\n" );
    CloseHandle( handle );
    return SetupOpenInfFileA( tmpfilename, 0, INF_STYLE_WIN4, err_line );
}

Referenced by test_SetupAddInstallSectionToDiskSpaceListA(), and test_SetupAddSectionToDiskSpaceListA().

START_TEST()

START_TEST

(

diskspace

)

Definition at line 1119 of file diskspace.c.

{
    is_win9x = !SetupCreateDiskSpaceListW((void *)0xdeadbeef, 0xdeadbeef, 0) &&
                GetLastError() == ERROR_CALL_NOT_IMPLEMENTED;
    test_SetupCreateDiskSpaceListA();
    test_SetupCreateDiskSpaceListW();
    test_SetupDuplicateDiskSpaceListA();
    test_SetupDuplicateDiskSpaceListW();
    test_SetupQuerySpaceRequiredOnDriveA();
    test_SetupQuerySpaceRequiredOnDriveW();
    test_SetupAddToDiskSpaceListA();
    test_SetupQueryDrivesInDiskSpaceListA();
    test_SetupAddSectionToDiskSpaceListA();
    test_SetupAddInstallSectionToDiskSpaceListA();
}

test_SetupAddInstallSectionToDiskSpaceListA()

static void test_SetupAddInstallSectionToDiskSpaceListA ( void  )

static

Definition at line 1012 of file diskspace.c.

{
    char tmp[MAX_PATH];
    char tmpfilename[MAX_PATH];
    char buffer[MAX_PATH];
    HDSKSPC diskspace;
    LONGLONG space;
    UINT err_line;
    BOOL ret;
    int i, j;
    HINF inf;
 
    if (!GetTempPathA(MAX_PATH, tmp))
    {
        win_skip("GetTempPath failed with error %d\n", GetLastError());
        return;
    }
 
    if (!GetTempFileNameA(tmp, "inftest", 0, tmpfilename))
    {
        win_skip("GetTempFileNameA failed with error %d\n", GetLastError());
        return;
    }
 
    inf = inf_open_file_content(tmpfilename, STD_HEADER "[a]\nCopyFiles=b\n[b]\ntest,,,\n[SourceDisksFiles]\ntest=1,,4096\r\n", &err_line);
    ok(!!inf, "Failed to open inf file (%d, line %d)\n", GetLastError(), err_line);
 
    diskspace = SetupCreateDiskSpaceListA(NULL, 0, SPDSL_IGNORE_DISK);
    ok(diskspace != NULL,"Expected SetupCreateDiskSpaceListA to return a valid handle\n");
 
    ret = SetupAddInstallSectionToDiskSpaceListA(diskspace, NULL, NULL, "a", 0, 0);
    ok(ret, "Expected SetupAddInstallSectionToDiskSpaceListA to succeed\n");
 
    ret = SetupAddInstallSectionToDiskSpaceListA(NULL, inf, NULL, "a", 0, 0);
    ok(!ret, "Expected SetupAddInstallSectionToDiskSpaceListA to fail\n");
    ok(GetLastError() == ERROR_INVALID_HANDLE, "Expected ERROR_INVALID_HANDLE as error, got %u\n",
       GetLastError());
 
    ret = SetupAddInstallSectionToDiskSpaceListA(diskspace, inf, NULL, NULL, 0, 0);
    ok(!ret || broken(ret), "Expected SetupAddSectionToDiskSpaceListA to fail\n");
    ok(GetLastError() == ERROR_INVALID_PARAMETER || broken(ret),
       "Expected ERROR_INVALID_PARAMETER as error, got %u\n", GetLastError());
 
    ret = SetupAddInstallSectionToDiskSpaceListA(diskspace, inf, NULL, "", 0, 0);
    ok(ret, "Expected SetupAddInstallSectionToDiskSpaceListA to succeed (%u)\n", GetLastError());
 
    ok(SetupDestroyDiskSpaceList(diskspace),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    for (i = 0; i < sizeof(section_test_i) / sizeof(section_test_i[0]); i++)
    {
        err_line = 0;
 
        inf = inf_open_file_content(tmpfilename, section_test_i[i].data, &err_line);
        ok(!!inf, "test %d: Failed to open inf file (%d, line %d)\n", i, GetLastError(), err_line);
        if (!inf) continue;
 
        diskspace = SetupCreateDiskSpaceListA(NULL, 0, SPDSL_IGNORE_DISK);
        ok(diskspace != NULL,"Expected SetupCreateDiskSpaceListA to return a valid handle\n");
 
        for (j = 0; j < 2; j++)
        {
            const struct section_i *section = &section_test_i[i].sections[j];
            if (!section->name)
                continue;
 
            SetLastError(0xdeadbeef);
            ret = SetupAddInstallSectionToDiskSpaceListA(diskspace, inf, NULL, section->name, 0, 0);
            if (section->result)
                ok(ret, "test %d: Expected adding section %d to succeed (%u)\n", i, j, GetLastError());
            else
            {
                ok(!ret, "test %d: Expected adding section %d to fail\n", i, j);
                ok(GetLastError() == section->error_code, "test %d: Expected %u as error, got %u\n",
                   i, section->error_code, GetLastError());
            }
        }
 
        memset(buffer, 0x0, sizeof(buffer));
        ret = SetupQueryDrivesInDiskSpaceListA(diskspace, buffer, sizeof(buffer), NULL);
        ok(ret, "test %d: Expected SetupQueryDrivesInDiskSpaceListA to succeed (%u)\n", i, GetLastError());
        ok(!memcmp(section_test_i[i].devices, buffer, section_test_i[i].device_length),
           "test %d: Device list (%s) does not match\n", i, buffer);
 
        for (j = 0; j < 2; j++)
        {
            const struct device_usage *usage = &section_test_i[i].usage[j];
            if (!usage->dev)
                continue;
 
            space = 0;
            ret = SetupQuerySpaceRequiredOnDriveA(diskspace, usage->dev, &space, NULL, 0);
            ok(ret, "test %d: Expected SetupQuerySpaceRequiredOnDriveA to succeed for device %s (%u)\n",
               i, usage->dev, GetLastError());
            ok(space == usage->usage, "test %d: Expected size %u for device %s, got %u\n",
               i, (DWORD)usage->usage, usage->dev, (DWORD)space);
        }
 
        ok(SetupDestroyDiskSpaceList(diskspace),
           "Expected SetupDestroyDiskSpaceList to succeed\n");
 
        SetupCloseInfFile(inf);
    }
 
    DeleteFileA(tmpfilename);
}

Referenced by START_TEST().

test_SetupAddSectionToDiskSpaceListA()

static void test_SetupAddSectionToDiskSpaceListA ( void  )

static

Definition at line 841 of file diskspace.c.

{
    char tmp[MAX_PATH];
    char tmpfilename[MAX_PATH];
    char buffer[MAX_PATH];
    HDSKSPC diskspace;
    UINT err_line;
    LONGLONG space;
    BOOL ret;
    int i, j;
    HINF inf;
 
    if (!GetTempPathA(MAX_PATH, tmp))
    {
        win_skip("GetTempPath failed with error %d\n", GetLastError());
        return;
    }
 
    if (!GetTempFileNameA(tmp, "inftest", 0, tmpfilename))
    {
        win_skip("GetTempFileNameA failed with error %d\n", GetLastError());
        return;
    }
 
    inf = inf_open_file_content(tmpfilename, STD_HEADER "[a]\ntest,,,\n[SourceDisksFiles]\ntest=1,,4096\r\n", &err_line);
    ok(!!inf, "Failed to open inf file (%d, line %d)\n", GetLastError(), err_line);
 
    diskspace = SetupCreateDiskSpaceListA(NULL, 0, SPDSL_IGNORE_DISK);
    ok(diskspace != NULL,"Expected SetupCreateDiskSpaceListA to return a valid handle\n");
 
    ret = SetupAddSectionToDiskSpaceListA(diskspace, NULL, NULL, "a", FILEOP_COPY, 0, 0);
    ok(!ret, "Expected SetupAddSectionToDiskSpaceListA to fail\n");
    ok(GetLastError() == ERROR_SECTION_NOT_FOUND, "Expected ERROR_SECTION_NOT_FOUND as error, got %u\n",
       GetLastError());
 
    ret = SetupAddSectionToDiskSpaceListA(NULL, inf, NULL, "a", FILEOP_COPY, 0, 0);
    ok(!ret, "Expected SetupAddSectionToDiskSpaceListA to fail\n");
    ok(GetLastError() == ERROR_INVALID_HANDLE, "Expected ERROR_INVALID_HANDLE as error, got %u\n",
       GetLastError());
 
    ret = SetupAddSectionToDiskSpaceListA(NULL, inf, NULL, "b", FILEOP_COPY, 0, 0);
    ok(!ret, "Expected SetupAddSectionToDiskSpaceListA to fail\n");
    ok(GetLastError() == ERROR_INVALID_HANDLE, "Expected ERROR_INVALID_HANDLE as error, got %u\n",
       GetLastError());
 
    ret = SetupAddSectionToDiskSpaceListA(diskspace, inf, NULL, "a", 0, 0, 0);
    ok(ret, "Expected SetupAddSectionToDiskSpaceListA to succeed (%u)\n", GetLastError());
 
    ok(SetupDestroyDiskSpaceList(diskspace),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    for (i = 0; i < sizeof(section_test) / sizeof(section_test[0]); i++)
    {
        err_line = 0;
 
        inf = inf_open_file_content(tmpfilename, section_test[i].data, &err_line);
        ok(!!inf, "test %d: Failed to open inf file (%d, line %d)\n", i, GetLastError(), err_line);
        if (!inf) continue;
 
        diskspace = SetupCreateDiskSpaceListA(NULL, 0, SPDSL_IGNORE_DISK);
        ok(diskspace != NULL, "Expected SetupCreateDiskSpaceListA to return a valid handle\n");
 
        for (j = 0; j < 2; j++)
        {
            const struct section *section = &section_test[i].sections[j];
            if (!section->name)
                continue;
 
            SetLastError(0xdeadbeef);
            ret = SetupAddSectionToDiskSpaceListA(diskspace, inf, NULL, section->name, section->fileop, 0, 0);
            if (section->result)
                ok(ret, "test %d: Expected adding section %d to succeed (%u)\n", i, j, GetLastError());
            else
            {
                ok(!ret, "test %d: Expected adding section %d to fail\n", i, j);
                ok(GetLastError() == section->error_code, "test %d: Expected %u as error, got %u\n",
                   i, section->error_code, GetLastError());
            }
        }
 
        memset(buffer, 0x0, sizeof(buffer));
        ret = SetupQueryDrivesInDiskSpaceListA(diskspace, buffer, sizeof(buffer), NULL);
        ok(ret, "test %d: Expected SetupQueryDrivesInDiskSpaceListA to succeed (%u)\n", i, GetLastError());
        ok(!memcmp(section_test[i].devices, buffer, section_test[i].device_length),
           "test %d: Device list (%s) does not match\n", i, buffer);
 
        for (j = 0; j < 2; j++)
        {
            const struct device_usage *usage = &section_test[i].usage[j];
            if (!usage->dev)
                continue;
 
            space = 0;
            ret = SetupQuerySpaceRequiredOnDriveA(diskspace, usage->dev, &space, NULL, 0);
            ok(ret, "test %d: Expected SetupQuerySpaceRequiredOnDriveA to succeed for device %s (%u)\n",
               i, usage->dev, GetLastError());
            ok(space == usage->usage, "test %d: Expected size %u for device %s, got %u\n",
               i, (DWORD)usage->usage, usage->dev, (DWORD)space);
        }
 
        ok(SetupDestroyDiskSpaceList(diskspace),
           "Expected SetupDestroyDiskSpaceList to succeed\n");
 
        SetupCloseInfFile(inf);
    }
 
    DeleteFileA(tmpfilename);
}

Referenced by START_TEST().

test_SetupAddToDiskSpaceListA()

static void test_SetupAddToDiskSpaceListA ( void  )

static

Definition at line 669 of file diskspace.c.

{
    HDSKSPC handle;
    BOOL ret;
 
    ret = SetupAddToDiskSpaceListA(NULL, "C:\\some-file.dat", 0, FILEOP_COPY, 0, 0);
    ok(!ret, "Expected SetupAddToDiskSpaceListA to return FALSE, got %d\n", ret);
    ok(GetLastError() == ERROR_INVALID_HANDLE,
       "Expected GetLastError() to return ERROR_INVALID_HANDLE, got %u\n", GetLastError());
 
    handle = SetupCreateDiskSpaceListA(NULL, 0, 0);
    ok(handle != NULL,"Expected SetupCreateDiskSpaceListA to return a valid handle\n");
 
    ret = SetupAddToDiskSpaceListA(handle, NULL, 0, FILEOP_COPY, 0, 0);
    ok(ret || broken(!ret) /* >= Vista */, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    ret = SetupAddToDiskSpaceListA(handle, "C:\\some-file.dat", -20, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    ret = SetupAddToDiskSpaceListA(handle, "C:\\some-file.dat", 0, FILEOP_RENAME, 0, 0);
    ok(!ret, "Expected SetupAddToDiskSpaceListA to return FALSE\n");
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    ret = SetupAddToDiskSpaceListA(handle, NULL, 0, FILEOP_RENAME, 0, 0);
    ok(ret || broken(!ret) /* >= Vista */, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    ret = SetupAddToDiskSpaceListA(NULL, NULL, 0, FILEOP_RENAME, 0, 0);
    ok(ret || broken(!ret) /* >= Vista */, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    ok(SetupDestroyDiskSpaceList(handle),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
}

Referenced by START_TEST().

test_SetupCreateDiskSpaceListA()

static void test_SetupCreateDiskSpaceListA ( void  )

static

Definition at line 59 of file diskspace.c.

{
    HDSKSPC ret;
 
    ret = SetupCreateDiskSpaceListA(NULL, 0, 0);
    ok(ret != NULL,
       "Expected SetupCreateDiskSpaceListA to return a valid handle, got NULL\n");
 
    ok(SetupDestroyDiskSpaceList(ret), "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    ret = SetupCreateDiskSpaceListA(NULL, 0, SPDSL_IGNORE_DISK);
    ok(ret != NULL,
       "Expected SetupCreateDiskSpaceListA to return a valid handle, got NULL\n");
 
    ok(SetupDestroyDiskSpaceList(ret), "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    SetLastError(0xdeadbeef);
    ret = SetupCreateDiskSpaceListA(NULL, 0, ~0U);
    ok(ret == NULL ||
       broken(ret != NULL), /* NT4/Win9x/Win2k */
       "Expected SetupCreateDiskSpaceListA to return NULL, got %p\n", ret);
    if (!ret)
        ok(GetLastError() == ERROR_INVALID_PARAMETER,
           "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
           GetLastError());
    else
        ok(SetupDestroyDiskSpaceList(ret), "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    SetLastError(0xdeadbeef);
    ret = SetupCreateDiskSpaceListA(NULL, 0xdeadbeef, 0);
    ok(ret == NULL,
       "Expected SetupCreateDiskSpaceListA to return NULL, got %p\n", ret);
    ok(GetLastError() == ERROR_INVALID_PARAMETER ||
       broken(GetLastError() == 0xdeadbeef), /* NT4/Win9x/Win2k */
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    ret = SetupCreateDiskSpaceListA((void *)0xdeadbeef, 0, 0);
    ok(ret == NULL,
       "Expected SetupCreateDiskSpaceListA to return NULL, got %p\n", ret);
    ok(GetLastError() == ERROR_INVALID_PARAMETER ||
       broken(GetLastError() == 0xdeadbeef), /* NT4/Win9x/Win2k */
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    ret = SetupCreateDiskSpaceListA((void *)0xdeadbeef, 0xdeadbeef, 0);
    ok(ret == NULL,
       "Expected SetupCreateDiskSpaceListA to return NULL, got %p\n", ret);
    ok(GetLastError() == ERROR_INVALID_PARAMETER ||
       broken(GetLastError() == 0xdeadbeef), /* NT4/Win9x/Win2k */
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
}

Referenced by START_TEST().

test_SetupCreateDiskSpaceListW()

static void test_SetupCreateDiskSpaceListW ( void  )

static

Definition at line 115 of file diskspace.c.

{
    HDSKSPC ret;
 
    ret = SetupCreateDiskSpaceListW(NULL, 0, 0);
    if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
    {
        win_skip("SetupCreateDiskSpaceListW is not implemented\n");
        return;
    }
    ok(ret != NULL,
       "Expected SetupCreateDiskSpaceListW to return a valid handle, got NULL\n");
 
    ok(SetupDestroyDiskSpaceList(ret), "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    ret = SetupCreateDiskSpaceListW(NULL, 0, SPDSL_IGNORE_DISK);
    ok(ret != NULL,
       "Expected SetupCreateDiskSpaceListW to return a valid handle, got NULL\n");
 
    ok(SetupDestroyDiskSpaceList(ret), "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    SetLastError(0xdeadbeef);
    ret = SetupCreateDiskSpaceListW(NULL, 0, ~0U);
    ok(ret == NULL ||
       broken(ret != NULL), /* NT4/Win2k */
       "Expected SetupCreateDiskSpaceListW to return NULL, got %p\n", ret);
    if (!ret)
        ok(GetLastError() == ERROR_INVALID_PARAMETER,
           "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
           GetLastError());
    else
        ok(SetupDestroyDiskSpaceList(ret), "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    SetLastError(0xdeadbeef);
    ret = SetupCreateDiskSpaceListW(NULL, 0xdeadbeef, 0);
    ok(ret == NULL,
       "Expected SetupCreateDiskSpaceListW to return NULL, got %p\n", ret);
    ok(GetLastError() == ERROR_INVALID_PARAMETER ||
       broken(GetLastError() == 0xdeadbeef), /* NT4/Win2k */
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    ret = SetupCreateDiskSpaceListW((void *)0xdeadbeef, 0, 0);
    ok(ret == NULL,
       "Expected SetupCreateDiskSpaceListW to return NULL, got %p\n", ret);
    ok(GetLastError() == ERROR_INVALID_PARAMETER ||
       broken(GetLastError() == 0xdeadbeef), /* NT4/Win2k */
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    ret = SetupCreateDiskSpaceListW((void *)0xdeadbeef, 0xdeadbeef, 0);
    ok(ret == NULL,
       "Expected SetupCreateDiskSpaceListW to return NULL, got %p\n", ret);
    ok(GetLastError() == ERROR_INVALID_PARAMETER ||
       broken(GetLastError() == 0xdeadbeef), /* NT4/Win2k */
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
}

Referenced by START_TEST().

test_SetupDuplicateDiskSpaceListA()

static void test_SetupDuplicateDiskSpaceListA ( void  )

static

Definition at line 176 of file diskspace.c.

{
    HDSKSPC handle, duplicate;
 
    if (is_win9x)
        win_skip("SetupDuplicateDiskSpaceListA crashes with NULL disk space handle on Win9x\n");
    else
    {
        SetLastError(0xdeadbeef);
        duplicate = SetupDuplicateDiskSpaceListA(NULL, NULL, 0, 0);
        ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
        ok(GetLastError() == ERROR_INVALID_HANDLE,
           "Expected GetLastError() to return ERROR_INVALID_HANDLE, got %u\n", GetLastError());
 
        SetLastError(0xdeadbeef);
        duplicate = SetupDuplicateDiskSpaceListA(NULL, (void *)0xdeadbeef, 0, 0);
        ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
        ok(GetLastError() == ERROR_INVALID_PARAMETER,
           "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
        SetLastError(0xdeadbeef);
        duplicate = SetupDuplicateDiskSpaceListA(NULL, NULL, 0xdeadbeef, 0);
        ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
        ok(GetLastError() == ERROR_INVALID_PARAMETER,
           "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
        SetLastError(0xdeadbeef);
        duplicate = SetupDuplicateDiskSpaceListA(NULL, NULL, 0, ~0U);
        ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
        ok(GetLastError() == ERROR_INVALID_PARAMETER,
           "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
    }
 
    handle = SetupCreateDiskSpaceListA(NULL, 0, 0);
    ok(handle != NULL,
       "Expected SetupCreateDiskSpaceListA to return a valid handle, got NULL\n");
 
    if (!handle)
    {
        skip("Failed to create a disk space handle\n");
        return;
    }
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListA(handle, (void *)0xdeadbeef, 0, 0);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListA(handle, NULL, 0xdeadbeef, 0);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListA(handle, NULL, 0, SPDSL_IGNORE_DISK);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListA(handle, NULL, 0, ~0U);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    duplicate = SetupDuplicateDiskSpaceListA(handle, NULL, 0, 0);
    ok(duplicate != NULL, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(duplicate != handle,
       "Expected new handle (%p) to be different from the old handle (%p)\n", duplicate, handle);
 
    ok(SetupDestroyDiskSpaceList(duplicate), "Expected SetupDestroyDiskSpaceList to succeed\n");
    ok(SetupDestroyDiskSpaceList(handle), "Expected SetupDestroyDiskSpaceList to succeed\n");
}

Referenced by START_TEST().

test_SetupDuplicateDiskSpaceListW()

static void test_SetupDuplicateDiskSpaceListW ( void  )

static

Definition at line 252 of file diskspace.c.

{
    HDSKSPC handle, duplicate;
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListW(NULL, NULL, 0, 0);
    if (!duplicate && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
    {
        win_skip("SetupDuplicateDiskSpaceListW is not available\n");
        return;
    }
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_HANDLE,
       "Expected GetLastError() to return ERROR_INVALID_HANDLE, got %u\n", GetLastError());
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListW(NULL, (void *)0xdeadbeef, 0, 0);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListW(NULL, NULL, 0xdeadbeef, 0);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListW(NULL, NULL, 0, ~0U);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    handle = SetupCreateDiskSpaceListW(NULL, 0, 0);
    ok(handle != NULL,
       "Expected SetupCreateDiskSpaceListW to return a valid handle, got NULL\n");
 
    if (!handle)
    {
        skip("Failed to create a disk space handle\n");
        return;
    }
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListW(handle, (void *)0xdeadbeef, 0, 0);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListW(handle, NULL, 0xdeadbeef, 0);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListW(handle, NULL, 0, SPDSL_IGNORE_DISK);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    SetLastError(0xdeadbeef);
    duplicate = SetupDuplicateDiskSpaceListW(handle, NULL, 0, ~0U);
    ok(!duplicate, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(GetLastError() == ERROR_INVALID_PARAMETER,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
 
    duplicate = SetupDuplicateDiskSpaceListW(handle, NULL, 0, 0);
    ok(duplicate != NULL, "Expected SetupDuplicateDiskSpaceList to return NULL, got %p\n", duplicate);
    ok(duplicate != handle,
       "Expected new handle (%p) to be different from the old handle (%p)\n", duplicate, handle);
 
    ok(SetupDestroyDiskSpaceList(duplicate), "Expected SetupDestroyDiskSpaceList to succeed\n");
    ok(SetupDestroyDiskSpaceList(handle), "Expected SetupDestroyDiskSpaceList to succeed\n");
}

Referenced by START_TEST().

test_SetupQueryDrivesInDiskSpaceListA()

static void test_SetupQueryDrivesInDiskSpaceListA ( void  )

static

Definition at line 703 of file diskspace.c.

{
    char buffer[MAX_PATH];
    HDSKSPC handle;
    DWORD size;
    BOOL ret;
 
    handle = SetupCreateDiskSpaceListA(NULL, 0, SPDSL_IGNORE_DISK);
    ok(handle != NULL,"Expected SetupCreateDiskSpaceListA to return a valid handle\n");
 
    ret = SetupQueryDrivesInDiskSpaceListA(handle, NULL, 0, NULL);
    ok(ret, "Expected SetupQueryDrivesInDiskSpaceListA to succeed\n");
 
    size = 0;
    ret = SetupQueryDrivesInDiskSpaceListA(handle, NULL, 0, &size);
    ok(ret, "Expected SetupQueryDrivesInDiskSpaceListA to succeed\n");
    ok(size == 1, "Expected size 1, got %u\n", size);
 
    ret = SetupAddToDiskSpaceListA(handle, "F:\\random-file.dat", 0, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    ret = SetupAddToDiskSpaceListA(handle, "G:\\random-file.dat", 0, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    ret = SetupAddToDiskSpaceListA(handle, "G:\\random-file2.dat", 0, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    ret = SetupAddToDiskSpaceListA(handle, "X:\\random-file.dat", 0, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    size = 0;
    ret = SetupQueryDrivesInDiskSpaceListA(handle, NULL, 0, &size);
    ok(ret, "Expected SetupQueryDrivesInDiskSpaceListA to succeed\n");
    ok(size == 10, "Expected size 10, got %u\n", size);
 
    size = 0;
    ret = SetupQueryDrivesInDiskSpaceListA(handle, buffer, 0, &size);
    ok(!ret, "Expected SetupQueryDrivesInDiskSpaceListA to fail\n");
    ok(size == 4, "Expected size 4, got %u\n", size);
    ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
       "Expected GetLastError() to return ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
 
    size = 0;
    ret = SetupQueryDrivesInDiskSpaceListA(handle, buffer, 4, &size);
    ok(!ret, "Expected SetupQueryDrivesInDiskSpaceListA to fail\n");
    ok(size == 7, "Expected size 7, got %u\n", size);
    ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
       "Expected GetLastError() to return ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
 
    size = 0;
    ret = SetupQueryDrivesInDiskSpaceListA(handle, buffer, 7, &size);
    ok(!ret, "Expected SetupQueryDrivesInDiskSpaceListA to fail\n");
    ok(size == 10, "Expected size 10, got %u\n", size);
    ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
       "Expected GetLastError() to return ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
 
    size = 0;
    memset(buffer, 0xff, sizeof(buffer));
    ret = SetupQueryDrivesInDiskSpaceListA(handle, buffer, sizeof(buffer), &size);
    ok(ret, "Expected SetupQueryDrivesInDiskSpaceListA to succeed\n");
    ok(size == 10, "Expected size 10, got %u\n", size);
    ok(!memcmp("f:\0g:\0x:\0\0", buffer, 10), "Device list does not match\n");
 
    memset(buffer, 0xff, sizeof(buffer));
    ret = SetupQueryDrivesInDiskSpaceListA(handle, buffer, sizeof(buffer), NULL);
    ok(ret, "Expected SetupQueryDrivesInDiskSpaceListA to succeed\n");
    ok(!memcmp("f:\0g:\0x:\0\0", buffer, 10), "Device list does not match\n");
 
    ok(SetupDestroyDiskSpaceList(handle),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
}

Referenced by START_TEST().

test_SetupQuerySpaceRequiredOnDriveA()

static void test_SetupQuerySpaceRequiredOnDriveA ( void  )

static

Definition at line 344 of file diskspace.c.

{
    BOOL ret;
    HDSKSPC handle;
    LONGLONG space;
    char windir[MAX_PATH];
    char drive[3];
    char tmp[MAX_PATH];
    LONGLONG size;
 
    if (is_win9x)
        win_skip("SetupQuerySpaceRequiredOnDriveA crashes with NULL disk space handle on Win9x\n");
    else
    {
        SetLastError(0xdeadbeef);
        ret = SetupQuerySpaceRequiredOnDriveA(NULL, NULL, NULL, NULL, 0);
        ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveA to return FALSE, got %d\n", ret);
        ok(GetLastError() == ERROR_INVALID_PARAMETER,
           "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
           GetLastError());
 
        SetLastError(0xdeadbeef);
        space = 0xdeadbeef;
        ret = SetupQuerySpaceRequiredOnDriveA(NULL, NULL, &space, NULL, 0);
        ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveA to return FALSE, got %d\n", ret);
        ok(space == 0xdeadbeef, "Expected output space parameter to be untouched\n");
        ok(GetLastError() == ERROR_INVALID_PARAMETER,
           "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
           GetLastError());
 
        SetLastError(0xdeadbeef);
        ret = SetupQuerySpaceRequiredOnDriveA(NULL, "", NULL, NULL, 0);
        ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveA to return FALSE, got %d\n", ret);
        ok(GetLastError() == ERROR_INVALID_HANDLE,
           "Expected GetLastError() to return ERROR_INVALID_HANDLE, got %u\n",
           GetLastError());
 
        SetLastError(0xdeadbeef);
        space = 0xdeadbeef;
        ret = SetupQuerySpaceRequiredOnDriveA(NULL, "", &space, NULL, 0);
        ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveA to return FALSE, got %d\n", ret);
        ok(space == 0xdeadbeef, "Expected output space parameter to be untouched\n");
        ok(GetLastError() == ERROR_INVALID_HANDLE,
           "Expected GetLastError() to return ERROR_INVALID_HANDLE, got %u\n",
           GetLastError());
    }
 
    handle = SetupCreateDiskSpaceListA(NULL, 0, 0);
    ok(handle != NULL,
       "Expected SetupCreateDiskSpaceListA to return a valid handle, got NULL\n");
 
    SetLastError(0xdeadbeef);
    ret = SetupQuerySpaceRequiredOnDriveA(handle, NULL, NULL, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveA to return FALSE, got %d\n", ret);
    ok(GetLastError() == ERROR_INVALID_PARAMETER ||
       GetLastError() == ERROR_INVALID_DRIVE, /* Win9x */
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    space = 0xdeadbeef;
    ret = SetupQuerySpaceRequiredOnDriveA(handle, NULL, &space, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveA to return FALSE, got %d\n", ret);
    ok(space == 0xdeadbeef, "Expected output space parameter to be untouched\n");
    ok(GetLastError() == ERROR_INVALID_PARAMETER ||
       GetLastError() == ERROR_INVALID_DRIVE, /* Win9x */
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    ret = SetupQuerySpaceRequiredOnDriveA(handle, "", NULL, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveA to return FALSE, got %d\n", ret);
    ok(GetLastError() == ERROR_INVALID_DRIVE,
       "Expected GetLastError() to return ERROR_INVALID_DRIVE, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    space = 0xdeadbeef;
    ret = SetupQuerySpaceRequiredOnDriveA(handle, "", &space, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveA to return FALSE, got %d\n", ret);
    ok(space == 0xdeadbeef, "Expected output space parameter to be untouched\n");
    ok(GetLastError() == ERROR_INVALID_DRIVE,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    GetWindowsDirectoryA(windir, MAX_PATH);
    drive[0] = windir[0]; drive[1] = windir[1]; drive[2] = 0;
 
    snprintf(tmp, MAX_PATH, "%c:\\wine-test-should-not-exist.txt", drive[0]);
    ret = SetupAddToDiskSpaceListA(handle, tmp, 0x100000, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    space = 0;
    ret = SetupQuerySpaceRequiredOnDriveA(handle, drive, &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space == 0x100000, "Expected 0x100000 as required space, got %s\n", debugstr_longlong(space));
 
    /* adding the same file again doesn't sum up the size */
    ret = SetupAddToDiskSpaceListA(handle, tmp, 0x200000, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    space = 0;
    ret = SetupQuerySpaceRequiredOnDriveA(handle, drive, &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space == 0x200000, "Expected 0x200000 as required space, got %s\n", debugstr_longlong(space));
 
    /* the device doesn't need to exist */
    snprintf(tmp, MAX_PATH, "F:\\wine-test-should-not-exist.txt");
    ret = SetupAddToDiskSpaceListA(handle, tmp, 0x200000, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    ret = SetupQuerySpaceRequiredOnDriveA(handle, "F:", &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space == 0x200000, "Expected 0x200000 as required space, got %s\n", debugstr_longlong(space));
 
    snprintf(tmp, MAX_PATH, "F:\\wine-test-should-not-exist.txt");
    ret = SetupAddToDiskSpaceListA(handle, tmp, 0x200000, FILEOP_DELETE, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    ret = SetupQuerySpaceRequiredOnDriveA(handle, "F:", &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space == 0x200000, "Expected 0x200000 as required space, got %s\n", debugstr_longlong(space));
 
    ok(SetupDestroyDiskSpaceList(handle),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    handle = SetupCreateDiskSpaceListA(NULL, 0, 0);
    ok(handle != NULL,
       "Expected SetupCreateDiskSpaceListA to return a valid handle, got NULL\n");
 
    /* the real size is subtracted unless SPDSL_IGNORE_DISK is specified */
    snprintf(tmp, MAX_PATH, "%s\\regedit.exe", windir);
 
    size = get_file_size(tmp);
    ret = SetupAddToDiskSpaceListA(handle, tmp, size, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
    space = 0;
    ret = SetupQuerySpaceRequiredOnDriveA(handle, drive, &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space == 0 || broken(space == -0x5000) || broken(space == -0x7000),
       "Expected 0x0 as required space, got %s\n", debugstr_longlong(space));
 
    ret = SetupAddToDiskSpaceListA(handle, tmp, size + 0x100000, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
    ret = SetupQuerySpaceRequiredOnDriveA(handle, drive, &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space == 0x100000 || broken(space == 0xf9000) || broken(space == 0xfb000),
       "Expected 0x100000 as required space, got %s\n", debugstr_longlong(space));
 
    ret = SetupAddToDiskSpaceListA(handle, tmp, size - 0x1000, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
    ret = SetupQuerySpaceRequiredOnDriveA(handle, drive, &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space == -0x1000 || broken(space == -0x6000) || broken(space == -0x8000),
       "Expected -0x1000 as required space, got %s\n", debugstr_longlong(space));
 
    ok(SetupDestroyDiskSpaceList(handle),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    /* test FILEOP_DELETE, then FILEOP_COPY */
    handle = SetupCreateDiskSpaceListA(NULL, 0, 0);
    ok(handle != NULL,
       "Expected SetupCreateDiskSpaceListA to return a valid handle, got NULL\n");
 
    ret = SetupAddToDiskSpaceListA(handle, tmp, size, FILEOP_DELETE, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
    ret = SetupAddToDiskSpaceListA(handle, tmp, size, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    space = 0;
    ret = SetupQuerySpaceRequiredOnDriveA(handle, drive, &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space == 0 || broken(space == -0x5000) || broken(space == -0x7000),
       "Expected 0x0 as required space, got %s\n", debugstr_longlong(space));
 
    ok(SetupDestroyDiskSpaceList(handle),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    /* test FILEOP_COPY, then FILEOP_DELETE */
    handle = SetupCreateDiskSpaceListA(NULL, 0, 0);
    ok(handle != NULL,
       "Expected SetupCreateDiskSpaceListA to return a valid handle, got NULL\n");
 
    ret = SetupAddToDiskSpaceListA(handle, tmp, size, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
    ret = SetupAddToDiskSpaceListA(handle, tmp, size, FILEOP_DELETE, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    space = 0;
    ret = SetupQuerySpaceRequiredOnDriveA(handle, drive, &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space == 0 || broken(space == -0x5000) || broken(space == -0x7000),
       "Expected 0x0 as required space, got %s\n", debugstr_longlong(space));
 
    ok(SetupDestroyDiskSpaceList(handle),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    /* test FILEOP_DELETE without SPDSL_IGNORE_DISK */
    handle = SetupCreateDiskSpaceListA(NULL, 0, 0);
    ok(handle != NULL,
       "Expected SetupCreateDiskSpaceListA to return a valid handle, got NULL\n");
 
    ret = SetupAddToDiskSpaceListA(handle, tmp, size, FILEOP_DELETE, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
    space = 0;
    ret = SetupQuerySpaceRequiredOnDriveA(handle, drive, &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space <= -size, "Expected space <= -size, got %s\n", debugstr_longlong(space));
 
    ok(SetupDestroyDiskSpaceList(handle),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
 
    /* test FILEOP_COPY and FILEOP_DELETE with SPDSL_IGNORE_DISK */
    handle = SetupCreateDiskSpaceListA(NULL, 0, SPDSL_IGNORE_DISK);
    ok(handle != NULL,
       "Expected SetupCreateDiskSpaceListA to return a valid handle, got NULL\n");
 
    ret = SetupAddToDiskSpaceListA(handle, tmp, size, FILEOP_DELETE, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
    space = 0;
    ret = SetupQuerySpaceRequiredOnDriveA(handle, drive, &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space == 0, "Expected size = 0, got %s\n", debugstr_longlong(space));
 
    ret = SetupAddToDiskSpaceListA(handle, tmp, size, FILEOP_COPY, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
    ret = SetupAddToDiskSpaceListA(handle, tmp, size, FILEOP_DELETE, 0, 0);
    ok(ret, "Expected SetupAddToDiskSpaceListA to succeed\n");
 
    space = 0;
    ret = SetupQuerySpaceRequiredOnDriveA(handle, drive, &space, NULL, 0);
    ok(ret, "Expected SetupQuerySpaceRequiredOnDriveA to succeed\n");
    ok(space >= size, "Expected size >= %s\n", debugstr_longlong(space));
 
    ok(SetupDestroyDiskSpaceList(handle),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
}

Referenced by START_TEST().

test_SetupQuerySpaceRequiredOnDriveW()

static void test_SetupQuerySpaceRequiredOnDriveW ( void  )

static

Definition at line 582 of file diskspace.c.

{
    static const WCHAR emptyW[] = {0};
 
    BOOL ret;
    HDSKSPC handle;
    LONGLONG space;
 
    SetLastError(0xdeadbeef);
    ret = SetupQuerySpaceRequiredOnDriveW(NULL, NULL, NULL, NULL, 0);
    if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
    {
        win_skip("SetupQuerySpaceRequiredOnDriveW is not available\n");
        return;
    }
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveW to return FALSE, got %d\n", ret);
    ok(GetLastError() == ERROR_INVALID_HANDLE,
       "Expected GetLastError() to return ERROR_INVALID_HANDLE, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    space = 0xdeadbeef;
    ret = SetupQuerySpaceRequiredOnDriveW(NULL, NULL, &space, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveW to return FALSE, got %d\n", ret);
    ok(space == 0xdeadbeef, "Expected output space parameter to be untouched\n");
    ok(GetLastError() == ERROR_INVALID_HANDLE,
       "Expected GetLastError() to return ERROR_INVALID_HANDLE, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    ret = SetupQuerySpaceRequiredOnDriveW(NULL, emptyW, NULL, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveW to return FALSE, got %d\n", ret);
    ok(GetLastError() == ERROR_INVALID_HANDLE,
       "Expected GetLastError() to return ERROR_INVALID_HANDLE, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    space = 0xdeadbeef;
    ret = SetupQuerySpaceRequiredOnDriveW(NULL, emptyW, &space, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveW to return FALSE, got %d\n", ret);
    ok(space == 0xdeadbeef, "Expected output space parameter to be untouched\n");
    ok(GetLastError() == ERROR_INVALID_HANDLE,
       "Expected GetLastError() to return ERROR_INVALID_HANDLE, got %u\n",
       GetLastError());
 
    handle = SetupCreateDiskSpaceListA(NULL, 0, 0);
    ok(handle != NULL,
       "Expected SetupCreateDiskSpaceListA to return a valid handle, got NULL\n");
 
    SetLastError(0xdeadbeef);
    ret = SetupQuerySpaceRequiredOnDriveW(handle, NULL, NULL, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveW to return FALSE, got %d\n", ret);
    ok(GetLastError() == ERROR_INVALID_PARAMETER ||
       GetLastError() == ERROR_INVALID_DRIVE, /* NT4/Win2k/XP/Win2k3 */
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    space = 0xdeadbeef;
    ret = SetupQuerySpaceRequiredOnDriveW(handle, NULL, &space, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveW to return FALSE, got %d\n", ret);
    ok(space == 0xdeadbeef, "Expected output space parameter to be untouched\n");
    ok(GetLastError() == ERROR_INVALID_PARAMETER ||
       GetLastError() == ERROR_INVALID_DRIVE, /* NT4/Win2k/XP/Win2k3 */
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    ret = SetupQuerySpaceRequiredOnDriveW(handle, emptyW, NULL, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveW to return FALSE, got %d\n", ret);
    ok(GetLastError() == ERROR_INVALID_DRIVE,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    SetLastError(0xdeadbeef);
    space = 0xdeadbeef;
    ret = SetupQuerySpaceRequiredOnDriveW(handle, emptyW, &space, NULL, 0);
    ok(!ret, "Expected SetupQuerySpaceRequiredOnDriveW to return FALSE, got %d\n", ret);
    ok(space == 0xdeadbeef, "Expected output space parameter to be untouched\n");
    ok(GetLastError() == ERROR_INVALID_DRIVE,
       "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
       GetLastError());
 
    ok(SetupDestroyDiskSpaceList(handle),
       "Expected SetupDestroyDiskSpaceList to succeed\n");
}

Referenced by START_TEST().

Variable Documentation

data

const char* data

Definition at line 791 of file diskspace.c.

device_length

int device_length

Definition at line 794 of file diskspace.c.

Referenced by test_SetupAddInstallSectionToDiskSpaceListA(), and test_SetupAddSectionToDiskSpaceListA().

devices

const char* devices

Definition at line 793 of file diskspace.c.

Referenced by test_SetupAddInstallSectionToDiskSpaceListA(), and test_SetupAddSectionToDiskSpaceListA().

is_win9x

BOOL is_win9x

static

Definition at line 33 of file diskspace.c.

Referenced by clip_emf_enum_proc(), START_TEST(), test_SendMessageTimeout(), test_SetupDuplicateDiskSpaceListA(), and test_SetupQuerySpaceRequiredOnDriveA().

const struct { ... } section_test[]

Referenced by test_SetupAddSectionToDiskSpaceListA().

const struct { ... } section_test_i[]

Referenced by test_SetupAddInstallSectionToDiskSpaceListA().

sections

struct section_i sections[2]

Definition at line 792 of file diskspace.c.

Referenced by create_test_dll_sections(), dwarf2_parse_compilation_unit(), dwarf2_parse_line_numbers(), find_pe_resource(), macho_load_section_info(), map_image_section(), and SetupGetLineByIndexW().

usage

struct device_usage usage[2]

(

void

)

Definition at line 795 of file diskspace.c.