aboutsummaryrefslogtreecommitdiff
path: root/kernel/src/devices/storage/management.cpp
blob: d440bf0cb051c64f42c718f6af4bd3e8402f6b6e (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
#include "kernel/devices/storage/management.hpp"

#include "kapi/boot_modules.hpp"
#include "kapi/system.hpp"

#include "kapi/devices/device.hpp"
#include "kernel/devices/storage/controller.hpp"
#include "kernel/devices/storage/ram_disk/controller.hpp"

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

#include <algorithm>
#include <cstddef>
#include <optional>

namespace kernel::devices::storage
{
  namespace
  {
    constexpr size_t static MINORS_PER_DEVICE = 16;
    constexpr size_t static START_MAJOR = 1;
    constinit size_t static next_free_major = START_MAJOR;

    constinit auto static active_storage_management = std::optional<management>{};
  }  // namespace

  auto management::init() -> void
  {
    if (active_storage_management)
    {
      kapi::system::panic("[DEVICES] Storage management has already been initialized.");
    }
    active_storage_management.emplace(management{});

    auto current_ram_disk_controller =
        kstd::make_shared<ram_disk::controller>(&kapi::boot_modules::get_boot_module_registry());
    active_storage_management->add_controller(current_ram_disk_controller);

    std::ranges::for_each(active_storage_management->m_controllers, [](auto controller) { controller->probe(); });
  }

  auto management::get() -> management &
  {
    if (!active_storage_management)
    {
      kapi::system::panic("[DEVICES] Storage management has not been initialized.");
    }

    return *active_storage_management;
  }

  auto management::add_controller(kstd::shared_ptr<controller> const & controller) -> void
  {
    controller->set_ids(next_free_major++, MINORS_PER_DEVICE);
    m_controllers.push_back(controller);
  }

  auto management::all_controllers() const -> kstd::vector<kstd::shared_ptr<controller>> const &
  {
    return m_controllers;
  }

  auto management::device_by_major_minor(size_t major, size_t minor) -> kstd::shared_ptr<kapi::devices::device>
  {
    kstd::shared_ptr<kapi::devices::device> found = nullptr;

    std::ranges::find_if(m_controllers, [&](auto const & controller) {
      if (controller != nullptr && controller->major() == major)
      {
        found = controller->device_by_minor(minor);
        return found != nullptr;
      }
      return false;
    });

    return found;
  }

  auto management::determine_boot_device() -> kstd::shared_ptr<kapi::devices::device>
  {
    return device_by_major_minor(START_MAJOR, 0);
  }

}  // namespace kernel::devices::storage