aboutsummaryrefslogtreecommitdiff
path: root/kernel/src/filesystem/mount_table.cpp
blob: 74c18ca3bb3aba875200fae10e4f0e3396c16176 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#include <kernel/filesystem/mount_table.hpp>

#include <kernel/filesystem/dentry.hpp>
#include <kernel/filesystem/mount.hpp>

#include <kstd/memory>
#include <kstd/vector>

#include <algorithm>
#include <ranges>
#include <string_view>

namespace kernel::filesystem
{
  namespace
  {
    auto is_descendant_of(kstd::shared_ptr<mount> const & candidate, kstd::shared_ptr<mount> const & ancestor) -> bool
    {
      for (auto current = candidate; current; current = current->get_parent_mount())
      {
        if (current == ancestor)
        {
          return true;
        }
      }

      return false;
    }

    auto is_strict_prefix(std::string_view prefix, std::string_view path) -> bool
    {
      return prefix != "/" && path.starts_with(prefix) && path.size() > prefix.size() && path[prefix.size()] == '/';
    }

    auto is_visible_mount(kstd::shared_ptr<mount> const & candidate,
                          kstd::vector<kstd::shared_ptr<mount>> const & mounts) -> bool
    {
      return std::ranges::none_of(mounts, [&](auto const & other) {
        // TODO BA-FS26 really correct?
        return other != candidate &&
               is_strict_prefix(other->get_mount_path().view(), candidate->get_mount_path().view()) &&
               !is_descendant_of(candidate, other);
      });
    }
  }  // namespace

  auto mount_table::has_child_mounts(kstd::shared_ptr<mount> const & parent_mount) const -> bool
  {
    return std::ranges::any_of(
        m_mounts, [&parent_mount](auto const & mount) { return mount->get_parent_mount() == parent_mount; });
  }

  void mount_table::add_mount(kstd::shared_ptr<mount> const & mount)
  {
    m_mounts.push_back(mount);
    if (auto mount_dentry = mount->get_mount_dentry())
    {
      mount_dentry->set_flag(dentry::dentry_flags::is_mount_point);
    }
    if (auto root_dentry = mount->get_root_dentry())
    {
      root_dentry->set_flag(dentry::dentry_flags::is_mount_root);
    }
  }

  auto mount_table::remove_mount(std::string_view path) -> operation_result
  {
    auto mount_it = std::ranges::find_if(std::ranges::reverse_view(m_mounts), [&](auto const & mount) {
      return mount->get_mount_path() == path && is_visible_mount(mount, m_mounts);
    });

    if (mount_it == std::ranges::reverse_view(m_mounts).end())
    {
      return operation_result::mount_not_found;
    }

    auto const & mount = *mount_it;
    if (has_child_mounts(mount))
    {
      return operation_result::has_child_mounts;
    }

    mount->get_mount_dentry()->unset_flag(dentry::dentry_flags::is_mount_point);
    m_mounts.erase(std::ranges::find(m_mounts, mount));
    return operation_result::removed;
  }

  auto mount_table::find_exact_mount(std::string_view path) const -> kstd::shared_ptr<mount>
  {
    auto reversed_mounts = std::ranges::reverse_view(m_mounts);
    auto mount_it =
        std::ranges::find_if(reversed_mounts, [&](auto const & mount) { return mount->get_mount_path() == path; });
    return (mount_it != reversed_mounts.end()) ? *mount_it : nullptr;
  }
}  // namespace kernel::filesystem