aboutsummaryrefslogtreecommitdiffstats
path: root/src/charon/testing/generator_test.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/charon/testing/generator_test.c')
-rw-r--r--src/charon/testing/generator_test.c1410
1 files changed, 1410 insertions, 0 deletions
diff --git a/src/charon/testing/generator_test.c b/src/charon/testing/generator_test.c
new file mode 100644
index 000000000..004c700e6
--- /dev/null
+++ b/src/charon/testing/generator_test.c
@@ -0,0 +1,1410 @@
+/**
+ * @file generator_test.c
+ *
+ * @brief Tests for the generator_t class.
+ *
+ */
+
+/*
+ * Copyright (C) 2005 Jan Hutter, Martin Willi
+ * 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(1);
+ proposal1->add_algorithm(proposal1, PROTO_IKE, ENCRYPTION_ALGORITHM, 1, 20);
+ proposal1->add_algorithm(proposal1, PROTO_IKE, PSEUDO_RANDOM_FUNCTION, 2, 22);
+ proposal1->add_algorithm(proposal1, PROTO_IKE, INTEGRITY_ALGORITHM, 3, 24);
+ proposal1->add_algorithm(proposal1, PROTO_IKE, DIFFIE_HELLMAN_GROUP, 4, 0);
+
+ proposal2 = proposal_create(2);
+ proposal2->add_algorithm(proposal2, PROTO_IKE, ENCRYPTION_ALGORITHM, 5, 26);
+ proposal2->add_algorithm(proposal2, PROTO_IKE, PSEUDO_RANDOM_FUNCTION, 6, 28);
+ proposal2->add_algorithm(proposal2, PROTO_IKE, INTEGRITY_ALGORITHM, 7, 30);
+ proposal2->add_algorithm(proposal2, PROTO_IKE, 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);
+
+
+ /* --------------------------- */
+ /* test with automatic created child proposals */
+
+ generator = generator_create();
+ tester->assert_true(tester,(generator != NULL), "generator create check");
+
+
+ proposal1 = proposal_create(1);
+
+ proposal1->add_algorithm(proposal1, PROTO_AH, INTEGRITY_ALGORITHM, AUTH_HMAC_MD5_96, 20);
+ proposal1->add_algorithm(proposal1, PROTO_AH, DIFFIE_HELLMAN_GROUP, MODP_2048_BIT, 0);
+ proposal1->add_algorithm(proposal1, PROTO_AH, EXTENDED_SEQUENCE_NUMBERS, EXT_SEQ_NUMBERS, 0);
+ proposal1->set_spi(proposal1, PROTO_AH, 0x01010101l);
+
+ proposal1->add_algorithm(proposal1, PROTO_ESP, ENCRYPTION_ALGORITHM, ENCR_AES_CBC, 20);
+ proposal1->add_algorithm(proposal1, PROTO_ESP, DIFFIE_HELLMAN_GROUP, MODP_1024_BIT, 0);
+ proposal1->set_spi(proposal1, PROTO_ESP, 0x02020202);
+
+
+ proposal2->add_algorithm(proposal2, PROTO_AH, INTEGRITY_ALGORITHM, AUTH_HMAC_MD5_96, 20);
+ proposal2->add_algorithm(proposal2, PROTO_AH, DIFFIE_HELLMAN_GROUP, MODP_2048_BIT, 0);
+ proposal2->add_algorithm(proposal2, PROTO_AH, EXTENDED_SEQUENCE_NUMBERS, EXT_SEQ_NUMBERS, 0);
+ proposal2->set_spi(proposal2, PROTO_AH, 0x01010101);
+
+ proposal2->add_algorithm(proposal2, PROTO_ESP, ENCRYPTION_ALGORITHM, ENCR_AES_CBC, 32);
+ proposal2->add_algorithm(proposal2, PROTO_ESP, INTEGRITY_ALGORITHM, AUTH_HMAC_MD5_96, 20);
+ proposal2->add_algorithm(proposal2, PROTO_ESP, DIFFIE_HELLMAN_GROUP, MODP_1024_BIT, 0);
+ proposal2->set_spi(proposal2, PROTO_ESP, 0x02020202);
+
+ 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_generation3[] = {
+ 0x00,0x00,0x00,0xA0, /* payload header*/
+
+ /* suite 1 */
+ 0x02,0x00,0x00,0x28, /* a proposal */
+ 0x01,0x02,0x04,0x03,
+ 0x01,0x01,0x01,0x01,
+ 0x03,0x00,0x00,0x0C, /* transform 1 */
+ 0x03,0x00,0x00,0x01,
+ 0x80,0x0E,0x00,0x14, /* keylength attribute with 20 bytes length */
+
+ 0x03,0x00,0x00,0x08, /* transform 2 */
+ 0x04,0x00,0x00,0x0E,
+
+ 0x00,0x00,0x00,0x08, /* transform 3 */
+ 0x05,0x00,0x00,0x01,
+
+
+ 0x02,0x00,0x00,0x20, /* a proposal */
+ 0x01,0x03,0x04,0x02,
+ 0x02,0x02,0x02,0x02,
+
+ 0x03,0x00,0x00,0x0C, /* transform 1 */
+ 0x01,0x00,0x00,0x0C,
+ 0x80,0x0E,0x00,0x20, /* keylength attribute with 32 bytes length */
+
+ 0x00,0x00,0x00,0x08, /* transform 2 */
+ 0x04,0x00,0x00,0x02,
+
+ /* suite 2 */
+ 0x02,0x00,0x00,0x28, /* a proposal */
+ 0x02,0x02,0x04,0x03,
+ 0x01,0x01,0x01,0x01,
+ 0x03,0x00,0x00,0x0C, /* transform 1 */
+ 0x03,0x00,0x00,0x01,
+ 0x80,0x0E,0x00,0x14, /* keylength attribute with 20 bytes length */
+
+ 0x03,0x00,0x00,0x08, /* transform 2 */
+ 0x04,0x00,0x00,0x0E,
+
+ 0x00,0x00,0x00,0x08, /* transform 3 */
+ 0x05,0x00,0x00,0x01,
+
+
+ 0x00,0x00,0x00,0x2C, /* a proposal */
+ 0x02,0x03,0x04,0x03,
+ 0x02,0x02,0x02,0x02,
+
+ 0x03,0x00,0x00,0x0C, /* transform 1 */
+ 0x01,0x00,0x00,0x0C,
+ 0x80,0x0E,0x00,0x20, /* keylength attribute with 32 bytes length */
+
+ 0x03,0x00,0x00,0x0C, /* transform 2 */
+ 0x03,0x00,0x00,0x01,
+ 0x80,0x0E,0x00,0x14, /* keylength attribute with 20 bytes length */
+
+ 0x00,0x00,0x00,0x08, /* transform 3 */
+ 0x04,0x00,0x00,0x02,
+
+ };
+
+
+ logger->log_bytes(logger,RAW,"expected",expected_generation3,sizeof(expected_generation3));
+
+ tester->assert_true(tester,(memcmp(expected_generation3,generated_data.ptr,sizeof(expected_generation3)) == 0), "compare generated data");
+
+ sa_payload->destroy(sa_payload);
+ proposal1->destroy(proposal1);
+ proposal2->destroy(proposal2);
+ list->destroy(list);
+ 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 spi,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();
+
+
+ spi.ptr = "12345";
+ spi.len = strlen(spi.ptr);
+
+ notification_data.ptr = "67890";
+ notification_data.len = strlen(notification_data.ptr);
+
+ notify_payload->set_protocol_id(notify_payload,255);
+ notify_payload->set_notify_message_type(notify_payload,63333); /* Hex F765 */
+ notify_payload->set_spi(notify_payload,spi);
+ 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,0x12,
+ 0xFF,0x05,0xF7,0x65,
+ /* spi */
+ 0x31,0x32,0x33,0x34,
+ 0x35,
+ /* 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(FALSE);
+
+
+ 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,PGP_CERTIFICATE);
+ 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,PGP_CERTIFICATE);
+ 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();
+
+
+ spis.ptr = "123456789012";
+ spis.len = strlen(spis.ptr);
+
+ delete_payload->set_protocol_id(delete_payload, PROTO_AH);
+ 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);
+}