增加HTTP解析层目录结构,增加插件自注册功能。
* 增加HTTP解析层目录结构,集成CMakeLists.txt编译文件; * 调整编译顺序,先编译插件再编译平台; * 增加TFE_PLUGIN_REGISTER宏,在TFE启动时自注册插件; * 修改了stream_open接口,在插件流初始化接口不传入数据。
This commit is contained in:
@@ -7,6 +7,6 @@ set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
|||||||
|
|
||||||
add_subdirectory(vendor)
|
add_subdirectory(vendor)
|
||||||
add_subdirectory(common)
|
add_subdirectory(common)
|
||||||
|
add_subdirectory(plugin)
|
||||||
add_subdirectory(platform)
|
add_subdirectory(platform)
|
||||||
add_subdirectory(cache)
|
add_subdirectory(cache)
|
||||||
add_subdirectory(plugin)
|
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
|
|
||||||
#include <tfe_future.h>
|
#include <tfe_future.h>
|
||||||
#include <tfe_connection.h>
|
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
|
||||||
struct tfe_http_req_spec
|
struct tfe_http_req_spec
|
||||||
@@ -19,17 +18,21 @@ struct tfe_http_resp_spec
|
|||||||
char * content_encoding;
|
char * content_encoding;
|
||||||
};
|
};
|
||||||
|
|
||||||
#define HTTP_REQUEST 1
|
enum tfe_http_direction
|
||||||
#define HTTP_RESPONSE 2
|
{
|
||||||
|
TFE_HTTP_REQUEST = 1,
|
||||||
|
TFE_HTTP_RESPONSE = 2
|
||||||
|
};
|
||||||
|
|
||||||
struct tfe_http_half
|
struct tfe_http_half
|
||||||
{
|
{
|
||||||
int direction; //HTTP_REQUEST or HTTP_RESPONSE
|
enum tfe_http_direction direction;
|
||||||
union
|
union
|
||||||
{
|
{
|
||||||
struct tfe_http_req_spec req_spec;
|
struct tfe_http_req_spec req_spec;
|
||||||
struct tfe_http_resp_spec resp_spec;
|
struct tfe_http_resp_spec resp_spec;
|
||||||
};
|
};
|
||||||
|
|
||||||
size_t field_cnt;
|
size_t field_cnt;
|
||||||
uint64_t cont_len;
|
uint64_t cont_len;
|
||||||
uint64_t cont_range_from;
|
uint64_t cont_range_from;
|
||||||
@@ -119,6 +122,7 @@ enum http_ev_bit_number
|
|||||||
RESP_BODY_CONT_BITNUM,
|
RESP_BODY_CONT_BITNUM,
|
||||||
RESP_BODY_END_BITNUM
|
RESP_BODY_END_BITNUM
|
||||||
};
|
};
|
||||||
|
|
||||||
#define EV_HTTP_REQ_HDR ((uint64_t)1<<REQ_HDR_BITNUM)
|
#define EV_HTTP_REQ_HDR ((uint64_t)1<<REQ_HDR_BITNUM)
|
||||||
#define EV_HTTP_SESSION_BEGIN EV_HTTP_REQ_HDR
|
#define EV_HTTP_SESSION_BEGIN EV_HTTP_REQ_HDR
|
||||||
#define EV_HTTP_REQ_BODY_BEGIN ((uint64_t)1<<REQ_HDR_BITNUM)
|
#define EV_HTTP_REQ_BODY_BEGIN ((uint64_t)1<<REQ_HDR_BITNUM)
|
||||||
|
|||||||
57
common/include/tfe_plugin.h
Normal file
57
common/include/tfe_plugin.h
Normal file
@@ -0,0 +1,57 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <tfe_stream.h>
|
||||||
|
|
||||||
|
struct tfe_proxy;
|
||||||
|
struct tfe_thread_ctx;
|
||||||
|
|
||||||
|
typedef int plugin_init_cb_t(struct tfe_proxy * proxy);
|
||||||
|
typedef void plugin_deinit_cb_t(struct tfe_proxy * proxy);
|
||||||
|
|
||||||
|
//Return 1 for identify as its traffic;
|
||||||
|
//Return 0 for unknown traffic;
|
||||||
|
typedef int stream_open_cb_t(const struct tfe_stream * stream, unsigned int thread_id,
|
||||||
|
enum tfe_conn_dir dir, void ** pme);
|
||||||
|
|
||||||
|
typedef tfe_stream_action stream_data_cb_t(const struct tfe_stream * stream, unsigned int thread_id,
|
||||||
|
enum tfe_conn_dir dir, const unsigned char * data, size_t len, void ** pme);
|
||||||
|
|
||||||
|
typedef void stream_close_cb_t(const struct tfe_stream * stream, unsigned int thread_id,
|
||||||
|
enum tfe_stream_close_reason reason, void ** pme);
|
||||||
|
|
||||||
|
enum tfe_plugin_type
|
||||||
|
{
|
||||||
|
TFE_PLUGIN_TYPE_PLATFORM,
|
||||||
|
TFE_PLUGIN_TYPE_PROTOCOL,
|
||||||
|
TFE_PLUGIN_TYPE_BUSINESS
|
||||||
|
};
|
||||||
|
|
||||||
|
struct tfe_plugin
|
||||||
|
{
|
||||||
|
/* SYMBOL & PROTOCOL */
|
||||||
|
const char * symbol;
|
||||||
|
enum tfe_plugin_type type;
|
||||||
|
enum tfe_app_proto proto;
|
||||||
|
|
||||||
|
/* PLUGIN INIT & DEINIT */
|
||||||
|
plugin_init_cb_t * on_init;
|
||||||
|
plugin_deinit_cb_t * on_deinit;
|
||||||
|
|
||||||
|
/* PLUGIN STREAM ENTRIES */
|
||||||
|
stream_open_cb_t * on_open;
|
||||||
|
stream_data_cb_t * on_data;
|
||||||
|
stream_close_cb_t * on_close;
|
||||||
|
};
|
||||||
|
|
||||||
|
/* Register plugin */
|
||||||
|
int tfe_plugin_register(struct tfe_plugin * plugin);
|
||||||
|
/* Unregister plugin */
|
||||||
|
void tfe_plugin_unregister(struct tfe_plugin * plugin);
|
||||||
|
|
||||||
|
#define TFE_PLUGIN_REGISTER(_symbol, _plugin) \
|
||||||
|
static void __attribute__((constructor, used)) __plugin_loader_##_symbol(); \
|
||||||
|
static void __plugin_loader_##_symbol() \
|
||||||
|
{ \
|
||||||
|
_plugin.symbol = #_symbol; \
|
||||||
|
tfe_plugin_register(&_plugin); \
|
||||||
|
}
|
||||||
6
common/include/tfe_proxy.h
Normal file
6
common/include/tfe_proxy.h
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
struct tfe_proxy;
|
||||||
|
|
||||||
|
const char * tfe_proxy_default_conffile();
|
||||||
|
const char * tfe_proxy_default_logger();
|
||||||
@@ -86,17 +86,6 @@ struct tfe_stream_write_ctx * tfe_stream_write_frag_start(const struct tfe_strea
|
|||||||
int tfe_stream_write_frag(struct tfe_stream_write_ctx * w_ctx, const unsigned char * data, size_t size);
|
int tfe_stream_write_frag(struct tfe_stream_write_ctx * w_ctx, const unsigned char * data, size_t size);
|
||||||
void tfe_stream_write_frag_end(struct tfe_stream_write_ctx * w_ctx);
|
void tfe_stream_write_frag_end(struct tfe_stream_write_ctx * w_ctx);
|
||||||
|
|
||||||
//Return 1 for identify as its traffic;
|
|
||||||
//Return 0 for unknown traffic;
|
|
||||||
typedef enum tfe_stream_action stream_open_cb_t(const struct tfe_stream * stream, unsigned int thread_id,
|
|
||||||
enum tfe_conn_dir dir, const unsigned char * data, size_t len, void ** pme);
|
|
||||||
|
|
||||||
typedef enum tfe_stream_action stream_data_cb_t(const struct tfe_stream * stream, unsigned int thread_id,
|
|
||||||
enum tfe_conn_dir dir, const unsigned char * data, size_t len, void ** pme);
|
|
||||||
|
|
||||||
typedef void stream_close_cb_t(const struct tfe_stream * stream, unsigned int thread_id,
|
|
||||||
enum tfe_stream_close_reason reason, void ** pme);
|
|
||||||
|
|
||||||
void tfe_stream_detach(const struct tfe_stream * stream);
|
void tfe_stream_detach(const struct tfe_stream * stream);
|
||||||
int tfe_stream_preempt(const struct tfe_stream * stream);
|
int tfe_stream_preempt(const struct tfe_stream * stream);
|
||||||
struct promise * tfe_stream_suspend(const struct tfe_stream * stream);
|
struct promise * tfe_stream_suspend(const struct tfe_stream * stream);
|
||||||
@@ -106,11 +95,4 @@ void tfe_stream_resume(struct promise * promise);
|
|||||||
int tfe_stream_shutdown(const struct tfe_stream * stream);
|
int tfe_stream_shutdown(const struct tfe_stream * stream);
|
||||||
int tfe_stream_shutdown_dir(const struct tfe_stream * stream, enum tfe_conn_dir dir);
|
int tfe_stream_shutdown_dir(const struct tfe_stream * stream, enum tfe_conn_dir dir);
|
||||||
|
|
||||||
struct tfe_plugin
|
|
||||||
{
|
|
||||||
char symbol[TFE_SYMBOL_MAX];
|
|
||||||
enum tfe_app_proto proto;
|
|
||||||
stream_open_cb_t * on_open;
|
|
||||||
stream_data_cb_t * on_data;
|
|
||||||
stream_close_cb_t * on_close;
|
|
||||||
};
|
|
||||||
|
|||||||
@@ -25,6 +25,8 @@
|
|||||||
#include <tfe_utils.h>
|
#include <tfe_utils.h>
|
||||||
#include <tfe_future.h>
|
#include <tfe_future.h>
|
||||||
#include <tfe_stream.h>
|
#include <tfe_stream.h>
|
||||||
|
#include <tfe_proxy.h>
|
||||||
|
|
||||||
#include <platform.h>
|
#include <platform.h>
|
||||||
#include <proxy.h>
|
#include <proxy.h>
|
||||||
#include <kni_acceptor.h>
|
#include <kni_acceptor.h>
|
||||||
|
|||||||
@@ -23,6 +23,7 @@
|
|||||||
#include <tfe_stream.h>
|
#include <tfe_stream.h>
|
||||||
#include <tfe_utils.h>
|
#include <tfe_utils.h>
|
||||||
#include <tfe_future.h>
|
#include <tfe_future.h>
|
||||||
|
#include <tfe_plugin.h>
|
||||||
|
|
||||||
#include <platform.h>
|
#include <platform.h>
|
||||||
#include <ssl_stream.h>
|
#include <ssl_stream.h>
|
||||||
@@ -301,8 +302,7 @@ static void __stream_bev_readcb(struct bufferevent * bev, void * arg)
|
|||||||
|
|
||||||
if (_stream->is_plugin_opened == 0)
|
if (_stream->is_plugin_opened == 0)
|
||||||
{
|
{
|
||||||
action_tmp = plugins[i].on_open(&_stream->head, _stream->thread_ref->thread_id,
|
plugins[i].on_open(&_stream->head, _stream->thread_ref->thread_id, dir, &(plug_ctx->pme));
|
||||||
dir, contiguous_data, contigous_len, &(plug_ctx->pme));
|
|
||||||
_stream->is_plugin_opened = 1;
|
_stream->is_plugin_opened = 1;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
|
|||||||
@@ -0,0 +1 @@
|
|||||||
|
add_subdirectory(http)
|
||||||
|
|||||||
4
plugin/protocol/http/CMakeLists.txt
Normal file
4
plugin/protocol/http/CMakeLists.txt
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
|
||||||
|
add_library(http src/http.cpp)
|
||||||
|
target_link_libraries(http common)
|
||||||
|
target_link_libraries(http http-parser-static)
|
||||||
79
plugin/protocol/http/src/http.cpp
Normal file
79
plugin/protocol/http/src/http.cpp
Normal file
@@ -0,0 +1,79 @@
|
|||||||
|
#include <MESA/MESA_list_queue.h>
|
||||||
|
#include <tfe_stream.h>
|
||||||
|
#include <tfe_utils.h>
|
||||||
|
#include <tfe_plugin.h>
|
||||||
|
#include <tfe_http.h>
|
||||||
|
#include <http_parser.h>
|
||||||
|
|
||||||
|
extern "C"
|
||||||
|
{
|
||||||
|
#include <sys/queue.h>
|
||||||
|
}
|
||||||
|
|
||||||
|
struct http_plugin
|
||||||
|
{
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
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;
|
||||||
|
}
|
||||||
|
|
||||||
|
TAILQ_HEAD(http_session_private_list, http_session_private);
|
||||||
|
struct http_session_private
|
||||||
|
{
|
||||||
|
TAILQ_ENTRY(http_session_private) s_next;
|
||||||
|
struct tfe_http_session s_public;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct http_connection_private
|
||||||
|
{
|
||||||
|
struct layer_addr * layer_addr;
|
||||||
|
struct http_session_private_list session_private_list;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct http_half_private
|
||||||
|
{
|
||||||
|
struct tfe_http_half head;
|
||||||
|
};
|
||||||
|
|
||||||
|
int http_connection_entry_open(const struct tfe_stream * stream, unsigned int thread_id,
|
||||||
|
enum tfe_conn_dir dir, void ** pme)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
{
|
||||||
|
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)
|
||||||
|
{
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
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