当你的U盘,移动硬盘遗失了,被盗了,或者是遗忘在公共场所,里面的资料就会完全被别人窥探,如果涉及一些隐私,机密,那后果是很严重的。本课题就是在这样一个情况下产生的。这里剖析的只是一个最初的演示原型,详细设计,以及一些细节可以参考源代码。源代码在WinXP,VC6.0编译通过。
1、访问注册表读取计算机上的移动存储设备
在注册表
1.
HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Disk\Enum
下面可以看到计算机当前的状态,这里可以看到所有的存储设备的情况,包括计算机的硬盘,通过情况下,名称为0的是计算机硬盘,所以在列出的选择设备时,会把名称为0的屏蔽,不然太危险,一旦造成系统的破坏,后果会很麻烦。对注册表的读写:
01.
HKEY
hkey;
02.
char
sz[256];
03.
DWORD
dwtype,sl = 256;
04.
for
(
int
i=1;i<8;i++)
05.
{
06.
if
(RegOpenKeyEx(HKEY_LOCAL_MACHINE,"SYSTEM\\CurrentControlSet
07.
\\Services\\Disk\\Enum",\
08.
NULL, KEY_ALL_ACCESS, &hkey)==ERROR_SUCCESS)
09.
{
10.
CString id;
11.
id.Format(
"%d"
,i);
12.
if
(RegQueryValueEx(hkey,id,NULL,&dwtype,(
LPBYTE
)sz,&sl)==ERROR_SUCCESS)
13.
{
14.
CString str=(CString)sz;
15.
m_select.AddString(sz);
16.
}
17.
}
18.
}
19.
RegCloseKey(hkey);
2、对磁盘的扇区的操作
Windows 操作系统在很大程度上采取了访问安全保护机制(例如,在Windows操作系统下不能直接访问物理内存、不能使用各种DOS、BIOS中断等等),其实Windows在采取“实保护”措施的同时也提供了另外的一种有别于在DOS下访问硬件设备的方法,即把所有的硬件设备全部看做“文件”,并允许按照对文件的读写方式来对其进行数据存取访问。对于磁盘扇区的读写,可以通过C++的CreateFile()函数来实现。由MSDN可查询到该函数原型:
1.
HANDLE
CreateFile(
2.
LPCTSTR
lpFileName,
3.
DWORD
dwDesiredAccess,
4.
DWORD
dwShareMode,
5.
LPSECURITY_ATTRIBUTES lpSecurityAttributes,
6.
DWORD
dwCreationDisposition,
7.
DWORD
dwFlagsAndAttributes,
8.
HANDLE
hTemplateFile );
由于访问的是事实上已经存在的磁盘扇区,因此只能以OPEN_EXISTING标志设置dwCreationDisposition参数指出将要打开已经存在的文件(设备)。至于其他参数的使用与操作普通文件时的用法相同。
使用的时候,如果需要定位到某一个具体的扇区,可以使用SetFilePointer()函数:
1.
DWORD
SetFilePointer(
HANDLE
hFile,
LONG
lDistanceToMove,
PLONG
lpDistanceToMoveHigh,
DWORD
dwMoveMethod);
在定位到要访问的扇区开始位置后就可以通过ReadFile()或WriteFile()函数实施相应的读写访问了,具体操作与文件读写并没有什么太大的差别。最后,在完成访问操作后以CloseHandle()关闭文件句柄释放资源,从而完成一次完整的磁盘扇区数据访问操作。
01.
int
ReadDisk(CString driver,unsigned
char
*Buf,
long
addr)
02.
{
03.
HANDLE
hDevice;
04.
BOOL
bResult;
05.
DWORD
bytesread;
06.
07.
hDevice=CreateFile(driver,GENERIC_READ|GENERIC_WRITE,
08.
FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_EXISTING,0,NULL);
09.
if
(hDevice==INVALID_HANDLE_VALUE)
10.
{
11.
AfxMessageBox(
"Error!"
);
12.
return
0;
13.
}
14.
if
(addr!=0)
15.
{
16.
SetFilePointer(hDevice,512*addr,NULL,NULL);
17.
}
18.
19.
bResult=ReadFile(hDevice,Buf,512,&bytesread,NULL);
20.
if
((bResult==FALSE)||(bytesread<512))
21.
{
22.
AfxMessageBox(
"Error!"
);
23.
return
0;
24.
}
25.
CloseHandle(hDevice);
26.
return
1;
27.
}
3、用RC4加密算法对磁盘加密。
1.
/* RC4加密,KEY是密钥,此处Key[]="MobileStorageSecurity",后期可以用户输入的密码作为密钥 */
2.
RC4_KEY rc4_key;
3.
build_rc4_key(Key,
strlen
((
char
*)Key),&rc4_key);
4.
rc4_handler(MBRBuf,
strlen
((
char
*)MBRBuf),&rc4_key);
4、U盘插入计算机时的自动感知
1.
LRESULT
CRawDiskDlg::WindowProc(
UINT
message,
WPARAM
wParam,
LPARAM
lParam)
2.
{
3.
if
( WM_DEVICECHANGE == message && FALSE == bCopy )
4.
{
5.
//刷新设备列表
6.
}
7.
return
CDialog::WindowProc(message, wParam, lParam);
8.
}
后期改进
1.这样就必须要求在电脑上有这个程序,所以在后期采用同样的原理,将这个程序放在U盘里,在加密的时候实现对自身的屏蔽,这样大大提高程序的易用性。
2.可以让用户输入密码,将密码作为密钥去加密和解密。
3.这个只是对一个磁盘最初的512字节加密,后期可以通过对扇区的定位和计算,对目录表及数据加密。