diff options
Diffstat (limited to 'Source/charon/testcases')
| -rw-r--r-- | Source/charon/testcases/generator_test.c | 62 | ||||
| -rw-r--r-- | Source/charon/testcases/generator_test.h | 8 | ||||
| -rw-r--r-- | Source/charon/testcases/testcases.c | 7 |
3 files changed, 74 insertions, 3 deletions
diff --git a/Source/charon/testcases/generator_test.c b/Source/charon/testcases/generator_test.c index a28b8f37e..76150060c 100644 --- a/Source/charon/testcases/generator_test.c +++ b/Source/charon/testcases/generator_test.c @@ -36,6 +36,7 @@ #include "../payloads/proposal_substructure.h" #include "../payloads/sa_payload.h" #include "../payloads/ke_payload.h" +#include "../payloads/notify_payload.h" /* * Described in Header @@ -673,3 +674,64 @@ void test_generator_with_ke_payload(tester_t *tester) } +/* + * Described in header + */ +void test_generator_with_notify_payload(tester_t *tester) +{ + generator_t *generator; + notify_payload_t *notify_payload; + logger_t *logger; + status_t status; + chunk_t generated_data; + chunk_t spi,notification_data; + + logger = global_logger_manager->create_logger(global_logger_manager,TESTER,"Message with Notify Payload"); + + /* 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); + + status = generator->generate_payload(generator,(payload_t *)notify_payload); + tester->assert_true(tester,(status == SUCCESS),"generate_payload call check"); + tester->assert_true(tester,(generator->write_to_chunk(generator,&generated_data) == SUCCESS),"write_to_chunk call check"); + 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"); + + allocator_free_chunk(generated_data); + + tester->assert_true(tester,(notify_payload->destroy(notify_payload) == SUCCESS), "notify_payload destroy call check"); + tester->assert_true(tester,(generator->destroy(generator) == SUCCESS), "generator destroy call check"); + + global_logger_manager->destroy_logger(global_logger_manager,logger); + +}
\ No newline at end of file diff --git a/Source/charon/testcases/generator_test.h b/Source/charon/testcases/generator_test.h index 6b72994b4..0c6fc22f4 100644 --- a/Source/charon/testcases/generator_test.h +++ b/Source/charon/testcases/generator_test.h @@ -74,5 +74,13 @@ void test_generator_with_sa_payload(tester_t *tester); */ void test_generator_with_ke_payload(tester_t *tester); +/** + * @brief Test function used to test the generator with Notify payload + * + * + * @param tester associated tester object + */ +void test_generator_with_notify_payload(tester_t *tester); + #endif /*GENERATOR_TEST_H_*/ diff --git a/Source/charon/testcases/testcases.c b/Source/charon/testcases/testcases.c index b99c3e640..80fa21ef7 100644 --- a/Source/charon/testcases/testcases.c +++ b/Source/charon/testcases/testcases.c @@ -127,6 +127,7 @@ test_t generator_test3 = {test_generator_with_transform_substructure,"Generator: test_t generator_test4 = {test_generator_with_proposal_substructure,"Generator: proposal substructure"}; test_t generator_test5 = {test_generator_with_sa_payload,"Generator: Message with SA Payload"}; test_t generator_test6 = {test_generator_with_ke_payload,"Generator: KE Payload"}; +test_t generator_test7 = {test_generator_with_notify_payload,"Generator: Notify Payload"}; /** @@ -210,6 +211,7 @@ logger_manager_t *global_logger_manager; &generator_test4, &generator_test5, &generator_test6, + &generator_test7, &ike_sa_manager_test, &packet_test, NULL @@ -226,9 +228,8 @@ logger_manager_t *global_logger_manager; tester_t *tester = tester_create(test_output, FALSE); - -// tester->perform_tests(tester,all_tests); - tester->perform_test(tester,&parser_test4); + tester->perform_tests(tester,all_tests); +// tester->perform_test(tester,&generator_test7); tester->destroy(tester); |