diff options
Diffstat (limited to 'Source/charon/tests/generator_test.c')
| -rw-r--r-- | Source/charon/tests/generator_test.c | 58 |
1 files changed, 49 insertions, 9 deletions
diff --git a/Source/charon/tests/generator_test.c b/Source/charon/tests/generator_test.c index 49ffb0829..228d50184 100644 --- a/Source/charon/tests/generator_test.c +++ b/Source/charon/tests/generator_test.c @@ -20,6 +20,7 @@ * for more details. */ +#include <string.h> #include <freeswan.h> #include <pluto/constants.h> #include <pluto/defs.h> @@ -33,6 +34,9 @@ extern payload_info_t *payload_infos[]; +/* + * Described in Header + */ void test_generator_with_unsupported_payload(tester_t *tester) { generator_t *generator; @@ -46,11 +50,15 @@ void test_generator_with_unsupported_payload(tester_t *tester) tester->assert_true(tester,(generator->destroy(generator) == SUCCESS), "generator destroy call check"); } +/* + * Described in Header + */ void test_generator_with_header_payload(tester_t *tester) { generator_t *generator; ike_header_t header_data; chunk_t generated_data; + status_t status; header_data.initiator_spi = 1; header_data.responder_spi = 2; @@ -59,24 +67,56 @@ void test_generator_with_header_payload(tester_t *tester) header_data.min_version = 5; header_data.exchange_type = 6; header_data.flags.initiator = TRUE; - header_data.flags.version = TRUE; + header_data.flags.version = FALSE; header_data.flags.response = TRUE; header_data.message_id = 7; header_data.length = 8; generator = generator_create(payload_infos); tester->assert_true(tester,(generator != NULL), "generator create check"); - tester->assert_true(tester,(generator->generate_payload(generator,HEADER,&header_data,&generated_data) == SUCCESS),"generate_payload call check"); - int i; - u_int8_t *data = generated_data.ptr; - - for (i = 0; i < generated_data.len;i++) - { - fprintf(stderr,"%x\n",data[i]); - } + status = generator->generate_payload(generator,HEADER,&header_data,&generated_data); + tester->assert_true(tester,(status == SUCCESS),"generate_payload call check"); + u_int8_t expected_generation[] = { + 0x00,0x00,0x00,0x00, + 0x00,0x00,0x00,0x01, + 0x00,0x00,0x00,0x00, + 0x00,0x00,0x00,0x02, + 0x03,0x45,0x06,0x28, + 0x00,0x00,0x00,0x07, + 0x00,0x00,0x00,0x08, + }; + tester->assert_true(tester,(memcmp(expected_generation,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); + allocator_free_chunk(generated_data); + + header_data.initiator_spi = 0x22000054231234; + header_data.responder_spi = 0x122398; + header_data.next_payload = 0xF3; + header_data.maj_version = 0x2; + header_data.min_version = 0x0; + header_data.exchange_type = 0x12; + header_data.flags.initiator = FALSE; + header_data.flags.version = FALSE; + header_data.flags.response = FALSE; + header_data.message_id = 0x33AFF3; + header_data.length = 0xAA11F; + + status = generator->generate_payload(generator,HEADER,&header_data,&generated_data); + tester->assert_true(tester,(status == SUCCESS),"generate_payload call check"); + + u_int8_t expected_generation2[] = { + 0x00,0x22,0x00,0x00, + 0x54,0x23,0x12,0x34, + 0x00,0x00,0x00,0x00, + 0x00,0x12,0x23,0x98, + 0xF3,0x20,0x12,0x00, + 0x00,0x33,0xAF,0xF3, + 0x00,0x0A,0xA1,0x1F, + }; + + tester->assert_true(tester,(memcmp(expected_generation2,generated_data.ptr,sizeof(expected_generation)) == 0), "compare generated data"); allocator_free_chunk(generated_data); tester->assert_true(tester,(generator->destroy(generator) == SUCCESS), "generator destroy call check"); |