76
|
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_imap.h>
|
|
12
|
|
13
|
|
14 typedef struct {
|
|
15 ngx_peers_t *peers;
|
|
16
|
|
17 ngx_msec_t timeout;
|
|
18
|
|
19 ngx_str_t host_header;
|
|
20 ngx_str_t uri;
|
|
21 } ngx_imap_auth_http_conf_t;
|
|
22
|
|
23
|
|
24 typedef struct ngx_imap_auth_http_ctx_s ngx_imap_auth_http_ctx_t;
|
|
25
|
|
26 typedef void (*ngx_imap_auth_http_handler_pt)(ngx_imap_session_t *s,
|
|
27 ngx_imap_auth_http_ctx_t *ctx);
|
|
28
|
|
29 struct ngx_imap_auth_http_ctx_s {
|
|
30 ngx_buf_t *request;
|
|
31 ngx_buf_t *response;
|
|
32 ngx_peer_connection_t peer;
|
|
33
|
|
34 ngx_imap_auth_http_handler_pt handler;
|
|
35
|
|
36 ngx_uint_t state;
|
|
37 ngx_uint_t hash; /* no needed ? */
|
|
38
|
|
39 u_char *header_name_start;
|
|
40 u_char *header_name_end;
|
|
41 u_char *header_start;
|
|
42 u_char *header_end;
|
|
43
|
|
44 ngx_str_t addr;
|
|
45 ngx_str_t port;
|
|
46 ngx_str_t err;
|
|
47
|
|
48 ngx_msec_t sleep;
|
|
49
|
|
50 ngx_peers_t *peers;
|
|
51 };
|
|
52
|
|
53
|
|
54 static void ngx_imap_auth_http_write_handler(ngx_event_t *wev);
|
|
55 static void ngx_imap_auth_http_read_handler(ngx_event_t *rev);
|
|
56 static void ngx_imap_auth_http_ignore_status_line(ngx_imap_session_t *s,
|
|
57 ngx_imap_auth_http_ctx_t *ctx);
|
|
58 static void ngx_imap_auth_http_process_headers(ngx_imap_session_t *s,
|
|
59 ngx_imap_auth_http_ctx_t *ctx);
|
|
60 static void ngx_imap_auth_sleep_handler(ngx_event_t *rev);
|
|
61 static ngx_int_t ngx_imap_auth_http_parse_header_line(ngx_imap_session_t *s,
|
|
62 ngx_imap_auth_http_ctx_t *ctx);
|
|
63 static void ngx_imap_auth_http_block_read(ngx_event_t *rev);
|
|
64 static void ngx_imap_auth_http_dummy_handler(ngx_event_t *ev);
|
|
65 static ngx_buf_t *ngx_imap_auth_http_create_request(ngx_imap_session_t *s,
|
|
66 ngx_imap_auth_http_conf_t *ahcf);
|
|
67
|
|
68 static void *ngx_imap_auth_http_create_conf(ngx_conf_t *cf);
|
|
69 static char *ngx_imap_auth_http_merge_conf(ngx_conf_t *cf, void *parent,
|
|
70 void *child);
|
|
71 static char *ngx_imap_auth_http(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
|
|
72
|
|
73
|
|
74 static ngx_command_t ngx_imap_auth_http_commands[] = {
|
|
75
|
|
76 { ngx_string("auth_http"),
|
|
77 NGX_IMAP_MAIN_CONF|NGX_IMAP_SRV_CONF|NGX_CONF_TAKE1,
|
|
78 ngx_imap_auth_http,
|
|
79 NGX_IMAP_SRV_CONF_OFFSET,
|
|
80 0,
|
|
81 NULL },
|
|
82
|
|
83 { ngx_string("auth_http_timeout"),
|
|
84 NGX_IMAP_MAIN_CONF|NGX_IMAP_SRV_CONF|NGX_CONF_TAKE1,
|
|
85 ngx_conf_set_msec_slot,
|
|
86 NGX_IMAP_SRV_CONF_OFFSET,
|
|
87 offsetof(ngx_imap_auth_http_conf_t, timeout),
|
|
88 NULL },
|
|
89
|
|
90 ngx_null_command
|
|
91 };
|
|
92
|
|
93
|
|
94 static ngx_imap_module_t ngx_imap_auth_http_module_ctx = {
|
|
95 NULL, /* create main configuration */
|
|
96 NULL, /* init main configuration */
|
|
97
|
|
98 ngx_imap_auth_http_create_conf, /* create server configuration */
|
|
99 ngx_imap_auth_http_merge_conf /* merge server configuration */
|
|
100 };
|
|
101
|
|
102
|
|
103 ngx_module_t ngx_imap_auth_http_module = {
|
|
104 NGX_MODULE_V1,
|
|
105 &ngx_imap_auth_http_module_ctx, /* module context */
|
|
106 ngx_imap_auth_http_commands, /* module directives */
|
|
107 NGX_IMAP_MODULE, /* module type */
|
|
108 NULL, /* init module */
|
|
109 NULL /* init process */
|
|
110 };
|
|
111
|
|
112
|
|
113 static char *ngx_imap_auth_http_protocol[] = { "pop3", "imap" };
|
|
114
|
|
115
|
|
116 void
|
|
117 ngx_imap_auth_http_init(ngx_imap_session_t *s)
|
|
118 {
|
|
119 ngx_int_t rc;
|
|
120 ngx_imap_auth_http_ctx_t *ctx;
|
|
121 ngx_imap_auth_http_conf_t *ahcf;
|
|
122
|
|
123 ctx = ngx_pcalloc(s->connection->pool, sizeof(ngx_imap_auth_http_ctx_t));
|
|
124 if (ctx == NULL) {
|
|
125 ngx_imap_session_internal_server_error(s);
|
|
126 return;
|
|
127 }
|
|
128
|
|
129 ahcf = ngx_imap_get_module_srv_conf(s, ngx_imap_auth_http_module);
|
|
130
|
|
131 ctx->request = ngx_imap_auth_http_create_request(s, ahcf);
|
|
132 if (ctx->request == NULL) {
|
|
133 ngx_imap_session_internal_server_error(s);
|
|
134 return;
|
|
135 }
|
|
136
|
|
137 ngx_imap_set_ctx(s, ctx, ngx_imap_auth_http_module);
|
|
138
|
|
139 ctx->peer.peers = ahcf->peers;
|
|
140 ctx->peer.log = s->connection->log;
|
|
141 ctx->peer.log_error = NGX_ERROR_ERR;
|
|
142
|
|
143 rc = ngx_event_connect_peer(&ctx->peer);
|
|
144
|
|
145 if (rc == NGX_ERROR) {
|
|
146 ngx_imap_session_internal_server_error(s);
|
|
147 return;
|
|
148 }
|
|
149
|
|
150 ctx->peer.connection->data = s;
|
|
151 ctx->peer.connection->pool = s->connection->pool;
|
|
152
|
|
153 s->connection->read->handler = ngx_imap_auth_http_block_read;
|
|
154 ctx->peer.connection->read->handler = ngx_imap_auth_http_read_handler;
|
|
155 ctx->peer.connection->write->handler = ngx_imap_auth_http_write_handler;
|
|
156
|
|
157 ctx->handler = ngx_imap_auth_http_ignore_status_line;
|
|
158
|
|
159 if (rc == NGX_OK) {
|
|
160 ngx_imap_auth_http_write_handler(ctx->peer.connection->write);
|
|
161 return;
|
|
162 }
|
|
163
|
|
164 ngx_add_timer(ctx->peer.connection->read, ahcf->timeout);
|
|
165 ngx_add_timer(ctx->peer.connection->write, ahcf->timeout);
|
|
166 }
|
|
167
|
|
168
|
|
169 static void
|
|
170 ngx_imap_auth_http_write_handler(ngx_event_t *wev)
|
|
171 {
|
|
172 ssize_t n, size;
|
|
173 ngx_connection_t *c;
|
|
174 ngx_imap_session_t *s;
|
|
175 ngx_imap_auth_http_ctx_t *ctx;
|
|
176 ngx_imap_auth_http_conf_t *ahcf;
|
|
177
|
|
178 c = wev->data;
|
|
179 s = c->data;
|
|
180
|
|
181 ctx = ngx_imap_get_module_ctx(s, ngx_imap_auth_http_module);
|
|
182
|
|
183 ngx_log_debug0(NGX_LOG_DEBUG_IMAP, wev->log, 0,
|
|
184 "imap auth http write handler");
|
|
185
|
|
186 if (wev->timedout) {
|
|
187 ngx_log_error(NGX_LOG_ERR, wev->log, NGX_ETIMEDOUT,
|
|
188 "auth http server timed out");
|
|
189 ngx_close_connection(ctx->peer.connection);
|
|
190 ngx_imap_session_internal_server_error(s);
|
|
191 return;
|
|
192 }
|
|
193
|
|
194 size = ctx->request->last - ctx->request->pos;
|
|
195
|
|
196 n = ngx_send(c, ctx->request->pos, size);
|
|
197
|
|
198 if (n == NGX_ERROR) {
|
|
199 ngx_close_connection(ctx->peer.connection);
|
|
200 ngx_imap_session_internal_server_error(s);
|
|
201 return;
|
|
202 }
|
|
203
|
|
204 if (n > 0) {
|
|
205 ctx->request->pos += n;
|
|
206
|
|
207 if (n == size) {
|
|
208 wev->handler = ngx_imap_auth_http_dummy_handler;
|
|
209
|
|
210 if (wev->timer_set) {
|
|
211 ngx_del_timer(wev);
|
|
212 }
|
|
213
|
|
214 return;
|
|
215 }
|
|
216 }
|
|
217
|
|
218 if (!wev->timer_set) {
|
|
219 ahcf = ngx_imap_get_module_srv_conf(s, ngx_imap_auth_http_module);
|
|
220 ngx_add_timer(wev, ahcf->timeout);
|
|
221 }
|
|
222 }
|
|
223
|
|
224
|
|
225 static void
|
|
226 ngx_imap_auth_http_read_handler(ngx_event_t *rev)
|
|
227 {
|
|
228 ssize_t n, size;
|
|
229 ngx_connection_t *c;
|
|
230 ngx_imap_session_t *s;
|
|
231 ngx_imap_auth_http_ctx_t *ctx;
|
|
232
|
|
233 c = rev->data;
|
|
234 s = c->data;
|
|
235
|
|
236 ngx_log_debug0(NGX_LOG_DEBUG_IMAP, rev->log, 0,
|
|
237 "imap auth http read handler");
|
|
238
|
|
239 ctx = ngx_imap_get_module_ctx(s, ngx_imap_auth_http_module);
|
|
240
|
|
241 if (rev->timedout) {
|
|
242 ngx_log_error(NGX_LOG_ERR, rev->log, NGX_ETIMEDOUT,
|
|
243 "auth http server timed out");
|
|
244 ngx_close_connection(ctx->peer.connection);
|
|
245 ngx_imap_session_internal_server_error(s);
|
|
246 return;
|
|
247 }
|
|
248
|
|
249 if (ctx->response == NULL) {
|
|
250 ctx->response = ngx_create_temp_buf(s->connection->pool, 1024);
|
|
251 if (ctx->response == NULL) {
|
|
252 ngx_close_connection(ctx->peer.connection);
|
|
253 ngx_imap_session_internal_server_error(s);
|
|
254 return;
|
|
255 }
|
|
256 }
|
|
257
|
|
258 size = ctx->response->end - ctx->response->last;
|
|
259
|
|
260 n = ngx_recv(c, ctx->response->pos, size);
|
|
261
|
|
262 if (n > 0) {
|
|
263 ctx->response->last += n;
|
|
264
|
|
265 ctx->handler(s, ctx);
|
|
266 return;
|
|
267 }
|
|
268
|
|
269 if (n == NGX_AGAIN) {
|
|
270 return;
|
|
271 }
|
|
272
|
|
273 ngx_close_connection(ctx->peer.connection);
|
|
274 ngx_imap_session_internal_server_error(s);
|
|
275 }
|
|
276
|
|
277
|
|
278 static void
|
|
279 ngx_imap_auth_http_ignore_status_line(ngx_imap_session_t *s,
|
|
280 ngx_imap_auth_http_ctx_t *ctx)
|
|
281 {
|
|
282 u_char *p, ch;
|
|
283 enum {
|
|
284 sw_start = 0,
|
|
285 sw_H,
|
|
286 sw_HT,
|
|
287 sw_HTT,
|
|
288 sw_HTTP,
|
|
289 sw_skip,
|
|
290 sw_almost_done
|
|
291 } state;
|
|
292
|
|
293 ngx_log_debug0(NGX_LOG_DEBUG_IMAP, s->connection->log, 0,
|
|
294 "imap auth http process status line");
|
|
295
|
|
296 state = ctx->state;
|
|
297
|
|
298 for (p = ctx->response->pos; p < ctx->response->last; p++) {
|
|
299 ch = *p;
|
|
300
|
|
301 switch (state) {
|
|
302
|
|
303 /* "HTTP/" */
|
|
304 case sw_start:
|
|
305 if (ch == 'H') {
|
|
306 state = sw_H;
|
|
307 break;
|
|
308 }
|
|
309 goto next;
|
|
310
|
|
311 case sw_H:
|
|
312 if (ch == 'T') {
|
|
313 state = sw_HT;
|
|
314 break;
|
|
315 }
|
|
316 goto next;
|
|
317
|
|
318 case sw_HT:
|
|
319 if (ch == 'T') {
|
|
320 state = sw_HTT;
|
|
321 break;
|
|
322 }
|
|
323 goto next;
|
|
324
|
|
325 case sw_HTT:
|
|
326 if (ch == 'P') {
|
|
327 state = sw_HTTP;
|
|
328 break;
|
|
329 }
|
|
330 goto next;
|
|
331
|
|
332 case sw_HTTP:
|
|
333 if (ch == '/') {
|
|
334 state = sw_skip;
|
|
335 break;
|
|
336 }
|
|
337 goto next;
|
|
338
|
|
339 /* any text until end of line */
|
|
340 case sw_skip:
|
|
341 switch (ch) {
|
|
342 case CR:
|
|
343 state = sw_almost_done;
|
|
344
|
|
345 break;
|
|
346 case LF:
|
|
347 goto done;
|
|
348 }
|
|
349 break;
|
|
350
|
|
351 /* end of status line */
|
|
352 case sw_almost_done:
|
|
353 if (ch == LF) {
|
|
354 goto done;
|
|
355 }
|
|
356
|
|
357 ngx_log_error(NGX_LOG_ERR, s->connection->log, 0,
|
|
358 "auth http server sent invalid response");
|
|
359 ngx_close_connection(ctx->peer.connection);
|
|
360 ngx_imap_session_internal_server_error(s);
|
|
361 return;
|
|
362 }
|
|
363 }
|
|
364
|
|
365 ctx->response->pos = p;
|
|
366 ctx->state = state;
|
|
367
|
|
368 return;
|
|
369
|
|
370 next:
|
|
371
|
|
372 p = ctx->response->start - 1;
|
|
373
|
|
374 done:
|
|
375
|
|
376 ctx->response->pos = p + 1;
|
|
377 ctx->state = 0;
|
|
378 ctx->handler = ngx_imap_auth_http_process_headers;
|
|
379 ctx->handler(s, ctx);
|
|
380 }
|
|
381
|
|
382
|
|
383 static void
|
|
384 ngx_imap_auth_http_process_headers(ngx_imap_session_t *s,
|
|
385 ngx_imap_auth_http_ctx_t *ctx)
|
|
386 {
|
|
387 u_char *p;
|
|
388 size_t len, size;
|
|
389 ngx_int_t rc, port, n;
|
|
390 struct sockaddr_in *sin;
|
|
391
|
|
392 ngx_log_debug0(NGX_LOG_DEBUG_IMAP, s->connection->log, 0,
|
|
393 "imap auth http process headers");
|
|
394
|
|
395 for ( ;; ) {
|
|
396 rc = ngx_imap_auth_http_parse_header_line(s, ctx);
|
|
397
|
|
398 if (rc == NGX_OK) {
|
|
399
|
|
400 #if (NGX_DEBUG)
|
|
401 {
|
|
402 ngx_str_t key, value;
|
|
403
|
|
404 key.len = ctx->header_name_end - ctx->header_name_start;
|
|
405 key.data = ctx->header_name_start;
|
|
406 value.len = ctx->header_end - ctx->header_start;
|
|
407 value.data = ctx->header_start;
|
|
408
|
|
409 ngx_log_debug2(NGX_LOG_DEBUG_IMAP, s->connection->log, 0,
|
|
410 "auth http header: \"%V: %V\"",
|
|
411 &key, &value);
|
|
412 }
|
|
413 #endif
|
|
414
|
|
415 len = ctx->header_name_end - ctx->header_name_start;
|
|
416
|
|
417 if (len == sizeof("Auth-Status") - 1
|
|
418 && ngx_strncasecmp(ctx->header_name_start, "Auth-Status",
|
|
419 sizeof("Auth-Status") - 1) == 0)
|
|
420 {
|
|
421 len = ctx->header_end - ctx->header_start;
|
|
422
|
|
423 if (len == 2
|
|
424 && ctx->header_start[0] == 'O'
|
|
425 && ctx->header_start[1] == 'K')
|
|
426 {
|
|
427 continue;
|
|
428 }
|
|
429
|
|
430 if (s->protocol == NGX_IMAP_POP3_PROTOCOL) {
|
|
431 size = sizeof("-ERR") - 1 + len + sizeof(CRLF) - 1;
|
|
432
|
|
433 } else {
|
|
434 size = s->tag.len + sizeof("NO") - 1 + len
|
|
435 + sizeof(CRLF) - 1;
|
|
436 }
|
|
437
|
|
438 p = ngx_pcalloc(s->connection->pool, size);
|
|
439 if (p == NULL) {
|
|
440 ngx_imap_session_internal_server_error(s);
|
|
441 return;
|
|
442 }
|
|
443
|
|
444 ctx->err.data = p;
|
|
445
|
|
446 if (s->protocol == NGX_IMAP_POP3_PROTOCOL) {
|
|
447 *p++ = '-'; *p++ = 'E'; *p++ = 'R'; *p++ = 'R';
|
|
448
|
|
449 } else {
|
|
450 p = ngx_cpymem(p, s->tag.data, s->tag.len);
|
|
451 *p++ = 'N'; *p++ = 'O';
|
|
452 }
|
|
453
|
|
454 *p++ = ' ';
|
|
455 p = ngx_cpymem(p, ctx->header_start, len);
|
|
456 *p++ = CR; *p++ = LF;
|
|
457
|
|
458 ctx->err.len = p - ctx->err.data;
|
|
459
|
|
460 continue;
|
|
461 }
|
|
462
|
|
463 if (len == sizeof("Auth-Server") - 1
|
|
464 && ngx_strncasecmp(ctx->header_name_start, "Auth-Server",
|
|
465 sizeof("Auth-Server") - 1) == 0)
|
|
466 {
|
|
467 ctx->addr.len = ctx->header_end - ctx->header_start;
|
|
468 ctx->addr.data = ctx->header_start;
|
|
469
|
|
470 continue;
|
|
471 }
|
|
472
|
|
473 if (len == sizeof("Auth-Port") - 1
|
|
474 && ngx_strncasecmp(ctx->header_name_start, "Auth-Port",
|
|
475 sizeof("Auth-Port") - 1) == 0)
|
|
476 {
|
|
477 ctx->port.len = ctx->header_end - ctx->header_start;
|
|
478 ctx->port.data = ctx->header_start;
|
|
479
|
|
480 continue;
|
|
481 }
|
|
482
|
|
483 if (len == sizeof("Auth-User") - 1
|
|
484 && ngx_strncasecmp(ctx->header_name_start, "Auth-User",
|
|
485 sizeof("Auth-User") - 1) == 0)
|
|
486 {
|
|
487 s->login.len = ctx->header_end - ctx->header_start;
|
|
488 s->login.data = ctx->header_start;
|
|
489
|
|
490 continue;
|
|
491 }
|
|
492
|
|
493 if (len == sizeof("Auth-Wait") - 1
|
|
494 && ngx_strncasecmp(ctx->header_name_start, "Auth-Wait",
|
|
495 sizeof("Auth-Wait") - 1) == 0)
|
|
496 {
|
|
497 n = ngx_atoi(ctx->header_start,
|
|
498 ctx->header_end - ctx->header_start);
|
|
499
|
|
500 if (n != NGX_ERROR) {
|
|
501 ctx->sleep = n;
|
|
502 }
|
|
503
|
|
504 continue;
|
|
505 }
|
|
506
|
|
507 /* ignore other headers */
|
|
508
|
|
509 continue;
|
|
510 }
|
|
511
|
|
512 if (rc == NGX_DONE) {
|
|
513 ngx_log_debug0(NGX_LOG_DEBUG_IMAP, s->connection->log, 0,
|
|
514 "auth http header done");
|
|
515
|
|
516 ngx_close_connection(ctx->peer.connection);
|
|
517
|
|
518 if (ctx->err.len) {
|
88
|
519 s->out = ctx->err;
|
76
|
520
|
|
521 if (ctx->sleep == 0) {
|
88
|
522 s->quit = 1;
|
76
|
523 }
|
|
524
|
88
|
525 ngx_imap_send(s->connection->write);
|
|
526
|
76
|
527 ngx_add_timer(s->connection->read, ctx->sleep * 1000);
|
|
528
|
|
529 s->connection->read->handler = ngx_imap_auth_sleep_handler;
|
|
530
|
|
531 return;
|
|
532 }
|
|
533
|
|
534 if (ctx->addr.len == 0 || ctx->port.len == 0) {
|
|
535 ngx_log_error(NGX_LOG_ERR, s->connection->log, 0,
|
|
536 "auth http server did not send server or port");
|
|
537 ngx_imap_session_internal_server_error(s);
|
|
538 return;
|
|
539 }
|
|
540
|
|
541 ctx->peers = ngx_pcalloc(s->connection->pool, sizeof(ngx_peers_t));
|
|
542 if (ctx->peers == NULL) {
|
|
543 ngx_imap_session_internal_server_error(s);
|
|
544 return;
|
|
545 }
|
|
546
|
|
547 sin = ngx_pcalloc(s->connection->pool, sizeof(struct sockaddr_in));
|
|
548 if (sin == NULL) {
|
|
549 ngx_imap_session_internal_server_error(s);
|
|
550 return;
|
|
551 }
|
|
552
|
|
553 sin->sin_family = AF_INET;
|
|
554
|
|
555 port = ngx_atoi(ctx->port.data, ctx->port.len);
|
|
556 if (port == NGX_ERROR || port < 1 || port > 65536) {
|
|
557 ngx_log_error(NGX_LOG_ERR, s->connection->log, 0,
|
|
558 "auth http server sent invalid server "
|
|
559 "port:\"%V\"", &ctx->port);
|
|
560 ngx_imap_session_internal_server_error(s);
|
|
561 return;
|
|
562 }
|
|
563
|
|
564 sin->sin_port = htons((in_port_t) port);
|
|
565
|
|
566 ctx->addr.data[ctx->addr.len] = '\0';
|
|
567 sin->sin_addr.s_addr = inet_addr((char *) ctx->addr.data);
|
|
568 if (sin->sin_addr.s_addr == INADDR_NONE) {
|
|
569 ngx_log_error(NGX_LOG_ERR, s->connection->log, 0,
|
|
570 "auth http server sent invalid server "
|
|
571 "address:\"%V\"", &ctx->addr);
|
|
572 ngx_imap_session_internal_server_error(s);
|
|
573 return;
|
|
574 }
|
|
575
|
|
576 ctx->peers->number = 1;
|
|
577
|
|
578 ctx->peers->peer[0].sockaddr = (struct sockaddr *) sin;
|
|
579 ctx->peers->peer[0].socklen = sizeof(struct sockaddr_in);
|
|
580
|
|
581 len = ctx->addr.len + 1 + ctx->port.len;
|
|
582
|
|
583 ctx->peers->peer[0].name.len = len;
|
|
584
|
|
585 ctx->peers->peer[0].name.data = ngx_palloc(s->connection->pool,
|
|
586 len);
|
|
587 if (ctx->peers->peer[0].name.data == NULL) {
|
|
588 ngx_imap_session_internal_server_error(s);
|
|
589 return;
|
|
590 }
|
|
591
|
|
592 len = ctx->addr.len;
|
|
593
|
|
594 ngx_memcpy(ctx->peers->peer[0].name.data, ctx->addr.data, len);
|
|
595
|
|
596 ctx->peers->peer[0].name.data[len++] = ':';
|
|
597
|
|
598 ngx_memcpy(ctx->peers->peer[0].name.data + len,
|
|
599 ctx->port.data, ctx->port.len);
|
|
600
|
|
601 ctx->peers->peer[0].uri_separator = "";
|
|
602
|
|
603 ngx_imap_proxy_init(s, ctx->peers);
|
|
604
|
|
605 return;
|
|
606 }
|
|
607
|
|
608 if (rc == NGX_AGAIN ) {
|
|
609 return;
|
|
610 }
|
|
611
|
|
612 /* rc == NGX_ERROR */
|
|
613
|
|
614 ngx_log_error(NGX_LOG_ERR, s->connection->log, 0,
|
|
615 "auth http server sent invalid header in response");
|
|
616 ngx_close_connection(ctx->peer.connection);
|
|
617 ngx_imap_session_internal_server_error(s);
|
|
618
|
|
619 return;
|
|
620 }
|
|
621 }
|
|
622
|
|
623
|
|
624 static void
|
|
625 ngx_imap_auth_sleep_handler(ngx_event_t *rev)
|
|
626 {
|
|
627 ngx_connection_t *c;
|
|
628 ngx_imap_session_t *s;
|
|
629
|
|
630 ngx_log_debug0(NGX_LOG_DEBUG_IMAP, rev->log, 0, "imap auth sleep handler");
|
|
631
|
|
632 c = rev->data;
|
|
633 s = c->data;
|
|
634
|
|
635 if (rev->timedout) {
|
|
636
|
|
637 rev->timedout = 0;
|
|
638
|
|
639 if (s->protocol == NGX_IMAP_POP3_PROTOCOL) {
|
|
640 s->imap_state = ngx_pop3_start;
|
|
641 s->connection->read->handler = ngx_pop3_auth_state;
|
|
642
|
|
643 } else {
|
|
644 s->imap_state = ngx_imap_start;
|
|
645 s->connection->read->handler = ngx_imap_auth_state;
|
|
646 }
|
|
647
|
|
648 if (rev->ready) {
|
|
649 s->connection->read->handler(rev);
|
|
650 return;
|
|
651 }
|
|
652
|
|
653 if (ngx_handle_read_event(rev, 0) == NGX_ERROR) {
|
|
654 ngx_imap_close_connection(s->connection);
|
|
655 }
|
|
656
|
|
657 return;
|
|
658 }
|
|
659
|
|
660 if (rev->active) {
|
|
661 if (ngx_handle_read_event(rev, 0) == NGX_ERROR) {
|
|
662 ngx_imap_close_connection(s->connection);
|
|
663 }
|
|
664 }
|
|
665 }
|
|
666
|
|
667
|
|
668 static ngx_int_t
|
|
669 ngx_imap_auth_http_parse_header_line(ngx_imap_session_t *s,
|
|
670 ngx_imap_auth_http_ctx_t *ctx)
|
|
671 {
|
|
672 u_char c, ch, *p;
|
|
673 ngx_uint_t hash;
|
|
674 enum {
|
|
675 sw_start = 0,
|
|
676 sw_name,
|
|
677 sw_space_before_value,
|
|
678 sw_value,
|
|
679 sw_space_after_value,
|
|
680 sw_almost_done,
|
|
681 sw_header_almost_done
|
|
682 } state;
|
|
683
|
|
684 state = ctx->state;
|
|
685 hash = ctx->hash;
|
|
686
|
|
687 for (p = ctx->response->pos; p < ctx->response->last; p++) {
|
|
688 ch = *p;
|
|
689
|
|
690 switch (state) {
|
|
691
|
|
692 /* first char */
|
|
693 case sw_start:
|
|
694
|
|
695 switch (ch) {
|
|
696 case CR:
|
|
697 ctx->header_end = p;
|
|
698 state = sw_header_almost_done;
|
|
699 break;
|
|
700 case LF:
|
|
701 ctx->header_end = p;
|
|
702 goto header_done;
|
|
703 default:
|
|
704 state = sw_name;
|
|
705 ctx->header_name_start = p;
|
|
706
|
|
707 c = (u_char) (ch | 0x20);
|
|
708 if (c >= 'a' && c <= 'z') {
|
|
709 hash = c;
|
|
710 break;
|
|
711 }
|
|
712
|
|
713 if (ch >= '0' && ch <= '9') {
|
|
714 hash = ch;
|
|
715 break;
|
|
716 }
|
|
717
|
|
718 return NGX_ERROR;
|
|
719 }
|
|
720 break;
|
|
721
|
|
722 /* header name */
|
|
723 case sw_name:
|
|
724 c = (u_char) (ch | 0x20);
|
|
725 if (c >= 'a' && c <= 'z') {
|
|
726 hash += c;
|
|
727 break;
|
|
728 }
|
|
729
|
|
730 if (ch == ':') {
|
|
731 ctx->header_name_end = p;
|
|
732 state = sw_space_before_value;
|
|
733 break;
|
|
734 }
|
|
735
|
|
736 if (ch == '-') {
|
|
737 hash += ch;
|
|
738 break;
|
|
739 }
|
|
740
|
|
741 if (ch >= '0' && ch <= '9') {
|
|
742 hash += ch;
|
|
743 break;
|
|
744 }
|
|
745
|
|
746 if (ch == CR) {
|
|
747 ctx->header_name_end = p;
|
|
748 ctx->header_start = p;
|
|
749 ctx->header_end = p;
|
|
750 state = sw_almost_done;
|
|
751 break;
|
|
752 }
|
|
753
|
|
754 if (ch == LF) {
|
|
755 ctx->header_name_end = p;
|
|
756 ctx->header_start = p;
|
|
757 ctx->header_end = p;
|
|
758 goto done;
|
|
759 }
|
|
760
|
|
761 return NGX_ERROR;
|
|
762
|
|
763 /* space* before header value */
|
|
764 case sw_space_before_value:
|
|
765 switch (ch) {
|
|
766 case ' ':
|
|
767 break;
|
|
768 case CR:
|
|
769 ctx->header_start = p;
|
|
770 ctx->header_end = p;
|
|
771 state = sw_almost_done;
|
|
772 break;
|
|
773 case LF:
|
|
774 ctx->header_start = p;
|
|
775 ctx->header_end = p;
|
|
776 goto done;
|
|
777 default:
|
|
778 ctx->header_start = p;
|
|
779 state = sw_value;
|
|
780 break;
|
|
781 }
|
|
782 break;
|
|
783
|
|
784 /* header value */
|
|
785 case sw_value:
|
|
786 switch (ch) {
|
|
787 case ' ':
|
|
788 ctx->header_end = p;
|
|
789 state = sw_space_after_value;
|
|
790 break;
|
|
791 case CR:
|
|
792 ctx->header_end = p;
|
|
793 state = sw_almost_done;
|
|
794 break;
|
|
795 case LF:
|
|
796 ctx->header_end = p;
|
|
797 goto done;
|
|
798 }
|
|
799 break;
|
|
800
|
|
801 /* space* before end of header line */
|
|
802 case sw_space_after_value:
|
|
803 switch (ch) {
|
|
804 case ' ':
|
|
805 break;
|
|
806 case CR:
|
|
807 state = sw_almost_done;
|
|
808 break;
|
|
809 case LF:
|
|
810 goto done;
|
|
811 default:
|
|
812 state = sw_value;
|
|
813 break;
|
|
814 }
|
|
815 break;
|
|
816
|
|
817 /* end of header line */
|
|
818 case sw_almost_done:
|
|
819 switch (ch) {
|
|
820 case LF:
|
|
821 goto done;
|
|
822 default:
|
|
823 return NGX_ERROR;
|
|
824 }
|
|
825
|
|
826 /* end of header */
|
|
827 case sw_header_almost_done:
|
|
828 switch (ch) {
|
|
829 case LF:
|
|
830 goto header_done;
|
|
831 default:
|
|
832 return NGX_ERROR;
|
|
833 }
|
|
834 }
|
|
835 }
|
|
836
|
|
837 ctx->response->pos = p;
|
|
838 ctx->state = state;
|
|
839 ctx->hash = hash;
|
|
840
|
|
841 return NGX_AGAIN;
|
|
842
|
|
843 done:
|
|
844
|
|
845 ctx->response->pos = p + 1;
|
|
846 ctx->state = sw_start;
|
|
847 ctx->hash = hash;
|
|
848
|
|
849 return NGX_OK;
|
|
850
|
|
851 header_done:
|
|
852
|
|
853 ctx->response->pos = p + 1;
|
|
854 ctx->state = sw_start;
|
|
855
|
|
856 return NGX_DONE;
|
|
857 }
|
|
858
|
|
859
|
|
860 static void
|
|
861 ngx_imap_auth_http_block_read(ngx_event_t *rev)
|
|
862 {
|
|
863 ngx_connection_t *c;
|
|
864 ngx_imap_session_t *s;
|
|
865 ngx_imap_auth_http_ctx_t *ctx;
|
|
866
|
|
867 ngx_log_debug0(NGX_LOG_DEBUG_IMAP, rev->log, 0,
|
|
868 "imap auth http block read");
|
|
869
|
|
870 if (ngx_handle_read_event(rev, 0) == NGX_ERROR) {
|
|
871 c = rev->data;
|
|
872 s = c->data;
|
|
873
|
|
874 ctx = ngx_imap_get_module_ctx(s, ngx_imap_auth_http_module);
|
|
875
|
|
876 ngx_close_connection(ctx->peer.connection);
|
|
877 ngx_imap_session_internal_server_error(s);
|
|
878 }
|
|
879 }
|
|
880
|
|
881
|
|
882 static void
|
|
883 ngx_imap_auth_http_dummy_handler(ngx_event_t *ev)
|
|
884 {
|
|
885 ngx_log_debug0(NGX_LOG_DEBUG_IMAP, ev->log, 0,
|
|
886 "imap auth http dummy handler");
|
|
887 }
|
|
888
|
|
889
|
|
890 static ngx_buf_t *
|
|
891 ngx_imap_auth_http_create_request(ngx_imap_session_t *s,
|
|
892 ngx_imap_auth_http_conf_t *ahcf)
|
|
893 {
|
|
894 size_t len;
|
|
895 ngx_buf_t *b;
|
|
896
|
|
897 len = sizeof("GET ") - 1 + ahcf->uri.len + sizeof(" HTTP/1.0" CRLF) - 1
|
|
898 + sizeof("Host: ") - 1 + ahcf->host_header.len + sizeof(CRLF) - 1
|
|
899 + sizeof("Auth-Method: plain" CRLF) - 1
|
|
900 + sizeof("Auth-User: ") - 1 + s->login.len + sizeof(CRLF) - 1
|
|
901 + sizeof("Auth-Pass: ") - 1 + s->passwd.len + sizeof(CRLF) - 1
|
|
902 + sizeof("Auth-Protocol: imap" CRLF) - 1
|
|
903 + sizeof("Auth-Login-Attempt: ") - 1 + NGX_INT_T_LEN
|
|
904 + sizeof(CRLF) - 1
|
|
905 + sizeof("Client-IP: ") - 1 + s->connection->addr_text.len
|
|
906 + sizeof(CRLF) - 1
|
|
907 + sizeof(CRLF) - 1;
|
|
908
|
|
909 b = ngx_create_temp_buf(s->connection->pool, len);
|
|
910 if (b == NULL) {
|
|
911 return NULL;
|
|
912 }
|
|
913
|
|
914 b->last = ngx_cpymem(b->last, "GET ", sizeof("GET ") - 1);
|
|
915 b->last = ngx_cpymem(b->last, ahcf->uri.data, ahcf->uri.len);
|
|
916 b->last = ngx_cpymem(b->last, " HTTP/1.0" CRLF,
|
|
917 sizeof(" HTTP/1.0" CRLF) - 1);
|
|
918
|
|
919 b->last = ngx_cpymem(b->last, "Host: ", sizeof("Host: ") - 1);
|
|
920 b->last = ngx_cpymem(b->last, ahcf->host_header.data,
|
|
921 ahcf->host_header.len);
|
|
922 *b->last++ = CR; *b->last++ = LF;
|
|
923
|
|
924 b->last = ngx_cpymem(b->last, "Auth-Method: plain" CRLF,
|
|
925 sizeof("Auth-Method: plain" CRLF) - 1);
|
|
926
|
|
927 b->last = ngx_cpymem(b->last, "Auth-User: ", sizeof("Auth-User: ") - 1);
|
|
928 b->last = ngx_cpymem(b->last, s->login.data, s->login.len);
|
|
929 *b->last++ = CR; *b->last++ = LF;
|
|
930
|
|
931 b->last = ngx_cpymem(b->last, "Auth-Pass: ", sizeof("Auth-Pass: ") - 1);
|
|
932 b->last = ngx_cpymem(b->last, s->passwd.data, s->passwd.len);
|
|
933 *b->last++ = CR; *b->last++ = LF;
|
|
934
|
|
935 b->last = ngx_cpymem(b->last, "Auth-Protocol: ",
|
|
936 sizeof("Auth-Protocol: ") - 1);
|
|
937 b->last = ngx_cpymem(b->last, ngx_imap_auth_http_protocol[s->protocol],
|
|
938 sizeof("imap") - 1);
|
|
939 *b->last++ = CR; *b->last++ = LF;
|
|
940
|
|
941 b->last = ngx_sprintf(b->last, "Auth-Login-Attempt: %ui" CRLF,
|
|
942 s->login_attempt);
|
|
943
|
|
944 b->last = ngx_cpymem(b->last, "Client-IP: ", sizeof("Client-IP: ") - 1);
|
|
945 b->last = ngx_cpymem(b->last, s->connection->addr_text.data,
|
|
946 s->connection->addr_text.len);
|
|
947 *b->last++ = CR; *b->last++ = LF;
|
|
948
|
|
949 /* add "\r\n" at the header end */
|
|
950 *b->last++ = CR; *b->last++ = LF;
|
|
951
|
|
952 #if (NGX_DEBUG)
|
|
953 {
|
|
954 ngx_str_t l;
|
|
955
|
|
956 l.len = b->last - b->pos;
|
|
957 l.data = b->pos;
|
|
958 ngx_log_debug1(NGX_LOG_DEBUG_IMAP, s->connection->log, 0,
|
|
959 "imap auth http header:\n\"%V\"", &l);
|
|
960 }
|
|
961 #endif
|
|
962
|
|
963 return b;
|
|
964 }
|
|
965
|
|
966
|
|
967 static void *
|
|
968 ngx_imap_auth_http_create_conf(ngx_conf_t *cf)
|
|
969 {
|
|
970 ngx_imap_auth_http_conf_t *ahcf;
|
|
971
|
|
972 ahcf = ngx_pcalloc(cf->pool, sizeof(ngx_imap_auth_http_conf_t));
|
|
973 if (ahcf == NULL) {
|
|
974 return NGX_CONF_ERROR;
|
|
975 }
|
|
976
|
|
977 ahcf->timeout = NGX_CONF_UNSET_MSEC;
|
|
978
|
|
979 return ahcf;
|
|
980 }
|
|
981
|
|
982
|
|
983 static char *
|
|
984 ngx_imap_auth_http_merge_conf(ngx_conf_t *cf, void *parent, void *child)
|
|
985 {
|
|
986 ngx_imap_auth_http_conf_t *prev = parent;
|
|
987 ngx_imap_auth_http_conf_t *conf = child;
|
|
988
|
|
989 if (conf->peers == NULL) {
|
|
990 conf->peers = prev->peers;
|
|
991 conf->host_header = prev->host_header;
|
|
992 conf->uri = prev->uri;
|
|
993 }
|
|
994
|
|
995 ngx_conf_merge_msec_value(conf->timeout, prev->timeout, 60000);
|
|
996
|
|
997 return NGX_CONF_OK;
|
|
998 }
|
|
999
|
|
1000
|
|
1001 static char *
|
|
1002 ngx_imap_auth_http(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
1003 {
|
|
1004 ngx_imap_auth_http_conf_t *ahcf = conf;
|
|
1005
|
|
1006 ngx_uint_t i;
|
|
1007 ngx_str_t *value, *url;
|
|
1008 ngx_inet_upstream_t inet_upstream;
|
|
1009 #if (NGX_HAVE_UNIX_DOMAIN)
|
|
1010 ngx_unix_domain_upstream_t unix_upstream;
|
|
1011 #endif
|
|
1012
|
|
1013 value = cf->args->elts;
|
|
1014
|
|
1015 url = &value[1];
|
|
1016
|
|
1017 if (ngx_strncasecmp(url->data, "unix:", 5) == 0) {
|
|
1018
|
|
1019 #if (NGX_HAVE_UNIX_DOMAIN)
|
|
1020
|
|
1021 ngx_memzero(&unix_upstream, sizeof(ngx_unix_domain_upstream_t));
|
|
1022
|
|
1023 unix_upstream.name = *url;
|
|
1024 unix_upstream.url = *url;
|
|
1025 unix_upstream.uri_part = 1;
|
|
1026
|
|
1027 ahcf->peers = ngx_unix_upstream_parse(cf, &unix_upstream);
|
|
1028 if (ahcf->peers == NULL) {
|
|
1029 return NGX_CONF_ERROR;
|
|
1030 }
|
|
1031
|
|
1032 ahcf->peers->peer[0].uri_separator = ":";
|
|
1033
|
|
1034 ahcf->host_header.len = sizeof("localhost") - 1;
|
|
1035 ahcf->host_header.data = (u_char *) "localhost";
|
|
1036 ahcf->uri = unix_upstream.uri;
|
|
1037
|
|
1038 #else
|
|
1039 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1040 "the unix domain sockets are not supported "
|
|
1041 "on this platform");
|
|
1042 return NGX_CONF_ERROR;
|
|
1043
|
|
1044 #endif
|
|
1045
|
|
1046 } else {
|
|
1047 ngx_memzero(&inet_upstream, sizeof(ngx_inet_upstream_t));
|
|
1048
|
|
1049 inet_upstream.name = *url;
|
|
1050 inet_upstream.url = *url;
|
|
1051 inet_upstream.default_port_value = 80;
|
|
1052 inet_upstream.uri_part = 1;
|
|
1053
|
|
1054 ahcf->peers = ngx_inet_upstream_parse(cf, &inet_upstream);
|
|
1055 if (ahcf->peers == NULL) {
|
|
1056 return NGX_CONF_ERROR;
|
|
1057 }
|
|
1058
|
|
1059 for (i = 0; i < ahcf->peers->number; i++) {
|
|
1060 ahcf->peers->peer[i].uri_separator = "";
|
|
1061 }
|
|
1062
|
|
1063 ahcf->host_header = inet_upstream.host_header;
|
|
1064 ahcf->uri = inet_upstream.uri;
|
|
1065 }
|
|
1066
|
|
1067 return NGX_CONF_OK;
|
|
1068 }
|