src/network/networkd-speed-meter.c

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