| |
| /* |
| * Copyright (C) Igor Sysoev |
| */ |
| |
| |
| #include <ngx_config.h> |
| #include <ngx_core.h> |
| #include <ngx_event.h> |
| #include <ngx_imap.h> |
| |
| |
| static void ngx_imap_init_session(ngx_connection_t *c); |
| static void ngx_imap_init_protocol(ngx_event_t *rev); |
| static ngx_int_t ngx_imap_read_command(ngx_imap_session_t *s); |
| static u_char *ngx_imap_log_error(ngx_log_t *log, u_char *buf, size_t len); |
| |
| #if (NGX_IMAP_SSL) |
| static void ngx_imap_ssl_init_connection(ngx_ssl_t *ssl, ngx_connection_t *c); |
| static void ngx_imap_ssl_handshake_handler(ngx_connection_t *c); |
| #endif |
| |
| |
| static ngx_str_t greetings[] = { |
| ngx_string("+OK POP3 ready" CRLF), |
| ngx_string("* OK IMAP4 ready" CRLF) |
| }; |
| |
| static ngx_str_t internal_server_errors[] = { |
| ngx_string("-ERR internal server error" CRLF), |
| ngx_string("* BAD internal server error" CRLF), |
| }; |
| |
| static u_char pop3_ok[] = "+OK" CRLF; |
| static u_char pop3_invalid_command[] = "-ERR invalid command" CRLF; |
| |
| static u_char imap_star[] = "* "; |
| static u_char imap_ok[] = "OK completed" CRLF; |
| static u_char imap_next[] = "+ OK" CRLF; |
| static u_char imap_bye[] = "* BYE" CRLF; |
| static u_char imap_invalid_command[] = "BAD invalid command" CRLF; |
| |
| |
| void |
| ngx_imap_init_connection(ngx_connection_t *c) |
| { |
| in_addr_t in_addr; |
| socklen_t len; |
| ngx_uint_t i; |
| struct sockaddr_in sin; |
| ngx_imap_log_ctx_t *ctx; |
| ngx_imap_in_port_t *imip; |
| ngx_imap_in_addr_t *imia; |
| ngx_imap_session_t *s; |
| #if (NGX_IMAP_SSL) |
| ngx_imap_ssl_conf_t *sslcf; |
| #endif |
| |
| |
| /* find the server configuration for the address:port */ |
| |
| /* AF_INET only */ |
| |
| imip = c->listening->servers; |
| imia = imip->addrs; |
| |
| i = 0; |
| |
| if (imip->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_WIN32) |
| if (c->local_sockaddr) { |
| in_addr = |
| ((struct sockaddr_in *) c->local_sockaddr)->sin_addr.s_addr; |
| |
| } else |
| #endif |
| { |
| len = sizeof(struct sockaddr_in); |
| if (getsockname(c->fd, (struct sockaddr *) &sin, &len) == -1) { |
| ngx_connection_error(c, ngx_socket_errno, |
| "getsockname() failed"); |
| ngx_imap_close_connection(c); |
| return; |
| } |
| |
| in_addr = sin.sin_addr.s_addr; |
| } |
| |
| /* the last address is "*" */ |
| |
| for ( /* void */ ; i < imip->naddrs - 1; i++) { |
| if (in_addr == imia[i].addr) { |
| break; |
| } |
| } |
| } |
| |
| |
| s = ngx_pcalloc(c->pool, sizeof(ngx_imap_session_t)); |
| if (s == NULL) { |
| ngx_imap_close_connection(c); |
| return; |
| } |
| |
| s->main_conf = imia[i].ctx->main_conf; |
| s->srv_conf = imia[i].ctx->srv_conf; |
| |
| s->addr_text = &imia[i].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_imap_log_ctx_t)); |
| if (ctx == NULL) { |
| ngx_imap_close_connection(c); |
| return; |
| } |
| |
| ctx->client = &c->addr_text; |
| ctx->session = s; |
| |
| c->log->connection = c->number; |
| c->log->handler = ngx_imap_log_error; |
| c->log->data = ctx; |
| c->log->action = "sending client greeting line"; |
| |
| c->log_error = NGX_ERROR_INFO; |
| |
| #if (NGX_IMAP_SSL) |
| |
| sslcf = ngx_imap_get_module_srv_conf(s, ngx_imap_ssl_module); |
| |
| if (sslcf->enable) { |
| ngx_imap_ssl_init_connection(&sslcf->ssl, c); |
| return; |
| } |
| |
| #endif |
| |
| ngx_imap_init_session(c); |
| } |
| |
| |
| #if (NGX_IMAP_SSL) |
| |
| static void |
| ngx_imap_starttls_handler(ngx_event_t *rev) |
| { |
| ngx_connection_t *c; |
| ngx_imap_session_t *s; |
| ngx_imap_ssl_conf_t *sslcf; |
| |
| c = rev->data; |
| s = c->data; |
| s->starttls = 1; |
| |
| c->log->action = "in starttls state"; |
| |
| sslcf = ngx_imap_get_module_srv_conf(s, ngx_imap_ssl_module); |
| |
| ngx_imap_ssl_init_connection(&sslcf->ssl, c); |
| } |
| |
| |
| static void |
| ngx_imap_ssl_init_connection(ngx_ssl_t *ssl, ngx_connection_t *c) |
| { |
| ngx_imap_session_t *s; |
| ngx_imap_core_srv_conf_t *cscf; |
| |
| if (ngx_ssl_create_connection(ssl, c, 0) == NGX_ERROR) { |
| ngx_imap_close_connection(c); |
| return; |
| } |
| |
| if (ngx_ssl_handshake(c) == NGX_AGAIN) { |
| |
| s = c->data; |
| |
| cscf = ngx_imap_get_module_srv_conf(s, ngx_imap_core_module); |
| |
| ngx_add_timer(c->read, cscf->timeout); |
| |
| c->ssl->handler = ngx_imap_ssl_handshake_handler; |
| |
| return; |
| } |
| |
| ngx_imap_ssl_handshake_handler(c); |
| } |
| |
| |
| static void |
| ngx_imap_ssl_handshake_handler(ngx_connection_t *c) |
| { |
| ngx_imap_session_t *s; |
| |
| if (c->ssl->handshaked) { |
| |
| s = c->data; |
| |
| if (s->starttls) { |
| c->read->handler = ngx_imap_init_protocol; |
| c->write->handler = ngx_imap_send; |
| |
| ngx_imap_init_protocol(c->read); |
| |
| return; |
| } |
| |
| ngx_imap_init_session(c); |
| return; |
| } |
| |
| ngx_imap_close_connection(c); |
| } |
| |
| #endif |
| |
| |
| static void |
| ngx_imap_init_session(ngx_connection_t *c) |
| { |
| ngx_imap_session_t *s; |
| ngx_imap_core_srv_conf_t *cscf; |
| |
| c->read->handler = ngx_imap_init_protocol; |
| c->write->handler = ngx_imap_send; |
| |
| s = c->data; |
| |
| cscf = ngx_imap_get_module_srv_conf(s, ngx_imap_core_module); |
| |
| s->protocol = cscf->protocol; |
| |
| s->ctx = ngx_pcalloc(c->pool, sizeof(void *) * ngx_imap_max_module); |
| if (s->ctx == NULL) { |
| ngx_imap_session_internal_server_error(s); |
| return; |
| } |
| |
| s->out = greetings[s->protocol]; |
| |
| ngx_add_timer(c->read, cscf->timeout); |
| |
| if (ngx_handle_read_event(c->read, 0) == NGX_ERROR) { |
| ngx_imap_close_connection(c); |
| } |
| |
| ngx_imap_send(c->write); |
| } |
| |
| |
| void |
| ngx_imap_send(ngx_event_t *wev) |
| { |
| ngx_int_t n; |
| ngx_connection_t *c; |
| ngx_imap_session_t *s; |
| ngx_imap_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_imap_close_connection(c); |
| return; |
| } |
| |
| if (s->out.len == 0) { |
| if (ngx_handle_write_event(c->write, 0) == NGX_ERROR) { |
| ngx_imap_close_connection(c); |
| } |
| |
| return; |
| } |
| |
| n = c->send(c, s->out.data, s->out.len); |
| |
| if (n > 0) { |
| s->out.len -= n; |
| |
| if (wev->timer_set) { |
| ngx_del_timer(wev); |
| } |
| |
| if (s->quit) { |
| ngx_imap_close_connection(c); |
| return; |
| } |
| |
| if (s->blocked) { |
| c->read->handler(c->read); |
| } |
| |
| return; |
| } |
| |
| if (n == NGX_ERROR) { |
| ngx_imap_close_connection(c); |
| return; |
| } |
| |
| /* n == NGX_AGAIN */ |
| |
| cscf = ngx_imap_get_module_srv_conf(s, ngx_imap_core_module); |
| |
| ngx_add_timer(c->write, cscf->timeout); |
| |
| if (ngx_handle_write_event(c->write, 0) == NGX_ERROR) { |
| ngx_imap_close_connection(c); |
| return; |
| } |
| } |
| |
| |
| static void |
| ngx_imap_init_protocol(ngx_event_t *rev) |
| { |
| size_t size; |
| ngx_connection_t *c; |
| ngx_imap_session_t *s; |
| ngx_imap_core_srv_conf_t *cscf; |
| |
| c = rev->data; |
| |
| c->log->action = "in auth state"; |
| |
| if (rev->timedout) { |
| ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out"); |
| c->timedout = 1; |
| ngx_imap_close_connection(c); |
| return; |
| } |
| |
| s = c->data; |
| |
| if (s->protocol == NGX_IMAP_POP3_PROTOCOL) { |
| size = 128; |
| s->imap_state = ngx_pop3_start; |
| c->read->handler = ngx_pop3_auth_state; |
| |
| } else { |
| cscf = ngx_imap_get_module_srv_conf(s, ngx_imap_core_module); |
| size = cscf->imap_client_buffer_size; |
| s->imap_state = ngx_imap_start; |
| c->read->handler = ngx_imap_auth_state; |
| } |
| |
| if (s->buffer == NULL) { |
| if (ngx_array_init(&s->args, c->pool, 2, sizeof(ngx_str_t)) |
| == NGX_ERROR) |
| { |
| ngx_imap_session_internal_server_error(s); |
| return; |
| } |
| |
| s->buffer = ngx_create_temp_buf(c->pool, size); |
| if (s->buffer == NULL) { |
| ngx_imap_session_internal_server_error(s); |
| return; |
| } |
| } |
| |
| c->read->handler(rev); |
| } |
| |
| |
| void |
| ngx_imap_auth_state(ngx_event_t *rev) |
| { |
| u_char *text, *last, *p, *dst, *src, *end; |
| ssize_t text_len, last_len; |
| ngx_str_t *arg; |
| ngx_int_t rc; |
| ngx_uint_t tag, i; |
| ngx_connection_t *c; |
| ngx_imap_session_t *s; |
| ngx_imap_core_srv_conf_t *cscf; |
| #if (NGX_IMAP_SSL) |
| ngx_imap_ssl_conf_t *sslcf; |
| #endif |
| |
| c = rev->data; |
| s = c->data; |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_IMAP, c->log, 0, "imap auth state"); |
| |
| if (rev->timedout) { |
| ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out"); |
| c->timedout = 1; |
| ngx_imap_close_connection(c); |
| return; |
| } |
| |
| if (s->out.len) { |
| ngx_log_debug0(NGX_LOG_DEBUG_IMAP, c->log, 0, "imap send handler busy"); |
| s->blocked = 1; |
| return; |
| } |
| |
| s->blocked = 0; |
| |
| rc = ngx_imap_read_command(s); |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_IMAP, c->log, 0, "imap auth: %i", rc); |
| |
| if (rc == NGX_AGAIN || rc == NGX_ERROR) { |
| return; |
| } |
| |
| tag = 1; |
| |
| text = NULL; |
| text_len = 0; |
| |
| last = imap_ok; |
| last_len = sizeof(imap_ok) - 1; |
| |
| if (rc == NGX_OK) { |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_IMAP, c->log, 0, "imap auth command: %i", |
| s->command); |
| |
| if (s->backslash) { |
| |
| arg = s->args.elts; |
| |
| for (i = 0; i < s->args.nelts; i++) { |
| dst = arg[i].data; |
| end = dst + arg[i].len; |
| |
| for (src = dst; src < end; dst++) { |
| *dst = *src; |
| if (*src++ == '\\') { |
| *dst = *src++; |
| } |
| } |
| |
| arg[i].len = dst - arg[i].data; |
| } |
| |
| s->backslash = 0; |
| } |
| |
| switch (s->command) { |
| |
| case NGX_IMAP_LOGIN: |
| |
| #if (NGX_IMAP_SSL) |
| |
| if (c->ssl == NULL) { |
| sslcf = ngx_imap_get_module_srv_conf(s, ngx_imap_ssl_module); |
| |
| if (sslcf->starttls == NGX_IMAP_STARTTLS_ONLY) { |
| rc = NGX_IMAP_PARSE_INVALID_COMMAND; |
| break; |
| } |
| } |
| #endif |
| |
| arg = s->args.elts; |
| |
| if (s->args.nelts == 2 && arg[0].len) { |
| |
| s->login.len = arg[0].len; |
| s->login.data = ngx_palloc(c->pool, s->login.len); |
| if (s->login.data == NULL) { |
| ngx_imap_session_internal_server_error(s); |
| return; |
| } |
| |
| ngx_memcpy(s->login.data, arg[0].data, s->login.len); |
| |
| s->passwd.len = arg[1].len; |
| s->passwd.data = ngx_palloc(c->pool, s->passwd.len); |
| if (s->passwd.data == NULL) { |
| ngx_imap_session_internal_server_error(s); |
| return; |
| } |
| |
| ngx_memcpy(s->passwd.data, arg[1].data, s->passwd.len); |
| |
| #if (NGX_DEBUG_IMAP_PASSWD) |
| ngx_log_debug2(NGX_LOG_DEBUG_IMAP, c->log, 0, |
| "imap login:\"%V\" passwd:\"%V\"", |
| &s->login, &s->passwd); |
| #else |
| ngx_log_debug1(NGX_LOG_DEBUG_IMAP, c->log, 0, |
| "imap login:\"%V\"", &s->login); |
| #endif |
| |
| s->args.nelts = 0; |
| s->buffer->pos = s->buffer->start; |
| s->buffer->last = s->buffer->start; |
| |
| if (rev->timer_set) { |
| ngx_del_timer(rev); |
| } |
| |
| s->login_attempt++; |
| |
| ngx_imap_auth_http_init(s); |
| |
| return; |
| |
| } else { |
| rc = NGX_IMAP_PARSE_INVALID_COMMAND; |
| } |
| |
| break; |
| |
| case NGX_IMAP_CAPABILITY: |
| cscf = ngx_imap_get_module_srv_conf(s, ngx_imap_core_module); |
| |
| #if (NGX_IMAP_SSL) |
| |
| if (c->ssl == NULL) { |
| sslcf = ngx_imap_get_module_srv_conf(s, ngx_imap_ssl_module); |
| |
| if (sslcf->starttls == NGX_IMAP_STARTTLS_ON) { |
| text_len = cscf->imap_starttls_capability.len; |
| text = cscf->imap_starttls_capability.data; |
| break; |
| } |
| |
| if (sslcf->starttls == NGX_IMAP_STARTTLS_ONLY) { |
| text_len = cscf->imap_starttls_only_capability.len; |
| text = cscf->imap_starttls_only_capability.data; |
| break; |
| } |
| } |
| #endif |
| |
| text_len = cscf->imap_capability.len; |
| text = cscf->imap_capability.data; |
| break; |
| |
| case NGX_IMAP_LOGOUT: |
| s->quit = 1; |
| text = imap_bye; |
| text_len = sizeof(imap_bye) - 1; |
| break; |
| |
| case NGX_IMAP_NOOP: |
| break; |
| |
| #if (NGX_IMAP_SSL) |
| |
| case NGX_IMAP_STARTTLS: |
| if (c->ssl == NULL) { |
| sslcf = ngx_imap_get_module_srv_conf(s, ngx_imap_ssl_module); |
| if (sslcf->starttls) { |
| c->read->handler = ngx_imap_starttls_handler; |
| break; |
| } |
| } |
| |
| rc = NGX_IMAP_PARSE_INVALID_COMMAND; |
| break; |
| #endif |
| |
| default: |
| rc = NGX_IMAP_PARSE_INVALID_COMMAND; |
| break; |
| } |
| |
| } else if (rc == NGX_IMAP_NEXT) { |
| last = imap_next; |
| last_len = sizeof(imap_next) - 1; |
| tag = 0; |
| } |
| |
| if (rc == NGX_IMAP_PARSE_INVALID_COMMAND) { |
| last = imap_invalid_command; |
| last_len = sizeof(imap_invalid_command) - 1; |
| } |
| |
| if (tag) { |
| if (s->tag.len == 0) { |
| s->tag.len = sizeof(imap_star) - 1; |
| s->tag.data = (u_char *) imap_star; |
| } |
| |
| if (s->tagged_line.len < s->tag.len + text_len + last_len) { |
| s->tagged_line.len = s->tag.len + text_len + last_len; |
| s->tagged_line.data = ngx_palloc(c->pool, s->tagged_line.len); |
| if (s->tagged_line.data == NULL) { |
| ngx_imap_close_connection(c); |
| return; |
| } |
| } |
| |
| s->out.data = s->tagged_line.data; |
| s->out.len = s->tag.len + text_len + last_len; |
| |
| p = s->out.data; |
| |
| if (text) { |
| p = ngx_cpymem(p, text, text_len); |
| } |
| p = ngx_cpymem(p, s->tag.data, s->tag.len); |
| ngx_memcpy(p, last, last_len); |
| |
| |
| } else { |
| s->out.data = last; |
| s->out.len = last_len; |
| } |
| |
| if (rc != NGX_IMAP_NEXT) { |
| s->args.nelts = 0; |
| s->buffer->pos = s->buffer->start; |
| s->buffer->last = s->buffer->start; |
| s->tag.len = 0; |
| } |
| |
| ngx_imap_send(c->write); |
| } |
| |
| |
| void |
| ngx_pop3_auth_state(ngx_event_t *rev) |
| { |
| u_char *text; |
| ssize_t size; |
| ngx_int_t rc; |
| ngx_str_t *arg; |
| ngx_connection_t *c; |
| ngx_imap_session_t *s; |
| ngx_imap_core_srv_conf_t *cscf; |
| #if (NGX_IMAP_SSL) |
| ngx_imap_ssl_conf_t *sslcf; |
| #endif |
| |
| c = rev->data; |
| s = c->data; |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_IMAP, c->log, 0, "pop3 auth state"); |
| |
| if (rev->timedout) { |
| ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out"); |
| c->timedout = 1; |
| ngx_imap_close_connection(c); |
| return; |
| } |
| |
| if (s->out.len) { |
| ngx_log_debug0(NGX_LOG_DEBUG_IMAP, c->log, 0, "imap send handler busy"); |
| s->blocked = 1; |
| return; |
| } |
| |
| s->blocked = 0; |
| |
| rc = ngx_imap_read_command(s); |
| |
| if (rc == NGX_AGAIN || rc == NGX_ERROR) { |
| return; |
| } |
| |
| text = pop3_ok; |
| size = sizeof(pop3_ok) - 1; |
| |
| if (rc == NGX_OK) { |
| switch (s->imap_state) { |
| |
| case ngx_pop3_start: |
| |
| switch (s->command) { |
| |
| case NGX_POP3_USER: |
| if (s->args.nelts == 1) { |
| s->imap_state = ngx_pop3_user; |
| |
| arg = s->args.elts; |
| s->login.len = arg[0].len; |
| s->login.data = ngx_palloc(c->pool, s->login.len); |
| if (s->login.data == NULL) { |
| ngx_imap_session_internal_server_error(s); |
| return; |
| } |
| |
| ngx_memcpy(s->login.data, arg[0].data, s->login.len); |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_IMAP, c->log, 0, |
| "pop3 login: \"%V\"", &s->login); |
| |
| } else { |
| rc = NGX_IMAP_PARSE_INVALID_COMMAND; |
| } |
| |
| break; |
| |
| case NGX_POP3_CAPA: |
| cscf = ngx_imap_get_module_srv_conf(s, ngx_imap_core_module); |
| |
| #if (NGX_IMAP_SSL) |
| |
| if (c->ssl == NULL) { |
| sslcf = ngx_imap_get_module_srv_conf(s, |
| ngx_imap_ssl_module); |
| if (sslcf->starttls) { |
| size = cscf->pop3_starttls_capability.len; |
| text = cscf->pop3_starttls_capability.data; |
| break; |
| } |
| } |
| #endif |
| |
| size = cscf->pop3_capability.len; |
| text = cscf->pop3_capability.data; |
| break; |
| |
| case NGX_POP3_QUIT: |
| s->quit = 1; |
| break; |
| |
| case NGX_POP3_NOOP: |
| break; |
| |
| #if (NGX_IMAP_SSL) |
| |
| case NGX_POP3_STLS: |
| if (c->ssl == NULL) { |
| sslcf = ngx_imap_get_module_srv_conf(s, |
| ngx_imap_ssl_module); |
| if (sslcf->starttls) { |
| c->read->handler = ngx_imap_starttls_handler; |
| break; |
| } |
| } |
| |
| rc = NGX_IMAP_PARSE_INVALID_COMMAND; |
| break; |
| #endif |
| |
| default: |
| s->imap_state = ngx_pop3_start; |
| rc = NGX_IMAP_PARSE_INVALID_COMMAND; |
| break; |
| } |
| |
| break; |
| |
| case ngx_pop3_user: |
| |
| switch (s->command) { |
| |
| case NGX_POP3_PASS: |
| if (s->args.nelts == 1) { |
| /* STUB */ s->imap_state = ngx_pop3_start; |
| |
| arg = s->args.elts; |
| s->passwd.len = arg[0].len; |
| s->passwd.data = ngx_palloc(c->pool, s->passwd.len); |
| if (s->passwd.data == NULL) { |
| ngx_imap_session_internal_server_error(s); |
| return; |
| } |
| |
| ngx_memcpy(s->passwd.data, arg[0].data, s->passwd.len); |
| |
| #if (NGX_DEBUG_IMAP_PASSWD) |
| ngx_log_debug1(NGX_LOG_DEBUG_IMAP, c->log, 0, |
| "pop3 passwd: \"%V\"", &s->passwd); |
| #endif |
| |
| s->args.nelts = 0; |
| s->buffer->pos = s->buffer->start; |
| s->buffer->last = s->buffer->start; |
| |
| if (rev->timer_set) { |
| ngx_del_timer(rev); |
| } |
| |
| ngx_imap_auth_http_init(s); |
| |
| return; |
| |
| } else { |
| rc = NGX_IMAP_PARSE_INVALID_COMMAND; |
| } |
| |
| break; |
| |
| case NGX_POP3_CAPA: |
| cscf = ngx_imap_get_module_srv_conf(s, ngx_imap_core_module); |
| size = cscf->pop3_capability.len; |
| text = cscf->pop3_capability.data; |
| break; |
| |
| case NGX_POP3_QUIT: |
| s->quit = 1; |
| break; |
| |
| case NGX_POP3_NOOP: |
| break; |
| |
| default: |
| s->imap_state = ngx_pop3_start; |
| rc = NGX_IMAP_PARSE_INVALID_COMMAND; |
| break; |
| } |
| |
| break; |
| |
| /* suppress warinings */ |
| case ngx_pop3_passwd: |
| break; |
| } |
| } |
| |
| if (rc == NGX_IMAP_PARSE_INVALID_COMMAND) { |
| text = pop3_invalid_command; |
| size = sizeof(pop3_invalid_command) - 1; |
| } |
| |
| s->args.nelts = 0; |
| s->buffer->pos = s->buffer->start; |
| s->buffer->last = s->buffer->start; |
| |
| s->out.data = text; |
| s->out.len = size; |
| |
| ngx_imap_send(c->write); |
| } |
| |
| |
| static ngx_int_t |
| ngx_imap_read_command(ngx_imap_session_t *s) |
| { |
| ssize_t n; |
| ngx_int_t rc; |
| |
| n = s->connection->recv(s->connection, s->buffer->last, |
| s->buffer->end - s->buffer->last); |
| |
| if (n == NGX_ERROR || n == 0) { |
| ngx_imap_close_connection(s->connection); |
| return NGX_ERROR; |
| } |
| |
| if (n > 0) { |
| s->buffer->last += n; |
| } |
| |
| if (n == NGX_AGAIN) { |
| if (ngx_handle_read_event(s->connection->read, 0) == NGX_ERROR) { |
| ngx_imap_session_internal_server_error(s); |
| return NGX_ERROR; |
| } |
| |
| return NGX_AGAIN; |
| } |
| |
| if (s->protocol == NGX_IMAP_POP3_PROTOCOL) { |
| rc = ngx_pop3_parse_command(s); |
| } else { |
| rc = ngx_imap_parse_command(s); |
| } |
| |
| if (rc == NGX_AGAIN |
| || rc == NGX_IMAP_NEXT |
| || rc == NGX_IMAP_PARSE_INVALID_COMMAND) |
| { |
| return rc; |
| } |
| |
| if (rc == NGX_ERROR) { |
| ngx_imap_close_connection(s->connection); |
| return NGX_ERROR; |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| void |
| ngx_imap_session_internal_server_error(ngx_imap_session_t *s) |
| { |
| s->out = internal_server_errors[s->protocol]; |
| s->quit = 1; |
| |
| ngx_imap_send(s->connection->write); |
| } |
| |
| |
| void |
| ngx_imap_close_connection(ngx_connection_t *c) |
| { |
| ngx_pool_t *pool; |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_IMAP, c->log, 0, |
| "close imap connection: %d", c->fd); |
| |
| #if (NGX_IMAP_SSL) |
| |
| if (c->ssl) { |
| if (ngx_ssl_shutdown(c) == NGX_AGAIN) { |
| c->ssl->handler = ngx_imap_close_connection; |
| return; |
| } |
| } |
| |
| #endif |
| |
| c->destroyed = 1; |
| |
| pool = c->pool; |
| |
| ngx_close_connection(c); |
| |
| ngx_destroy_pool(pool); |
| } |
| |
| |
| static u_char * |
| ngx_imap_log_error(ngx_log_t *log, u_char *buf, size_t len) |
| { |
| u_char *p; |
| ngx_imap_session_t *s; |
| ngx_imap_log_ctx_t *ctx; |
| |
| if (log->action) { |
| p = ngx_snprintf(buf, len, " while %s", log->action); |
| len -= p - buf; |
| buf = p; |
| } |
| |
| ctx = log->data; |
| |
| p = ngx_snprintf(buf, len, ", client: %V", ctx->client); |
| len -= p - buf; |
| buf = p; |
| |
| s = ctx->session; |
| |
| if (s == NULL) { |
| return p; |
| } |
| |
| p = ngx_snprintf(buf, len, ", server: %V", s->addr_text); |
| len -= p - buf; |
| buf = p; |
| |
| if (s->login.len == 0) { |
| return p; |
| } |
| |
| p = ngx_snprintf(buf, len, ", login: \"%V\"", &s->login); |
| len -= p - buf; |
| buf = p; |
| |
| if (s->proxy == NULL) { |
| return p; |
| } |
| |
| p = ngx_snprintf(buf, len, ", upstream: %V", |
| &s->proxy->upstream.peers->peer[0].name); |
| |
| return p; |
| } |