diff options
author | bossiel <bossiel@yahoo.fr> | 2011-08-10 22:59:15 +0000 |
---|---|---|
committer | bossiel <bossiel@yahoo.fr> | 2011-08-10 22:59:15 +0000 |
commit | 1ebf5a5fcda0c9154e22ed02404fd46525a7fd9f (patch) | |
tree | 4b6214a7142ab1035cb0e47444e88af38e712421 /tinyHTTP/src/thttp_message.c | |
download | doubango-1.0.zip doubango-1.0.tar.gz |
Move deprecated v1.0 from trunk to branches1.0
Diffstat (limited to 'tinyHTTP/src/thttp_message.c')
-rw-r--r-- | tinyHTTP/src/thttp_message.c | 503 |
1 files changed, 503 insertions, 0 deletions
diff --git a/tinyHTTP/src/thttp_message.c b/tinyHTTP/src/thttp_message.c new file mode 100644 index 0000000..454aa71 --- /dev/null +++ b/tinyHTTP/src/thttp_message.c @@ -0,0 +1,503 @@ +/* +* Copyright (C) 2009-2010 Mamadou Diop. +* +* Contact: Mamadou Diop <diopmamadou(at)doubango.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. +* +*/ + +/**@file thttp_message.c + * @brief Represents a HTTP message. A HTTP message is either a request from a client to a server, or a + * response from a server to a client. + * + * @author Mamadou Diop <diopmamadou(at)doubango.org> + * + * @date Created: Sat Nov 8 16:54:58 2009 mdiop + */ +#include "tinyhttp/thttp_message.h" + +#include "tsk_debug.h" + +/**@defgroup thttp_message_group HTTP Message +*/ + +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; +} + +/*== 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; +} + +/*== 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; +} + + +/**@ingroup thttp_message_group +* Creates new HTTP message. Could be either a request or a response. +* @retval @ref thttp_message_t object. +* @sa @ref thttp_request_create()<br>@ref thttp_response_create() +*/ +thttp_message_t* thttp_message_create() +{ + return tsk_object_new(thttp_message_def_t, thttp_unknown); +} + +/**@ingroup thttp_message_group +* Creates new HTTP request. +* @param method The method (const char*). e.g. GET, POST, HEAD ... +* @param url The url (@ref thttp_url_t). +* @retval @ref thttp_request_t object. +* +* @code +// example +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); +} +* @endcode +*/ +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); +} + + + +/**@ingroup thttp_message_group +* Creates a HTTP response. +* @param request The request (@ref thttp_request_t) from which to create the response. +* @param status_code The status code (short). +* @param reason_phrase The reason phrase (const char*). +* @retval @ref thttp_response_t object. +* +* @code +// example +//thttp_request_t* request; +thttp_response_t* response; +if((response = thttp_response_create(request, 200, "OK"))){ + TSK_OBJECT_SAFE_FREE(response); +} +* @endcode +*/ +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); +} + + +/**@ingroup thttp_message_group +*/ +int thttp_message_add_header(thttp_message_t *self, const thttp_header_t *hdr) +{ + #define ADD_HEADER(type, field) \ + case thttp_htype_##type: \ + { \ + if(!self->field) \ + { \ + self->field = (thttp_header_##type##_t*)header; \ + return 0; \ + } \ + 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); + + default: break; + } + + tsk_list_push_back_data(self->headers, (void**)&header); + + 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; +} + +/**@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; +} + +/**@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 += 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; + } + } + } + } + +bail: + 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); +} + +/**@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; + } + } + } + +bail: + 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 */ + 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 */ + 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(); + + if(!thttp_message_serialize(self, output)){ + ret = tsk_strndup(output->data, output->size); + } + + 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; +} + +/**@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; + + 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; +} + + + + + + + + + +//======================================================== +// HTTP message object definition +// + +/**@ingroup thttp_message_group +*/ +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*); +#if defined(__GNUC__) + message->line.response.status_code = (short)va_arg(*app, int); +#else + 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; +} + +/**@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; +} + +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; |