diff options
Diffstat (limited to 'Source')
| -rw-r--r-- | Source/charon/generator.c | 144 |
1 files changed, 140 insertions, 4 deletions
diff --git a/Source/charon/generator.c b/Source/charon/generator.c index 46334ccf8..9f72fce5e 100644 --- a/Source/charon/generator.c +++ b/Source/charon/generator.c @@ -38,11 +38,128 @@ struct private_generator_s { * Public part of a generator object */ generator_t public; + + /* private functions and fields */ + + /** + * Generates a chunk_t with specific encoding rules + * + * items are bytewhise written + * + * @param this private_generator_t-object + * @param data_struct data_struct to read data from + * @param encoding_rules pointer to first encoding_rule of encoding rules array + * @param encoding_rules_count number of encoding rules in encoding rules array + * @param data pointer to chunk where to write the data in + * + * @return SUCCESS if succeeded, + * OUT_OF_RES if out of ressources + */ + status_t (*generate) (private_generator_t *this,void * data_struct,encoding_rule_t *encoding_rules, size_t encoding_rules_count, chunk_t *data); + + /** + * TODO + */ + status_t (*generate_u_int_type) (private_generator_t *this,encoding_type_t int_type,u_int8_t **buffer,u_int8_t **out_position,u_int8_t **roof_position,size_t *current_bit); + + /** + * Pointer to the payload informations needed to automatic + * generate a specific payload type + */ + payload_info_t **payload_infos; }; + +/** + * implements private_generator_t's generate_u_int_type function + */ + +static status_t generate_u_int_type (private_generator_t *this,encoding_type_t int_type,u_int8_t **buffer,u_int8_t **out_position,u_int8_t **roof_position,size_t *current_bit) +{ + size_t number_of_bits = 0; + + switch (int_type) + { + case U_INT_4: + number_of_bits = 4; + break; + case U_INT_8: + number_of_bits = 8; + break; + case U_INT_16: + number_of_bits = 16; + break; + case U_INT_32: + number_of_bits = 32; + break; + case U_INT_64: + number_of_bits = 64; + break; + default: + return FAILED; + } + return SUCCESS; +} + +/** + * implements private_generator_t's generate function + */ +static status_t generate (private_generator_t *this,void * data_struct,encoding_rule_t *encoding_rules, size_t encoding_rules_count, chunk_t *data) +{ + u_int8_t * buffer = alloc_bytes(GENERATOR_DATA_BUFFER_SIZE, "generator buffer"); + u_int8_t * out_position = buffer; + u_int8_t * roof_position = buffer + GENERATOR_DATA_BUFFER_SIZE; + size_t current_bit = 0; + int i; + + if (buffer == NULL) + { + return OUT_OF_RES; + } + for (i = 0; i < encoding_rules_count;i++) + { + status_t status = SUCCESS; + switch (encoding_rules[i].type) + { + case U_INT_4: + case U_INT_8: + case U_INT_16: + case U_INT_32: + case U_INT_64: + status = this->generate_u_int_type(this,encoding_rules[i].type,&buffer,&out_position,&roof_position,¤t_bit); + break; + case RESERVED_BIT: + case RESERVED_BYTE: + case FLAG: + case LENGTH: + case SPI_SIZE: + default: + break; + } + if (status != SUCCESS) + { + pfree(buffer); + return status; + } + } + + return SUCCESS; +} + static status_t generate_payload (private_generator_t *this,payload_type_t payload_type,void * data_struct, chunk_t *data) { - return FAILED; + int i; + + /* check every payload info for specific type */ + for (i = 0; this->payload_infos[i] != NULL; i++) + { + if (this->payload_infos[i]->payload_type == payload_type) + { + /* found payload informations, generating is done in private function generate() */ + return (this->generate(this, data_struct,this->payload_infos[i]->ecoding_rules,this->payload_infos[i]->encoding_rules_count,data)); + } + } + return NOT_SUPPORTED; } /** @@ -59,13 +176,32 @@ static status_t destroy(private_generator_t *this) return SUCCESS; } - -generator_t * generator_create() +/* + * Described in header + */ +generator_t * generator_create(payload_info_t ** payload_infos) { - private_generator_t *this = alloc_thing(private_generator_t,"private_generator_t"); + private_generator_t *this; + + if (payload_infos == NULL) + { + return NULL; + } + + this = alloc_thing(private_generator_t,"private_generator_t"); + if (this == NULL) + { + return NULL; + } this->public.generate_payload = (status_t(*)(generator_t*, payload_type_t, void *, chunk_t *)) generate_payload; this->public.destroy = (status_t(*)(generator_t*)) destroy; + /* initiate private fields */ + this->generate = generate; + this->generate_u_int_type = generate_u_int_type; + + this->payload_infos = payload_infos; + return &(this->public); } |