From 116835ed0d245f3e73173f2f2057265827416ce6 Mon Sep 17 00:00:00 2001
From: Vadim Kochan <vadim4j@gmail.com>
Date: Mon, 8 Feb 2016 08:01:52 +0200
Subject: trafgen: l2: Add MPLS header generation

Add support for MPLS header creating with fields:

    Label, TClass, Bottom-Stack, TTL

By default S-field is set to 1 (last label), but resets to 0
if lower MPLS header is added.

Signed-off-by: Vadim Kochan <vadim4j@gmail.com>
Signed-off-by: Tobias Klauser <tklauser@distanz.ch>
---
 trafgen_l2.c    | 32 ++++++++++++++++++++++++++++++++
 trafgen_l2.h    |  7 +++++++
 trafgen_proto.h |  1 +
 3 files changed, 40 insertions(+)

diff --git a/trafgen_l2.c b/trafgen_l2.c
index a7387b8..e6c5b59 100644
--- a/trafgen_l2.c
+++ b/trafgen_l2.c
@@ -26,6 +26,8 @@ static uint16_t pid_to_eth(enum proto_id pid)
 		return ETH_P_IP;
 	case PROTO_IP6:
 		return ETH_P_IPV6;
+	case PROTO_MPLS:
+		return ETH_P_MPLS_UC;
 	case PROTO_VLAN:
 		return ETH_P_8021Q;
 	default:
@@ -137,9 +139,39 @@ static struct proto_hdr arp_hdr = {
 	.header_init	= arp_header_init,
 };
 
+static struct proto_field mpls_fields[] = {
+	{ .id = MPLS_LABEL, .len = 4, .shift = 12, .mask = 0xfffff000 },
+	{ .id = MPLS_TC,    .len = 4, .shift = 9,  .mask = 0xe00 },
+	{ .id = MPLS_LAST,  .len = 4, .shift = 8,  .mask = 0x100 },
+	{ .id = MPLS_TTL,   .len = 4, .shift = 0,  .mask = 0xff },
+};
+
+static void mpls_header_init(struct proto_hdr *hdr)
+{
+	proto_lower_default_add(hdr, PROTO_ETH);
+
+	proto_header_fields_add(hdr, mpls_fields, array_size(mpls_fields));
+
+	proto_field_set_default_be32(hdr, MPLS_LAST, 1);
+}
+
+static void mpls_set_next_proto(struct proto_hdr *hdr, enum proto_id pid)
+{
+	if (pid == PROTO_MPLS)
+		proto_field_set_default_be32(hdr, MPLS_LAST, 0);
+}
+
+static struct proto_hdr mpls_hdr = {
+	.id		= PROTO_MPLS,
+	.layer		= PROTO_L2,
+	.header_init	= mpls_header_init,
+	.set_next_proto = mpls_set_next_proto,
+};
+
 void protos_l2_init(void)
 {
 	proto_header_register(&eth_hdr);
 	proto_header_register(&vlan_hdr);
 	proto_header_register(&arp_hdr);
+	proto_header_register(&mpls_hdr);
 }
diff --git a/trafgen_l2.h b/trafgen_l2.h
index 9beb2aa..14f0e84 100644
--- a/trafgen_l2.h
+++ b/trafgen_l2.h
@@ -28,6 +28,13 @@ enum vlan_field {
 	VLAN_ETYPE,
 };
 
+enum mpls_field {
+	MPLS_LABEL,
+	MPLS_TC,
+	MPLS_LAST,
+	MPLS_TTL,
+};
+
 extern void protos_l2_init(void);
 
 #endif /* TRAFGEN_L2_H */
diff --git a/trafgen_proto.h b/trafgen_proto.h
index 10ded79..31ac1c8 100644
--- a/trafgen_proto.h
+++ b/trafgen_proto.h
@@ -14,6 +14,7 @@ enum proto_id {
 	PROTO_ETH,
 	PROTO_VLAN,
 	PROTO_ARP,
+	PROTO_MPLS,
 	PROTO_IP4,
 	PROTO_IP6,
 	PROTO_UDP,
-- 
cgit v1.2.3-54-g00ecf