变更tfe_stream.h中的部分公用结构体定义。

This commit is contained in:
Lu Qiuwen
2018-09-03 10:30:47 +08:00
parent a473aa3c3c
commit 8ddb847224
6 changed files with 24 additions and 24 deletions

View File

@@ -5,12 +5,12 @@
#include <tfe_types.h> #include <tfe_types.h>
#include <tfe_utils.h> #include <tfe_utils.h>
enum tfe_session_proto enum tfe_stream_proto
{ {
SESSION_PROTO_PLAIN = 0, STREAM_PROTO_PLAIN = 0,
SESSION_PROTO_SSL, STREAM_PROTO_SSL,
SESSION_PROTO_QUIC, STREAM_PROTO_QUIC,
SESSION_PROTO_SPDY STREAM_PROTO_SPDY
}; };
enum tfe_app_proto enum tfe_app_proto
@@ -37,13 +37,13 @@ enum tfe_conn_status
/* single dst or src socket bufferevent descriptor */ /* single dst or src socket bufferevent descriptor */
struct tfe_conn struct tfe_conn
{ {
struct layer_addr addr;
enum tfe_conn_status status; enum tfe_conn_status status;
struct bufferevent * bev;
}; };
struct tfe_stream struct tfe_stream
{ {
struct layer_addr addr;
enum tfe_stream_proto proto;
struct tfe_conn upstream; struct tfe_conn upstream;
struct tfe_conn downstream; struct tfe_conn downstream;
}; };
@@ -88,10 +88,10 @@ void tfe_stream_write_frag_end(struct tfe_stream_write_ctx * w_ctx);
//Return 1 for identify as its traffic; //Return 1 for identify as its traffic;
//Return 0 for unknown traffic; //Return 0 for unknown traffic;
typedef tfe_stream_action stream_open_cb_t(const struct tfe_stream * stream, unsigned int thread_id, 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); enum tfe_conn_dir dir, const unsigned char * data, size_t len, void ** pme);
typedef tfe_stream_action stream_data_cb_t(const struct tfe_stream * stream, unsigned int thread_id, 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); 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, typedef void stream_close_cb_t(const struct tfe_stream * stream, unsigned int thread_id,
@@ -100,11 +100,11 @@ typedef void stream_close_cb_t(const struct tfe_stream * stream, unsigned int th
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);
void tfe_stream_resume(struct promisc * promisc); void tfe_stream_resume(struct promise * promise);
//close both sides of the stream. //close both sides of the stream.
int stream_shutdown(const struct tfe_stream * stream); int tfe_stream_shutdown(const struct tfe_stream * stream);
int 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 struct tfe_plugin
{ {

View File

@@ -59,7 +59,7 @@ struct tfe_stream_private
struct tfe_proxy * proxy_ref; struct tfe_proxy * proxy_ref;
struct tfe_thread_ctx * thread_ref; struct tfe_thread_ctx * thread_ref;
enum tfe_session_proto session_type; enum tfe_stream_proto session_type;
struct tfe_conn_private * conn_upstream; struct tfe_conn_private * conn_upstream;
struct tfe_conn_private * conn_downstream; struct tfe_conn_private * conn_downstream;

View File

@@ -38,7 +38,7 @@ struct tfe_proxy_accept_para
/* Session Type */ /* Session Type */
bool is_set_session_type; bool is_set_session_type;
enum tfe_session_proto session_type; enum tfe_stream_proto session_type;
bool passthrough; bool passthrough;
}; };

View File

@@ -126,14 +126,14 @@ void __kni_event_cb(evutil_socket_t fd, short what, void * user)
assert(__tlv_info->type == KNI_TLV_TYPE_PRO); assert(__tlv_info->type == KNI_TLV_TYPE_PRO);
enum tfe_session_proto __session_proto; enum tfe_stream_proto __session_proto;
if (__tlv_info->value == KNI_TLV_VALUE_HTTP) if (__tlv_info->value == KNI_TLV_VALUE_HTTP)
{ {
__session_proto = SESSION_PROTO_PLAIN; __session_proto = STREAM_PROTO_PLAIN;
} }
else if (__tlv_info->value == KNI_TLV_VALUE_SSL) else if (__tlv_info->value == KNI_TLV_VALUE_SSL)
{ {
__session_proto = SESSION_PROTO_SSL; __session_proto = STREAM_PROTO_SSL;
} }
else else
{ {

View File

@@ -73,7 +73,7 @@ int tfe_proxy_fds_accept(struct tfe_proxy * ctx, const struct tfe_proxy_accept_p
if (para->passthrough || ctx->tcp_all_passthrough) if (para->passthrough || ctx->tcp_all_passthrough)
{ {
bool __true = true; bool __true = true;
enum tfe_session_proto __session_type = SESSION_PROTO_PLAIN; enum tfe_stream_proto __session_type = STREAM_PROTO_PLAIN;
tfe_stream_option_set(stream, TFE_STREAM_OPT_PASSTHROUGH, &__true, sizeof(__true)); tfe_stream_option_set(stream, TFE_STREAM_OPT_PASSTHROUGH, &__true, sizeof(__true));
tfe_stream_option_set(stream, TFE_STREAM_OPT_SESSION_TYPE, &__session_type, sizeof(__session_type)); tfe_stream_option_set(stream, TFE_STREAM_OPT_SESSION_TYPE, &__session_type, sizeof(__session_type));

View File

@@ -60,7 +60,7 @@ static inline enum tfe_conn_dir __BEV_DIR(struct tfe_stream_private * _stream, s
static inline bool __IS_SSL(struct tfe_stream_private * _stream) static inline bool __IS_SSL(struct tfe_stream_private * _stream)
{ {
return (_stream->session_type == SESSION_PROTO_SSL); return (_stream->session_type == STREAM_PROTO_SSL);
} }
void tfe_stream_detach(const struct tfe_stream * stream) void tfe_stream_detach(const struct tfe_stream * stream)
@@ -601,7 +601,7 @@ void tfe_stream_init_by_fds(struct tfe_stream * stream, evutil_socket_t fd_downs
evutil_make_socket_nonblocking(fd_downstream); evutil_make_socket_nonblocking(fd_downstream);
evutil_make_socket_nonblocking(fd_upstream); evutil_make_socket_nonblocking(fd_upstream);
if (_stream->session_type == SESSION_PROTO_PLAIN) if (_stream->session_type == STREAM_PROTO_PLAIN)
{ {
_stream->conn_downstream = __conn_private_create(_stream, fd_downstream); _stream->conn_downstream = __conn_private_create(_stream, fd_downstream);
_stream->conn_upstream = __conn_private_create(_stream, fd_upstream); _stream->conn_upstream = __conn_private_create(_stream, fd_upstream);
@@ -613,7 +613,7 @@ void tfe_stream_init_by_fds(struct tfe_stream * stream, evutil_socket_t fd_downs
__conn_private_enable(_stream->conn_upstream); __conn_private_enable(_stream->conn_upstream);
} }
if (_stream->session_type == SESSION_PROTO_SSL) if (_stream->session_type == STREAM_PROTO_SSL)
{ {
_stream->ssl_mgr = _stream->proxy_ref->ssl_mgr_handler; _stream->ssl_mgr = _stream->proxy_ref->ssl_mgr_handler;
@@ -635,8 +635,8 @@ int tfe_stream_option_set(struct tfe_stream * stream, enum tfe_stream_option opt
if (opt == TFE_STREAM_OPT_SESSION_TYPE) if (opt == TFE_STREAM_OPT_SESSION_TYPE)
{ {
assert(sz_arg == sizeof(enum tfe_session_proto)); assert(sz_arg == sizeof(enum tfe_stream_proto));
_stream->session_type = *(enum tfe_session_proto *)arg; _stream->session_type = *(enum tfe_stream_proto *)arg;
} }
else if (opt == TFE_STREAM_OPT_PASSTHROUGH) else if (opt == TFE_STREAM_OPT_PASSTHROUGH)
{ {