Win32程序与Qt程序对U盘的探测实践

最新推荐文章于 2023-06-27 16:49:01 发布

Joven_xxx

最新推荐文章于 2023-06-27 16:49:01 发布

阅读量968

点赞数 1

分类专栏： Qt/QML的实践积累文章标签： u盘 win32 QtU盘探测

本文链接：https://blog.csdn.net/Shado_walker/article/details/78768400

版权

Qt/QML的实践积累专栏收录该内容

42 篇文章 10 订阅

订阅专栏

在Windows操作系统下，所有的事件都是以消息为驱动的，因此，当我们插U盘和拔U盘时，也是触发了Windows的消息，我们对此消息进行监听并接收该消息，就可以探测对U盘的插拔。

—

Win32程序

对于消息的监听，熟悉Windows中Win32程序原理的你，一定非常熟悉，其六大步骤如下：

1. 声明消息类（WNDCLASS）

2. 注册消息类（RegisterClass）

3. 创建窗口（CreateWindow）

4. 获取消息队列（GetMessage）

5. 消息传送（TranslateMessage）

6. 消息派发（DispatchMessage）

在声明消息类WNDCLASS时，设置消息回调wc.lpfnWndProc为我们自己的函数，即可以接收到系统的消息到该函数中。

以上过程，在main函数中的源码如下：

int _tmain(int argc, _TCHAR* argv[])
{
	isNumber("5.8.");

	WNDCLASS wc;
	ZeroMemory(&wc, sizeof(wc));
	wc.lpszClassName = TEXT("myusbmsg");
	wc.lpfnWndProc = WndProc;

	RegisterClass(&wc);
	HWND h = CreateWindow(TEXT("myusbmsg"), TEXT(""), 0, 0, 0, 0, 0,
		0, 0, GetModuleHandle(0), 0);
	MSG msg;
	while (GetMessage(&msg, 0, 0, 0) > 0) {
		TranslateMessage(&msg);
		DispatchMessage(&msg);
	}

	return 0;
}

要接收U盘的插拔消息，就可以在我们设置的回调函数中判断，假如我们的回调函数声明如下：

LRESULT CALLBACK WndProc(HWND h, UINT msg, WPARAM wp, LPARAM lp);

首先得判断收到的消息msg是不是U盘的插拔，因为Windows的消息非常多，因此得屏蔽掉无关的消息，只处理我们想接受的消息。U盘的插拔所触发的消息消息为 WM_DEVICECHANGE 消息，不管是插入U盘，韩式拔掉U盘，都会触发 WM_DEVICECHANGE 消息。

Windows的消息回调函数还会携带WPARAM wp, LPARAM lp连个参数，可以帮助我们判断U盘是插入还是拔出，以及U盘插入后是

哪个盘符，其详细解释如下：

wp的值为 DBT_DEVICEARRIVAL 时，表示U盘插入了。其值为 DBT_DEVICEREMOVECOMPLETE 时，表示U盘拔出了。
将lp进行转换 DEV_BROADCAST_VOLUME* p = (DEV_BROADCAST_VOLUME*)lp;得到DEV_BROADCAST_VOLUME的指针p,当 p->dbcv_devicetype == DBT_DEVTYP_VOLUME 时，表示卷标发生变化，即可以知道U盘发生了插入或拔出。
通过 p->dbcv_unitmask 可以根据其位操作中1的标识位判断具体是哪个盘发生了插入或拔出，其判断函数如下：

char FirstDriveFromMask(ULONG unitmask)

{

char i;

for (i = 0; i < 26; ++i)

{

if (unitmask & 0x1)

break;

unitmask = unitmask >> 1;

}

return (i + 'A');

}

通过代码一目了然可知，当p->dbcv_unitmask为0x1时为A盘，为0x10时为B盘，为0x100时为C盘，为0x1000时为D盘...以此类推。

以下奉上接收消息的函数，以及U盘插拔判断的逻辑，如下仅供参考：

LRESULT CALLBACK WndProc(HWND h, UINT msg, WPARAM wp, LPARAM lp)

{

/*

* system uses the WM_DEVICECHANGE message to notify device is changed(USB is inserted or removed).

*/

if (msg == WM_DEVICECHANGE) {

printf("注意---Msg-Change===wp=%d---lp=%d\n", wp, lp);

/*

* #define DBT_DEVNODES_CHANGED 0x0007

* Message = WM_DEVICECHANGE

* wParam = DBT_DEVNODES_CHANGED

* lParam = 0

*

* send when configmg finished a process tree batch. Some devnodes

* may have been added or removed. This is used by ring3 people which

* need to be refreshed whenever any devnode changed occur (like

* device manager). People specific to certain devices should use

* DBT_DEVICE* instead.

*

* The upon windows-msg-param is received when device list is changed, but this time

* system don't know which device is add or remove, after second signal received, the

* below wparam can judge U-Disk or other device.

*/

/*

* The below wparam DBT_DEVICEARRIVAL(system detected a new device) & DBT_DEVICEREMOVECOMPLETE(device is gone)

* can just detect U-Disk inserted or removed(VOLUME List Changed), other device didn't show in volume list

* will not receive DBT_DEVICEARRIVAL and DBT_DEVICEREMOVECOMPLETE(Like Mouse and KeyBorad with not line).

*/

if ((DWORD)wp == DBT_DEVICEARRIVAL) {

DEV_BROADCAST_VOLUME* p = (DEV_BROADCAST_VOLUME*)lp;

if (p->dbcv_devicetype == DBT_DEVTYP_VOLUME) {

char disk = FirstDriveFromMask(p->dbcv_unitmask);

printf("注意---%c盘插进来了\n", disk);

}

}

else if ((DWORD)wp == DBT_DEVICEREMOVECOMPLETE) {

DEV_BROADCAST_VOLUME* p = (DEV_BROADCAST_VOLUME*)lp;

if (p->dbcv_devicetype == DBT_DEVTYP_VOLUME) {

char disk = FirstDriveFromMask(p->dbcv_unitmask);

printf("注意---%c盘被拔掉了\n", disk);

}

}

return TRUE;

}

else return DefWindowProc(h, msg, wp, lp);

}