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;
245 int bestMatch = -1, bestSize, bestBlock, bestOfs, bestNeedsSplit;
248 ENTER("pNode sName pChild",
251 blockData = malloc(disk->BlockSize);
253 // Read child inode (get's the file type)
254 Ext2_int_ReadInode(disk, Child->Inode, &inode);
256 // Create a stub entry
257 newEntry.inode = Child->Inode;
258 newEntry.name_len = strlen(Name);
259 newEntry.rec_len = ((newEntry.name_len+3)&~3) + EXT2_DIRENT_SIZE;
260 newEntry.type = inode.i_mode >> 12;
261 memcpy(newEntry.name, Name, newEntry.name_len);
263 // Read directory's inode
264 Ext2_int_ReadInode(disk, Node->Inode, &inode);
267 // Get a lock on the inode
268 //Ext2_int_LockInode(disk, Node->Inode);
269 Mutex_Acquire(&Node->Lock);
271 // if( !Node->Data ) {
275 // - Do this ourselves as it is a simple operation
276 base = inode.i_block[0] * disk->BlockSize;
277 VFS_ReadAt( disk->FD, base, disk->BlockSize, blockData );
282 dirent = blockData + ofs;
283 // Sanity Check the entry
284 if(ofs + dirent->rec_len > disk->BlockSize) {
286 "Directory entry %i of inode 0x%x extends over a block boundary",
287 nEntries, (Uint)Node->Inode);
291 LOG("Entry %i: %x %i bytes", nEntries, dirent->type, dirent->rec_len);
293 if(dirent->type == 0)
295 if( dirent->rec_len >= newEntry.rec_len
296 && (bestMatch == -1 || bestSize > dirent->rec_len) )
298 bestMatch = nEntries;
299 bestSize = dirent->rec_len;
305 // Non free - check name to avoid duplicates
308 LOG(" name='%.*s'", dirent->name_len, dirent->name);
309 if(strncmp(Name, dirent->name, dirent->name_len) == 0) {
310 //Ext2_int_UnlockInode(disk, Node->Inode);
311 Mutex_Release(&Node->Lock);
316 int spare_space = dirent->rec_len - (dirent->name_len + EXT2_DIRENT_SIZE);
317 if( spare_space > newEntry.rec_len
318 && (bestMatch == -1 || bestSize > spare_space) )
320 bestMatch = nEntries;
321 bestSize = spare_space;
329 // Increment the pointer
331 ofs += dirent->rec_len;
332 size -= dirent->rec_len;
333 if( size > 0 && ofs >= disk->BlockSize ) {
334 // Read the next block if needed
335 // BLOCK_DIR_OFS(Node->Data, block) = nEntries;
338 base = Ext2_int_GetBlockAddr(disk, inode.i_block, block);
339 VFS_ReadAt( disk->FD, base, disk->BlockSize, blockData );
343 LOG("bestMatch = %i", bestMatch);
344 // If EOF was reached with no space, check if we can fit one on the end
345 if( bestMatch < 0 && ofs + newEntry.rec_len < disk->BlockSize ) {
346 Node->Size += newEntry.rec_len;
347 Node->Flags |= VFS_FFLAG_DIRTY;
350 bestSize = newEntry.rec_len;
353 // Check if a free slot was found
357 base = Ext2_int_GetBlockAddr(disk, inode.i_block, bestBlock);
358 VFS_ReadAt( disk->FD, base, disk->BlockSize, blockData );
359 dirent = blockData + bestOfs;
360 // Shorten a pre-existing entry
363 dirent->rec_len = EXT2_DIRENT_SIZE + dirent->name_len;
364 bestOfs += dirent->rec_len;
365 //bestSize -= dirent->rec_len; // (not needed, bestSize is the spare space after)
366 dirent = blockData + bestOfs;
368 // Insert new file entry
369 memcpy(dirent, &newEntry, newEntry.rec_len);
370 // Create a new blank entry
371 if( bestSize != newEntry.rec_len )
373 bestOfs += newEntry.rec_len;
374 dirent = blockData + bestOfs;
376 dirent->rec_len = bestSize - newEntry.rec_len;
380 VFS_WriteAt( disk->FD, base, disk->BlockSize, blockData );
383 // Allocate block, Write
384 Uint32 newblock = Ext2_int_AllocateBlock(disk, base / disk->BlockSize);
385 Ext2_int_AppendBlock(disk, &inode, newblock);
386 base = newblock * disk->BlockSize;
387 Node->Size += newEntry.rec_len;
388 Node->Flags |= VFS_FFLAG_DIRTY;
389 memcpy(blockData, &newEntry, newEntry.rec_len);
390 memset(blockData + newEntry.rec_len, 0, disk->BlockSize - newEntry.rec_len);
391 VFS_WriteAt( disk->FD, base, disk->BlockSize, blockData );
395 Child->Flags |= VFS_FFLAG_DIRTY;
397 //Ext2_int_UnlockInode(disk, Node->Inode);
398 Mutex_Release(&Node->Lock);
402 Log_Warning("Ext2", "TODO: Impliment Ext2_Link");
407 // ---- INTERNAL FUNCTIONS ----
409 * \fn vfs_node *Ext2_int_CreateNode(tExt2_Disk *Disk, Uint InodeID)
410 * \brief Create a new VFS Node
412 tVFS_Node *Ext2_int_CreateNode(tExt2_Disk *Disk, Uint InodeID)
418 if( !Ext2_int_ReadInode(Disk, InodeID, &inode) )
421 if( (tmpNode = Inode_GetCache(Disk->CacheID, InodeID)) )
424 memset(&retNode, 0, sizeof(retNode));
427 retNode.Inode = InodeID;
428 retNode.ImplPtr = Disk;
429 retNode.ImplInt = inode.i_links_count;
432 retNode.Size = inode.i_size;
434 // Set Access Permissions
435 retNode.UID = inode.i_uid;
436 retNode.GID = inode.i_gid;
438 retNode.ACLs = VFS_UnixToAcessACL(inode.i_mode & 0777, inode.i_uid, inode.i_gid);
440 // Set Function Pointers
441 retNode.Type = &gExt2_FileType;
443 switch(inode.i_mode & EXT2_S_IFMT)
447 retNode.Flags = VFS_FFLAG_SYMLINK;
452 retNode.Size |= (Uint64)inode.i_dir_acl << 32;
456 retNode.Type = &gExt2_DirType;
457 retNode.Flags = VFS_FFLAG_DIRECTORY;
458 retNode.Data = calloc( sizeof(Uint16), DivUp(retNode.Size, Disk->BlockSize) );
460 // Unknown, Write protect it to be safe
462 retNode.Flags = VFS_FFLAG_READONLY;
467 retNode.ATime = inode.i_atime * 1000;
468 retNode.MTime = inode.i_mtime * 1000;
469 retNode.CTime = inode.i_ctime * 1000;
471 // Save in node cache and return saved node
472 return Inode_CacheNode(Disk->CacheID, &retNode);
475 int Ext2_int_WritebackNode(tExt2_Disk *Disk, tVFS_Node *Node)
477 Log_Warning("Ext2","TODO: Impliment Ext2_int_WritebackNode");