tor-browser

The Tor Browser
git clone https://git.dasho.dev/tor-browser.git
Log | Files | Refs | README | LICENSE

FileSystem.cpp (3878B)

      1 // Windows/FileSystem.cpp
      2 
      3 #include "StdAfx.h"
      4 
      5 #ifndef UNDER_CE
      6 
      7 #ifndef _UNICODE
      8 #include "../Common/StringConvert.h"
      9 #endif
     10 
     11 #include "FileSystem.h"
     12 #include "Defs.h"
     13 
     14 #ifndef _UNICODE
     15 extern bool g_IsNT;
     16 #endif
     17 
     18 namespace NWindows {
     19 namespace NFile {
     20 namespace NSystem {
     21 
     22 bool MyGetVolumeInformation(
     23    CFSTR rootPath,
     24    UString &volumeName,
     25    LPDWORD volumeSerialNumber,
     26    LPDWORD maximumComponentLength,
     27    LPDWORD fileSystemFlags,
     28    UString &fileSystemName)
     29 {
     30  BOOL res;
     31  #ifndef _UNICODE
     32  if (!g_IsNT)
     33  {
     34    TCHAR v[MAX_PATH + 2]; v[0] = 0;
     35    TCHAR f[MAX_PATH + 2]; f[0] = 0;
     36    res = GetVolumeInformation(fs2fas(rootPath),
     37        v, MAX_PATH,
     38        volumeSerialNumber, maximumComponentLength, fileSystemFlags,
     39        f, MAX_PATH);
     40    volumeName = MultiByteToUnicodeString(v);
     41    fileSystemName = MultiByteToUnicodeString(f);
     42  }
     43  else
     44  #endif
     45  {
     46    WCHAR v[MAX_PATH + 2]; v[0] = 0;
     47    WCHAR f[MAX_PATH + 2]; f[0] = 0;
     48    res = GetVolumeInformationW(fs2us(rootPath),
     49        v, MAX_PATH,
     50        volumeSerialNumber, maximumComponentLength, fileSystemFlags,
     51        f, MAX_PATH);
     52    volumeName = v;
     53    fileSystemName = f;
     54  }
     55  return BOOLToBool(res);
     56 }
     57 
     58 UINT MyGetDriveType(CFSTR pathName)
     59 {
     60  #ifndef _UNICODE
     61  if (!g_IsNT)
     62  {
     63    return GetDriveType(fs2fas(pathName));
     64  }
     65  else
     66  #endif
     67  {
     68    return GetDriveTypeW(fs2us(pathName));
     69  }
     70 }
     71 
     72 typedef BOOL (WINAPI * GetDiskFreeSpaceExA_Pointer)(
     73  LPCSTR lpDirectoryName,                  // directory name
     74  PULARGE_INTEGER lpFreeBytesAvailable,    // bytes available to caller
     75  PULARGE_INTEGER lpTotalNumberOfBytes,    // bytes on disk
     76  PULARGE_INTEGER lpTotalNumberOfFreeBytes // free bytes on disk
     77 );
     78 
     79 typedef BOOL (WINAPI * GetDiskFreeSpaceExW_Pointer)(
     80  LPCWSTR lpDirectoryName,                 // directory name
     81  PULARGE_INTEGER lpFreeBytesAvailable,    // bytes available to caller
     82  PULARGE_INTEGER lpTotalNumberOfBytes,    // bytes on disk
     83  PULARGE_INTEGER lpTotalNumberOfFreeBytes // free bytes on disk
     84 );
     85 
     86 bool MyGetDiskFreeSpace(CFSTR rootPath, UInt64 &clusterSize, UInt64 &totalSize, UInt64 &freeSize)
     87 {
     88  DWORD numSectorsPerCluster, bytesPerSector, numFreeClusters, numClusters;
     89  bool sizeIsDetected = false;
     90  #ifndef _UNICODE
     91  if (!g_IsNT)
     92  {
     93    GetDiskFreeSpaceExA_Pointer pGetDiskFreeSpaceEx = (GetDiskFreeSpaceExA_Pointer)GetProcAddress(
     94        GetModuleHandle(TEXT("kernel32.dll")), "GetDiskFreeSpaceExA");
     95    if (pGetDiskFreeSpaceEx)
     96    {
     97      ULARGE_INTEGER freeBytesToCaller2, totalSize2, freeSize2;
     98      sizeIsDetected = BOOLToBool(pGetDiskFreeSpaceEx(fs2fas(rootPath), &freeBytesToCaller2, &totalSize2, &freeSize2));
     99      totalSize = totalSize2.QuadPart;
    100      freeSize = freeSize2.QuadPart;
    101    }
    102    if (!::GetDiskFreeSpace(fs2fas(rootPath), &numSectorsPerCluster, &bytesPerSector, &numFreeClusters, &numClusters))
    103      return false;
    104  }
    105  else
    106  #endif
    107  {
    108    GetDiskFreeSpaceExW_Pointer pGetDiskFreeSpaceEx = (GetDiskFreeSpaceExW_Pointer)GetProcAddress(
    109        GetModuleHandle(TEXT("kernel32.dll")), "GetDiskFreeSpaceExW");
    110    if (pGetDiskFreeSpaceEx)
    111    {
    112      ULARGE_INTEGER freeBytesToCaller2, totalSize2, freeSize2;
    113      sizeIsDetected = BOOLToBool(pGetDiskFreeSpaceEx(fs2us(rootPath), &freeBytesToCaller2, &totalSize2, &freeSize2));
    114      totalSize = totalSize2.QuadPart;
    115      freeSize = freeSize2.QuadPart;
    116    }
    117    if (!::GetDiskFreeSpaceW(fs2us(rootPath), &numSectorsPerCluster, &bytesPerSector, &numFreeClusters, &numClusters))
    118      return false;
    119  }
    120  clusterSize = (UInt64)bytesPerSector * (UInt64)numSectorsPerCluster;
    121  if (!sizeIsDetected)
    122  {
    123    totalSize = clusterSize * (UInt64)numClusters;
    124    freeSize = clusterSize * (UInt64)numFreeClusters;
    125  }
    126  return true;
    127 }
    128 
    129 }}}
    130 
    131 #endif