ngx_http_lua_socket_tcp.h

/*
 * Copyright (C) Yichun Zhang (agentzh)
 */


#ifndef_NGX_HTTP_LUA_SOCKET_TCP_H_INCLUDED_
#define_NGX_HTTP_LUA_SOCKET_TCP_H_INCLUDED_


#include"ngx_http_lua_common.h"


#defineNGX_HTTP_LUA_SOCKET_FT_ERROR 0x0001
#defineNGX_HTTP_LUA_SOCKET_FT_TIMEOUT 0x0002
#defineNGX_HTTP_LUA_SOCKET_FT_CLOSED 0x0004
#defineNGX_HTTP_LUA_SOCKET_FT_RESOLVER 0x0008
#defineNGX_HTTP_LUA_SOCKET_FT_BUFTOOSMALL 0x0010
#defineNGX_HTTP_LUA_SOCKET_FT_NOMEM 0x0020
#defineNGX_HTTP_LUA_SOCKET_FT_PARTIALWRITE 0x0040
#defineNGX_HTTP_LUA_SOCKET_FT_CLIENTABORT 0x0080
#defineNGX_HTTP_LUA_SOCKET_FT_SSL 0x0100


typedefstructngx_http_lua_socket_tcp_upstream_s
ngx_http_lua_socket_tcp_upstream_t;


typedefstructngx_http_lua_socket_udata_queue_s
ngx_http_lua_socket_udata_queue_t;


typedef
int (*ngx_http_lua_socket_tcp_retval_handler)(ngx_http_request_t*r,
ngx_http_lua_socket_tcp_upstream_t*u, lua_State*L);


typedefvoid (*ngx_http_lua_socket_tcp_upstream_handler_pt)
 (ngx_http_request_t*r, ngx_http_lua_socket_tcp_upstream_t*u);


typedefstruct {
ngx_event_tevent;
ngx_queue_tqueue;
ngx_str_thost;
ngx_http_cleanup_pt*cleanup;
ngx_http_lua_socket_tcp_upstream_t*u;
in_port_tport;
} ngx_http_lua_socket_tcp_conn_op_ctx_t;


#definengx_http_lua_socket_tcp_free_conn_op_ctx(conn_op_ctx) \
 ngx_free(conn_op_ctx->host.data); \
 ngx_free(conn_op_ctx)


typedefstruct {
lua_State*lua_vm;

ngx_int_tsize;
ngx_queue_tcache_connect_op;
ngx_queue_twait_connect_op;

/* connections == active connections + pending connect operations,
 * while active connections == out-of-pool reused connections
 * + in-pool connections */
ngx_int_tconnections;

/* queues of ngx_http_lua_socket_pool_item_t: */
ngx_queue_tcache;
ngx_queue_tfree;

ngx_int_tbacklog;

u_charkey[1];

} ngx_http_lua_socket_pool_t;


structngx_http_lua_socket_tcp_upstream_s {
ngx_http_lua_socket_tcp_retval_handlerread_prepare_retvals;
ngx_http_lua_socket_tcp_retval_handlerwrite_prepare_retvals;
ngx_http_lua_socket_tcp_upstream_handler_ptread_event_handler;
ngx_http_lua_socket_tcp_upstream_handler_ptwrite_event_handler;

ngx_http_lua_socket_udata_queue_t*udata_queue;

ngx_http_lua_socket_pool_t*socket_pool;

ngx_http_lua_loc_conf_t*conf;
ngx_http_cleanup_pt*cleanup;
ngx_http_request_t*request;
ngx_peer_connection_tpeer;

ngx_msec_tread_timeout;
ngx_msec_tsend_timeout;
ngx_msec_tconnect_timeout;

ngx_http_upstream_resolved_t*resolved;

ngx_chain_t*bufs_in; /* input data buffers */
ngx_chain_t*buf_in; /* last input data buffer */
ngx_buf_tbuffer; /* receive buffer */

size_tlength;
size_trest;

ngx_err_tsocket_errno;

ngx_int_t (*input_filter)(void*data, ssize_tbytes);
void*input_filter_ctx;

size_trequest_len;
ngx_chain_t*request_bufs;

ngx_http_lua_co_ctx_t*read_co_ctx;
ngx_http_lua_co_ctx_t*write_co_ctx;

ngx_uint_treused;

#if (NGX_HTTP_SSL)
ngx_str_tssl_name;
ngx_ssl_session_t*ssl_session_ret;
constchar*error_ret;
intopenssl_error_code_ret;
#endif

ngx_chain_t*busy_bufs;

unsignedft_type:16;
unsignedno_close:1;
unsignedconn_waiting:1;
unsignedread_waiting:1;
unsignedwrite_waiting:1;
unsignedeof:1;
unsignedbody_downstream:1;
unsignedraw_downstream:1;
unsignedread_closed:1;
unsignedwrite_closed:1;
unsignedconn_closed:1;
#if (NGX_HTTP_SSL)
unsignedssl_verify:1;
unsignedssl_session_reuse:1;
#endif
};


typedefstructngx_http_lua_dfa_edge_sngx_http_lua_dfa_edge_t;


structngx_http_lua_dfa_edge_s {
u_charchr;
intnew_state;
ngx_http_lua_dfa_edge_t*next;
};


typedefstruct {
ngx_http_lua_socket_tcp_upstream_t*upstream;

ngx_str_tpattern;
intstate;
ngx_http_lua_dfa_edge_t**recovering;

unsignedinclusive:1;
} ngx_http_lua_socket_compiled_pattern_t;


typedefstruct {
ngx_http_lua_socket_pool_t*socket_pool;

ngx_queue_tqueue;
ngx_connection_t*connection;

socklen_tsocklen;
structsockaddr_storagesockaddr;

ngx_uint_treused;

ngx_http_lua_socket_udata_queue_t*udata_queue;
} ngx_http_lua_socket_pool_item_t;


structngx_http_lua_socket_udata_queue_s {
ngx_pool_t*pool;
ngx_queue_tqueue;
ngx_queue_tfree;
intlen;
intcapacity;
};


typedefstruct {
ngx_queue_tqueue;
uint64_tkey;
uint64_tvalue;
} ngx_http_lua_socket_node_t;


voidngx_http_lua_inject_socket_tcp_api(ngx_log_t*log, lua_State*L);
voidngx_http_lua_inject_req_socket_api(lua_State*L);
voidngx_http_lua_cleanup_conn_pools(lua_State*L);


#endif/* _NGX_HTTP_LUA_SOCKET_TCP_H_INCLUDED_ */

/* vi:set ft=c ts=4 sw=4 et fdm=marker: */