tango-tfe/common/src/tfe_http.cpp

#include <stdio.h>
#include <stdlib.h>
#include <tfe_http.h>
#include <tfe_plugin.h>

struct tfe_http_half * tfe_http_request_create(int major_version, int method, const char * uri, const char * host)
{
	return NULL;
}

struct tfe_http_half * tfe_http_response_create(int major_version, int resp_code)
{
	return NULL;
}

static const char * __str_std_header_field_map[] =
{
	[TFE_HTTP_UNKNOWN_FIELD] = NULL,
	[TFE_HTTP_HOST] = "Host",
	[TFE_HTTP_REFERER] = "Referer",
	[TFE_HTTP_USER_AGENT] = "User-Agent",
	[TFE_HTTP_COOKIE] = "Cookie",
	[TFE_HTTP_PROXY_AUTHORIZATION] = "Proxy-Authorization",
	[TFE_HTTP_AUTHORIZATION] = "Authorization",
	[TFE_HTTP_LOCATION] = "Location",
	[TFE_HTTP_SERVER] = "Server",
	[TFE_HTTP_ETAG] = "Etag",
	[TFE_HTTP_DATE] = "Date",
	[TFE_HTTP_TRAILER] = "Trailer",
	[TFE_HTTP_TRANSFER_ENCODING] = "Transfer-Encoding",
	[TFE_HTTP_VIA] = "Via",
	[TFE_HTTP_PRAGMA] = "Pragma",
	[TFE_HTTP_CONNECTION] = "Connection",
	[TFE_HTTP_CONT_ENCODING] = "Content-Encoding",
	[TFE_HTTP_CONT_LANGUAGE] = "Content-Language",
	[TFE_HTTP_CONT_LOCATION] = "Content-Location",
	[TFE_HTTP_CONT_RANGE] = "Content-Range",
	[TFE_HTTP_CONT_LENGTH] = "Content-Length",
	[TFE_HTTP_CONT_TYPE] = "Content-Type",
	[TFE_HTTP_CONT_DISPOSITION] = "Content-Disposition",
	[TFE_HTTP_EXPIRES] = "Expires",
	[TFE_HTTP_ACCEPT_ENCODING] = "Accept-Encoding"
};

struct http_field_name * http_field_name_duplicate(const struct http_field_name * orig)
{
	struct http_field_name * __duplicated = ALLOC(struct http_field_name, 1);
	assert(__duplicated != NULL);

	if (orig->field_id == TFE_HTTP_UNKNOWN_FIELD)
	{
		__duplicated->field_id = TFE_HTTP_UNKNOWN_FIELD;
		__duplicated->field_name = tfe_strdup(orig->field_name);
	}
	else
	{
		__duplicated->field_id = orig->field_id;
		__duplicated->field_name = NULL;
	}

	return __duplicated;
}

int http_field_name_compare(const struct http_field_name * lvalue, const struct http_field_name * rvalue)
{
	if (lvalue->field_id != rvalue->field_id)
	{
		return (lvalue->field_id - rvalue->field_id);
	}

	/* unknown field, compare field_name in string */
	if (lvalue->field_id == TFE_HTTP_UNKNOWN_FIELD)
	{
		return strcasecmp(lvalue->field_name, rvalue->field_name);
	}

	/* field_id is equal, but not unknown, hit */
	return 0;
}

struct http_field_name * http_field_construct_from_string(const char * str_field)
{
	return NULL;
}

const char * http_field_to_string(const struct http_field_name * field)
{
	if (field->field_id != TFE_HTTP_UNKNOWN_FIELD) return __str_std_header_field_map[field->field_id];
	return field->field_name;
}

void http_field_destory(struct http_field_name * field)
{
	free(field);
}

struct http_frame_session_ctx
{
	struct http_frame_plugin_status * plugin_status;
	unsigned int nr_plugin_status;

	struct tfe_plugin * calling_plugin;
	struct http_frame_plugin_status * calling_plugin_status;
};

struct http_frame_plugin_status * http_frame_current_plugin_status(struct http_frame_session_ctx * ht_frame)
{
	return ht_frame->calling_plugin_status;
}

struct http_frame_session_ctx * http_frame_raise_session_begin(const struct tfe_stream * stream,
	struct tfe_http_session * ht_session, unsigned int thread_id)
{
	struct http_frame_session_ctx * ht_frame = ALLOC(struct http_frame_session_ctx, 1);
	ht_frame->nr_plugin_status = tfe_plugin_total_counts();
	ht_frame->plugin_status = ALLOC(struct http_frame_plugin_status, ht_frame->nr_plugin_status);

	unsigned int __for_each_iterator = 0;
	unsigned int __plugin_id = 0;

	struct tfe_plugin * plugin_info_iter;
	TFE_PLUGIN_FOREACH(plugin_info_iter, &__for_each_iterator)
	{
		if (plugin_info_iter->on_session_begin == NULL) continue;

		/* Calling ctx, in callback can fetch by calling frame_plugin_status_get_XXX */
		ht_frame->calling_plugin = plugin_info_iter;
		ht_frame->calling_plugin_status = &ht_frame->plugin_status[__plugin_id];

		/* Call session begin */
		void ** calling_pme = &ht_frame->calling_plugin_status->pme;
		plugin_info_iter->on_session_begin(stream, ht_session, thread_id, calling_pme);
	}

	/* Clear calling ctx */
	ht_frame->calling_plugin = NULL;
	ht_frame->calling_plugin_status = NULL;
	return ht_frame;
};

void http_frame_raise_session_end(struct http_frame_session_ctx * ht_frame, struct tfe_stream * stream,
	struct tfe_http_session * ht_session, unsigned int thread_id)
{
	unsigned int __for_each_iterator = 0;
	unsigned int __plugin_id = 0;
	struct tfe_plugin * plugin_info_iter;

	TFE_PLUGIN_FOREACH(plugin_info_iter, &__for_each_iterator)
	{
		if (plugin_info_iter->on_session_end == NULL) continue;

		/* Calling ctx, in callback can fetch by calling frame_plugin_status_get_XXX */
		ht_frame->calling_plugin = plugin_info_iter;
		ht_frame->calling_plugin_status = &ht_frame->plugin_status[__plugin_id];

		/* Call session end */
		void ** calling_pme = &ht_frame->calling_plugin_status->pme;
		plugin_info_iter->on_session_end(stream, ht_session, thread_id, calling_pme);
	}

	ht_frame->calling_plugin = NULL;
	ht_frame->calling_plugin_status = NULL;

	free(ht_frame->plugin_status);
	free(ht_frame);
}

void http_frame_raise_event(struct http_frame_session_ctx * ht_frame,
	const struct tfe_stream * stream, struct tfe_http_session * ht_session, enum tfe_http_event event,
	const unsigned char * data, size_t datalen, unsigned int thread_id)
{
	unsigned int __for_each_iterator = 0;
	unsigned int __plugin_id = 0;

	struct tfe_plugin * plugin_info_iter;
	TFE_PLUGIN_FOREACH(plugin_info_iter, &__for_each_iterator)
	{
		if (plugin_info_iter->on_session_data == NULL) continue;

		/* Calling ctx, in callback can fetch by calling frame_plugin_status_get_XXX */
		ht_frame->calling_plugin = plugin_info_iter;
		ht_frame->calling_plugin_status = &ht_frame->plugin_status[__plugin_id];

		void ** calling_pme = &ht_frame->calling_plugin_status->pme;
		plugin_info_iter->on_session_data(stream, ht_session, event, data, datalen, thread_id, calling_pme);
	}

	ht_frame->calling_plugin = NULL;
	ht_frame->calling_plugin_status = NULL;
}

void http_frame_currect_plugin_detach(struct http_frame_session_ctx * ht_frame)
{
	ht_frame->calling_plugin_status->detached = 1;
}

int http_frame_currect_plugin_preempt(struct http_frame_session_ctx * ht_frame)
{
	for(unsigned int i = 0; i < ht_frame->nr_plugin_status; i++)
	{
		struct http_frame_plugin_status * __plugin_status = &ht_frame->plugin_status[i];
		if (__plugin_status->preempt) return -1;
	}

	assert(ht_frame->calling_plugin_status != NULL);
	ht_frame->calling_plugin_status->preempt = 1;
	return 0;
}