aboutsummaryrefslogtreecommitdiff
path: root/kernel/kapi/devices/power.cpp
blob: 3bc668480054b0b358e74c580df3ee325b7518f1 (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
#include <kapi/devices/power.hpp>

#include <kapi/devices.hpp>

#include <kstd/memory.hpp>
#include <kstd/print.hpp>
#include <kstd/result.hpp>
#include <kstd/vector.hpp>

#include <ranges>

namespace kapi::devices
{

  namespace
  {
    auto resume_subtree(device & node) -> void;

    // NOLINTNEXTLINE(misc-no-recursion)
    auto suspend_subtree(device & node) -> kstd::result<void>
    {
      auto suspended_children = kstd::vector<kstd::shared_ptr<device>>{};

      if (auto as_bus = node.facet<bus>())
      {
        for (auto const & child : as_bus->children())
        {
          auto result = suspend_subtree(*child);
          if (!result)
          {
            for (auto const & already_suspended : std::views::reverse(suspended_children))
            {
              resume_subtree(*already_suspended);
            }

            return result;
          }

          suspended_children.push_back(child);
        }
      }

      if (auto driver = node.bound_driver())
      {
        auto result = driver->suspend(node);
        if (!result)
        {
          for (auto const & already_suspended : std::views::reverse(suspended_children))
          {
            resume_subtree(*already_suspended);
          }

          return result;
        }
      }

      return kstd::success();
    }

    // NOLINTNEXTLINE(misc-no-recursion)
    auto resume_subtree(device & node) -> void
    {
      if (auto driver = node.bound_driver())
      {
        auto result = driver->resume(node);
        if (!result)
        {
          kstd::println(kstd::print_sink::stderr, "[OS:PM] failed to resume device '{}': {}", node.name(),
                        result.error().message());
        }
      }

      if (auto as_bus = node.facet<bus>())
      {
        for (auto const & child : as_bus->children())
        {
          resume_subtree(*child);
        }
      }
    }
  }  // namespace

  auto suspend_tree(bus & root) -> kstd::result<void>
  {
    return suspend_subtree(root);
  }

  auto resume_tree(bus & root) -> void
  {
    resume_subtree(root);
  }

}  // namespace kapi::devices