Mercurial > hg > nginx-mail
annotate src/mail/ngx_mail_proxy_module.c @ 665:0b460e61bdcd default tip
Merge with nginx 1.0.0.
author | Maxim Dounin <mdounin@mdounin.ru> |
---|---|
date | Mon, 25 Apr 2011 04:22:17 +0400 |
parents | 9773720b845e 8246d8a2c2be |
children |
rev | line source |
---|---|
290 | 1 |
2 /* | |
3 * Copyright (C) Igor Sysoev | |
4 */ | |
5 | |
6 | |
7 #include <ngx_config.h> | |
8 #include <ngx_core.h> | |
9 #include <ngx_event.h> | |
10 #include <ngx_event_connect.h> | |
11 #include <ngx_mail.h> | |
12 | |
13 | |
14 typedef struct { | |
15 ngx_flag_t enable; | |
16 ngx_flag_t pass_error_message; | |
17 ngx_flag_t xclient; | |
18 size_t buffer_size; | |
19 ngx_msec_t timeout; | |
20 } ngx_mail_proxy_conf_t; | |
21 | |
22 | |
23 static void ngx_mail_proxy_block_read(ngx_event_t *rev); | |
24 static void ngx_mail_proxy_pop3_handler(ngx_event_t *rev); | |
25 static void ngx_mail_proxy_imap_handler(ngx_event_t *rev); | |
26 static void ngx_mail_proxy_smtp_handler(ngx_event_t *rev); | |
27 static void ngx_mail_proxy_dummy_handler(ngx_event_t *ev); | |
28 static ngx_int_t ngx_mail_proxy_read_response(ngx_mail_session_t *s, | |
29 ngx_uint_t state); | |
30 static void ngx_mail_proxy_handler(ngx_event_t *ev); | |
31 static void ngx_mail_proxy_upstream_error(ngx_mail_session_t *s); | |
32 static void ngx_mail_proxy_internal_server_error(ngx_mail_session_t *s); | |
33 static void ngx_mail_proxy_close_session(ngx_mail_session_t *s); | |
34 static void *ngx_mail_proxy_create_conf(ngx_conf_t *cf); | |
35 static char *ngx_mail_proxy_merge_conf(ngx_conf_t *cf, void *parent, | |
36 void *child); | |
37 | |
38 | |
39 static ngx_command_t ngx_mail_proxy_commands[] = { | |
40 | |
41 { ngx_string("proxy"), | |
42 NGX_MAIL_MAIN_CONF|NGX_MAIL_SRV_CONF|NGX_CONF_FLAG, | |
43 ngx_conf_set_flag_slot, | |
44 NGX_MAIL_SRV_CONF_OFFSET, | |
45 offsetof(ngx_mail_proxy_conf_t, enable), | |
46 NULL }, | |
47 | |
48 { ngx_string("proxy_buffer"), | |
49 NGX_MAIL_MAIN_CONF|NGX_MAIL_SRV_CONF|NGX_CONF_TAKE1, | |
50 ngx_conf_set_size_slot, | |
51 NGX_MAIL_SRV_CONF_OFFSET, | |
52 offsetof(ngx_mail_proxy_conf_t, buffer_size), | |
53 NULL }, | |
54 | |
55 { ngx_string("proxy_timeout"), | |
56 NGX_MAIL_MAIN_CONF|NGX_MAIL_SRV_CONF|NGX_CONF_TAKE1, | |
57 ngx_conf_set_msec_slot, | |
58 NGX_MAIL_SRV_CONF_OFFSET, | |
59 offsetof(ngx_mail_proxy_conf_t, timeout), | |
60 NULL }, | |
61 | |
62 { ngx_string("proxy_pass_error_message"), | |
63 NGX_MAIL_MAIN_CONF|NGX_MAIL_SRV_CONF|NGX_CONF_TAKE1, | |
64 ngx_conf_set_flag_slot, | |
65 NGX_MAIL_SRV_CONF_OFFSET, | |
66 offsetof(ngx_mail_proxy_conf_t, pass_error_message), | |
67 NULL }, | |
68 | |
69 { ngx_string("xclient"), | |
70 NGX_MAIL_MAIN_CONF|NGX_MAIL_SRV_CONF|NGX_CONF_FLAG, | |
71 ngx_conf_set_flag_slot, | |
72 NGX_MAIL_SRV_CONF_OFFSET, | |
73 offsetof(ngx_mail_proxy_conf_t, xclient), | |
74 NULL }, | |
75 | |
76 ngx_null_command | |
77 }; | |
78 | |
79 | |
80 static ngx_mail_module_t ngx_mail_proxy_module_ctx = { | |
336 | 81 NULL, /* protocol */ |
82 | |
290 | 83 NULL, /* create main configuration */ |
84 NULL, /* init main configuration */ | |
85 | |
86 ngx_mail_proxy_create_conf, /* create server configuration */ | |
87 ngx_mail_proxy_merge_conf /* merge server configuration */ | |
88 }; | |
89 | |
90 | |
91 ngx_module_t ngx_mail_proxy_module = { | |
92 NGX_MODULE_V1, | |
93 &ngx_mail_proxy_module_ctx, /* module context */ | |
94 ngx_mail_proxy_commands, /* module directives */ | |
95 NGX_MAIL_MODULE, /* module type */ | |
96 NULL, /* init master */ | |
97 NULL, /* init module */ | |
98 NULL, /* init process */ | |
99 NULL, /* init thread */ | |
100 NULL, /* exit thread */ | |
101 NULL, /* exit process */ | |
102 NULL, /* exit master */ | |
103 NGX_MODULE_V1_PADDING | |
104 }; | |
105 | |
106 | |
473 | 107 static u_char smtp_auth_ok[] = "235 2.0.0 OK" CRLF; |
290 | 108 |
109 | |
110 void | |
583 | 111 ngx_mail_proxy_init(ngx_mail_session_t *s, ngx_addr_t *peer) |
290 | 112 { |
113 int keepalive; | |
114 ngx_int_t rc; | |
115 ngx_mail_proxy_ctx_t *p; | |
116 ngx_mail_proxy_conf_t *pcf; | |
117 ngx_mail_core_srv_conf_t *cscf; | |
118 | |
119 s->connection->log->action = "connecting to upstream"; | |
120 | |
121 cscf = ngx_mail_get_module_srv_conf(s, ngx_mail_core_module); | |
122 | |
123 if (cscf->so_keepalive) { | |
124 keepalive = 1; | |
125 | |
126 if (setsockopt(s->connection->fd, SOL_SOCKET, SO_KEEPALIVE, | |
127 (const void *) &keepalive, sizeof(int)) | |
128 == -1) | |
129 { | |
130 ngx_log_error(NGX_LOG_ALERT, s->connection->log, ngx_socket_errno, | |
131 "setsockopt(SO_KEEPALIVE) failed"); | |
132 } | |
133 } | |
134 | |
135 p = ngx_pcalloc(s->connection->pool, sizeof(ngx_mail_proxy_ctx_t)); | |
136 if (p == NULL) { | |
137 ngx_mail_session_internal_server_error(s); | |
138 return; | |
139 } | |
140 | |
141 s->proxy = p; | |
142 | |
143 p->upstream.sockaddr = peer->sockaddr; | |
144 p->upstream.socklen = peer->socklen; | |
145 p->upstream.name = &peer->name; | |
146 p->upstream.get = ngx_event_get_peer; | |
147 p->upstream.log = s->connection->log; | |
148 p->upstream.log_error = NGX_ERROR_ERR; | |
149 | |
150 rc = ngx_event_connect_peer(&p->upstream); | |
151 | |
152 if (rc == NGX_ERROR || rc == NGX_BUSY || rc == NGX_DECLINED) { | |
153 ngx_mail_proxy_internal_server_error(s); | |
154 return; | |
155 } | |
156 | |
157 ngx_add_timer(p->upstream.connection->read, cscf->timeout); | |
158 | |
159 p->upstream.connection->data = s; | |
160 p->upstream.connection->pool = s->connection->pool; | |
161 | |
162 s->connection->read->handler = ngx_mail_proxy_block_read; | |
163 p->upstream.connection->write->handler = ngx_mail_proxy_dummy_handler; | |
164 | |
165 pcf = ngx_mail_get_module_srv_conf(s, ngx_mail_proxy_module); | |
166 | |
167 s->proxy->buffer = ngx_create_temp_buf(s->connection->pool, | |
168 pcf->buffer_size); | |
169 if (s->proxy->buffer == NULL) { | |
170 ngx_mail_proxy_internal_server_error(s); | |
171 return; | |
172 } | |
173 | |
372 | 174 s->out.len = 0; |
175 | |
290 | 176 switch (s->protocol) { |
177 | |
178 case NGX_MAIL_POP3_PROTOCOL: | |
179 p->upstream.connection->read->handler = ngx_mail_proxy_pop3_handler; | |
180 s->mail_state = ngx_pop3_start; | |
181 break; | |
182 | |
183 case NGX_MAIL_IMAP_PROTOCOL: | |
184 p->upstream.connection->read->handler = ngx_mail_proxy_imap_handler; | |
185 s->mail_state = ngx_imap_start; | |
186 break; | |
187 | |
188 default: /* NGX_MAIL_SMTP_PROTOCOL */ | |
189 p->upstream.connection->read->handler = ngx_mail_proxy_smtp_handler; | |
190 s->mail_state = ngx_smtp_start; | |
191 break; | |
192 } | |
193 } | |
194 | |
195 | |
196 static void | |
197 ngx_mail_proxy_block_read(ngx_event_t *rev) | |
198 { | |
199 ngx_connection_t *c; | |
200 ngx_mail_session_t *s; | |
201 | |
202 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, "mail proxy block read"); | |
203 | |
459 | 204 if (ngx_handle_read_event(rev, 0) != NGX_OK) { |
290 | 205 c = rev->data; |
206 s = c->data; | |
207 | |
208 ngx_mail_proxy_close_session(s); | |
209 } | |
210 } | |
211 | |
212 | |
213 static void | |
214 ngx_mail_proxy_pop3_handler(ngx_event_t *rev) | |
215 { | |
216 u_char *p; | |
217 ngx_int_t rc; | |
218 ngx_str_t line; | |
219 ngx_connection_t *c; | |
220 ngx_mail_session_t *s; | |
221 ngx_mail_proxy_conf_t *pcf; | |
222 | |
223 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, | |
224 "mail proxy pop3 auth handler"); | |
225 | |
226 c = rev->data; | |
227 s = c->data; | |
228 | |
229 if (rev->timedout) { | |
230 ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, | |
231 "upstream timed out"); | |
232 c->timedout = 1; | |
233 ngx_mail_proxy_internal_server_error(s); | |
234 return; | |
235 } | |
236 | |
237 rc = ngx_mail_proxy_read_response(s, 0); | |
238 | |
239 if (rc == NGX_AGAIN) { | |
240 return; | |
241 } | |
242 | |
243 if (rc == NGX_ERROR) { | |
244 ngx_mail_proxy_upstream_error(s); | |
245 return; | |
246 } | |
247 | |
248 switch (s->mail_state) { | |
249 | |
250 case ngx_pop3_start: | |
251 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, "mail proxy send user"); | |
252 | |
253 s->connection->log->action = "sending user name to upstream"; | |
254 | |
255 line.len = sizeof("USER ") - 1 + s->login.len + 2; | |
382 | 256 line.data = ngx_pnalloc(c->pool, line.len); |
290 | 257 if (line.data == NULL) { |
258 ngx_mail_proxy_internal_server_error(s); | |
259 return; | |
260 } | |
261 | |
262 p = ngx_cpymem(line.data, "USER ", sizeof("USER ") - 1); | |
263 p = ngx_cpymem(p, s->login.data, s->login.len); | |
264 *p++ = CR; *p = LF; | |
265 | |
266 s->mail_state = ngx_pop3_user; | |
267 break; | |
268 | |
269 case ngx_pop3_user: | |
270 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, "mail proxy send pass"); | |
271 | |
272 s->connection->log->action = "sending password to upstream"; | |
273 | |
274 line.len = sizeof("PASS ") - 1 + s->passwd.len + 2; | |
382 | 275 line.data = ngx_pnalloc(c->pool, line.len); |
290 | 276 if (line.data == NULL) { |
277 ngx_mail_proxy_internal_server_error(s); | |
278 return; | |
279 } | |
280 | |
281 p = ngx_cpymem(line.data, "PASS ", sizeof("PASS ") - 1); | |
282 p = ngx_cpymem(p, s->passwd.data, s->passwd.len); | |
283 *p++ = CR; *p = LF; | |
284 | |
285 s->mail_state = ngx_pop3_passwd; | |
286 break; | |
287 | |
288 case ngx_pop3_passwd: | |
289 s->connection->read->handler = ngx_mail_proxy_handler; | |
290 s->connection->write->handler = ngx_mail_proxy_handler; | |
291 rev->handler = ngx_mail_proxy_handler; | |
292 c->write->handler = ngx_mail_proxy_handler; | |
293 | |
294 pcf = ngx_mail_get_module_srv_conf(s, ngx_mail_proxy_module); | |
295 ngx_add_timer(s->connection->read, pcf->timeout); | |
296 ngx_del_timer(c->read); | |
297 | |
298 c->log->action = NULL; | |
299 ngx_log_error(NGX_LOG_INFO, c->log, 0, "client logged in"); | |
300 | |
301 ngx_mail_proxy_handler(s->connection->write); | |
302 | |
303 return; | |
304 | |
305 default: | |
306 #if (NGX_SUPPRESS_WARN) | |
613 | 307 ngx_str_null(&line); |
290 | 308 #endif |
309 break; | |
310 } | |
311 | |
312 if (c->send(c, line.data, line.len) < (ssize_t) line.len) { | |
313 /* | |
314 * we treat the incomplete sending as NGX_ERROR | |
315 * because it is very strange here | |
316 */ | |
317 ngx_mail_proxy_internal_server_error(s); | |
318 return; | |
319 } | |
320 | |
321 s->proxy->buffer->pos = s->proxy->buffer->start; | |
322 s->proxy->buffer->last = s->proxy->buffer->start; | |
323 } | |
324 | |
325 | |
326 static void | |
327 ngx_mail_proxy_imap_handler(ngx_event_t *rev) | |
328 { | |
329 u_char *p; | |
330 ngx_int_t rc; | |
331 ngx_str_t line; | |
332 ngx_connection_t *c; | |
333 ngx_mail_session_t *s; | |
334 ngx_mail_proxy_conf_t *pcf; | |
335 | |
336 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, | |
337 "mail proxy imap auth handler"); | |
338 | |
339 c = rev->data; | |
340 s = c->data; | |
341 | |
342 if (rev->timedout) { | |
343 ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, | |
344 "upstream timed out"); | |
345 c->timedout = 1; | |
346 ngx_mail_proxy_internal_server_error(s); | |
347 return; | |
348 } | |
349 | |
350 rc = ngx_mail_proxy_read_response(s, s->mail_state); | |
351 | |
352 if (rc == NGX_AGAIN) { | |
353 return; | |
354 } | |
355 | |
356 if (rc == NGX_ERROR) { | |
357 ngx_mail_proxy_upstream_error(s); | |
358 return; | |
359 } | |
360 | |
361 switch (s->mail_state) { | |
362 | |
363 case ngx_imap_start: | |
364 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, | |
365 "mail proxy send login"); | |
366 | |
367 s->connection->log->action = "sending LOGIN command to upstream"; | |
368 | |
369 line.len = s->tag.len + sizeof("LOGIN ") - 1 | |
370 + 1 + NGX_SIZE_T_LEN + 1 + 2; | |
382 | 371 line.data = ngx_pnalloc(c->pool, line.len); |
290 | 372 if (line.data == NULL) { |
373 ngx_mail_proxy_internal_server_error(s); | |
374 return; | |
375 } | |
376 | |
377 line.len = ngx_sprintf(line.data, "%VLOGIN {%uz}" CRLF, | |
378 &s->tag, s->login.len) | |
379 - line.data; | |
380 | |
381 s->mail_state = ngx_imap_login; | |
382 break; | |
383 | |
384 case ngx_imap_login: | |
385 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, "mail proxy send user"); | |
386 | |
387 s->connection->log->action = "sending user name to upstream"; | |
388 | |
389 line.len = s->login.len + 1 + 1 + NGX_SIZE_T_LEN + 1 + 2; | |
382 | 390 line.data = ngx_pnalloc(c->pool, line.len); |
290 | 391 if (line.data == NULL) { |
392 ngx_mail_proxy_internal_server_error(s); | |
393 return; | |
394 } | |
395 | |
396 line.len = ngx_sprintf(line.data, "%V {%uz}" CRLF, | |
397 &s->login, s->passwd.len) | |
398 - line.data; | |
399 | |
400 s->mail_state = ngx_imap_user; | |
401 break; | |
402 | |
403 case ngx_imap_user: | |
404 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, | |
405 "mail proxy send passwd"); | |
406 | |
407 s->connection->log->action = "sending password to upstream"; | |
408 | |
409 line.len = s->passwd.len + 2; | |
382 | 410 line.data = ngx_pnalloc(c->pool, line.len); |
290 | 411 if (line.data == NULL) { |
412 ngx_mail_proxy_internal_server_error(s); | |
413 return; | |
414 } | |
415 | |
416 p = ngx_cpymem(line.data, s->passwd.data, s->passwd.len); | |
417 *p++ = CR; *p = LF; | |
418 | |
419 s->mail_state = ngx_imap_passwd; | |
420 break; | |
421 | |
422 case ngx_imap_passwd: | |
423 s->connection->read->handler = ngx_mail_proxy_handler; | |
424 s->connection->write->handler = ngx_mail_proxy_handler; | |
425 rev->handler = ngx_mail_proxy_handler; | |
426 c->write->handler = ngx_mail_proxy_handler; | |
427 | |
428 pcf = ngx_mail_get_module_srv_conf(s, ngx_mail_proxy_module); | |
429 ngx_add_timer(s->connection->read, pcf->timeout); | |
430 ngx_del_timer(c->read); | |
431 | |
432 c->log->action = NULL; | |
433 ngx_log_error(NGX_LOG_INFO, c->log, 0, "client logged in"); | |
434 | |
435 ngx_mail_proxy_handler(s->connection->write); | |
436 | |
437 return; | |
438 | |
439 default: | |
440 #if (NGX_SUPPRESS_WARN) | |
613 | 441 ngx_str_null(&line); |
290 | 442 #endif |
443 break; | |
444 } | |
445 | |
446 if (c->send(c, line.data, line.len) < (ssize_t) line.len) { | |
447 /* | |
448 * we treat the incomplete sending as NGX_ERROR | |
449 * because it is very strange here | |
450 */ | |
451 ngx_mail_proxy_internal_server_error(s); | |
452 return; | |
453 } | |
454 | |
455 s->proxy->buffer->pos = s->proxy->buffer->start; | |
456 s->proxy->buffer->last = s->proxy->buffer->start; | |
457 } | |
458 | |
459 | |
460 static void | |
461 ngx_mail_proxy_smtp_handler(ngx_event_t *rev) | |
462 { | |
463 u_char *p; | |
464 ngx_int_t rc; | |
465 ngx_str_t line; | |
449 | 466 ngx_buf_t *b; |
290 | 467 ngx_connection_t *c; |
468 ngx_mail_session_t *s; | |
469 ngx_mail_proxy_conf_t *pcf; | |
470 ngx_mail_core_srv_conf_t *cscf; | |
471 | |
472 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, | |
473 "mail proxy smtp auth handler"); | |
474 | |
475 c = rev->data; | |
476 s = c->data; | |
477 | |
478 if (rev->timedout) { | |
479 ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, | |
480 "upstream timed out"); | |
481 c->timedout = 1; | |
482 ngx_mail_proxy_internal_server_error(s); | |
483 return; | |
484 } | |
485 | |
486 rc = ngx_mail_proxy_read_response(s, s->mail_state); | |
487 | |
488 if (rc == NGX_AGAIN) { | |
489 return; | |
490 } | |
491 | |
492 if (rc == NGX_ERROR) { | |
493 ngx_mail_proxy_upstream_error(s); | |
494 return; | |
495 } | |
496 | |
497 switch (s->mail_state) { | |
498 | |
499 case ngx_smtp_start: | |
500 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, "mail proxy send ehlo"); | |
501 | |
502 s->connection->log->action = "sending HELO/EHLO to upstream"; | |
503 | |
504 cscf = ngx_mail_get_module_srv_conf(s, ngx_mail_core_module); | |
505 | |
506 line.len = sizeof("HELO ") - 1 + cscf->server_name.len + 2; | |
382 | 507 line.data = ngx_pnalloc(c->pool, line.len); |
290 | 508 if (line.data == NULL) { |
509 ngx_mail_proxy_internal_server_error(s); | |
510 return; | |
511 } | |
512 | |
513 pcf = ngx_mail_get_module_srv_conf(s, ngx_mail_proxy_module); | |
514 | |
515 p = ngx_cpymem(line.data, | |
516 ((s->esmtp || pcf->xclient) ? "EHLO " : "HELO "), | |
517 sizeof("HELO ") - 1); | |
518 | |
519 p = ngx_cpymem(p, cscf->server_name.data, cscf->server_name.len); | |
520 *p++ = CR; *p = LF; | |
521 | |
449 | 522 if (pcf->xclient) { |
523 s->mail_state = ngx_smtp_helo_xclient; | |
524 | |
525 } else if (s->auth_method == NGX_MAIL_AUTH_NONE) { | |
526 s->mail_state = ngx_smtp_helo_from; | |
527 | |
528 } else { | |
529 s->mail_state = ngx_smtp_helo; | |
530 } | |
290 | 531 |
532 break; | |
533 | |
449 | 534 case ngx_smtp_helo_xclient: |
290 | 535 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, |
536 "mail proxy send xclient"); | |
537 | |
538 s->connection->log->action = "sending XCLIENT to upstream"; | |
539 | |
482 | 540 line.len = sizeof("XCLIENT ADDR= LOGIN= NAME=" |
366 | 541 CRLF) - 1 |
542 + s->connection->addr_text.len + s->login.len + s->host.len; | |
290 | 543 |
382 | 544 line.data = ngx_pnalloc(c->pool, line.len); |
290 | 545 if (line.data == NULL) { |
546 ngx_mail_proxy_internal_server_error(s); | |
547 return; | |
548 } | |
549 | |
449 | 550 line.len = ngx_sprintf(line.data, |
482 | 551 "XCLIENT ADDR=%V%s%V NAME=%V" CRLF, |
449 | 552 &s->connection->addr_text, |
553 (s->login.len ? " LOGIN=" : ""), &s->login, &s->host) | |
554 - line.data; | |
555 | |
482 | 556 if (s->smtp_helo.len) { |
557 s->mail_state = ngx_smtp_xclient_helo; | |
558 | |
559 } else if (s->auth_method == NGX_MAIL_AUTH_NONE) { | |
560 s->mail_state = ngx_smtp_xclient_from; | |
561 | |
562 } else { | |
563 s->mail_state = ngx_smtp_xclient; | |
564 } | |
565 | |
566 break; | |
567 | |
568 case ngx_smtp_xclient_helo: | |
569 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, | |
570 "mail proxy send client ehlo"); | |
571 | |
572 s->connection->log->action = "sending client HELO/EHLO to upstream"; | |
573 | |
574 line.len = sizeof("HELO " CRLF) - 1 + s->smtp_helo.len; | |
575 | |
576 line.data = ngx_pnalloc(c->pool, line.len); | |
577 if (line.data == NULL) { | |
578 ngx_mail_proxy_internal_server_error(s); | |
579 return; | |
580 } | |
581 | |
582 line.len = ngx_sprintf(line.data, | |
583 ((s->esmtp) ? "EHLO %V" CRLF : "HELO %V" CRLF), | |
584 &s->smtp_helo) | |
585 - line.data; | |
586 | |
449 | 587 s->mail_state = (s->auth_method == NGX_MAIL_AUTH_NONE) ? |
482 | 588 ngx_smtp_helo_from : ngx_smtp_helo; |
449 | 589 |
590 break; | |
591 | |
592 case ngx_smtp_helo_from: | |
593 case ngx_smtp_xclient_from: | |
594 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, | |
595 "mail proxy send mail from"); | |
596 | |
597 s->connection->log->action = "sending MAIL FROM to upstream"; | |
598 | |
599 line.len = s->smtp_from.len + sizeof(CRLF) - 1; | |
600 line.data = ngx_pnalloc(c->pool, line.len); | |
601 if (line.data == NULL) { | |
602 ngx_mail_proxy_internal_server_error(s); | |
603 return; | |
290 | 604 } |
605 | |
449 | 606 p = ngx_cpymem(line.data, s->smtp_from.data, s->smtp_from.len); |
607 *p++ = CR; *p = LF; | |
608 | |
609 s->mail_state = ngx_smtp_from; | |
610 | |
290 | 611 break; |
612 | |
449 | 613 case ngx_smtp_from: |
614 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, rev->log, 0, | |
615 "mail proxy send rcpt to"); | |
616 | |
617 s->connection->log->action = "sending RCPT TO to upstream"; | |
618 | |
619 line.len = s->smtp_to.len + sizeof(CRLF) - 1; | |
620 line.data = ngx_pnalloc(c->pool, line.len); | |
621 if (line.data == NULL) { | |
622 ngx_mail_proxy_internal_server_error(s); | |
623 return; | |
624 } | |
625 | |
626 p = ngx_cpymem(line.data, s->smtp_to.data, s->smtp_to.len); | |
627 *p++ = CR; *p = LF; | |
290 | 628 |
449 | 629 s->mail_state = ngx_smtp_to; |
630 | |
631 break; | |
632 | |
633 case ngx_smtp_helo: | |
634 case ngx_smtp_xclient: | |
635 case ngx_smtp_to: | |
290 | 636 |
449 | 637 b = s->proxy->buffer; |
638 | |
639 if (s->auth_method == NGX_MAIL_AUTH_NONE) { | |
640 b->pos = b->start; | |
641 | |
642 } else { | |
473 | 643 ngx_memcpy(b->start, smtp_auth_ok, sizeof(smtp_auth_ok) - 1); |
644 b->last = b->start + sizeof(smtp_auth_ok) - 1; | |
449 | 645 } |
290 | 646 |
647 s->connection->read->handler = ngx_mail_proxy_handler; | |
648 s->connection->write->handler = ngx_mail_proxy_handler; | |
649 rev->handler = ngx_mail_proxy_handler; | |
650 c->write->handler = ngx_mail_proxy_handler; | |
651 | |
652 pcf = ngx_mail_get_module_srv_conf(s, ngx_mail_proxy_module); | |
653 ngx_add_timer(s->connection->read, pcf->timeout); | |
654 ngx_del_timer(c->read); | |
655 | |
656 c->log->action = NULL; | |
657 ngx_log_error(NGX_LOG_INFO, c->log, 0, "client logged in"); | |
658 | |
436
9b19e26b2660
Mail: smtp pipelining support.
Maxim Dounin <mdounin@mdounin.ru>
parents:
434
diff
changeset
|
659 if (s->buffer->pos == s->buffer->last) { |
9b19e26b2660
Mail: smtp pipelining support.
Maxim Dounin <mdounin@mdounin.ru>
parents:
434
diff
changeset
|
660 ngx_mail_proxy_handler(s->connection->write); |
9b19e26b2660
Mail: smtp pipelining support.
Maxim Dounin <mdounin@mdounin.ru>
parents:
434
diff
changeset
|
661 |
9b19e26b2660
Mail: smtp pipelining support.
Maxim Dounin <mdounin@mdounin.ru>
parents:
434
diff
changeset
|
662 } else { |
9b19e26b2660
Mail: smtp pipelining support.
Maxim Dounin <mdounin@mdounin.ru>
parents:
434
diff
changeset
|
663 ngx_mail_proxy_handler(c->write); |
9b19e26b2660
Mail: smtp pipelining support.
Maxim Dounin <mdounin@mdounin.ru>
parents:
434
diff
changeset
|
664 } |
290 | 665 |
666 return; | |
667 | |
668 default: | |
669 #if (NGX_SUPPRESS_WARN) | |
613 | 670 ngx_str_null(&line); |
290 | 671 #endif |
672 break; | |
673 } | |
674 | |
675 if (c->send(c, line.data, line.len) < (ssize_t) line.len) { | |
676 /* | |
677 * we treat the incomplete sending as NGX_ERROR | |
678 * because it is very strange here | |
679 */ | |
680 ngx_mail_proxy_internal_server_error(s); | |
681 return; | |
682 } | |
683 | |
684 s->proxy->buffer->pos = s->proxy->buffer->start; | |
685 s->proxy->buffer->last = s->proxy->buffer->start; | |
686 } | |
687 | |
688 | |
689 static void | |
690 ngx_mail_proxy_dummy_handler(ngx_event_t *wev) | |
691 { | |
692 ngx_connection_t *c; | |
693 ngx_mail_session_t *s; | |
694 | |
695 ngx_log_debug0(NGX_LOG_DEBUG_MAIL, wev->log, 0, "mail proxy dummy handler"); | |
696 | |
459 | 697 if (ngx_handle_write_event(wev, 0) != NGX_OK) { |
290 | 698 c = wev->data; |
699 s = c->data; | |
700 | |
701 ngx_mail_proxy_close_session(s); | |
702 } | |
703 } | |
704 | |
705 | |
706 static ngx_int_t | |
707 ngx_mail_proxy_read_response(ngx_mail_session_t *s, ngx_uint_t state) | |
708 { | |
709 u_char *p; | |
710 ssize_t n; | |
711 ngx_buf_t *b; | |
712 ngx_mail_proxy_conf_t *pcf; | |
713 | |
714 s->connection->log->action = "reading response from upstream"; | |
715 | |
716 b = s->proxy->buffer; | |
717 | |
718 n = s->proxy->upstream.connection->recv(s->proxy->upstream.connection, | |
719 b->last, b->end - b->last); | |
720 | |
721 if (n == NGX_ERROR || n == 0) { | |
722 return NGX_ERROR; | |
723 } | |
724 | |
725 if (n == NGX_AGAIN) { | |
726 return NGX_AGAIN; | |
727 } | |
728 | |
729 b->last += n; | |
730 | |
613 | 731 if (b->last - b->pos < 4) { |
290 | 732 return NGX_AGAIN; |
733 } | |
734 | |
735 if (*(b->last - 2) != CR || *(b->last - 1) != LF) { | |
736 if (b->last == b->end) { | |
737 *(b->last - 1) = '\0'; | |
738 ngx_log_error(NGX_LOG_ERR, s->connection->log, 0, | |
739 "upstream sent too long response line: \"%s\"", | |
740 b->pos); | |
741 return NGX_ERROR; | |
742 } | |
743 | |
744 return NGX_AGAIN; | |
745 } | |
746 | |
747 p = b->pos; | |
748 | |
749 switch (s->protocol) { | |
750 | |
751 case NGX_MAIL_POP3_PROTOCOL: | |
752 if (p[0] == '+' && p[1] == 'O' && p[2] == 'K') { | |
753 return NGX_OK; | |
754 } | |
755 break; | |
756 | |
757 case NGX_MAIL_IMAP_PROTOCOL: | |
758 switch (state) { | |
759 | |
760 case ngx_imap_start: | |
761 if (p[0] == '*' && p[1] == ' ' && p[2] == 'O' && p[3] == 'K') { | |
762 return NGX_OK; | |
763 } | |
764 break; | |
765 | |
766 case ngx_imap_login: | |
767 case ngx_imap_user: | |
768 if (p[0] == '+') { | |
769 return NGX_OK; | |
770 } | |
771 break; | |
772 | |
773 case ngx_imap_passwd: | |
774 if (ngx_strncmp(p, s->tag.data, s->tag.len) == 0) { | |
775 p += s->tag.len; | |
776 if (p[0] == 'O' && p[1] == 'K') { | |
777 return NGX_OK; | |
778 } | |
779 } | |
780 break; | |
781 } | |
782 | |
783 break; | |
784 | |
785 default: /* NGX_MAIL_SMTP_PROTOCOL */ | |
786 switch (state) { | |
787 | |
449 | 788 case ngx_smtp_start: |
789 if (p[0] == '2' && p[1] == '2' && p[2] == '0') { | |
790 return NGX_OK; | |
791 } | |
792 break; | |
793 | |
290 | 794 case ngx_smtp_helo: |
449 | 795 case ngx_smtp_helo_xclient: |
796 case ngx_smtp_helo_from: | |
797 case ngx_smtp_from: | |
296 | 798 if (p[0] == '2' && p[1] == '5' && p[2] == '0') { |
799 return NGX_OK; | |
800 } | |
801 break; | |
290 | 802 |
803 case ngx_smtp_xclient: | |
449 | 804 case ngx_smtp_xclient_from: |
482 | 805 case ngx_smtp_xclient_helo: |
449 | 806 if (p[0] == '2' && (p[1] == '2' || p[1] == '5') && p[2] == '0') { |
296 | 807 return NGX_OK; |
808 } | |
809 break; | |
449 | 810 |
811 case ngx_smtp_to: | |
812 return NGX_OK; | |
290 | 813 } |
814 | |
296 | 815 break; |
290 | 816 } |
817 | |
818 pcf = ngx_mail_get_module_srv_conf(s, ngx_mail_proxy_module); | |
819 | |
820 if (pcf->pass_error_message == 0) { | |
821 *(b->last - 2) = '\0'; | |
822 ngx_log_error(NGX_LOG_ERR, s->connection->log, 0, | |
823 "upstream sent invalid response: \"%s\"", p); | |
824 return NGX_ERROR; | |
825 } | |
826 | |
827 s->out.len = b->last - p - 2; | |
828 s->out.data = p; | |
829 | |
830 ngx_log_error(NGX_LOG_INFO, s->connection->log, 0, | |
831 "upstream sent invalid response: \"%V\"", &s->out); | |
832 | |
833 s->out.len = b->last - b->pos; | |
834 s->out.data = b->pos; | |
835 | |
836 return NGX_ERROR; | |
837 } | |
838 | |
839 | |
840 static void | |
841 ngx_mail_proxy_handler(ngx_event_t *ev) | |
842 { | |
843 char *action, *recv_action, *send_action; | |
844 size_t size; | |
845 ssize_t n; | |
846 ngx_buf_t *b; | |
847 ngx_uint_t do_write; | |
848 ngx_connection_t *c, *src, *dst; | |
849 ngx_mail_session_t *s; | |
850 ngx_mail_proxy_conf_t *pcf; | |
851 | |
852 c = ev->data; | |
853 s = c->data; | |
854 | |
855 if (ev->timedout) { | |
856 c->log->action = "proxying"; | |
857 | |
858 if (c == s->connection) { | |
859 ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, | |
860 "client timed out"); | |
861 c->timedout = 1; | |
862 | |
863 } else { | |
864 ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, | |
865 "upstream timed out"); | |
866 } | |
867 | |
868 ngx_mail_proxy_close_session(s); | |
869 return; | |
870 } | |
871 | |
872 if (c == s->connection) { | |
873 if (ev->write) { | |
874 recv_action = "proxying and reading from upstream"; | |
875 send_action = "proxying and sending to client"; | |
876 src = s->proxy->upstream.connection; | |
877 dst = c; | |
878 b = s->proxy->buffer; | |
879 | |
880 } else { | |
881 recv_action = "proxying and reading from client"; | |
882 send_action = "proxying and sending to upstream"; | |
883 src = c; | |
884 dst = s->proxy->upstream.connection; | |
885 b = s->buffer; | |
886 } | |
887 | |
888 } else { | |
889 if (ev->write) { | |
890 recv_action = "proxying and reading from client"; | |
891 send_action = "proxying and sending to upstream"; | |
892 src = s->connection; | |
893 dst = c; | |
894 b = s->buffer; | |
895 | |
896 } else { | |
897 recv_action = "proxying and reading from upstream"; | |
898 send_action = "proxying and sending to client"; | |
899 src = c; | |
900 dst = s->connection; | |
901 b = s->proxy->buffer; | |
902 } | |
903 } | |
904 | |
905 do_write = ev->write ? 1 : 0; | |
906 | |
907 ngx_log_debug3(NGX_LOG_DEBUG_MAIL, ev->log, 0, | |
908 "mail proxy handler: %d, #%d > #%d", | |
909 do_write, src->fd, dst->fd); | |
910 | |
911 for ( ;; ) { | |
912 | |
913 if (do_write) { | |
914 | |
915 size = b->last - b->pos; | |
916 | |
917 if (size && dst->write->ready) { | |
918 c->log->action = send_action; | |
919 | |
920 n = dst->send(dst, b->pos, size); | |
921 | |
922 if (n == NGX_ERROR) { | |
923 ngx_mail_proxy_close_session(s); | |
924 return; | |
925 } | |
926 | |
927 if (n > 0) { | |
928 b->pos += n; | |
929 | |
930 if (b->pos == b->last) { | |
931 b->pos = b->start; | |
932 b->last = b->start; | |
933 } | |
934 } | |
935 } | |
936 } | |
937 | |
938 size = b->end - b->last; | |
939 | |
940 if (size && src->read->ready) { | |
941 c->log->action = recv_action; | |
942 | |
943 n = src->recv(src, b->last, size); | |
944 | |
945 if (n == NGX_AGAIN || n == 0) { | |
946 break; | |
947 } | |
948 | |
949 if (n > 0) { | |
950 do_write = 1; | |
951 b->last += n; | |
952 | |
953 continue; | |
954 } | |
955 | |
956 if (n == NGX_ERROR) { | |
957 src->read->eof = 1; | |
958 } | |
959 } | |
960 | |
961 break; | |
962 } | |
963 | |
964 c->log->action = "proxying"; | |
965 | |
324 | 966 if ((s->connection->read->eof && s->buffer->pos == s->buffer->last) |
967 || (s->proxy->upstream.connection->read->eof | |
968 && s->proxy->buffer->pos == s->proxy->buffer->last) | |
969 || (s->connection->read->eof | |
970 && s->proxy->upstream.connection->read->eof)) | |
290 | 971 { |
972 action = c->log->action; | |
973 c->log->action = NULL; | |
974 ngx_log_error(NGX_LOG_INFO, c->log, 0, "proxied session done"); | |
975 c->log->action = action; | |
976 | |
977 ngx_mail_proxy_close_session(s); | |
978 return; | |
979 } | |
980 | |
459 | 981 if (ngx_handle_write_event(dst->write, 0) != NGX_OK) { |
290 | 982 ngx_mail_proxy_close_session(s); |
983 return; | |
984 } | |
985 | |
459 | 986 if (ngx_handle_read_event(dst->read, 0) != NGX_OK) { |
290 | 987 ngx_mail_proxy_close_session(s); |
988 return; | |
989 } | |
990 | |
459 | 991 if (ngx_handle_write_event(src->write, 0) != NGX_OK) { |
290 | 992 ngx_mail_proxy_close_session(s); |
993 return; | |
994 } | |
995 | |
459 | 996 if (ngx_handle_read_event(src->read, 0) != NGX_OK) { |
290 | 997 ngx_mail_proxy_close_session(s); |
998 return; | |
999 } | |
1000 | |
1001 if (c == s->connection) { | |
1002 pcf = ngx_mail_get_module_srv_conf(s, ngx_mail_proxy_module); | |
1003 ngx_add_timer(c->read, pcf->timeout); | |
1004 } | |
1005 } | |
1006 | |
1007 | |
1008 static void | |
1009 ngx_mail_proxy_upstream_error(ngx_mail_session_t *s) | |
1010 { | |
1011 if (s->proxy->upstream.connection) { | |
1012 ngx_log_debug1(NGX_LOG_DEBUG_MAIL, s->connection->log, 0, | |
1013 "close mail proxy connection: %d", | |
1014 s->proxy->upstream.connection->fd); | |
1015 | |
1016 ngx_close_connection(s->proxy->upstream.connection); | |
1017 } | |
1018 | |
1019 if (s->out.len == 0) { | |
1020 ngx_mail_session_internal_server_error(s); | |
1021 return; | |
1022 } | |
1023 | |
1024 s->quit = 1; | |
1025 ngx_mail_send(s->connection->write); | |
1026 } | |
1027 | |
1028 | |
1029 static void | |
1030 ngx_mail_proxy_internal_server_error(ngx_mail_session_t *s) | |
1031 { | |
1032 if (s->proxy->upstream.connection) { | |
1033 ngx_log_debug1(NGX_LOG_DEBUG_MAIL, s->connection->log, 0, | |
1034 "close mail proxy connection: %d", | |
1035 s->proxy->upstream.connection->fd); | |
1036 | |
1037 ngx_close_connection(s->proxy->upstream.connection); | |
1038 } | |
1039 | |
1040 ngx_mail_session_internal_server_error(s); | |
1041 } | |
1042 | |
1043 | |
1044 static void | |
1045 ngx_mail_proxy_close_session(ngx_mail_session_t *s) | |
1046 { | |
1047 if (s->proxy->upstream.connection) { | |
1048 ngx_log_debug1(NGX_LOG_DEBUG_MAIL, s->connection->log, 0, | |
1049 "close mail proxy connection: %d", | |
1050 s->proxy->upstream.connection->fd); | |
1051 | |
1052 ngx_close_connection(s->proxy->upstream.connection); | |
1053 } | |
1054 | |
1055 ngx_mail_close_connection(s->connection); | |
1056 } | |
1057 | |
1058 | |
1059 static void * | |
1060 ngx_mail_proxy_create_conf(ngx_conf_t *cf) | |
1061 { | |
1062 ngx_mail_proxy_conf_t *pcf; | |
1063 | |
1064 pcf = ngx_pcalloc(cf->pool, sizeof(ngx_mail_proxy_conf_t)); | |
1065 if (pcf == NULL) { | |
536 | 1066 return NULL; |
290 | 1067 } |
1068 | |
1069 pcf->enable = NGX_CONF_UNSET; | |
1070 pcf->pass_error_message = NGX_CONF_UNSET; | |
1071 pcf->xclient = NGX_CONF_UNSET; | |
1072 pcf->buffer_size = NGX_CONF_UNSET_SIZE; | |
1073 pcf->timeout = NGX_CONF_UNSET_MSEC; | |
1074 | |
1075 return pcf; | |
1076 } | |
1077 | |
1078 | |
1079 static char * | |
1080 ngx_mail_proxy_merge_conf(ngx_conf_t *cf, void *parent, void *child) | |
1081 { | |
1082 ngx_mail_proxy_conf_t *prev = parent; | |
1083 ngx_mail_proxy_conf_t *conf = child; | |
1084 | |
1085 ngx_conf_merge_value(conf->enable, prev->enable, 0); | |
1086 ngx_conf_merge_value(conf->pass_error_message, prev->pass_error_message, 0); | |
1087 ngx_conf_merge_value(conf->xclient, prev->xclient, 1); | |
1088 ngx_conf_merge_size_value(conf->buffer_size, prev->buffer_size, | |
1089 (size_t) ngx_pagesize); | |
1090 ngx_conf_merge_msec_value(conf->timeout, prev->timeout, 24 * 60 * 60000); | |
1091 | |
1092 return NGX_CONF_OK; | |
1093 } |