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 at 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

Reply via email to