diskspace.c File Reference

#include "setupapi_private.h"

Include dependency graph for diskspace.c:

Go to the source code of this file.

Classes
struct	DRIVE_ENTRY
struct	DISKSPACELIST

Typedefs
typedef struct DRIVE_ENTRY *	LPDRIVE_ENTRY
typedef struct DISKSPACELIST *	LPDISKSPACELIST

Functions
HDSKSPC WINAPI	SetupCreateDiskSpaceListW (PVOID Reserved1, DWORD Reserved2, UINT Flags)
HDSKSPC WINAPI	SetupCreateDiskSpaceListA (PVOID Reserved1, DWORD Reserved2, UINT Flags)
HDSKSPC WINAPI	SetupDuplicateDiskSpaceListW (HDSKSPC DiskSpace, PVOID Reserved1, DWORD Reserved2, UINT Flags)
HDSKSPC WINAPI	SetupDuplicateDiskSpaceListA (HDSKSPC DiskSpace, PVOID Reserved1, DWORD Reserved2, UINT Flags)
BOOL WINAPI	SetupAddInstallSectionToDiskSpaceListA (HDSKSPC DiskSpace, HINF InfHandle, HINF LayoutInfHandle, LPCSTR SectionName, PVOID Reserved1, UINT Reserved2)
BOOL WINAPI	SetupQuerySpaceRequiredOnDriveW (HDSKSPC DiskSpace, LPCWSTR DriveSpec, LONGLONG *SpaceRequired, PVOID Reserved1, UINT Reserved2)
BOOL WINAPI	SetupQuerySpaceRequiredOnDriveA (HDSKSPC DiskSpace, LPCSTR DriveSpec, LONGLONG *SpaceRequired, PVOID Reserved1, UINT Reserved2)
BOOL WINAPI	SetupDestroyDiskSpaceList (HDSKSPC DiskSpace)
BOOL WINAPI	SetupAddToDiskSpaceListA (HDSKSPC diskspace, PCSTR targetfile, LONGLONG filesize, UINT operation, PVOID reserved1, UINT reserved2)
BOOL WINAPI	SetupAddToDiskSpaceListW (HDSKSPC diskspace, PCWSTR targetfile, LONGLONG filesize, UINT operation, PVOID reserved1, UINT reserved2)

Typedef Documentation

LPDISKSPACELIST

typedef struct DISKSPACELIST * LPDISKSPACELIST

LPDRIVE_ENTRY

typedef struct DRIVE_ENTRY * LPDRIVE_ENTRY

Function Documentation

SetupAddInstallSectionToDiskSpaceListA()

BOOL WINAPI SetupAddInstallSectionToDiskSpaceListA	(	HDSKSPC	DiskSpace,
		HINF	InfHandle,
		HINF	LayoutInfHandle,
		LPCSTR	SectionName,
		PVOID	Reserved1,
		UINT	Reserved2
	)

Definition at line 136 of file diskspace.c.

{
    FIXME ("Stub\n");
    return TRUE;
}

Referenced by test_SetupAddInstallSectionToDiskSpaceListA().

SetupAddToDiskSpaceListA()

BOOL WINAPI SetupAddToDiskSpaceListA	(	HDSKSPC	diskspace,
		PCSTR	targetfile,
		LONGLONG	filesize,
		UINT	operation,
		PVOID	reserved1,
		UINT	reserved2
	)

Definition at line 255 of file diskspace.c.

{
    FIXME(": stub\n");
    SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
    return FALSE;
}

Referenced by test_SetupAddToDiskSpaceListA(), test_SetupQueryDrivesInDiskSpaceListA(), and test_SetupQuerySpaceRequiredOnDriveA().

SetupAddToDiskSpaceListW()

BOOL WINAPI SetupAddToDiskSpaceListW	(	HDSKSPC	diskspace,
		PCWSTR	targetfile,
		LONGLONG	filesize,
		UINT	operation,
		PVOID	reserved1,
		UINT	reserved2
	)

Definition at line 267 of file diskspace.c.

{
    FIXME(": stub\n");
    SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
    return FALSE;
}

SetupCreateDiskSpaceListA()

HDSKSPC WINAPI SetupCreateDiskSpaceListA	(	PVOID	Reserved1,
		DWORD	Reserved2,
		UINT	Flags
	)

Definition at line 89 of file diskspace.c.

{
    return SetupCreateDiskSpaceListW( Reserved1, Reserved2, Flags );
}

Referenced by test_SetupAddInstallSectionToDiskSpaceListA(), test_SetupAddSectionToDiskSpaceListA(), test_SetupAddToDiskSpaceListA(), test_SetupCreateDiskSpaceListA(), test_SetupDuplicateDiskSpaceListA(), test_SetupQueryDrivesInDiskSpaceListA(), test_SetupQuerySpaceRequiredOnDriveA(), and test_SetupQuerySpaceRequiredOnDriveW().

SetupCreateDiskSpaceListW()

HDSKSPC WINAPI SetupCreateDiskSpaceListW	(	PVOID	Reserved1,
		DWORD	Reserved2,
		UINT	Flags
	)

Definition at line 38 of file diskspace.c.

{
    WCHAR drives[255];
    DWORD rc;
    WCHAR *ptr;
    LPDISKSPACELIST list=NULL;
 
    TRACE("(%p, %u, 0x%08x)\n", Reserved1, Reserved2, Flags);
 
    if (Reserved1 || Reserved2 || Flags & ~SPDSL_IGNORE_DISK)
    {
        SetLastError(ERROR_INVALID_PARAMETER);
        return NULL;
    }
 
    rc = GetLogicalDriveStringsW(255,drives);
 
    if (rc == 0)
        return NULL;
 
    list = HeapAlloc(GetProcessHeap(),0,sizeof(DISKSPACELIST));
 
    list->dwDriveCount = 0;
 
    ptr = drives;
 
    while (*ptr)
    {
        DWORD type = GetDriveTypeW(ptr);
        if (type == DRIVE_FIXED)
        {
            DWORD clusters;
            DWORD sectors;
            DWORD bytes;
            DWORD total;
            lstrcpyW(list->Drives[list->dwDriveCount].lpzName,ptr);
            GetDiskFreeSpaceW(ptr,&sectors,&bytes,&clusters,&total);
            list->Drives[list->dwDriveCount].dwFreeSpace = clusters * sectors *
                                                           bytes;
            list->Drives[list->dwDriveCount].dwWantedSpace = 0;
            list->dwDriveCount++;
        }
       ptr += lstrlenW(ptr) + 1;
    }
    return list;
}

Referenced by SetupCreateDiskSpaceListA(), START_TEST(), test_SetupCreateDiskSpaceListW(), and test_SetupDuplicateDiskSpaceListW().

SetupDestroyDiskSpaceList()

BOOL WINAPI SetupDestroyDiskSpaceList

(

HDSKSPC

DiskSpace

)

Definition at line 245 of file diskspace.c.

{
    LPDISKSPACELIST list = (LPDISKSPACELIST)DiskSpace;
    HeapFree(GetProcessHeap(),0,list);
    return TRUE;
}

Referenced by test_SetupAddInstallSectionToDiskSpaceListA(), test_SetupAddSectionToDiskSpaceListA(), test_SetupAddToDiskSpaceListA(), test_SetupCreateDiskSpaceListA(), test_SetupCreateDiskSpaceListW(), test_SetupDuplicateDiskSpaceListA(), test_SetupDuplicateDiskSpaceListW(), test_SetupQueryDrivesInDiskSpaceListA(), test_SetupQuerySpaceRequiredOnDriveA(), and test_SetupQuerySpaceRequiredOnDriveW().

SetupDuplicateDiskSpaceListA()

HDSKSPC WINAPI SetupDuplicateDiskSpaceListA	(	HDSKSPC	DiskSpace,
		PVOID	Reserved1,
		DWORD	Reserved2,
		UINT	Flags
	)

Definition at line 128 of file diskspace.c.

{
    return SetupDuplicateDiskSpaceListW(DiskSpace, Reserved1, Reserved2, Flags);
}

Referenced by test_SetupDuplicateDiskSpaceListA().

SetupDuplicateDiskSpaceListW()

HDSKSPC WINAPI SetupDuplicateDiskSpaceListW	(	HDSKSPC	DiskSpace,
		PVOID	Reserved1,
		DWORD	Reserved2,
		UINT	Flags
	)

Definition at line 97 of file diskspace.c.

{
    DISKSPACELIST *list_copy, *list_original = DiskSpace;
 
    if (Reserved1 || Reserved2 || Flags)
    {
        SetLastError(ERROR_INVALID_PARAMETER);
        return NULL;
    }
 
    if (!DiskSpace)
    {
        SetLastError(ERROR_INVALID_HANDLE);
        return NULL;
    }
 
    list_copy = HeapAlloc(GetProcessHeap(), 0, sizeof(DISKSPACELIST));
    if (!list_copy)
    {
        SetLastError(ERROR_NOT_ENOUGH_MEMORY);
        return NULL;
    }
 
    *list_copy = *list_original;
 
    return list_copy;
}

Referenced by SetupDuplicateDiskSpaceListA(), and test_SetupDuplicateDiskSpaceListW().

SetupQuerySpaceRequiredOnDriveA()

BOOL WINAPI SetupQuerySpaceRequiredOnDriveA	(	HDSKSPC	DiskSpace,
		LPCSTR	DriveSpec,
		LONGLONG *	SpaceRequired,
		PVOID	Reserved1,
		UINT	Reserved2
	)

Definition at line 201 of file diskspace.c.

{
    DWORD len;
    LPWSTR DriveSpecW;
    BOOL ret;
 
    /* The parameter validation checks are in a different order from the
     * Unicode variant of SetupQuerySpaceRequiredOnDrive. */
    if (!DriveSpec)
    {
        SetLastError(ERROR_INVALID_PARAMETER);
        return FALSE;
    }
 
    if (!DiskSpace)
    {
        SetLastError(ERROR_INVALID_HANDLE);
        return FALSE;
    }
 
    len = MultiByteToWideChar(CP_ACP, 0, DriveSpec, -1, NULL, 0);
 
    DriveSpecW = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
    if (!DriveSpecW)
    {
        SetLastError(ERROR_NOT_ENOUGH_MEMORY);
        return FALSE;
    }
 
    MultiByteToWideChar(CP_ACP, 0, DriveSpec, -1, DriveSpecW, len);
 
    ret = SetupQuerySpaceRequiredOnDriveW(DiskSpace, DriveSpecW, SpaceRequired,
                                          Reserved1, Reserved2);
 
    HeapFree(GetProcessHeap(), 0, DriveSpecW);
 
    return ret;
}

Referenced by test_SetupAddInstallSectionToDiskSpaceListA(), test_SetupAddSectionToDiskSpaceListA(), and test_SetupQuerySpaceRequiredOnDriveA().

SetupQuerySpaceRequiredOnDriveW()

BOOL WINAPI SetupQuerySpaceRequiredOnDriveW	(	HDSKSPC	DiskSpace,
		LPCWSTR	DriveSpec,
		LONGLONG *	SpaceRequired,
		PVOID	Reserved1,
		UINT	Reserved2
	)

Definition at line 147 of file diskspace.c.

{
    WCHAR *driveW;
    unsigned int i;
    LPDISKSPACELIST list = DiskSpace;
    BOOL rc = FALSE;
    static const WCHAR bkslsh[]= {'\\',0};
 
    if (!DiskSpace)
    {
        SetLastError(ERROR_INVALID_HANDLE);
        return FALSE;
    }
 
    if (!DriveSpec)
    {
        SetLastError(ERROR_INVALID_PARAMETER);
        return FALSE;
    }
 
    driveW = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(DriveSpec) + 2) * sizeof(WCHAR));
    if (!driveW)
    {
        SetLastError(ERROR_NOT_ENOUGH_MEMORY);
        return FALSE;
    }
 
    lstrcpyW(driveW,DriveSpec);
    lstrcatW(driveW,bkslsh);
 
    TRACE("Looking for drive %s\n",debugstr_w(driveW));
 
    for (i = 0; i < list->dwDriveCount; i++)
    {
        TRACE("checking drive %s\n",debugstr_w(list->Drives[i].lpzName));
        if (lstrcmpW(driveW,list->Drives[i].lpzName)==0)
        {
            rc = TRUE;
            *SpaceRequired = list->Drives[i].dwWantedSpace;
            break;
        }
    }
 
    HeapFree(GetProcessHeap(), 0, driveW);
 
    if (!rc) SetLastError(ERROR_INVALID_DRIVE);
    return rc;
}

Referenced by SetupQuerySpaceRequiredOnDriveA(), and test_SetupQuerySpaceRequiredOnDriveW().