[linux-2.6-block.git] / samples / bpf / sockex1_kern.c

#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";
Commit	Line	Data
a8085782 AS	1	#include <uapi/linux/bpf.h>
a8085782 AS	2	#include <uapi/linux/if_ether.h>
614cd3bd	3	#include <uapi/linux/if_packet.h>
a8085782 AS	4	#include <uapi/linux/ip.h>
	5	#include "bpf_helpers.h"
	6
	7	struct bpf_map_def SEC("maps") my_map = {
	8	.type = BPF_MAP_TYPE_ARRAY,
	9	.key_size = sizeof(u32),
	10	.value_size = sizeof(long),
	11	.max_entries = 256,
	12	};
	13
	14	SEC("socket1")
614cd3bd	15	int bpf_prog1(struct __sk_buff *skb)
a8085782 AS	16	{
	17	int index = load_byte(skb, ETH_HLEN + offsetof(struct iphdr, protocol));
	18	long *value;
	19
614cd3bd AS	20	if (skb->pkt_type != PACKET_OUTGOING)
	21	return 0;
	22
a8085782 AS	23	value = bpf_map_lookup_elem(&my_map, &index);
a8085782 AS	24	if (value)
614cd3bd	25	__sync_fetch_and_add(value, skb->len);
a8085782 AS	26
	27	return 0;
	28	}
	29	char _license[] SEC("license") = "GPL";