]>
Commit | Line | Data |
---|---|---|
db9ecf05 | 1 | /* SPDX-License-Identifier: LGPL-2.1-or-later */ |
a879e1a4 YW |
2 | |
3 | #include <errno.h> | |
4 | ||
5 | #include "sd-event.h" | |
6 | #include "sd-netlink.h" | |
7 | ||
6a1af3d4 | 8 | #include "networkd-link-bus.h" |
a879e1a4 YW |
9 | #include "networkd-link.h" |
10 | #include "networkd-manager.h" | |
11 | #include "networkd-speed-meter.h" | |
12 | ||
13 | static int process_message(Manager *manager, sd_netlink_message *message) { | |
14 | uint16_t type; | |
15 | int ifindex, r; | |
16 | Link *link; | |
17 | ||
18 | r = sd_netlink_message_get_type(message, &type); | |
19 | if (r < 0) | |
20 | return r; | |
21 | ||
22 | if (type != RTM_NEWLINK) | |
23 | return 0; | |
24 | ||
25 | r = sd_rtnl_message_link_get_ifindex(message, &ifindex); | |
26 | if (r < 0) | |
27 | return r; | |
28 | ||
6eab614d YW |
29 | r = link_get_by_index(manager, ifindex, &link); |
30 | if (r < 0) | |
31 | return r; | |
a879e1a4 YW |
32 | |
33 | link->stats_old = link->stats_new; | |
34 | ||
35 | r = sd_netlink_message_read(message, IFLA_STATS64, sizeof link->stats_new, &link->stats_new); | |
36 | if (r < 0) | |
37 | return r; | |
38 | ||
39 | link->stats_updated = true; | |
40 | ||
41 | return 0; | |
42 | } | |
43 | ||
44 | static int speed_meter_handler(sd_event_source *s, uint64_t usec, void *userdata) { | |
45 | _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *req = NULL, *reply = NULL; | |
99534007 | 46 | Manager *manager = ASSERT_PTR(userdata); |
a879e1a4 | 47 | usec_t usec_now; |
a879e1a4 YW |
48 | Link *link; |
49 | int r; | |
50 | ||
51 | assert(s); | |
a879e1a4 YW |
52 | |
53 | r = sd_event_now(sd_event_source_get_event(s), CLOCK_MONOTONIC, &usec_now); | |
54 | if (r < 0) | |
55 | return r; | |
56 | ||
57 | r = sd_event_source_set_time(s, usec_now + manager->speed_meter_interval_usec); | |
58 | if (r < 0) | |
59 | return r; | |
60 | ||
61 | manager->speed_meter_usec_old = manager->speed_meter_usec_new; | |
62 | manager->speed_meter_usec_new = usec_now; | |
63 | ||
6eab614d | 64 | HASHMAP_FOREACH(link, manager->links_by_index) |
a879e1a4 YW |
65 | link->stats_updated = false; |
66 | ||
67 | r = sd_rtnl_message_new_link(manager->rtnl, &req, RTM_GETLINK, 0); | |
68 | if (r < 0) { | |
69 | log_warning_errno(r, "Failed to allocate RTM_GETLINK netlink message, ignoring: %m"); | |
70 | return 0; | |
71 | } | |
72 | ||
24c0f385 | 73 | r = sd_netlink_message_set_request_dump(req, true); |
a879e1a4 YW |
74 | if (r < 0) { |
75 | log_warning_errno(r, "Failed to set dump flag, ignoring: %m"); | |
76 | return 0; | |
77 | } | |
78 | ||
79 | r = sd_netlink_call(manager->rtnl, req, 0, &reply); | |
80 | if (r < 0) { | |
81 | log_warning_errno(r, "Failed to call RTM_GETLINK, ignoring: %m"); | |
82 | return 0; | |
83 | } | |
84 | ||
cda7fc8d | 85 | for (sd_netlink_message *i = reply; i; i = sd_netlink_message_next(i)) |
a879e1a4 YW |
86 | (void) process_message(manager, i); |
87 | ||
88 | return 0; | |
89 | } | |
90 | ||
91 | int manager_start_speed_meter(Manager *manager) { | |
92 | _cleanup_(sd_event_source_unrefp) sd_event_source *s = NULL; | |
93 | int r; | |
94 | ||
95 | assert(manager); | |
96 | assert(manager->event); | |
97 | ||
98 | if (!manager->use_speed_meter) | |
99 | return 0; | |
100 | ||
101 | r = sd_event_add_time(manager->event, &s, CLOCK_MONOTONIC, 0, 0, speed_meter_handler, manager); | |
102 | if (r < 0) | |
103 | return r; | |
104 | ||
105 | r = sd_event_source_set_enabled(s, SD_EVENT_ON); | |
106 | if (r < 0) | |
107 | return r; | |
108 | ||
109 | manager->speed_meter_event_source = TAKE_PTR(s); | |
110 | return 0; | |
111 | } |