current state

Source/charon/states/responder_init.c

@@ -22,7 +22,12 @@
  
 #include "responder_init.h"
 
+#include "../globals.h"
 #include "../utils/allocator.h"
+#include "../payloads/sa_payload.h"
+#include "../payloads/ke_payload.h"
+#include "../payloads/nonce_payload.h"
+#include "../transforms/diffie_hellman.h"
 
 /**
  * Private data of a responder_init_t object.
@@ -35,6 +40,51 @@ struct private_responder_init_s {
 	 */
 	responder_init_t public;
 	
+	/**
+	 * Assigned IKE_SA
+	 */
+	protected_ike_sa_t *ike_sa;
+	
+	/**
+	 * Diffie Hellman object used to compute shared secret
+	 */
+	diffie_hellman_t *diffie_hellman;
+		
+	/**
+	 * Diffie Hellman group number
+	 */
+	u_int16_t dh_group_number;	
+	
+	/**
+	 * Priority used get matching dh_group number
+	 */
+	u_int16_t dh_group_priority;
+
+	/**
+	 * Sent nonce value
+	 */
+	chunk_t sent_nonce;
+	
+	/**
+	 * Received nonce value
+	 */
+	chunk_t received_nonce;
+	
+	/**
+	 * Logger used to log data 
+	 * 
+	 * Is logger of ike_sa!
+	 */
+	logger_t *logger;
+	
+	/**
+	 * Proposals used to initiate connection
+	 */
+	linked_list_t *proposals;
+	
+	status_t (*build_sa_payload) (private_responder_init_t *this, payload_t **payload);
+	status_t (*build_ke_payload) (private_responder_init_t *this, payload_t **payload);
+	status_t (*build_nonce_payload) (private_responder_init_t *this, payload_t **payload);	
 };
 
 /**
@@ -42,16 +92,412 @@ struct private_responder_init_s {
  */
 static status_t process_message(private_responder_init_t *this, message_t *message, state_t **new_state)
 {
-	*new_state = (state_t *) this;
+	linked_list_iterator_t *payloads;
+	host_t *source, *destination;
+	status_t status;
+	message_t *response;
+	payload_t *payload;
+	packet_t *packet;
+	
+	/* this is the first message we process, so copy host infos */
+	message->get_source(message, &source);
+	message->get_destination(message, &destination);
+	
+	/* we need to clone them, since we destroy the message later */
+	destination->clone(destination, &(this->ike_sa->me.host));
+	source->clone(source, &(this->ike_sa->other.host));
+	
+	/* parse incoming message */
+	status = message->parse_body(message);
+	if (status != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR | MORE, "Could not parse body of request message");
+		return status;	
+	}
+
+	/* iterate over incoming payloads. We can be sure, the message contains only accepted payloads! */
+	status = message->get_payload_iterator(message, &payloads);
+	if (status != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR, "Fatal error: Could not get payload interator");
+		return status;
+	}
+	
+	while (payloads->has_next(payloads))
+	{
+		payload_t *payload;
+		
+		/* get current payload */
+		payloads->current(payloads, (void**)&payload);
+		
+		this->logger->log(this->logger, CONTROL|MORE, "Processing payload of type %s", mapping_find(payload_type_m, payload->get_type(payload)));
+		switch (payload->get_type(payload))
+		{
+			case SECURITY_ASSOCIATION:
+			{
+				sa_payload_t *sa_payload = (sa_payload_t*)payload;
+				linked_list_iterator_t *suggested_proposals, *accepted_proposals;
+
+				status = this->proposals->create_iterator(this->proposals, &accepted_proposals, FALSE);
+				if (status != SUCCESS)
+				{
+					this->logger->log(this->logger, ERROR, "Fatal error: Could not create iterator on list for proposals");
+					payloads->destroy(payloads);
+					return status;	
+				}
+				
+				/* get the list of suggested proposals */ 
+				status = sa_payload->create_proposal_substructure_iterator(sa_payload, &suggested_proposals, TRUE);
+				if (status != SUCCESS)
+				{	
+					this->logger->log(this->logger, ERROR, "Fatal error: Could not create iterator on suggested proposals");
+					accepted_proposals->destroy(accepted_proposals);
+					payloads->destroy(payloads);
+					return status;
+				}
+				
+				/* now let the configuration-manager select a subset of the proposals */
+				status = global_configuration_manager->select_proposals_for_host(global_configuration_manager,
+									this->ike_sa->other.host, suggested_proposals, accepted_proposals);
+				if (status != SUCCESS)
+				{
+					this->logger->log(this->logger, CONTROL | MORE, "No proposal of suggested proposals selected");
+					suggested_proposals->destroy(suggested_proposals);
+					accepted_proposals->destroy(accepted_proposals);
+					payloads->destroy(payloads);
+					return status;
+				}
+				
+				/* iterators are not needed anymore */			
+				suggested_proposals->destroy(suggested_proposals);
+				accepted_proposals->destroy(accepted_proposals);
+				
+				this->logger->log(this->logger, CONTROL | MORE, "SA Payload processed");
+				/* ok, we have what we need for sa_payload (proposals are stored in this->proposals)*/
+				break;
+			}
+			case KEY_EXCHANGE:
+			{
+				ke_payload_t *ke_payload = (ke_payload_t*)payload;
+				diffie_hellman_group_t group;
+				diffie_hellman_t *dh;
+				bool allowed_group;
+				
+				group = ke_payload->get_dh_group_number(ke_payload);
+				
+				status = global_configuration_manager->is_dh_group_allowed_for_host(global_configuration_manager,
+								this->ike_sa->other.host, group, &allowed_group);
+
+				if (status != SUCCESS)
+				{
+					this->logger->log(this->logger, ERROR | MORE, "Could not get informations about DH group");
+					payloads->destroy(payloads);
+					return status;
+				}
+				if (!allowed_group)
+				{
+					/** @todo Send info reply */	
+				}
+				
+				/* create diffie hellman object to handle DH exchange */
+				dh = diffie_hellman_create(group);
+				if (dh == NULL)
+				{
+					this->logger->log(this->logger, ERROR, "Could not generate DH object");
+					payloads->destroy(payloads);
+					return OUT_OF_RES;
+				}
+
+				this->logger->log(this->logger, CONTROL | MORE, "Set other DH public value");
+				
+				status = dh->set_other_public_value(dh, ke_payload->get_key_exchange_data(ke_payload));
+				if (status != SUCCESS)
+				{
+					this->logger->log(this->logger, ERROR, "Could not set other DH public value");
+					dh->destroy(dh);
+					payloads->destroy(payloads);
+					return OUT_OF_RES;
+				}
+
+				this->diffie_hellman = dh;
+				
+				this->logger->log(this->logger, CONTROL | MORE, "KE Payload processed");
+				break;
+			}
+			case NONCE:
+			{
+				nonce_payload_t *nonce_payload = (nonce_payload_t*)payload;
+				chunk_t nonce;
+
+				this->logger->log(this->logger, CONTROL | MORE, "Get nonce value and store it");
+				nonce_payload->get_nonce(nonce_payload, &nonce);
+				/** @todo free if there is already one */
+				this->received_nonce.ptr = allocator_clone_bytes(nonce.ptr, nonce.len);
+				this->received_nonce.len = nonce.len;
+				if (this->received_nonce.ptr == NULL)
+				{
+					payloads->destroy(payloads);
+					return OUT_OF_RES;
+				}
+				
+				this->logger->log(this->logger, CONTROL | MORE, "Nonce Payload processed");
+				break;
+			}
+			default:
+			{
+				this->logger->log(this->logger, ERROR | MORE, "Payload type not supported!");
+				payloads->destroy(payloads);
+				return OUT_OF_RES;
+			}
+				
+		}
+			
+	}
+	/* iterator can be destroyed */
+	payloads->destroy(payloads);
+	
+	this->logger->log(this->logger, CONTROL | MORE, "Request successfully handled. Going to create reply.");
+
+	this->logger->log(this->logger, CONTROL | MOST, "Going to create nonce.");		
+	if (this->ike_sa->randomizer->allocate_pseudo_random_bytes(this->ike_sa->randomizer, NONCE_SIZE, &(this->sent_nonce)) != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR, "Could not create nonce!");
+		return OUT_OF_RES;
+	}
+		
+
+	/* set up the reply */
+	status = this->ike_sa->build_message(this->ike_sa, IKE_SA_INIT, FALSE, &response);
+	if (status != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR, "Could not create empty message");
+		return status;	
+	}
+	
+	/* build SA payload */		
+	status = this->build_sa_payload(this, &payload);
+	if (status != SUCCESS)
+	{	
+		this->logger->log(this->logger, ERROR, "Could not build SA payload");
+		return status;
+	}
+	
+	this	->logger->log(this->logger, CONTROL|MOST, "add SA payload to message");
+	status = response->add_payload(response, payload);
+	if (status != SUCCESS)
+	{	
+		this->logger->log(this->logger, ERROR, "Could not add SA payload to message");
+		return status;
+	}
+	
+	/* build KE payload */
+	status = this->build_ke_payload(this,&payload);
+	if (status != SUCCESS)
+	{	
+		this->logger->log(this->logger, ERROR, "Could not build KE payload");
+		return status;
+	}
+
+	this	->logger->log(this->logger, CONTROL|MOST, "add KE payload to message");
+	status = response->add_payload(response, payload);
+	if (status != SUCCESS)
+	{	
+		this->logger->log(this->logger, ERROR, "Could not add KE payload to message");
+		return status;
+	}
+	
+	/* build Nonce payload */
+	status = this->build_nonce_payload(this, &payload);
+	if (status != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR, "Could not build NONCE payload");
+		return status;
+	}
+
+	this	->logger->log(this->logger, CONTROL|MOST, "add nonce payload to message");
+	status = response->add_payload(response, payload);
+	if (status != SUCCESS)
+	{	
+		this->logger->log(this->logger, ERROR, "Could not add nonce payload to message");
+		return status;
+	}
+	
+	/* generate packet */	
+	this	->logger->log(this->logger, CONTROL|MOST, "generate packet from message");
+	status = response->generate(response, &packet);
+	if (status != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR, "Fatal error: could not generate packet from message");
+		return status;
+	}
+	
+	this	->logger->log(this->logger, CONTROL|MOST, "Add packet to global send queue");
+	status = global_send_queue->add(global_send_queue, packet);
+	if (status != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR, "Could not add packet to send queue");
+		return status;
+	}
+
+	if (	this->ike_sa->last_responded_message != NULL)
+	{
+		/* destroy message */
+		this	->logger->log(this->logger, CONTROL|MOST, "Destroy stored last responded message");
+		this->ike_sa->last_responded_message->destroy(this->ike_sa->last_responded_message);
+	}
+
+	this->ike_sa->last_responded_message	 = response;
+
+	/* state has NOW changed :-) */
+//	this	->logger->log(this->logger, CONTROL|MORE, "Change state of IKE_SA from %s to %s",mapping_find(ike_sa_state_m,this->state),mapping_find(ike_sa_state_m,IKE_SA_INIT_REQUESTED) );
+
+	
 	return SUCCESS;
 }
 
+/**
+ * implements private_initiator_init_t.build_sa_payload
+ */
+static status_t build_sa_payload(private_responder_init_t *this, payload_t **payload)
+{
+	sa_payload_t* sa_payload;
+	linked_list_iterator_t *proposal_iterator;
+	status_t status;
+	
+	
+	/* SA payload takes proposals from this->ike_sa_init_data.proposals and writes them to the created sa_payload */
+
+	this->logger->log(this->logger, CONTROL|MORE, "building sa payload");
+	
+	status = this->proposals->create_iterator(this->proposals, &proposal_iterator, FALSE);
+	if (status != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR, "Fatal error: Could not create iterator on list for proposals");
+		return status;	
+	}
+	
+	sa_payload = sa_payload_create();
+	if (sa_payload == NULL)
+	{
+		this->logger->log(this->logger, ERROR, "Fatal error: Could not create SA payload object");
+		return OUT_OF_RES;
+	}
+	
+	while (proposal_iterator->has_next(proposal_iterator))
+	{
+		proposal_substructure_t *current_proposal;
+		proposal_substructure_t *current_proposal_clone;
+		status = proposal_iterator->current(proposal_iterator,(void **) &current_proposal);
+		if (status != SUCCESS)
+		{
+			this->logger->log(this->logger, ERROR, "Could not get current proposal needed to copy");
+			sa_payload->destroy(sa_payload);
+			return status;	
+		}
+		status = current_proposal->clone(current_proposal,&current_proposal_clone);
+		if (status != SUCCESS)
+		{
+			this->logger->log(this->logger, ERROR, "Could not clone current proposal");
+			sa_payload->destroy(sa_payload);
+			return status;	
+		}
+		
+		status = sa_payload->add_proposal_substructure(sa_payload,current_proposal_clone);
+		if (status != SUCCESS)
+		{
+			this->logger->log(this->logger, ERROR, "Could not add cloned proposal to SA payload");
+			sa_payload->destroy(sa_payload);
+			return status;	
+		}
+
+	}
+	
+	this->logger->log(this->logger, CONTROL|MORE, "sa payload builded");
+	
+	*payload = (payload_t *) sa_payload;
+	
+	return SUCCESS;
+}
+
+/**
+ * implements private_initiator_init_t.build_ke_payload
+ */
+static status_t build_ke_payload(private_responder_init_t *this, payload_t **payload)
+{
+	ke_payload_t *ke_payload;
+	chunk_t key_data;
+	status_t status;
+
+	this->logger->log(this->logger, CONTROL|MORE, "building ke payload");
+	
+
+	this	->logger->log(this->logger, CONTROL|MORE, "get public dh value to send in ke payload");
+	status = this->diffie_hellman->get_my_public_value(this->diffie_hellman,&key_data);
+	if (status != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR, "Could not get my DH public value");
+		return status;
+	}
+
+	ke_payload = ke_payload_create();
+	if (ke_payload == NULL)
+	{
+		this->logger->log(this->logger, ERROR, "Could not create KE payload");
+		allocator_free_chunk(key_data);
+		return OUT_OF_RES;	
+	}
+	ke_payload->set_dh_group_number(ke_payload, MODP_1024_BIT);
+	if (ke_payload->set_key_exchange_data(ke_payload, key_data) != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR, "Could not set key exchange data of KE payload");
+		ke_payload->destroy(ke_payload);
+		allocator_free_chunk(key_data);
+		return OUT_OF_RES;
+	}
+	allocator_free_chunk(key_data);
+
+	*payload = (payload_t *) ke_payload;
+	return SUCCESS;			
+}
+
+/**
+ * implements private_initiator_init_t.build_nonce_payload
+ */
+static status_t build_nonce_payload(private_responder_init_t *this, payload_t **payload)
+{
+	nonce_payload_t *nonce_payload;
+	status_t status;
+	
+	this->logger->log(this->logger, CONTROL|MORE, "building nonce payload");
+
+	nonce_payload = nonce_payload_create();
+	if (nonce_payload == NULL)
+	{	
+		this->logger->log(this->logger, ERROR, "Fatal error: could not create nonce payload object");
+		return OUT_OF_RES;	
+	}
+
+	status = nonce_payload->set_nonce(nonce_payload, this->sent_nonce);
+	
+	if (status != SUCCESS)
+	{
+		this->logger->log(this->logger, ERROR, "Fatal error: could not set nonce data of payload");
+		nonce_payload->destroy(nonce_payload);
+		return status;
+	}
+		
+	*payload = (payload_t *) nonce_payload;
+	
+	return SUCCESS;
+}
+
+
 /**
  * Implements state_t.get_state
  */
 static ike_sa_state_t get_state(private_responder_init_t *this)
 {
-	return INITIATOR_INIT;
+	return RESPONDER_INIT;
 }
 
 /**
@@ -59,6 +505,15 @@ static ike_sa_state_t get_state(private_responder_init_t *this)
  */
 static status_t destroy(private_responder_init_t *this)
 {
+	/* destroy stored proposal */
+	this->logger->log(this->logger, CONTROL | MOST, "Destroy stored proposals");
+	while (this->proposals->get_count(this->proposals) > 0)
+	{
+		proposal_substructure_t *current_proposal;
+		this->proposals->remove_first(this->proposals,(void **)&current_proposal);
+		current_proposal->destroy(current_proposal);
+	}
+	this->proposals->destroy(this->proposals);
 	allocator_free(this);
 	return SUCCESS;
 }
@@ -66,7 +521,7 @@ static status_t destroy(private_responder_init_t *this)
 /* 
  * Described in header.
  */
-responder_init_t *responder_init_create()
+responder_init_t *responder_init_create(protected_ike_sa_t *ike_sa)
 {
 	private_responder_init_t *this = allocator_alloc_thing(private_responder_init_t);
 	
@@ -80,5 +535,24 @@ responder_init_t *responder_init_create()
 	this->public.state_interface.get_state = (ike_sa_state_t (*) (state_t *)) get_state;
 	this->public.state_interface.destroy  = (status_t (*) (state_t *)) destroy;
 	
+	/* private functions */
+	this->build_sa_payload = build_sa_payload;
+	this->build_ke_payload = build_ke_payload;
+	this->build_nonce_payload = build_nonce_payload;
+	
+	/* private data */
+	this->ike_sa = ike_sa;
+	this->logger = this->ike_sa->logger;
+	this->sent_nonce.ptr = NULL;
+	this->sent_nonce.len = 0;
+	this->received_nonce.ptr = NULL;
+	this->received_nonce.len = 0;
+	this->proposals = linked_list_create();
+	if (this->proposals == NULL)
+	{
+		allocator_free(this);
+		return NULL;
+	}
+	
 	return &(this->public);
 }