#include <uapi/linux/bpf.h> | |
#include <uapi/linux/if_ether.h> | |
#include <uapi/linux/if_packet.h> | |
#include <uapi/linux/ip.h> | |
#include "bpf_helpers.h" | |
struct bpf_map_def SEC("maps") my_map = { | |
.type = BPF_MAP_TYPE_ARRAY, | |
.key_size = sizeof(u32), | |
.value_size = sizeof(long), | |
.max_entries = 256, | |
}; | |
SEC("socket1") | |
int bpf_prog1(struct __sk_buff *skb) | |
{ | |
int index = load_byte(skb, ETH_HLEN + offsetof(struct iphdr, protocol)); | |
long *value; | |
if (skb->pkt_type != PACKET_OUTGOING) | |
return 0; | |
value = bpf_map_lookup_elem(&my_map, &index); | |
if (value) | |
__sync_fetch_and_add(value, skb->len); | |
return 0; | |
} | |
char _license[] SEC("license") = "GPL"; |