mirror of
https://github.com/xdp-project/bpf-examples.git
synced 2024-05-06 15:54:53 +00:00
583f7a213f
Makes sure the ethertype is set correctly depending on the type of encapsulation. Fixes #2. Signed-off-by: Toke Høiland-Jørgensen <toke@redhat.com>
51 lines
914 B
C
51 lines
914 B
C
#include <linux/bpf.h>
|
|
#include <linux/in6.h>
|
|
#include <bpf/bpf_helpers.h>
|
|
#include <xdp/parsing_helpers.h>
|
|
#include "encap.h"
|
|
|
|
#ifdef IPV6
|
|
#define OFFSET sizeof(struct ipv6hdr)
|
|
#else
|
|
#define OFFSET sizeof(struct iphdr)
|
|
#endif
|
|
|
|
SEC("prog") int xdp_encap(struct xdp_md *ctx)
|
|
{
|
|
volatile struct ethhdr *ehdr, old_ehdr = {};
|
|
volatile void *data, *data_end;
|
|
size_t offset = OFFSET;
|
|
int ret = XDP_ABORTED;
|
|
|
|
data = (void *)(long)ctx->data;
|
|
data_end = (void *)(long)ctx->data_end;
|
|
|
|
ehdr = data;
|
|
if (ehdr + 1 > data_end)
|
|
goto out;
|
|
old_ehdr = *ehdr;
|
|
|
|
if (bpf_xdp_adjust_head(ctx, -offset))
|
|
goto out;
|
|
|
|
data = (void *)(long)ctx->data;
|
|
data_end = (void *)(long)ctx->data_end;
|
|
|
|
ehdr = data;
|
|
|
|
if (ehdr + 1 > data_end)
|
|
goto out;
|
|
*ehdr = old_ehdr;
|
|
|
|
#ifdef IPV6
|
|
encap_ipv6(data, data_end);
|
|
#else
|
|
encap_ipv4(data, data_end);
|
|
#endif
|
|
ret = XDP_PASS;
|
|
out:
|
|
return ret;
|
|
}
|
|
|
|
char _license[] SEC("license") = "GPL";
|