Modules/FAT - Fixed edge cases in read
[tpg/acess2.git] / Modules / Filesystems / NTFS / main.c
1 /*
2  * Acess2 - NTFS Driver
3  * By John Hodge (thePowersGang)
4  *
5  * main.c - Driver core
6  */
7 #define DEBUG   1
8 #define VERBOSE 0
9 #include <acess.h>
10 #include <vfs.h>
11 #include "common.h"
12 #include <modules.h>
13
14 // === IMPORTS ===
15 extern char     *NTFS_ReadDir(tVFS_Node *Node, int Pos);
16 extern tVFS_Node        *NTFS_FindDir(tVFS_Node *Node, const char *Name);
17
18 // === PROTOTYPES ===
19  int    NTFS_Install(char **Arguments);
20 tVFS_Node       *NTFS_InitDevice(const char *Devices, const char **Options);
21 void    NTFS_Unmount(tVFS_Node *Node);
22 void    NTFS_DumpEntry(tNTFS_Disk *Disk, Uint32 Entry);
23
24 // === GLOBALS ===
25 MODULE_DEFINE(0, 0x0A /*v0.1*/, FS_NTFS, NTFS_Install, NULL);
26 tVFS_Driver     gNTFS_FSInfo = {"ntfs", 0, NTFS_InitDevice, NTFS_Unmount, NULL};
27
28 tNTFS_Disk      gNTFS_Disks;
29
30 // === CODE ===
31 /**
32  * \brief Installs the NTFS driver
33  */
34 int NTFS_Install(char **Arguments)
35 {
36         VFS_AddDriver( &gNTFS_FSInfo );
37         return 0;
38 }
39
40 /**
41  * \brief Mount a NTFS volume
42  */
43 tVFS_Node *NTFS_InitDevice(const char *Device, const char **Options)
44 {
45         tNTFS_Disk      *disk;
46         tNTFS_BootSector        bs;
47         
48         disk = malloc( sizeof(tNTFS_Disk) );
49         
50         disk->FD = VFS_Open(Device, VFS_OPENFLAG_READ);
51         if(!disk->FD) {
52                 free(disk);
53                 return NULL;
54         }
55         
56         Log_Debug("FS_NTFS", "&bs = %p", &bs);
57         VFS_ReadAt(disk->FD, 0, 512, &bs);
58         
59         Log_Debug("FS_NTFS", "Jump = %02x%02x%02x",
60                 bs.Jump[0],
61                 bs.Jump[1],
62                 bs.Jump[2]);
63         Log_Debug("FS_NTFS", "SystemID = %02x%02x%02x%02x%02x%02x%02x%02x (%8C)",
64                 bs.SystemID[0], bs.SystemID[1], bs.SystemID[2], bs.SystemID[3],
65                 bs.SystemID[4], bs.SystemID[5], bs.SystemID[6], bs.SystemID[7],
66                 bs.SystemID
67                 );
68         Log_Debug("FS_NTFS", "BytesPerSector = %i", bs.BytesPerSector);
69         Log_Debug("FS_NTFS", "SectorsPerCluster = %i", bs.SectorsPerCluster);
70         Log_Debug("FS_NTFS", "MediaDescriptor = 0x%x", bs.MediaDescriptor);
71         Log_Debug("FS_NTFS", "SectorsPerTrack = %i", bs.SectorsPerTrack);
72         Log_Debug("FS_NTFS", "Heads = %i", bs.Heads);
73         Log_Debug("FS_NTFS", "TotalSectorCount = 0x%llx", bs.TotalSectorCount);
74         Log_Debug("FS_NTFS", "MFTStart = 0x%llx", bs.MFTStart);
75         Log_Debug("FS_NTFS", "MFTMirrorStart = 0x%llx", bs.MFTMirrorStart);
76         Log_Debug("FS_NTFS", "ClustersPerMFTRecord = %i", bs.ClustersPerMFTRecord);
77         Log_Debug("FS_NTFS", "ClustersPerIndexRecord = %i", bs.ClustersPerIndexRecord);
78         Log_Debug("FS_NTFS", "SerialNumber = 0x%llx", bs.SerialNumber);
79         
80         disk->ClusterSize = bs.BytesPerSector * bs.SectorsPerCluster;
81         Log_Debug("NTFS", "Cluster Size = %i KiB", disk->ClusterSize/1024);
82         disk->MFTBase = bs.MFTStart;
83         Log_Debug("NTFS", "MFT Base = %i", disk->MFTBase);
84         Log_Debug("NTFS", "TotalSectorCount = 0x%x", bs.TotalSectorCount);
85         
86         if( bs.ClustersPerMFTRecord < 0 ) {
87                 disk->MFTRecSize = 1 << (-bs.ClustersPerMFTRecord);
88         }
89         else {
90                 disk->MFTRecSize = bs.ClustersPerMFTRecord * disk->ClusterSize;
91         }
92         
93         disk->RootNode.Inode = 5;       // MFT Ent #5 is filesystem root
94         disk->RootNode.ImplPtr = disk;
95         
96         disk->RootNode.UID = 0;
97         disk->RootNode.GID = 0;
98         
99         disk->RootNode.NumACLs = 1;
100         disk->RootNode.ACLs = &gVFS_ACL_EveryoneRX;
101         
102         disk->RootNode.ReadDir = NTFS_ReadDir;
103         disk->RootNode.FindDir = NTFS_FindDir;
104         disk->RootNode.MkNod = NULL;
105         disk->RootNode.Link = NULL;
106         disk->RootNode.Relink = NULL;
107         disk->RootNode.Close = NULL;
108         
109         NTFS_DumpEntry(disk, 5);
110         
111         return &disk->RootNode;
112 }
113
114 /**
115  * \brief Unmount an NTFS Disk
116  */
117 void NTFS_Unmount(tVFS_Node *Node)
118 {
119         
120 }
121
122 /**
123  * \brief Dumps a MFT Entry
124  */
125 void NTFS_DumpEntry(tNTFS_Disk *Disk, Uint32 Entry)
126 {
127         void    *buf = malloc( Disk->MFTRecSize );
128         tNTFS_FILE_Header       *hdr = buf;
129         tNTFS_FILE_Attrib       *attr;
130          int    i;
131         
132         if(!buf) {
133                 Log_Warning("FS_NTFS", "malloc() fail!");
134                 return ;
135         }
136         
137         VFS_ReadAt( Disk->FD,
138                 Disk->MFTBase * Disk->ClusterSize + Entry * Disk->MFTRecSize,
139                 Disk->MFTRecSize,
140                 buf);
141         
142         Log_Debug("FS_NTFS", "MFT Entry #%i", Entry);
143         Log_Debug("FS_NTFS", "- Magic = 0x%08x (%4C)", hdr->Magic, &hdr->Magic);
144         Log_Debug("FS_NTFS", "- UpdateSequenceOfs = 0x%x", hdr->UpdateSequenceOfs);
145         Log_Debug("FS_NTFS", "- UpdateSequenceSize = 0x%x", hdr->UpdateSequenceSize);
146         Log_Debug("FS_NTFS", "- LSN = 0x%x", hdr->LSN);
147         Log_Debug("FS_NTFS", "- SequenceNumber = %i", hdr->SequenceNumber);
148         Log_Debug("FS_NTFS", "- HardLinkCount = %i", hdr->HardLinkCount);
149         Log_Debug("FS_NTFS", "- FirstAttribOfs = 0x%x", hdr->FirstAttribOfs);
150         Log_Debug("FS_NTFS", "- Flags = 0x%x", hdr->Flags);
151         Log_Debug("FS_NTFS", "- RecordSize = 0x%x", hdr->RecordSize);
152         Log_Debug("FS_NTFS", "- RecordSpace = 0x%x", hdr->RecordSpace);
153         Log_Debug("FS_NTFS", "- Reference = 0x%llx", hdr->Reference);
154         Log_Debug("FS_NTFS", "- NextAttribID = 0x%04x", hdr->NextAttribID);
155         
156         attr = (void*)( (char*)hdr + hdr->FirstAttribOfs );
157         i = 0;
158         while( (tVAddr)attr < (tVAddr)hdr + hdr->RecordSize )
159         {
160                 if(attr->Type == 0xFFFFFFFF)    break;
161                 Log_Debug("FS_NTFS", "- Attribute %i", i ++);
162                 Log_Debug("FS_NTFS", " > Type = 0x%x", attr->Type);
163                 Log_Debug("FS_NTFS", " > Size = 0x%x", attr->Size);
164                 Log_Debug("FS_NTFS", " > ResidentFlag = 0x%x", attr->ResidentFlag);
165                 Log_Debug("FS_NTFS", " > NameLength = %i", attr->NameLength);
166                 Log_Debug("FS_NTFS", " > NameOffset = 0x%x", attr->NameOffset);
167                 Log_Debug("FS_NTFS", " > Flags = 0x%x", attr->Flags);
168                 Log_Debug("FS_NTFS", " > AttributeID = 0x%x", attr->AttributeID);
169                 if( !attr->ResidentFlag ) {
170                         Log_Debug("FS_NTFS", " > AttribLen = 0x%x", attr->Resident.AttribLen);
171                         Log_Debug("FS_NTFS", " > AttribOfs = 0x%x", attr->Resident.AttribOfs);
172                         Log_Debug("FS_NTFS", " > IndexedFlag = 0x%x", attr->Resident.IndexedFlag);
173                         Log_Debug("FS_NTFS", " > Name = '%*C'", attr->NameLength, attr->Resident.Name);
174                         Debug_HexDump("FS_NTFS",
175                                 (void*)( (tVAddr)attr + attr->Resident.AttribOfs ),
176                                 attr->Resident.AttribLen
177                                 );
178                 }
179                 else {
180                         Log_Debug("FS_NTFS", " > StartingVCN = 0x%llx", attr->NonResident.StartingVCN);
181                         Log_Debug("FS_NTFS", " > LastVCN = 0x%llx", attr->NonResident.LastVCN);
182                         Log_Debug("FS_NTFS", " > DataRunOfs = 0x%x", attr->NonResident.DataRunOfs);
183                         Log_Debug("FS_NTFS", " > CompressionUnitSize = 0x%x", attr->NonResident.CompressionUnitSize);
184                         Log_Debug("FS_NTFS", " > AllocatedSize = 0x%llx", attr->NonResident.AllocatedSize);
185                         Log_Debug("FS_NTFS", " > RealSize = 0x%llx", attr->NonResident.RealSize);
186                         Log_Debug("FS_NTFS", " > InitiatedSize = 0x%llx", attr->NonResident.InitiatedSize);
187                         Log_Debug("FS_NTFS", " > Name = '%*C'", attr->NameLength, attr->NonResident.Name);
188                 }
189                 
190                 attr = (void*)( (tVAddr)attr + attr->Size );
191         }
192         
193         free(buf);
194 }

UCC git Repository :: git.ucc.asn.au