Add one more check.

2021-11-12 21:14:20 +00:00
parent fe09dece3a
commit d9d6167b4b
1 changed files with 2 additions and 2 deletions
--- 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);
                }