// Copyright (c) PLUMgrid, Inc.
|
// Licensed under the Apache License, Version 2.0 (the "License")
|
|
BPF_PROG_ARRAY(jump, 64);
|
BPF_ARRAY(stats, u64, 64);
|
|
enum states {
|
S_EOP = 1,
|
S_ETHER,
|
S_ARP,
|
S_IP
|
};
|
|
int parse_ether(struct __sk_buff *skb) {
|
size_t cur = 0;
|
size_t next = cur + 14;
|
|
int key = S_ETHER;
|
u64 *leaf = stats.lookup(&key);
|
if (leaf) (*leaf)++;
|
|
switch (bpf_dext_pkt(skb, cur + 12, 0, 16)) {
|
case 0x0800: jump.call(skb, S_IP);
|
case 0x0806: jump.call(skb, S_ARP);
|
}
|
jump.call(skb, S_EOP);
|
return 1;
|
}
|
|
int parse_arp(struct __sk_buff *skb) {
|
size_t cur = 14; // TODO: get from ctx
|
size_t next = cur + 28;
|
|
int key = S_ARP;
|
u64 *leaf = stats.lookup(&key);
|
if (leaf) (*leaf)++;
|
|
jump.call(skb, S_EOP);
|
return 1;
|
}
|
|
int parse_ip(struct __sk_buff *skb) {
|
size_t cur = 14; // TODO: get from ctx
|
size_t next = cur + 20;
|
|
int key = S_IP;
|
u64 *leaf = stats.lookup(&key);
|
if (leaf) (*leaf)++;
|
|
jump.call(skb, S_EOP);
|
return 1;
|
}
|
|
int eop(struct __sk_buff *skb) {
|
int key = S_EOP;
|
u64 *leaf = stats.lookup(&key);
|
if (leaf) (*leaf)++;
|
return 1;
|
}
|
