src/mail/ngx_mail_pop3_handler.c - nginx-1.31.3 nginx/ @ 42f8df65b

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.         return;
  55.     }

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


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

  63.     c = rev->data;

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

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

  71.     s = c->data;

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

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

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

  87.     ngx_mail_pop3_auth_state(rev);
  88. }


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

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

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

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

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

  107.         if (ngx_handle_read_event(c->read, 0) != NGX_OK) {
  108.             ngx_mail_close_connection(c);
  109.             return;
  110.         }

  111.         return;
  112.     }

  113.     s->blocked = 0;

  114.     rc = ngx_mail_read_command(s, c);

  115.     if (rc == NGX_AGAIN) {
  116.         if (ngx_handle_read_event(c->read, 0) != NGX_OK) {
  117.             ngx_mail_session_internal_server_error(s);
  118.             return;
  119.         }

  120.         return;
  121.     }

  122.     if (rc == NGX_ERROR) {
  123.         return;
  124.     }

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

  126.     if (rc == NGX_OK) {
  127.         switch (s->mail_state) {

  128.         case ngx_pop3_start:

  129.             switch (s->command) {

  130.             case NGX_POP3_USER:
  131.                 rc = ngx_mail_pop3_user(s, c);
  132.                 break;

  133.             case NGX_POP3_CAPA:
  134.                 rc = ngx_mail_pop3_capa(s, c, 1);
  135.                 break;

  136.             case NGX_POP3_APOP:
  137.                 rc = ngx_mail_pop3_apop(s, c);
  138.                 break;

  139.             case NGX_POP3_AUTH:
  140.                 rc = ngx_mail_pop3_auth(s, c);
  141.                 break;

  142.             case NGX_POP3_QUIT:
  143.                 s->quit = 1;
  144.                 break;

  145.             case NGX_POP3_NOOP:
  146.                 break;

  147.             case NGX_POP3_STLS:
  148.                 rc = ngx_mail_pop3_stls(s, c);
  149.                 break;

  150.             default:
  151.                 rc = NGX_MAIL_PARSE_INVALID_COMMAND;
  152.                 break;
  153.             }

  154.             break;

  155.         case ngx_pop3_user:

  156.             switch (s->command) {

  157.             case NGX_POP3_PASS:
  158.                 rc = ngx_mail_pop3_pass(s, c);
  159.                 break;

  160.             case NGX_POP3_CAPA:
  161.                 rc = ngx_mail_pop3_capa(s, c, 0);
  162.                 break;

  163.             case NGX_POP3_QUIT:
  164.                 s->quit = 1;
  165.                 break;

  166.             case NGX_POP3_NOOP:
  167.                 break;

  168.             default:
  169.                 rc = NGX_MAIL_PARSE_INVALID_COMMAND;
  170.                 break;
  171.             }

  172.             break;

  173.         /* suppress warnings */
  174.         case ngx_pop3_passwd:
  175.             break;

  176.         case ngx_pop3_auth_login_username:
  177.             rc = ngx_mail_auth_login_username(s, c, 0);

  178.             ngx_str_set(&s->out, pop3_password);
  179.             s->mail_state = ngx_pop3_auth_login_password;
  180.             break;

  181.         case ngx_pop3_auth_login_password:
  182.             rc = ngx_mail_auth_login_password(s, c);
  183.             break;

  184.         case ngx_pop3_auth_plain:
  185.             rc = ngx_mail_auth_plain(s, c, 0);
  186.             break;

  187.         case ngx_pop3_auth_cram_md5:
  188.             rc = ngx_mail_auth_cram_md5(s, c);
  189.             break;

  190.         case ngx_pop3_auth_external:
  191.             rc = ngx_mail_auth_external(s, c, 0);
  192.             break;
  193.         }
  194.     }

  195.     if (s->buffer->pos < s->buffer->last) {
  196.         s->blocked = 1;
  197.     }

  198.     switch (rc) {

  199.     case NGX_DONE:
  200.         ngx_mail_auth(s, c);
  201.         return;

  202.     case NGX_ERROR:
  203.         ngx_mail_session_internal_server_error(s);
  204.         return;

  205.     case NGX_MAIL_PARSE_INVALID_COMMAND:
  206.         s->mail_state = ngx_pop3_start;
  207.         s->state = 0;

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

  209.         /* fall through */

  210.     case NGX_OK:

  211.         s->args.nelts = 0;

  212.         if (s->buffer->pos == s->buffer->last) {
  213.             s->buffer->pos = s->buffer->start;
  214.             s->buffer->last = s->buffer->start;
  215.         }

  216.         if (s->state) {
  217.             s->arg_start = s->buffer->pos;
  218.         }

  219.         if (ngx_handle_read_event(c->read, 0) != NGX_OK) {
  220.             ngx_mail_session_internal_server_error(s);
  221.             return;
  222.         }

  223.         ngx_mail_send(c->write);
  224.     }
  225. }

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

  230. #if (NGX_MAIL_SSL)
  231.     if (ngx_mail_starttls_only(s, c)) {
  232.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  233.     }
  234. #endif

  235.     if (s->args.nelts != 1) {
  236.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  237.     }

  238.     arg = s->args.elts;
  239.     s->login.len = arg[0].len;
  240.     s->login.data = ngx_pnalloc(c->pool, s->login.len);
  241.     if (s->login.data == NULL) {
  242.         return NGX_ERROR;
  243.     }

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

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

  247.     s->mail_state = ngx_pop3_user;

  248.     return NGX_OK;
  249. }


  250. static ngx_int_t
  251. ngx_mail_pop3_pass(ngx_mail_session_t *s, ngx_connection_t *c)
  252. {
  253.     ngx_str_t  *arg;

  254.     if (s->args.nelts != 1) {
  255.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  256.     }

  257.     arg = s->args.elts;
  258.     s->passwd.len = arg[0].len;
  259.     s->passwd.data = ngx_pnalloc(c->pool, s->passwd.len);
  260.     if (s->passwd.data == NULL) {
  261.         return NGX_ERROR;
  262.     }

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

  264. #if (NGX_DEBUG_MAIL_PASSWD)
  265.     ngx_log_debug1(NGX_LOG_DEBUG_MAIL, c->log, 0,
  266.                    "pop3 passwd: \"%V\"", &s->passwd);
  267. #endif

  268.     return NGX_DONE;
  269. }


  270. static ngx_int_t
  271. ngx_mail_pop3_capa(ngx_mail_session_t *s, ngx_connection_t *c, ngx_int_t stls)
  272. {
  273.     ngx_mail_pop3_srv_conf_t  *pscf;

  274.     pscf = ngx_mail_get_module_srv_conf(s, ngx_mail_pop3_module);

  275. #if (NGX_MAIL_SSL)

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

  278.         sslcf = ngx_mail_get_module_srv_conf(s, ngx_mail_ssl_module);

  279.         if (sslcf->starttls == NGX_MAIL_STARTTLS_ON) {
  280.             s->out = pscf->starttls_capability;
  281.             return NGX_OK;
  282.         }

  283.         if (sslcf->starttls == NGX_MAIL_STARTTLS_ONLY) {
  284.             s->out = pscf->starttls_only_capability;
  285.             return NGX_OK;
  286.         }
  287.     }

  288. #endif

  289.     s->out = pscf->capability;
  290.     return NGX_OK;
  291. }


  292. static ngx_int_t
  293. ngx_mail_pop3_stls(ngx_mail_session_t *s, ngx_connection_t *c)
  294. {
  295. #if (NGX_MAIL_SSL)
  296.     ngx_mail_ssl_conf_t  *sslcf;

  297.     if (c->ssl == NULL) {
  298.         sslcf = ngx_mail_get_module_srv_conf(s, ngx_mail_ssl_module);
  299.         if (sslcf->starttls) {
  300.             s->buffer->pos = s->buffer->start;
  301.             s->buffer->last = s->buffer->start;
  302.             c->read->handler = ngx_mail_starttls_handler;
  303.             return NGX_OK;
  304.         }
  305.     }

  306. #endif

  307.     return NGX_MAIL_PARSE_INVALID_COMMAND;
  308. }


  309. static ngx_int_t
  310. ngx_mail_pop3_apop(ngx_mail_session_t *s, ngx_connection_t *c)
  311. {
  312.     ngx_str_t                 *arg;
  313.     ngx_mail_pop3_srv_conf_t  *pscf;

  314. #if (NGX_MAIL_SSL)
  315.     if (ngx_mail_starttls_only(s, c)) {
  316.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  317.     }
  318. #endif

  319.     if (s->args.nelts != 2) {
  320.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  321.     }

  322.     pscf = ngx_mail_get_module_srv_conf(s, ngx_mail_pop3_module);

  323.     if (!(pscf->auth_methods & NGX_MAIL_AUTH_APOP_ENABLED)) {
  324.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  325.     }

  326.     arg = s->args.elts;

  327.     s->login.len = arg[0].len;
  328.     s->login.data = ngx_pnalloc(c->pool, s->login.len);
  329.     if (s->login.data == NULL) {
  330.         return NGX_ERROR;
  331.     }

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

  333.     s->passwd.len = arg[1].len;
  334.     s->passwd.data = ngx_pnalloc(c->pool, s->passwd.len);
  335.     if (s->passwd.data == NULL) {
  336.         return NGX_ERROR;
  337.     }

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

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

  341.     s->auth_method = NGX_MAIL_AUTH_APOP;

  342.     return NGX_DONE;
  343. }


  344. static ngx_int_t
  345. ngx_mail_pop3_auth(ngx_mail_session_t *s, ngx_connection_t *c)
  346. {
  347.     ngx_int_t                  rc;
  348.     ngx_mail_pop3_srv_conf_t  *pscf;

  349. #if (NGX_MAIL_SSL)
  350.     if (ngx_mail_starttls_only(s, c)) {
  351.         return NGX_MAIL_PARSE_INVALID_COMMAND;
  352.     }
  353. #endif

  354.     pscf = ngx_mail_get_module_srv_conf(s, ngx_mail_pop3_module);

  355.     if (s->args.nelts == 0) {
  356.         s->out = pscf->auth_capability;
  357.         s->state = 0;

  358.         return NGX_OK;
  359.     }

  360.     rc = ngx_mail_auth_parse(s, c);

  361.     switch (rc) {

  362.     case NGX_MAIL_AUTH_LOGIN:

  363.         ngx_str_set(&s->out, pop3_username);
  364.         s->mail_state = ngx_pop3_auth_login_username;

  365.         return NGX_OK;

  366.     case NGX_MAIL_AUTH_LOGIN_USERNAME:

  367.         ngx_str_set(&s->out, pop3_password);
  368.         s->mail_state = ngx_pop3_auth_login_password;

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

  370.     case NGX_MAIL_AUTH_PLAIN:

  371.         ngx_str_set(&s->out, pop3_next);
  372.         s->mail_state = ngx_pop3_auth_plain;

  373.         return NGX_OK;

  374.     case NGX_MAIL_AUTH_CRAM_MD5:

  375.         if (!(pscf->auth_methods & NGX_MAIL_AUTH_CRAM_MD5_ENABLED)) {
  376.             return NGX_MAIL_PARSE_INVALID_COMMAND;
  377.         }

  378.         if (ngx_mail_auth_cram_md5_salt(s, c, "+ ", 2) == NGX_OK) {
  379.             s->mail_state = ngx_pop3_auth_cram_md5;
  380.             return NGX_OK;
  381.         }

  382.         return NGX_ERROR;

  383.     case NGX_MAIL_AUTH_EXTERNAL:

  384.         if (!(pscf->auth_methods & NGX_MAIL_AUTH_EXTERNAL_ENABLED)) {
  385.             return NGX_MAIL_PARSE_INVALID_COMMAND;
  386.         }

  387.         ngx_str_set(&s->out, pop3_username);
  388.         s->mail_state = ngx_pop3_auth_external;

  389.         return NGX_OK;
  390.     }

  391.     return rc;
  392. }