diff options
Diffstat (limited to 'tinyHTTP/src/thttp_message.c')
-rwxr-xr-x | tinyHTTP/src/thttp_message.c | 603 |
1 files changed, 295 insertions, 308 deletions
diff --git a/tinyHTTP/src/thttp_message.c b/tinyHTTP/src/thttp_message.c index 619576e..4c26c93 100755 --- a/tinyHTTP/src/thttp_message.c +++ b/tinyHTTP/src/thttp_message.c @@ -2,19 +2,19 @@ * Copyright (C) 2010-2011 Mamadou Diop. * * Contact: Mamadou Diop <diopmamadou(at)doubango[dot]org> -* +* * This file is part of Open Source Doubango Framework. * * DOUBANGO is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. -* +* * DOUBANGO is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. -* +* * You should have received a copy of the GNU General Public License * along with DOUBANGO. * @@ -37,35 +37,32 @@ static int pred_find_string_by_value(const tsk_list_item_t *item, const void *stringVal) { - if(item && item->data) - { - tsk_string_t *string = item->data; - return tsk_stricmp(string->value, stringVal); - } - return -1; + if(item && item->data) { + tsk_string_t *string = item->data; + return tsk_stricmp(string->value, stringVal); + } + return -1; } /*== Predicate function to find thttp_header_t object by type. */ static int pred_find_header_by_type(const tsk_list_item_t *item, const void *thttp_htype) { - if(item && item->data) - { - thttp_header_t *header = item->data; - thttp_header_type_t htype = *((thttp_header_type_t*)thttp_htype); - return (header->type - htype); - } - return -1; + if(item && item->data) { + thttp_header_t *header = item->data; + thttp_header_type_t htype = *((thttp_header_type_t*)thttp_htype); + return (header->type - htype); + } + return -1; } /*== Predicate function to find thttp_header_t object by name. */ static int pred_find_header_by_name(const tsk_list_item_t *item, const void *name) { - if(item && item->data) - { - thttp_header_t *header = item->data; - return tsk_stricmp(thttp_header_get_nameex(header), name); - } - return -1; + if(item && item->data) { + thttp_header_t *header = item->data; + return tsk_stricmp(thttp_header_get_nameex(header), name); + } + return -1; } @@ -76,7 +73,7 @@ static int pred_find_header_by_name(const tsk_list_item_t *item, const void *nam */ thttp_message_t* thttp_message_create() { - return tsk_object_new(thttp_message_def_t, thttp_unknown); + return tsk_object_new(thttp_message_def_t, thttp_unknown); } /**@ingroup thttp_message_group @@ -91,7 +88,7 @@ thttp_url_t* url; thttp_request_t* request; if((url = thttp_url_parse("http://www.google.com", tsk_strlen("http://www.google.com")))){ request = thttp_request_create("GET", url); - // ... + // ... TSK_OBJECT_SAFE_FREE(url); TSK_OBJECT_SAFE_FREE(request); } @@ -99,7 +96,7 @@ if((url = thttp_url_parse("http://www.google.com", tsk_strlen("http://www.google */ thttp_request_t* thttp_request_create(const char* method, const thttp_url_t* url) { - return tsk_object_new(thttp_message_def_t, thttp_request, method, url); + return tsk_object_new(thttp_message_def_t, thttp_request, method, url); } @@ -122,7 +119,7 @@ if((response = thttp_response_create(request, 200, "OK"))){ */ thttp_response_t* thttp_response_create(const thttp_request_t* request, short status_code, const char* reason_phrase) { - return tsk_object_new(thttp_message_def_t, thttp_response, request, status_code, reason_phrase); + return tsk_object_new(thttp_message_def_t, thttp_response, request, status_code, reason_phrase); } @@ -130,7 +127,7 @@ thttp_response_t* thttp_response_create(const thttp_request_t* request, short st */ int thttp_message_add_header(thttp_message_t *self, const thttp_header_t *hdr) { - #define ADD_HEADER(type, field) \ +#define ADD_HEADER(type, field) \ case thttp_htype_##type: \ { \ if(!self->field) \ @@ -140,299 +137,295 @@ int thttp_message_add_header(thttp_message_t *self, const thttp_header_t *hdr) } \ break; \ } - - if(self && hdr) - { - thttp_header_t *header = tsk_object_ref((void*)hdr); - switch(header->type) - { - ADD_HEADER(Content_Type, Content_Type); - ADD_HEADER(Content_Length, Content_Length); + if(self && hdr) { + thttp_header_t *header = tsk_object_ref((void*)hdr); + + switch(header->type) { + ADD_HEADER(Content_Type, Content_Type); + ADD_HEADER(Content_Length, Content_Length); - default: break; - } + default: + break; + } - tsk_list_push_back_data(self->headers, (void**)&header); + tsk_list_push_back_data(self->headers, (void**)&header); - return 0; - } - return -1; + return 0; + } + return -1; } /**@ingroup thttp_message_group */ int thttp_message_add_headers(thttp_message_t *self, const thttp_headers_L_t *headers) { - tsk_list_item_t *item = 0; - if(self) - { - tsk_list_foreach(item, headers){ - thttp_message_add_header(self, item->data); - } - return 0; - } - return -1; + tsk_list_item_t *item = 0; + if(self) { + tsk_list_foreach(item, headers) { + thttp_message_add_header(self, item->data); + } + return 0; + } + return -1; } int thttp_message_add_headers_2(thttp_message_t *self, ...) { - const tsk_object_def_t* objdef; - thttp_header_t *header; - va_list ap; - - if(!self){ - TSK_DEBUG_ERROR("Invalid parameter"); - return -1; - } - - va_start(ap, self); - while((objdef = va_arg(ap, const tsk_object_def_t*))){ - if((header = tsk_object_new_2(objdef, &ap))){ - thttp_message_add_header(self, header); - TSK_OBJECT_SAFE_FREE(header); - } - } - va_end(ap); - - return 0; + const tsk_object_def_t* objdef; + thttp_header_t *header; + va_list ap; + + if(!self) { + TSK_DEBUG_ERROR("Invalid parameter"); + return -1; + } + + va_start(ap, self); + while((objdef = va_arg(ap, const tsk_object_def_t*))) { + if((header = tsk_object_new_2(objdef, &ap))) { + thttp_message_add_header(self, header); + TSK_OBJECT_SAFE_FREE(header); + } + } + va_end(ap); + + return 0; } /**@ingroup thttp_message_group */ int thttp_message_add_content(thttp_message_t *self, const char* content_type, const void* content, tsk_size_t size) { - if(self && content && size) - { - if(content_type){ - TSK_OBJECT_SAFE_FREE(self->Content_Type); - } - TSK_OBJECT_SAFE_FREE(self->Content_Length); - TSK_OBJECT_SAFE_FREE(self->Content); - - if(content_type){ - THTTP_MESSAGE_ADD_HEADER(self, THTTP_HEADER_CONTENT_TYPE_VA_ARGS(content_type)); - } - THTTP_MESSAGE_ADD_HEADER(self, THTTP_HEADER_CONTENT_LENGTH_VA_ARGS(size)); - self->Content = tsk_buffer_create(content, size); - - return 0; - } - return -1; + if(self && content && size) { + if(content_type) { + TSK_OBJECT_SAFE_FREE(self->Content_Type); + } + TSK_OBJECT_SAFE_FREE(self->Content_Length); + TSK_OBJECT_SAFE_FREE(self->Content); + + if(content_type) { + THTTP_MESSAGE_ADD_HEADER(self, THTTP_HEADER_CONTENT_TYPE_VA_ARGS(content_type)); + } + THTTP_MESSAGE_ADD_HEADER(self, THTTP_HEADER_CONTENT_LENGTH_VA_ARGS(size)); + self->Content = tsk_buffer_create(content, size); + + return 0; + } + return -1; } /**@ingroup thttp_message_group */ int thttp_message_append_content(thttp_message_t *self, const void* content, tsk_size_t size) { - if(self && content && size){ - if(!self->Content){ - self->Content = tsk_buffer_create(content, size); - } - else{ - tsk_buffer_append(self->Content, content, size); - } - - if(!self->Content_Length){ - THTTP_MESSAGE_ADD_HEADER(self, THTTP_HEADER_CONTENT_LENGTH_VA_ARGS(size)); - } - else{ - self->Content_Length->length += (uint32_t)size; - } - return 0; - } - - return -1; + if(self && content && size) { + if(!self->Content) { + self->Content = tsk_buffer_create(content, size); + } + else { + tsk_buffer_append(self->Content, content, size); + } + + if(!self->Content_Length) { + THTTP_MESSAGE_ADD_HEADER(self, THTTP_HEADER_CONTENT_LENGTH_VA_ARGS(size)); + } + else { + self->Content_Length->length += (uint32_t)size; + } + return 0; + } + + return -1; } /**@ingroup thttp_message_group */ const thttp_header_t *thttp_message_get_headerAt(const thttp_message_t *self, thttp_header_type_t type, tsk_size_t index) { - tsk_size_t pos = 0; - tsk_list_item_t *item; - const thttp_header_t* hdr = tsk_null; - - if(self) - { - switch(type) - { - case thttp_htype_Content_Type: - if(index == 0){ - hdr = (const thttp_header_t*)self->Content_Type; - goto bail; - }else pos++; break; - case thttp_htype_Content_Length: - if(index == 0){ - hdr = (const thttp_header_t*)self->Content_Length; - goto bail; - }else pos++; break; - default: - break; - } - - tsk_list_foreach(item, self->headers) - { - if(!pred_find_header_by_type(item, &type)) - { - if(pos++ >= index) - { - hdr = item->data; - break; - } - } - } - } + tsk_size_t pos = 0; + tsk_list_item_t *item; + const thttp_header_t* hdr = tsk_null; + + if(self) { + switch(type) { + case thttp_htype_Content_Type: + if(index == 0) { + hdr = (const thttp_header_t*)self->Content_Type; + goto bail; + } + else { + pos++; + } + break; + case thttp_htype_Content_Length: + if(index == 0) { + hdr = (const thttp_header_t*)self->Content_Length; + goto bail; + } + else { + pos++; + } + break; + default: + break; + } + + tsk_list_foreach(item, self->headers) { + if(!pred_find_header_by_type(item, &type)) { + if(pos++ >= index) { + hdr = item->data; + break; + } + } + } + } bail: - return hdr; + return hdr; } /**@ingroup thttp_message_group */ const thttp_header_t *thttp_message_get_header(const thttp_message_t *self, thttp_header_type_t type) { - return thttp_message_get_headerAt(self, type, 0); + return thttp_message_get_headerAt(self, type, 0); } /**@ingroup thttp_message_group */ const thttp_header_t *thttp_message_get_headerByName(const thttp_message_t *self, const char* name) { - //tsk_size_t pos = 0; - tsk_list_item_t *item; - const thttp_header_t* hdr = tsk_null; - - if(self) - { - if(tsk_striequals(name, "Content-Type")){ - hdr = (const thttp_header_t*)self->Content_Type; - goto bail; - } - - if(tsk_striequals(name, "Content-Length")){ - hdr = (const thttp_header_t*)self->Content_Length; - goto bail; - } - - tsk_list_foreach(item, self->headers) - { - if(!pred_find_header_by_name(item, name)) - { - hdr = item->data; - break; - } - } - } + //tsk_size_t pos = 0; + tsk_list_item_t *item; + const thttp_header_t* hdr = tsk_null; + + if(self) { + if(tsk_striequals(name, "Content-Type")) { + hdr = (const thttp_header_t*)self->Content_Type; + goto bail; + } + + if(tsk_striequals(name, "Content-Length")) { + hdr = (const thttp_header_t*)self->Content_Length; + goto bail; + } + + tsk_list_foreach(item, self->headers) { + if(!pred_find_header_by_name(item, name)) { + hdr = item->data; + break; + } + } + } bail: - return hdr; + return hdr; } /**@ingroup thttp_message_group */ int thttp_message_serialize(const thttp_message_t *self, tsk_buffer_t *output) { - if(!self || !output){ - return -1; - } - - if(THTTP_MESSAGE_IS_REQUEST(self)){ - /*Method SP Request-URI SP HTTP-Version CRLF*/ - /* Method */ - tsk_buffer_append_2(output, "%s ", self->line.request.method); - /* Request URI: hpath?search */ - if(self->line.request.url){ - tsk_buffer_append_2(output, "/%s%s%s ", - self->line.request.url->hpath ? self->line.request.url->hpath : "", - self->line.request.url->search ? "?" : "", - self->line.request.url->search ? self->line.request.url->search : "" - ); - } - /* HTTP VERSION */ - tsk_buffer_append_2(output, "%s\r\n", THTTP_MESSAGE_VERSION_DEFAULT); - /* HOST */ - if(self->line.request.url && self->line.request.url->host && self->line.request.url->port){ - tsk_buffer_append_2(output, "Host: %s:%u\r\n", self->line.request.url->host, self->line.request.url->port); - } - } - else{ - /*HTTP-Version SP Status-Code SP Reason-Phrase CRLF*/ - tsk_buffer_append_2(output, "%s %hi %s\r\n", THTTP_MESSAGE_VERSION_DEFAULT, THTTP_RESPONSE_CODE(self), THTTP_RESPONSE_PHRASE(self)); - } - - /* Content-Type */ - if(self->Content_Type){ - thttp_header_serialize(THTTP_HEADER(self->Content_Type), output); - } - /* Content-Length*/ - if(self->Content_Length){ - thttp_header_serialize(THTTP_HEADER(self->Content_Length), output); - } - - /* All other headers */ - { - tsk_list_item_t *item; - tsk_list_foreach(item, self->headers) - { - thttp_header_t *hdr = item->data; - thttp_header_serialize(hdr, output); - } - } - - /* EMPTY LINE */ - tsk_buffer_append(output, "\r\n", 2); - - /* CONTENT */ - if(THTTP_MESSAGE_HAS_CONTENT(self)){ - tsk_buffer_append(output, TSK_BUFFER_TO_STRING(self->Content), TSK_BUFFER_SIZE(self->Content)); - } - - return 0; + if(!self || !output) { + return -1; + } + + if(THTTP_MESSAGE_IS_REQUEST(self)) { + /*Method SP Request-URI SP HTTP-Version CRLF*/ + /* Method */ + tsk_buffer_append_2(output, "%s ", self->line.request.method); + /* Request URI: hpath?search */ + if(self->line.request.url) { + tsk_buffer_append_2(output, "/%s%s%s ", + self->line.request.url->hpath ? self->line.request.url->hpath : "", + self->line.request.url->search ? "?" : "", + self->line.request.url->search ? self->line.request.url->search : "" + ); + } + /* HTTP VERSION */ + tsk_buffer_append_2(output, "%s\r\n", THTTP_MESSAGE_VERSION_DEFAULT); + /* HOST */ + if(self->line.request.url && self->line.request.url->host && self->line.request.url->port) { + tsk_buffer_append_2(output, "Host: %s:%u\r\n", self->line.request.url->host, self->line.request.url->port); + } + } + else { + /*HTTP-Version SP Status-Code SP Reason-Phrase CRLF*/ + tsk_buffer_append_2(output, "%s %hi %s\r\n", THTTP_MESSAGE_VERSION_DEFAULT, THTTP_RESPONSE_CODE(self), THTTP_RESPONSE_PHRASE(self)); + } + + /* Content-Type */ + if(self->Content_Type) { + thttp_header_serialize(THTTP_HEADER(self->Content_Type), output); + } + /* Content-Length*/ + if(self->Content_Length) { + thttp_header_serialize(THTTP_HEADER(self->Content_Length), output); + } + + /* All other headers */ + { + tsk_list_item_t *item; + tsk_list_foreach(item, self->headers) { + thttp_header_t *hdr = item->data; + thttp_header_serialize(hdr, output); + } + } + + /* EMPTY LINE */ + tsk_buffer_append(output, "\r\n", 2); + + /* CONTENT */ + if(THTTP_MESSAGE_HAS_CONTENT(self)) { + tsk_buffer_append(output, TSK_BUFFER_TO_STRING(self->Content), TSK_BUFFER_SIZE(self->Content)); + } + + return 0; } /**@ingroup thttp_message_group */ char* thttp_message_tostring(const thttp_message_t *self) { - char* ret = tsk_null; - tsk_buffer_t* output = tsk_buffer_create_null(); + char* ret = tsk_null; + tsk_buffer_t* output = tsk_buffer_create_null(); - if(!thttp_message_serialize(self, output)){ - ret = tsk_strndup(output->data, output->size); - } + if(!thttp_message_serialize(self, output)) { + ret = tsk_strndup(output->data, output->size); + } - TSK_OBJECT_SAFE_FREE(output); - return ret; + TSK_OBJECT_SAFE_FREE(output); + return ret; } /**@ingroup thttp_message_group */ thttp_request_t *thttp_request_new(const char* method, const thttp_url_t *request_url) { - thttp_request_t* request = 0; - - if((request = thttp_request_create(method, request_url))){ - THTTP_MESSAGE_ADD_HEADER(request, THTTP_HEADER_CONTENT_LENGTH_VA_ARGS(0)); - } - return request; + thttp_request_t* request = 0; + + if((request = thttp_request_create(method, request_url))) { + THTTP_MESSAGE_ADD_HEADER(request, THTTP_HEADER_CONTENT_LENGTH_VA_ARGS(0)); + } + return request; } /**@ingroup thttp_message_group */ thttp_response_t *thttp_response_new(short status_code, const char* reason_phrase, const thttp_request_t *request) { - thttp_response_t *response = 0; + thttp_response_t *response = 0; - if(request){ - response = thttp_response_create(request, status_code, reason_phrase); - THTTP_MESSAGE_ADD_HEADER(response, THTTP_HEADER_CONTENT_LENGTH_VA_ARGS(0)); - /* - Copy other headers - */ - } + if(request) { + response = thttp_response_create(request, status_code, reason_phrase); + THTTP_MESSAGE_ADD_HEADER(response, THTTP_HEADER_CONTENT_LENGTH_VA_ARGS(0)); + /* + Copy other headers + */ + } - return response; + return response; } @@ -451,80 +444,74 @@ thttp_response_t *thttp_response_new(short status_code, const char* reason_phras */ static tsk_object_t* thttp_message_ctor(tsk_object_t *self, va_list * app) { - thttp_message_t *message = self; - if(message) - { - message->type = va_arg(*app, thttp_message_type_t); - message->headers = tsk_list_create(); - - switch(message->type) - { - case thttp_unknown: - { - break; - } - - case thttp_request: - { - message->line.request.method = tsk_strdup(va_arg(*app, const char*)); - message->line.request.url = tsk_object_ref((void*)va_arg(*app, const thttp_url_t*)); - break; - } - - case thttp_response: - { - /*const thttp_request_t* request =*/ va_arg(*app, const thttp_request_t*); + thttp_message_t *message = self; + if(message) { + message->type = va_arg(*app, thttp_message_type_t); + message->headers = tsk_list_create(); + + switch(message->type) { + case thttp_unknown: { + break; + } + + case thttp_request: { + message->line.request.method = tsk_strdup(va_arg(*app, const char*)); + message->line.request.url = tsk_object_ref((void*)va_arg(*app, const thttp_url_t*)); + break; + } + + case thttp_response: { + /*const thttp_request_t* request =*/ va_arg(*app, const thttp_request_t*); #if defined(__GNUC__) - message->line.response.status_code = (short)va_arg(*app, int); + message->line.response.status_code = (short)va_arg(*app, int); #else - message->line.response.status_code = va_arg(*app, short); + message->line.response.status_code = va_arg(*app, short); #endif - message->line.response.reason_phrase = tsk_strdup(va_arg(*app, const char*)); - break; - } - } - } - else{ - TSK_DEBUG_ERROR("Failed to create new http message."); - } - return self; + message->line.response.reason_phrase = tsk_strdup(va_arg(*app, const char*)); + break; + } + } + } + else { + TSK_DEBUG_ERROR("Failed to create new http message."); + } + return self; } /**@ingroup thttp_message_group */ static tsk_object_t* thttp_message_dtor(tsk_object_t *self) { - thttp_message_t *message = self; - if(message){ - if(THTTP_MESSAGE_IS_REQUEST(message)){ - TSK_FREE(message->line.request.method); - TSK_OBJECT_SAFE_FREE(message->line.request.url); - } - else if(THTTP_MESSAGE_IS_RESPONSE(message)){ - TSK_FREE(message->line.response.reason_phrase); - } - - TSK_FREE(message->http_version); - - TSK_OBJECT_SAFE_FREE(message->Content_Length); - TSK_OBJECT_SAFE_FREE(message->Content_Type); - - TSK_OBJECT_SAFE_FREE(message->Content); - - TSK_OBJECT_SAFE_FREE(message->headers); - } - else{ - TSK_DEBUG_ERROR("Null HTTP message."); - } - - return self; + thttp_message_t *message = self; + if(message) { + if(THTTP_MESSAGE_IS_REQUEST(message)) { + TSK_FREE(message->line.request.method); + TSK_OBJECT_SAFE_FREE(message->line.request.url); + } + else if(THTTP_MESSAGE_IS_RESPONSE(message)) { + TSK_FREE(message->line.response.reason_phrase); + } + + TSK_FREE(message->http_version); + + TSK_OBJECT_SAFE_FREE(message->Content_Length); + TSK_OBJECT_SAFE_FREE(message->Content_Type); + + TSK_OBJECT_SAFE_FREE(message->Content); + + TSK_OBJECT_SAFE_FREE(message->headers); + } + else { + TSK_DEBUG_ERROR("Null HTTP message."); + } + + return self; } -static const tsk_object_def_t thttp_message_def_s = -{ - sizeof(thttp_message_t), - thttp_message_ctor, - thttp_message_dtor, - tsk_null +static const tsk_object_def_t thttp_message_def_s = { + sizeof(thttp_message_t), + thttp_message_ctor, + thttp_message_dtor, + tsk_null }; const tsk_object_def_t *thttp_message_def_t = &thttp_message_def_s; |