mirror of https://github.com/Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
141 lines
3.9 KiB
141 lines
3.9 KiB
// SPDX-License-Identifier: GPL-2.0 |
|
#ifndef _XDP_SAMPLE_BPF_H |
|
#define _XDP_SAMPLE_BPF_H |
|
|
|
#include "vmlinux.h" |
|
#include <bpf/bpf_tracing.h> |
|
#include <bpf/bpf_core_read.h> |
|
#include <bpf/bpf_helpers.h> |
|
|
|
#include "xdp_sample_shared.h" |
|
|
|
#define ETH_ALEN 6 |
|
#define ETH_P_802_3_MIN 0x0600 |
|
#define ETH_P_8021Q 0x8100 |
|
#define ETH_P_8021AD 0x88A8 |
|
#define ETH_P_IP 0x0800 |
|
#define ETH_P_IPV6 0x86DD |
|
#define ETH_P_ARP 0x0806 |
|
#define IPPROTO_ICMPV6 58 |
|
|
|
#define EINVAL 22 |
|
#define ENETDOWN 100 |
|
#define EMSGSIZE 90 |
|
#define EOPNOTSUPP 95 |
|
#define ENOSPC 28 |
|
|
|
typedef struct { |
|
__uint(type, BPF_MAP_TYPE_ARRAY); |
|
__uint(map_flags, BPF_F_MMAPABLE); |
|
__type(key, unsigned int); |
|
__type(value, struct datarec); |
|
} array_map; |
|
|
|
extern array_map rx_cnt; |
|
extern const volatile int nr_cpus; |
|
|
|
enum { |
|
XDP_REDIRECT_SUCCESS = 0, |
|
XDP_REDIRECT_ERROR = 1 |
|
}; |
|
|
|
static __always_inline void swap_src_dst_mac(void *data) |
|
{ |
|
unsigned short *p = data; |
|
unsigned short dst[3]; |
|
|
|
dst[0] = p[0]; |
|
dst[1] = p[1]; |
|
dst[2] = p[2]; |
|
p[0] = p[3]; |
|
p[1] = p[4]; |
|
p[2] = p[5]; |
|
p[3] = dst[0]; |
|
p[4] = dst[1]; |
|
p[5] = dst[2]; |
|
} |
|
|
|
#if defined(__BYTE_ORDER__) && defined(__ORDER_LITTLE_ENDIAN__) && \ |
|
__BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ |
|
#define bpf_ntohs(x) __builtin_bswap16(x) |
|
#define bpf_htons(x) __builtin_bswap16(x) |
|
#elif defined(__BYTE_ORDER__) && defined(__ORDER_BIG_ENDIAN__) && \ |
|
__BYTE_ORDER__ == __ORDER_BIG_ENDIAN__ |
|
#define bpf_ntohs(x) (x) |
|
#define bpf_htons(x) (x) |
|
#else |
|
# error "Endianness detection needs to be set up for your compiler?!" |
|
#endif |
|
|
|
/* |
|
* Note: including linux/compiler.h or linux/kernel.h for the macros below |
|
* conflicts with vmlinux.h include in BPF files, so we define them here. |
|
* |
|
* Following functions are taken from kernel sources and |
|
* break aliasing rules in their original form. |
|
* |
|
* While kernel is compiled with -fno-strict-aliasing, |
|
* perf uses -Wstrict-aliasing=3 which makes build fail |
|
* under gcc 4.4. |
|
* |
|
* Using extra __may_alias__ type to allow aliasing |
|
* in this case. |
|
*/ |
|
typedef __u8 __attribute__((__may_alias__)) __u8_alias_t; |
|
typedef __u16 __attribute__((__may_alias__)) __u16_alias_t; |
|
typedef __u32 __attribute__((__may_alias__)) __u32_alias_t; |
|
typedef __u64 __attribute__((__may_alias__)) __u64_alias_t; |
|
|
|
static __always_inline void __read_once_size(const volatile void *p, void *res, int size) |
|
{ |
|
switch (size) { |
|
case 1: *(__u8_alias_t *) res = *(volatile __u8_alias_t *) p; break; |
|
case 2: *(__u16_alias_t *) res = *(volatile __u16_alias_t *) p; break; |
|
case 4: *(__u32_alias_t *) res = *(volatile __u32_alias_t *) p; break; |
|
case 8: *(__u64_alias_t *) res = *(volatile __u64_alias_t *) p; break; |
|
default: |
|
asm volatile ("" : : : "memory"); |
|
__builtin_memcpy((void *)res, (const void *)p, size); |
|
asm volatile ("" : : : "memory"); |
|
} |
|
} |
|
|
|
static __always_inline void __write_once_size(volatile void *p, void *res, int size) |
|
{ |
|
switch (size) { |
|
case 1: *(volatile __u8_alias_t *) p = *(__u8_alias_t *) res; break; |
|
case 2: *(volatile __u16_alias_t *) p = *(__u16_alias_t *) res; break; |
|
case 4: *(volatile __u32_alias_t *) p = *(__u32_alias_t *) res; break; |
|
case 8: *(volatile __u64_alias_t *) p = *(__u64_alias_t *) res; break; |
|
default: |
|
asm volatile ("" : : : "memory"); |
|
__builtin_memcpy((void *)p, (const void *)res, size); |
|
asm volatile ("" : : : "memory"); |
|
} |
|
} |
|
|
|
#define READ_ONCE(x) \ |
|
({ \ |
|
union { typeof(x) __val; char __c[1]; } __u = \ |
|
{ .__c = { 0 } }; \ |
|
__read_once_size(&(x), __u.__c, sizeof(x)); \ |
|
__u.__val; \ |
|
}) |
|
|
|
#define WRITE_ONCE(x, val) \ |
|
({ \ |
|
union { typeof(x) __val; char __c[1]; } __u = \ |
|
{ .__val = (val) }; \ |
|
__write_once_size(&(x), __u.__c, sizeof(x)); \ |
|
__u.__val; \ |
|
}) |
|
|
|
/* Add a value using relaxed read and relaxed write. Less expensive than |
|
* fetch_add when there is no write concurrency. |
|
*/ |
|
#define NO_TEAR_ADD(x, val) WRITE_ONCE((x), READ_ONCE(x) + (val)) |
|
#define NO_TEAR_INC(x) NO_TEAR_ADD((x), 1) |
|
|
|
#define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0])) |
|
|
|
#endif
|
|
|