Chinaunix首页 | 论坛 | 博客
  • 博客访问: 617361
  • 博文数量: 178
  • 博客积分: 1400
  • 博客等级: 上尉
  • 技术积分: 2162
  • 用 户 组: 普通用户
  • 注册时间: 2009-05-12 20:06
文章分类

全部博文(178)

文章存档

2011年(1)

2010年(94)

2009年(86)

我的朋友

分类:

2010-03-13 10:09:00

Windows下判断U盘插入并取出盘符路径的代码。


#include
#include

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;
}
阅读(3482) | 评论(0) | 转发(2) |
给主人留下些什么吧!~~