- support for key exchange payload

- tested

5 files changed, 59 insertions, 7 deletions

diff --git a/Source/charon/parser.c b/Source/charon/parser.c
index e65f88ff6..fc034c5b6 100644
--- a/Source/charon/parser.c
+++ b/Source/charon/parser.c
@@ -740,6 +740,16 @@ static status_t parse_payload(private_parser_t *this, payload_type_t payload_typ
 				}		
 				break;			
 			}
+			case KEY_EXCHANGE_DATA:
+			{
+				size_t keydata_length = payload_length - 8;
+				if (this->parse_chunk(this, rule_number, output + rule->offset, keydata_length) != SUCCESS) 
+				{
+					pld->destroy(pld);
+					return PARSE_ERROR;
+				}		
+				break;			
+			}
 			default:
 			{
 				this->logger->log(this->logger, ERROR, "  no rule to parse rule %d %s (%d)", rule_number, mapping_find(encoding_type_t_mappings, rule->type), rule->type);
diff --git a/Source/charon/payloads/payload.c b/Source/charon/payloads/payload.c
index 22bb85bdc..f04413849 100644
--- a/Source/charon/payloads/payload.c
+++ b/Source/charon/payloads/payload.c
@@ -27,6 +27,7 @@
 #include "ike_header.h"
 #include "sa_payload.h"
 #include "nonce_payload.h"
+#include "ke_payload.h"
 
 
 
@@ -79,6 +80,8 @@ payload_t *payload_create(payload_type_t type)
 			return (payload_t*)transform_attribute_create();
 		case NONCE:
 			return (payload_t*)nonce_payload_create();
+		case KEY_EXCHANGE:
+			return (payload_t*)ke_payload_create();
 		default:
 			return NULL;
 	}
diff --git a/Source/charon/testcases/parser_test.c b/Source/charon/testcases/parser_test.c
index 354073912..27b552338 100644
--- a/Source/charon/testcases/parser_test.c
+++ b/Source/charon/testcases/parser_test.c
@@ -31,6 +31,7 @@
 #include "../payloads/ike_header.h"
 #include "../payloads/sa_payload.h"
 #include "../payloads/nonce_payload.h"
+#include "../payloads/ke_payload.h"
 
 
 extern logger_manager_t *global_logger_manager;
@@ -192,7 +193,6 @@ void test_parser_with_nonce_payload(tester_t *tester)
 	status_t status;
 	chunk_t nonce_chunk, result;
 	
-	
 	u_int8_t nonce_bytes[] = {
 		0x00,0x00,0x00,0x14, /* payload header */
 			0x00,0x01,0x02,0x03,  /* 16 Byte nonce */
@@ -204,7 +204,6 @@ void test_parser_with_nonce_payload(tester_t *tester)
 	nonce_chunk.ptr = nonce_bytes;
 	nonce_chunk.len = sizeof(nonce_bytes);
 
-	
 	parser = parser_create(nonce_chunk);
 	tester->assert_true(tester,(parser != NULL), "parser create check");
 	status = parser->parse_payload(parser, NONCE, (payload_t**)&nonce_payload);
@@ -215,14 +214,47 @@ void test_parser_with_nonce_payload(tester_t *tester)
 	{
 		return;	
 	}
-
 	nonce_payload->get_nonce(nonce_payload, &result);
-	
 	tester->assert_true(tester,(result.len == 16), "parsed nonce lenght");
 	tester->assert_false(tester,(memcmp(nonce_bytes + 4, result.ptr, result.len)), "parsed nonce data");
+	nonce_payload->destroy(nonce_payload);
+}
+
+/*
+ * Described in Header 
+ */
+void test_parser_with_ke_payload(tester_t *tester)
+{
+	parser_t *parser;
+	ke_payload_t *ke_payload;
+	status_t status;
+	chunk_t ke_chunk, result;
 	
+	u_int8_t ke_bytes[] = {
+		0x00,0x00,0x00,0x18, /* payload header */
+		0x00,0x03,0x00,0x00, /* dh group 3 */ 
+			0x01,0x02,0x03,0x03, /* 16 Byte dh data */
+			0x04,0x05,0x06,0x07,
+			0x08,0x09,0x0A,0x2B,
+			0x0C,0x0D,0x0E,0x0F
+	};
 	
-	
+	ke_chunk.ptr = ke_bytes;
+	ke_chunk.len = sizeof(ke_bytes);
 
-	nonce_payload->destroy(nonce_payload);
+	parser = parser_create(ke_chunk); 
+	tester->assert_true(tester,(parser != NULL), "parser create check");
+	status = parser->parse_payload(parser, KEY_EXCHANGE, (payload_t**)&ke_payload);
+	tester->assert_true(tester,(status == SUCCESS),"parse_payload call check");
+	tester->assert_true(tester,(parser->destroy(parser) == SUCCESS), "parser destroy call check");
+	
+	if (status != SUCCESS)
+	{
+		return;	
+	}
+	tester->assert_true(tester,(ke_payload->get_dh_group_number(ke_payload) == 3), "DH group");
+	result = ke_payload->get_key_exchange_data(ke_payload);
+	tester->assert_true(tester,(result.len == 16), "parsed key lenght");
+	tester->assert_false(tester,(memcmp(ke_bytes + 8, result.ptr, result.len)), "parsed key data");
+	ke_payload->destroy(ke_payload);
 }
diff --git a/Source/charon/testcases/parser_test.h b/Source/charon/testcases/parser_test.h
index a59896b1b..6a36f6659 100644
--- a/Source/charon/testcases/parser_test.h
+++ b/Source/charon/testcases/parser_test.h
@@ -31,4 +31,6 @@ void test_parser_with_sa_payload(tester_t *tester);
 
 void test_parser_with_nonce_payload(tester_t *tester);
 
+void test_parser_with_ke_payload(tester_t *tester);
+
 #endif /*PARSER_TEST_H_*/
diff --git a/Source/charon/testcases/testcases.c b/Source/charon/testcases/testcases.c
index 106ecee8d..b99c3e640 100644
--- a/Source/charon/testcases/testcases.c
+++ b/Source/charon/testcases/testcases.c
@@ -145,6 +145,11 @@ test_t parser_test2 = {test_parser_with_sa_payload, "Parser: sa payload"};
  */
 test_t parser_test3 = {test_parser_with_nonce_payload, "Parser: nonce payload"};
 
+/**
+ * Parser test for ike nonce payload
+ */
+test_t parser_test4 = {test_parser_with_ke_payload, "Parser: key exchange payload"};
+
 
 /**
  * Test for packet_t
@@ -223,7 +228,7 @@ logger_manager_t *global_logger_manager;
 
 
 //	tester->perform_tests(tester,all_tests);
-	tester->perform_test(tester,&parser_test3);   
+	tester->perform_test(tester,&parser_test4);   
 
  	
 	tester->destroy(tester);