src/mail/ngx_mail_pop3_handler.c - nginx-1.7.10

Global variables defined

Functions defined

Source code


  1. /*
  2. * Copyright (C) Igor Sysoev
  3. * Copyright (C) Nginx, Inc.
  4. */


  5. #include <ngx_config.h>
  6. #include <ngx_core.h>
  7. #include <ngx_event.h>
  8. #include <ngx_mail.h>
  9. #include <ngx_mail_pop3_module.h>


  10. static ngx_int_t ngx_mail_pop3_user(ngx_mail_session_t *s, ngx_connection_t *c);
  11. static ngx_int_t ngx_mail_pop3_pass(ngx_mail_session_t *s, ngx_connection_t *c);
  12. static ngx_int_t ngx_mail_pop3_capa(ngx_mail_session_t *s, ngx_connection_t *c,
  13.     ngx_int_t stls);
  14. static ngx_int_t ngx_mail_pop3_stls(ngx_mail_session_t *s, ngx_connection_t *c);
  15. static ngx_int_t ngx_mail_pop3_apop(ngx_mail_session_t *s, ngx_connection_t *c);
  16. static ngx_int_t ngx_mail_pop3_auth(ngx_mail_session_t *s, ngx_connection_t *c);


  17. static u_char  pop3_greeting[] = "+OK POP3 ready" CRLF;
  18. static u_char  pop3_ok[] = "+OK" CRLF;
  19. static u_char  pop3_next[] = "+ " CRLF;
  20. static u_char  pop3_username[] = "+ VXNlcm5hbWU6" CRLF;
  21. static u_char  pop3_password[] = "+ UGFzc3dvcmQ6" CRLF;
  22. static u_char  pop3_invalid_command[] = "-ERR invalid command" CRLF;


  23. void
  24. ngx_mail_pop3_init_session(ngx_mail_session_t *s, ngx_connection_t *c)
  25. {
  26.     u_char                    *p;
  27.     ngx_mail_core_srv_conf_t  *cscf;
  28.     ngx_mail_pop3_srv_conf_t  *pscf;

  29.     pscf = ngx_mail_get_module_srv_conf(s, ngx_mail_pop3_module);
  30.     cscf = ngx_mail_get_module_srv_conf(s, ngx_mail_core_module);

  31.     if (pscf->auth_methods
  32.         & (NGX_MAIL_AUTH_APOP_ENABLED|NGX_MAIL_AUTH_CRAM_MD5_ENABLED))
  33.     {
  34.         if (ngx_mail_salt(s, c, cscf) != NGX_OK) {
  35.             ngx_mail_session_internal_server_error(s);
  36.             return;
  37.         }

  38.         s->out.data = ngx_pnalloc(c->pool, sizeof(pop3_greeting) + s->salt.len);
  39.         if (s->out.data == NULL) {
  40.             ngx_mail_session_internal_server_error(s);
  41.             return;
  42.         }

  43.         p = ngx_cpymem(s->out.data, pop3_greeting, sizeof(pop3_greeting) - 3);
  44.         *p++ = ' ';
  45.         p = ngx_cpymem(p, s->salt.data, s->salt.len);

  46.         s->out.len = p - s->out.data;

  47.     } else {
  48.         ngx_str_set(&s->out, pop3_greeting);
  49.     }

  50.     c->read->handler = ngx_mail_pop3_init_protocol;

  51.     ngx_add_timer(c->read, cscf->timeout);

  52.     if (ngx_handle_read_event(c->read, 0) != NGX_OK) {
  53.         ngx_mail_close_connection(c);
  54.     }

  55.     ngx_mail_send(c->write);
  56. }


  57. void
  58. ngx_mail_pop3_init_protocol(ngx_event_t *rev)
  59. {
  60.     ngx_connection_t    *c;
  61.     ngx_mail_session_t  *s;

  62.     c = rev->data;

  63.     c->log->action = "in auth state";

  64.     if (rev->timedout) {
  65.         ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out");
  66.         c->timedout = 1;
  67.         ngx_mail_close_connection(c);
  68.         return;
  69.     }

  70.     s = c->data;

  71.     if (s->buffer == NULL) {
  72.         if (ngx_array_init(&s->args, c->pool, 2, sizeof(ngx_str_t))
  73.             == NGX_ERROR)
  74.         {
  75.             ngx_mail_session_internal_server_error(s);
  76.             return;
  77.         }

  78.         s->buffer = ngx_create_temp_buf(c->pool, 128);
  79.         if (s->buffer == NULL) {
  80.             ngx_mail_session_internal_server_error(s);
  81.             return;
  82.         }
  83.     }

  84.     s->mail_state = ngx_pop3_start;
  85.     c->read->handler = ngx_mail_pop3_auth_state;

  86.     ngx_mail_pop3_auth_state(rev);
  87. }


  88. void
  89. ngx_mail_pop3_auth_state(ngx_event_t *rev)
  90. {
  91.     ngx_int_t            rc;
  92.     ngx_connection_t    *c;
  93.     ngx_mail_session_t  *s;

  94.     c = rev->data;
  95.     s = c->data;

  96.     ngx_log_debug0(NGX_LOG_DEBUG_MAIL, c->log, 0, "pop3 auth state");

  97.     if (rev->timedout) {
  98.         ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out");
  99.         c->timedout = 1;
  100.         ngx_mail_close_connection(c);
  101.         return;
  102.     }

  103.     if (s->out.len) {
  104.         ngx_log_debug0(NGX_LOG_DEBUG_MAIL, c->log, 0, "pop3 send handler busy");
  105.         s->blocked = 1;
  106.         return;
  107.     }

  108.     s->blocked = 0;

  109.     rc = ngx_mail_read_command(s, c);

  110.     if (rc == NGX_AGAIN || rc == NGX_ERROR) {
  111.         return;
  112.     }

  113.     ngx_str_set(&s->out, pop3_ok);

  114.     if (rc == NGX_OK) {
  115.         switch (s->mail_state) {

  116.         case ngx_pop3_start:

  117.             switch (s->command) {

  118.             case NGX_POP3_USER:
  119.                 rc = ngx_mail_pop3_user(s, c);
  120.                 break;

  121.             case NGX_POP3_CAPA:
  122.                 rc = ngx_mail_pop3_capa(s, c, 1);
  123.                 break;

  124.             case NGX_POP3_APOP:
  125.                 rc = ngx_mail_pop3_apop(s, c);
  126.                 break;

  127.             case NGX_POP3_AUTH:
  128.                 rc = ngx_mail_pop3_auth(s, c);
  129.                 break;

  130.             case NGX_POP3_QUIT:
  131.                 s->quit = 1;
  132.                 break;

  133.             case NGX_POP3_NOOP:
  134.                 break;

  135.             case NGX_POP3_STLS:
  136.                 rc = ngx_mail_pop3_stls(s, c);
  137.                 break;

  138.             default:
  139.                 rc = NGX_MAIL_PARSE_INVALID_COMMAND;
  140.                 break;
  141.             }

  142.             break;

  143.         case ngx_pop3_user:

  144.             switch (s->command) {

  145.             case NGX_POP3_PASS:
  146.                 rc = ngx_mail_pop3_pass(s, c);
  147.                 break;

  148.             case NGX_POP3_CAPA:
  149.                 rc = ngx_mail_pop3_capa(s, c, 0);
  150.                 break;

  151.             case NGX_POP3_QUIT:
  152.                 s->quit = 1;
  153.                 break;

  154.             case NGX_POP3_NOOP:
  155.                 break;

  156.             default:
  157.                 rc = NGX_MAIL_PARSE_INVALID_COMMAND;
  158.                 break;
  159.             }

  160.             break;

  161.         /* suppress warnings */
  162.         case ngx_pop3_passwd:
  163.             break;

  164.         case ngx_pop3_auth_login_username:
  165.             rc = ngx_mail_auth_login_username(s, c, 0);

  166.             ngx_str_set(&s->out, pop3_password);
  167.             s->mail_state = ngx_pop3_auth_login_password;
  168.             break;

  169.         case ngx_pop3_auth_login_password:
  170.             rc = ngx_mail_auth_login_password(s, c);
  171.             break;

  172.         case ngx_pop3_auth_plain:
  173.             rc = ngx_mail_auth_plain(s, c, 0);
  174.             break;

  175.         case ngx_pop3_auth_cram_md5:
  176.             rc = ngx_mail_auth_cram_md5(s, c);
  177.             break;
  178.         }
  179.     }

  180.     switch (rc) {

  181.     case NGX_DONE:
  182.         ngx_mail_auth(s, c);
  183.         return;

  184.     case NGX_ERROR:
  185.         ngx_mail_session_internal_server_error(s);
  186.         return;

  187.     case NGX_MAIL_PARSE_INVALID_COMMAND:
  188.         s->mail_state = ngx_pop3_start;
  189.         s->state = 0;

  190.         ngx_str_set(&s->out, pop3_invalid_command);

  191.         /* fall through */

  192.     case NGX_OK:

  193.         s->args.nelts = 0;
  194.         s->buffer->pos = s->buffer->start;
  195.         s->buffer->last = s->buffer->start;

  196.         if (s->state) {
  197.             s->arg_start = s->buffer->start;
  198.         }

  199.         ngx_mail_send(c->write);
  200.     }
  201. }

  202. static ngx_int_t
  203. ngx_mail_pop3_user(ngx_mail_session_t *s, ngx_connection_t *c)
  204. {
  205.     ngx_str_t  *arg;

  206. #if (NGX_MAIL_SSL)
  207.     if (ngx_mail_starttls_only(s, c)) {
  208.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  209.     }
  210. #endif

  211.     if (s->args.nelts != 1) {
  212.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  213.     }

  214.     arg = s->args.elts;
  215.     s->login.len = arg[0].len;
  216.     s->login.data = ngx_pnalloc(c->pool, s->login.len);
  217.     if (s->login.data == NULL) {
  218.         return NGX_ERROR;
  219.     }

  220.     ngx_memcpy(s->login.data, arg[0].data, s->login.len);

  221.     ngx_log_debug1(NGX_LOG_DEBUG_MAIL, c->log, 0,
  222.                    "pop3 login: \"%V\"", &s->login);

  223.     s->mail_state = ngx_pop3_user;

  224.     return NGX_OK;
  225. }


  226. static ngx_int_t
  227. ngx_mail_pop3_pass(ngx_mail_session_t *s, ngx_connection_t *c)
  228. {
  229.     ngx_str_t  *arg;

  230.     if (s->args.nelts != 1) {
  231.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  232.     }

  233.     arg = s->args.elts;
  234.     s->passwd.len = arg[0].len;
  235.     s->passwd.data = ngx_pnalloc(c->pool, s->passwd.len);
  236.     if (s->passwd.data == NULL) {
  237.         return NGX_ERROR;
  238.     }

  239.     ngx_memcpy(s->passwd.data, arg[0].data, s->passwd.len);

  240. #if (NGX_DEBUG_MAIL_PASSWD)
  241.     ngx_log_debug1(NGX_LOG_DEBUG_MAIL, c->log, 0,
  242.                    "pop3 passwd: \"%V\"", &s->passwd);
  243. #endif

  244.     return NGX_DONE;
  245. }


  246. static ngx_int_t
  247. ngx_mail_pop3_capa(ngx_mail_session_t *s, ngx_connection_t *c, ngx_int_t stls)
  248. {
  249.     ngx_mail_pop3_srv_conf_t  *pscf;

  250.     pscf = ngx_mail_get_module_srv_conf(s, ngx_mail_pop3_module);

  251. #if (NGX_MAIL_SSL)

  252.     if (stls && c->ssl == NULL) {
  253.         ngx_mail_ssl_conf_t  *sslcf;

  254.         sslcf = ngx_mail_get_module_srv_conf(s, ngx_mail_ssl_module);

  255.         if (sslcf->starttls == NGX_MAIL_STARTTLS_ON) {
  256.             s->out = pscf->starttls_capability;
  257.             return NGX_OK;
  258.         }

  259.         if (sslcf->starttls == NGX_MAIL_STARTTLS_ONLY) {
  260.             s->out = pscf->starttls_only_capability;
  261.             return NGX_OK;
  262.         }
  263.     }

  264. #endif

  265.     s->out = pscf->capability;
  266.     return NGX_OK;
  267. }


  268. static ngx_int_t
  269. ngx_mail_pop3_stls(ngx_mail_session_t *s, ngx_connection_t *c)
  270. {
  271. #if (NGX_MAIL_SSL)
  272.     ngx_mail_ssl_conf_t  *sslcf;

  273.     if (c->ssl == NULL) {
  274.         sslcf = ngx_mail_get_module_srv_conf(s, ngx_mail_ssl_module);
  275.         if (sslcf->starttls) {
  276.             c->read->handler = ngx_mail_starttls_handler;
  277.             return NGX_OK;
  278.         }
  279.     }

  280. #endif

  281.     return NGX_MAIL_PARSE_INVALID_COMMAND;
  282. }


  283. static ngx_int_t
  284. ngx_mail_pop3_apop(ngx_mail_session_t *s, ngx_connection_t *c)
  285. {
  286.     ngx_str_t                 *arg;
  287.     ngx_mail_pop3_srv_conf_t  *pscf;

  288. #if (NGX_MAIL_SSL)
  289.     if (ngx_mail_starttls_only(s, c)) {
  290.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  291.     }
  292. #endif

  293.     if (s->args.nelts != 2) {
  294.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  295.     }

  296.     pscf = ngx_mail_get_module_srv_conf(s, ngx_mail_pop3_module);

  297.     if (!(pscf->auth_methods & NGX_MAIL_AUTH_APOP_ENABLED)) {
  298.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  299.     }

  300.     arg = s->args.elts;

  301.     s->login.len = arg[0].len;
  302.     s->login.data = ngx_pnalloc(c->pool, s->login.len);
  303.     if (s->login.data == NULL) {
  304.         return NGX_ERROR;
  305.     }

  306.     ngx_memcpy(s->login.data, arg[0].data, s->login.len);

  307.     s->passwd.len = arg[1].len;
  308.     s->passwd.data = ngx_pnalloc(c->pool, s->passwd.len);
  309.     if (s->passwd.data == NULL) {
  310.         return NGX_ERROR;
  311.     }

  312.     ngx_memcpy(s->passwd.data, arg[1].data, s->passwd.len);

  313.     ngx_log_debug2(NGX_LOG_DEBUG_MAIL, c->log, 0,
  314.                    "pop3 apop: \"%V\" \"%V\"", &s->login, &s->passwd);

  315.     s->auth_method = NGX_MAIL_AUTH_APOP;

  316.     return NGX_DONE;
  317. }


  318. static ngx_int_t
  319. ngx_mail_pop3_auth(ngx_mail_session_t *s, ngx_connection_t *c)
  320. {
  321.     ngx_int_t                  rc;
  322.     ngx_mail_pop3_srv_conf_t  *pscf;

  323. #if (NGX_MAIL_SSL)
  324.     if (ngx_mail_starttls_only(s, c)) {
  325.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  326.     }
  327. #endif

  328.     pscf = ngx_mail_get_module_srv_conf(s, ngx_mail_pop3_module);

  329.     if (s->args.nelts == 0) {
  330.         s->out = pscf->auth_capability;
  331.         s->state = 0;

  332.         return NGX_OK;
  333.     }

  334.     rc = ngx_mail_auth_parse(s, c);

  335.     switch (rc) {

  336.     case NGX_MAIL_AUTH_LOGIN:

  337.         ngx_str_set(&s->out, pop3_username);
  338.         s->mail_state = ngx_pop3_auth_login_username;

  339.         return NGX_OK;

  340.     case NGX_MAIL_AUTH_LOGIN_USERNAME:

  341.         ngx_str_set(&s->out, pop3_password);
  342.         s->mail_state = ngx_pop3_auth_login_password;

  343.         return ngx_mail_auth_login_username(s, c, 1);

  344.     case NGX_MAIL_AUTH_PLAIN:

  345.         ngx_str_set(&s->out, pop3_next);
  346.         s->mail_state = ngx_pop3_auth_plain;

  347.         return NGX_OK;

  348.     case NGX_MAIL_AUTH_CRAM_MD5:

  349.         if (!(pscf->auth_methods & NGX_MAIL_AUTH_CRAM_MD5_ENABLED)) {
  350.             return NGX_MAIL_PARSE_INVALID_COMMAND;
  351.         }

  352.         if (ngx_mail_auth_cram_md5_salt(s, c, "+ ", 2) == NGX_OK) {
  353.             s->mail_state = ngx_pop3_auth_cram_md5;
  354.             return NGX_OK;
  355.         }

  356.         return NGX_ERROR;
  357.     }

  358.     return rc;
  359. }