#include <windows.h> 
#include <dbt.h> 
 
char FirstDriveFromMask(ULONG unitmask) 
{ 
    char i; 
    for (i = 0; i < 26; ++i) 
    { 
        if (unitmask & 0x1) 
            break; 
        unitmask = unitmask >> 1; 
    } 
    return (i + 'A'); 
} 
 
bool Application::winEventFilter(MSG *msg, long *result) 
{ 
    bool ret = false; 
    if (msg->message != WM_DEVICECHANGE) 
        return ret; 
 
    PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)msg->lParam; 
    static int iCnt = 0; 
    char szDrivePath[3]; 
 
    switch (msg->wParam) 
    { 
    case DBT_DEVICEARRIVAL: 
        if (lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME) 
        { 
            PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb; 
            //if (lpdbv->dbcv_flags & DBTF_MEDIA) 
            { 
                sprintf(szDrivePath, "%c:\\", 
                        FirstDriveFromMask(lpdbv->dbcv_unitmask)); 
                const wchar_t *path = QString(szDrivePath).toStdWString().c_str(); 
                if (szDrivePath[0] > 'B' 
                    && GetDriveType(path) == DRIVE_REMOVABLE) 
                { 
                    iCnt++; 
                    if (iCnt%3 == 0) 
                    { 
                        qDebug() << "Incoming: " << szDrivePath; 
                        ret = true; 
                    } 
                } 
            } 
        } 
        break; 
    case DBT_DEVICEREMOVECOMPLETE: 
        break; 
    default:; 
    } 
    return ret; 
}