From d053535a66ff864d309dec2303927e410b27e5b6 Mon Sep 17 00:00:00 2001
From: Vadim Kochan <vadim4j@gmail.com>
Date: Tue, 26 Jan 2016 22:25:05 +0200
Subject: trafgen: proto: Add functon to fill field with device ipv4 address

Add helper which fills a device's ipv4 addr to the specified protocol field.

It will be used by protocols like ARP, IPv4, etc.

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

diff --git a/trafgen_proto.c b/trafgen_proto.c
index 4976e54..2c090b2 100644
--- a/trafgen_proto.c
+++ b/trafgen_proto.c
@@ -5,6 +5,7 @@
 
 #include <stddef.h>
 #include <string.h>
+#include <netinet/in.h>
 #include <linux/if_ether.h>
 
 #include "dev.h"
@@ -321,6 +322,37 @@ void proto_field_set_default_dev_mac(struct proto_hdr *hdr, uint32_t fid)
 	__proto_field_set_dev_mac(hdr, fid, true);
 }
 
+static void __proto_field_set_dev_ipv4(struct proto_hdr *hdr, uint32_t fid,
+				       bool is_default)
+{
+	struct sockaddr_storage ss = { };
+	struct sockaddr_in *ss4;
+	uint32_t ip_addr;
+	int ret;
+
+	if (proto_field_is_set(hdr, fid))
+		return;
+
+	ret = device_address(hdr->ctx->dev, AF_INET, &ss);
+	if (ret < 0)
+		panic("Could not get device IPv4 address\n");
+
+	ss4 = (struct sockaddr_in *) &ss;
+	ip_addr = ss4->sin_addr.s_addr;
+
+	__proto_field_set_bytes(hdr, fid, (uint8_t *)&ip_addr, is_default, false);
+}
+
+void proto_field_set_dev_ipv4(struct proto_hdr *hdr, uint32_t fid)
+{
+	__proto_field_set_dev_ipv4(hdr, fid, false);
+}
+
+void proto_field_set_default_dev_ipv4(struct proto_hdr *hdr, uint32_t fid)
+{
+	__proto_field_set_dev_ipv4(hdr, fid, true);
+}
+
 void protos_init(const char *dev)
 {
 	struct proto_hdr *p;
diff --git a/trafgen_proto.h b/trafgen_proto.h
index 517b278..0267cf6 100644
--- a/trafgen_proto.h
+++ b/trafgen_proto.h
@@ -97,4 +97,7 @@ extern void proto_field_set_default_be32(struct proto_hdr *hdr, uint32_t fid,
 extern void proto_field_set_dev_mac(struct proto_hdr *hdr, uint32_t fid);
 extern void proto_field_set_default_dev_mac(struct proto_hdr *hdr, uint32_t fid);
 
+extern void proto_field_set_dev_ipv4(struct proto_hdr *hdr, uint32_t fid);
+extern void proto_field_set_default_dev_ipv4(struct proto_hdr *hdr, uint32_t fid);
+
 #endif /* TRAFGEN_PROTO_I_H */
-- 
cgit v1.2.3-54-g00ecf