3 * - By John Hodge (thePowersGang)
10 #include "ext2_common.h"
13 #define BLOCK_DIR_OFS(_data, _block) (((Uint16*)(_data))[(_block)])
16 char *Ext2_ReadDir(tVFS_Node *Node, int Pos);
17 tVFS_Node *Ext2_FindDir(tVFS_Node *Node, const char *FileName);
18 int Ext2_MkNod(tVFS_Node *Node, const char *Name, Uint Flags);
19 int Ext2_Unlink(tVFS_Node *Node, const char *OldName);
20 int Ext2_Link(tVFS_Node *Parent, const char *Name, tVFS_Node *Node);
22 tVFS_Node *Ext2_int_CreateNode(tExt2_Disk *Disk, Uint InodeId);
25 tVFS_NodeType gExt2_DirType = {
26 .TypeName = "ext2-dir",
27 .ReadDir = Ext2_ReadDir,
28 .FindDir = Ext2_FindDir,
30 .Unlink = Ext2_Unlink,
32 .Close = Ext2_CloseFile
34 tVFS_NodeType gExt2_FileType = {
35 .TypeName = "ext2-file",
38 .Close = Ext2_CloseFile
43 * \brief Reads a directory entry
44 * \param Node Directory node
45 * \param Pos Position of desired element
47 char *Ext2_ReadDir(tVFS_Node *Node, int Pos)
51 Uint64 Base; // Block's Base Address
55 tExt2_Disk *disk = Node->ImplPtr;
58 ENTER("pNode iPos", Node, Pos);
60 // Read directory's inode
61 Ext2_int_ReadInode(disk, Node->Inode, &inode);
64 LOG("inode={.i_block[0]= 0x%x, .i_size=0x%x}", inode.i_block[0], inode.i_size);
68 // - Do this ourselves as it is a simple operation
69 Base = inode.i_block[0] * disk->BlockSize;
71 while(Pos -- && size > 0 && size <= inode.i_size)
73 VFS_ReadAt( disk->FD, Base+ofs, sizeof(tExt2_DirEnt), &dirent);
74 ofs += dirent.rec_len;
75 size -= dirent.rec_len;
78 if(ofs >= disk->BlockSize) {
80 if( ofs > disk->BlockSize ) {
81 Log_Warning("EXT2", "Directory Entry %i of inode %i extends over a block boundary, ignoring",
82 entNum-1, Node->Inode);
85 Base = Ext2_int_GetBlockAddr( disk, inode.i_block, block );
93 // Check for the end of the list
94 if(size <= 0 || size > inode.i_size) {
100 VFS_ReadAt( disk->FD, Base+ofs, sizeof(tExt2_DirEnt), &dirent );
101 LOG("dirent={.rec_len=%i,.inode=0x%x,.name_len=%i}",
102 dirent.rec_len, dirent.inode, dirent.name_len);
103 dirent.name[ dirent.name_len ] = '\0'; // Cap off string
105 if( dirent.name_len == 0 ) {
106 LEAVE('p', VFS_SKIP);
110 // Ignore . and .. (these are done in the VFS)
111 if( (dirent.name[0] == '.' && dirent.name[1] == '\0')
112 || (dirent.name[0] == '.' && dirent.name[1] == '.' && dirent.name[2]=='\0')) {
113 LEAVE('p', VFS_SKIP);
114 return VFS_SKIP; // Skip
117 LEAVE('s', dirent.name);
119 return strdup(dirent.name);
123 * \brief Gets information about a file
124 * \param Node Parent Node
125 * \param Filename Name of wanted file
126 * \return VFS Node of file
128 tVFS_Node *Ext2_FindDir(tVFS_Node *Node, const char *Filename)
130 tExt2_Disk *disk = Node->ImplPtr;
133 Uint64 Base; // Block's Base Address
138 int filenameLen = strlen(Filename);
140 // Read directory's inode
141 Ext2_int_ReadInode(disk, Node->Inode, &inode);
145 // - Do this ourselves as it is a simple operation
146 Base = inode.i_block[0] * disk->BlockSize;
150 VFS_ReadAt( disk->FD, Base+ofs, sizeof(tExt2_DirEnt), &dirent);
151 // TODO: Possible overrun if name_len == 255?
152 dirent.name[ dirent.name_len ] = '\0'; // Cap off string
153 // If it matches, create a node and return it
154 if(dirent.name_len == filenameLen && strcmp(dirent.name, Filename) == 0)
155 return Ext2_int_CreateNode( disk, dirent.inode );
156 // Increment pointers
157 ofs += dirent.rec_len;
158 size -= dirent.rec_len;
161 // Check for end of block
162 if(ofs >= disk->BlockSize) {
164 if( ofs > disk->BlockSize ) {
165 Log_Warning("EXT2", "Directory Entry %i of inode %i extends over a block boundary, ignoring",
166 entNum-1, Node->Inode);
169 Base = Ext2_int_GetBlockAddr( disk, inode.i_block, block );
177 * \fn int Ext2_MkNod(tVFS_Node *Parent, const char *Name, Uint Flags)
178 * \brief Create a new node
180 int Ext2_MkNod(tVFS_Node *Parent, const char *Name, Uint Flags)
182 ENTER("pParent sName xFlags", Parent, Name, Flags);
184 Uint64 inodeNum = Ext2_int_AllocateInode(Parent->ImplPtr, Parent->Inode);
185 if( inodeNum == 0 ) {
186 LOG("Inode allocation failed");
190 tVFS_Node *child = Ext2_int_CreateNode(Parent->ImplPtr, inodeNum);
192 Ext2_int_DereferenceInode(Parent->ImplPtr, inodeNum);
193 Log_Warning("Ext2", "Ext2_MkNod - Node creation failed");
198 child->Flags = Flags & (VFS_FFLAG_DIRECTORY|VFS_FFLAG_SYMLINK|VFS_FFLAG_READONLY);
199 child->UID = Threads_GetUID();
200 child->GID = Threads_GetGID();
205 child->ImplInt = 0; // ImplInt is the link count
208 int rv = Ext2_Link(Parent, Name, child);
209 child->Type->Close(child);
215 * \brief Rename a file
216 * \param Node This (directory) node
217 * \param OldName Old name of file
218 * \param NewName New name for file
219 * \return Boolean Failure - See ::tVFS_Node.Unlink for info
221 int Ext2_Unlink(tVFS_Node *Node, const char *OldName)
223 Log_Warning("Ext2", "TODO: Impliment Ext2_Unlink");
228 * \brief Links an existing node to a new name
229 * \param Parent Parent (directory) node
230 * \param Name New name for the node
231 * \param Node Node to link
232 * \return Boolean Failure - See ::tVFS_Node.Link for info
234 int Ext2_Link(tVFS_Node *Node, const char *Name, tVFS_Node *Child)
237 tExt2_Disk *disk = Node->ImplPtr;
239 tExt2_DirEnt *dirent;
240 tExt2_DirEnt newEntry;
241 Uint64 base; // Block's Base Address
242 int block = 0, ofs = 0;
246 int bestSize=0, bestBlock=0, bestOfs=0, bestNeedsSplit=0;
249 ENTER("pNode sName pChild",
252 blockData = malloc(disk->BlockSize);
254 // Read child inode (get's the file type)
255 Ext2_int_ReadInode(disk, Child->Inode, &inode);
257 // Create a stub entry
258 newEntry.inode = Child->Inode;
259 newEntry.name_len = strlen(Name);
260 newEntry.rec_len = ((newEntry.name_len+3)&~3) + EXT2_DIRENT_SIZE;
261 newEntry.type = inode.i_mode >> 12;
262 memcpy(newEntry.name, Name, newEntry.name_len);
264 // Read directory's inode
265 Ext2_int_ReadInode(disk, Node->Inode, &inode);
268 // Get a lock on the inode
269 //Ext2_int_LockInode(disk, Node->Inode);
270 Mutex_Acquire(&Node->Lock);
272 // if( !Node->Data ) {
276 // - Do this ourselves as it is a simple operation
277 base = inode.i_block[0] * disk->BlockSize;
278 VFS_ReadAt( disk->FD, base, disk->BlockSize, blockData );
284 dirent = blockData + ofs;
285 // Sanity Check the entry
286 if(ofs + dirent->rec_len > disk->BlockSize) {
288 "Directory entry %i of inode 0x%x extends over a block boundary",
289 nEntries, (Uint)Node->Inode);
293 LOG("Entry %i: %x %i bytes", nEntries, dirent->type, dirent->rec_len);
295 if(dirent->type == 0)
297 if( dirent->rec_len >= newEntry.rec_len
298 && (bestMatch == -1 || bestSize > dirent->rec_len) )
300 bestMatch = nEntries;
301 bestSize = dirent->rec_len;
307 // Non free - check name to avoid duplicates
310 LOG(" name='%.*s'", dirent->name_len, dirent->name);
311 if(strncmp(Name, dirent->name, dirent->name_len) == 0) {
312 //Ext2_int_UnlockInode(disk, Node->Inode);
313 Mutex_Release(&Node->Lock);
318 int spare_space = dirent->rec_len - (dirent->name_len + EXT2_DIRENT_SIZE);
319 if( spare_space > newEntry.rec_len
320 && (bestMatch == -1 || bestSize > spare_space) )
322 bestMatch = nEntries;
323 bestSize = spare_space;
331 // Increment the pointer
333 ofs += dirent->rec_len;
334 size -= dirent->rec_len;
335 if( size > 0 && ofs >= disk->BlockSize ) {
336 // Read the next block if needed
337 // BLOCK_DIR_OFS(Node->Data, block) = nEntries;
340 base = Ext2_int_GetBlockAddr(disk, inode.i_block, block);
341 VFS_ReadAt( disk->FD, base, disk->BlockSize, blockData );
345 LOG("bestMatch = %i", bestMatch);
346 // If EOF was reached with no space, check if we can fit one on the end
347 if( bestMatch < 0 && ofs + newEntry.rec_len < disk->BlockSize ) {
348 Node->Size += newEntry.rec_len;
349 Node->Flags |= VFS_FFLAG_DIRTY;
352 bestSize = newEntry.rec_len;
355 // Check if a free slot was found
359 base = Ext2_int_GetBlockAddr(disk, inode.i_block, bestBlock);
360 VFS_ReadAt( disk->FD, base, disk->BlockSize, blockData );
361 dirent = blockData + bestOfs;
362 // Shorten a pre-existing entry
365 dirent->rec_len = EXT2_DIRENT_SIZE + dirent->name_len;
366 bestOfs += dirent->rec_len;
367 //bestSize -= dirent->rec_len; // (not needed, bestSize is the spare space after)
368 dirent = blockData + bestOfs;
370 // Insert new file entry
371 memcpy(dirent, &newEntry, newEntry.rec_len);
372 // Create a new blank entry
373 if( bestSize != newEntry.rec_len )
375 bestOfs += newEntry.rec_len;
376 dirent = blockData + bestOfs;
378 dirent->rec_len = bestSize - newEntry.rec_len;
382 VFS_WriteAt( disk->FD, base, disk->BlockSize, blockData );
385 // Allocate block, Write
386 Uint32 newblock = Ext2_int_AllocateBlock(disk, base / disk->BlockSize);
387 Ext2_int_AppendBlock(disk, &inode, newblock);
388 base = newblock * disk->BlockSize;
389 Node->Size += newEntry.rec_len;
390 Node->Flags |= VFS_FFLAG_DIRTY;
391 memcpy(blockData, &newEntry, newEntry.rec_len);
392 memset(blockData + newEntry.rec_len, 0, disk->BlockSize - newEntry.rec_len);
393 VFS_WriteAt( disk->FD, base, disk->BlockSize, blockData );
397 Child->Flags |= VFS_FFLAG_DIRTY;
399 //Ext2_int_UnlockInode(disk, Node->Inode);
400 Mutex_Release(&Node->Lock);
404 Log_Warning("Ext2", "TODO: Impliment Ext2_Link");
409 // ---- INTERNAL FUNCTIONS ----
411 * \fn vfs_node *Ext2_int_CreateNode(tExt2_Disk *Disk, Uint InodeID)
412 * \brief Create a new VFS Node
414 tVFS_Node *Ext2_int_CreateNode(tExt2_Disk *Disk, Uint InodeID)
420 if( !Ext2_int_ReadInode(Disk, InodeID, &inode) )
423 if( (tmpNode = Inode_GetCache(Disk->CacheID, InodeID)) )
426 memset(&retNode, 0, sizeof(retNode));
429 retNode.Inode = InodeID;
430 retNode.ImplPtr = Disk;
431 retNode.ImplInt = inode.i_links_count;
434 retNode.Size = inode.i_size;
436 // Set Access Permissions
437 retNode.UID = inode.i_uid;
438 retNode.GID = inode.i_gid;
440 retNode.ACLs = VFS_UnixToAcessACL(inode.i_mode & 0777, inode.i_uid, inode.i_gid);
442 // Set Function Pointers
443 retNode.Type = &gExt2_FileType;
445 switch(inode.i_mode & EXT2_S_IFMT)
449 retNode.Flags = VFS_FFLAG_SYMLINK;
454 retNode.Size |= (Uint64)inode.i_dir_acl << 32;
458 retNode.Type = &gExt2_DirType;
459 retNode.Flags = VFS_FFLAG_DIRECTORY;
460 retNode.Data = calloc( sizeof(Uint16), DivUp(retNode.Size, Disk->BlockSize) );
462 // Unknown, Write protect it to be safe
464 retNode.Flags = VFS_FFLAG_READONLY;
469 retNode.ATime = inode.i_atime * 1000;
470 retNode.MTime = inode.i_mtime * 1000;
471 retNode.CTime = inode.i_ctime * 1000;
473 // Save in node cache and return saved node
474 return Inode_CacheNode(Disk->CacheID, &retNode);
477 int Ext2_int_WritebackNode(tExt2_Disk *Disk, tVFS_Node *Node)
479 Log_Warning("Ext2","TODO: Impliment Ext2_int_WritebackNode");