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