trafgen: l2: Add support for PFC header

Add PFC header fields setting (prio vector, time per priority).
By default Ethernet header is pushed with required fields.

Signed-off-by: Vadim Kochan <vadim4j@gmail.com>
Signed-off-by: Tobias Klauser <tklauser@distanz.ch>

3 files changed, 64 insertions, 0 deletions

diff --git a/trafgen_l2.c b/trafgen_l2.c
index 8d2d285..b27c9dd 100644
--- a/trafgen_l2.c
+++ b/trafgen_l2.c
@@ -31,6 +31,7 @@ static uint16_t pid_to_eth(enum proto_id pid)
 	case PROTO_VLAN:
 		return ETH_P_8021Q;
 	case PROTO_PAUSE:
+	case PROTO_PFC:
 		return ETH_P_PAUSE;
 	default:
 		bug();
@@ -80,6 +81,46 @@ static struct proto_ops pause_proto_ops = {
 	.header_init	= pause_header_init,
 };
 
+static struct proto_field pfc_fields[] = {
+	{ .id = PFC_OPCODE,  .len = 2, .offset = 0 },
+	{ .id = PFC_PRIO,    .len = 2, .offset = 2 },
+	{ .id = PFC_PRIO_0,  .len = 2, .offset = 2, .mask = 0x0001 },
+	{ .id = PFC_PRIO_1,  .len = 2, .offset = 2, .mask = 0x0002, .shift = 1 },
+	{ .id = PFC_PRIO_2,  .len = 2, .offset = 2, .mask = 0x0004, .shift = 2 },
+	{ .id = PFC_PRIO_3,  .len = 2, .offset = 2, .mask = 0x0008, .shift = 3 },
+	{ .id = PFC_PRIO_4,  .len = 2, .offset = 2, .mask = 0x0010, .shift = 4 },
+	{ .id = PFC_PRIO_5,  .len = 2, .offset = 2, .mask = 0x0020, .shift = 5 },
+	{ .id = PFC_PRIO_6,  .len = 2, .offset = 2, .mask = 0x0040, .shift = 6 },
+	{ .id = PFC_PRIO_7,  .len = 2, .offset = 2, .mask = 0x0080, .shift = 7 },
+	{ .id = PFC_TIME_0,  .len = 2, .offset = 4,  },
+	{ .id = PFC_TIME_1,  .len = 2, .offset = 6,  },
+	{ .id = PFC_TIME_2,  .len = 2, .offset = 8,  },
+	{ .id = PFC_TIME_3,  .len = 2, .offset = 10, },
+	{ .id = PFC_TIME_4,  .len = 2, .offset = 12, },
+	{ .id = PFC_TIME_5,  .len = 2, .offset = 14, },
+	{ .id = PFC_TIME_6,  .len = 2, .offset = 16, },
+	{ .id = PFC_TIME_7,  .len = 2, .offset = 18, },
+};
+
+static void pfc_header_init(struct proto_hdr *hdr)
+{
+	uint8_t eth_dst[6] = { 0x01, 0x80, 0xC2, 0x00, 0x00, 0x01 };
+
+	struct proto_hdr *lower;
+
+	lower = proto_lower_default_add(hdr, PROTO_ETH);
+	proto_field_set_default_bytes(lower, ETH_DST_ADDR, eth_dst);
+
+	proto_header_fields_add(hdr, pfc_fields, array_size(pfc_fields));
+	proto_field_set_default_be16(hdr, PFC_OPCODE, 0x0101);
+}
+
+static struct proto_ops pfc_proto_ops = {
+	.id		= PROTO_PFC,
+	.layer		= PROTO_L2,
+	.header_init	= pfc_header_init,
+};
+
 static struct proto_field vlan_fields[] = {
 	/* TPID overlaps with Ethernet header and points to ether type */
 	{ .id = VLAN_TPID, .len = 2, .offset = -2 },
@@ -194,6 +235,7 @@ void protos_l2_init(void)
 {
 	proto_ops_register(&eth_proto_ops);
 	proto_ops_register(&pause_proto_ops);
+	proto_ops_register(&pfc_proto_ops);
 	proto_ops_register(&vlan_proto_ops);
 	proto_ops_register(&arp_proto_ops);
 	proto_ops_register(&mpls_proto_ops);
diff --git a/trafgen_l2.h b/trafgen_l2.h
index 7c8ef7d..dd1d947 100644
--- a/trafgen_l2.h
+++ b/trafgen_l2.h
@@ -12,6 +12,27 @@ enum pause_field {
 	PAUSE_TIME,
 };
 
+enum pfc_field {
+	PFC_OPCODE,
+	PFC_PRIO,
+	PFC_PRIO_0,
+	PFC_PRIO_1,
+	PFC_PRIO_2,
+	PFC_PRIO_3,
+	PFC_PRIO_4,
+	PFC_PRIO_5,
+	PFC_PRIO_6,
+	PFC_PRIO_7,
+	PFC_TIME_0,
+	PFC_TIME_1,
+	PFC_TIME_2,
+	PFC_TIME_3,
+	PFC_TIME_4,
+	PFC_TIME_5,
+	PFC_TIME_6,
+	PFC_TIME_7,
+};
+
 enum arp_field {
 	ARP_HTYPE,
 	ARP_PTYPE,
diff --git a/trafgen_proto.h b/trafgen_proto.h
index a1f6b66..ec152b4 100644
--- a/trafgen_proto.h
+++ b/trafgen_proto.h
@@ -9,6 +9,7 @@ enum proto_id {
 	PROTO_NONE = 0,
 	PROTO_ETH,
 	PROTO_PAUSE,
+	PROTO_PFC,
 	PROTO_VLAN,
 	PROTO_ARP,
 	PROTO_MPLS,