TLDKv2dev-next-socket

Signed-off-by: Jianfeng Tan <henry.tjf@antfin.com>
Signed-off-by: Jielong Zhou <jielong.zjl@antfin.com>
Signed-off-by: Jian Zhang <wuzai.zj@antfin.com>
Signed-off-by: Chen Zhao <winters.zc@antfin.com>
Change-Id: I55c39de4c6cd30f991f35631eb507f770230f08e

1 files changed, 56 insertions, 0 deletions

diff --git a/test/packetdrill/gre_packet.c b/test/packetdrill/gre_packet.c
new file mode 100644
index 0000000..235e6d9
--- /dev/null
+++ b/test/packetdrill/gre_packet.c
@@ -0,0 +1,56 @@
+/*
+ * 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 GRE packets.
+ */
+
+#include "gre_packet.h"
+
+#include "ip_packet.h"
+#include "gre.h"
+
+int gre_header_append(struct packet *packet, const struct gre *gre, char **error)
+{
+	struct header *header;
+
+	header = packet_append_header(packet, HEADER_GRE, gre_len(gre));
+	if (header == NULL) {
+		asprintf(error, "too many headers");
+		return STATUS_ERR;
+	}
+
+	memcpy(header->h.gre, gre, gre_len(gre));
+
+	return STATUS_OK;
+}
+
+int gre_header_finish(struct packet *packet,
+		      struct header *header, struct header *next_inner)
+{
+	struct gre *gre = header->h.gre;
+	int gre_bytes = gre_len(gre) + next_inner->total_bytes;
+
+	gre->proto = htons(header_type_info(next_inner->type)->eth_proto);
+
+	header->total_bytes = gre_bytes;
+
+	return STATUS_OK;
+}