132 lines
2.1 KiB
C
132 lines
2.1 KiB
C
#pragma once
|
|
|
|
#include <linux/types.h>
|
|
|
|
#define MAX_PCKT_LENGTH 65535
|
|
#define MAX_FILTERS 100
|
|
#define MAX_TRACK_IPS 100000
|
|
|
|
#ifdef __BPF__
|
|
#define likely(x) __builtin_expect(!!(x), 1)
|
|
#define unlikely(x) __builtin_expect(!!(x), 0)
|
|
#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
|
|
#define htons(x) ((__be16)___constant_swab16((x)))
|
|
#define ntohs(x) ((__be16)___constant_swab16((x)))
|
|
#define htonl(x) ((__be32)___constant_swab32((x)))
|
|
#define ntohl(x) ((__be32)___constant_swab32((x)))
|
|
#elif __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
|
|
#define htons(x) (x)
|
|
#define ntohs(X) (x)
|
|
#define htonl(x) (x)
|
|
#define ntohl(x) (x)
|
|
#endif
|
|
|
|
#define uint128_t __uint128_t
|
|
#endif
|
|
|
|
struct tcpopts
|
|
{
|
|
unsigned int enabled : 1;
|
|
|
|
unsigned int do_sport : 1;
|
|
__u16 sport;
|
|
|
|
unsigned int do_dport : 1;
|
|
__u16 dport;
|
|
|
|
// TCP flags.
|
|
unsigned int do_urg : 1;
|
|
unsigned int urg : 1;
|
|
|
|
unsigned int do_ack : 1;
|
|
unsigned int ack : 1;
|
|
|
|
unsigned int do_rst : 1;
|
|
unsigned int rst : 1;
|
|
|
|
unsigned int do_psh : 1;
|
|
unsigned int psh : 1;
|
|
|
|
unsigned int do_syn : 1;
|
|
unsigned int syn : 1;
|
|
|
|
unsigned int do_fin : 1;
|
|
unsigned int fin : 1;
|
|
};
|
|
|
|
struct udpopts
|
|
{
|
|
unsigned int enabled : 1;
|
|
|
|
unsigned int do_sport : 1;
|
|
__u16 sport;
|
|
|
|
unsigned int do_dport : 1;
|
|
__u16 dport;
|
|
};
|
|
|
|
struct icmpopts
|
|
{
|
|
unsigned int enabled : 1;
|
|
|
|
unsigned int do_code : 1;
|
|
__u8 code;
|
|
|
|
unsigned int do_type : 1;
|
|
__u8 type;
|
|
};
|
|
|
|
struct filter
|
|
{
|
|
__u8 id;
|
|
|
|
unsigned int enabled : 1;
|
|
|
|
__u8 action;
|
|
|
|
__u32 srcip;
|
|
__u32 dstip;
|
|
|
|
__u32 srcip6[4];
|
|
__u32 dstip6[4];
|
|
|
|
unsigned int do_min_ttl : 1;
|
|
__u8 min_ttl;
|
|
|
|
unsigned int do_max_ttl : 1;
|
|
__u8 max_ttl;
|
|
|
|
unsigned int do_min_len : 1;
|
|
__u16 min_len;
|
|
|
|
unsigned int do_max_len : 1;
|
|
__u16 max_len;
|
|
|
|
unsigned int do_tos : 1;
|
|
int8_t tos;
|
|
|
|
unsigned int do_pps : 1;
|
|
__u64 pps;
|
|
|
|
unsigned int do_bps : 1;
|
|
__u64 bps;
|
|
|
|
__u64 blocktime;
|
|
|
|
struct tcpopts tcpopts;
|
|
struct udpopts udpopts;
|
|
struct icmpopts icmpopts;
|
|
};
|
|
|
|
struct stats
|
|
{
|
|
__u64 allowed;
|
|
__u64 dropped;
|
|
};
|
|
|
|
struct ip_stats
|
|
{
|
|
__u64 pps;
|
|
__u64 bps;
|
|
__u64 tracking;
|
|
}; |