diff options
Diffstat (limited to 'Source/charon/generator.c')
-rw-r--r-- | Source/charon/generator.c | 26 |
1 files changed, 25 insertions, 1 deletions
diff --git a/Source/charon/generator.c b/Source/charon/generator.c index 9fb304787..523af80a6 100644 --- a/Source/charon/generator.c +++ b/Source/charon/generator.c @@ -36,7 +36,7 @@ #include "payloads/proposal_substructure.h" #include "payloads/transform_substructure.h" #include "payloads/sa_payload.h" - +#include "payloads/ke_payload.h" extern logger_manager_t *global_logger_manager; @@ -736,6 +736,30 @@ static status_t generate_payload (private_generator_t *this,payload_t *payload) status = this->generate_from_chunk(this,rules[i].offset); break; } + case KEY_EXCHANGE_DATA: + { + /* the Key Exchange Data value is generated from chunk */ + status = this->generate_from_chunk(this,rules[i].offset); + if (status != SUCCESS) + { + this->logger->log(this->logger,CONTROL_MORE,"Could no write key exchange data from chunk"); + return status; + } + + /* before iterative generate the transforms, store the current payload length position */ + u_int32_t payload_length_position_offset = this->last_payload_length_position_offset; + /* Length of KE_PAYLOAD is calculated */ + u_int16_t length_of_ke_payload = KE_PAYLOAD_HEADER_LENGTH + ((chunk_t *)(this->data_struct + rules[i].offset))->len; + + u_int16_t int16_val = htons(length_of_ke_payload); + status = this->write_bytes_to_buffer_at_offset(this,&int16_val,sizeof(u_int16_t),payload_length_position_offset); + if (status != SUCCESS) + { + this->logger->log(this->logger,CONTROL_MORE,"Could no write payload length into buffer"); + return status; + } + break; + } case PROPOSALS: { /* before iterative generate the transforms, store the current payload length position */ |