1 //main.cpp
 2 #include <stdio.h>
 3 #include <windows.h>
 4 #include "ntfs_types.h"
 5 #include "ntfs_diskio.h"
 6 #include "ntfs_basic.h"
 7 
 8 DPT_Info DPT[10];
 9 MBR_Info MBR[10];
10 int main(void)
11 {
12     GetPartitioninformation(DPT);
13     DisplayPartitionInformation(DPT);
14     for(int i = 0; i < 7; i++)
15     {
16         GetMBR(&MBR[i], &DPT[i]);
17     }
18     printf("\n");
19     DisplayMBR(MBR);
20 
21     return 0;
22 }

 1 //ntfs_diskio.cpp
 2 #include <stdio.h>
 3 #include <windows.h>
 4 #include "ntfs_diskio.h"
 5 #include "ntfs_basic.h"
 6 
 7 BOOL ReadSectorData( HANDLE& hDevice,UINT64 redpos, char * lpOutBuffer512 )
 8 {
 9     memset(lpOutBuffer512,0,512);
10     LARGE_INTEGER li;
11     li.QuadPart = redpos*0x200;//0x200 = 512,求出扇区的 字节地址，通过设置读取的地址和长度进行read
12     SetFilePointer(hDevice,li.LowPart,&li.HighPart,FILE_BEGIN);
13     DWORD DCount=0; //计数
14     BOOL bResult=ReadFile(hDevice, lpOutBuffer512, 512, &DCount, NULL);
15     return bResult;
16 }
17 
18 //通过给定磁盘的编号，获取到磁盘的句柄
19 HANDLE GetDiskHandle(int iDiskNo)
20 {
21     char szDriverBuffer[128];
22     memset(szDriverBuffer,0,128);
23     //格式化设备文件名称
24     sprintf(szDriverBuffer,"\\\\.\\PhysicalDrive%d",iDiskNo);
25     HANDLE m_hDevice = NULL;
26     //CreateFile获取到设备句柄
27     m_hDevice = CreateFileA(
28         szDriverBuffer,// 设备名称,这里指第一块硬盘，多个硬盘的自己修改就好了
29         GENERIC_READ, // 指定读访问方式
30         FILE_SHARE_READ | FILE_SHARE_WRITE, // 共享模式为读|写，0表示不能共享
31         NULL, // NULL表示该句柄不能被子程序继承
32         OPEN_EXISTING, // 打开已经存在的文件，文件不存在则函数调用失败
33         NULL, // 指定文件属性
34         NULL);
35     if (m_hDevice==INVALID_HANDLE_VALUE)
36     {
37         m_hDevice = NULL;
38         //无效
39         return INVALID_HANDLE_VALUE;
40     }
41     //设备句柄
42     return m_hDevice;
43 }

 1 //ntfs_diskio.h
 2 #ifndef __NTFS_DISKIO_H
 3 #define __NTFS_DISKIO_H
 4 
 5 #include "ntfs_types.h"
 6 
 7 BOOL ReadSectorData( HANDLE& hDevice,UINT64 redpos, char * lpOutBuffer512 );//读取扇区函数，移植到MCU改变函数实现
 8 HANDLE GetDiskHandle(int iDiskNo);
 9 
10 #endif

 1 //ntfs_basic.h
 2 #ifndef __NTFS_BASIC_H
 3 #define __NTFS_BASIC_H
 4 
 5 #include "ntfs_types.h"
 6 
 7 uint16_t ArrayToU16LittleEnd(uint8_t *str);
 8 uint32_t ArrayToU32LittleEnd(uint8_t *str);
 9 uint64_t ArrayToU64LittleEnd(uint8_t *str);
10 //获取扇区MBR信息
11 void GetMBR(MBR_Info *MBR, DPT_Info *DPT);
12 //获取磁盘分区信息，不区分扩展分区和主分区，均以“分区”表示，DPT_Info中保存物理起始扇区
13 void GetPartitioninformation(DPT_Info* DPT);
14 //扇区信息转成可以识别的Info结构
15 void DptTransferLittleEnd(DPT_Byte* src, DPT_Info *dest);
16 void EbrTransferLittleEnd(DPT_Byte* src, DPT_Info *dest);
17 void MbrTransferLittleEnd(MBR_Byte* src, MBR_Info *dest);
18 
19 //以下为PC机调试显示用，移植到MCU需要移除
20 void DisplayPartitionInformation(DPT_Info* DPT);
21 void DisplayMBR(MBR_Info *MBR);
22 
23 #endif

  1 //ntfs_basic.cpp
  2 #include <stdio.h>
  3 #include <windows.h>
  4 #include "ntfs_basic.h"
  5 #include "ntfs_diskio.h"
  6 
  7 #define PHYDRIVE 4
  8 
  9 uint16_t ArrayToU16LittleEnd(uint8_t *str)
 10 {
 11     return (str[0] + (str[1] << 8));
 12 }
 13 
 14 uint32_t ArrayToU32LittleEnd(uint8_t *str)
 15 {
 16     return (str[0] + (str[1] << 8) + (str[2] << 16) + (str[3] << 24));
 17 }
 18 
 19 uint64_t ArrayToU64LittleEnd(uint8_t *str)
 20 {
 21     return (str[0] + (str[1] << 8) + (str[2] << 16) + (str[3] << 24) + (str[4] << 32) + (str[5] << 40) + (str[6] << 48) + (str[7] << 56));
 22 }
 23 
 24 void GetMBR(MBR_Info *MBR, DPT_Info *DPT)
 25 {
 26     MBR_Byte MBRsector;
 27     HANDLE hdl = GetDiskHandle(PHYDRIVE);//获取一个指定物理驱动器句柄
 28     ReadSectorData(hdl, DPT->RealOffset, (char*)(void*)&MBRsector);
 29     MbrTransferLittleEnd(&MBRsector, MBR);
 30 }
 31 void GetPartitioninformation(DPT_Info* DPT)
 32 {
 33     DPT_Byte DPTsector;//定义DPT扇区作为buffer，注意MCU的栈尺寸要大于512字节
 34     unsigned long sectorsoffset = 0;
 35     unsigned long extendsoffset = 0;
 36     HANDLE hdl = GetDiskHandle(PHYDRIVE);//获取一个指定物理驱动器句柄
 37     ReadSectorData(hdl, 0, (char*)(void*)&DPTsector);//载入DPT
 38     DptTransferLittleEnd(&DPTsector, DPT);//小端模式转换扇区数据为分区信息
 39     int k;
 40     for(k = 0; k < 4; k++)
 41     {
 42         if(DPT[k].PartitionType == fsptWin95_Extended_8GB || DPT[k].PartitionType == fsptExtended)//两种扩展分区
 43             break;
 44         DPT[k].RealOffset = DPT[k].UsedSector;
 45     }
 46     extendsoffset = DPT[k].UsedSector;
 47     sectorsoffset = extendsoffset;
 48     int i = k;
 49     unsigned char Partitions = 255;//sizeof(DPT) / sizeof(DPT_Info);
 50     while(i < Partitions)//最大遍历分区数量Partitions，防止由于分区过多导致下标越界
 51     {
 52         if(DPT[i].PartitionType == fsptWin95_Extended_8GB || DPT[i].PartitionType == fsptExtended)//两种扩展分区
 53         {
 54             if(i == k)
 55                 DPT[i].RealOffset = extendsoffset;
 56             else
 57                 DPT[i].RealOffset = extendsoffset + DPT[i].UsedSector + DPT[i+1].UsedSector;
 58             ReadSectorData(hdl, sectorsoffset, (char*)(void*)&DPTsector);//载入EBR
 59             DptTransferLittleEnd(&DPTsector, &DPT[i]);//小端模式转换扇区数据为分区信息
 60             sectorsoffset = extendsoffset + DPT[i+1].UsedSector;
 61             DPT[i].RealOffset+=DPT[i].UsedSector;//对扩展扇区MBR地址做补偿
 62             i++;
 63         }
 64         else if(DPT[i].TotalSector == NULL)
 65         {
 66             break;
 67         }
 68     }
 69     
 70 }
 71 
 72 void DisplayPartitionInformation(DPT_Info* DPT)
 73 {
 74     printf("磁盘%d：\n", PHYDRIVE);
 75     printf("               活动分区     分区类型     起始扇区     总扇区数      容量\n");
 76     printf("------------------------------------------------------------------------\n");
 77     int i = 0;
 78     unsigned char Partitions = 255;//sizeof(DPT) / sizeof(DPT_Info);
 79     while(DPT[i].TotalSector != NULL && i < Partitions)
 80     {
 81         printf("磁盘4分区%d：  ", i);
 82         if(DPT[i].ActivePartition)
 83             printf("       是");
 84         else
 85             printf("       否");
 86         printf("%13d",    DPT[i].PartitionType);
 87         printf("%13d",    DPT[i].RealOffset);
 88         printf("%13d",    DPT[i].TotalSector);
 89         printf("%8.2fGB", DPT[i].TotalSector/2097152.0);
 90         printf("\n");
 91         i++;
 92     }
 93 }
 94 
 95 void DisplayMBR(MBR_Info *MBR)
 96 {
 97     printf("磁盘%d：\n", PHYDRIVE);
 98     printf("OEM名        扇区总数    MFT起始簇号    MFT备份簇号    MFT大小    索引大小\n");
 99     printf("--------------------------------------------------------------------------\n");
100     int i = 0;
101     unsigned char Partitions = 255;//sizeof(MBR) / sizeof(MBR_Info);
102     while(MBR[i].TotalSectors != NULL && i < Partitions)
103     {
104         printf("%s",     MBR[i].OEMName);
105         printf("%13lld", MBR[i].TotalSectors);
106         printf("%15lld", MBR[i].MFTStartCluster);
107         printf("%15lld", MBR[i].MFTBackupStartCluster);
108         printf("%11d",   MBR[i].SizePerMFT);
109         printf("%12lld", MBR[i].ClusterPerIndex);
110         printf("\n");
111         i++;
112     }
113 }
114 void DptTransferLittleEnd(DPT_Byte* src, DPT_Info *dest)
115 {
116     for(int i = 0; i < 4; i++)
117     {
118         dest[i].ActivePartition   = src->DPT_Table[i].ActivePartition == 0x80 ? true : false;//0x80是活动扇区，0x00是非活动扇区
119         dest[i].StartMagneticHead = src->DPT_Table[i].StartInfo[0];
120         dest[i].StartSector       = src->DPT_Table[i].StartInfo[1] & 0x3F;
121         dest[i].StartCylinder     = src->DPT_Table[i].StartInfo[2] + ((src->DPT_Table[i].StartInfo[1] & 0xC0) << 2);
122         dest[i].PartitionType     = (Partition_Type)src->DPT_Table[i].PartitionType;
123         dest[i].EndMagneticHead   = src->DPT_Table[i].EndInfo[0];
124         dest[i].EndSector         = src->DPT_Table[i].EndInfo[1] & 0x3F;
125         dest[i].EndCylinder       = src->DPT_Table[i].EndInfo[2] + ((src->DPT_Table[i].EndInfo[1] & 0xC0) << 2);
126         dest[i].UsedSector        = ArrayToU32LittleEnd(src->DPT_Table[i].UsedSector);
127         dest[i].TotalSector       = ArrayToU32LittleEnd(src->DPT_Table[i].TotalSector);
128     }
129 }
130 
131 void EbrTransferLittleEnd(DPT_Byte* src, DPT_Info *dest)
132 {
133     for(int i = 0; i < 2; i++)
134     {
135         dest[i].ActivePartition   = src->DPT_Table[i].ActivePartition == 0x80 ? true : false;//0x80是活动扇区，0x00是非活动扇区
136         dest[i].StartMagneticHead = src->DPT_Table[i].StartInfo[0];
137         dest[i].StartSector       = src->DPT_Table[i].StartInfo[1] & 0x3F;
138         dest[i].StartCylinder     = src->DPT_Table[i].StartInfo[2] + ((src->DPT_Table[i].StartInfo[1] & 0xC0) << 2);
139         dest[i].PartitionType     = (Partition_Type)src->DPT_Table[i].PartitionType;
140         dest[i].EndMagneticHead   = src->DPT_Table[i].EndInfo[0];
141         dest[i].EndSector         = src->DPT_Table[i].EndInfo[1] & 0x3F;
142         dest[i].EndCylinder       = src->DPT_Table[i].EndInfo[2] + ((src->DPT_Table[i].EndInfo[1] & 0xC0) << 2);
143         dest[i].UsedSector        = ArrayToU32LittleEnd(src->DPT_Table[i].UsedSector);
144         dest[i].TotalSector       = ArrayToU32LittleEnd(src->DPT_Table[i].TotalSector);
145     }
146 }
147 
148 void MbrTransferLittleEnd(MBR_Byte* src, MBR_Info *dest)
149 {
150     dest->OEMName[0]              = src->OEMName[0];
151     dest->OEMName[1]              = src->OEMName[1];
152     dest->OEMName[2]              = src->OEMName[2];
153     dest->OEMName[3]              = src->OEMName[3];
154     dest->OEMName[4]              = src->OEMName[4];
155     dest->OEMName[5]              = src->OEMName[5];
156     dest->OEMName[6]              = src->OEMName[6];
157     dest->OEMName[7]              = src->OEMName[7];
158     dest->OEMName[8]              = 0;
159     dest->BytesPerSector          = ArrayToU16LittleEnd(src->BytesPerSector);
160     dest->SectorsPerCluster       = src->SectorsPerCluster;
161     dest->ReservedSector          = ArrayToU16LittleEnd(src->ReservedSector);
162     dest->MediaDescriptor         = src->MediaDescriptor;
163     dest->SectorsPerTrack         = ArrayToU16LittleEnd(src->SectorsPerTrack);
164     dest->MegnaticHeadPerCylinder = ArrayToU16LittleEnd(src->MegnaticHeadPerCylinder);
165     dest->HiddenSectors           = ArrayToU32LittleEnd(src->HiddenSectors);
166     dest->TotalSectors            = ArrayToU64LittleEnd(src->TotalSectors);
167     dest->MFTStartCluster         = ArrayToU64LittleEnd(src->MFTStartCluster);
168     dest->MFTBackupStartCluster   = ArrayToU64LittleEnd(src->MFTBackupStartCluster);
169     dest->SizePerMFT              = src->SizePerMFT;
170     dest->ClusterPerIndex         = src->ClusterPerIndex;
171     dest->SerialNumber            = ArrayToU64LittleEnd(src->SerialNumber);
172     dest->CheckSum                = ArrayToU32LittleEnd(src->CheckSum);
173 }