161 lines
2.6 KiB
C++
161 lines
2.6 KiB
C++
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 1998 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
volumes.cpp
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
SIS Groveler SIS volume mount classes
|
||
|
|
|
||
|
|
Authors:
|
||
|
|
|
||
|
|
John Douceur, 1998
|
||
|
|
|
||
|
|
Environment:
|
||
|
|
|
||
|
|
User Mode
|
||
|
|
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
#include "all.hxx"
|
||
|
|
|
||
|
|
Volumes::Volumes()
|
||
|
|
{
|
||
|
|
volume_handle = INVALID_HANDLE_VALUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
Volumes::~Volumes()
|
||
|
|
{
|
||
|
|
if (volume_handle != INVALID_HANDLE_VALUE)
|
||
|
|
{
|
||
|
|
FindVolumeClose(volume_handle);
|
||
|
|
volume_handle = INVALID_HANDLE_VALUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
Volumes::find(
|
||
|
|
_TCHAR *volume_name,
|
||
|
|
DWORD length)
|
||
|
|
{
|
||
|
|
DWORD error_code = NO_ERROR;
|
||
|
|
if (volume_handle == INVALID_HANDLE_VALUE)
|
||
|
|
{
|
||
|
|
volume_handle = FindFirstVolume(volume_name, length);
|
||
|
|
if (volume_handle == INVALID_HANDLE_VALUE)
|
||
|
|
{
|
||
|
|
error_code = GetLastError();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
BOOL ok = FindNextVolume(volume_handle, volume_name, length);
|
||
|
|
if (!ok)
|
||
|
|
{
|
||
|
|
error_code = GetLastError();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return error_code;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
VolumeMountPoints::VolumeMountPoints(
|
||
|
|
_TCHAR *volume_name)
|
||
|
|
{
|
||
|
|
this->volume_name = volume_name;
|
||
|
|
drive_mask = 0;
|
||
|
|
drive_letter = 0;
|
||
|
|
volume_handle = INVALID_HANDLE_VALUE;
|
||
|
|
if (volume_name == 0)
|
||
|
|
{
|
||
|
|
drive_mask = GetLogicalDrives();
|
||
|
|
_TCHAR drive_name[4];
|
||
|
|
drive_name[0] = _T('a');
|
||
|
|
drive_name[1] = _T(':');
|
||
|
|
drive_name[2] = _T('\\');
|
||
|
|
drive_name[3] = _T('\0');
|
||
|
|
DWORD drive_bit = 1;
|
||
|
|
while (drive_name[0] <= _T('z'))
|
||
|
|
{
|
||
|
|
if ((drive_mask & drive_bit) != 0)
|
||
|
|
{
|
||
|
|
UINT drive_type = GetDriveType(drive_name);
|
||
|
|
if (drive_type != DRIVE_FIXED)
|
||
|
|
{
|
||
|
|
drive_mask &= ~drive_bit;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
drive_bit <<= 1;
|
||
|
|
drive_name[0]++;
|
||
|
|
}
|
||
|
|
drive_mask <<= 1;
|
||
|
|
drive_letter = _T('a') - 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
VolumeMountPoints::~VolumeMountPoints()
|
||
|
|
{
|
||
|
|
if (volume_handle != INVALID_HANDLE_VALUE)
|
||
|
|
{
|
||
|
|
FindVolumeMountPointClose(volume_handle);
|
||
|
|
volume_handle = INVALID_HANDLE_VALUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
VolumeMountPoints::find(
|
||
|
|
_TCHAR *volume_mount_point,
|
||
|
|
DWORD length)
|
||
|
|
{
|
||
|
|
if (volume_name == 0)
|
||
|
|
{
|
||
|
|
if ((drive_mask & 0xfffffffe) == 0)
|
||
|
|
{
|
||
|
|
return ERROR_NO_MORE_FILES;
|
||
|
|
}
|
||
|
|
if (length < 4)
|
||
|
|
{
|
||
|
|
return ERROR_FILENAME_EXCED_RANGE;
|
||
|
|
}
|
||
|
|
do
|
||
|
|
{
|
||
|
|
drive_mask >>= 1;
|
||
|
|
drive_letter++;
|
||
|
|
} while ((drive_mask & 1) == 0);
|
||
|
|
volume_mount_point[0] = drive_letter;
|
||
|
|
volume_mount_point[1] = _T(':');
|
||
|
|
volume_mount_point[2] = _T('\\');
|
||
|
|
volume_mount_point[3] = _T('\0');
|
||
|
|
return NO_ERROR;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
if (volume_handle == INVALID_HANDLE_VALUE)
|
||
|
|
{
|
||
|
|
volume_handle = FindFirstVolumeMountPoint(
|
||
|
|
volume_name, volume_mount_point, length);
|
||
|
|
if (volume_handle == INVALID_HANDLE_VALUE)
|
||
|
|
{
|
||
|
|
return GetLastError();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
BOOL ok = FindNextVolumeMountPoint(
|
||
|
|
volume_handle, volume_mount_point, length);
|
||
|
|
if (!ok)
|
||
|
|
{
|
||
|
|
return GetLastError();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return NO_ERROR;
|
||
|
|
}
|
||
|
|
}
|