From 01f19563d47cd5e5269da0a060f473a34e1b183d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20De=20Keersmaeker?=
<francois.dekeersmaeker@uclouvain.be>
Date: Fri, 29 Nov 2024 11:06:30 +0100
Subject: [PATCH] Templates: callback: fixed variable
---
profile_translator_blocklist/templates/callback.c.j2 | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/profile_translator_blocklist/templates/callback.c.j2 b/profile_translator_blocklist/templates/callback.c.j2
index 2e06fef..5801752 100644
--- a/profile_translator_blocklist/templates/callback.c.j2
+++ b/profile_translator_blocklist/templates/callback.c.j2
@@ -207,6 +207,7 @@ uint32_t callback_{{nfqueue_name}}(int pkt_id, int pkt_len, uint8_t *payload, vo
else if ({{ template.format(domain_name) }}) {
{% endif %}
domain_name = "{{domain_name}}";
+ subdomain = "{{domain_name.split('.', 1)[1]}}";
ip_list = dns_get_ip_from_name(dns_message.answers, dns_message.header.ancount, domain_name);
}
{% endif %}
@@ -215,6 +216,7 @@ uint32_t callback_{{nfqueue_name}}(int pkt_id, int pkt_len, uint8_t *payload, vo
{% if "domain_name" in nfq_match["template"] %}
{% set domain_name = nfq_match["match"] %}
domain_name = "{{domain_name}}";
+ subdomain = "{{domain_name.split('.', 1)[1]}}";
ip_list = dns_get_ip_from_name(dns_message.answers, dns_message.header.ancount, domain_name);
{% endif %}
{% endif %}
@@ -225,7 +227,7 @@ uint32_t callback_{{nfqueue_name}}(int pkt_id, int pkt_len, uint8_t *payload, vo
{% if is_dns_response.value %}
if (ip_list.ip_count > 0) {
// Add IP addresses to DNS map
- dns_map_add(dns_map, "{{domain_name.split('.', 1)[1]}}", ip_list);
+ dns_map_add(dns_map, subdomain, ip_list);
}
{% endif %}
--
GitLab