qosify.h

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Copyright (C) 2021 Felix Fietkau <nbd@nbd.name>
   4  */
   5 #ifndef __QOS_CLASSIFY_H
   6 #define __QOS_CLASSIFY_H
   7
   8 #include <stdbool.h>
   9 #include <regex.h>
  10
  11 #include <bpf/bpf.h>
  12 #include <bpf/libbpf.h>
  13
  14 #include "qosify-bpf.h"
  15
  16 #include <libubox/utils.h>
  17 #include <libubox/avl.h>
  18 #include <libubox/blobmsg.h>
  19 #include <libubox/ulog.h>
  20
  21 #include <netinet/in.h>
  22
  23 #define CLASSIFY_PROG_PATH      "/lib/bpf/qosify-bpf.o"
  24 #define CLASSIFY_PIN_PATH       "/sys/fs/bpf/qosify"
  25 #define CLASSIFY_DATA_PATH      "/sys/fs/bpf/qosify_data"
  26
  27 enum qosify_map_id {
  28         CL_MAP_TCP_PORTS,
  29         CL_MAP_UDP_PORTS,
  30         CL_MAP_IPV4_ADDR,
  31         CL_MAP_IPV6_ADDR,
  32         CL_MAP_CLASS,
  33         CL_MAP_CONFIG,
  34         CL_MAP_DNS,
  35         __CL_MAP_MAX,
  36 };
  37
  38 struct qosify_map_data {
  39         enum qosify_map_id id;
  40
  41         bool file : 1;
  42         bool user : 1;
  43
  44         uint8_t dscp;
  45         uint8_t file_dscp;
  46
  47         union {
  48                 uint32_t port;
  49                 struct in_addr ip;
  50                 struct in6_addr ip6;
  51                 struct {
  52                         const char *pattern;
  53                         regex_t regex;
  54                 } dns;
  55         } addr;
  56 };
  57
  58 struct qosify_map_entry {
  59         struct avl_node avl;
  60
  61         uint32_t timeout;
  62
  63         struct qosify_map_data data;
  64 };
  65
  66
  67 extern int qosify_map_timeout;
  68 extern int qosify_active_timeout;
  69 extern struct qosify_config config;
  70 extern struct qosify_flow_config flow_config;
  71
  72 int qosify_run_cmd(char *cmd, bool ignore_error);
  73
  74 int qosify_loader_init(void);
  75
  76 int qosify_map_init(void);
  77 int qosify_map_dscp_value(const char *val, uint8_t *dscp);
  78 int qosify_map_load_file(const char *file);
  79 int qosify_map_set_entry(enum qosify_map_id id, bool file, const char *str,
  80                          uint8_t dscp);
  81 void qosify_map_reload(void);
  82 void qosify_map_clear_files(void);
  83 void qosify_map_gc(void);
  84 void qosify_map_dump(struct blob_buf *b);
  85 void qosify_map_set_dscp_default(enum qosify_map_id id, uint8_t val);
  86 void qosify_map_reset_config(void);
  87 void qosify_map_update_config(void);
  88 void qosify_map_set_classes(struct blob_attr *val);
  89 int qosify_map_lookup_dns_entry(char *host, uint8_t *dscp);
  90 int qosify_map_add_dns_host(char *host, const char *addr, const char *type, int ttl);
  91 int map_parse_flow_config(struct qosify_flow_config *cfg, struct blob_attr *attr,
  92                           bool reset);
  93 int map_fill_dscp_value(uint8_t *dest, struct blob_attr *attr, bool reset);
  94
  95 int qosify_iface_init(void);
  96 void qosify_iface_config_update(struct blob_attr *ifaces, struct blob_attr *devs);
  97 void qosify_iface_check(void);
  98 void qosify_iface_status(struct blob_buf *b);
  99 void qosify_iface_stop(void);
 100
 101 int qosify_ubus_init(void);
 102 void qosify_ubus_stop(void);
 103 int qosify_ubus_check_interface(const char *name, char *ifname, int ifname_len);
 104
 105 #endif