diff --git a/sys/netgraph/netflow/ng_netflow.c b/sys/netgraph/netflow/ng_netflow.c index df5caa391dc9..f0cb86792699 100644 --- a/sys/netgraph/netflow/ng_netflow.c +++ b/sys/netgraph/netflow/ng_netflow.c @@ -42,6 +42,7 @@ static const char rcs_id[] = #include #include #include +#include #include #include #include @@ -514,6 +515,19 @@ ng_netflow_rcvdata (hook_p hook, item_p item) eh = mtod(m, struct ether_header *); ip = (struct ip *)(eh + 1); break; + case ETHERTYPE_VLAN: + { + struct ether_vlan_header *evh; + + M_CHECK(sizeof(struct ether_vlan_header) - + sizeof(struct ether_header)); + evh = mtod(m, struct ether_vlan_header *); + if (ntohs(evh->evl_proto) == ETHERTYPE_IP) { + M_CHECK(sizeof(struct ip)); + ip = (struct ip *)(evh + 1); + break; + } + } default: goto bypass; /* pass this frame */ } @@ -551,7 +565,21 @@ ng_netflow_rcvdata (hook_p hook, item_p item) struct ether_header *eh; eh = mtod(m, struct ether_header *); - ip = (struct ip *)(eh + 1); + switch (ntohs(eh->ether_type)) { + case ETHERTYPE_IP: + ip = (struct ip *)(eh + 1); + break; + case ETHERTYPE_VLAN: + { + struct ether_vlan_header *evh; + + evh = mtod(m, struct ether_vlan_header *); + ip = (struct ip *)(evh + 1); + break; + } + default: + panic("ng_netflow entered deadcode"); + } break; } case DLT_RAW: