- all header field types implemented and tested

This commit is contained in:
Jan Hutter
2005-11-09 16:07:33 +00:00
parent 61c3cdc26a
commit e6fbf1d259
2 changed files with 91 additions and 12 deletions
+42 -3
View File
@@ -350,10 +350,7 @@ static status_t generate_u_int_type (private_generator_t *this,encoding_type_t i
return FAILED;
}
return SUCCESS;
}
@@ -379,6 +376,7 @@ static status_t generate (private_generator_t *this,void * data_struct,encoding_
status = SUCCESS;
switch (encoding_rules[i].type)
{
/* all u int values are generated in generate_u_int_type */
case U_INT_4:
case U_INT_8:
case U_INT_16:
@@ -387,10 +385,51 @@ static status_t generate (private_generator_t *this,void * data_struct,encoding_
status = this->generate_u_int_type(this,encoding_rules[i].type,encoding_rules[i].offset,infos);
break;
case RESERVED_BIT:
{
u_int8_t reserved_bit = ~(1 << (7 - infos->current_bit));
*(infos->out_position) = *(infos->out_position) & reserved_bit;
infos->current_bit++;
if (infos->current_bit >= 8)
{
infos->current_bit = infos->current_bit % 8;
status = infos->make_space_available(infos,8);
infos->out_position++;
}
break;
}
case RESERVED_BYTE:
{
if (infos->current_bit > 0)
{
return FAILED;
}
*(infos->out_position) = 0x00;
infos->out_position++;
break;
}
case FLAG:
{
u_int8_t flag_value = (*((bool *) (infos->data_struct + encoding_rules[i].offset))) ? 1 : 0;
u_int8_t flag = (flag_value << (7 - infos->current_bit));
*(infos->out_position) = *(infos->out_position) | flag;
infos->current_bit++;
if (infos->current_bit >= 8)
{
infos->current_bit = infos->current_bit % 8;
status = infos->make_space_available(infos,8);
infos->out_position++;
}
break;
}
case LENGTH:
/* length is generated like an U_INT_32 */
status = this->generate_u_int_type(this,U_INT_32,encoding_rules[i].offset,infos);
case SPI_SIZE:
/* actually not implemented */
default:
break;
}
+49 -9
View File
@@ -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");