MBR的读取

1. void  CKillScriptDlg::ReadMBR(BYTE* pMBR, UINT nLen)     
2. {     
3.     HANDLE   hDevice;        
4.     LPSTR   szBuff = NULL;     
5.     DISK_GEOMETRY   Geometry = {0};     
6.     BOOL   bRet;     
7.     DWORD  bytes = 0,bread = 0,count = 0;     
8.     int    i = 0;     
9.        
10.     hDevice   =   CreateFileA( "////.//PHYSICALDRIVE0",     
11.         GENERIC_READ | GENERIC_WRITE,    
12.         FILE_SHARE_READ | FILE_SHARE_WRITE,     
13.         NULL, OPEN_EXISTING, 0, NULL );     
14.     if (hDevice == INVALID_HANDLE_VALUE)     
15.     {     
16.         MessageBoxA(NULL,"打开设备出错",NULL,MB_OK);     
17.         return;     
18.     }
19.     // Lock volume for read operation
20.     bRet = DeviceIoControl(hDevice,
21.                     FSCTL_LOCK_VOLUME,
22.                     NULL, 0,
23.                     NULL, 0,
24.                     &count,    NULL); 
25.     if(!bRet)
26.     {
27.         CloseHandle(hDevice);
28.         return;
29.     }
31.     bRet = DeviceIoControl(hDevice, 
32.                     IOCTL_DISK_GET_DRIVE_GEOMETRY, 
33.                     NULL, 0, 
34.                     &Geometry,
35.                     sizeof(DISK_GEOMETRY),
36.                     &count,NULL);
37.     if(!bRet)
38.     {
39.         goto END;
40.     }
41.     szBuff = (LPSTR)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, Geometry.BytesPerSector);     
42.     if( szBuff == NULL)     
43.     {
44.         MessageBoxA(NULL,"分配内存出错",NULL,MB_OK);     
45.         goto END;
46.     }
48.     bytes = 512;     
49.     bRet = ReadFile(hDevice, szBuff, bytes, &bread, NULL);     
50.     if (bRet == FALSE || bread<512)     
51.     {     
52.         MessageBoxA(NULL,"读MBR出错",NULL,MB_OK);     
53.         goto END;   
54.     }     
55.     for(int n=0; n<512; n++)
56.     {
57.         pMBR[n] = szBuff[n];   
58.     }
59. END:
60.     //unlock volume
61.     DeviceIoControl(hDevice, FSCTL_UNLOCK_VOLUME, NULL, 0, NULL, 0, &count,NULL);
62.     CloseHandle(hDevice);     
63. }