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
|
#include <kapi/interrupts.hpp>
#include <kstd/flat_map.hpp>
#include <kstd/memory.hpp>
#include <kstd/print.hpp>
#include <kstd/vector.hpp>
#include <algorithm>
#include <cstdint>
#include <utility>
namespace kapi::interrupts
{
namespace
{
auto constinit handlers = kstd::flat_map<std::uint32_t, kstd::vector<std::pair<handler *, kstd::weak_ptr<void>>>>{};
} // namespace
auto register_handler(std::uint32_t irq_number, handler & handler, kstd::weak_ptr<void> context) -> void
{
if (handlers.contains(irq_number))
{
auto & handler_list = handlers.at(irq_number);
handler_list.push_back(std::pair{&handler, context});
}
else
{
handlers.emplace(irq_number, kstd::vector{
std::pair{&handler, context}
});
}
}
auto unregister_handler(std::uint32_t irq_number, handler & handler) -> void
{
auto & handler_list = handlers.at(irq_number);
auto [first, last] =
std::ranges::remove_if(handler_list, [&](auto const & entry) { return entry.first == &handler; });
handler_list.erase(first, last);
}
auto dispatch(std::uint32_t irq_number) -> status
{
if (!handlers.contains(irq_number))
{
kstd::println(kstd::print_sink::stderr, "[OS:interrupts] No handler for IRQ{}", irq_number);
return status::unhandled;
}
auto & handler_list = handlers.at(irq_number);
if (handler_list.empty())
{
kstd::println(kstd::print_sink::stderr, "[OS:interrupts] No handler for IRQ{}", irq_number);
return status::unhandled;
}
for (auto const & [handler, context] : handler_list)
{
if (handler && handler->handle_interrupt(irq_number, context) == status::handled)
{
return status::handled;
}
}
return status::unhandled;
}
} // namespace kapi::interrupts
|