Saving console output to an int array - c++

Question

I want to run this command:
wmic diskdrive get index
and store the indexes in an int array.
Could you help me?

Answer 1

I found this idea elsewhere on this site: try calling CreateFile() for all possible indexes of "\\\\.\\PhysicalDriveN".
There are claims that Windows only support up to 16 physical drives, and that those indexes are sequential - you need to verify that. If there might be more - increase the limit. And if they ARE in fact sequential - you may break out of the loop after first ERROR_FILE_NOT_FOUND.
Here is what I got:
#include <Windows.h>
#include <iostream>
#include <sstream>
#include <vector>
int main() {
std::vector<int> indexes;
for (int index = 0; index < 16; ++index) {
std::stringstream ss;
ss << "\\\\.\\PhysicalDrive" << index;
//std::cout << ss.str() << std::endl;
HANDLE disk = CreateFileA(ss.str().c_str(),
GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL,
NULL);
if (INVALID_HANDLE_VALUE == disk) {
DWORD err = GetLastError();
if (ERROR_ACCESS_DENIED == err) {
// No access - you need Admin rights to open this file
// But the drive is there!
//std::cout << "ERROR_ACCESS_DENIED" << std::endl;
indexes.push_back(index);
}
else if (ERROR_FILE_NOT_FOUND == err) {
// No such drive
//std::cout << "ERROR_FILE_NOT_FOUND" << std::endl;
}
else {
// TODO: research - what else could've happened?
std::cout << err << std::endl;
}
}
else {
indexes.push_back(index);
CloseHandle(disk);
}
}
std::cout << "Found indexes:" << std::endl;
for (auto index : indexes) {
std::cout << index << std::endl;
}
}