简单MBR读取库的实现

20011010wo

1. <div class="blockcode"><blockquote>#define WIN32_LEAN_AND_MEAN            
   
2. 
   
3. #include <windows.h>
   
4. #include <io.h>
   
5. #include <sys\stat.h>
   
6. #include <sys\types.h>
   
7. #include <share.h>
   
8. #include <fcntl.h>
   
9. #include <cstdlib>
   
10. 
    
11. #include <string>
    
12. struct MBRPartitionTable
    
13. {
    
14.         unsigned char bootsign;
    
15.         unsigned char start_chsaddress[3];
    
16.         unsigned char type;
    
17.         unsigned char end_chsaddress[3];
    
18.         unsigned char used_sectorsnum[4];
    
19.         unsigned char total_sectorsnum[4];
    
20. };
    
21. struct MBR
    
22. {
    
23.         unsigned char bootstrapcode[446];
    
24.         MBRPartitionTable a;
    
25.         MBRPartitionTable b;
    
26.         MBRPartitionTable c;
    
27.         MBRPartitionTable d;
    
28.         unsigned char endsign[2];
    
29. };
    
30. // 物理驱动器对象
    
31. class PhysicalDrive
    
32. {
    
33. public:
    
34.         PhysicalDrive(void) : drive(L"\\\\.\\PhysicalDrive"), drive_index(0){}        //构造函数
    
35.         std::wstring GetDriveName(void)                                                                                        //获取存储设备名称  
    
36.         {
    
37.                 wchar_t drive_index_buffer[8];
    
38.                 _itow_s(drive_index, drive_index_buffer, 8, 10);                                        //int 转 wcahr_t
    
39.                 return drive + drive_index_buffer;
    
40.         }
    
41. private:
    
42.         std::wstring drive;                                                     //存储设备部分名称
    
43.         int drive_index;                                                                              //存储设备号
    
44. };
    
45. 
    
46. class Libmbr {
    
47. public:
    
48.         Libmbr(void) = delete;                                                          //不允许合成构造函数
    
49.         Libmbr(PhysicalDrive);                                                                        //构造函数
    
50.         bool ReadMBR();                                                                                       //读MBR函数
    
51.         bool WriteMBR();                                                                                    //写MBR函数
    
52.         MBR code = {0};                                                              //将要写入的数据
    
53. private:
    
54.         bool ReadWriteMBR(bool);                                                     //内部实现函数
    
55.         PhysicalDrive m_drive;                                                          //存储设备对象
    
56.         int iFileHandler;                                                              //文件句柄
    
57. };
    

复制代码

1. #include "libmbr.h"
   
2. 
   
3. 
   
4. 
   
5. inline Libmbr::Libmbr(PhysicalDrive drive) :m_drive(drive) {}        //构造函数
   
6. 
   
7. inline bool Libmbr::ReadMBR()
   
8. {
   
9.         return ReadWriteMBR(true);
   
10. }
    
11. 
    
12. inline bool Libmbr::WriteMBR()
    
13. {
    
14.         return ReadWriteMBR(false);
    
15. }
    
16. 
    
17. 
    
18. bool Libmbr::ReadWriteMBR(bool isRead)
    
19. {
    
20.         //打开设备
    
21.         if (_wsopen_s(&iFileHandler, m_drive.GetDriveName().c_str(), _O_BINARY | _O_RDWR, _SH_DENYNO, _S_IREAD | _S_IWRITE))
    
22.                 return false;
    
23.         //判断是否至EOF
    
24.         if (_eof(iFileHandler))
    
25.                 return false;
    
26.         //移动至起始点
    
27.         if (_lseek(iFileHandler, 0, SEEK_SET) == -1)
    
28.                 return false;
    
29.         //读写属性
    
30.         if (isRead)
    
31.         {
    
32.                 if (_read(iFileHandler, &code, 512) == -1)        //读512个字
    
33.                         return false;
    
34.         }
    
35.         else
    
36.         {
    
37.                 if (_write(iFileHandler, &code, 512) == -1)        //写512个字
    
38.                         return false;
    
39.         }
    
40.         //刷新设备缓冲
    
41.         if (_commit(iFileHandler) == -1)
    
42.                 return false;
    
43.         //关闭句柄
    
44.         if (_close(iFileHandler) == -1)
    
45.                 return false;
    
46.         return true;
    
47. }
    

复制代码

20011010wo

补充：还有数据结构还没加入

20011010wo

加入了

nkc3g4

赞~