ext2: write and truncate stub

Signed-off-by: Andy-Python-Programmer <[email protected]>

Author: Andy-Python-Programmer

src/aero_kernel/src/fs/ext2/disk.rs

@@ -1,3 +1,24 @@
+/*
+ * Copyright (C) 2021-2022 The Aero Project Developers.
+ *
+ * This file is part of The Aero Project.
+ *
+ * Aero is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Aero is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Aero. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+use bit_field::BitField;
+
 use crate::fs::inode;
 use crate::utils::CeilDiv;
 
@@ -197,6 +218,12 @@ impl INode {
         self.type_and_perm = file_type.bits() | (self.type_and_perm & mask);
     }
 
+    pub fn set_permissions(&mut self, permissions: u16) {
+        let mut val = self.type_and_perm;
+        val.set_bits(..13, permissions);
+        self.type_and_perm = val;
+    }
+
     pub fn file_type(&self) -> FileType {
         let ty = self.type_and_perm >> 12;
 

src/aero_kernel/src/fs/ext2/group_desc.rs

@@ -1,3 +1,22 @@
+/*
+ * Copyright (C) 2021-2022 The Aero Project Developers.
+ *
+ * This file is part of The Aero Project.
+ *
+ * Aero is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Aero is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Aero. If not, see <https://www.gnu.org/licenses/>.
+ */
+
 use core::mem::MaybeUninit;
 
 use alloc::boxed::Box;

src/aero_kernel/src/fs/ext2/mod.rs

@@ -84,7 +84,6 @@ impl INode {
         let block_size = filesystem.superblock.block_size();
 
         let mut progress = 0;
-
         let count = core::cmp::min(inode.size_lower as usize - offset, buffer.len());
 
         while progress < count {
@@ -113,7 +112,43 @@ impl INode {
         Ok(count)
     }
 
-    /// Allocates and appends a new block to the inode.
+    pub fn write(&self, offset: usize, buffer: &[u8]) -> super::Result<usize> {
+        let filesystem = self.fs.upgrade().unwrap();
+        let block_size = filesystem.superblock.block_size();
+
+        let mut progress = 0;
+        let count = buffer.len();
+
+        while progress < count {
+            let block = (offset + progress) / block_size;
+            let loc = (offset + progress) % block_size;
+
+            let mut chunk = count - progress;
+
+            if chunk > block_size - loc {
+                chunk = block_size - loc;
+            }
+
+            let mut block_index = self.get_block(block).unwrap() as usize;
+
+            if block_index == 0 {
+                block_index = self.append_block().unwrap();
+            }
+
+            filesystem
+                .block
+                .write(
+                    (block_index * block_size) + loc,
+                    &buffer[progress..progress + chunk],
+                )
+                .expect("inode: write failed");
+
+            progress += chunk;
+        }
+
+        Ok(count)
+    }
+
     pub fn append_block(&self) -> Option<usize> {
         let fs = self.fs.upgrade().expect("ext2: filesystem was dropped");
         let block_size = fs.superblock.block_size();
@@ -215,7 +250,11 @@ impl INode {
 
         {
             let mut inode = ext2_inode.inode.write();
+            **inode = disk::INode::default();
+
             inode.set_file_type(typ);
+            inode.set_permissions(0o755);
+
             inode.hl_count += 1;
         }
 
@@ -334,6 +373,18 @@ impl INodeInterface for INode {
         Ok(count)
     }
 
+    fn write_at(&self, offset: usize, usr_buffer: &[u8]) -> super::Result<usize> {
+        if !self.metadata()?.is_file() {
+            return Err(FileSystemError::NotSupported);
+        }
+
+        self.write(offset, usr_buffer)
+    }
+
+    fn truncate(&self, _size: usize) -> super::Result<()> {
+        Ok(())
+    }
+
     fn touch(&self, parent: DirCacheItem, name: &str) -> super::Result<DirCacheItem> {
         let inode = self.make_inode(name, FileType::File)?;
         Ok(DirEntry::new(parent, inode, name.to_string()))