forked from dolphin-emu/dolphin
		
	
		
			
				
	
	
		
			116 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			116 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
// Copyright 2008 Dolphin Emulator Project
 | 
						|
// Licensed under GPLv2+
 | 
						|
// Refer to the license.txt file included.
 | 
						|
 | 
						|
#pragma once
 | 
						|
 | 
						|
#include <cstddef>
 | 
						|
#include <map>
 | 
						|
#include <memory>
 | 
						|
#include <optional>
 | 
						|
#include <string>
 | 
						|
#include <vector>
 | 
						|
 | 
						|
#include "Common/CommonTypes.h"
 | 
						|
#include "DiscIO/Filesystem.h"
 | 
						|
 | 
						|
namespace DiscIO
 | 
						|
{
 | 
						|
class Volume;
 | 
						|
struct Partition;
 | 
						|
 | 
						|
class FileInfoGCWii : public FileInfo
 | 
						|
{
 | 
						|
public:
 | 
						|
  // None of the constructors take ownership of FST pointers
 | 
						|
 | 
						|
  // Set everything manually
 | 
						|
  FileInfoGCWii(const u8* fst, u8 offset_shift, u32 index, u32 total_file_infos);
 | 
						|
  // For the root object only
 | 
						|
  FileInfoGCWii(const u8* fst, u8 offset_shift);
 | 
						|
  // Copy another object
 | 
						|
  FileInfoGCWii(const FileInfoGCWii& file_info) = default;
 | 
						|
  // Copy data that is common to the whole file system
 | 
						|
  FileInfoGCWii(const FileInfoGCWii& file_info, u32 index);
 | 
						|
 | 
						|
  ~FileInfoGCWii() override;
 | 
						|
 | 
						|
  std::unique_ptr<FileInfo> clone() const override;
 | 
						|
  const_iterator begin() const override;
 | 
						|
  const_iterator end() const override;
 | 
						|
 | 
						|
  u64 GetOffset() const override;
 | 
						|
  u32 GetSize() const override;
 | 
						|
  bool IsDirectory() const override;
 | 
						|
  u32 GetTotalChildren() const override;
 | 
						|
  std::string GetName() const override;
 | 
						|
  std::string GetPath() const override;
 | 
						|
 | 
						|
  bool IsValid(u64 fst_size, const FileInfoGCWii& parent_directory) const;
 | 
						|
 | 
						|
protected:
 | 
						|
  uintptr_t GetAddress() const override;
 | 
						|
  FileInfo& operator++() override;
 | 
						|
 | 
						|
private:
 | 
						|
  enum class EntryProperty
 | 
						|
  {
 | 
						|
    // NAME_OFFSET's lower 3 bytes are the name's offset within the name table.
 | 
						|
    // NAME_OFFSET's upper 1 byte is 1 for directories and 0 for files.
 | 
						|
    NAME_OFFSET = 0,
 | 
						|
    // For files, FILE_OFFSET is the file offset in the partition,
 | 
						|
    // and for directories, it's the FST index of the parent directory.
 | 
						|
    // The root directory has its parent directory index set to 0.
 | 
						|
    FILE_OFFSET = 1,
 | 
						|
    // For files, FILE_SIZE is the file size, and for directories,
 | 
						|
    // it's the FST index of the next entry that isn't in the directory.
 | 
						|
    FILE_SIZE = 2
 | 
						|
  };
 | 
						|
 | 
						|
  // For files, returns the index of the next item. For directories,
 | 
						|
  // returns the index of the next item that isn't inside it.
 | 
						|
  u32 GetNextIndex() const;
 | 
						|
  // Returns one of the three properties of this FST entry.
 | 
						|
  // Read the comments in EntryProperty for details.
 | 
						|
  u32 Get(EntryProperty entry_property) const;
 | 
						|
  // Returns the name offset, excluding the directory identification byte
 | 
						|
  u64 GetNameOffset() const;
 | 
						|
 | 
						|
  const u8* m_fst;
 | 
						|
  u8 m_offset_shift;
 | 
						|
  u32 m_index;
 | 
						|
  u32 m_total_file_infos;
 | 
						|
};
 | 
						|
 | 
						|
class FileSystemGCWii : public FileSystem
 | 
						|
{
 | 
						|
public:
 | 
						|
  FileSystemGCWii(const Volume* volume, const Partition& partition);
 | 
						|
  ~FileSystemGCWii() override;
 | 
						|
 | 
						|
  bool IsValid() const override { return m_valid; }
 | 
						|
  const FileInfo& GetRoot() const override;
 | 
						|
  std::unique_ptr<FileInfo> FindFileInfo(const std::string& path) const override;
 | 
						|
  std::unique_ptr<FileInfo> FindFileInfo(u64 disc_offset) const override;
 | 
						|
 | 
						|
  u64 ReadFile(const FileInfo* file_info, u8* buffer, u64 max_buffer_size,
 | 
						|
               u64 offset_in_file) const override;
 | 
						|
  bool ExportFile(const FileInfo* file_info, const std::string& export_filename) const override;
 | 
						|
  bool ExportApploader(const std::string& export_folder) const override;
 | 
						|
  bool ExportDOL(const std::string& export_folder) const override;
 | 
						|
  std::optional<u64> GetBootDOLOffset() const override;
 | 
						|
  std::optional<u32> GetBootDOLSize(u64 dol_offset) const override;
 | 
						|
 | 
						|
private:
 | 
						|
  bool m_valid;
 | 
						|
  u32 m_offset_shift;
 | 
						|
  std::vector<u8> m_file_system_table;
 | 
						|
  FileInfoGCWii m_root;
 | 
						|
  // Maps the end offset of files to FST indexes
 | 
						|
  mutable std::map<u64, u32> m_offset_file_info_cache;
 | 
						|
 | 
						|
  std::unique_ptr<FileInfo> FindFileInfo(const std::string& path, const FileInfo& file_info) const;
 | 
						|
};
 | 
						|
 | 
						|
}  // namespace
 |