[dpdk-dev] [PATCH] examples: add ip version check for l3fwd app

[Andrey Chilikin <andrey.chilikin@intel.com>]

Added optional ip version check to l3fwd app to allow to detect
the ip version if mbuf ol_flags are not set in case of running
in a VM with emulated network controllers

Signed-off-by: Andrey Chilikin <andrey.chilikin@intel.com>
---
 examples/l3fwd/main.c |   22 ++++++++++++++++++++++
 1 files changed, 22 insertions(+), 0 deletions(-)

diff --git a/examples/l3fwd/main.c b/examples/l3fwd/main.c
index e32512e..4d4e5bc 100644
--- a/examples/l3fwd/main.c
+++ b/examples/l3fwd/main.c
@@ -76,6 +76,8 @@
 #define APP_LOOKUP_LPM                  1
 #define DO_RFC_1812_CHECKS
 
+#define DO_IP_VERSION_CHECK             0
+
 #ifndef APP_LOOKUP_METHOD
 #define APP_LOOKUP_METHOD             APP_LOOKUP_LPM
 #endif
@@ -953,6 +955,15 @@ l3fwd_simple_forward(struct rte_mbuf *m, uint8_t portid, struct lcore_conf *qcon
 	void *d_addr_bytes;
 	uint8_t dst_port;
 
+#if DO_IP_VERSION_CHECK
+	if (!(m->ol_flags & (PKT_RX_IPV4_HDR | PKT_RX_IPV6_HDR))) {
+		uint8_t ip_ver = *(uint8_t *)(rte_pktmbuf_mtod(m, unsigned char *) +
+			sizeof(struct ether_hdr)) >> 4;
+		if (ip_ver == 4)
+			m->ol_flags |= PKT_RX_IPV4_HDR;
+	}
+#endif
+
 	eth_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
 
 	if (m->ol_flags & PKT_RX_IPV4_HDR) {
@@ -1071,6 +1082,17 @@ get_dst_port(const struct lcore_conf *qconf, struct rte_mbuf *pkt,
 	struct ipv6_hdr *ipv6_hdr;
 	struct ether_hdr *eth_hdr;
 
+#if DO_IP_VERSION_CHECK
+	if (!(pkt->ol_flags & (PKT_RX_IPV4_HDR | PKT_RX_IPV6_HDR))) {
+		uint8_t ip_ver = *(uint8_t *)(rte_pktmbuf_mtod(pkt, unsigned char *) +
+			sizeof(struct ether_hdr)) >> 4;
+		if (ip_ver == 4)
+			pkt->ol_flags |= PKT_RX_IPV4_HDR;
+		else if (ip_ver == 6)
+			pkt->ol_flags |= PKT_RX_IPV6_HDR;
+	}
+#endif
+
 	if (pkt->ol_flags & PKT_RX_IPV4_HDR) {
 		if (rte_lpm_lookup(qconf->ipv4_lookup_struct, dst_ipv4,
 				&next_hop) != 0)
-- 
1.7.4.1