- begun to implement generator
This commit is contained in:
parent
d94fe8b3b9
commit
5f08382cbe
1 changed files with 140 additions and 4 deletions
|
@ -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);
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue