2 * SetupAPI DiskSpace functions
4 * Copyright 2004 CodeWeavers (Aric Stewart)
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include "setupapi_private.h"
26 LONGLONG dwWantedSpace
;
27 } DRIVE_ENTRY
, *LPDRIVE_ENTRY
;
31 DRIVE_ENTRY Drives
[26];
32 } DISKSPACELIST
, *LPDISKSPACELIST
;
35 /***********************************************************************
36 * SetupCreateDiskSpaceListW (SETUPAPI.@)
38 HDSKSPC WINAPI
SetupCreateDiskSpaceListW(PVOID Reserved1
, DWORD Reserved2
, UINT Flags
)
43 LPDISKSPACELIST list
=NULL
;
45 TRACE("(%p, %u, 0x%08x)\n", Reserved1
, Reserved2
, Flags
);
47 if (Reserved1
|| Reserved2
|| Flags
& ~SPDSL_IGNORE_DISK
)
49 SetLastError(ERROR_INVALID_PARAMETER
);
53 rc
= GetLogicalDriveStringsW(255,drives
);
58 list
= HeapAlloc(GetProcessHeap(),0,sizeof(DISKSPACELIST
));
60 list
->dwDriveCount
= 0;
66 DWORD type
= GetDriveTypeW(ptr
);
67 if (type
== DRIVE_FIXED
)
73 lstrcpyW(list
->Drives
[list
->dwDriveCount
].lpzName
,ptr
);
74 GetDiskFreeSpaceW(ptr
,§ors
,&bytes
,&clusters
,&total
);
75 list
->Drives
[list
->dwDriveCount
].dwFreeSpace
= clusters
* sectors
*
77 list
->Drives
[list
->dwDriveCount
].dwWantedSpace
= 0;
80 ptr
+= lstrlenW(ptr
) + 1;
86 /***********************************************************************
87 * SetupCreateDiskSpaceListA (SETUPAPI.@)
89 HDSKSPC WINAPI
SetupCreateDiskSpaceListA(PVOID Reserved1
, DWORD Reserved2
, UINT Flags
)
91 return SetupCreateDiskSpaceListW( Reserved1
, Reserved2
, Flags
);
94 /***********************************************************************
95 * SetupDuplicateDiskSpaceListW (SETUPAPI.@)
97 HDSKSPC WINAPI
SetupDuplicateDiskSpaceListW(HDSKSPC DiskSpace
, PVOID Reserved1
, DWORD Reserved2
, UINT Flags
)
99 DISKSPACELIST
*list_copy
, *list_original
= DiskSpace
;
101 if (Reserved1
|| Reserved2
|| Flags
)
103 SetLastError(ERROR_INVALID_PARAMETER
);
109 SetLastError(ERROR_INVALID_HANDLE
);
113 list_copy
= HeapAlloc(GetProcessHeap(), 0, sizeof(DISKSPACELIST
));
116 SetLastError(ERROR_NOT_ENOUGH_MEMORY
);
120 *list_copy
= *list_original
;
125 /***********************************************************************
126 * SetupDuplicateDiskSpaceListA (SETUPAPI.@)
128 HDSKSPC WINAPI
SetupDuplicateDiskSpaceListA(HDSKSPC DiskSpace
, PVOID Reserved1
, DWORD Reserved2
, UINT Flags
)
130 return SetupDuplicateDiskSpaceListW(DiskSpace
, Reserved1
, Reserved2
, Flags
);
133 /***********************************************************************
134 * SetupAddInstallSectionToDiskSpaceListA (SETUPAPI.@)
136 BOOL WINAPI
SetupAddInstallSectionToDiskSpaceListA(HDSKSPC DiskSpace
,
137 HINF InfHandle
, HINF LayoutInfHandle
,
138 LPCSTR SectionName
, PVOID Reserved1
, UINT Reserved2
)
144 /***********************************************************************
145 * SetupQuerySpaceRequiredOnDriveW (SETUPAPI.@)
147 BOOL WINAPI
SetupQuerySpaceRequiredOnDriveW(HDSKSPC DiskSpace
,
148 LPCWSTR DriveSpec
, LONGLONG
*SpaceRequired
,
149 PVOID Reserved1
, UINT Reserved2
)
153 LPDISKSPACELIST list
= DiskSpace
;
155 static const WCHAR bkslsh
[]= {'\\',0};
159 SetLastError(ERROR_INVALID_HANDLE
);
165 SetLastError(ERROR_INVALID_PARAMETER
);
169 driveW
= HeapAlloc(GetProcessHeap(), 0, (lstrlenW(DriveSpec
) + 2) * sizeof(WCHAR
));
172 SetLastError(ERROR_NOT_ENOUGH_MEMORY
);
176 lstrcpyW(driveW
,DriveSpec
);
177 lstrcatW(driveW
,bkslsh
);
179 TRACE("Looking for drive %s\n",debugstr_w(driveW
));
181 for (i
= 0; i
< list
->dwDriveCount
; i
++)
183 TRACE("checking drive %s\n",debugstr_w(list
->Drives
[i
].lpzName
));
184 if (lstrcmpW(driveW
,list
->Drives
[i
].lpzName
)==0)
187 *SpaceRequired
= list
->Drives
[i
].dwWantedSpace
;
192 HeapFree(GetProcessHeap(), 0, driveW
);
194 if (!rc
) SetLastError(ERROR_INVALID_DRIVE
);
198 /***********************************************************************
199 * SetupQuerySpaceRequiredOnDriveA (SETUPAPI.@)
201 BOOL WINAPI
SetupQuerySpaceRequiredOnDriveA(HDSKSPC DiskSpace
,
202 LPCSTR DriveSpec
, LONGLONG
*SpaceRequired
,
203 PVOID Reserved1
, UINT Reserved2
)
209 /* The parameter validation checks are in a different order from the
210 * Unicode variant of SetupQuerySpaceRequiredOnDrive. */
213 SetLastError(ERROR_INVALID_PARAMETER
);
219 SetLastError(ERROR_INVALID_HANDLE
);
223 len
= MultiByteToWideChar(CP_ACP
, 0, DriveSpec
, -1, NULL
, 0);
225 DriveSpecW
= HeapAlloc(GetProcessHeap(), 0, len
* sizeof(WCHAR
));
228 SetLastError(ERROR_NOT_ENOUGH_MEMORY
);
232 MultiByteToWideChar(CP_ACP
, 0, DriveSpec
, -1, DriveSpecW
, len
);
234 ret
= SetupQuerySpaceRequiredOnDriveW(DiskSpace
, DriveSpecW
, SpaceRequired
,
235 Reserved1
, Reserved2
);
237 HeapFree(GetProcessHeap(), 0, DriveSpecW
);
242 /***********************************************************************
243 * SetupDestroyDiskSpaceList (SETUPAPI.@)
245 BOOL WINAPI
SetupDestroyDiskSpaceList(HDSKSPC DiskSpace
)
247 LPDISKSPACELIST list
= (LPDISKSPACELIST
)DiskSpace
;
248 HeapFree(GetProcessHeap(),0,list
);