diff options
Diffstat (limited to 'src/charon/testing/generator_test.c')
| -rw-r--r-- | src/charon/testing/generator_test.c | 1291 |
1 files changed, 0 insertions, 1291 deletions
diff --git a/src/charon/testing/generator_test.c b/src/charon/testing/generator_test.c deleted file mode 100644 index 2ba772ce0..000000000 --- a/src/charon/testing/generator_test.c +++ /dev/null @@ -1,1291 +0,0 @@ -/** - * @file generator_test.c - * - * @brief Tests for the generator_t class. - * - */ - -/* - * Copyright (C) 2005-2006 Martin Willi - * Copyright (C) 2005 Jan Hutter - * Hochschule fuer Technik Rapperswil - * - * This program is free software; you can redistribute it and/or modify it - * under the terms of the GNU General Public License as published by the - * Free Software Foundation; either version 2 of the License, or (at your - * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>. - * - * This program is distributed in the hope that it will be useful, but - * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY - * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License - * for more details. - */ - -#include <string.h> - -#include "generator_test.h" - -#include <daemon.h> -#include <encoding/generator.h> -#include <utils/logger_manager.h> -#include <utils/logger.h> -#include <encoding/payloads/encodings.h> -#include <encoding/payloads/ike_header.h> -#include <encoding/payloads/transform_attribute.h> -#include <encoding/payloads/transform_substructure.h> -#include <encoding/payloads/proposal_substructure.h> -#include <encoding/payloads/sa_payload.h> -#include <encoding/payloads/ke_payload.h> -#include <encoding/payloads/notify_payload.h> -#include <encoding/payloads/nonce_payload.h> -#include <encoding/payloads/id_payload.h> -#include <encoding/payloads/auth_payload.h> -#include <encoding/payloads/cert_payload.h> -#include <encoding/payloads/certreq_payload.h> -#include <encoding/payloads/ts_payload.h> -#include <encoding/payloads/delete_payload.h> -#include <encoding/payloads/vendor_id_payload.h> -#include <encoding/payloads/cp_payload.h> -#include <encoding/payloads/eap_payload.h> - -/* - * Described in Header - */ -void test_generator_with_header_payload(protected_tester_t *tester) -{ - generator_t *generator; - ike_header_t *header_data; - chunk_t generated_data; - logger_t *logger; - - logger = logger_manager->get_logger(logger_manager, TESTER); - - header_data = ike_header_create(); - header_data->set_initiator_spi(header_data,1); - header_data->set_responder_spi(header_data,2); - ((payload_t *) header_data)->set_next_type((payload_t *) header_data, 3); - header_data->set_exchange_type(header_data, 6); - header_data->set_initiator_flag(header_data, TRUE); - header_data->set_response_flag(header_data, TRUE); - header_data->set_message_id(header_data,7); - - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - generator->generate_payload(generator,(payload_t *) header_data); - - generator->write_to_chunk(generator,&generated_data); - - u_int8_t expected_generation[] = { - 0x01,0x00,0x00,0x00, - 0x00,0x00,0x00,0x00, - 0x02,0x00,0x00,0x00, - 0x00,0x00,0x00,0x00, - 0x03,0x20,0x06,0x28, - 0x00,0x00,0x00,0x07, - 0x00,0x00,0x00,0x1C, - }; - - logger->log_bytes(logger,RAW,"expected header",expected_generation,sizeof(expected_generation)); - tester->assert_true(tester,(generated_data.len == sizeof(expected_generation)), "compare generated data length"); - logger->log_chunk(logger,RAW,"generated header",generated_data); - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data 1"); - chunk_free(&generated_data); - - generator->destroy(generator); - - header_data->set_initiator_spi(header_data,0x22000054231234LL); - header_data->set_responder_spi(header_data,0x122398); - ((payload_t *) header_data)->set_next_type((payload_t *) header_data,0xF3); - header_data->set_exchange_type(header_data, 0x12); - header_data->set_initiator_flag(header_data, TRUE); - header_data->set_response_flag(header_data, TRUE); - header_data->set_message_id(header_data,0x33AFF3); - - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - generator->generate_payload(generator,(payload_t *)header_data); - - generator->write_to_chunk(generator,&generated_data); - - u_int8_t expected_generation2[] = { - 0x34,0x12,0x23,0x54, - 0x00,0x00,0x22,0x00, - 0x98,0x23,0x12,0x00, - 0x00,0x00,0x00,0x00, - 0xF3,0x20,0x12,0x28, - 0x00,0x33,0xAF,0xF3, - 0x00,0x00,0x00,0x1C, - }; - - - logger->log_bytes(logger,RAW,"expected header",expected_generation2,sizeof(expected_generation2)); - - logger->log_chunk(logger,RAW,"generated header",generated_data); - - tester->assert_true(tester,(memcmp(expected_generation2,generated_data.ptr,sizeof(expected_generation2)) == 0), "compare generated data 2"); - chunk_free(&generated_data); - - header_data->destroy(header_data); - - generator->destroy(generator); -} - -/* - * Described in header - */ -void test_generator_with_transform_attribute(protected_tester_t *tester) -{ - generator_t *generator; - transform_attribute_t *attribute; - chunk_t generated_data; - logger_t *logger; - - logger = logger_manager->get_logger(logger_manager, TESTER); - - - /* test empty attribute */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - attribute = transform_attribute_create(); - generator->generate_payload(generator,(payload_t *)attribute); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated attribute",generated_data); - - u_int8_t expected_generation[] = { - 0x80,0x00,0x00,0x00, - }; - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - chunk_free(&generated_data); - attribute->destroy(attribute); - generator->destroy(generator); - - /* test attribute with 2 byte data */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - attribute = transform_attribute_create(); - u_int16_t dataval = 5768; - chunk_t data; - data.ptr = (void *) &dataval; - data.len = 2; - - attribute->set_value_chunk(attribute,data); - - generator->generate_payload(generator,(payload_t *)attribute); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated attribute",generated_data); - - u_int8_t expected_generation2[] = { - 0x80,0x00,0x16,0x88, - }; - tester->assert_true(tester,(memcmp(expected_generation2,generated_data.ptr,sizeof(expected_generation2)) == 0), "compare generated data"); - - chunk_free(&generated_data); - attribute->destroy(attribute); - generator->destroy(generator); - - - - /* test attribute with 25 byte data */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - attribute = transform_attribute_create(); - char *stringval = "ddddddddddeeeeeeeeeefffff"; - data.ptr = (void *) stringval; - data.len = 25; - - attribute->set_value_chunk(attribute,data); - - attribute->set_attribute_type(attribute,456); - - - generator->generate_payload(generator,(payload_t *)attribute); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated attribute",generated_data); - - u_int8_t expected_generation3[] = { - 0x01,0xC8,0x00,0x19, - 0x64,0x64,0x64,0x64, - 0x64,0x64,0x64,0x64, - 0x64,0x64,0x65,0x65, - 0x65,0x65,0x65,0x65, - 0x65,0x65,0x65,0x65, - 0x66,0x66,0x66,0x66, - 0x66 - }; - tester->assert_true(tester,(memcmp(expected_generation3,generated_data.ptr,sizeof(expected_generation3)) == 0), "compare generated data"); - - chunk_free(&generated_data); - attribute->destroy(attribute); - generator->destroy(generator); -} - - - -/* - * Described in header - */ -void test_generator_with_transform_substructure(protected_tester_t *tester) -{ - generator_t *generator; - transform_attribute_t *attribute1, *attribute2; - transform_substructure_t *transform; - chunk_t data; - chunk_t generated_data; - logger_t *logger; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - /* create attribute 1 */ - attribute1 = transform_attribute_create(); - char *stringval = "abcd"; - data.ptr = (void *) stringval; - data.len = 4; - attribute1->set_value_chunk(attribute1,data); - attribute1->set_attribute_type(attribute1,0); - logger->log(logger,CONTROL,"attribute1 created"); - - /* create attribute 2 */ - attribute2 = transform_attribute_create(); - stringval = "efgh"; - data.ptr = (void *) stringval; - data.len = 4; - attribute2->set_value_chunk(attribute2,data); - attribute2->set_attribute_type(attribute2,0); - logger->log(logger,CONTROL,"attribute2 created"); - - /* create transform */ - transform = transform_substructure_create(); - tester->assert_true(tester,(transform != NULL), "transform create check"); - transform->add_transform_attribute(transform,attribute1); - transform->add_transform_attribute(transform,attribute2); - transform->set_transform_type(transform,5); /* hex 5 */ - transform->set_transform_id(transform,65000); /* hex FDE8 */ - - - logger->log(logger,CONTROL,"transform created"); - - generator->generate_payload(generator,(payload_t *)transform); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated transform",generated_data); - - u_int8_t expected_generation3[] = { - 0x00,0x00,0x00,0x18, - 0x05,0x00,0xFD,0xE8, - 0x00,0x00,0x00,0x04, - 0x61,0x62,0x63,0x64, - 0x00,0x00,0x00,0x04, - 0x65,0x66,0x67,0x68, - }; - tester->assert_true(tester,(memcmp(expected_generation3,generated_data.ptr,sizeof(expected_generation3)) == 0), "compare generated data"); - - chunk_free(&generated_data); - transform->destroy(transform); - generator->destroy(generator); -} - - -/* - * Described in header - */ -void test_generator_with_proposal_substructure(protected_tester_t *tester) -{ - generator_t *generator; - transform_attribute_t *attribute1, *attribute2, *attribute3; - transform_substructure_t *transform1, *transform2; - proposal_substructure_t *proposal; - chunk_t data; - chunk_t generated_data; - logger_t *logger; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - /* create attribute 1 */ - attribute1 = transform_attribute_create(); - char *stringval = "abcd"; - data.ptr = (void *) stringval; - data.len = 4; - attribute1->set_value_chunk(attribute1,data); - attribute1->set_attribute_type(attribute1,0); - - logger->log(logger,CONTROL,"attribute1 created"); - - /* create attribute 2 */ - attribute2 = transform_attribute_create(); - stringval = "efgh"; - data.ptr = (void *) stringval; - data.len = 4; - attribute2->set_value_chunk(attribute2,data); - attribute2->set_attribute_type(attribute2,0); - logger->log(logger,CONTROL,"attribute2 created"); - - /* create attribute 3 */ - attribute3 = transform_attribute_create(); - stringval = "ijkl"; - data.ptr = (void *) stringval; - data.len = 4; - attribute3->set_value_chunk(attribute3,data); - attribute3->set_attribute_type(attribute3,0); - logger->log(logger,CONTROL,"attribute3 created"); - - /* create transform 1*/ - transform1 = transform_substructure_create(); - tester->assert_true(tester,(transform1 != NULL), "transform create check"); - transform1->add_transform_attribute(transform1,attribute1); - transform1->add_transform_attribute(transform1,attribute2); - transform1->set_transform_type(transform1,5); /* hex 5 */ - transform1->set_transform_id(transform1,65000); /* hex FDE8 */ - - /* create transform 2*/ - transform2 = transform_substructure_create(); - tester->assert_true(tester,(transform2 != NULL), "transform create check"); - transform2->add_transform_attribute(transform2,attribute3); - transform2->set_transform_type(transform2,3); /* hex 3 */ - transform2->set_transform_id(transform2,4); /* hex 4 */ - - logger->log(logger,CONTROL,"transforms created"); - - proposal = proposal_substructure_create(); - tester->assert_true(tester,(proposal != NULL), "proposal create check"); - - stringval = "ABCDEFGH"; - data.ptr = (void *) stringval; - data.len = 8; - - proposal->add_transform_substructure(proposal,transform1); - proposal->add_transform_substructure(proposal,transform2); - proposal->set_spi(proposal,data); - proposal->set_proposal_number(proposal,7); - proposal->set_protocol_id(proposal,4); - - generator->generate_payload(generator,(payload_t *)proposal); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated transform",generated_data); - - u_int8_t expected_generation[] = { - /* proposal header */ - 0x00,0x00,0x00,0x38, - 0x07,0x04,0x08,0x02, - /* SPI */ - 0x41,0x42,0x43,0x44, - 0x45,0x46,0x47,0x48, - /* first transform */ - 0x03,0x00,0x00,0x18, - 0x05,0x00,0xFD,0xE8, - /* first transform attributes */ - 0x00,0x00,0x00,0x04, - 0x61,0x62,0x63,0x64, - 0x00,0x00,0x00,0x04, - 0x65,0x66,0x67,0x68, - /* second transform */ - 0x00,0x00,0x00,0x10, - 0x03,0x00,0x00,0x04, - /* second transform attributes */ - 0x00,0x00,0x00,0x04, - 0x69,0x6A,0x6B,0x6C - }; - logger->log_bytes(logger,RAW,"expected transform",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - proposal->destroy(proposal); - generator->destroy(generator); -} - -/* - * Described in header - */ -void test_generator_with_sa_payload(protected_tester_t *tester) -{ - generator_t *generator; - transform_attribute_t *attribute1, *attribute2, *attribute3; - transform_substructure_t *transform1, *transform2; - proposal_substructure_t *proposal_str1, *proposal_str2; - linked_list_t *list; - proposal_t *proposal1, *proposal2; - sa_payload_t *sa_payload; - ike_header_t *ike_header; - - chunk_t data; - chunk_t generated_data; - logger_t *logger; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - /* --------------------------- */ - /* test first with self created proposals */ - - /* create attribute 1 */ - attribute1 = transform_attribute_create(); - char *stringval = "abcd"; - data.ptr = (void *) stringval; - data.len = 4; - attribute1->set_value_chunk(attribute1,data); - attribute1->set_attribute_type(attribute1,0); - logger->log(logger,CONTROL,"attribute1 created"); - - /* create attribute 2 */ - attribute2 = transform_attribute_create(); - stringval = "efgh"; - data.ptr = (void *) stringval; - data.len = 4; - attribute2->set_value_chunk(attribute2,data); - attribute2->set_attribute_type(attribute2,0); - logger->log(logger,CONTROL,"attribute2 created"); - - /* create attribute 3 */ - attribute3 = transform_attribute_create(); - stringval = "ijkl"; - data.ptr = (void *) stringval; - data.len = 4; - attribute3->set_value_chunk(attribute3,data); - attribute3->set_attribute_type(attribute3,0); - logger->log(logger,CONTROL,"attribute3 created"); - - /* create transform 1*/ - transform1 = transform_substructure_create(); - tester->assert_true(tester,(transform1 != NULL), "transform create check"); - transform1->add_transform_attribute(transform1,attribute1); - transform1->add_transform_attribute(transform1,attribute2); - transform1->set_transform_type(transform1,5); /* hex 5 */ - transform1->set_transform_id(transform1,65000); /* hex FDE8 */ - - /* create transform 2*/ - transform2 = transform_substructure_create(); - tester->assert_true(tester,(transform2 != NULL), "transform create check"); - transform2->add_transform_attribute(transform2,attribute3); - transform2->set_transform_type(transform2,3); /* hex 3 */ - transform2->set_transform_id(transform2,4); /* hex 4 */ - - logger->log(logger,CONTROL,"transforms created"); - - /* create proposal 1 */ - proposal_str1 = proposal_substructure_create(); - tester->assert_true(tester,(proposal1 != NULL), "proposal create check"); - - stringval = "ABCDEFGH"; - data.ptr = (void *) stringval; - data.len = 8; - - proposal_str1->add_transform_substructure(proposal_str1,transform1); - proposal_str1->add_transform_substructure(proposal_str1,transform2); - proposal_str1->set_spi(proposal_str1,data); - proposal_str1->set_proposal_number(proposal_str1,7); - proposal_str1->set_protocol_id(proposal_str1,4); - - /* create proposal 2 */ - proposal_str2 = proposal_substructure_create(); - tester->assert_true(tester,(proposal_str2 != NULL), "proposal create check"); - proposal_str2->set_proposal_number(proposal_str2,7); - proposal_str2->set_protocol_id(proposal_str2,5); - - /* create sa_payload */ - sa_payload = sa_payload_create(); - - sa_payload->add_proposal_substructure(sa_payload,proposal_str1); - sa_payload->add_proposal_substructure(sa_payload,proposal_str2); - - ike_header = ike_header_create(); - ike_header->set_initiator_spi(ike_header,0x22000054231234LL); - ike_header->set_responder_spi(ike_header,0x122398); - ((payload_t *) ike_header)->set_next_type((payload_t *) ike_header,SECURITY_ASSOCIATION); - ike_header->set_exchange_type(ike_header, 0x12); - ike_header->set_initiator_flag(ike_header, TRUE); - ike_header->set_response_flag(ike_header, TRUE); - ike_header->set_message_id(ike_header,0x33AFF3); - - generator->generate_payload(generator,(payload_t *)ike_header); - generator->generate_payload(generator,(payload_t *)sa_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated transform",generated_data); - - u_int8_t expected_generation[] = { - /* sa payload header */ - 0x34,0x12,0x23,0x54, - 0x00,0x00,0x22,0x00, - 0x98,0x23,0x12,0x00, - 0x00,0x00,0x00,0x00, - 0x21,0x20,0x12,0x28, - 0x00,0x33,0xAF,0xF3, - 0x00,0x00,0x00,0x60, - - /* sa payload header */ - 0x00,0x00,0x00,0x44, - /* proposal header */ - 0x02,0x00,0x00,0x38, - 0x07,0x04,0x08,0x02, - /* SPI */ - 0x41,0x42,0x43,0x44, - 0x45,0x46,0x47,0x48, - /* first transform */ - 0x03,0x00,0x00,0x18, - 0x05,0x00,0xFD,0xE8, - /* first transform attributes */ - 0x00,0x00,0x00,0x04, - 0x61,0x62,0x63,0x64, - 0x00,0x00,0x00,0x04, - 0x65,0x66,0x67,0x68, - /* second transform */ - 0x00,0x00,0x00,0x10, - 0x03,0x00,0x00,0x04, - /* second transform attributes */ - 0x00,0x00,0x00,0x04, - 0x69,0x6A,0x6B,0x6C, - /* proposal header 2*/ - 0x00,0x00,0x00,0x08, - 0x07,0x05,0x00,0x00, - - }; - - logger->log_bytes(logger,RAW,"expected transform",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - ike_header->destroy(ike_header); - sa_payload->destroy(sa_payload); - generator->destroy(generator); - - /* --------------------------- */ - /* test with automatic created proposals */ - - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - - proposal1 = proposal_create(PROTO_IKE); - proposal1->add_algorithm(proposal1, ENCRYPTION_ALGORITHM, 1, 20); - proposal1->add_algorithm(proposal1, PSEUDO_RANDOM_FUNCTION, 2, 22); - proposal1->add_algorithm(proposal1, INTEGRITY_ALGORITHM, 3, 24); - proposal1->add_algorithm(proposal1, DIFFIE_HELLMAN_GROUP, 4, 0); - - proposal2 = proposal_create(PROTO_IKE); - proposal2->add_algorithm(proposal2, ENCRYPTION_ALGORITHM, 5, 26); - proposal2->add_algorithm(proposal2, PSEUDO_RANDOM_FUNCTION, 6, 28); - proposal2->add_algorithm(proposal2, INTEGRITY_ALGORITHM, 7, 30); - proposal2->add_algorithm(proposal2, DIFFIE_HELLMAN_GROUP, 8, 0); - - list = linked_list_create(); - list->insert_last(list, (void*)proposal1); - list->insert_last(list, (void*)proposal2); - sa_payload = sa_payload_create_from_proposal_list(list); - tester->assert_true(tester,(sa_payload != NULL), "sa_payload create check"); - - generator->generate_payload(generator,(payload_t *)sa_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated",generated_data); - - u_int8_t expected_generation2[] = { - 0x00,0x00,0x00,0x6C, /* payload header*/ - 0x02,0x00,0x00,0x34, /* a proposal */ - 0x01,0x01,0x00,0x04, - 0x03,0x00,0x00,0x0C, /* transform 1 */ - 0x01,0x00,0x00,0x01, - 0x80,0x0E,0x00,0x14, /* keylength attribute with 20 bytes length */ - 0x03,0x00,0x00,0x0C, /* transform 2 */ - 0x02,0x00,0x00,0x02, - 0x80,0x0E,0x00,0x16, /* keylength attribute with 20 bytes length */ - 0x03,0x00,0x00,0x0C, /* transform 3 */ - 0x03,0x00,0x00,0x03, - 0x80,0x0E,0x00,0x18, /* keylength attribute with 20 bytes length */ - 0x00,0x00,0x00,0x08, /* transform 4 */ - 0x04,0x00,0x00,0x04, - 0x00,0x00,0x00,0x34, /* a proposal */ - 0x02,0x01,0x00,0x04, - 0x03,0x00,0x00,0x0C, /* transform 1 */ - 0x01,0x00,0x00,0x05, - 0x80,0x0E,0x00,0x1A, /* keylength attribute with 16 bytes length */ - 0x03,0x00,0x00,0x0C, /* transform 2 */ - 0x02,0x00,0x00,0x06, - 0x80,0x0E,0x00,0x1C, /* keylength attribute with 16 bytes length */ - 0x03,0x00,0x00,0x0C, /* transform 3 */ - 0x03,0x00,0x00,0x07, - 0x80,0x0E,0x00,0x1E, /* keylength attribute with 16 bytes length */ - 0x00,0x00,0x00,0x08, /* transform 4 */ - 0x04,0x00,0x00,0x08, - - }; - - logger->log_bytes(logger,RAW,"expected",expected_generation2,sizeof(expected_generation2)); - - tester->assert_true(tester,(memcmp(expected_generation2,generated_data.ptr,sizeof(expected_generation2)) == 0), "compare generated data"); - - sa_payload->destroy(sa_payload); - list->destroy(list); - proposal1->destroy(proposal1); - proposal2->destroy(proposal2); - chunk_free(&generated_data); - generator->destroy(generator); -} - -/* - * Described in header - */ -void test_generator_with_ke_payload(protected_tester_t *tester) -{ - generator_t *generator; - ke_payload_t *ke_payload; - logger_t *logger; - chunk_t generated_data; - chunk_t key_exchange_data; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - ke_payload = ke_payload_create(); - - - key_exchange_data.ptr = "test-text"; - key_exchange_data.len = strlen(key_exchange_data.ptr); - - ke_payload->set_key_exchange_data(ke_payload,key_exchange_data); - - ke_payload->set_dh_group_number(ke_payload,7777); - - generator->generate_payload(generator,(payload_t *)ke_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x11, - 0x1E,0x61,0x00,0x00, - /* key exchange data */ - 0x74,0x65,0x73,0x74, - 0x2D,0x74,0x65,0x78, - 0x74 - }; - - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - ke_payload->destroy(ke_payload); - generator->destroy(generator); - -} - -/* - * Described in header - */ -void test_generator_with_notify_payload(protected_tester_t *tester) -{ - generator_t *generator; - notify_payload_t *notify_payload; - logger_t *logger; - chunk_t generated_data; - chunk_t notification_data; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - notify_payload = notify_payload_create(); - - notification_data.ptr = "67890"; - notification_data.len = strlen(notification_data.ptr); - - notify_payload->set_protocol_id(notify_payload,255); - notify_payload->set_notify_type(notify_payload,63333); /* Hex F765 */ - notify_payload->set_spi(notify_payload, 0x31323334); - notify_payload->set_notification_data(notify_payload,notification_data); - - generator->generate_payload(generator,(payload_t *)notify_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x11, - 0xFF,0x04,0xF7,0x65, - /* spi */ - 0x31,0x32,0x33,0x34, - /* notification data */ - 0x36,0x37,0x38,0x39, - 0x30, - }; - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - notify_payload->destroy(notify_payload); - generator->destroy(generator); -} - -/* - * Described in header - */ -void test_generator_with_nonce_payload(protected_tester_t *tester) -{ - generator_t *generator; - nonce_payload_t *nonce_payload; - logger_t *logger; - chunk_t generated_data; - chunk_t nonce; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - nonce_payload = nonce_payload_create(); - - - nonce.ptr = "1234567890123456"; - nonce.len = strlen("1234567890123456"); - - nonce_payload->set_nonce(nonce_payload,nonce); - - generator->generate_payload(generator,(payload_t *)nonce_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x14, - /* nonce data */ - 0x31,0x32,0x33,0x34, - 0x35,0x36,0x37,0x38, - 0x39,0x30,0x31,0x32, - 0x33,0x34,0x35,0x36 - }; - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - - nonce_payload->destroy(nonce_payload); - generator->destroy(generator); -} - -/* - * Described in header. - */ -void test_generator_with_id_payload(protected_tester_t *tester) -{ - generator_t *generator; - id_payload_t *id_payload; - logger_t *logger; - chunk_t generated_data; - chunk_t id; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - id_payload = id_payload_create(FALSE); - - - id.ptr = "123456789012"; - id.len = strlen(id.ptr); - - id_payload->set_id_type(id_payload,ID_IPV4_ADDR); - id_payload->set_data(id_payload,id); - - generator->generate_payload(generator,(payload_t *)id_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x14, - 0x01,0x00,0x00,0x00, - /* id data */ - 0x31,0x32,0x33,0x34, - 0x35,0x36,0x37,0x38, - 0x39,0x30,0x31,0x32, - }; - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - id_payload->destroy(id_payload); - generator->destroy(generator); -} - -/* - * Described in header. - */ -void test_generator_with_auth_payload(protected_tester_t *tester) -{ - generator_t *generator; - auth_payload_t *auth_payload; - logger_t *logger; - chunk_t generated_data; - chunk_t auth; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - auth_payload = auth_payload_create(); - - - auth.ptr = "123456789012"; - auth.len = strlen(auth.ptr); - - auth_payload->set_auth_method(auth_payload,SHARED_KEY_MESSAGE_INTEGRITY_CODE); - auth_payload->set_data(auth_payload,auth); - - generator->generate_payload(generator,(payload_t *)auth_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x14, - 0x02,0x00,0x00,0x00, - /* auth data */ - 0x31,0x32,0x33,0x34, - 0x35,0x36,0x37,0x38, - 0x39,0x30,0x31,0x32, - }; - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - auth_payload->destroy(auth_payload); - generator->destroy(generator); -} - -/* - * Described in header. - */ -void test_generator_with_ts_payload(protected_tester_t *tester) -{ - generator_t *generator; - ts_payload_t *ts_payload; - traffic_selector_substructure_t *ts1, *ts2; - host_t *start_host1, *start_host2, *end_host1, *end_host2; - logger_t *logger; - chunk_t generated_data; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - ts_payload = ts_payload_create(TRUE); - - /* first traffic selector */ - ts1 = traffic_selector_substructure_create(); - - start_host1 = host_create(AF_INET,"192.168.1.0",500); - ts1->set_start_host(ts1,start_host1); - start_host1->destroy(start_host1); - - end_host1 = host_create(AF_INET,"192.168.1.255",500); - ts1->set_end_host(ts1,end_host1); - end_host1->destroy(end_host1); - - ts_payload->add_traffic_selector_substructure(ts_payload,ts1); - - /* second traffic selector */ - - ts2 = traffic_selector_substructure_create(); - - start_host2 = host_create(AF_INET,"0.0.0.0",0); - ts2->set_start_host(ts2,start_host2); - ts2->set_protocol_id(ts2,3); - start_host2->destroy(start_host2); - - end_host2 = host_create(AF_INET,"255.255.255.255",65535); - ts2->set_end_host(ts2,end_host2); - end_host2->destroy(end_host2); - - ts_payload->add_traffic_selector_substructure(ts_payload,ts2); - - - generator->generate_payload(generator,(payload_t *)ts_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x28, - 0x02,0x00,0x00,0x00, - - /* traffic selector 1 */ - 0x07,0x00,0x00,0x10, - 0x01,0xF4,0x01,0xF4, - 0xC0,0xA8,0x01,0x00, - 0xC0,0xA8,0x01,0xFF, - - /* traffic selector 2 */ - 0x07,0x03,0x00,0x10, - 0x00,0x00,0xFF,0xFF, - 0x00,0x00,0x00,0x00, - 0xFF,0xFF,0xFF,0xFF, - }; - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - ts_payload->destroy(ts_payload); - generator->destroy(generator); -} - -/* - * Described in header. - */ -void test_generator_with_cert_payload(protected_tester_t *tester) -{ - generator_t *generator; - cert_payload_t *cert_payload; - logger_t *logger; - chunk_t generated_data; - chunk_t cert; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - cert_payload = cert_payload_create(); - - - cert.ptr = "123456789012"; - cert.len = strlen(cert.ptr); - - cert_payload->set_cert_encoding(cert_payload, CERT_PGP); - cert_payload->set_data(cert_payload,cert); - - generator->generate_payload(generator,(payload_t *)cert_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x11, - 0x02, - /* cert data */ - 0x31,0x32,0x33,0x34, - 0x35,0x36,0x37,0x38, - 0x39,0x30,0x31,0x32, - }; - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - cert_payload->destroy(cert_payload); - generator->destroy(generator); -} - -/* - * Described in header. - */ -void test_generator_with_certreq_payload(protected_tester_t *tester) -{ - generator_t *generator; - certreq_payload_t *certreq_payload; - logger_t *logger; - chunk_t generated_data; - chunk_t certreq; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - certreq_payload = certreq_payload_create(); - - - certreq.ptr = "123456789012"; - certreq.len = strlen(certreq.ptr); - - certreq_payload->set_cert_encoding(certreq_payload, CERT_PGP); - certreq_payload->set_data(certreq_payload,certreq); - - generator->generate_payload(generator,(payload_t *)certreq_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x11, - 0x02, - /* certreq data */ - 0x31,0x32,0x33,0x34, - 0x35,0x36,0x37,0x38, - 0x39,0x30,0x31,0x32, - }; - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - certreq_payload->destroy(certreq_payload); - generator->destroy(generator); -} - -/* - * Described in header. - */ -void test_generator_with_delete_payload(protected_tester_t *tester) -{ - generator_t *generator; - delete_payload_t *delete_payload; - logger_t *logger; - chunk_t generated_data; - chunk_t spis; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - delete_payload = delete_payload_create(PROTO_AH); - - - spis.ptr = "123456789012"; - spis.len = strlen(spis.ptr); - - delete_payload->set_spi_count(delete_payload,3); - delete_payload->set_spi_size(delete_payload,4); - delete_payload->set_spis(delete_payload,spis); - - generator->generate_payload(generator,(payload_t *)delete_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x14, - 0x02,0x04,0x00,0x03, - /* delete data */ - 0x31,0x32,0x33,0x34, - 0x35,0x36,0x37,0x38, - 0x39,0x30,0x31,0x32, - }; - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - delete_payload->destroy(delete_payload); - generator->destroy(generator); -} - -/* - * Described in header. - */ -void test_generator_with_vendor_id_payload(protected_tester_t *tester) -{ - generator_t *generator; - vendor_id_payload_t *vendor_id_payload; - logger_t *logger; - chunk_t generated_data; - chunk_t data; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - vendor_id_payload = vendor_id_payload_create(); - - - data.ptr = "123456789012"; - data.len = strlen(data.ptr); -; - vendor_id_payload->set_data(vendor_id_payload,data); - generator->generate_payload(generator,(payload_t *)vendor_id_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x10, - /* vendor_id data */ - 0x31,0x32,0x33,0x34, - 0x35,0x36,0x37,0x38, - 0x39,0x30,0x31,0x32, - }; - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - vendor_id_payload->destroy(vendor_id_payload); - generator->destroy(generator); -} - -/* - * Described in header - */ -void test_generator_with_cp_payload(protected_tester_t *tester) -{ - generator_t *generator; - configuration_attribute_t *attribute1, *attribute2; - cp_payload_t *configuration; - chunk_t data; - chunk_t generated_data; - logger_t *logger; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - /* create attribute 1 */ - attribute1 = configuration_attribute_create(); - char *stringval = "abcd"; - data.ptr = (void *) stringval; - data.len = 4; - attribute1->set_value(attribute1,data); - attribute1->set_attribute_type(attribute1,3); - logger->log(logger,CONTROL,"attribute1 created"); - - /* create attribute 2 */ - attribute2 = configuration_attribute_create(); - stringval = "efgh"; - data.ptr = (void *) stringval; - data.len = 4; - attribute2->set_value(attribute2,data); - attribute2->set_attribute_type(attribute2,4); - logger->log(logger,CONTROL,"attribute2 created"); - - /* create configuration */ - configuration = cp_payload_create(); - tester->assert_true(tester,(configuration != NULL), "configuration create check"); - configuration->add_configuration_attribute(configuration,attribute1); - configuration->add_configuration_attribute(configuration,attribute2); - configuration->set_config_type(configuration,5); /* hex 5 */ - - - logger->log(logger,CONTROL,"cp payload created"); - - generator->generate_payload(generator,(payload_t *)configuration); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated configuration",generated_data); - - u_int8_t expected_generation3[] = { - /* cp payload header */ - 0x00,0x00,0x00,0x18, - 0x05,0x00,0x00,0x00, - /* configuration attribute 1*/ - 0x00,0x03,0x00,0x04, - 0x61,0x62,0x63,0x64, - /* configuration attribute 2*/ - 0x00,0x04,0x00,0x04, - 0x65,0x66,0x67,0x68, - }; - - logger->log_bytes(logger,RAW,"expected configuration",expected_generation3,sizeof(expected_generation3)); - - tester->assert_true(tester,(memcmp(expected_generation3,generated_data.ptr,sizeof(expected_generation3)) == 0), "compare generated data"); - - chunk_free(&generated_data); - configuration->destroy(configuration); - generator->destroy(generator); -} - -/* - * Described in header. - */ -void test_generator_with_eap_payload(protected_tester_t *tester) -{ - generator_t *generator; - eap_payload_t *eap_payload; - logger_t *logger; - chunk_t generated_data; - chunk_t message; - - logger = logger_manager->get_logger(logger_manager,TESTER); - - /* create generator */ - generator = generator_create(); - tester->assert_true(tester,(generator != NULL), "generator create check"); - - eap_payload = eap_payload_create(); - - - message.ptr = "123456789012"; - message.len = strlen(message.ptr); -; - eap_payload->set_message(eap_payload,message); - generator->generate_payload(generator,(payload_t *)eap_payload); - generator->write_to_chunk(generator,&generated_data); - logger->log_chunk(logger,RAW,"generated payload",generated_data); - - u_int8_t expected_generation[] = { - /* payload header */ - 0x00,0x00,0x00,0x10, - /* eap data */ - 0x31,0x32,0x33,0x34, - 0x35,0x36,0x37,0x38, - 0x39,0x30,0x31,0x32, - }; - - logger->log_bytes(logger,RAW,"expected payload",expected_generation,sizeof(expected_generation)); - - tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); - - chunk_free(&generated_data); - - eap_payload->destroy(eap_payload); - generator->destroy(generator); -} |