5efc3669d57bd067f779f745b1302a9eb6fbb577
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"
23 WINE_DEFAULT_DEBUG_CHANNEL(setupapi
);
28 LONGLONG dwWantedSpace
;
29 } DRIVE_ENTRY
, *LPDRIVE_ENTRY
;
33 DRIVE_ENTRY Drives
[26];
34 } DISKSPACELIST
, *LPDISKSPACELIST
;
37 /***********************************************************************
38 * SetupCreateDiskSpaceListW (SETUPAPI.@)
40 HDSKSPC WINAPI
SetupCreateDiskSpaceListW(PVOID Reserved1
, DWORD Reserved2
, UINT Flags
)
45 LPDISKSPACELIST list
=NULL
;
47 rc
= GetLogicalDriveStringsW(255,drives
);
52 list
= HeapAlloc(GetProcessHeap(),0,sizeof(DISKSPACELIST
));
54 list
->dwDriveCount
= 0;
60 DWORD type
= GetDriveTypeW(ptr
);
61 if (type
== DRIVE_FIXED
)
67 lstrcpyW(list
->Drives
[list
->dwDriveCount
].lpzName
,ptr
);
68 GetDiskFreeSpaceW(ptr
,§ors
,&bytes
,&clusters
,&total
);
69 list
->Drives
[list
->dwDriveCount
].dwFreeSpace
= clusters
* sectors
*
71 list
->Drives
[list
->dwDriveCount
].dwWantedSpace
= 0;
74 ptr
+= lstrlenW(ptr
) + 1;
80 /***********************************************************************
81 * SetupCreateDiskSpaceListA (SETUPAPI.@)
83 HDSKSPC WINAPI
SetupCreateDiskSpaceListA(PVOID Reserved1
, DWORD Reserved2
, UINT Flags
)
85 return SetupCreateDiskSpaceListW( Reserved1
, Reserved2
, Flags
);
88 /***********************************************************************
89 * SetupDuplicateDiskSpaceListW (SETUPAPI.@)
91 HDSKSPC WINAPI
SetupDuplicateDiskSpaceListW(HDSKSPC DiskSpace
, PVOID Reserved1
, DWORD Reserved2
, UINT Flags
)
93 DISKSPACELIST
*list_copy
, *list_original
= DiskSpace
;
95 if (Reserved1
|| Reserved2
|| Flags
)
97 SetLastError(ERROR_INVALID_PARAMETER
);
103 SetLastError(ERROR_INVALID_HANDLE
);
107 list_copy
= HeapAlloc(GetProcessHeap(), 0, sizeof(DISKSPACELIST
));
110 SetLastError(ERROR_NOT_ENOUGH_MEMORY
);
114 *list_copy
= *list_original
;
119 /***********************************************************************
120 * SetupDuplicateDiskSpaceListA (SETUPAPI.@)
122 HDSKSPC WINAPI
SetupDuplicateDiskSpaceListA(HDSKSPC DiskSpace
, PVOID Reserved1
, DWORD Reserved2
, UINT Flags
)
124 return SetupDuplicateDiskSpaceListW(DiskSpace
, Reserved1
, Reserved2
, Flags
);
127 /***********************************************************************
128 * SetupAddInstallSectionToDiskSpaceListA (SETUPAPI.@)
130 BOOL WINAPI
SetupAddInstallSectionToDiskSpaceListA(HDSKSPC DiskSpace
,
131 HINF InfHandle
, HINF LayoutInfHandle
,
132 LPCSTR SectionName
, PVOID Reserved1
, UINT Reserved2
)
138 /***********************************************************************
139 * SetupQuerySpaceRequiredOnDriveW (SETUPAPI.@)
141 BOOL WINAPI
SetupQuerySpaceRequiredOnDriveW(HDSKSPC DiskSpace
,
142 LPCWSTR DriveSpec
, LONGLONG
*SpaceRequired
,
143 PVOID Reserved1
, UINT Reserved2
)
147 LPDISKSPACELIST list
= DiskSpace
;
149 static const WCHAR bkslsh
[]= {'\\',0};
153 SetLastError(ERROR_INVALID_HANDLE
);
159 SetLastError(ERROR_INVALID_PARAMETER
);
163 driveW
= HeapAlloc(GetProcessHeap(), 0, (lstrlenW(DriveSpec
) + 2) * sizeof(WCHAR
));
166 SetLastError(ERROR_NOT_ENOUGH_MEMORY
);
170 lstrcpyW(driveW
,DriveSpec
);
171 lstrcatW(driveW
,bkslsh
);
173 TRACE("Looking for drive %s\n",debugstr_w(driveW
));
175 for (i
= 0; i
< list
->dwDriveCount
; i
++)
177 TRACE("checking drive %s\n",debugstr_w(list
->Drives
[i
].lpzName
));
178 if (lstrcmpW(driveW
,list
->Drives
[i
].lpzName
)==0)
181 *SpaceRequired
= list
->Drives
[i
].dwWantedSpace
;
186 HeapFree(GetProcessHeap(), 0, driveW
);
188 if (!rc
) SetLastError(ERROR_INVALID_DRIVE
);
192 /***********************************************************************
193 * SetupQuerySpaceRequiredOnDriveA (SETUPAPI.@)
195 BOOL WINAPI
SetupQuerySpaceRequiredOnDriveA(HDSKSPC DiskSpace
,
196 LPCSTR DriveSpec
, LONGLONG
*SpaceRequired
,
197 PVOID Reserved1
, UINT Reserved2
)
203 /* The parameter validation checks are in a different order from the
204 * Unicode variant of SetupQuerySpaceRequiredOnDrive. */
207 SetLastError(ERROR_INVALID_PARAMETER
);
213 SetLastError(ERROR_INVALID_HANDLE
);
217 len
= MultiByteToWideChar(CP_ACP
, 0, DriveSpec
, -1, NULL
, 0);
219 DriveSpecW
= HeapAlloc(GetProcessHeap(), 0, len
* sizeof(WCHAR
));
222 SetLastError(ERROR_NOT_ENOUGH_MEMORY
);
226 MultiByteToWideChar(CP_ACP
, 0, DriveSpec
, -1, DriveSpecW
, len
);
228 ret
= SetupQuerySpaceRequiredOnDriveW(DiskSpace
, DriveSpecW
, SpaceRequired
,
229 Reserved1
, Reserved2
);
231 HeapFree(GetProcessHeap(), 0, DriveSpecW
);
236 /***********************************************************************
237 * SetupDestroyDiskSpaceList (SETUPAPI.@)
239 BOOL WINAPI
SetupDestroyDiskSpaceList(HDSKSPC DiskSpace
)
241 LPDISKSPACELIST list
= (LPDISKSPACELIST
)DiskSpace
;
242 HeapFree(GetProcessHeap(),0,list
);