Windows下判断U盘插入并取出盘符路径的代码。 [复制链接]

#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;
}

正在做U盘的识别，正好用得上！
我现在的项目是在做一个特定U盘设备的识别与操作。在U盘插上后先判断是不是我需要的设备，然后对U盘中的文件进行操作。
现在利用LibUSB-Win32能够识别出U盘设备了，但是无法与盘符相对应，现在一直没有解决。
目前我改变了思路，首先对是否有U盘插入进行响应，得到U盘的盘符，然后在判断该设备是否是我想要的设备。
不知这个方法是否可行，还望指点

发现了先获取盘符再判断设备的问题了，如果程序启动的时候，就有多个u盘已经插在了计算机上，这种情况下就没法判断哪个u盘是我想要的设备了。
问题又回到了如何获取u盘盘符的问题了

U盘上放个很诡异的保护文件，读取下判断可否？比如就写入一行复杂点儿的字符串。能读到该字符串的便是想要的U盘了吧~~~
（或者修改U盘备注名？）

强贴留名～

这种消息感觉需要hook

mark 也许以后能用上

谢谢共享