From d9d6167b4bfe5a7989c8154d4e3f1012e2b3b9ac Mon Sep 17 00:00:00 2001 From: gamemann Date: Fri, 12 Nov 2021 21:14:20 +0000 Subject: [PATCH] Add one more check. --- src/xdpfw_kern.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/xdpfw_kern.c b/src/xdpfw_kern.c index 63722c3..ee1df01 100644 --- a/src/xdpfw_kern.c +++ b/src/xdpfw_kern.c @@ -599,11 +599,11 @@ int xdp_prog_main(struct xdp_md *ctx) { __u64 newTime = now + (blocktime * 1000000000); - if (eth->h_proto == htons(ETH_P_IPV6)) + if (iph6) { bpf_map_update_elem(&ip6_blacklist_map, &srcip6, &newTime, BPF_ANY); } - else + else if (iph) { bpf_map_update_elem(&ip_blacklist_map, &iph->saddr, &newTime, BPF_ANY); }