Isakmp_dpd task added.

src/libcharon/sa/ikev1/tasks/aggressive_mode.c

@@ -164,7 +164,7 @@ static status_t send_notify(private_aggressive_mode_t *this, notify_type_t type)
 	notify->set_spi_data(notify, spi);
 
 	this->ike_sa->queue_task(this->ike_sa,
-						(task_t*)informational_create(this->ike_sa, notify));
+						(task_t*)informational_create(this->ike_sa, notify, 0));
 	/* cancel all active/passive tasks in favour of informational */
 	return ALREADY_DONE;
 }

src/libcharon/sa/ikev1/tasks/informational.c

@@ -18,6 +18,8 @@
 #include <daemon.h>
 #include <sa/ikev1/tasks/isakmp_delete.h>
 #include <sa/ikev1/tasks/quick_delete.h>
+#include <sa/ikev1/tasks/isakmp_dpd.h>
+
 #include <encoding/payloads/delete_payload.h>
 
 typedef struct private_informational_t private_informational_t;
@@ -46,6 +48,16 @@ struct private_informational_t {
 	 * Delete subtask
 	 */
 	task_t *del;
+
+	/**
+	 * DPD subtask
+	 */
+	task_t *dpd;
+
+	/**
+	 * DPD sequence number
+	 */
+	u_int32_t dpd_seqnr;
 };
 
 METHOD(task_t, build_i, status_t,
@@ -80,6 +92,15 @@ METHOD(task_t, process_r, status_t,
 					this->ike_sa->set_condition(this->ike_sa,
 												COND_INIT_CONTACT_SEEN, TRUE);
 				}
+				else if (type == DPD_R_U_THERE)
+				{
+					DBG3(DBG_IKE, "received DPD request");
+					this->dpd = (task_t*)isakmp_dpd_create(this->ike_sa, notify, this->dpd_seqnr);
+				}
+				else if (type == DPD_R_U_THERE_ACK)
+				{
+					DBG3(DBG_IKE, "received DPD request ack");
+				}
 				else if (type < 16384)
 				{
 					DBG1(DBG_IKE, "received %N error notify",
@@ -123,6 +144,11 @@ METHOD(task_t, process_r, status_t,
 	{
 		return this->del->process(this->del, message);
 	}
+
+	if (this->dpd && status == SUCCESS)
+	{
+		return this->dpd->process(this->dpd, message);
+	}
 	return status;
 }
 
@@ -133,6 +159,13 @@ METHOD(task_t, build_r, status_t,
 	{
 		return this->del->build(this->del, message);
 	}
+
+	if (this->dpd)
+	{
+		status_t status = this->dpd->build(this->dpd, message);
+		this->dpd->destroy(this->dpd);
+		return status;
+	}
 	return FAILED;
 }
 
@@ -165,7 +198,7 @@ METHOD(task_t, destroy, void,
 /*
  * Described in header.
  */
-informational_t *informational_create(ike_sa_t *ike_sa, notify_payload_t *notify)
+informational_t *informational_create(ike_sa_t *ike_sa, notify_payload_t *notify, u_int32_t dpd_seqnr)
 {
 	private_informational_t *this;
 
@@ -179,6 +212,7 @@ informational_t *informational_create(ike_sa_t *ike_sa, notify_payload_t *notify
 		},
 		.ike_sa = ike_sa,
 		.notify = notify,
+		.dpd_seqnr = dpd_seqnr,
 	);
 
 	if (notify)

src/libcharon/sa/ikev1/tasks/informational.h

@@ -44,8 +44,9 @@ struct informational_t {
  *
  * @param ike_sa		IKE_SA this task works for
  * @param notify		notify to send as initiator, NULL if responder
+ * @param dpd_seqnr	DPD sequence number, incoming or outgoing
  * @return				task to handle by the task_manager
  */
-informational_t *informational_create(ike_sa_t *ike_sa, notify_payload_t *notify);
+informational_t *informational_create(ike_sa_t *ike_sa, notify_payload_t *notify, u_int32_t dpd_seqnr);
 
 #endif /** INFORMATIONAL_H_ @}*/

src/libcharon/sa/ikev1/tasks/isakmp_dpd.c

@@ -0,0 +1,278 @@
+#include "isakmp_dpd.h"
+
+#include <encoding/payloads/notify_payload.h>
+#include <sa/ikev1/tasks/informational.h>
+
+#include <daemon.h>
+
+#ifdef SLIPSTREAM
+/* Should be the last include */
+#include <ikev2_mem.h>
+#endif /* SLIPSTREAM */
+
+typedef struct private_isakmp_dpd_t private_isakmp_dpd_t;
+
+/**
+ * Private members of a isakmp_dpd_t task.
+ */
+struct private_isakmp_dpd_t {
+
+	/**
+	 * Public methods and task_t interface.
+	 */
+	isakmp_dpd_t public;
+
+	/**
+	 * Sequence number.
+	 */
+	u_int32_t seqnr;
+
+	/**
+	 * Notify payload, only provided for requests.
+	 */
+	notify_payload_t *notify;
+
+	/**
+	 * IKE SA we are serving.
+	 */
+	ike_sa_t *ike_sa;
+};
+
+/**
+ * Get DPD sequence number from notify payload.
+ */
+static bool get_seqnr(notify_payload_t *notify, u_int32_t *seqnr)
+{
+	chunk_t chunk = notify->get_notification_data(notify);
+
+	if( chunk.ptr && chunk.len == 4)
+	{
+		u_int32_t seqnr_read = *((u_int32_t*)chunk.ptr);
+
+		*seqnr = ntohl(seqnr_read);
+
+		return TRUE;
+	}
+
+	DBG1(DBG_IKE, "no DPD seqnr received");
+
+	return FALSE;
+}
+
+/**
+ * Add notify payload to message.
+ */
+static void add_notify(private_isakmp_dpd_t *this, message_t *message, notify_type_t type)
+{
+	notify_payload_t *notify;
+
+	ike_sa_id_t *ike_sa_id;
+	u_int64_t spi_i, spi_r;
+	u_int32_t seqnr;
+	chunk_t spi;
+
+	notify = notify_payload_create_from_protocol_and_type(NOTIFY_V1,
+		PROTO_IKE, type);
+
+	seqnr = htonl(this->seqnr);
+	ike_sa_id = this->ike_sa->get_id(this->ike_sa);
+	spi_i = ike_sa_id->get_initiator_spi(ike_sa_id);
+	spi_r = ike_sa_id->get_responder_spi(ike_sa_id);
+	spi = chunk_cata("cc", chunk_from_thing(spi_i), chunk_from_thing(spi_r));
+
+	notify->set_spi_data(notify, spi);
+	notify->set_notification_data(notify, chunk_from_thing(seqnr));
+
+	message->add_payload(message, (payload_t*)notify);
+}
+
+METHOD(isakmp_dpd_t, get_dpd_seqnr, u_int32_t,
+	private_isakmp_dpd_t *this)
+{
+	return this->seqnr;
+}
+
+METHOD(task_t, build_i, status_t,
+	private_isakmp_dpd_t *this, message_t *message)
+{
+	add_notify(this, message, DPD_R_U_THERE);
+
+	return NEED_MORE;
+}
+
+METHOD(task_t, build_r, status_t,
+			 private_isakmp_dpd_t *this, message_t *message)
+{
+	add_notify(this, message, DPD_R_U_THERE_ACK);
+
+	return SUCCESS;
+}
+
+METHOD(task_t, process_i, status_t,
+	private_isakmp_dpd_t *this, message_t *message)
+{
+	enumerator_t *enumerator;
+	notify_payload_t *notify;
+	notify_type_t type;
+	payload_t *payload;
+	task_t *info_task = NULL;
+
+	enumerator = message->create_payload_enumerator(message);
+	while (enumerator->enumerate(enumerator, &payload))
+	{
+		switch (payload->get_type(payload))
+		{
+			case NOTIFY_V1:
+				notify = (notify_payload_t*)payload;
+				type = notify->get_notify_type(notify);
+
+				if (type == DPD_R_U_THERE_ACK)
+				{
+					u_int32_t seqnr;
+
+					if (!get_seqnr(notify, &seqnr))
+					{
+						return FAILED;
+					}
+
+					if (this->seqnr != seqnr)
+					{
+						DBG1(DBG_IKE, "received DPD Ack with unexpected seqnr (%u) expect (%u)",seqnr,this->seqnr);
+						return SUCCESS;
+					}
+
+					DBG4(DBG_IKE, "received DPD Ack with seqnr (%u)",seqnr);
+
+					return SUCCESS;
+
+				}
+				else if (type == DPD_R_U_THERE)
+				{
+					u_int32_t expected = this->seqnr + 1;
+
+					if (!get_seqnr(notify, &this->seqnr))
+					{
+						return FAILED;
+					}
+
+					if (expected != 1 && this->seqnr != expected)
+					{
+						DBG1(DBG_IKE, "received DPD request with unexpected seqnr (%u) expect (%u)",
+							this->seqnr,expected);
+						return SUCCESS;
+					}
+
+					DBG4(DBG_IKE, "received DPD request with seqnr %u",this->seqnr);
+
+					this->public.task.build = _build_r;
+					return NEED_MORE;
+				}
+				else
+				{
+					info_task = (task_t*)informational_create(this->ike_sa, NULL, 0);
+				}
+				continue;
+
+			default:
+				continue;
+		}
+		break;
+	}
+	enumerator->destroy(enumerator);
+
+	if (info_task)
+	{
+		status_t status = info_task->process(info_task, message);
+		/* Assuming that the informational task will not need to send other replies than dpd */
+		info_task->destroy(info_task);
+		return status;
+	}
+
+	return SUCCESS;
+}
+
+METHOD(task_t, process_r, status_t,
+			 private_isakmp_dpd_t *this, message_t *message)
+{
+	u_int32_t expected = this->seqnr + 1;
+
+	if (this->notify)
+	{
+		if (!get_seqnr(this->notify, &this->seqnr))
+		{
+			return FAILED;
+		}
+
+		if (expected != 1 && this->seqnr != expected)
+		{
+			DBG1(DBG_IKE, "received DPD request with unexpected seqnr (%u) expect (%u)",
+				this->seqnr,expected);
+			return SUCCESS;
+		}
+
+		DBG4(DBG_IKE, "DPD request received with seqnr %u",this->seqnr);
+	}
+	else
+	{
+		DBG1(DBG_IKE, "no notify provided");
+		return FAILED;
+	}
+	return NEED_MORE;
+}
+
+
+METHOD(task_t, get_type, task_type_t,
+	private_isakmp_dpd_t *this)
+{
+	return TASK_ISAKMP_DPD;
+}
+
+
+METHOD(task_t, migrate, void,
+	private_isakmp_dpd_t *this, ike_sa_t *ike_sa)
+{
+	this->ike_sa = ike_sa;
+	this->seqnr = 0;
+
+}
+
+METHOD(task_t, destroy, void,
+	private_isakmp_dpd_t *this)
+{
+	free(this);
+}
+
+/*
+ * Described in header.
+ */
+isakmp_dpd_t *isakmp_dpd_create(ike_sa_t *ike_sa, notify_payload_t *notify, u_int32_t seqnr)
+{
+	private_isakmp_dpd_t *this;
+
+	INIT(this,
+		.public = {
+			.task = {
+				.get_type = _get_type,
+				.migrate = _migrate,
+				.destroy = _destroy,
+			},
+			.get_dpd_seqnr = _get_dpd_seqnr,
+		},
+		.notify = notify,
+		.ike_sa = ike_sa,
+		.seqnr = seqnr,
+	);
+
+	if (!notify)
+	{
+		this->public.task.build = _build_i;
+		this->public.task.process = _process_i;
+	}
+	else
+	{
+		this->public.task.build = _build_r;
+		this->public.task.process = _process_r;
+	}
+
+	return &this->public;
+}

src/libcharon/sa/ikev1/tasks/isakmp_dpd.h

@@ -0,0 +1,38 @@
+#ifndef ISAKMP_DPD_H_
+#define ISAPMP_DPD_H_
+
+typedef struct isakmp_dpd_t isakmp_dpd_t;
+
+#include <library.h>
+#include <sa/ike_sa.h>
+#include <sa/task.h>
+
+/**
+ * Task of type isakmp_dpd, detects dead peers.
+ *
+ *
+ */
+struct isakmp_dpd_t {
+
+	/**
+	 * Implements the task_t interface
+	 */
+	task_t task;
+
+	/**
+	 * Get the received dpd seqnr.
+	 *
+	 * @return				protocol ID
+	 */
+	u_int32_t (*get_dpd_seqnr) (isakmp_dpd_t *dpd_task);
+};
+
+/**
+ * Create a new isakmp_dpd task.
+ *
+ * @param initiator		TRUE if task is the original initiator
+ * @return				isakmp_dpd task to handle by the task_manager
+ */
+isakmp_dpd_t *isakmp_dpd_create(ike_sa_t *ike_sa, notify_payload_t *notify, u_int32_t seqnr);
+
+#endif /** ISAKMP_DPD_H_ @}*/

src/libcharon/sa/ikev1/tasks/main_mode.c

@@ -176,7 +176,7 @@ static status_t send_notify(private_main_mode_t *this, notify_type_t type)
 	notify->set_spi_data(notify, spi);
 
 	this->ike_sa->queue_task(this->ike_sa,
-						(task_t*)informational_create(this->ike_sa, notify));
+						(task_t*)informational_create(this->ike_sa, notify, 0));
 	/* cancel all active/passive tasks in favour of informational */
 	return ALREADY_DONE;
 }

src/libcharon/sa/ikev1/tasks/quick_mode.c

@@ -539,7 +539,7 @@ static status_t send_notify(private_quick_mode_t *this, notify_type_t type)
 	notify->set_spi(notify, this->spi_i);
 
 	this->ike_sa->queue_task(this->ike_sa,
-						(task_t*)informational_create(this->ike_sa, notify));
+						(task_t*)informational_create(this->ike_sa, notify, 0));
 	/* cancel all active/passive tasks in favour of informational */
 	return ALREADY_DONE;
 }