summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--trafgen_l2.c32
-rw-r--r--trafgen_l2.h7
-rw-r--r--trafgen_proto.h1
3 files changed, 40 insertions, 0 deletions
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,