核心代码如下：

#include "StdAfx.h"
#include "Function.h"CString GetDiskNumber(CString name)
{HKEY hkey;char sz[256];DWORD dwtype,sl = 256;int number=0;// 确定选择的磁盘for(int i=1;i<8;i++){if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,"SYSTEM//CurrentControlSet//Services//Disk//Enum",/NULL,KEY_ALL_ACCESS,&hkey)==ERROR_SUCCESS){ CString id;id.Format("%d",i);if(RegQueryValueEx(hkey,id,NULL,&dwtype,(LPBYTE)sz,&sl)==ERROR_SUCCESS){CString str=(CString)sz;if(str.Compare(name)==0){number=i;break;}}}}CString driver=".//PHYSICALDRIVE";CString num;num.Format("%d",number);driver+=num;return driver;
}int ReadDisk(CString driver,unsigned char *Buf,long addr)
{HANDLE hDevice;BOOL bResult;DWORD bytesread;hDevice=CreateFile(driver,GENERIC_READ|GENERIC_WRITE,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_EXISTING,0,NULL);if(hDevice==INVALID_HANDLE_VALUE){   AfxMessageBox("Error!");return 0;}if(addr!=0){SetFilePointer(hDevice,512*addr,NULL,NULL);}bResult=ReadFile(hDevice,Buf,512,&bytesread,NULL);if((bResult==FALSE)||(bytesread<512))   {   AfxMessageBox("Error!");return 0;}CloseHandle(hDevice);return 1;
}int WriteDisk(CString driver,unsigned char *Buf,long addr)
{HANDLE hDevice;BOOL bResult;DWORD bytesread;hDevice=CreateFile(driver,GENERIC_READ|GENERIC_WRITE,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_EXISTING,0,NULL);if(hDevice==INVALID_HANDLE_VALUE){   AfxMessageBox("Error!");return 0;}if(addr!=0){SetFilePointer(hDevice,512*addr,NULL,NULL);}bResult=WriteFile(hDevice,Buf,512,&bytesread,NULL);if((bResult==FALSE)||(bytesread<512))   {   AfxMessageBox("Error!");return 0;}CloseHandle(hDevice);return 1;
}void PopupUSBDevice()
{char strSystemDirectory[256];GetSystemDirectory( strSystemDirectory, 256 );CString strTemp = strSystemDirectory;strTemp += "//rundll32.exe shell32.dll,Control_RunDLL hotplug.dll";WinExec( strTemp, SW_SHOW );
}

调用代码如下

BOOL CMSSDlg::OnInitDialog()
{CDialog::OnInitDialog();// Add "About..." menu item to system menu.// IDM_ABOUTBOX must be in the system command range.ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);ASSERT(IDM_ABOUTBOX < 0xF000);CMenu* pSysMenu = GetSystemMenu(FALSE);if (pSysMenu != NULL){CString strAboutMenu;strAboutMenu.LoadString(IDS_ABOUTBOX);if (!strAboutMenu.IsEmpty()){pSysMenu->AppendMenu(MF_SEPARATOR);pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);}}// Set the icon for this dialog.  The framework does this automatically//  when the application's main window is not a dialogSetIcon(m_hIcon, TRUE);            // Set big iconSetIcon(m_hIcon, FALSE);     // Set small icon// TODO: Add extra initialization hereHKEY hkey;char sz[256];DWORD dwtype,sl = 256;for(int i=1;i<8;i++){if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,"SYSTEM//CurrentControlSet//Services//Disk//Enum",/NULL, KEY_ALL_ACCESS, &hkey)==ERROR_SUCCESS){CString id;id.Format("%d",i);if(RegQueryValueEx(hkey,id,NULL,&dwtype,(LPBYTE)sz,&sl)==ERROR_SUCCESS){CString str=(CString)sz;m_select.AddString(sz);}}}RegCloseKey(hkey);m_select.SetCurSel(0);if(m_select.GetCount()==0){AfxMessageBox("没有发现移动设备!");}return TRUE;  // return TRUE  unless you set the focus to a control
}void CMSSDlg::OnSysCommand(UINT nID, LPARAM lParam)
{if ((nID & 0xFFF0) == IDM_ABOUTBOX){CAboutDlg dlgAbout;dlgAbout.DoModal();}else{CDialog::OnSysCommand(nID, lParam);}
}// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.void CMSSDlg::OnPaint()
{if (IsIconic()){CPaintDC dc(this); // device context for paintingSendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);// Center icon in client rectangleint cxIcon = GetSystemMetrics(SM_CXICON);int cyIcon = GetSystemMetrics(SM_CYICON);CRect rect;GetClientRect(&rect);int x = (rect.Width() - cxIcon + 1) / 2;int y = (rect.Height() - cyIcon + 1) / 2;// Draw the icondc.DrawIcon(x, y, m_hIcon);}else{CDialog::OnPaint();}
}// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CMSSDlg::OnQueryDragIcon()
{return (HCURSOR) m_hIcon;
}void CMSSDlg::OnExit()
{// TODO: Add your control notification handler code hereCMSSDlg::OnCancel();
}LRESULT CMSSDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)
{if (WM_DEVICECHANGE == message){m_select.ResetContent();HKEY hkey;char sz[256];DWORD dwtype,sl = 256;for(int i=1;i<8;i++){if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,"SYSTEM//CurrentControlSet//Services//Disk//Enum",/NULL,KEY_ALL_ACCESS,&hkey)==ERROR_SUCCESS){CString id;id.Format("%d",i);if(RegQueryValueEx(hkey,id,NULL,&dwtype,(LPBYTE)sz,&sl)==ERROR_SUCCESS){CString str=(CString)sz;m_select.AddString(sz);}}}m_select.SetCurSel(0);RegCloseKey(hkey);}return CDialog::WindowProc(message, wParam, lParam);
}void CMSSDlg::OnAddPassword()
{// TODO: Add your control notification handler code hereunsigned char MBRBuf[512];CString name;CString driver;int id;              // 选择移动设备的编号if(m_select.GetCurSel()==CB_ERR){AfxMessageBox("请选择要加密的设备！");return;}id=m_select.GetCurSel();m_select.GetLBText(id,name);// 确定选择的磁盘driver=GetDiskNumber(name);// 读磁盘的MBR区if(ReadDisk(driver,MBRBuf,0)==0)return;/* RC4加密，KEY是密钥，此处Key[]="MobileStorageSecurity",后期可以用户输入的密码作为密钥 */RC4_KEY rc4_key;build_rc4_key(Key,strlen((char*)Key),&rc4_key);rc4_handler(MBRBuf,strlen((char*)MBRBuf),&rc4_key);// 将加密后的MBR写入磁盘if(WriteDisk(driver,MBRBuf,0)==0)return;AfxMessageBox("加密成功！");// 弹出USB存储设备PopupUSBDevice();
}void CMSSDlg::OnRemovePassword()
{// TODO: Add your control notification handler code hereunsigned char MBRBuf[512];CString name;CString driver;int id;              // 选择移动设备的编号if(m_select.GetCurSel()==CB_ERR){AfxMessageBox("请选择要加密的设备！");return;}id=m_select.GetCurSel();m_select.GetLBText(id,name);// 确定选择的磁盘driver=GetDiskNumber(name);// 读磁盘的MBR区if(ReadDisk(driver,MBRBuf,0)==0)return;/* RC4加密，KEY是密钥，此处Key[]="MobileStorageSecurity",后期可以用户输入的密码作为密钥 */RC4_KEY rc4_key;build_rc4_key(Key,strlen((char*)Key),&rc4_key);rc4_handler(MBRBuf,strlen((char*)MBRBuf),&rc4_key);// 将解密后的MBR写入磁盘if(WriteDisk(driver,MBRBuf,0)==0)return;AfxMessageBox("解密成功！");// 弹出USB存储设备PopupUSBDevice();
}

原文链接： http://blog.csdn.net/yincheng01/article/details/4668226