driveutils.cpp

#include "driveutils.h"

#ifdef Q_OS_WIN
#include <windows.h>
#include <winioctl.h>
#include <QDir>
#endif

QString driveLetterToPhysicalDrive(const QString &driveLetter) {
#ifdef Q_OS_WIN
    // Expect something like "G:/" or "G:"
    QString driveRoot = driveLetter.left(2); // "G:"
    QString volumePath = QString("\\\\.\\%1").arg(driveRoot);

    HANDLE hDevice = CreateFileW((LPCWSTR)volumePath.utf16(),
                                 0, FILE_SHARE_READ | FILE_SHARE_WRITE,
                                 NULL, OPEN_EXISTING, 0, NULL);

    if (hDevice == INVALID_HANDLE_VALUE) {
        return QString(); // Failed
    }

    STORAGE_DEVICE_NUMBER sdn;
    DWORD bytesReturned = 0;
    bool success = DeviceIoControl(hDevice,
                                   IOCTL_STORAGE_GET_DEVICE_NUMBER,
                                   NULL, 0,
                                   &sdn, sizeof(sdn),
                                   &bytesReturned, NULL);

    CloseHandle(hDevice);

    if (success) {
        return QString("\\\\.\\PhysicalDrive%1").arg(sdn.DeviceNumber);
    } else {
        return QString();
    }
#else
    return driveLetter; // Non-Windows fallback
#endif
}

QString getSystemDriveLetter() {
#ifdef Q_OS_WIN
    wchar_t sysPath[MAX_PATH];
    GetSystemWindowsDirectoryW(sysPath, MAX_PATH);
    QString sysDir = QString::fromWCharArray(sysPath);
    return sysDir.left(3).toUpper();  // Example: "C:/"
#else
    return "/"; // Unix-based fallback
#endif
}

bool isSystemDrive(const QString &path) {
#ifdef Q_OS_WIN
    QString normalized = QDir::toNativeSeparators(path).toUpper();
    QString sysDrive = getSystemDriveLetter();
    return normalized.startsWith(sysDrive);
#else
    return false; // On Linux/macOS, scanning root "/" isn't blocked here
#endif
}