Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-04-23 17:03:12 -05:00
Code Issues Packages Projects Releases Wiki Activity

fsptool: lsvol

Browse Source
This commit is contained in:
Bill Zissimopoulos 2017-07-19 15:22:43 -07:00
parent af37424ecc
commit 690d3e4c8e
1 changed files with 58 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
src/fsptool/fsptool.c
View File

@ -68,23 +68,29 @@ static void usage(void)
PROGNAME); PROGNAME);
} }
static NTSTATUS lsvol_dev(PWSTR DeviceName) static NTSTATUS FspToolGetVolumeList(PWSTR DeviceName,
PWCHAR *PVolumeListBuf, PSIZE_T PVolumeListSize)
{ {
NTSTATUS Result; NTSTATUS Result;
PWCHAR VolumeListBuf, VolumeListBufEnd; PWCHAR VolumeListBuf;
SIZE_T VolumeListSize; SIZE_T VolumeListSize;
*PVolumeListBuf = 0;
*PVolumeListSize = 0;
for (VolumeListSize = 1024;; VolumeListSize *= 2) for (VolumeListSize = 1024;; VolumeListSize *= 2)
{ {
VolumeListBuf = MemAlloc(VolumeListSize); VolumeListBuf = MemAlloc(VolumeListSize);
if (0 == VolumeListBuf) if (0 == VolumeListBuf)
return STATUS_INSUFFICIENT_RESOURCES; return STATUS_INSUFFICIENT_RESOURCES;
Result = FspFsctlGetVolumeList(DeviceName, VolumeListBuf, &VolumeListSize); Result = FspFsctlGetVolumeList(DeviceName,
VolumeListBuf, &VolumeListSize);
if (NT_SUCCESS(Result)) if (NT_SUCCESS(Result))
{ {
VolumeListBufEnd = (PVOID)((PUINT8)VolumeListBuf + VolumeListSize); *PVolumeListBuf = VolumeListBuf;
break; *PVolumeListSize = VolumeListSize;
return Result;
} }
MemFree(VolumeListBuf); MemFree(VolumeListBuf);
@ -92,11 +98,57 @@ static NTSTATUS lsvol_dev(PWSTR DeviceName)
if (STATUS_BUFFER_TOO_SMALL != Result) if (STATUS_BUFFER_TOO_SMALL != Result)
return Result; return Result;
} }
}
static WCHAR FspToolGetDriveLetter(PDWORD PLogicalDrives, PWSTR VolumeName)
{
WCHAR VolumeNameBuf[MAX_PATH];
WCHAR LocalNameBuf[3];
WCHAR Drive;
if (0 == *PLogicalDrives)
return 0;
LocalNameBuf[1] = L':';
LocalNameBuf[2] = L'\0';
for (Drive = 'Z'; 'A' <= Drive; Drive--)
if (0 != (*PLogicalDrives & (1 << (Drive - 'A'))))
{
LocalNameBuf[0] = Drive;
if (QueryDosDeviceW(LocalNameBuf, VolumeNameBuf, sizeof VolumeNameBuf / sizeof(WCHAR)))
{
if (0 == invariant_wcscmp(VolumeNameBuf, VolumeName))
{
*PLogicalDrives &= ~(1 << (Drive - 'A'));
return Drive;
}
}
}
return 0;
}
static NTSTATUS lsvol_dev(PWSTR DeviceName)
{
NTSTATUS Result;
PWCHAR VolumeListBuf, VolumeListBufEnd;
SIZE_T VolumeListSize;
DWORD LogicalDrives;
WCHAR Drive[3] = L"\0:";
Result = FspToolGetVolumeList(DeviceName, &VolumeListBuf, &VolumeListSize);
if (!NT_SUCCESS(Result))
return Result;
VolumeListBufEnd = (PVOID)((PUINT8)VolumeListBuf + VolumeListSize);
LogicalDrives = GetLogicalDrives();
for (PWCHAR P = VolumeListBuf, VolumeName = P; VolumeListBufEnd > P; P++) for (PWCHAR P = VolumeListBuf, VolumeName = P; VolumeListBufEnd > P; P++)
if (L'\0' == *P) if (L'\0' == *P)
{ {
info("%S", VolumeName); Drive[0] = FspToolGetDriveLetter(&LogicalDrives, VolumeName);
info("%-4S%S", Drive[0] ? Drive : L"", VolumeName);
VolumeName = P + 1; VolumeName = P + 1;
} }