CPP: Windows: FindFirstVolumeW
Retrieves the name of a volume on a computer. FindFirstVolume is used to begin scanning the volumes of a computer. See line 85 for sample usage
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#include <windows.h> | |
#include <stdio.h> | |
void DisplayVolumePaths( | |
__in PWCHAR VolumeName | |
) | |
{ | |
DWORD CharCount = MAX_PATH + 1; | |
PWCHAR Names = NULL; | |
PWCHAR NameIdx = NULL; | |
BOOL Success = FALSE; | |
for (;;) | |
{ | |
// | |
// Allocate a buffer to hold the paths. | |
Names = (PWCHAR) new BYTE[CharCount * sizeof(WCHAR)]; | |
if (!Names) | |
{ | |
// | |
// If memory can't be allocated, return. | |
return; | |
} | |
// | |
// Obtain all of the paths | |
// for this volume. | |
Success = GetVolumePathNamesForVolumeNameW( | |
VolumeName, Names, CharCount, &CharCount | |
); | |
if (Success) | |
{ | |
break; | |
} | |
if (GetLastError() != ERROR_MORE_DATA) | |
{ | |
break; | |
} | |
// | |
// Try again with the | |
// new suggested size. | |
delete[] Names; | |
Names = NULL; | |
} | |
if (Success) | |
{ | |
// | |
// Display the various paths. | |
for (NameIdx = Names; | |
NameIdx[0] != L'\0'; | |
NameIdx += wcslen(NameIdx) + 1) | |
{ | |
wprintf(L" %s", NameIdx); | |
} | |
wprintf(L"\n"); | |
} | |
if (Names != NULL) | |
{ | |
delete[] Names; | |
Names = NULL; | |
} | |
return; | |
} | |
void __cdecl wmain(void) | |
{ | |
DWORD CharCount = 0; | |
WCHAR DeviceName[MAX_PATH] = L""; | |
DWORD Error = ERROR_SUCCESS; | |
HANDLE FindHandle = INVALID_HANDLE_VALUE; | |
BOOL Found = FALSE; | |
size_t Index = 0; | |
BOOL Success = FALSE; | |
WCHAR VolumeName[MAX_PATH] = L""; | |
// | |
// Enumerate all volumes in the system. | |
FindHandle = FindFirstVolumeW(VolumeName, ARRAYSIZE(VolumeName)); | |
if (FindHandle == INVALID_HANDLE_VALUE) | |
{ | |
Error = GetLastError(); | |
wprintf(L"FindFirstVolumeW failed with error code %d\n", Error); | |
return; | |
} | |
for (;;) | |
{ | |
// | |
// Skip the \\?\ prefix and remove the trailing backslash. | |
Index = wcslen(VolumeName) - 1; | |
if (VolumeName[0] != L'\\' || | |
VolumeName[1] != L'\\' || | |
VolumeName[2] != L'?' || | |
VolumeName[3] != L'\\' || | |
VolumeName[Index] != L'\\') | |
{ | |
Error = ERROR_BAD_PATHNAME; | |
wprintf(L"FindFirstVolumeW/FindNextVolumeW returned a bad path: %s\n", VolumeName); | |
break; | |
} | |
// | |
// QueryDosDeviceW does not allow a trailing backslash, | |
// so temporarily remove it. | |
VolumeName[Index] = L'\0'; | |
CharCount = QueryDosDeviceW(&VolumeName[4], DeviceName, ARRAYSIZE(DeviceName)); | |
VolumeName[Index] = L'\\'; | |
if (CharCount == 0) | |
{ | |
Error = GetLastError(); | |
wprintf(L"QueryDosDeviceW failed with error code %d\n", Error); | |
break; | |
} | |
wprintf(L"\nFound a device:\n %s", DeviceName); | |
wprintf(L"\nVolume name: %s", VolumeName); | |
wprintf(L"\nPaths:"); | |
DisplayVolumePaths(VolumeName); | |
// | |
// Move on to the next volume. | |
Success = FindNextVolumeW(FindHandle, VolumeName, ARRAYSIZE(VolumeName)); | |
if (!Success) | |
{ | |
Error = GetLastError(); | |
if (Error != ERROR_NO_MORE_FILES) | |
{ | |
wprintf(L"FindNextVolumeW failed with error code %d\n", Error); | |
break; | |
} | |
// | |
// Finished iterating | |
// through all the volumes. | |
Error = ERROR_SUCCESS; | |
break; | |
} | |
} | |
FindVolumeClose(FindHandle); | |
FindHandle = INVALID_HANDLE_VALUE; | |
return; | |
} |
Notes:
- Tested using Visual Studio 2022 (17.2.6) C++ Windows Console template
Reference(s):
https://docs.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-findfirstvolumew
No comments:
Post a Comment