void
ngx_tcp_send(ngx_event_t *wev)
{
ngx_int_t n;
ngx_connection_t *c;
ngx_tcp_session_t *s;
ngx_tcp_core_srv_conf_t *cscf;
c = wev->data;
s = c->data;
if (wev->timedout) {
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out");
c->timedout = 1;
ngx_tcp_close_connection(c);
return;
}
if (s->out.len == 0) {
if (ngx_handle_write_event(c->write, 0) != NGX_OK) {
ngx_tcp_close_connection(c);
}
return;
}
n = c->send(c, s->out.data, s->out.len);
ngx_log_debug1(NGX_LOG_DEBUG_EVENT, wev->log, 0, "nginx tcp send:%d", n);
if (n > 0) {
s->out.len -= n;
if (wev->timer_set) {
ngx_del_timer(wev);
}
if (s->quit) {
ngx_tcp_close_connection(c);
return;
}
return;
}
if (n == NGX_ERROR) {
ngx_tcp_close_connection(c);
return;
}
/* n == NGX_AGAIN */
cscf = ngx_tcp_get_module_srv_conf(s, ngx_tcp_core_module);
ngx_add_timer(c->write, cscf->timeout);
if (ngx_handle_write_event(c->write, 0) != NGX_OK) {
ngx_tcp_close_connection(c);
return;
}
}
void
ngx_tcp_send(ngx_event_t *wev)
{
ngx_int_t rc;
ngx_connection_t *c;
ngx_tcp_session_t *s;
ngx_tcp_core_srv_conf_t *cscf;
c = wev->data;
s = c->data;
if (wev->timedout) {
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "%s|%d|%s|client timed out",__FILE__, __LINE__, __FUNCTION__);
c->timedout = 1;
ngx_tcp_close_connection(c);
return;
}
if (s->output_buffer_chain == NULL) {
return;
}
// rc = s->output_ctx->output_filter(s->output_ctx->filter_ctx,
//s->output_buffer_chain);
rc = s->output_ctx->output_filter(s);
ngx_chain_update_chains(s->output_ctx->pool,
&s->output_ctx->free, &s->output_ctx->busy,
&s->output_buffer_chain, s->output_ctx->tag);
s->output_buffer_chain = NULL;
if (rc == NGX_OK || rc == NGX_DONE) {
if (wev->timer_set) {
ngx_del_timer(wev);
}
return;
}
if (rc == NGX_ERROR) {
ngx_log_error(NGX_LOG_ERR, c->log, 0,
"ngx_tcp_send|client=%V\n", &c->addr_text);
ngx_tcp_close_connection(c);
return;
}
/* rc == NGX_AGAIN */
/*
ngx_log_error(NGX_LOG_INFO, c->log, 0,
"ngx_tcp_send|NGX_AGAIN|client=%V\n", &c->addr_text);
*/
cscf = ngx_tcp_get_module_srv_conf(s, ngx_tcp_core_module);
ngx_add_timer(c->write, cscf->timeout);
if (ngx_handle_write_event(c->write, 0) != NGX_OK) {
ngx_tcp_close_connection(c);
return;
}
}
void
ngx_tcp_init_connection(ngx_connection_t *c)
{
ngx_event_t *rev;
ngx_tcp_log_ctx_t *ctx;
ctx = ngx_palloc(c->pool, sizeof(ngx_tcp_log_ctx_t));
if (ctx == NULL) {
ngx_tcp_close_connection(c);
return;
}
ctx->client = NULL;
ctx->session = NULL;
c->log->connection = c->number;
c->log->handler = ngx_tcp_log_error;
c->log->data = ctx;
c->log->action = "client init tcp connection";
c->log_error = NGX_ERROR_INFO;
rev = c->read;
rev->handler = ngx_tcp_init_session;
c->write->handler = ngx_tcp_empty_handler;
/*
#if (NGX_STAT_STUB)
(void) ngx_atomic_fetch_add(ngx_stat_reading, 1);
#endif
*/
if (rev->ready) {
/* the deferred accept(), rtsig, aio, iocp */
if (ngx_use_accept_mutex) {
ngx_post_event(rev, &ngx_posted_events);
return;
}
ngx_tcp_init_session(rev);
return;
}
ngx_add_timer(rev, c->listening->post_accept_timeout);
if (ngx_handle_read_event(rev, 0) != NGX_OK) {
/*
#if (NGX_STAT_STUB)
(void) ngx_atomic_fetch_add(ngx_stat_reading, -1);
#endif
*/
ngx_tcp_close_connection(c);
return;
}
}
static void
ngx_tcp_ssl_init_connection(ngx_ssl_t *ssl, ngx_connection_t *c)
{
ngx_tcp_session_t *s;
ngx_tcp_core_srv_conf_t *cscf;
if (ngx_ssl_create_connection(ssl, c, NGX_SSL_BUFFER) == NGX_ERROR) {
ngx_tcp_close_connection(c);
return;
}
if (ngx_ssl_handshake(c) == NGX_AGAIN) {
s = c->data;
cscf = ngx_tcp_get_module_srv_conf(s, ngx_tcp_core_module);
ngx_add_timer(c->read, cscf->timeout);
c->ssl->handler = ngx_tcp_ssl_handshake_handler;
return;
}
ngx_tcp_ssl_handshake_handler(c);
}
static void
ngx_tcp_ssl_handshake_handler(ngx_connection_t *c)
{
ngx_tcp_session_t *s;
ngx_tcp_core_srv_conf_t *cscf;
if (c->ssl->handshaked) {
s = c->data;
if (s->starttls) {
cscf = ngx_tcp_get_module_srv_conf(s, ngx_tcp_core_module);
c->read->handler = cscf->protocol->init_protocol;
c->write->handler = ngx_tcp_send;
cscf->protocol->init_protocol(c->read);
return;
}
c->read->ready = 0;
ngx_tcp_init_session(c);
return;
}
ngx_tcp_close_connection(c);
}
void
ngx_tcp_lua_close_session(ngx_tcp_session_t *s)
{
ngx_connection_t *c;
ngx_tcp_cleanup_t *cln;
#if (NGX_STAT_STUB)
if (s->stat_reading) {
(void) ngx_atomic_fetch_add(ngx_stat_reading, -1);
}
if (s->stat_writing) {
(void) ngx_atomic_fetch_add(ngx_stat_writing, -1);
}
#endif
c = s->connection;
ngx_log_debug0(NGX_LOG_DEBUG_TCP, c->log, 0, "ngx_tcp_lua_close_session");
for (cln = s->cleanup; cln; cln = cln->next) {
if (cln->handler) {
cln->handler(cln->data);
cln->handler = NULL;
}
}
ngx_destroy_pool(s->pool);
ngx_tcp_close_connection(c);
return;
}
static void
ngx_tcp_lua_req_keepalive_handler(ngx_tcp_session_t *s)
{
ngx_connection_t *c;
ngx_event_t *rev;
c = s->connection;
rev = c->read;
if (rev->timedout) {
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out");
ngx_tcp_lua_close_session(s);
return;
}
if (c->close) {
ngx_tcp_lua_close_session(s);
return;
}
//have data or close event
//tmp
ngx_tcp_test_reading(s);
if (!c->read->ready) {
if (!c->error) {
ngx_log_error(NGX_LOG_INFO, c->log, 0, "weird, no event call this function");
return;// XXX . add return due to this weird thing really happened.
}
ngx_tcp_lua_close_session(s);
return;
}
c->idle = 0;
ngx_reusable_connection(c, 0);
//s->write_event_handler = ngx_tcp_session_empty_handler;
s->read_event_handler = ngx_tcp_lua_check_client_abort_handler;
if (rev->timer_set) {
ngx_del_timer(rev);
}
if (ngx_tcp_lua_init_light_session(s) != NGX_OK) {
ngx_tcp_close_connection(c);
return;
}
c->log->action = "start new session";
ngx_tcp_lua_req_resume(s);
}
static void
ngx_tcp_ssl_handshake_handler(ngx_connection_t *c)
{
if (c->ssl->handshaked) {
c->read->ready = 0;
ngx_tcp_init_session(c);
return;
}
ngx_tcp_close_connection(c);
}
void
ngx_tcp_finalize_session(ngx_tcp_session_t *s)
{
ngx_connection_t *c;
ngx_tcp_cleanup_t *cln;
c = s->connection;
//ngx_tcp_access_log_handler(s);
ngx_log_debug1(NGX_LOG_DEBUG_EVENT, c->log, 0,
"close tcp session: %d", c->fd);
for (cln = s->cleanup; cln; cln = cln->next) {
if (cln->handler) {
cln->handler(cln->data);
cln->handler = NULL;
}
}
ngx_tcp_close_connection(c);
return;
}
static void
ngx_tcp_init_session(ngx_event_t *rev)
{
ngx_time_t *tp;
ngx_uint_t i;
ngx_connection_t *c;
struct sockaddr_in *sin;
ngx_tcp_port_t *port;
ngx_tcp_in_addr_t *addr;
ngx_tcp_log_ctx_t *ctx;
ngx_tcp_addr_conf_t *addr_conf;
ngx_tcp_session_t *s;
#if (NGX_HAVE_INET6)
struct sockaddr_in6 *sin6;
ngx_tcp_in6_addr_t *addr6;
#endif
ngx_tcp_core_srv_conf_t *cscf;
/*
#if (NGX_STAT_STUB)
(void) ngx_atomic_fetch_add(ngx_stat_reading, -1);
#endif
*/
c = rev->data;
s = c->data;
if (s == NULL) {
s = ngx_pcalloc(c->pool, sizeof(ngx_tcp_session_t));
if (s == NULL) {
ngx_tcp_close_connection(c);
return;
}
c->data = s;
}
ctx = c->log->data;
ctx->session= s;
ctx->client= &c->addr_text;
c->requests++;
s->connection = c;
s->signature = NGX_TCP_MODULE;
/* find the server configuration for the address:port */
port = c->listening->servers;
if (port->naddrs > 1) {
/*
* there are several addresses on this port and one of them
* is an "*:port" wildcard so getsockname() in ngx_tcp_server_addr()
* is required to determine a server address
*/
if (ngx_connection_local_sockaddr(c, NULL, 0) != NGX_OK) {
ngx_tcp_close_connection(c);
return;
}
switch (c->local_sockaddr->sa_family) {
#if (NGX_HAVE_INET6)
case AF_INET6:
sin6 = (struct sockaddr_in6 *) c->local_sockaddr;
addr6 = port->addrs;
/* the last address is "*" */
for (i = 0; i < port->naddrs - 1; i++) {
if (ngx_memcmp(&addr6[i].addr6, &sin6->sin6_addr, 16) == 0) {
break;
}
}
addr_conf = &addr6[i].conf;
break;
#endif
default: /* AF_INET */
sin = (struct sockaddr_in *) c->local_sockaddr;
addr = port->addrs;
/* the last address is "*" */
for (i = 0; i < port->naddrs - 1; i++) {
if (addr[i].addr == sin->sin_addr.s_addr) {
break;
}
}
addr_conf = &addr[i].conf;
break;
}
} else {
switch (c->local_sockaddr->sa_family) {
#if (NGX_HAVE_INET6)
case AF_INET6:
addr6 = port->addrs;
addr_conf = &addr6[0].conf;
break;
#endif
default: /* AF_INET */
addr = port->addrs;
addr_conf = &addr[0].conf;
break;
}
}
if (addr_conf->default_ctx) {
s->main_conf = addr_conf->default_ctx->main_conf;
s->srv_conf = addr_conf->default_ctx->srv_conf;
}
else {
s->main_conf = addr_conf->ctx->main_conf;
s->srv_conf = addr_conf->ctx->srv_conf;
}
s->addr_text = &addr_conf->addr_text;
if (rev->timedout) {
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out");
ngx_tcp_close_connection(c);
return;
}
rev->handler = ngx_tcp_process_session;
//s->read_event_handler = ngx_tcp_block_reading;
/*
#if (NGX_HTTP_SSL)
{
ngx_tcp_ssl_srv_conf_t *sscf;
sscf = ngx_tcp_get_module_srv_conf(r, ngx_tcp_ssl_module);
if (sscf->enable || addr_conf->ssl) {
if (c->ssl == NULL) {
c->log->action = "SSL handshaking";
if (addr_conf->ssl && sscf->ssl.ctx == NULL) {
ngx_log_error(NGX_LOG_ERR, c->log, 0,
"no \"ssl_certificate\" is defined "
"in server listening on SSL port");
ngx_tcp_close_connection(c);
return;
}
if (ngx_ssl_create_connection(&sscf->ssl, c, NGX_SSL_BUFFER)
!= NGX_OK)
{
ngx_tcp_close_connection(c);
return;
}
rev->handler = ngx_tcp_ssl_handshake;
}
r->main_filter_need_in_memory = 1;
}
}
#endif
*/
s->pool = c->pool;
cscf = ngx_tcp_get_module_srv_conf(s, ngx_tcp_core_module);
if (cscf == NULL) {
ngx_tcp_finalize_session(s);
return;
}
s->ctx = ngx_pcalloc(s->pool, sizeof(void *) * ngx_tcp_max_module);
if (s->ctx == NULL) {
ngx_tcp_finalize_session(s);
return;
}
tp = ngx_timeofday();
s->start_sec = tp->sec;
s->start_msec = tp->msec;
s->bytes_read = 0;
s->bytes_write = 0;
ngx_tcp_set_session_socket(s);
/*
#if (NGX_STAT_STUB)
(void) ngx_atomic_fetch_add(ngx_stat_reading, 1);
r->stat_reading = 1;
(void) ngx_atomic_fetch_add(ngx_stat_requests, 1);
#endif
*/
rev->handler(rev);
}
void
ngx_tcp_init_connection(ngx_connection_t *c)
{
ngx_uint_t i;
ngx_tcp_port_t *port;
struct sockaddr *sa;
struct sockaddr_in *sin;
ngx_tcp_log_ctx_t *ctx;
ngx_tcp_in_addr_t *addr;
ngx_tcp_session_t *s;
ngx_tcp_addr_conf_t *addr_conf;
ngx_tcp_core_srv_conf_t *cscf;
#if (NGX_HAVE_INET6)
struct sockaddr_in6 *sin6;
ngx_tcp_in6_addr_t *addr6;
#endif
/* find the server configuration for the address:port */
port = c->listening->servers;
if (port && port->naddrs > 1) {
/*
* There are several addresses on this port and one of them
* is the "*:port" wildcard so getsockname() is needed to determine
* the server address.
*
* AcceptEx() already gave this address.
*/
if (ngx_connection_local_sockaddr(c, NULL, 0) != NGX_OK) {
ngx_tcp_close_connection(c);
return;
}
sa = c->local_sockaddr;
switch (sa->sa_family) {
#if (NGX_HAVE_INET6)
case AF_INET6:
sin6 = (struct sockaddr_in6 *) sa;
addr6 = port->addrs;
/* the last address is "*" */
for (i = 0; i < port->naddrs - 1; i++) {
if (ngx_memcmp(&addr6[i].addr6, &sin6->sin6_addr, 16) == 0) {
break;
}
}
addr_conf = &addr6[i].conf;
break;
#endif
default: /* AF_INET */
sin = (struct sockaddr_in *) sa;
addr = port->addrs;
/* the last address is "*" */
for (i = 0; i < port->naddrs - 1; i++) {
if (addr[i].addr == sin->sin_addr.s_addr) {
break;
}
}
addr_conf = &addr[i].conf;
break;
}
} else {
switch (c->local_sockaddr->sa_family) {
#if (NGX_HAVE_INET6)
case AF_INET6:
addr6 = port->addrs;
addr_conf = &addr6[0].conf;
break;
#endif
case AF_INET:
addr = port->addrs;
addr_conf = &addr[0].conf;
break;
default: /* AF_UNIX */
addr = port->addrs;
addr_conf = &addr[0].conf;
break;
}
}
cscf = ngx_tcp_get_module_srv_conf(addr_conf->ctx,
ngx_tcp_core_module);
s = cscf->protocol->create_session(c);
if (s == NULL) {
ngx_tcp_close_connection(c);
return;
}
s->main_conf = addr_conf->ctx->main_conf;
s->srv_conf = addr_conf->ctx->srv_conf;
s->addr_text = &addr_conf->addr_text;
c->data = s;
s->connection = c;
ngx_log_error(NGX_LOG_INFO, c->log, 0, "*%ui client %V connected to %V",
c->number, &c->addr_text, s->addr_text);
ctx = ngx_palloc(c->pool, sizeof(ngx_tcp_log_ctx_t));
if (ctx == NULL) {
ngx_tcp_close_connection(c);
return;
}
ctx->client = &c->addr_text;
ctx->session = s;
c->log->connection = c->number;
c->log->handler = ngx_tcp_log_error_msg;
c->log->data = ctx;
c->log->action = "sending client greeting line";
c->log_error = NGX_ERROR_INFO;
#if (NGX_TCP_SSL)
{
ngx_tcp_ssl_conf_t *sslcf;
sslcf = ngx_tcp_get_module_srv_conf(s, ngx_tcp_ssl_module);
if (sslcf->enable) {
c->log->action = "SSL handshaking";
ngx_tcp_ssl_init_connection(&sslcf->ssl, c);
return;
}
if (addr_conf->ssl) {
c->log->action = "SSL handshaking";
if (sslcf->ssl.ctx == NULL) {
ngx_log_error(NGX_LOG_ERR, c->log, 0,
"no \"ssl_certificate\" is defined "
"in server listening on SSL port");
ngx_tcp_close_connection(c);
return;
}
ngx_tcp_ssl_init_connection(&sslcf->ssl, c);
return;
}
}
#endif
ngx_tcp_init_session(c);
}
