637 lines
16 KiB
C
637 lines
16 KiB
C
#include <linux/if_ether.h>
|
|
#include <linux/ip.h>
|
|
#include <linux/ipv6.h>
|
|
#include <linux/udp.h>
|
|
#include <linux/tcp.h>
|
|
#include <linux/icmp.h>
|
|
#include <linux/icmpv6.h>
|
|
#include <linux/in.h>
|
|
#include <stdatomic.h>
|
|
|
|
#include <linux/bpf.h>
|
|
#include <linux/bpf_common.h>
|
|
|
|
#include <bpf_helpers.h>
|
|
#include <xdp/xdp_helpers.h>
|
|
#include <xdp/prog_dispatcher.h>
|
|
|
|
#include "xdpfw.h"
|
|
|
|
#ifndef memcpy
|
|
#define memcpy(dest, src, n) __builtin_memcpy((dest), (src), (n))
|
|
#endif
|
|
|
|
struct
|
|
{
|
|
__uint(priority, 10);
|
|
__uint(XDP_PASS, 1);
|
|
} XDP_RUN_CONFIG(xdp_prog_main);
|
|
|
|
struct
|
|
{
|
|
__uint(type, BPF_MAP_TYPE_ARRAY);
|
|
__uint(max_entries, MAX_FILTERS);
|
|
__type(key, __u32);
|
|
__type(value, struct filter);
|
|
} filters_map SEC(".maps");
|
|
|
|
struct
|
|
{
|
|
__uint(type, BPF_MAP_TYPE_PERCPU_ARRAY);
|
|
__uint(max_entries, 1);
|
|
__type(key, __u32);
|
|
__type(value, struct stats);
|
|
} stats_map SEC(".maps");
|
|
|
|
struct
|
|
{
|
|
__uint(type, BPF_MAP_TYPE_LRU_HASH);
|
|
__uint(max_entries, MAX_TRACK_IPS);
|
|
__type(key, __u32);
|
|
__type(value, struct ip_stats);
|
|
} ip_stats_map SEC(".maps");
|
|
|
|
struct
|
|
{
|
|
__uint(type, BPF_MAP_TYPE_LRU_HASH);
|
|
__uint(max_entries, MAX_TRACK_IPS);
|
|
__type(key, __u32);
|
|
__type(value, __u64);
|
|
} ip_blacklist_map SEC(".maps");
|
|
|
|
struct
|
|
{
|
|
__uint(type, BPF_MAP_TYPE_LRU_HASH);
|
|
__uint(max_entries, MAX_TRACK_IPS);
|
|
__type(key, __u128);
|
|
__type(value, struct ip_stats);
|
|
} ip6_stats_map SEC(".maps");
|
|
|
|
struct
|
|
{
|
|
__uint(type, BPF_MAP_TYPE_LRU_HASH);
|
|
__uint(max_entries, MAX_TRACK_IPS);
|
|
__type(key, __u128);
|
|
__type(value, __u64);
|
|
} ip6_blacklist_map SEC(".maps");
|
|
|
|
SEC("xdp_prog")
|
|
int xdp_prog_main(struct xdp_md *ctx)
|
|
{
|
|
// Initialize data.
|
|
void *data_end = (void *)(long)ctx->data_end;
|
|
void *data = (void *)(long)ctx->data;
|
|
|
|
// Scan ethernet header.
|
|
struct ethhdr *eth = data;
|
|
|
|
// Check if the ethernet header is valid.
|
|
if (eth + 1 > (struct ethhdr *)data_end)
|
|
{
|
|
return XDP_DROP;
|
|
}
|
|
|
|
// Check Ethernet protocol.
|
|
if (unlikely(eth->h_proto != htons(ETH_P_IP) && eth->h_proto != htons(ETH_P_IPV6)))
|
|
{
|
|
return XDP_PASS;
|
|
}
|
|
|
|
__u8 action = 0;
|
|
__u64 blocktime = 1;
|
|
|
|
// Initialize IP headers.
|
|
struct iphdr *iph = NULL;
|
|
struct ipv6hdr *iph6 = NULL;
|
|
__u128 srcip6 = 0;
|
|
|
|
// Set IPv4 and IPv6 common variables.
|
|
if (eth->h_proto == htons(ETH_P_IPV6))
|
|
{
|
|
iph6 = (data + sizeof(struct ethhdr));
|
|
|
|
if (unlikely(iph6 + 1 > (struct ipv6hdr *)data_end))
|
|
{
|
|
return XDP_DROP;
|
|
}
|
|
|
|
memcpy(&srcip6, &iph6->saddr.in6_u.u6_addr32, sizeof(srcip6));
|
|
}
|
|
else
|
|
{
|
|
iph = (data + sizeof(struct ethhdr));
|
|
|
|
if (unlikely(iph + 1 > (struct iphdr *)data_end))
|
|
{
|
|
return XDP_DROP;
|
|
}
|
|
}
|
|
|
|
// Check IP header protocols.
|
|
if ((iph6 && iph6->nexthdr != IPPROTO_UDP && iph6->nexthdr != IPPROTO_TCP && iph6->nexthdr != IPPROTO_ICMP) && (iph && iph->protocol != IPPROTO_UDP && iph->protocol != IPPROTO_TCP && iph->protocol != IPPROTO_ICMP))
|
|
{
|
|
return XDP_PASS;
|
|
}
|
|
|
|
// Get stats map.
|
|
__u32 key = 0;
|
|
struct stats *stats = bpf_map_lookup_elem(&stats_map, &key);
|
|
|
|
__u64 now = bpf_ktime_get_ns();
|
|
|
|
// Check blacklist map.
|
|
__u64 *blocked = NULL;
|
|
|
|
if (iph6)
|
|
{
|
|
blocked = bpf_map_lookup_elem(&ip6_blacklist_map, &srcip6);
|
|
}
|
|
else if (iph)
|
|
{
|
|
blocked = bpf_map_lookup_elem(&ip_blacklist_map, &iph->saddr);
|
|
}
|
|
|
|
if (blocked != NULL && *blocked > 0)
|
|
{
|
|
#ifdef DEBUG
|
|
bpf_printk("Checking for blocked packet... Block time %llu.\n", *blocked);
|
|
#endif
|
|
|
|
if (now > *blocked)
|
|
{
|
|
// Remove element from map.
|
|
if (iph6)
|
|
{
|
|
bpf_map_delete_elem(&ip6_blacklist_map, &srcip6);
|
|
}
|
|
else if (iph)
|
|
{
|
|
bpf_map_delete_elem(&ip_blacklist_map, &iph->saddr);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
#ifdef DOSTATSONBLOCKMAP
|
|
// Increase blocked stats entry.
|
|
if (stats)
|
|
{
|
|
stats->dropped++;
|
|
}
|
|
#endif
|
|
|
|
// They're still blocked. Drop the packet.
|
|
return XDP_DROP;
|
|
}
|
|
}
|
|
|
|
// Update IP stats (PPS/BPS).
|
|
__u64 pps = 0;
|
|
__u64 bps = 0;
|
|
|
|
struct ip_stats *ip_stats = NULL;
|
|
|
|
if (iph6)
|
|
{
|
|
ip_stats = bpf_map_lookup_elem(&ip6_stats_map, &srcip6);
|
|
}
|
|
else if (iph)
|
|
{
|
|
ip_stats = bpf_map_lookup_elem(&ip_stats_map, &iph->saddr);
|
|
}
|
|
|
|
if (ip_stats)
|
|
{
|
|
// Check for reset.
|
|
if ((now - ip_stats->tracking) > 1000000000)
|
|
{
|
|
ip_stats->pps = 0;
|
|
ip_stats->bps = 0;
|
|
ip_stats->tracking = now;
|
|
}
|
|
|
|
// Increment PPS and BPS using built-in functions.
|
|
__sync_fetch_and_add(&ip_stats->pps, 1);
|
|
__sync_fetch_and_add(&ip_stats->bps, ctx->data_end - ctx->data);
|
|
|
|
pps = ip_stats->pps;
|
|
bps = ip_stats->bps;
|
|
}
|
|
else
|
|
{
|
|
// Create new entry.
|
|
struct ip_stats new;
|
|
|
|
new.pps = 1;
|
|
new.bps = ctx->data_end - ctx->data;
|
|
new.tracking = now;
|
|
|
|
pps = new.pps;
|
|
bps = new.bps;
|
|
|
|
if (iph6)
|
|
{
|
|
bpf_map_update_elem(&ip6_stats_map, &srcip6, &new, BPF_ANY);
|
|
}
|
|
else if (iph)
|
|
{
|
|
bpf_map_update_elem(&ip_stats_map, &iph->saddr, &new, BPF_ANY);
|
|
}
|
|
}
|
|
|
|
struct tcphdr *tcph = NULL;
|
|
struct udphdr *udph = NULL;
|
|
struct icmphdr *icmph = NULL;
|
|
struct icmp6hdr *icmp6h = NULL;
|
|
|
|
// Check protocol.
|
|
if (iph6)
|
|
{
|
|
switch (iph6->nexthdr)
|
|
{
|
|
case IPPROTO_TCP:
|
|
// Scan TCP header.
|
|
tcph = (data + sizeof(struct ethhdr) + sizeof(struct ipv6hdr));
|
|
|
|
// Check TCP header.
|
|
if (tcph + 1 > (struct tcphdr *)data_end)
|
|
{
|
|
return XDP_DROP;
|
|
}
|
|
|
|
break;
|
|
|
|
case IPPROTO_UDP:
|
|
// Scan UDP header.
|
|
udph = (data + sizeof(struct ethhdr) + sizeof(struct ipv6hdr));
|
|
|
|
// Check TCP header.
|
|
if (udph + 1 > (struct udphdr *)data_end)
|
|
{
|
|
return XDP_DROP;
|
|
}
|
|
|
|
break;
|
|
|
|
case IPPROTO_ICMPV6:
|
|
// Scan ICMPv6 header.
|
|
icmp6h = (data + sizeof(struct ethhdr) + sizeof(struct ipv6hdr));
|
|
|
|
// Check ICMPv6 header.
|
|
if (icmp6h + 1 > (struct icmp6hdr *)data_end)
|
|
{
|
|
return XDP_DROP;
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
else if (iph)
|
|
{
|
|
switch (iph->protocol)
|
|
{
|
|
case IPPROTO_TCP:
|
|
// Scan TCP header.
|
|
tcph = (data + sizeof(struct ethhdr) + (iph->ihl * 4));
|
|
|
|
// Check TCP header.
|
|
if (tcph + 1 > (struct tcphdr *)data_end)
|
|
{
|
|
return XDP_DROP;
|
|
}
|
|
|
|
break;
|
|
|
|
case IPPROTO_UDP:
|
|
// Scan UDP header.
|
|
udph = (data + sizeof(struct ethhdr) + (iph->ihl * 4));
|
|
|
|
// Check TCP header.
|
|
if (udph + 1 > (struct udphdr *)data_end)
|
|
{
|
|
return XDP_DROP;
|
|
}
|
|
|
|
break;
|
|
|
|
case IPPROTO_ICMP:
|
|
// Scan ICMP header.
|
|
icmph = (data + sizeof(struct ethhdr) + (iph->ihl * 4));
|
|
|
|
// Check ICMP header.
|
|
if (icmph + 1 > (struct icmphdr *)data_end)
|
|
{
|
|
return XDP_DROP;
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
for (__u8 i = 0; i < MAX_FILTERS; i++)
|
|
{
|
|
__u32 key = i;
|
|
|
|
struct filter *filter = bpf_map_lookup_elem(&filters_map, &key);
|
|
|
|
// Check if ID is above 0 (if 0, it's an invalid rule).
|
|
if (!filter || filter->id < 1)
|
|
{
|
|
break;
|
|
}
|
|
|
|
// Check if the rule is enabled.
|
|
if (!filter->enabled)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Do specific IPv6.
|
|
if (iph6)
|
|
{
|
|
// Source address.
|
|
if (filter->srcip6[0] != 0 && (iph6->saddr.in6_u.u6_addr32[0] != filter->srcip6[0] || iph6->saddr.in6_u.u6_addr32[1] != filter->srcip6[1] || iph6->saddr.in6_u.u6_addr32[2] != filter->srcip6[2] || iph6->saddr.in6_u.u6_addr32[3] != filter->srcip6[3]))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Destination address.
|
|
if (filter->dstip6[0] != 0 && (iph6->daddr.in6_u.u6_addr32[0] != filter->dstip6[0] || iph6->daddr.in6_u.u6_addr32[1] != filter->dstip6[1] || iph6->daddr.in6_u.u6_addr32[2] != filter->dstip6[2] || iph6->daddr.in6_u.u6_addr32[3] != filter->dstip6[3]))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
#ifdef ALLOWSINGLEIPV4V6
|
|
if (filter->srcip != 0 || filter->dstip != 0)
|
|
{
|
|
continue;
|
|
}
|
|
#endif
|
|
|
|
// Max TTL length.
|
|
if (filter->do_max_ttl && filter->max_ttl > iph6->hop_limit)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Min TTL length.
|
|
if (filter->do_min_ttl && filter->min_ttl < iph6->hop_limit)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Max packet length.
|
|
if (filter->do_max_len && filter->max_len > (ntohs(iph6->payload_len) + sizeof(struct ethhdr)))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Min packet length.
|
|
if (filter->do_min_len && filter->min_len < (ntohs(iph6->payload_len) + sizeof(struct ethhdr)))
|
|
{
|
|
continue;
|
|
}
|
|
}
|
|
else if (iph)
|
|
{
|
|
// Source address.
|
|
if (filter->srcip && iph->saddr != filter->srcip)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Destination address.
|
|
if (filter->dstip != 0 && iph->daddr != filter->dstip)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
#ifdef ALLOWSINGLEIPV4V6
|
|
if ((filter->srcip6[0] != 0 || filter->srcip6[1] != 0 || filter->srcip6[2] != 0 || filter->srcip6[3] != 0) || (filter->dstip6[0] != 0 || filter->dstip6[1] != 0 || filter->dstip6[2] != 0 || filter->dstip6[3] != 0))
|
|
{
|
|
continue;
|
|
}
|
|
#endif
|
|
|
|
// TOS.
|
|
if (filter->do_tos && filter->tos != iph->tos)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Max TTL length.
|
|
if (filter->do_max_ttl && filter->max_ttl < iph->ttl)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Min TTL length.
|
|
if (filter->do_min_ttl && filter->min_ttl > iph->ttl)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Max packet length.
|
|
if (filter->do_max_len && filter->max_len < (ntohs(iph->tot_len) + sizeof(struct ethhdr)))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Min packet length.
|
|
if (filter->do_min_len && filter->min_len > (ntohs(iph->tot_len) + sizeof(struct ethhdr)))
|
|
{
|
|
continue;
|
|
}
|
|
}
|
|
|
|
// PPS.
|
|
if (filter->do_pps && pps < filter->pps)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// BPS.
|
|
if (filter->do_bps && bps < filter->bps)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Do TCP options.
|
|
if (filter->tcpopts.enabled)
|
|
{
|
|
if (!tcph)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Source port.
|
|
if (filter->tcpopts.do_sport && htons(filter->tcpopts.sport) != tcph->source)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Destination port.
|
|
if (filter->tcpopts.do_dport && htons(filter->tcpopts.dport) != tcph->dest)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// URG flag.
|
|
if (filter->tcpopts.do_urg && filter->tcpopts.urg != tcph->urg)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// ACK flag.
|
|
if (filter->tcpopts.do_ack && filter->tcpopts.ack != tcph->ack)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// RST flag.
|
|
if (filter->tcpopts.do_rst && filter->tcpopts.rst != tcph->rst)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// PSH flag.
|
|
if (filter->tcpopts.do_psh && filter->tcpopts.psh != tcph->psh)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// SYN flag.
|
|
if (filter->tcpopts.do_syn && filter->tcpopts.syn != tcph->syn)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// FIN flag.
|
|
if (filter->tcpopts.do_fin && filter->tcpopts.fin != tcph->fin)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// ECE flag.
|
|
if (filter->tcpopts.do_ece && filter->tcpopts.ece != tcph->ece)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// CWR flag.
|
|
if (filter->tcpopts.do_cwr && filter->tcpopts.cwr != tcph->cwr)
|
|
{
|
|
continue;
|
|
}
|
|
}
|
|
else if (filter->udpopts.enabled)
|
|
{
|
|
if (!udph)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Source port.
|
|
if (filter->udpopts.do_sport && htons(filter->udpopts.sport) != udph->source)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Destination port.
|
|
if (filter->udpopts.do_dport && htons(filter->udpopts.dport) != udph->dest)
|
|
{
|
|
|
|
continue;
|
|
}
|
|
}
|
|
else if (filter->icmpopts.enabled)
|
|
{
|
|
if (icmph)
|
|
{
|
|
// Code.
|
|
if (filter->icmpopts.do_code && filter->icmpopts.code != icmph->code)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Type.
|
|
if (filter->icmpopts.do_type && filter->icmpopts.type != icmph->type)
|
|
{
|
|
continue;
|
|
}
|
|
}
|
|
else if (icmp6h)
|
|
{
|
|
// Code.
|
|
if (filter->icmpopts.do_code && filter->icmpopts.code != icmp6h->icmp6_code)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Type.
|
|
if (filter->icmpopts.do_type && filter->icmpopts.type != icmp6h->icmp6_type)
|
|
{
|
|
continue;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
continue;
|
|
}
|
|
}
|
|
|
|
// Matched.
|
|
#ifdef DEBUG
|
|
bpf_printk("Matched rule ID #%d.\n", filter->id);
|
|
#endif
|
|
|
|
action = filter->action;
|
|
blocktime = filter->blocktime;
|
|
|
|
goto matched;
|
|
}
|
|
|
|
return XDP_PASS;
|
|
|
|
matched:
|
|
if (action == 0)
|
|
{
|
|
#ifdef DEBUG
|
|
//bpf_printk("Matched with protocol %d and sAddr %lu.\n", iph->protocol, iph->saddr);
|
|
#endif
|
|
|
|
// Before dropping, update the blacklist map.
|
|
if (blocktime > 0)
|
|
{
|
|
__u64 newTime = now + (blocktime * 1000000000);
|
|
|
|
if (iph6)
|
|
{
|
|
bpf_map_update_elem(&ip6_blacklist_map, &srcip6, &newTime, BPF_ANY);
|
|
}
|
|
else if (iph)
|
|
{
|
|
bpf_map_update_elem(&ip_blacklist_map, &iph->saddr, &newTime, BPF_ANY);
|
|
}
|
|
}
|
|
|
|
if (stats)
|
|
{
|
|
stats->dropped++;
|
|
}
|
|
|
|
return XDP_DROP;
|
|
}
|
|
else
|
|
{
|
|
if (stats)
|
|
{
|
|
stats->allowed++;
|
|
}
|
|
}
|
|
|
|
return XDP_PASS;
|
|
}
|
|
|
|
char _license[] SEC("license") = "GPL";
|
|
|
|
__uint(xsk_prog_version, XDP_DISPATCHER_VERSION) SEC(XDP_METADATA_SECTION); |