/*
 * Copyright (C) 2010-2015 Mamadou Diop.
 *	
 * 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_parser_message.c
  * @brief HTTP parser.
  *
  */
 #include "tinyhttp/parsers/thttp_parser_message.h"
 #include "tinyhttp/parsers/thttp_parser_header.h"
 
 #include "tinyhttp/parsers/thttp_parser_url.h"
 
 #include "tsk_debug.h"
 #include "tsk_memory.h"
 
 static void thttp_message_parser_execute(tsk_ragel_state_t *state, thttp_message_t *message, tsk_bool_t extract_content);
 static void thttp_message_parser_init(tsk_ragel_state_t *state);
 static void thttp_message_parser_eoh(tsk_ragel_state_t *state, thttp_message_t *message, tsk_bool_t extract_content);
 
 /***********************************
 *	Ragel state machine.
 */
 %%{
 	machine thttp_machine_parser_message;
 
 	#/* Tag the buffer (start point). */
 	action tag{
 		state->tag_start = p;
 	}
 
 	#/* HTTP method */
 	action parse_method{
 		int len;
 		state->tag_end = p;
 		len = (int)(state->tag_end  - state->tag_start);
 
 		if(message->type == thttp_unknown){
 			message->type = thttp_request;
 			if(!message->line.request.method){
 			#if HAVE_CRT //Debug memory
 		message->line.request.method = calloc(1, len+1);
 	#else
 		message->line.request.method = tsk_calloc(1, len+1);
 	#endif //HAVE_CRT
 				
 				memcpy(message->line.request.method, state->tag_start, len);
 			}
 		}
 		else{
 			state->cs = thttp_machine_parser_message_error;
 		}
 	}
 
 	#/* Request URI parsing */
 	action parse_requesturl{
 		int len;
 		state->tag_end = p;
 		len = (int)(state->tag_end  - state->tag_start);
 		
 		if(!message->line.request.url){
 			message->line.request.url = thttp_url_parse(state->tag_start, (tsk_size_t)len);
 		}
 	}
 
 	#/* Sip Version */
 	action parse_httpversion{
 		int len;
 		state->tag_end = p;
 		len = (int)(state->tag_end  - state->tag_start);
 
 		if(!message->http_version){
 		#if HAVE_CRT //Debug memory
 			message->http_version = calloc(1, len+1);
 		
 	#else
 			message->http_version = tsk_calloc(1, len+1);
 		
 	#endif //HAVE_CRT
 			memcpy(message->http_version, state->tag_start, len);
 		}
 	}
 
 	#/* Status Code */
 	action parse_status_code{
 		int len;
 		state->tag_end = p;
 		len = (int)(state->tag_end  - state->tag_start);
 		
 		if(message->type == thttp_unknown){
 			message->type = thttp_response;
 			message->line.response.status_code = atoi(state->tag_start);
 		}
 		else{
 			state->cs = thttp_machine_parser_message_error;
 		}
 	}
 
 	#/* Reason Phrase */
 	action parse_reason_phrase{
 		int len;
 		state->tag_end = p;
 		len = (int)(state->tag_end  - state->tag_start);
 
 		if(!message->line.response.reason_phrase){
 		#if HAVE_CRT //Debug memory
 			message->line.response.reason_phrase = tsk_calloc(1, len+1);
 		
 	#else
 			message->line.response.reason_phrase = tsk_calloc(1, len+1);
 		
 	#endif //HAVE_CRT
 			memcpy(message->line.response.reason_phrase, state->tag_start, len);
 		}
 	}
 
 	#/* Parse http header */
 	action parse_header{
 		int len;
 		state->tag_end = p;
 		len = (int)(state->tag_end  - state->tag_start);
 		
 		if(thttp_header_parse(state, message)){
 			TSK_DEBUG_ERROR("Failed to parse header - %s", state->tag_start);
 		}
 		else{
 			//TSK_DEBUG_INFO("THTTP_MESSAGE_PARSER::PARSE_HEADER len=%d state=%d", len, state->cs);
 		}
 	}
 
 	#/* Parse http content/body. */
 	#action parse_body
 	#{
 	#	int len;
 	#	state->tag_end = p;
 	#	len = (int)(state->tag_end  - state->tag_start);
 	#	TSK_DEBUG_ERROR("==THTTP_MESSAGE_PARSER::PARSE_BODY==");
 	#}
 
 	#/* End-Of-Headers */
 	action eoh{
 		state->cs = cs;
 		state->p = p;
 		state->pe = pe;
 		state->eof = eof;
 
 		thttp_message_parser_eoh(state, message, extract_content);
 
 		cs = state->cs;
 		p = state->p;
 		pe = state->pe;
 		eof = state->eof;
 	}
 
 	# Includes
 	include thttp_machine_utils "./ragel/thttp_machine_utils.rl";
 	include thttp_machine_message "./ragel/thttp_machine_message.rl";
 	
 	# Entry point
 	main := HTTP_message;
 }%%
 
 
 TSK_RAGEL_DISABLE_WARNINGS_BEGIN()
 /* Regel data */
 %%write data;
 TSK_RAGEL_DISABLE_WARNINGS_END()
 
 /**	Parses raw HTTP buffer.
  *
  * @param state	Ragel state containing the buffer references.
  * @param result @ref thttp_message_t object representing the raw buffer.
  * @param	extract_content	Indicates wheteher to parse the message content or not. If set to true, then
  * only headers will be parsed.
  *
  * @retval	Zero if succeed and non-zero error code otherwise. 
 **/
 int thttp_message_parse(tsk_ragel_state_t *state, thttp_message_t **result, tsk_bool_t extract_content)
 {
 	if(!state || state->pe <= state->p){
 		return -1;
 	}
 
 	if(!*result){
 		*result = thttp_message_create();
 	}
 
 	/* Ragel init */
 	thttp_message_parser_init(state);
 
 	/*
 	*	State mechine execution.
 	*/
 	thttp_message_parser_execute(state, *result, extract_content);
 
 	/* Check result */
 
 	if( state->cs < %%{ write first_final; }%% ){
 		TSK_DEBUG_ERROR("Failed to parse HTTP message.");
 		TSK_OBJECT_SAFE_FREE(*result);
 		return -2;
 	}
 	return 0;
 }
 
 
 static void thttp_message_parser_init(tsk_ragel_state_t *state)
 {
 	int cs = 0;
 
 	/* Regel machine initialization. */
 	%% write init;
 	
 	state->cs = cs;
 }
 
 static void thttp_message_parser_execute(tsk_ragel_state_t *state, thttp_message_t *message, tsk_bool_t extract_content)
 {
 	int cs = state->cs;
 	const char *p = state->p;
 	const char *pe = state->pe;
 	const char *eof = state->eof;
 
 	TSK_RAGEL_DISABLE_WARNINGS_BEGIN()
 	%% write exec;
 	TSK_RAGEL_DISABLE_WARNINGS_END()
 
 	state->cs = cs;
 	state->p = p;
 	state->pe = pe;
 	state->eof = eof;
 }
 
 static void thttp_message_parser_eoh(tsk_ragel_state_t *state, thttp_message_t *message, tsk_bool_t extract_content)
 {
 	int cs = state->cs;
 	const char *p = state->p;
 	const char *pe = state->pe;
 	const char *eof = state->eof;
 
 	if(extract_content && message){
 		uint32_t clen = THTTP_MESSAGE_CONTENT_LENGTH(message);
 		if(clen){
 			if((p + clen)<pe && !message->Content){
 				message->Content = tsk_buffer_create((p+1), clen);
 				p = (p + clen);
 			}
 			else{
 				p = (pe - 1);
 			}
 		}
 	}
 	//%%write eof;
 
 	state->cs = cs;
 	state->p = p;
 	state->pe = pe;
 	state->eof = eof;
 }