Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275/* * SetupAPI DiskSpace functions * * Copyright 2004 CodeWeavers (Aric Stewart) * * 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 "setupapi_private.h"
typedef struct { WCHAR lpzName[20]; LONGLONG dwFreeSpace; LONGLONG dwWantedSpace;} DRIVE_ENTRY, *LPDRIVE_ENTRY;
typedef struct { DWORD dwDriveCount; DRIVE_ENTRY Drives[26];} DISKSPACELIST, *LPDISKSPACELIST;
/*********************************************************************** * SetupCreateDiskSpaceListW (SETUPAPI.@) */HDSKSPC WINAPI SetupCreateDiskSpaceListW(PVOID Reserved1, DWORD Reserved2, UINT Flags){ 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,§ors,&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;}
/*********************************************************************** * SetupCreateDiskSpaceListA (SETUPAPI.@) */HDSKSPC WINAPI SetupCreateDiskSpaceListA(PVOID Reserved1, DWORD Reserved2, UINT Flags){ return SetupCreateDiskSpaceListW( Reserved1, Reserved2, Flags );}
/*********************************************************************** * SetupDuplicateDiskSpaceListW (SETUPAPI.@) */HDSKSPC WINAPI SetupDuplicateDiskSpaceListW(HDSKSPC DiskSpace, PVOID Reserved1, DWORD Reserved2, UINT Flags){ 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;}
/*********************************************************************** * SetupDuplicateDiskSpaceListA (SETUPAPI.@) */HDSKSPC WINAPI SetupDuplicateDiskSpaceListA(HDSKSPC DiskSpace, PVOID Reserved1, DWORD Reserved2, UINT Flags){ return SetupDuplicateDiskSpaceListW(DiskSpace, Reserved1, Reserved2, Flags);}
/*********************************************************************** * SetupAddInstallSectionToDiskSpaceListA (SETUPAPI.@) */BOOL WINAPI SetupAddInstallSectionToDiskSpaceListA(HDSKSPC DiskSpace, HINF InfHandle, HINF LayoutInfHandle, LPCSTR SectionName, PVOID Reserved1, UINT Reserved2){ FIXME ("Stub\n"); return TRUE;}
/************************************************************************ SetupQuerySpaceRequiredOnDriveW (SETUPAPI.@)*/BOOL WINAPI SetupQuerySpaceRequiredOnDriveW(HDSKSPC DiskSpace, LPCWSTR DriveSpec, LONGLONG *SpaceRequired, PVOID Reserved1, UINT Reserved2){ 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;}
/************************************************************************ SetupQuerySpaceRequiredOnDriveA (SETUPAPI.@)*/BOOL WINAPI SetupQuerySpaceRequiredOnDriveA(HDSKSPC DiskSpace, LPCSTR DriveSpec, LONGLONG *SpaceRequired, PVOID Reserved1, UINT Reserved2){ 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;}
/************************************************************************ SetupDestroyDiskSpaceList (SETUPAPI.@)*/BOOL WINAPI SetupDestroyDiskSpaceList(HDSKSPC DiskSpace){ LPDISKSPACELIST list = (LPDISKSPACELIST)DiskSpace; HeapFree(GetProcessHeap(),0,list); return TRUE;}
/************************************************************************ SetupAddToDiskSpaceListA (SETUPAPI.@)*/BOOL WINAPI SetupAddToDiskSpaceListA(HDSKSPC diskspace, PCSTR targetfile, LONGLONG filesize, UINT operation, PVOID reserved1, UINT reserved2){ FIXME(": stub\n"); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); return FALSE;}
/************************************************************************ SetupAddToDiskSpaceListW (SETUPAPI.@)*/BOOL WINAPI SetupAddToDiskSpaceListW(HDSKSPC diskspace, PCWSTR targetfile, LONGLONG filesize, UINT operation, PVOID reserved1, UINT reserved2){ FIXME(": stub\n"); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); return FALSE;}