c++检测u盘插入c++代码示例
时间: 2024-09-12 10:13:33 浏览: 44
U盘偷猎者代码(c++)
在C++中,检测U盘插入并不是语言本身直接提供的功能,通常需要依赖操作系统提供的API。以Windows为例,可以使用Windows的设备通知机制(RegisterDeviceNotification API)来接收系统通知,当U盘等移动存储设备插入时,系统会发送一个通知消息。
以下是一个简单的C++示例代码,展示如何使用Windows API在Windows平台上检测U盘插入:
```cpp
#include <windows.h>
#include <dbt.h>
// 定义一个用于接收通知消息的回调函数
LRESULT CALLBACK NotifyFunc(WPARAM wParam, LPARAM lParam) {
DEV_BROADCAST_HDR* dbh = (DEV_BROADCAST_HDR*)lParam;
if (dbh->dbch_devicetype == DBT_DEVTYP_DEVICEINTERFACE) {
DEV_BROADCAST_DEVICEINTERFACE* dbdi = (DEV_BROADCAST_DEVICEINTERFACE*)dbh;
// 在这里处理设备插入事件
MessageBox(NULL, L"U盘插入了!", L"设备通知", MB_OK);
}
return TRUE;
}
int main() {
// 使用WM_DEVICECHANGE消息注册一个通知窗口
HWND hwnd = CreateWindow(L"STATIC", L"DeviceNotificationWindow", 0, 0, 0, 0, 0, HWND_MESSAGE, NULL, NULL, NULL);
WNDCLASSEX wc = {0};
wc.cbSize = sizeof(WNDCLASSEX);
wc.lpfnWndProc = DefWindowProc;
wc.hInstance = GetModuleHandle(NULL);
wc.lpszClassName = L"DeviceNotificationWindowClass";
RegisterClassEx(&wc);
// 填充DEV_BROADCAST_DEVICEINTERFACE结构体并注册
DEV_BROADCAST_DEVICEINTERFACE notificationFilter = {0};
notificationFilter.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
notificationFilter.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
// 这里可以指定特定设备的GUID,如果你想监听特定类型的设备
HDEVNOTIFY hDevNotify = RegisterDeviceNotification(hwnd, ¬ificationFilter, DEVICE_NOTIFY_WINDOW_HANDLE | DEVICE_NOTIFY_ALL_INTERFACE_CLASSES);
if (!hDevNotify) {
// 注册失败处理
MessageBox(NULL, L"注册设备通知失败!", L"错误", MB_ICONERROR);
return 1;
}
// 进入消息循环
MSG msg = {0};
while (GetMessage(&msg, NULL, 0, 0)) {
if (msg.message == WM_DEVICECHANGE) {
NotifyFunc(msg.wParam, msg.lParam);
}
}
// 清理
UnregisterDeviceNotification(hDevNotify);
DestroyWindow(hwnd);
return 0;
}
```
这段代码创建了一个隐藏的窗口,用于接收设备通知消息,并在U盘插入时弹出消息框。这里使用了`RegisterDeviceNotification`函数来注册设备通知。
需要注意的是,编写这样的代码需要有Windows开发环境,并且需要处理很多细节问题,例如消息循环的管理,以及错误处理等。
阅读全文