#include<DeviceInfos.h>

QString DisplayVolumePaths( 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 "memory can't be allocated";
        }

        //
        //  Obtain all of the paths
        //  for this volume.
        Success = GetVolumePathNamesForVolumeNameW(
            VolumeName, Names, CharCount, &CharCount
            );

        if ( Success )
        {
            qDebug()<<"GetVolumePathNamesForVolumeNameW succeed count"<<CharCount;
            break;
        }

        qDebug()<<"GetVolumePathNamesForVolumeNameW failed count"<<CharCount<<" error "<<GetLastError();
        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'';
              NameIdx += wcslen(NameIdx) + 1 )
        {
            qDebug()<<"  "<<QString::fromWCharArray(NameIdx);
            return QString::fromWCharArray(NameIdx);
        }
        qDebug()<<"n";
    }

    if ( Names != NULL )
    {
        delete [] Names;
        Names = NULL;
    }

    return "s";
}

QMap<QString,DeviceVolumeName> __cdecl GetDeviceVolumeInfos(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"";
    QMap<QString,DeviceVolumeName> mapDeviceInfos=QMap<QString,DeviceVolumeName>();


    //
    //  Enumerate all volumes in the system.
    FindHandle = FindFirstVolumeW(VolumeName, ARRAYSIZE(VolumeName));

    if (FindHandle == INVALID_HANDLE_VALUE)
    {
        Error = GetLastError();
        qDebug()<<"FindFirstVolumeW failed with error code %dn"<< QString::number( Error);

        return mapDeviceInfos;
    }

    for (;;)
    {
        DeviceVolumeName DVName;
        //
        //  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;
            qDebug()<<"FindFirstVolumeW/FindNextVolumeW returned a bad path: %sn"<<QString::fromWCharArray( VolumeName);
            break;
        }

        //
        //  QueryDosDeviceW does not allow a trailing backslash,
        //  so temporarily remove it.
        VolumeName[Index] = L'';

        CharCount = QueryDosDeviceW(&VolumeName[4], DeviceName, ARRAYSIZE(DeviceName));

        VolumeName[Index] = L'\';

        if ( CharCount == 0 )
        {
            Error = GetLastError();
            qDebug()<<L"QueryDosDeviceW failed with error code n"<<QString::number(  Error);
            break;
        }


        qDebug()<<"Found a device: "<<QString::fromWCharArray( DeviceName);
        qDebug()<<"Volume name: "<< QString::fromWCharArray(VolumeName);
        qDebug()<<"Paths:  ";
        QString VolPath=DisplayVolumePaths(VolumeName);
        qDebug()<<"VolPath.at(0)"<<VolPath.at(0);


        DVName.DeviceN=QString::fromWCharArray(DeviceName);
        DVName.VolumeN=QString::fromWCharArray(VolumeName);
        mapDeviceInfos.insert(VolPath.at(0),DVName);




        //
        //  Move on to the next volume.
        Success = FindNextVolumeW(FindHandle, VolumeName, ARRAYSIZE(VolumeName));

        if ( !Success )
        {
            Error = GetLastError();

            if (Error != ERROR_NO_MORE_FILES)
            {
                qDebug()<<"FindNextVolumeW failed with error code n"<<QString::number( Error);
                break;
            }

            //
            //  Finished iterating
            //  through all the volumes.
            Error = ERROR_SUCCESS;
            break;
        }
    }

    FindVolumeClose(FindHandle);
    FindHandle = INVALID_HANDLE_VALUE;

    return mapDeviceInfos;
}

Found a device:  "DeviceHarddiskVolume1"
Volume name:  "\?Volume{af2c546d-c067-11e1-9f1c-806e6f6e6963}"
Paths:
GetVolumePathNamesForVolumeNameW succeed count 5
   "C:"
VolPath.at(0) 'C'
Found a device:  "DeviceHarddiskVolume2"
Volume name:  "\?Volume{af2c546e-c067-11e1-9f1c-806e6f6e6963}"
Paths:
GetVolumePathNamesForVolumeNameW succeed count 5
   "D:"
VolPath.at(0) 'D'
Found a device:  "Device{A5E1065D-0AD1-48ED-8457-A80B2D9B6FE2}#0#0"
Volume name:  "\?Volume{4dee513c-0b96-11e2-9b3b-d4bed9a624c5}"
Paths:
GetVolumePathNamesForVolumeNameW failed count 261  error  2
VolPath.at(0) 's'
Found a device:  "DeviceCdRom0"
Volume name:  "\?Volume{af2c5471-c067-11e1-9f1c-806e6f6e6963}"
Paths:
GetVolumePathNamesForVolumeNameW succeed count 5
   "E:"
VolPath.at(0) 'E'

Found a device:  "Device{A5E1065D-0AD1-48ED-8457-A80B2D9B6FE2}#0#0"
    Volume name:  "\?Volume{4dee513c-0b96-11e2-9b3b-d4bed9a624c5}"
    Paths:
    GetVolumePathNamesForVolumeNameW failed count 261  error  2