From 733714125b0a14223b9292b6a57c58dfcd4168ff Mon Sep 17 00:00:00 2001
From: Neal Cardwell <ncardwell@google.com>
Date: Fri, 22 Nov 2013 11:45:11 -0500
Subject: [PATCH] net-test: packetdrill MPLS support: appending MPLS headers to
packets
Basic code for filling in MPLS headers and appending them to packets.
Change-Id: Id31ef8e97de3e9b95c3b05ff42aade6a3d8490a5
---
gtests/net/packetdrill/mpls_packet.c | 77 ++++++++++++++++++++++++++++
gtests/net/packetdrill/mpls_packet.h | 57 ++++++++++++++++++++
2 files changed, 134 insertions(+)
create mode 100644 gtests/net/packetdrill/mpls_packet.c
create mode 100644 gtests/net/packetdrill/mpls_packet.h
diff --git a/gtests/net/packetdrill/mpls_packet.c b/gtests/net/packetdrill/mpls_packet.c
new file mode 100644
index 00000000..24486810
--- /dev/null
+++ b/gtests/net/packetdrill/mpls_packet.c
@@ -0,0 +1,77 @@
+/*
+ * Copyright 2013 Google Inc.
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ * 02110-1301, USA.
+ */
+/*
+ * Author: ncardwell@google.com (Neal Cardwell)
+ *
+ * Implementation for module for formatting MPLS packets.
+ */
+
+#include "mpls_packet.h"
+
+#include "mpls.h"
+
+int new_mpls_stack_entry(s64 label, s64 traffic_class,
+ bool is_stack_bottom, s64 ttl,
+ struct mpls *mpls, char **error)
+{
+ if ((label < 0) || (label >= (1<<20))) {
+ asprintf(error, "MPLS label out of range for 20 bits");
+ return STATUS_ERR;
+ }
+
+ if ((traffic_class < 0) || (traffic_class >= (1<<3))) {
+ asprintf(error, "MPLS traffic_class out of range for 3 bits");
+ return STATUS_ERR;
+ }
+
+ if ((ttl < 0) || (ttl >= (1<<8))) {
+ asprintf(error, "MPLS ttl out of range for 8 bits");
+ return STATUS_ERR;
+ }
+
+ mpls_entry_set(label, traffic_class, is_stack_bottom, ttl, mpls);
+ return STATUS_OK;
+}
+
+int mpls_header_append(struct packet *packet, struct mpls_stack *mpls_stack,
+ char **error)
+{
+ struct header *header;
+ int mpls_bytes = mpls_stack->length * sizeof(struct mpls);
+
+ header = packet_append_header(packet, HEADER_MPLS, mpls_bytes);
+ if (header == NULL) {
+ asprintf(error, "too many headers");
+ return STATUS_ERR;
+ }
+
+ memcpy(header->h.mpls, mpls_stack->entries, mpls_bytes);
+
+ return STATUS_OK;
+}
+
+int mpls_header_finish(struct packet *packet,
+ struct header *header, struct header *next_inner)
+{
+ int mpls_bytes = header->header_bytes + next_inner->total_bytes;
+
+ header->total_bytes = mpls_bytes;
+
+ return STATUS_OK;
+}
diff --git a/gtests/net/packetdrill/mpls_packet.h b/gtests/net/packetdrill/mpls_packet.h
new file mode 100644
index 00000000..16079c0a
--- /dev/null
+++ b/gtests/net/packetdrill/mpls_packet.h
@@ -0,0 +1,57 @@
+/*
+ * Copyright 2013 Google Inc.
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ * 02110-1301, USA.
+ */
+/*
+ * Author: ncardwell@google.com (Neal Cardwell)
+ *
+ * Interface for module for formatting MPLS packets.
+ */
+
+#ifndef __MPLS_PACKET_H__
+#define __MPLS_PACKET_H__
+
+#include "types.h"
+
+#include "mpls.h"
+#include "packet.h"
+
+/* Fill in the given MPLS label stack entry with the given field
+ * values, validating that actual parameter value fits inside the
+ * width of the field on the wire. On success, return STATUS_OK; on
+ * error return STATUS_ERR and fill in a malloc-allocated error
+ * message in *error.
+ */
+extern int new_mpls_stack_entry(s64 label, s64 traffic_class,
+ bool is_stack_bottom, s64 ttl,
+ struct mpls *mpls, char **error);
+
+/* Append an MPLS header to the end of the given packet. On success,
+ * return STATUS_OK; on error return STATUS_ERR and fill in a
+ * malloc-allocated error message in *error.
+ */
+extern int mpls_header_append(struct packet *packet,
+ struct mpls_stack *mpls_stack,
+ char **error);
+
+/* Finalize the MPLS header by filling in all necessary fields that
+ * were not filled in at parse time.
+ */
+extern int mpls_header_finish(struct packet *packet,
+ struct header *header, struct header *next_inner);
+
+#endif /* __MPLS_PACKET_H__ */
--
GitLab