Ext2FileSystem.h 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. #pragma once
  2. #include "DiskBackedFileSystem.h"
  3. #include "UnixTypes.h"
  4. #include <AK/Buffer.h>
  5. #include <AK/OwnPtr.h>
  6. #include <AK/Lock.h>
  7. #include "ext2_fs.h"
  8. struct ext2_group_desc;
  9. struct ext2_inode;
  10. struct ext2_super_block;
  11. class Ext2FS;
  12. class Ext2FSInode final : public Inode {
  13. friend class Ext2FS;
  14. public:
  15. virtual ~Ext2FSInode() override;
  16. size_t size() const { return m_raw_inode.i_size; }
  17. bool is_symlink() const { return isSymbolicLink(m_raw_inode.i_mode); }
  18. // ^Inode (Retainable magic)
  19. virtual void one_retain_left() override;
  20. private:
  21. // ^Inode
  22. virtual ssize_t read_bytes(Unix::off_t, size_t, byte* buffer, FileDescriptor*) override;
  23. virtual InodeMetadata metadata() const override;
  24. virtual bool traverse_as_directory(Function<bool(const FS::DirectoryEntry&)>) override;
  25. virtual InodeIdentifier lookup(const String& name) override;
  26. virtual String reverse_lookup(InodeIdentifier) override;
  27. virtual void flush_metadata() override;
  28. virtual bool write(const ByteBuffer&) override;
  29. virtual bool add_child(InodeIdentifier child_id, const String& name, byte file_type, int& error) override;
  30. virtual bool remove_child(const String& name, int& error) override;
  31. virtual RetainPtr<Inode> parent() const override;
  32. virtual int set_atime(Unix::time_t) override;
  33. virtual int set_ctime(Unix::time_t) override;
  34. virtual int set_mtime(Unix::time_t) override;
  35. virtual int increment_link_count() override;
  36. virtual int decrement_link_count() override;
  37. void populate_lookup_cache();
  38. Ext2FS& fs();
  39. const Ext2FS& fs() const;
  40. Ext2FSInode(Ext2FS&, unsigned index, const ext2_inode&);
  41. Lock m_lock;
  42. Vector<unsigned> m_block_list;
  43. HashMap<String, unsigned> m_lookup_cache;
  44. ext2_inode m_raw_inode;
  45. mutable InodeIdentifier m_parent_id;
  46. };
  47. class Ext2FS final : public DiskBackedFS {
  48. friend class Ext2FSInode;
  49. public:
  50. static RetainPtr<Ext2FS> create(RetainPtr<DiskDevice>&&);
  51. virtual ~Ext2FS() override;
  52. virtual bool initialize() override;
  53. private:
  54. typedef unsigned BlockIndex;
  55. typedef unsigned GroupIndex;
  56. typedef unsigned InodeIndex;
  57. explicit Ext2FS(RetainPtr<DiskDevice>&&);
  58. const ext2_super_block& super_block() const;
  59. const ext2_group_desc& group_descriptor(unsigned groupIndex) const;
  60. unsigned first_block_of_group(unsigned groupIndex) const;
  61. unsigned inodes_per_block() const;
  62. unsigned inodes_per_group() const;
  63. unsigned blocks_per_group() const;
  64. unsigned inode_size() const;
  65. bool write_ext2_inode(unsigned, const ext2_inode&);
  66. ByteBuffer read_block_containing_inode(unsigned inode, unsigned& blockIndex, unsigned& offset) const;
  67. ByteBuffer read_super_block() const;
  68. bool write_super_block(const ext2_super_block&);
  69. virtual const char* class_name() const override;
  70. virtual InodeIdentifier root_inode() const override;
  71. virtual RetainPtr<Inode> create_inode(InodeIdentifier parentInode, const String& name, Unix::mode_t, unsigned size, int& error) override;
  72. virtual RetainPtr<Inode> create_directory(InodeIdentifier parentInode, const String& name, Unix::mode_t, int& error) override;
  73. virtual RetainPtr<Inode> get_inode(InodeIdentifier) const override;
  74. unsigned allocate_inode(unsigned preferredGroup, unsigned expectedSize);
  75. Vector<BlockIndex> allocate_blocks(unsigned group, unsigned count);
  76. unsigned group_index_from_inode(unsigned) const;
  77. Vector<unsigned> block_list_for_inode(const ext2_inode&, bool include_block_list_blocks = false) const;
  78. void dump_block_bitmap(unsigned groupIndex) const;
  79. void dump_inode_bitmap(unsigned groupIndex) const;
  80. template<typename F> void traverse_inode_bitmap(unsigned groupIndex, F) const;
  81. template<typename F> void traverse_block_bitmap(unsigned groupIndex, F) const;
  82. bool add_inode_to_directory(InodeIndex parent, InodeIndex child, const String& name, byte fileType, int& error);
  83. bool write_directory_inode(unsigned directoryInode, Vector<DirectoryEntry>&&);
  84. bool get_inode_allocation_state(InodeIndex) const;
  85. bool set_inode_allocation_state(unsigned inode, bool);
  86. bool set_block_allocation_state(GroupIndex, BlockIndex, bool);
  87. void uncache_inode(InodeIndex);
  88. unsigned m_blockGroupCount { 0 };
  89. mutable ByteBuffer m_cached_super_block;
  90. mutable ByteBuffer m_cached_group_descriptor_table;
  91. mutable Lock m_inode_cache_lock;
  92. mutable HashMap<BlockIndex, RetainPtr<Ext2FSInode>> m_inode_cache;
  93. };
  94. inline Ext2FS& Ext2FSInode::fs()
  95. {
  96. return static_cast<Ext2FS&>(Inode::fs());
  97. }
  98. inline const Ext2FS& Ext2FSInode::fs() const
  99. {
  100. return static_cast<const Ext2FS&>(Inode::fs());
  101. }