Added FS functions to Archive and Archive_RomFS

3 files changed, 57 insertions, 8 deletions

diff --git a/src/core/file_sys/archive.h b/src/core/file_sys/archive.h
index ed2d83640..ac5630bea 100644
--- a/src/core/file_sys/archive.h
+++ b/src/core/file_sys/archive.h
@@ -37,18 +37,33 @@ public:
 
     /**
      * Read data from the archive
-     * @param offset Offset in bytes to start reading archive from
-     * @param length Length in bytes to read data from archive
+     * @param offset Offset in bytes to start reading data from
+     * @param length Length in bytes of data to read from archive
      * @param buffer Buffer to read data into
      * @return Number of bytes read
      */
     virtual size_t Read(const u64 offset, const u32 length, u8* buffer) const = 0;
 
     /**
+     * Write data to the archive
+     * @param offset Offset in bytes to start writing data to
+     * @param length Length in bytes of data to write to archive
+     * @param buffer Buffer to write data from
+     * @param flush  The flush parameters (0 == do not flush)
+     * @return Number of bytes written
+     */
+    virtual size_t Write(const u64 offset, const u32 length, const u32 flush, u8* buffer) = 0;
+
+    /**
      * Get the size of the archive in bytes
      * @return Size of the archive in bytes
      */
     virtual size_t GetSize() const = 0;
+    
+    /**
+     * Set the size of the archive in bytes
+     */
+    virtual void SetSize(const u64 size) = 0;
 };
 
 } // namespace FileSys
diff --git a/src/core/file_sys/archive_romfs.cpp b/src/core/file_sys/archive_romfs.cpp
index fd84b9c8c..dc3fb1807 100644
--- a/src/core/file_sys/archive_romfs.cpp
+++ b/src/core/file_sys/archive_romfs.cpp
@@ -23,8 +23,8 @@ Archive_RomFS::~Archive_RomFS() {
 
 /**
  * Read data from the archive
- * @param offset Offset in bytes to start reading archive from
- * @param length Length in bytes to read data from archive
+ * @param offset Offset in bytes to start reading data from
+ * @param length Length in bytes of data to read from archive
  * @param buffer Buffer to read data into
  * @return Number of bytes read
  */
@@ -35,12 +35,31 @@ size_t Archive_RomFS::Read(const u64 offset, const u32 length, u8* buffer) const
 }
 
 /**
+ * Write data to the archive
+ * @param offset Offset in bytes to start writing data to
+ * @param length Length in bytes of data to write to archive
+ * @param buffer Buffer to write data from
+ * @param flush  The flush parameters (0 == do not flush)
+ * @return Number of bytes written
+ */
+size_t Archive_RomFS::Write(const u64 offset, const u32 length, const u32 flush, u8* buffer) {
+    ERROR_LOG(FILESYS, "Attempted to write to ROMFS.");
+    return 0;
+}
+
+/**
  * Get the size of the archive in bytes
  * @return Size of the archive in bytes
  */
 size_t Archive_RomFS::GetSize() const {
-    ERROR_LOG(FILESYS, "(UNIMPLEMENTED)");
-    return 0;
+    return sizeof(u8) * raw_data.size();
+}
+
+/**
+ * Set the size of the archive in bytes
+ */
+void Archive_RomFS::SetSize(const u64 size) {
+    ERROR_LOG(FILESYS, "Attempted to set the size of ROMFS");
 }
 
 } // namespace FileSys
diff --git a/src/core/file_sys/archive_romfs.h b/src/core/file_sys/archive_romfs.h
index 8a31190a9..e9ed6f77a 100644
--- a/src/core/file_sys/archive_romfs.h
+++ b/src/core/file_sys/archive_romfs.h
@@ -30,18 +30,33 @@ public:
 
     /**
      * Read data from the archive
-     * @param offset Offset in bytes to start reading archive from
-     * @param length Length in bytes to read data from archive
+     * @param offset Offset in bytes to start reading data from
+     * @param length Length in bytes of data to read from archive
      * @param buffer Buffer to read data into
      * @return Number of bytes read
      */
     size_t Read(const u64 offset, const u32 length, u8* buffer) const override;
 
     /**
+     * Write data to the archive
+     * @param offset Offset in bytes to start writing data to
+     * @param length Length in bytes of data to write to archive
+     * @param buffer Buffer to write data from
+     * @param flush  The flush parameters (0 == do not flush)
+     * @return Number of bytes written
+     */
+    size_t Write(const u64 offset, const u32 length, const u32 flush, u8* buffer) override;
+
+    /**
      * Get the size of the archive in bytes
      * @return Size of the archive in bytes
      */
     size_t GetSize() const override;
+    
+    /**
+     * Set the size of the archive in bytes
+     */
+    void SetSize(const u64 size) override;
 
 private:
     std::vector<u8> raw_data;