From 9d690477c9fee3eb9bc094dab91fd0f0b882b7ef Mon Sep 17 00:00:00 2001
From: Andreas Steffen <andreas.steffen@strongswan.org>
Date: Mon, 8 Aug 2011 20:13:32 +0200
Subject: [PATCH] output PEP device addresses as metadata

---
 .../plugins/tnc_ifmap/tnc_ifmap_listener.c    | 30 +++++++++
 .../plugins/tnc_ifmap/tnc_ifmap_soap.c        | 62 ++++++++++++++++++-
 .../plugins/tnc_ifmap/tnc_ifmap_soap.h        |  9 +++
 3 files changed, 100 insertions(+), 1 deletion(-)

diff --git a/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_listener.c b/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_listener.c
index aee36d6bb..4a2a2264d 100644
--- a/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_listener.c
+++ b/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_listener.c
@@ -17,6 +17,7 @@
 #include "tnc_ifmap_soap.h"
 
 #include <daemon.h>
+#include <hydra.h>
 #include <debug.h>
 
 typedef struct private_tnc_ifmap_listener_t private_tnc_ifmap_listener_t;
@@ -61,6 +62,30 @@ static bool publish_ike_sa(private_tnc_ifmap_listener_t *this,
 									   host, up);
 }
 
+/**
+ * Publish PEP device-ip metadata
+ */
+static bool publish_device_ip_addresses(private_tnc_ifmap_listener_t *this)
+{
+	enumerator_t *enumerator;
+	host_t *host;
+	bool success = TRUE;
+
+	enumerator = hydra->kernel_interface->create_address_enumerator(
+							hydra->kernel_interface, FALSE, FALSE);
+	while (enumerator->enumerate(enumerator, &host))
+	{
+		if (!this->ifmap->publish_device_ip(this->ifmap, host))
+		{
+			success = FALSE;
+			break;
+		}
+	}
+	enumerator->destroy(enumerator);
+
+	return success;
+}
+
 /**
  * Publish all IKE_SA metadata
  */
@@ -136,6 +161,11 @@ tnc_ifmap_listener_t *tnc_ifmap_listener_create(bool reload)
 		destroy(this);
 		return NULL;
 	}
+	if (!publish_device_ip_addresses(this))
+	{
+		destroy(this);
+		return NULL;
+	}
 	if (reload)
 	{
 		if (!reload_metadata(this))
diff --git a/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_soap.c b/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_soap.c
index b1c7f5a0f..d9b20df28 100644
--- a/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_soap.c
+++ b/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_soap.c
@@ -261,7 +261,35 @@ static axiom_node_t* create_ip_address(private_tnc_ifmap_soap_t *this,
 
 	el = axiom_element_create(this->env, NULL, "ip-address", NULL, &node);
 
-	snprintf(buf, BUF_LEN, "%H", host);
+	if (host->get_family(host) == AF_INET6)
+	{
+		chunk_t address;
+		int len, written, i;
+		char *pos;
+		bool first = TRUE;
+
+		/* output IPv6 address in canonical IF-MAP 2.0 format */
+		address = host->get_address(host);
+		pos = buf;
+		len = sizeof(buf);
+
+		for (i = 0; i < address.len; i = i + 2)
+		{
+			written = snprintf(pos, len, "%s%x", first ? "" : ":",
+							   256*address.ptr[i] +  address.ptr[i+1]);
+			if (written < 0 || written > len)
+			{
+				break;
+			}
+			pos += written;
+			len -= written;
+			first = FALSE;
+		}
+	}
+	else
+	{
+		snprintf(buf, BUF_LEN, "%H", host);
+	}
 	attr = axiom_attribute_create(this->env, "value", buf, NULL);	
 	axiom_element_add_attribute(el, this->env, attr, node);
 
@@ -427,6 +455,37 @@ METHOD(tnc_ifmap_soap_t, publish_ike_sa, bool,
 	return send_receive(this, "publish", request, "publishReceived", NULL);
 }
 
+METHOD(tnc_ifmap_soap_t, publish_device_ip, bool,
+	private_tnc_ifmap_soap_t *this, host_t *host)
+{
+	axiom_node_t *request, *node;
+	axiom_element_t *el;
+	axiom_namespace_t *ns, *ns_meta;
+	axiom_attribute_t *attr;
+
+	/* build publish request */
+ 	ns = axiom_namespace_create(this->env, IFMAP_NS, "ifmap");
+ 	el = axiom_element_create(this->env, NULL, "publish", ns, &request);
+	ns_meta = axiom_namespace_create(this->env, IFMAP_META_NS, "meta");
+	axiom_element_declare_namespace(el, this->env, request, ns_meta);	
+	attr = axiom_attribute_create(this->env, "session-id", this->session_id,
+								  NULL);	
+	axiom_element_add_attribute(el, this->env, attr, request);
+	el = axiom_element_create(this->env, NULL, "update", NULL, &node);
+	axiom_node_add_child(request, this->env, node);
+
+	/* add device, ip-address and metadata */
+	axiom_node_add_child(node, this->env,
+							 create_device(this));
+	axiom_node_add_child(node, this->env,
+							 create_ip_address(this, host));
+	axiom_node_add_child(node, this->env,
+							 create_metadata(this, "device-ip"));
+
+	/* send publish request and receive publishReceived */
+	return send_receive(this, "publish", request, "publishReceived", NULL);
+}
+
 METHOD(tnc_ifmap_soap_t, endSession, bool,
 	private_tnc_ifmap_soap_t *this)
 {
@@ -502,6 +561,7 @@ tnc_ifmap_soap_t *tnc_ifmap_soap_create()
 			.newSession = _newSession,
 			.purgePublisher = _purgePublisher,
 			.publish_ike_sa = _publish_ike_sa,
+			.publish_device_ip = _publish_device_ip,
 			.endSession = _endSession,
 			.destroy = _destroy,
 		},
diff --git a/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_soap.h b/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_soap.h
index 9d24425fe..833a748f4 100644
--- a/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_soap.h
+++ b/src/libcharon/plugins/tnc_ifmap/tnc_ifmap_soap.h
@@ -60,6 +60,15 @@ struct tnc_ifmap_soap_t {
 						   identification_t *id, bool is_user,
 						   host_t *host, bool up);
 
+	/**
+	 * Publish PEP device-ip metadata 
+	 *
+	 * @param host			IP address of local endpoint
+	 * @return				TRUE if command was successful
+	 */
+
+	bool (*publish_device_ip)(tnc_ifmap_soap_t *this, host_t *host);
+
 	/**
 	 * Ends an IF-MAP session
 	 *