README: fix typo

[project/qosify.git] / map.c

diff --git a/map.c b/map.c
index 2eacfcdc40b67ac391913aef7bd798b4da3bcfb1..05518f5225c72ac8315d8be00d48f075f396a623 100644 (file)
--- a/map.c
+++ b/map.c
@@ -38,6 +38,7 @@ static const struct {
        [CL_MAP_IPV4_ADDR] = { "ipv4_map", "ipv4_addr" },
        [CL_MAP_IPV6_ADDR] = { "ipv6_map", "ipv6_addr" },
        [CL_MAP_CONFIG] = { "config", "config" },
+       [CL_MAP_DNS] = { "dns", "dns" },
 };
 
 static const struct {
@@ -66,6 +67,7 @@ static const struct {
        { "AF43", 38 },
        { "EF", 46 },
        { "VA", 44 },
+       { "LE", 1 },
 };
 
 static void qosify_map_timer_cb(struct uloop_timeout *t)
@@ -169,7 +171,7 @@ int qosify_map_init(void)
 {
        int i;
 
-       for (i = 0; i < ARRAY_SIZE(qosify_map_fds); i++) {
+       for (i = 0; i < CL_MAP_DNS; i++) {
                qosify_map_fds[i] = qosify_map_get_fd(i);
                if (qosify_map_fds[i] < 0)
                        return -1;
@@ -210,9 +212,37 @@ static int qosify_map_entry_cmp(const void *k1, const void *k2, void *ptr)
        if (d1->id != d2->id)
                return d2->id - d1->id;
 
+       if (d1->id == CL_MAP_DNS)
+               return strcmp(d1->addr.dns.pattern, d2->addr.dns.pattern);
+
        return memcmp(&d1->addr, &d2->addr, sizeof(d1->addr));
 }
 
+static struct qosify_map_entry *
+__qosify_map_alloc_entry(struct qosify_map_data *data)
+{
+       struct qosify_map_entry *e;
+       char *pattern;
+
+       if (data->id < CL_MAP_DNS) {
+               e = calloc(1, sizeof(*e));
+               memcpy(&e->data.addr, &data->addr, sizeof(e->data.addr));
+
+               return e;
+       }
+
+       e = calloc_a(sizeof(*e), &pattern, strlen(data->addr.dns.pattern) + 1);
+       strcpy(pattern, data->addr.dns.pattern);
+       e->data.addr.dns.pattern = pattern;
+       if (regcomp(&e->data.addr.dns.regex, pattern,
+                   REG_EXTENDED | REG_ICASE | REG_NOSUB)) {
+               free(e);
+               return NULL;
+       }
+
+       return e;
+}
+
 static void __qosify_map_set_entry(struct qosify_map_data *data)
 {
        int fd = qosify_map_fds[data->id];
@@ -227,10 +257,12 @@ static void __qosify_map_set_entry(struct qosify_map_data *data)
                if (!add)
                        return;
 
-               e = calloc(1, sizeof(*e));
+               e = __qosify_map_alloc_entry(data);
+               if (!e)
+                       return;
+
                e->avl.key = &e->data;
                e->data.id = data->id;
-               memcpy(&e->data.addr, &data->addr, sizeof(e->data.addr));
                avl_insert(&map_data, &e->avl);
        } else {
                prev_dscp = e->data.dscp;
@@ -250,7 +282,7 @@ static void __qosify_map_set_entry(struct qosify_map_data *data)
                e->data.dscp = e->data.file_dscp;
        }
 
-       if (e->data.dscp != prev_dscp)
+       if (e->data.dscp != prev_dscp && data->id < CL_MAP_DNS)
                bpf_map_update_elem(fd, &data->addr, &e->data.dscp, BPF_ANY);
 
        if (add) {
@@ -320,6 +352,9 @@ int qosify_map_set_entry(enum qosify_map_id id, bool file, const char *str, uint
        };
 
        switch (id) {
+       case CL_MAP_DNS:
+               data.addr.dns.pattern = str;
+               break;
        case CL_MAP_TCP_PORTS:
        case CL_MAP_UDP_PORTS:
                return qosify_map_set_port(&data, str);
@@ -401,6 +436,8 @@ qosify_map_parse_line(char *str)
        if (dscp < 0)
                return;
 
+       if (!strncmp(key, "dns:", 4))
+               qosify_map_set_entry(CL_MAP_DNS, true, key + 4, dscp);
        if (!strncmp(key, "tcp:", 4))
                qosify_map_set_entry(CL_MAP_TCP_PORTS, true, key + 4, dscp);
        else if (!strncmp(key, "udp:", 4))
@@ -511,7 +548,8 @@ static void qosify_map_free_entry(struct qosify_map_entry *e)
        int fd = qosify_map_fds[e->data.id];
 
        avl_delete(&map_data, &e->avl);
-       bpf_map_delete_elem(fd, &e->data.addr);
+       if (e->data.id < CL_MAP_DNS)
+               bpf_map_delete_elem(fd, &e->data.addr);
        free(e);
 }
 
@@ -548,6 +586,52 @@ void qosify_map_gc(void)
        uloop_timeout_set(&qosify_map_timer, timeout * 1000);
 }
 
+
+int qosify_map_add_dns_host(const char *host, const char *addr, const char *type, int ttl)
+{
+       struct qosify_map_data data = {
+               .id = CL_MAP_DNS,
+               .addr.dns.pattern = "",
+       };
+       struct qosify_map_entry *e;
+       int prev_timeout = qosify_map_timeout;
+
+       e = avl_find_ge_element(&map_data, &data, e, avl);
+       if (!e)
+               return 0;
+
+       memset(&data, 0, sizeof(data));
+       data.user = true;
+       if (!strcmp(type, "A"))
+               data.id = CL_MAP_IPV4_ADDR;
+       else if (!strcmp(type, "AAAA"))
+               data.id = CL_MAP_IPV6_ADDR;
+       else
+               return 0;
+
+       if (qosify_map_fill_ip(&data, addr))
+               return -1;
+
+       avl_for_element_to_last(&map_data, e, e, avl) {
+               regex_t *regex = &e->data.addr.dns.regex;
+
+               if (e->data.id != CL_MAP_DNS)
+                       return 0;
+
+               if (regexec(regex, host, 0, NULL, 0) != 0)
+                       continue;
+
+               if (ttl)
+                       qosify_map_timeout = ttl;
+               data.dscp = e->data.dscp;
+               __qosify_map_set_entry(&data);
+               qosify_map_timeout = prev_timeout;
+       }
+
+       return 0;
+}
+
+
 void qosify_map_dump(struct blob_buf *b)
 {
        struct qosify_map_entry *e;
@@ -583,22 +667,25 @@ void qosify_map_dump(struct blob_buf *b)
 
                blobmsg_add_string(b, "type", qosify_map_info[e->data.id].type_name);
 
-               buf = blobmsg_alloc_string_buffer(b, "value", buf_len);
                switch (e->data.id) {
                case CL_MAP_TCP_PORTS:
                case CL_MAP_UDP_PORTS:
-                       snprintf(buf, buf_len, "%d", ntohs(e->data.addr.port));
+                       blobmsg_printf(b, "addr", "%d", ntohs(e->data.addr.port));
                        break;
                case CL_MAP_IPV4_ADDR:
                case CL_MAP_IPV6_ADDR:
+                       buf = blobmsg_alloc_string_buffer(b, "addr", buf_len);
                        af = e->data.id == CL_MAP_IPV6_ADDR ? AF_INET6 : AF_INET;
                        inet_ntop(af, &e->data.addr, buf, buf_len);
+                       blobmsg_add_string_buffer(b);
+                       break;
+               case CL_MAP_DNS:
+                       blobmsg_add_string(b, "addr", e->data.addr.dns.pattern);
                        break;
                default:
                        *buf = 0;
                        break;
                }
-               blobmsg_add_string_buffer(b);
                blobmsg_close_table(b, c);
        }
        blobmsg_close_array(b, a);