dll/win32/setupapi/diskspace.c

   1 /*
   2  * SetupAPI DiskSpace functions
   3  *
   4  * Copyright 2004 CodeWeavers (Aric Stewart)
   5  *
   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.
  10  *
  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.
  15  *
  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
  19  */
  20
  21 #include "setupapi_private.h"
  22
  23 WINE_DEFAULT_DEBUG_CHANNEL(setupapi);
  24
  25 typedef struct {
  26     WCHAR   lpzName[20];
  27     LONGLONG dwFreeSpace;
  28     LONGLONG dwWantedSpace;
  29 } DRIVE_ENTRY, *LPDRIVE_ENTRY;
  30
  31 typedef struct {
  32     DWORD   dwDriveCount;
  33     DRIVE_ENTRY Drives[26];
  34 } DISKSPACELIST, *LPDISKSPACELIST;
  35
  36
  37 /***********************************************************************
  38  *              SetupCreateDiskSpaceListW  (SETUPAPI.@)
  39  */
  40 HDSKSPC WINAPI SetupCreateDiskSpaceListW(PVOID Reserved1, DWORD Reserved2, UINT Flags)
  41 {
  42     WCHAR drives[255];
  43     DWORD rc;
  44     WCHAR *ptr;
  45     LPDISKSPACELIST list=NULL;
  46
  47     rc = GetLogicalDriveStringsW(255,drives);
  48
  49     if (rc == 0)
  50         return NULL;
  51
  52     list = HeapAlloc(GetProcessHeap(),0,sizeof(DISKSPACELIST));
  53
  54     list->dwDriveCount = 0;
  55
  56     ptr = drives;
  57
  58     while (*ptr)
  59     {
  60         DWORD type = GetDriveTypeW(ptr);
  61         if (type == DRIVE_FIXED)
  62         {
  63             DWORD clusters;
  64             DWORD sectors;
  65             DWORD bytes;
  66             DWORD total;
  67             lstrcpyW(list->Drives[list->dwDriveCount].lpzName,ptr);
  68             GetDiskFreeSpaceW(ptr,&sectors,&bytes,&clusters,&total);
  69             list->Drives[list->dwDriveCount].dwFreeSpace = clusters * sectors *
  70                                                            bytes;
  71             list->Drives[list->dwDriveCount].dwWantedSpace = 0;
  72             list->dwDriveCount++;
  73         }
  74        ptr += lstrlenW(ptr) + 1;
  75     }
  76     return  (HANDLE)list;
  77 }
  78
  79
  80 /***********************************************************************
  81  *              SetupCreateDiskSpaceListA  (SETUPAPI.@)
  82  */
  83 HDSKSPC WINAPI SetupCreateDiskSpaceListA(PVOID Reserved1, DWORD Reserved2, UINT Flags)
  84 {
  85     return SetupCreateDiskSpaceListW( Reserved1, Reserved2, Flags );
  86 }
  87
  88
  89 /***********************************************************************
  90  *              SetupAddInstallSectionToDiskSpaceListA  (SETUPAPI.@)
  91  */
  92 BOOL WINAPI SetupAddInstallSectionToDiskSpaceListA(HDSKSPC DiskSpace,
  93                         HINF InfHandle, HINF LayoutInfHandle,
  94                         LPCSTR SectionName, PVOID Reserved1, UINT Reserved2)
  95 {
  96     FIXME ("Stub\n");
  97     return TRUE;
  98 }
  99
 100 /***********************************************************************
 101 *               SetupQuerySpaceRequiredOnDriveA  (SETUPAPI.@)
 102 */
 103 BOOL WINAPI SetupQuerySpaceRequiredOnDriveA(HDSKSPC DiskSpace,
 104                         LPCSTR DriveSpec, LONGLONG* SpaceRequired,
 105                         PVOID Reserved1, UINT Reserved2)
 106 {
 107     WCHAR driveW[20];
 108     unsigned int i;
 109     LPDISKSPACELIST list = (LPDISKSPACELIST)DiskSpace;
 110     BOOL rc = FALSE;
 111     static const WCHAR bkslsh[]= {'\\',0};
 112
 113     MultiByteToWideChar(CP_ACP,0,DriveSpec,-1,driveW,20);
 114
 115     lstrcatW(driveW,bkslsh);
 116
 117     TRACE("Looking for drive %s\n",debugstr_w(driveW));
 118
 119     for (i = 0; i < list->dwDriveCount; i++)
 120     {
 121         TRACE("checking drive %s\n",debugstr_w(list->Drives[i].lpzName));
 122         if (lstrcmpW(driveW,list->Drives[i].lpzName)==0)
 123         {
 124             rc = TRUE;
 125             *SpaceRequired = list->Drives[i].dwWantedSpace;
 126             break;
 127         }
 128     }
 129
 130     return rc;
 131 }
 132
 133 /***********************************************************************
 134 *               SetupDestroyDiskSpaceList  (SETUPAPI.@)
 135 */
 136 BOOL WINAPI SetupDestroyDiskSpaceList(HDSKSPC DiskSpace)
 137 {
 138     LPDISKSPACELIST list = (LPDISKSPACELIST)DiskSpace;
 139     HeapFree(GetProcessHeap(),0,list);
 140     return TRUE;
 141 }