完成HTTP解析请求侧解析基本流程
* 集成HTTP Parser，解析HTTP请求侧数据； * 增加tfe_http.h中请求方法、应答状态的常量定义 * 变更tfe_http.h中HTTP头部标准定义，去掉非头部字段，增加TFE前缀避免冲突。
This commit is contained in:
Lu Qiuwen
196
plugin/protocol/http/src/http_entry.cpp
Normal file
196
plugin/protocol/http/src/http_entry.cpp
Normal file
@@ -0,0 +1,196 @@
|
||||
#include <MESA/MESA_list_queue.h>
|
||||
#include <tfe_stream.h>
|
||||
#include <tfe_utils.h>
|
||||
#include <tfe_plugin.h>
|
||||
#include <tfe_stream.h>
|
||||
#include <http_parser.h>
|
||||
#include <malloc.h>
|
||||
|
||||
#include <http_common.h>
|
||||
#include <http_half.h>
|
||||
#include <assert.h>
|
||||
|
||||
struct http_plugin __g_http_plugin;
|
||||
struct http_plugin * g_http_plugin = &__g_http_plugin;
|
||||
|
||||
int http_plugin_init(struct tfe_proxy * proxy)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
void http_plugin_deinit(struct tfe_proxy * proxy)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
struct http_session_private * hs_private_create( struct http_connection_private * hc_private,
|
||||
struct http_half_private * hf_private_req, struct http_half_private * hf_private_resp)
|
||||
{
|
||||
struct http_session_private * __hs_private = ALLOC(struct http_session_private, 1);
|
||||
|
||||
/* HS-PUBLIC */
|
||||
__hs_private->hs_public.major_version = 1;
|
||||
__hs_private->hs_public.req = hf_private_req != NULL ? to_hf_public(hf_private_req) : NULL;
|
||||
__hs_private->hs_public.resp = hf_private_req != NULL ? to_hf_public(hf_private_resp) : NULL;
|
||||
__hs_private->hs_public.session_id = hc_private->session_id_counter++;
|
||||
|
||||
/* HS-PRIVATE*/
|
||||
__hs_private->hc_private = hc_private;
|
||||
return __hs_private;
|
||||
}
|
||||
|
||||
void hs_private_destory(struct http_session_private * hs_private)
|
||||
{
|
||||
free(hs_private);
|
||||
}
|
||||
|
||||
static void __SET_PME_HC_PRIVATE(struct http_connection_private * h_conn, void ** pme)
|
||||
{
|
||||
*pme = (void *) h_conn;
|
||||
}
|
||||
|
||||
static struct http_connection_private * __GET_PME_HC_PRIVATE(void ** pme)
|
||||
{
|
||||
return (struct http_connection_private *) (*pme);
|
||||
}
|
||||
|
||||
int http_connection_entry_open(const struct tfe_stream * stream, unsigned int thread_id,
|
||||
enum tfe_conn_dir dir, void ** pme)
|
||||
{
|
||||
struct http_connection_private * ht_conn = ALLOC(struct http_connection_private, 1);
|
||||
TAILQ_INIT(&ht_conn->hs_private_list);
|
||||
__SET_PME_HC_PRIVATE(ht_conn, pme);
|
||||
return 0;
|
||||
}
|
||||
|
||||
enum tfe_stream_action __http_connection_entry_on_request(const struct tfe_stream * stream,
|
||||
struct http_connection_private * hc_private, unsigned int thread_id, const unsigned char * data, size_t len)
|
||||
{
|
||||
struct http_session_private * hs_private = TAILQ_LAST(&hc_private->hs_private_list, hs_private_list);
|
||||
struct http_half_private * hf_private_request = NULL;
|
||||
int ret = 0;
|
||||
|
||||
/* There is no available in session list,
|
||||
* that indicate all HTTP request has corresponding response */
|
||||
if (hs_private == NULL)
|
||||
{
|
||||
goto __new_session;
|
||||
}
|
||||
|
||||
/* The last request is finished, we need to create a new session,
|
||||
* or proceed parse content for last request */
|
||||
hf_private_request = to_hf_request_private(hs_private);
|
||||
if (hf_private_request->finished)
|
||||
{
|
||||
goto __new_session;
|
||||
}
|
||||
|
||||
goto __parse;
|
||||
|
||||
__new_session:
|
||||
hf_private_request = hf_private_create(TFE_HTTP_REQUEST, 1, 0);
|
||||
hs_private = hs_private_create(hc_private, hf_private_request, NULL);
|
||||
TAILQ_INSERT_TAIL(&hc_private->hs_private_list, hs_private, next);
|
||||
|
||||
__parse:
|
||||
ret = hf_private_parse(hf_private_request, data, len);
|
||||
/* Need more data, no boundary touched */
|
||||
if (ret == 0)
|
||||
{
|
||||
return ACTION_DEFER_DATA;
|
||||
}
|
||||
|
||||
/* Some kind of error happened, write log and detach the stream */
|
||||
if (ret == -1)
|
||||
{
|
||||
TFE_STREAM_LOG_ERROR(stream, "Failed at parsing stream as HTTP: %u, %s, %s",
|
||||
hf_private_request->parse_errno, http_errno_name(hf_private_request->parse_errno),
|
||||
http_errno_description(hf_private_request->parse_errno));
|
||||
|
||||
tfe_stream_detach(stream);
|
||||
return ACTION_FORWARD_DATA;
|
||||
}
|
||||
|
||||
assert(ret == 1);
|
||||
|
||||
/* Touch a boundary, such as the end of HTTP headers, bodys, et al.
|
||||
* need to call user's cb */
|
||||
size_t __forward_bytes = hf_private_request->parse_cursor;
|
||||
tfe_stream_action_set_opt(stream, ACTION_OPT_FOWARD_BYTES, &__forward_bytes, sizeof(__forward_bytes));
|
||||
return ACTION_FORWARD_DATA;
|
||||
}
|
||||
|
||||
enum tfe_stream_action __http_connection_entry_on_response(const struct tfe_stream * stream,
|
||||
struct http_connection_private * ht_conn, unsigned int thread_id, const unsigned char * data, size_t len)
|
||||
{
|
||||
return ACTION_FORWARD_DATA;
|
||||
}
|
||||
|
||||
int __http_connection_identify(const struct tfe_stream * stream,
|
||||
struct http_connection_private * ht_conn, const unsigned char * data, size_t len)
|
||||
{
|
||||
struct http_half_private * hf_private = hf_private_create(TFE_HTTP_RESPONSE, 1, 0);
|
||||
int ret = hf_private_parse(hf_private, data, len);
|
||||
hf_private_destory(hf_private);
|
||||
return ret;
|
||||
}
|
||||
|
||||
#define HTTP_INDENTIFY_LENGTH 8
|
||||
|
||||
enum tfe_stream_action http_connection_entry_data(const struct tfe_stream * stream, unsigned int thread_id,
|
||||
enum tfe_conn_dir dir, const unsigned char * data, size_t len, void ** pme)
|
||||
{
|
||||
struct http_connection_private * ht_conn = __GET_PME_HC_PRIVATE(pme);
|
||||
|
||||
if (ht_conn->is_preempted == 0)
|
||||
{
|
||||
/* If the server push response before client send request, this must not be HTTP, detach the stream */
|
||||
if (dir == CONN_DIR_UPSTREAM)
|
||||
goto __detach;
|
||||
|
||||
/* Protocol Identification, we need 8 bytes at least to tell it is HTTP or not */
|
||||
if (len < HTTP_INDENTIFY_LENGTH)
|
||||
{
|
||||
static const unsigned int __defer_bytes = HTTP_INDENTIFY_LENGTH;
|
||||
tfe_stream_action_set_opt(stream, ACTION_OPT_DEFER_BYTES, (void *) &__defer_bytes, sizeof(__defer_bytes));
|
||||
return ACTION_DEFER_DATA;
|
||||
}
|
||||
|
||||
/* Now, we want to identify this stream */
|
||||
int ret = __http_connection_identify(stream, ht_conn, data, len);
|
||||
if (ret != 0) goto __detach;
|
||||
|
||||
/* This is HTTP, try to preempt the stream
|
||||
* It may be failed because other plugin preempted before us */
|
||||
ret = tfe_stream_preempt(stream);
|
||||
if (ret != 0) goto __detach;
|
||||
}
|
||||
|
||||
/* This stream has been preempt, this plugin try to parse it */
|
||||
return (dir == CONN_DIR_DOWNSTREAM) ? __http_connection_entry_on_request(stream, ht_conn, thread_id,
|
||||
data, len) : __http_connection_entry_on_response(stream, ht_conn, thread_id, data, len);
|
||||
|
||||
__detach:
|
||||
tfe_stream_detach(stream);
|
||||
return ACTION_FORWARD_DATA;
|
||||
}
|
||||
|
||||
void http_connection_entry_close(const struct tfe_stream * stream, unsigned int thread_id,
|
||||
enum tfe_stream_close_reason reason, void ** pme)
|
||||
{
|
||||
struct http_connection_private * __ht_conn = __GET_PME_HC_PRIVATE(pme);
|
||||
}
|
||||
|
||||
static struct tfe_plugin __http_plugin_info =
|
||||
{
|
||||
.symbol = "HTTP",
|
||||
.type = TFE_PLUGIN_TYPE_PROTOCOL,
|
||||
.proto = APP_PROTO_HTTP1,
|
||||
.on_init = http_plugin_init,
|
||||
.on_deinit = http_plugin_deinit,
|
||||
.on_open = http_connection_entry_open,
|
||||
.on_data = http_connection_entry_data,
|
||||
.on_close = http_connection_entry_close
|
||||
};
|
||||
|
||||
TFE_PLUGIN_REGISTER(HTTP, __http_plugin_info)
|
||||
Reference in New Issue
Block a user