|
350
|
1
|
|
|
2 /*
|
|
|
3 * Copyright (C) Igor Sysoev
|
|
660
|
4 * Copyright (C) Nginx, Inc.
|
|
350
|
5 */
|
|
|
6
|
|
|
7
|
|
|
8 #include <ngx_config.h>
|
|
|
9 #include <ngx_core.h>
|
|
|
10 #include <ngx_event.h>
|
|
|
11
|
|
|
12
|
|
|
13 #define NGX_RESOLVER_UDP_SIZE 4096
|
|
|
14
|
|
|
15
|
|
|
16 typedef struct {
|
|
|
17 u_char ident_hi;
|
|
|
18 u_char ident_lo;
|
|
|
19 u_char flags_hi;
|
|
|
20 u_char flags_lo;
|
|
|
21 u_char nqs_hi;
|
|
|
22 u_char nqs_lo;
|
|
|
23 u_char nan_hi;
|
|
|
24 u_char nan_lo;
|
|
|
25 u_char nns_hi;
|
|
|
26 u_char nns_lo;
|
|
|
27 u_char nar_hi;
|
|
|
28 u_char nar_lo;
|
|
|
29 } ngx_resolver_query_t;
|
|
|
30
|
|
|
31
|
|
|
32 typedef struct {
|
|
|
33 u_char type_hi;
|
|
|
34 u_char type_lo;
|
|
|
35 u_char class_hi;
|
|
|
36 u_char class_lo;
|
|
|
37 } ngx_resolver_qs_t;
|
|
|
38
|
|
|
39
|
|
|
40 typedef struct {
|
|
|
41 u_char type_hi;
|
|
|
42 u_char type_lo;
|
|
|
43 u_char class_hi;
|
|
|
44 u_char class_lo;
|
|
|
45 u_char ttl[4];
|
|
|
46 u_char len_hi;
|
|
|
47 u_char len_lo;
|
|
|
48 } ngx_resolver_an_t;
|
|
|
49
|
|
|
50
|
|
|
51 ngx_int_t ngx_udp_connect(ngx_udp_connection_t *uc);
|
|
|
52
|
|
|
53
|
|
366
|
54 static void ngx_resolver_cleanup(void *data);
|
|
|
55 static void ngx_resolver_cleanup_tree(ngx_resolver_t *r, ngx_rbtree_t *tree);
|
|
350
|
56 static ngx_int_t ngx_resolve_name_locked(ngx_resolver_t *r,
|
|
|
57 ngx_resolver_ctx_t *ctx);
|
|
|
58 static void ngx_resolver_expire(ngx_resolver_t *r, ngx_rbtree_t *tree,
|
|
|
59 ngx_queue_t *queue);
|
|
|
60 static ngx_int_t ngx_resolver_send_query(ngx_resolver_t *r,
|
|
|
61 ngx_resolver_node_t *rn);
|
|
|
62 static ngx_int_t ngx_resolver_create_name_query(ngx_resolver_node_t *rn,
|
|
|
63 ngx_resolver_ctx_t *ctx);
|
|
|
64 static ngx_int_t ngx_resolver_create_addr_query(ngx_resolver_node_t *rn,
|
|
|
65 ngx_resolver_ctx_t *ctx);
|
|
|
66 static void ngx_resolver_resend_handler(ngx_event_t *ev);
|
|
|
67 static time_t ngx_resolver_resend(ngx_resolver_t *r, ngx_rbtree_t *tree,
|
|
|
68 ngx_queue_t *queue);
|
|
|
69 static void ngx_resolver_read_response(ngx_event_t *rev);
|
|
|
70 static void ngx_resolver_process_response(ngx_resolver_t *r, u_char *buf,
|
|
|
71 size_t n);
|
|
|
72 static void ngx_resolver_process_a(ngx_resolver_t *r, u_char *buf, size_t n,
|
|
356
|
73 ngx_uint_t ident, ngx_uint_t code, ngx_uint_t nan, ngx_uint_t ans);
|
|
350
|
74 static void ngx_resolver_process_ptr(ngx_resolver_t *r, u_char *buf, size_t n,
|
|
356
|
75 ngx_uint_t ident, ngx_uint_t code, ngx_uint_t nan);
|
|
350
|
76 static ngx_resolver_node_t *ngx_resolver_lookup_name(ngx_resolver_t *r,
|
|
|
77 ngx_str_t *name, uint32_t hash);
|
|
|
78 static ngx_resolver_node_t *ngx_resolver_lookup_addr(ngx_resolver_t *r,
|
|
|
79 in_addr_t addr);
|
|
|
80 static void ngx_resolver_rbtree_insert_value(ngx_rbtree_node_t *temp,
|
|
|
81 ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel);
|
|
|
82 static ngx_int_t ngx_resolver_copy(ngx_resolver_t *r, ngx_str_t *name,
|
|
|
83 u_char *buf, u_char *src, u_char *last);
|
|
|
84 static void ngx_resolver_timeout_handler(ngx_event_t *ev);
|
|
|
85 static void ngx_resolver_free_node(ngx_resolver_t *r, ngx_resolver_node_t *rn);
|
|
|
86 static void *ngx_resolver_alloc(ngx_resolver_t *r, size_t size);
|
|
366
|
87 static void *ngx_resolver_calloc(ngx_resolver_t *r, size_t size);
|
|
350
|
88 static void ngx_resolver_free(ngx_resolver_t *r, void *p);
|
|
|
89 static void ngx_resolver_free_locked(ngx_resolver_t *r, void *p);
|
|
|
90 static void *ngx_resolver_dup(ngx_resolver_t *r, void *src, size_t size);
|
|
688
|
91 static in_addr_t *ngx_resolver_rotate(ngx_resolver_t *r, in_addr_t *src,
|
|
|
92 ngx_uint_t n);
|
|
560
|
93 static u_char *ngx_resolver_log_error(ngx_log_t *log, u_char *buf, size_t len);
|
|
350
|
94
|
|
|
95
|
|
|
96 ngx_resolver_t *
|
|
646
|
97 ngx_resolver_create(ngx_conf_t *cf, ngx_str_t *names, ngx_uint_t n)
|
|
350
|
98 {
|
|
650
|
99 ngx_str_t s;
|
|
646
|
100 ngx_url_t u;
|
|
678
|
101 ngx_uint_t i, j;
|
|
350
|
102 ngx_resolver_t *r;
|
|
366
|
103 ngx_pool_cleanup_t *cln;
|
|
350
|
104 ngx_udp_connection_t *uc;
|
|
|
105
|
|
366
|
106 cln = ngx_pool_cleanup_add(cf->pool, 0);
|
|
|
107 if (cln == NULL) {
|
|
|
108 return NULL;
|
|
|
109 }
|
|
|
110
|
|
|
111 cln->handler = ngx_resolver_cleanup;
|
|
|
112
|
|
|
113 r = ngx_calloc(sizeof(ngx_resolver_t), cf->log);
|
|
350
|
114 if (r == NULL) {
|
|
|
115 return NULL;
|
|
|
116 }
|
|
|
117
|
|
366
|
118 cln->data = r;
|
|
|
119
|
|
|
120 r->event = ngx_calloc(sizeof(ngx_event_t), cf->log);
|
|
350
|
121 if (r->event == NULL) {
|
|
|
122 return NULL;
|
|
|
123 }
|
|
|
124
|
|
354
|
125 ngx_rbtree_init(&r->name_rbtree, &r->name_sentinel,
|
|
|
126 ngx_resolver_rbtree_insert_value);
|
|
|
127
|
|
|
128 ngx_rbtree_init(&r->addr_rbtree, &r->addr_sentinel,
|
|
|
129 ngx_rbtree_insert_value);
|
|
|
130
|
|
|
131 ngx_queue_init(&r->name_resend_queue);
|
|
|
132 ngx_queue_init(&r->addr_resend_queue);
|
|
|
133
|
|
|
134 ngx_queue_init(&r->name_expire_queue);
|
|
|
135 ngx_queue_init(&r->addr_expire_queue);
|
|
350
|
136
|
|
|
137 r->event->handler = ngx_resolver_resend_handler;
|
|
|
138 r->event->data = r;
|
|
484
|
139 r->event->log = &cf->cycle->new_log;
|
|
350
|
140 r->ident = -1;
|
|
|
141
|
|
|
142 r->resend_timeout = 5;
|
|
|
143 r->expire = 30;
|
|
650
|
144 r->valid = 0;
|
|
350
|
145
|
|
484
|
146 r->log = &cf->cycle->new_log;
|
|
598
|
147 r->log_level = NGX_LOG_ERR;
|
|
350
|
148
|
|
684
|
149 if (n) {
|
|
|
150 if (ngx_array_init(&r->udp_connections, cf->pool, n,
|
|
|
151 sizeof(ngx_udp_connection_t))
|
|
|
152 != NGX_OK)
|
|
|
153 {
|
|
|
154 return NULL;
|
|
|
155 }
|
|
|
156 }
|
|
|
157
|
|
646
|
158 for (i = 0; i < n; i++) {
|
|
650
|
159 if (ngx_strncmp(names[i].data, "valid=", 6) == 0) {
|
|
|
160 s.len = names[i].len - 6;
|
|
|
161 s.data = names[i].data + 6;
|
|
|
162
|
|
|
163 r->valid = ngx_parse_time(&s, 1);
|
|
|
164
|
|
662
|
165 if (r->valid == (time_t) NGX_ERROR) {
|
|
650
|
166 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
|
167 "invalid parameter: %V", &names[i]);
|
|
|
168 return NULL;
|
|
|
169 }
|
|
|
170
|
|
|
171 continue;
|
|
|
172 }
|
|
|
173
|
|
646
|
174 ngx_memzero(&u, sizeof(ngx_url_t));
|
|
|
175
|
|
676
|
176 u.url = names[i];
|
|
|
177 u.default_port = 53;
|
|
|
178
|
|
|
179 if (ngx_parse_url(cf->pool, &u) != NGX_OK) {
|
|
|
180 if (u.err) {
|
|
|
181 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
|
182 "%s in resolver \"%V\"",
|
|
|
183 u.err, &u.url);
|
|
|
184 }
|
|
|
185
|
|
646
|
186 return NULL;
|
|
|
187 }
|
|
|
188
|
|
678
|
189 uc = ngx_array_push_n(&r->udp_connections, u.naddrs);
|
|
354
|
190 if (uc == NULL) {
|
|
|
191 return NULL;
|
|
|
192 }
|
|
|
193
|
|
678
|
194 ngx_memzero(uc, u.naddrs * sizeof(ngx_udp_connection_t));
|
|
|
195
|
|
|
196 for (j = 0; j < u.naddrs; j++) {
|
|
|
197 uc[j].sockaddr = u.addrs[j].sockaddr;
|
|
|
198 uc[j].socklen = u.addrs[j].socklen;
|
|
|
199 uc[j].server = u.addrs[j].name;
|
|
|
200 }
|
|
354
|
201 }
|
|
350
|
202
|
|
|
203 return r;
|
|
|
204 }
|
|
|
205
|
|
|
206
|
|
366
|
207 static void
|
|
|
208 ngx_resolver_cleanup(void *data)
|
|
|
209 {
|
|
|
210 ngx_resolver_t *r = data;
|
|
|
211
|
|
646
|
212 ngx_uint_t i;
|
|
|
213 ngx_udp_connection_t *uc;
|
|
|
214
|
|
366
|
215 if (r) {
|
|
|
216 ngx_log_debug0(NGX_LOG_DEBUG_CORE, ngx_cycle->log, 0,
|
|
|
217 "cleanup resolver");
|
|
|
218
|
|
|
219 ngx_resolver_cleanup_tree(r, &r->name_rbtree);
|
|
|
220
|
|
|
221 ngx_resolver_cleanup_tree(r, &r->addr_rbtree);
|
|
|
222
|
|
|
223 if (r->event) {
|
|
|
224 ngx_free(r->event);
|
|
|
225 }
|
|
|
226
|
|
646
|
227
|
|
|
228 uc = r->udp_connections.elts;
|
|
|
229
|
|
|
230 for (i = 0; i < r->udp_connections.nelts; i++) {
|
|
|
231 if (uc[i].connection) {
|
|
|
232 ngx_close_connection(uc[i].connection);
|
|
366
|
233 }
|
|
|
234 }
|
|
|
235
|
|
|
236 ngx_free(r);
|
|
|
237 }
|
|
|
238 }
|
|
|
239
|
|
|
240
|
|
|
241 static void
|
|
|
242 ngx_resolver_cleanup_tree(ngx_resolver_t *r, ngx_rbtree_t *tree)
|
|
|
243 {
|
|
|
244 ngx_resolver_ctx_t *ctx, *next;
|
|
|
245 ngx_resolver_node_t *rn;
|
|
|
246
|
|
|
247 while (tree->root != tree->sentinel) {
|
|
|
248
|
|
|
249 rn = (ngx_resolver_node_t *) ngx_rbtree_min(tree->root, tree->sentinel);
|
|
|
250
|
|
|
251 ngx_queue_remove(&rn->queue);
|
|
|
252
|
|
|
253 for (ctx = rn->waiting; ctx; ctx = next) {
|
|
376
|
254 next = ctx->next;
|
|
366
|
255
|
|
|
256 if (ctx->event) {
|
|
|
257 ngx_resolver_free(r, ctx->event);
|
|
|
258 }
|
|
|
259
|
|
|
260 ngx_resolver_free(r, ctx);
|
|
|
261 }
|
|
|
262
|
|
|
263 ngx_rbtree_delete(tree, &rn->node);
|
|
|
264
|
|
|
265 ngx_resolver_free_node(r, rn);
|
|
|
266 }
|
|
|
267 }
|
|
|
268
|
|
|
269
|
|
350
|
270 ngx_resolver_ctx_t *
|
|
|
271 ngx_resolve_start(ngx_resolver_t *r, ngx_resolver_ctx_t *temp)
|
|
|
272 {
|
|
|
273 in_addr_t addr;
|
|
|
274 ngx_resolver_ctx_t *ctx;
|
|
|
275
|
|
|
276 if (temp) {
|
|
|
277 addr = ngx_inet_addr(temp->name.data, temp->name.len);
|
|
|
278
|
|
|
279 if (addr != INADDR_NONE) {
|
|
|
280 temp->resolver = r;
|
|
|
281 temp->state = NGX_OK;
|
|
|
282 temp->naddrs = 1;
|
|
|
283 temp->addrs = &temp->addr;
|
|
|
284 temp->addr = addr;
|
|
|
285 temp->quick = 1;
|
|
|
286
|
|
|
287 return temp;
|
|
|
288 }
|
|
|
289 }
|
|
|
290
|
|
646
|
291 if (r->udp_connections.nelts == 0) {
|
|
354
|
292 return NGX_NO_RESOLVER;
|
|
|
293 }
|
|
|
294
|
|
350
|
295 ctx = ngx_resolver_calloc(r, sizeof(ngx_resolver_ctx_t));
|
|
|
296
|
|
|
297 if (ctx) {
|
|
|
298 ctx->resolver = r;
|
|
|
299 }
|
|
|
300
|
|
|
301 return ctx;
|
|
|
302 }
|
|
|
303
|
|
|
304
|
|
|
305 ngx_int_t
|
|
|
306 ngx_resolve_name(ngx_resolver_ctx_t *ctx)
|
|
|
307 {
|
|
|
308 ngx_int_t rc;
|
|
|
309 ngx_resolver_t *r;
|
|
|
310
|
|
|
311 r = ctx->resolver;
|
|
|
312
|
|
|
313 ngx_log_debug1(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
|
314 "resolve: \"%V\"", &ctx->name);
|
|
|
315
|
|
|
316 if (ctx->quick) {
|
|
|
317 ctx->handler(ctx);
|
|
|
318 return NGX_OK;
|
|
|
319 }
|
|
|
320
|
|
|
321 /* lock name mutex */
|
|
|
322
|
|
|
323 rc = ngx_resolve_name_locked(r, ctx);
|
|
|
324
|
|
|
325 if (rc == NGX_OK) {
|
|
|
326 return NGX_OK;
|
|
|
327 }
|
|
|
328
|
|
|
329 /* unlock name mutex */
|
|
|
330
|
|
|
331 if (rc == NGX_AGAIN) {
|
|
|
332 return NGX_OK;
|
|
|
333 }
|
|
|
334
|
|
366
|
335 /* NGX_ERROR */
|
|
350
|
336
|
|
|
337 if (ctx->event) {
|
|
366
|
338 ngx_resolver_free(r, ctx->event);
|
|
350
|
339 }
|
|
|
340
|
|
366
|
341 ngx_resolver_free(r, ctx);
|
|
350
|
342
|
|
|
343 return NGX_ERROR;
|
|
|
344 }
|
|
|
345
|
|
|
346
|
|
|
347 void
|
|
|
348 ngx_resolve_name_done(ngx_resolver_ctx_t *ctx)
|
|
|
349 {
|
|
|
350 uint32_t hash;
|
|
|
351 ngx_resolver_t *r;
|
|
|
352 ngx_resolver_ctx_t *w, **p;
|
|
|
353 ngx_resolver_node_t *rn;
|
|
|
354
|
|
|
355 r = ctx->resolver;
|
|
|
356
|
|
|
357 ngx_log_debug1(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
|
358 "resolve name done: %i", ctx->state);
|
|
|
359
|
|
|
360 if (ctx->quick) {
|
|
|
361 return;
|
|
|
362 }
|
|
|
363
|
|
|
364 if (ctx->event && ctx->event->timer_set) {
|
|
|
365 ngx_del_timer(ctx->event);
|
|
|
366 }
|
|
|
367
|
|
|
368 /* lock name mutex */
|
|
|
369
|
|
|
370 if (ctx->state == NGX_AGAIN || ctx->state == NGX_RESOLVE_TIMEDOUT) {
|
|
|
371
|
|
|
372 hash = ngx_crc32_short(ctx->name.data, ctx->name.len);
|
|
|
373
|
|
|
374 rn = ngx_resolver_lookup_name(r, &ctx->name, hash);
|
|
|
375
|
|
|
376 if (rn) {
|
|
|
377 p = &rn->waiting;
|
|
|
378 w = rn->waiting;
|
|
|
379
|
|
|
380 while (w) {
|
|
|
381 if (w == ctx) {
|
|
|
382 *p = w->next;
|
|
|
383
|
|
|
384 goto done;
|
|
|
385 }
|
|
|
386
|
|
|
387 p = &w->next;
|
|
|
388 w = w->next;
|
|
|
389 }
|
|
|
390 }
|
|
|
391
|
|
|
392 ngx_log_error(NGX_LOG_ALERT, r->log, 0,
|
|
|
393 "could not cancel %V resolving", &ctx->name);
|
|
|
394 }
|
|
|
395
|
|
|
396 done:
|
|
|
397
|
|
|
398 ngx_resolver_expire(r, &r->name_rbtree, &r->name_expire_queue);
|
|
|
399
|
|
|
400 /* unlock name mutex */
|
|
|
401
|
|
366
|
402 /* lock alloc mutex */
|
|
|
403
|
|
|
404 if (ctx->event) {
|
|
|
405 ngx_resolver_free_locked(r, ctx->event);
|
|
|
406 }
|
|
|
407
|
|
|
408 ngx_resolver_free_locked(r, ctx);
|
|
|
409
|
|
|
410 /* unlock alloc mutex */
|
|
350
|
411 }
|
|
|
412
|
|
|
413
|
|
|
414 /* NGX_RESOLVE_A only */
|
|
|
415
|
|
|
416 static ngx_int_t
|
|
|
417 ngx_resolve_name_locked(ngx_resolver_t *r, ngx_resolver_ctx_t *ctx)
|
|
|
418 {
|
|
|
419 uint32_t hash;
|
|
|
420 in_addr_t addr, *addrs;
|
|
372
|
421 ngx_int_t rc;
|
|
350
|
422 ngx_uint_t naddrs;
|
|
|
423 ngx_resolver_ctx_t *next;
|
|
|
424 ngx_resolver_node_t *rn;
|
|
|
425
|
|
|
426 hash = ngx_crc32_short(ctx->name.data, ctx->name.len);
|
|
|
427
|
|
|
428 rn = ngx_resolver_lookup_name(r, &ctx->name, hash);
|
|
|
429
|
|
|
430 if (rn) {
|
|
|
431
|
|
|
432 if (rn->valid >= ngx_time()) {
|
|
|
433
|
|
|
434 ngx_log_debug0(NGX_LOG_DEBUG_CORE, r->log, 0, "resolve cached");
|
|
|
435
|
|
|
436 ngx_queue_remove(&rn->queue);
|
|
|
437
|
|
|
438 rn->expire = ngx_time() + r->expire;
|
|
|
439
|
|
|
440 ngx_queue_insert_head(&r->name_expire_queue, &rn->queue);
|
|
|
441
|
|
|
442 naddrs = rn->naddrs;
|
|
|
443
|
|
|
444 if (naddrs) {
|
|
|
445
|
|
|
446 /* NGX_RESOLVE_A answer */
|
|
|
447
|
|
|
448 if (naddrs != 1) {
|
|
|
449 addr = 0;
|
|
688
|
450 addrs = ngx_resolver_rotate(r, rn->u.addrs, naddrs);
|
|
350
|
451 if (addrs == NULL) {
|
|
|
452 return NGX_ERROR;
|
|
|
453 }
|
|
|
454
|
|
|
455 } else {
|
|
|
456 addr = rn->u.addr;
|
|
|
457 addrs = NULL;
|
|
|
458 }
|
|
|
459
|
|
|
460 ctx->next = rn->waiting;
|
|
|
461 rn->waiting = NULL;
|
|
|
462
|
|
|
463 /* unlock name mutex */
|
|
|
464
|
|
|
465 do {
|
|
|
466 ctx->state = NGX_OK;
|
|
|
467 ctx->naddrs = naddrs;
|
|
|
468 ctx->addrs = (naddrs == 1) ? &ctx->addr : addrs;
|
|
|
469 ctx->addr = addr;
|
|
|
470 next = ctx->next;
|
|
|
471
|
|
|
472 ctx->handler(ctx);
|
|
|
473
|
|
|
474 ctx = next;
|
|
|
475 } while (ctx);
|
|
|
476
|
|
|
477 if (addrs) {
|
|
|
478 ngx_resolver_free(r, addrs);
|
|
|
479 }
|
|
|
480
|
|
|
481 return NGX_OK;
|
|
|
482 }
|
|
|
483
|
|
|
484 /* NGX_RESOLVE_CNAME */
|
|
|
485
|
|
372
|
486 if (ctx->recursion++ < NGX_RESOLVER_MAX_RECURSION) {
|
|
|
487
|
|
|
488 ctx->name.len = rn->cnlen;
|
|
|
489 ctx->name.data = rn->u.cname;
|
|
|
490
|
|
|
491 return ngx_resolve_name_locked(r, ctx);
|
|
|
492 }
|
|
|
493
|
|
|
494 ctx->next = rn->waiting;
|
|
|
495 rn->waiting = NULL;
|
|
|
496
|
|
|
497 /* unlock name mutex */
|
|
|
498
|
|
|
499 do {
|
|
|
500 ctx->state = NGX_RESOLVE_NXDOMAIN;
|
|
|
501 next = ctx->next;
|
|
|
502
|
|
|
503 ctx->handler(ctx);
|
|
|
504
|
|
|
505 ctx = next;
|
|
|
506 } while (ctx);
|
|
|
507
|
|
|
508 return NGX_OK;
|
|
350
|
509 }
|
|
|
510
|
|
|
511 if (rn->waiting) {
|
|
|
512
|
|
|
513 ctx->next = rn->waiting;
|
|
|
514 rn->waiting = ctx;
|
|
542
|
515 ctx->state = NGX_AGAIN;
|
|
350
|
516
|
|
|
517 return NGX_AGAIN;
|
|
|
518 }
|
|
|
519
|
|
|
520 ngx_queue_remove(&rn->queue);
|
|
|
521
|
|
|
522 /* lock alloc mutex */
|
|
|
523
|
|
674
|
524 if (rn->query) {
|
|
|
525 ngx_resolver_free_locked(r, rn->query);
|
|
|
526 rn->query = NULL;
|
|
|
527 }
|
|
350
|
528
|
|
|
529 if (rn->cnlen) {
|
|
|
530 ngx_resolver_free_locked(r, rn->u.cname);
|
|
|
531 }
|
|
|
532
|
|
|
533 if (rn->naddrs > 1) {
|
|
|
534 ngx_resolver_free_locked(r, rn->u.addrs);
|
|
|
535 }
|
|
|
536
|
|
|
537 /* unlock alloc mutex */
|
|
|
538
|
|
|
539 } else {
|
|
|
540
|
|
|
541 rn = ngx_resolver_alloc(r, sizeof(ngx_resolver_node_t));
|
|
|
542 if (rn == NULL) {
|
|
|
543 return NGX_ERROR;
|
|
|
544 }
|
|
|
545
|
|
|
546 rn->name = ngx_resolver_dup(r, ctx->name.data, ctx->name.len);
|
|
|
547 if (rn->name == NULL) {
|
|
|
548 ngx_resolver_free(r, rn);
|
|
|
549 return NGX_ERROR;
|
|
|
550 }
|
|
|
551
|
|
|
552 rn->node.key = hash;
|
|
|
553 rn->nlen = (u_short) ctx->name.len;
|
|
372
|
554 rn->query = NULL;
|
|
350
|
555
|
|
|
556 ngx_rbtree_insert(&r->name_rbtree, &rn->node);
|
|
|
557 }
|
|
|
558
|
|
372
|
559 rc = ngx_resolver_create_name_query(rn, ctx);
|
|
|
560
|
|
|
561 if (rc == NGX_ERROR) {
|
|
350
|
562 goto failed;
|
|
|
563 }
|
|
|
564
|
|
372
|
565 if (rc == NGX_DECLINED) {
|
|
|
566 ngx_rbtree_delete(&r->name_rbtree, &rn->node);
|
|
|
567
|
|
|
568 ngx_resolver_free(r, rn->query);
|
|
|
569 ngx_resolver_free(r, rn->name);
|
|
|
570 ngx_resolver_free(r, rn);
|
|
|
571
|
|
|
572 ctx->state = NGX_RESOLVE_NXDOMAIN;
|
|
|
573 ctx->handler(ctx);
|
|
|
574
|
|
|
575 return NGX_OK;
|
|
|
576 }
|
|
|
577
|
|
350
|
578 if (ngx_resolver_send_query(r, rn) != NGX_OK) {
|
|
|
579 goto failed;
|
|
|
580 }
|
|
|
581
|
|
|
582 if (ctx->event == NULL) {
|
|
|
583 ctx->event = ngx_resolver_calloc(r, sizeof(ngx_event_t));
|
|
|
584 if (ctx->event == NULL) {
|
|
|
585 goto failed;
|
|
|
586 }
|
|
|
587
|
|
|
588 ctx->event->handler = ngx_resolver_timeout_handler;
|
|
|
589 ctx->event->data = ctx;
|
|
|
590 ctx->event->log = r->log;
|
|
|
591 ctx->ident = -1;
|
|
|
592
|
|
|
593 ngx_add_timer(ctx->event, ctx->timeout);
|
|
|
594 }
|
|
|
595
|
|
|
596 if (ngx_queue_empty(&r->name_resend_queue)) {
|
|
|
597 ngx_add_timer(r->event, (ngx_msec_t) (r->resend_timeout * 1000));
|
|
|
598 }
|
|
|
599
|
|
|
600 rn->expire = ngx_time() + r->resend_timeout;
|
|
|
601
|
|
|
602 ngx_queue_insert_head(&r->name_resend_queue, &rn->queue);
|
|
|
603
|
|
|
604 rn->cnlen = 0;
|
|
|
605 rn->naddrs = 0;
|
|
|
606 rn->valid = 0;
|
|
|
607 rn->waiting = ctx;
|
|
|
608
|
|
|
609 ctx->state = NGX_AGAIN;
|
|
|
610
|
|
|
611 return NGX_AGAIN;
|
|
|
612
|
|
|
613 failed:
|
|
|
614
|
|
|
615 ngx_rbtree_delete(&r->name_rbtree, &rn->node);
|
|
|
616
|
|
372
|
617 if (rn->query) {
|
|
|
618 ngx_resolver_free(r, rn->query);
|
|
|
619 }
|
|
|
620
|
|
350
|
621 ngx_resolver_free(r, rn->name);
|
|
|
622
|
|
|
623 ngx_resolver_free(r, rn);
|
|
|
624
|
|
|
625 return NGX_ERROR;
|
|
|
626 }
|
|
|
627
|
|
|
628
|
|
|
629 ngx_int_t
|
|
|
630 ngx_resolve_addr(ngx_resolver_ctx_t *ctx)
|
|
|
631 {
|
|
442
|
632 u_char *name;
|
|
350
|
633 ngx_resolver_t *r;
|
|
|
634 ngx_resolver_node_t *rn;
|
|
|
635
|
|
|
636 r = ctx->resolver;
|
|
|
637
|
|
|
638 ctx->addr = ntohl(ctx->addr);
|
|
|
639
|
|
|
640 /* lock addr mutex */
|
|
|
641
|
|
|
642 rn = ngx_resolver_lookup_addr(r, ctx->addr);
|
|
|
643
|
|
|
644 if (rn) {
|
|
|
645
|
|
|
646 if (rn->valid >= ngx_time()) {
|
|
|
647
|
|
|
648 ngx_log_debug0(NGX_LOG_DEBUG_CORE, r->log, 0, "resolve cached");
|
|
|
649
|
|
|
650 ngx_queue_remove(&rn->queue);
|
|
|
651
|
|
|
652 rn->expire = ngx_time() + r->expire;
|
|
|
653
|
|
|
654 ngx_queue_insert_head(&r->addr_expire_queue, &rn->queue);
|
|
|
655
|
|
442
|
656 name = ngx_resolver_dup(r, rn->name, rn->nlen);
|
|
|
657 if (name == NULL) {
|
|
350
|
658 goto failed;
|
|
|
659 }
|
|
|
660
|
|
442
|
661 ctx->name.len = rn->nlen;
|
|
|
662 ctx->name.data = name;
|
|
|
663
|
|
350
|
664 /* unlock addr mutex */
|
|
|
665
|
|
|
666 ctx->state = NGX_OK;
|
|
|
667
|
|
|
668 ctx->handler(ctx);
|
|
|
669
|
|
442
|
670 ngx_resolver_free(r, name);
|
|
350
|
671
|
|
|
672 return NGX_OK;
|
|
|
673 }
|
|
|
674
|
|
|
675 if (rn->waiting) {
|
|
|
676
|
|
|
677 ctx->next = rn->waiting;
|
|
|
678 rn->waiting = ctx;
|
|
542
|
679 ctx->state = NGX_AGAIN;
|
|
350
|
680
|
|
442
|
681 /* unlock addr mutex */
|
|
|
682
|
|
|
683 return NGX_OK;
|
|
350
|
684 }
|
|
|
685
|
|
|
686 ngx_queue_remove(&rn->queue);
|
|
|
687
|
|
|
688 ngx_resolver_free(r, rn->query);
|
|
372
|
689 rn->query = NULL;
|
|
350
|
690
|
|
|
691 } else {
|
|
|
692 rn = ngx_resolver_alloc(r, sizeof(ngx_resolver_node_t));
|
|
|
693 if (rn == NULL) {
|
|
|
694 goto failed;
|
|
|
695 }
|
|
|
696
|
|
|
697 rn->node.key = ctx->addr;
|
|
372
|
698 rn->query = NULL;
|
|
350
|
699
|
|
|
700 ngx_rbtree_insert(&r->addr_rbtree, &rn->node);
|
|
|
701 }
|
|
|
702
|
|
|
703 if (ngx_resolver_create_addr_query(rn, ctx) != NGX_OK) {
|
|
|
704 goto failed;
|
|
|
705 }
|
|
|
706
|
|
|
707 if (ngx_resolver_send_query(r, rn) != NGX_OK) {
|
|
|
708 goto failed;
|
|
|
709 }
|
|
|
710
|
|
|
711 ctx->event = ngx_resolver_calloc(r, sizeof(ngx_event_t));
|
|
|
712 if (ctx->event == NULL) {
|
|
|
713 goto failed;
|
|
|
714 }
|
|
|
715
|
|
|
716 ctx->event->handler = ngx_resolver_timeout_handler;
|
|
|
717 ctx->event->data = ctx;
|
|
|
718 ctx->event->log = r->log;
|
|
|
719 ctx->ident = -1;
|
|
|
720
|
|
|
721 ngx_add_timer(ctx->event, ctx->timeout);
|
|
|
722
|
|
|
723 if (ngx_queue_empty(&r->addr_resend_queue)) {
|
|
|
724 ngx_add_timer(r->event, (ngx_msec_t) (r->resend_timeout * 1000));
|
|
|
725 }
|
|
|
726
|
|
|
727 rn->expire = ngx_time() + r->resend_timeout;
|
|
|
728
|
|
|
729 ngx_queue_insert_head(&r->addr_resend_queue, &rn->queue);
|
|
|
730
|
|
|
731 rn->cnlen = 0;
|
|
|
732 rn->naddrs = 0;
|
|
|
733 rn->name = NULL;
|
|
|
734 rn->nlen = 0;
|
|
|
735 rn->valid = 0;
|
|
|
736 rn->waiting = ctx;
|
|
|
737
|
|
|
738 /* unlock addr mutex */
|
|
|
739
|
|
|
740 ctx->state = NGX_AGAIN;
|
|
|
741
|
|
|
742 return NGX_OK;
|
|
|
743
|
|
|
744 failed:
|
|
|
745
|
|
|
746 if (rn) {
|
|
|
747 ngx_rbtree_delete(&r->addr_rbtree, &rn->node);
|
|
|
748
|
|
372
|
749 if (rn->query) {
|
|
|
750 ngx_resolver_free(r, rn->query);
|
|
|
751 }
|
|
|
752
|
|
350
|
753 ngx_resolver_free(r, rn);
|
|
|
754 }
|
|
|
755
|
|
|
756 /* unlock addr mutex */
|
|
|
757
|
|
|
758 if (ctx->event) {
|
|
366
|
759 ngx_resolver_free(r, ctx->event);
|
|
350
|
760 }
|
|
|
761
|
|
366
|
762 ngx_resolver_free(r, ctx);
|
|
350
|
763
|
|
|
764 return NGX_ERROR;
|
|
|
765 }
|
|
|
766
|
|
|
767
|
|
|
768 void
|
|
|
769 ngx_resolve_addr_done(ngx_resolver_ctx_t *ctx)
|
|
|
770 {
|
|
|
771 in_addr_t addr;
|
|
|
772 ngx_resolver_t *r;
|
|
|
773 ngx_resolver_ctx_t *w, **p;
|
|
|
774 ngx_resolver_node_t *rn;
|
|
|
775
|
|
|
776 r = ctx->resolver;
|
|
|
777
|
|
|
778 ngx_log_debug1(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
|
779 "resolve addr done: %i", ctx->state);
|
|
|
780
|
|
|
781 if (ctx->event && ctx->event->timer_set) {
|
|
|
782 ngx_del_timer(ctx->event);
|
|
|
783 }
|
|
|
784
|
|
|
785 /* lock addr mutex */
|
|
|
786
|
|
|
787 if (ctx->state == NGX_AGAIN || ctx->state == NGX_RESOLVE_TIMEDOUT) {
|
|
|
788
|
|
|
789 rn = ngx_resolver_lookup_addr(r, ctx->addr);
|
|
|
790
|
|
|
791 if (rn) {
|
|
|
792 p = &rn->waiting;
|
|
|
793 w = rn->waiting;
|
|
|
794
|
|
|
795 while (w) {
|
|
|
796 if (w == ctx) {
|
|
|
797 *p = w->next;
|
|
|
798
|
|
|
799 goto done;
|
|
|
800 }
|
|
|
801
|
|
|
802 p = &w->next;
|
|
|
803 w = w->next;
|
|
|
804 }
|
|
|
805 }
|
|
|
806
|
|
|
807 addr = ntohl(ctx->addr);
|
|
|
808
|
|
|
809 ngx_log_error(NGX_LOG_ALERT, r->log, 0,
|
|
|
810 "could not cancel %ud.%ud.%ud.%ud resolving",
|
|
|
811 (addr >> 24) & 0xff, (addr >> 16) & 0xff,
|
|
|
812 (addr >> 8) & 0xff, addr & 0xff);
|
|
|
813 }
|
|
|
814
|
|
|
815 done:
|
|
|
816
|
|
|
817 ngx_resolver_expire(r, &r->addr_rbtree, &r->addr_expire_queue);
|
|
|
818
|
|
|
819 /* unlock addr mutex */
|
|
|
820
|
|
366
|
821 /* lock alloc mutex */
|
|
|
822
|
|
|
823 if (ctx->event) {
|
|
|
824 ngx_resolver_free_locked(r, ctx->event);
|
|
|
825 }
|
|
|
826
|
|
|
827 ngx_resolver_free_locked(r, ctx);
|
|
|
828
|
|
|
829 /* unlock alloc mutex */
|
|
350
|
830 }
|
|
|
831
|
|
|
832
|
|
|
833 static void
|
|
|
834 ngx_resolver_expire(ngx_resolver_t *r, ngx_rbtree_t *tree, ngx_queue_t *queue)
|
|
|
835 {
|
|
|
836 time_t now;
|
|
|
837 ngx_uint_t i;
|
|
|
838 ngx_queue_t *q;
|
|
|
839 ngx_resolver_node_t *rn;
|
|
|
840
|
|
|
841 ngx_log_debug0(NGX_LOG_DEBUG_CORE, r->log, 0, "resolver expire");
|
|
|
842
|
|
|
843 now = ngx_time();
|
|
|
844
|
|
|
845 for (i = 0; i < 2; i++) {
|
|
|
846 if (ngx_queue_empty(queue)) {
|
|
|
847 return;
|
|
|
848 }
|
|
|
849
|
|
|
850 q = ngx_queue_last(queue);
|
|
|
851
|
|
|
852 rn = ngx_queue_data(q, ngx_resolver_node_t, queue);
|
|
|
853
|
|
|
854 if (now <= rn->expire) {
|
|
|
855 return;
|
|
|
856 }
|
|
|
857
|
|
358
|
858 ngx_log_debug2(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
|
859 "resolver expire \"%*s\"", (size_t) rn->nlen, rn->name);
|
|
350
|
860
|
|
|
861 ngx_queue_remove(q);
|
|
|
862
|
|
|
863 ngx_rbtree_delete(tree, &rn->node);
|
|
|
864
|
|
|
865 ngx_resolver_free_node(r, rn);
|
|
|
866 }
|
|
|
867 }
|
|
|
868
|
|
|
869
|
|
|
870 static ngx_int_t
|
|
|
871 ngx_resolver_send_query(ngx_resolver_t *r, ngx_resolver_node_t *rn)
|
|
|
872 {
|
|
|
873 ssize_t n;
|
|
|
874 ngx_udp_connection_t *uc;
|
|
|
875
|
|
646
|
876 uc = r->udp_connections.elts;
|
|
|
877
|
|
|
878 uc = &uc[r->last_connection++];
|
|
|
879 if (r->last_connection == r->udp_connections.nelts) {
|
|
|
880 r->last_connection = 0;
|
|
|
881 }
|
|
350
|
882
|
|
|
883 if (uc->connection == NULL) {
|
|
664
|
884
|
|
|
885 uc->log = *r->log;
|
|
|
886 uc->log.handler = ngx_resolver_log_error;
|
|
|
887 uc->log.data = uc;
|
|
|
888 uc->log.action = "resolving";
|
|
|
889
|
|
350
|
890 if (ngx_udp_connect(uc) != NGX_OK) {
|
|
|
891 return NGX_ERROR;
|
|
|
892 }
|
|
|
893
|
|
|
894 uc->connection->data = r;
|
|
|
895 uc->connection->read->handler = ngx_resolver_read_response;
|
|
366
|
896 uc->connection->read->resolver = 1;
|
|
350
|
897 }
|
|
|
898
|
|
|
899 n = ngx_send(uc->connection, rn->query, rn->qlen);
|
|
|
900
|
|
|
901 if (n == -1) {
|
|
|
902 return NGX_ERROR;
|
|
|
903 }
|
|
|
904
|
|
|
905 if ((size_t) n != (size_t) rn->qlen) {
|
|
560
|
906 ngx_log_error(NGX_LOG_CRIT, &uc->log, 0, "send() incomplete");
|
|
350
|
907 return NGX_ERROR;
|
|
|
908 }
|
|
|
909
|
|
|
910 return NGX_OK;
|
|
|
911 }
|
|
|
912
|
|
|
913
|
|
|
914 static void
|
|
|
915 ngx_resolver_resend_handler(ngx_event_t *ev)
|
|
|
916 {
|
|
|
917 time_t timer, atimer, ntimer;
|
|
|
918 ngx_resolver_t *r;
|
|
|
919
|
|
|
920 r = ev->data;
|
|
|
921
|
|
|
922 ngx_log_debug0(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
|
923 "resolver resend handler");
|
|
|
924
|
|
|
925 /* lock name mutex */
|
|
|
926
|
|
|
927 ntimer = ngx_resolver_resend(r, &r->name_rbtree, &r->name_resend_queue);
|
|
|
928
|
|
|
929 /* unlock name mutex */
|
|
|
930
|
|
|
931 /* lock addr mutex */
|
|
|
932
|
|
|
933 atimer = ngx_resolver_resend(r, &r->addr_rbtree, &r->addr_resend_queue);
|
|
354
|
934
|
|
350
|
935 /* unlock addr mutex */
|
|
|
936
|
|
|
937 if (ntimer == 0) {
|
|
|
938 timer = atimer;
|
|
|
939
|
|
|
940 } else if (atimer == 0) {
|
|
|
941 timer = ntimer;
|
|
|
942
|
|
|
943 } else {
|
|
|
944 timer = (atimer < ntimer) ? atimer : ntimer;
|
|
|
945 }
|
|
|
946
|
|
|
947 if (timer) {
|
|
|
948 ngx_add_timer(r->event, (ngx_msec_t) (timer * 1000));
|
|
|
949 }
|
|
|
950 }
|
|
|
951
|
|
|
952
|
|
|
953 static time_t
|
|
|
954 ngx_resolver_resend(ngx_resolver_t *r, ngx_rbtree_t *tree, ngx_queue_t *queue)
|
|
|
955 {
|
|
|
956 time_t now;
|
|
|
957 ngx_queue_t *q;
|
|
|
958 ngx_resolver_node_t *rn;
|
|
|
959
|
|
|
960 now = ngx_time();
|
|
|
961
|
|
|
962 for ( ;; ) {
|
|
|
963 if (ngx_queue_empty(queue)) {
|
|
|
964 return 0;
|
|
|
965 }
|
|
|
966
|
|
|
967 q = ngx_queue_last(queue);
|
|
|
968
|
|
|
969 rn = ngx_queue_data(q, ngx_resolver_node_t, queue);
|
|
|
970
|
|
|
971 if (now < rn->expire) {
|
|
|
972 return rn->expire - now;
|
|
|
973 }
|
|
|
974
|
|
358
|
975 ngx_log_debug3(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
|
976 "resolver resend \"%*s\" %p",
|
|
|
977 (size_t) rn->nlen, rn->name, rn->waiting);
|
|
350
|
978
|
|
|
979 ngx_queue_remove(q);
|
|
|
980
|
|
|
981 if (rn->waiting) {
|
|
|
982
|
|
678
|
983 (void) ngx_resolver_send_query(r, rn);
|
|
|
984
|
|
|
985 rn->expire = now + r->resend_timeout;
|
|
|
986
|
|
|
987 ngx_queue_insert_head(queue, q);
|
|
364
|
988
|
|
|
989 continue;
|
|
350
|
990 }
|
|
|
991
|
|
|
992 ngx_rbtree_delete(tree, &rn->node);
|
|
|
993
|
|
|
994 ngx_resolver_free_node(r, rn);
|
|
|
995 }
|
|
|
996 }
|
|
|
997
|
|
|
998
|
|
|
999 static void
|
|
|
1000 ngx_resolver_read_response(ngx_event_t *rev)
|
|
|
1001 {
|
|
|
1002 ssize_t n;
|
|
|
1003 ngx_connection_t *c;
|
|
|
1004 u_char buf[NGX_RESOLVER_UDP_SIZE];
|
|
|
1005
|
|
|
1006 c = rev->data;
|
|
|
1007
|
|
|
1008 do {
|
|
354
|
1009 n = ngx_udp_recv(c, buf, NGX_RESOLVER_UDP_SIZE);
|
|
|
1010
|
|
|
1011 if (n < 0) {
|
|
350
|
1012 return;
|
|
|
1013 }
|
|
|
1014
|
|
|
1015 ngx_resolver_process_response(c->data, buf, n);
|
|
|
1016
|
|
|
1017 } while (rev->ready);
|
|
|
1018 }
|
|
|
1019
|
|
|
1020
|
|
|
1021 static void
|
|
|
1022 ngx_resolver_process_response(ngx_resolver_t *r, u_char *buf, size_t n)
|
|
|
1023 {
|
|
|
1024 char *err;
|
|
|
1025 size_t len;
|
|
416
|
1026 ngx_uint_t i, times, ident, qident, flags, code, nqs, nan,
|
|
|
1027 qtype, qclass;
|
|
|
1028 ngx_queue_t *q;
|
|
350
|
1029 ngx_resolver_qs_t *qs;
|
|
416
|
1030 ngx_resolver_node_t *rn;
|
|
350
|
1031 ngx_resolver_query_t *query;
|
|
|
1032
|
|
416
|
1033 if ((size_t) n < sizeof(ngx_resolver_query_t)) {
|
|
350
|
1034 goto short_response;
|
|
|
1035 }
|
|
|
1036
|
|
|
1037 query = (ngx_resolver_query_t *) buf;
|
|
|
1038
|
|
|
1039 ident = (query->ident_hi << 8) + query->ident_lo;
|
|
|
1040 flags = (query->flags_hi << 8) + query->flags_lo;
|
|
|
1041 nqs = (query->nqs_hi << 8) + query->nqs_lo;
|
|
|
1042 nan = (query->nan_hi << 8) + query->nan_lo;
|
|
|
1043
|
|
|
1044 ngx_log_debug6(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
676
|
1045 "resolver DNS response %ui fl:%04Xui %ui/%ui/%ud/%ud",
|
|
350
|
1046 ident, flags, nqs, nan,
|
|
|
1047 (query->nns_hi << 8) + query->nns_lo,
|
|
|
1048 (query->nar_hi << 8) + query->nar_lo);
|
|
|
1049
|
|
|
1050 if (!(flags & 0x8000)) {
|
|
|
1051 ngx_log_error(r->log_level, r->log, 0,
|
|
372
|
1052 "invalid DNS response %ui fl:%04Xui", ident, flags);
|
|
350
|
1053 return;
|
|
|
1054 }
|
|
|
1055
|
|
|
1056 code = flags & 0x7f;
|
|
|
1057
|
|
416
|
1058 if (code == NGX_RESOLVE_FORMERR) {
|
|
|
1059
|
|
|
1060 times = 0;
|
|
|
1061
|
|
|
1062 for (q = ngx_queue_head(&r->name_resend_queue);
|
|
|
1063 q != ngx_queue_sentinel(&r->name_resend_queue) || times++ < 100;
|
|
|
1064 q = ngx_queue_next(q))
|
|
|
1065 {
|
|
|
1066 rn = ngx_queue_data(q, ngx_resolver_node_t, queue);
|
|
|
1067 qident = (rn->query[0] << 8) + rn->query[1];
|
|
|
1068
|
|
|
1069 if (qident == ident) {
|
|
|
1070 ngx_log_error(r->log_level, r->log, 0,
|
|
|
1071 "DNS error (%ui: %s), query id:%ui, name:\"%*s\"",
|
|
|
1072 code, ngx_resolver_strerror(code), ident,
|
|
|
1073 rn->nlen, rn->name);
|
|
|
1074 return;
|
|
|
1075 }
|
|
|
1076 }
|
|
|
1077
|
|
|
1078 goto dns_error;
|
|
|
1079 }
|
|
|
1080
|
|
|
1081 if (code > NGX_RESOLVE_REFUSED) {
|
|
|
1082 goto dns_error;
|
|
350
|
1083 }
|
|
|
1084
|
|
|
1085 if (nqs != 1) {
|
|
|
1086 err = "invalid number of questions in DNS response";
|
|
|
1087 goto done;
|
|
|
1088 }
|
|
|
1089
|
|
|
1090 i = sizeof(ngx_resolver_query_t);
|
|
|
1091
|
|
|
1092 while (i < (ngx_uint_t) n) {
|
|
|
1093 if (buf[i] == '\0') {
|
|
|
1094 goto found;
|
|
|
1095 }
|
|
|
1096
|
|
|
1097 len = buf[i];
|
|
|
1098 i += 1 + len;
|
|
|
1099 }
|
|
|
1100
|
|
|
1101 goto short_response;
|
|
|
1102
|
|
|
1103 found:
|
|
|
1104
|
|
|
1105 if (i++ == 0) {
|
|
|
1106 err = "zero-length domain name in DNS response";
|
|
|
1107 goto done;
|
|
|
1108 }
|
|
|
1109
|
|
|
1110 if (i + sizeof(ngx_resolver_qs_t) + nan * (2 + sizeof(ngx_resolver_an_t))
|
|
|
1111 > (ngx_uint_t) n)
|
|
|
1112 {
|
|
|
1113 goto short_response;
|
|
|
1114 }
|
|
|
1115
|
|
|
1116 qs = (ngx_resolver_qs_t *) &buf[i];
|
|
|
1117
|
|
|
1118 qtype = (qs->type_hi << 8) + qs->type_lo;
|
|
|
1119 qclass = (qs->class_hi << 8) + qs->class_lo;
|
|
|
1120
|
|
|
1121 ngx_log_debug2(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
372
|
1122 "resolver DNS response qt:%ui cl:%ui", qtype, qclass);
|
|
350
|
1123
|
|
|
1124 if (qclass != 1) {
|
|
|
1125 ngx_log_error(r->log_level, r->log, 0,
|
|
372
|
1126 "unknown query class %ui in DNS response", qclass);
|
|
350
|
1127 return;
|
|
|
1128 }
|
|
|
1129
|
|
|
1130 switch (qtype) {
|
|
|
1131
|
|
|
1132 case NGX_RESOLVE_A:
|
|
|
1133
|
|
|
1134 ngx_resolver_process_a(r, buf, n, ident, code, nan,
|
|
356
|
1135 i + sizeof(ngx_resolver_qs_t));
|
|
350
|
1136
|
|
|
1137 break;
|
|
|
1138
|
|
|
1139 case NGX_RESOLVE_PTR:
|
|
|
1140
|
|
356
|
1141 ngx_resolver_process_ptr(r, buf, n, ident, code, nan);
|
|
350
|
1142
|
|
|
1143 break;
|
|
|
1144
|
|
|
1145 default:
|
|
|
1146 ngx_log_error(r->log_level, r->log, 0,
|
|
372
|
1147 "unknown query type %ui in DNS response", qtype);
|
|
350
|
1148 return;
|
|
|
1149 }
|
|
|
1150
|
|
|
1151 return;
|
|
|
1152
|
|
|
1153 short_response:
|
|
|
1154
|
|
|
1155 err = "short dns response";
|
|
|
1156
|
|
|
1157 done:
|
|
|
1158
|
|
|
1159 ngx_log_error(r->log_level, r->log, 0, err);
|
|
|
1160
|
|
|
1161 return;
|
|
416
|
1162
|
|
|
1163 dns_error:
|
|
|
1164
|
|
|
1165 ngx_log_error(r->log_level, r->log, 0,
|
|
|
1166 "DNS error (%ui: %s), query id:%ui",
|
|
|
1167 code, ngx_resolver_strerror(code), ident);
|
|
|
1168 return;
|
|
350
|
1169 }
|
|
|
1170
|
|
|
1171
|
|
|
1172 static void
|
|
|
1173 ngx_resolver_process_a(ngx_resolver_t *r, u_char *buf, size_t last,
|
|
|
1174 ngx_uint_t ident, ngx_uint_t code, ngx_uint_t nan, ngx_uint_t ans)
|
|
|
1175 {
|
|
|
1176 char *err;
|
|
|
1177 u_char *cname;
|
|
|
1178 size_t len;
|
|
650
|
1179 int32_t ttl;
|
|
350
|
1180 uint32_t hash;
|
|
|
1181 in_addr_t addr, *addrs;
|
|
|
1182 ngx_str_t name;
|
|
|
1183 ngx_uint_t qtype, qident, naddrs, a, i, n, start;
|
|
|
1184 ngx_resolver_an_t *an;
|
|
|
1185 ngx_resolver_ctx_t *ctx, *next;
|
|
|
1186 ngx_resolver_node_t *rn;
|
|
|
1187
|
|
|
1188 if (ngx_resolver_copy(r, &name, buf, &buf[12], &buf[last]) != NGX_OK) {
|
|
|
1189 return;
|
|
|
1190 }
|
|
|
1191
|
|
|
1192 ngx_log_debug1(NGX_LOG_DEBUG_CORE, r->log, 0, "resolver qs:%V", &name);
|
|
|
1193
|
|
|
1194 hash = ngx_crc32_short(name.data, name.len);
|
|
|
1195
|
|
|
1196 /* lock name mutex */
|
|
|
1197
|
|
|
1198 rn = ngx_resolver_lookup_name(r, &name, hash);
|
|
|
1199
|
|
|
1200 if (rn == NULL || rn->query == NULL) {
|
|
|
1201 ngx_log_error(r->log_level, r->log, 0,
|
|
|
1202 "unexpected response for %V", &name);
|
|
|
1203 goto failed;
|
|
|
1204 }
|
|
|
1205
|
|
|
1206 qident = (rn->query[0] << 8) + rn->query[1];
|
|
|
1207
|
|
|
1208 if (ident != qident) {
|
|
|
1209 ngx_log_error(r->log_level, r->log, 0,
|
|
372
|
1210 "wrong ident %ui response for %V, expect %ui",
|
|
350
|
1211 ident, &name, qident);
|
|
|
1212 goto failed;
|
|
|
1213 }
|
|
|
1214
|
|
528
|
1215 ngx_resolver_free(r, name.data);
|
|
|
1216
|
|
356
|
1217 if (code == 0 && nan == 0) {
|
|
|
1218 code = 3; /* NXDOMAIN */
|
|
|
1219 }
|
|
|
1220
|
|
350
|
1221 if (code) {
|
|
|
1222 next = rn->waiting;
|
|
|
1223 rn->waiting = NULL;
|
|
|
1224
|
|
|
1225 ngx_queue_remove(&rn->queue);
|
|
|
1226
|
|
|
1227 ngx_rbtree_delete(&r->name_rbtree, &rn->node);
|
|
|
1228
|
|
|
1229 ngx_resolver_free_node(r, rn);
|
|
|
1230
|
|
|
1231 /* unlock name mutex */
|
|
|
1232
|
|
|
1233 while (next) {
|
|
|
1234 ctx = next;
|
|
|
1235 ctx->state = code;
|
|
|
1236 next = ctx->next;
|
|
|
1237
|
|
|
1238 ctx->handler(ctx);
|
|
|
1239 }
|
|
|
1240
|
|
|
1241 return;
|
|
|
1242 }
|
|
|
1243
|
|
|
1244 i = ans;
|
|
|
1245 naddrs = 0;
|
|
|
1246 addr = 0;
|
|
|
1247 addrs = NULL;
|
|
|
1248 cname = NULL;
|
|
|
1249 qtype = 0;
|
|
650
|
1250 ttl = 0;
|
|
350
|
1251
|
|
|
1252 for (a = 0; a < nan; a++) {
|
|
|
1253
|
|
|
1254 start = i;
|
|
|
1255
|
|
|
1256 while (i < last) {
|
|
|
1257
|
|
|
1258 if (buf[i] & 0xc0) {
|
|
|
1259 i += 2;
|
|
|
1260 goto found;
|
|
|
1261 }
|
|
|
1262
|
|
|
1263 if (buf[i] == 0) {
|
|
|
1264 i++;
|
|
|
1265 goto test_length;
|
|
|
1266 }
|
|
|
1267
|
|
|
1268 i += 1 + buf[i];
|
|
|
1269 }
|
|
|
1270
|
|
|
1271 goto short_response;
|
|
|
1272
|
|
|
1273 test_length:
|
|
|
1274
|
|
|
1275 if (i - start < 2) {
|
|
|
1276 err = "invalid name in dns response";
|
|
|
1277 goto invalid;
|
|
|
1278 }
|
|
|
1279
|
|
|
1280 found:
|
|
|
1281
|
|
|
1282 if (i + sizeof(ngx_resolver_an_t) >= last) {
|
|
|
1283 goto short_response;
|
|
|
1284 }
|
|
|
1285
|
|
|
1286 an = (ngx_resolver_an_t *) &buf[i];
|
|
|
1287
|
|
|
1288 qtype = (an->type_hi << 8) + an->type_lo;
|
|
|
1289 len = (an->len_hi << 8) + an->len_lo;
|
|
650
|
1290 ttl = (an->ttl[0] << 24) + (an->ttl[1] << 16)
|
|
|
1291 + (an->ttl[2] << 8) + (an->ttl[3]);
|
|
|
1292
|
|
|
1293 if (ttl < 0) {
|
|
|
1294 ttl = 0;
|
|
|
1295 }
|
|
350
|
1296
|
|
|
1297 if (qtype == NGX_RESOLVE_A) {
|
|
|
1298
|
|
|
1299 i += sizeof(ngx_resolver_an_t);
|
|
|
1300
|
|
|
1301 if (i + len > last) {
|
|
|
1302 goto short_response;
|
|
|
1303 }
|
|
|
1304
|
|
|
1305 addr = htonl((buf[i] << 24) + (buf[i + 1] << 16)
|
|
|
1306 + (buf[i + 2] << 8) + (buf[i + 3]));
|
|
|
1307
|
|
|
1308 naddrs++;
|
|
|
1309
|
|
|
1310 i += len;
|
|
|
1311
|
|
|
1312 } else if (qtype == NGX_RESOLVE_CNAME) {
|
|
|
1313 cname = &buf[i] + sizeof(ngx_resolver_an_t);
|
|
|
1314 i += sizeof(ngx_resolver_an_t) + len;
|
|
372
|
1315
|
|
|
1316 } else if (qtype == NGX_RESOLVE_DNAME) {
|
|
|
1317 i += sizeof(ngx_resolver_an_t) + len;
|
|
|
1318
|
|
|
1319 } else {
|
|
|
1320 ngx_log_error(r->log_level, r->log, 0,
|
|
|
1321 "unexpected qtype %ui", qtype);
|
|
350
|
1322 }
|
|
|
1323 }
|
|
|
1324
|
|
650
|
1325 ngx_log_debug3(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
|
1326 "resolver naddrs:%ui cname:%p ttl:%d",
|
|
|
1327 naddrs, cname, ttl);
|
|
350
|
1328
|
|
|
1329 if (naddrs) {
|
|
|
1330
|
|
|
1331 if (naddrs == 1) {
|
|
|
1332 rn->u.addr = addr;
|
|
|
1333
|
|
|
1334 } else {
|
|
|
1335
|
|
|
1336 addrs = ngx_resolver_alloc(r, naddrs * sizeof(in_addr_t));
|
|
|
1337 if (addrs == NULL) {
|
|
|
1338 return;
|
|
|
1339 }
|
|
|
1340
|
|
|
1341 n = 0;
|
|
|
1342 i = ans;
|
|
|
1343
|
|
|
1344 for (a = 0; a < nan; a++) {
|
|
|
1345
|
|
|
1346 for ( ;; ) {
|
|
|
1347
|
|
|
1348 if (buf[i] & 0xc0) {
|
|
|
1349 i += 2;
|
|
|
1350 goto ok;
|
|
|
1351 }
|
|
|
1352
|
|
|
1353 if (buf[i] == 0) {
|
|
|
1354 i++;
|
|
|
1355 goto ok;
|
|
|
1356 }
|
|
|
1357
|
|
|
1358 i += 1 + buf[i];
|
|
|
1359 }
|
|
|
1360
|
|
|
1361 ok:
|
|
|
1362
|
|
|
1363 an = (ngx_resolver_an_t *) &buf[i];
|
|
|
1364
|
|
|
1365 qtype = (an->type_hi << 8) + an->type_lo;
|
|
|
1366 len = (an->len_hi << 8) + an->len_lo;
|
|
|
1367
|
|
|
1368 i += sizeof(ngx_resolver_an_t);
|
|
|
1369
|
|
|
1370 if (qtype == NGX_RESOLVE_A) {
|
|
|
1371
|
|
|
1372 addrs[n++] = htonl((buf[i] << 24) + (buf[i + 1] << 16)
|
|
|
1373 + (buf[i + 2] << 8) + (buf[i + 3]));
|
|
|
1374
|
|
|
1375 if (n == naddrs) {
|
|
|
1376 break;
|
|
|
1377 }
|
|
|
1378 }
|
|
|
1379
|
|
|
1380 i += len;
|
|
|
1381 }
|
|
|
1382
|
|
|
1383 rn->u.addrs = addrs;
|
|
|
1384
|
|
|
1385 addrs = ngx_resolver_dup(r, rn->u.addrs,
|
|
|
1386 naddrs * sizeof(in_addr_t));
|
|
|
1387 if (addrs == NULL) {
|
|
|
1388 return;
|
|
|
1389 }
|
|
|
1390 }
|
|
|
1391
|
|
|
1392 rn->naddrs = (u_short) naddrs;
|
|
|
1393
|
|
|
1394 ngx_queue_remove(&rn->queue);
|
|
|
1395
|
|
650
|
1396 rn->valid = ngx_time() + (r->valid ? r->valid : ttl);
|
|
350
|
1397 rn->expire = ngx_time() + r->expire;
|
|
|
1398
|
|
|
1399 ngx_queue_insert_head(&r->name_expire_queue, &rn->queue);
|
|
|
1400
|
|
|
1401 next = rn->waiting;
|
|
|
1402 rn->waiting = NULL;
|
|
|
1403
|
|
|
1404 /* unlock name mutex */
|
|
|
1405
|
|
|
1406 while (next) {
|
|
|
1407 ctx = next;
|
|
|
1408 ctx->state = NGX_OK;
|
|
|
1409 ctx->naddrs = naddrs;
|
|
|
1410 ctx->addrs = (naddrs == 1) ? &ctx->addr : addrs;
|
|
|
1411 ctx->addr = addr;
|
|
|
1412 next = ctx->next;
|
|
|
1413
|
|
|
1414 ctx->handler(ctx);
|
|
|
1415 }
|
|
|
1416
|
|
442
|
1417 if (naddrs > 1) {
|
|
350
|
1418 ngx_resolver_free(r, addrs);
|
|
|
1419 }
|
|
|
1420
|
|
674
|
1421 ngx_resolver_free(r, rn->query);
|
|
|
1422 rn->query = NULL;
|
|
|
1423
|
|
350
|
1424 return;
|
|
|
1425
|
|
|
1426 } else if (cname) {
|
|
|
1427
|
|
|
1428 /* CNAME only */
|
|
|
1429
|
|
|
1430 if (ngx_resolver_copy(r, &name, buf, cname, &buf[last]) != NGX_OK) {
|
|
|
1431 return;
|
|
|
1432 }
|
|
|
1433
|
|
|
1434 ngx_log_debug1(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
|
1435 "resolver cname:\"%V\"", &name);
|
|
|
1436
|
|
356
|
1437 ngx_queue_remove(&rn->queue);
|
|
|
1438
|
|
350
|
1439 rn->cnlen = (u_short) name.len;
|
|
|
1440 rn->u.cname = name.data;
|
|
650
|
1441
|
|
|
1442 rn->valid = ngx_time() + (r->valid ? r->valid : ttl);
|
|
350
|
1443 rn->expire = ngx_time() + r->expire;
|
|
|
1444
|
|
|
1445 ngx_queue_insert_head(&r->name_expire_queue, &rn->queue);
|
|
|
1446
|
|
|
1447 ctx = rn->waiting;
|
|
|
1448 rn->waiting = NULL;
|
|
|
1449
|
|
|
1450 if (ctx) {
|
|
|
1451 ctx->name = name;
|
|
|
1452
|
|
|
1453 (void) ngx_resolve_name_locked(r, ctx);
|
|
|
1454 }
|
|
|
1455
|
|
674
|
1456 ngx_resolver_free(r, rn->query);
|
|
|
1457 rn->query = NULL;
|
|
|
1458
|
|
350
|
1459 return;
|
|
|
1460 }
|
|
|
1461
|
|
|
1462 ngx_log_error(r->log_level, r->log, 0,
|
|
372
|
1463 "no A or CNAME types in DNS responses, unknown query type: %ui",
|
|
|
1464 qtype);
|
|
350
|
1465 return;
|
|
|
1466
|
|
|
1467 short_response:
|
|
|
1468
|
|
|
1469 err = "short dns response";
|
|
|
1470
|
|
|
1471 invalid:
|
|
|
1472
|
|
|
1473 /* unlock name mutex */
|
|
|
1474
|
|
|
1475 ngx_log_error(r->log_level, r->log, 0, err);
|
|
|
1476
|
|
|
1477 return;
|
|
|
1478
|
|
|
1479 failed:
|
|
|
1480
|
|
|
1481 /* unlock name mutex */
|
|
|
1482
|
|
528
|
1483 ngx_resolver_free(r, name.data);
|
|
|
1484
|
|
350
|
1485 return;
|
|
|
1486 }
|
|
|
1487
|
|
|
1488
|
|
|
1489 static void
|
|
|
1490 ngx_resolver_process_ptr(ngx_resolver_t *r, u_char *buf, size_t n,
|
|
356
|
1491 ngx_uint_t ident, ngx_uint_t code, ngx_uint_t nan)
|
|
350
|
1492 {
|
|
|
1493 char *err;
|
|
|
1494 size_t len;
|
|
|
1495 in_addr_t addr;
|
|
650
|
1496 int32_t ttl;
|
|
350
|
1497 ngx_int_t digit;
|
|
|
1498 ngx_str_t name;
|
|
622
|
1499 ngx_uint_t i, mask, qident;
|
|
350
|
1500 ngx_resolver_an_t *an;
|
|
|
1501 ngx_resolver_ctx_t *ctx, *next;
|
|
|
1502 ngx_resolver_node_t *rn;
|
|
|
1503
|
|
|
1504 if (ngx_resolver_copy(r, NULL, buf, &buf[12], &buf[n]) != NGX_OK) {
|
|
|
1505 goto invalid_in_addr_arpa;
|
|
|
1506 }
|
|
|
1507
|
|
|
1508 addr = 0;
|
|
|
1509 i = 12;
|
|
|
1510
|
|
|
1511 for (mask = 0; mask < 32; mask += 8) {
|
|
|
1512 len = buf[i++];
|
|
|
1513
|
|
|
1514 digit = ngx_atoi(&buf[i], len);
|
|
|
1515 if (digit == NGX_ERROR || digit > 255) {
|
|
|
1516 goto invalid_in_addr_arpa;
|
|
|
1517 }
|
|
|
1518
|
|
|
1519 addr += digit << mask;
|
|
|
1520 i += len;
|
|
|
1521 }
|
|
|
1522
|
|
|
1523 if (ngx_strcmp(&buf[i], "\7in-addr\4arpa") != 0) {
|
|
|
1524 goto invalid_in_addr_arpa;
|
|
|
1525 }
|
|
|
1526
|
|
|
1527 /* lock addr mutex */
|
|
|
1528
|
|
|
1529 rn = ngx_resolver_lookup_addr(r, addr);
|
|
|
1530
|
|
|
1531 if (rn == NULL || rn->query == NULL) {
|
|
|
1532 ngx_log_error(r->log_level, r->log, 0,
|
|
|
1533 "unexpected response for %ud.%ud.%ud.%ud",
|
|
|
1534 (addr >> 24) & 0xff, (addr >> 16) & 0xff,
|
|
|
1535 (addr >> 8) & 0xff, addr & 0xff);
|
|
|
1536 goto failed;
|
|
|
1537 }
|
|
|
1538
|
|
|
1539 qident = (rn->query[0] << 8) + rn->query[1];
|
|
|
1540
|
|
|
1541 if (ident != qident) {
|
|
|
1542 ngx_log_error(r->log_level, r->log, 0,
|
|
372
|
1543 "wrong ident %ui response for %ud.%ud.%ud.%ud, expect %ui",
|
|
|
1544 ident, (addr >> 24) & 0xff, (addr >> 16) & 0xff,
|
|
|
1545 (addr >> 8) & 0xff, addr & 0xff, qident);
|
|
350
|
1546 goto failed;
|
|
|
1547 }
|
|
|
1548
|
|
356
|
1549 if (code == 0 && nan == 0) {
|
|
|
1550 code = 3; /* NXDOMAIN */
|
|
|
1551 }
|
|
|
1552
|
|
350
|
1553 if (code) {
|
|
|
1554 next = rn->waiting;
|
|
|
1555 rn->waiting = NULL;
|
|
|
1556
|
|
|
1557 ngx_queue_remove(&rn->queue);
|
|
|
1558
|
|
|
1559 ngx_rbtree_delete(&r->addr_rbtree, &rn->node);
|
|
|
1560
|
|
|
1561 ngx_resolver_free_node(r, rn);
|
|
|
1562
|
|
|
1563 /* unlock addr mutex */
|
|
|
1564
|
|
|
1565 while (next) {
|
|
|
1566 ctx = next;
|
|
|
1567 ctx->state = code;
|
|
|
1568 next = ctx->next;
|
|
|
1569
|
|
|
1570 ctx->handler(ctx);
|
|
|
1571 }
|
|
|
1572
|
|
|
1573 return;
|
|
|
1574 }
|
|
|
1575
|
|
|
1576 i += sizeof("\7in-addr\4arpa") + sizeof(ngx_resolver_qs_t);
|
|
|
1577
|
|
|
1578 if (i + 2 + sizeof(ngx_resolver_an_t) > (ngx_uint_t) n) {
|
|
|
1579 goto short_response;
|
|
|
1580 }
|
|
|
1581
|
|
|
1582 /* compression pointer to "XX.XX.XX.XX.in-addr.arpa */
|
|
|
1583
|
|
|
1584 if (buf[i] != 0xc0 || buf[i + 1] != 0x0c) {
|
|
|
1585 err = "invalid in-addr.arpa name in DNS response";
|
|
|
1586 goto invalid;
|
|
|
1587 }
|
|
|
1588
|
|
|
1589 an = (ngx_resolver_an_t *) &buf[i + 2];
|
|
|
1590
|
|
|
1591 len = (an->len_hi << 8) + an->len_lo;
|
|
650
|
1592 ttl = (an->ttl[0] << 24) + (an->ttl[1] << 16)
|
|
|
1593 + (an->ttl[2] << 8) + (an->ttl[3]);
|
|
|
1594
|
|
|
1595 if (ttl < 0) {
|
|
|
1596 ttl = 0;
|
|
|
1597 }
|
|
350
|
1598
|
|
|
1599 ngx_log_debug3(NGX_LOG_DEBUG_CORE, r->log, 0,
|
|
622
|
1600 "resolver qt:%ui cl:%ui len:%uz",
|
|
|
1601 (an->type_hi << 8) + an->type_lo,
|
|
|
1602 (an->class_hi << 8) + an->class_lo, len);
|
|
350
|
1603
|
|
|
1604 i += 2 + sizeof(ngx_resolver_an_t);
|
|
|
1605
|
|
|
1606 if (i + len > (ngx_uint_t) n) {
|
|
|
1607 goto short_response;
|
|
|
1608 }
|
|
|
1609
|
|
|
1610 if (ngx_resolver_copy(r, &name, buf, &buf[i], &buf[n]) != NGX_OK) {
|
|
|
1611 return;
|
|
|
1612 }
|
|
|
1613
|
|
|
1614 ngx_log_debug1(NGX_LOG_DEBUG_CORE, r->log, 0, "resolver an:%V", &name);
|
|
|
1615
|
|
442
|
1616 if (name.len != (size_t) rn->nlen
|
|
|
1617 || ngx_strncmp(name.data, rn->name, name.len) != 0)
|
|
350
|
1618 {
|
|
442
|
1619 if (rn->nlen) {
|
|
|
1620 ngx_resolver_free(r, rn->name);
|
|
|
1621 }
|
|
|
1622
|
|
|
1623 rn->nlen = (u_short) name.len;
|
|
350
|
1624 rn->name = name.data;
|
|
|
1625
|
|
442
|
1626 name.data = ngx_resolver_dup(r, rn->name, name.len);
|
|
350
|
1627 if (name.data == NULL) {
|
|
|
1628 goto failed;
|
|
|
1629 }
|
|
|
1630 }
|
|
|
1631
|
|
|
1632 ngx_queue_remove(&rn->queue);
|
|
|
1633
|
|
650
|
1634 rn->valid = ngx_time() + (r->valid ? r->valid : ttl);
|
|
350
|
1635 rn->expire = ngx_time() + r->expire;
|
|
|
1636
|
|
|
1637 ngx_queue_insert_head(&r->addr_expire_queue, &rn->queue);
|
|
|
1638
|
|
|
1639 next = rn->waiting;
|
|
|
1640 rn->waiting = NULL;
|
|
|
1641
|
|
|
1642 /* unlock addr mutex */
|
|
|
1643
|
|
|
1644 while (next) {
|
|
|
1645 ctx = next;
|
|
|
1646 ctx->state = NGX_OK;
|
|
|
1647 ctx->name = name;
|
|
|
1648 next = ctx->next;
|
|
|
1649
|
|
|
1650 ctx->handler(ctx);
|
|
|
1651 }
|
|
|
1652
|
|
|
1653 ngx_resolver_free(r, name.data);
|
|
|
1654
|
|
|
1655 return;
|
|
|
1656
|
|
|
1657 invalid_in_addr_arpa:
|
|
|
1658
|
|
|
1659 ngx_log_error(r->log_level, r->log, 0,
|
|
|
1660 "invalid in-addr.arpa name in DNS response");
|
|
|
1661 return;
|
|
|
1662
|
|
|
1663 short_response:
|
|
|
1664
|
|
|
1665 err = "short DNS response";
|
|
|
1666
|
|
|
1667 invalid:
|
|
|
1668
|
|
|
1669 /* unlock addr mutex */
|
|
|
1670
|
|
|
1671 ngx_log_error(r->log_level, r->log, 0, err);
|
|
|
1672
|
|
|
1673 return;
|
|
|
1674
|
|
|
1675 failed:
|
|
|
1676
|
|
|
1677 /* unlock addr mutex */
|
|
|
1678
|
|
|
1679 return;
|
|
|
1680 }
|
|
|
1681
|
|
|
1682
|
|
|
1683 static ngx_resolver_node_t *
|
|
|
1684 ngx_resolver_lookup_name(ngx_resolver_t *r, ngx_str_t *name, uint32_t hash)
|
|
|
1685 {
|
|
|
1686 ngx_int_t rc;
|
|
|
1687 ngx_rbtree_node_t *node, *sentinel;
|
|
|
1688 ngx_resolver_node_t *rn;
|
|
|
1689
|
|
|
1690 node = r->name_rbtree.root;
|
|
|
1691 sentinel = r->name_rbtree.sentinel;
|
|
|
1692
|
|
|
1693 while (node != sentinel) {
|
|
|
1694
|
|
|
1695 if (hash < node->key) {
|
|
|
1696 node = node->left;
|
|
|
1697 continue;
|
|
|
1698 }
|
|
|
1699
|
|
|
1700 if (hash > node->key) {
|
|
|
1701 node = node->right;
|
|
|
1702 continue;
|
|
|
1703 }
|
|
|
1704
|
|
|
1705 /* hash == node->key */
|
|
|
1706
|
|
664
|
1707 rn = (ngx_resolver_node_t *) node;
|
|
|
1708
|
|
|
1709 rc = ngx_memn2cmp(name->data, rn->name, name->len, rn->nlen);
|
|
|
1710
|
|
|
1711 if (rc == 0) {
|
|
|
1712 return rn;
|
|
|
1713 }
|
|
|
1714
|
|
|
1715 node = (rc < 0) ? node->left : node->right;
|
|
350
|
1716 }
|
|
|
1717
|
|
|
1718 /* not found */
|
|
|
1719
|
|
|
1720 return NULL;
|
|
|
1721 }
|
|
|
1722
|
|
|
1723
|
|
|
1724 static ngx_resolver_node_t *
|
|
|
1725 ngx_resolver_lookup_addr(ngx_resolver_t *r, in_addr_t addr)
|
|
|
1726 {
|
|
|
1727 ngx_rbtree_node_t *node, *sentinel;
|
|
|
1728
|
|
|
1729 node = r->addr_rbtree.root;
|
|
|
1730 sentinel = r->addr_rbtree.sentinel;
|
|
|
1731
|
|
|
1732 while (node != sentinel) {
|
|
|
1733
|
|
|
1734 if (addr < node->key) {
|
|
|
1735 node = node->left;
|
|
|
1736 continue;
|
|
|
1737 }
|
|
|
1738
|
|
|
1739 if (addr > node->key) {
|
|
|
1740 node = node->right;
|
|
|
1741 continue;
|
|
|
1742 }
|
|
|
1743
|
|
|
1744 /* addr == node->key */
|
|
|
1745
|
|
|
1746 return (ngx_resolver_node_t *) node;
|
|
|
1747 }
|
|
|
1748
|
|
|
1749 /* not found */
|
|
|
1750
|
|
|
1751 return NULL;
|
|
|
1752 }
|
|
|
1753
|
|
|
1754
|
|
|
1755 static void
|
|
|
1756 ngx_resolver_rbtree_insert_value(ngx_rbtree_node_t *temp,
|
|
|
1757 ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel)
|
|
|
1758 {
|
|
|
1759 ngx_rbtree_node_t **p;
|
|
|
1760 ngx_resolver_node_t *rn, *rn_temp;
|
|
|
1761
|
|
|
1762 for ( ;; ) {
|
|
|
1763
|
|
|
1764 if (node->key < temp->key) {
|
|
|
1765
|
|
|
1766 p = &temp->left;
|
|
|
1767
|
|
|
1768 } else if (node->key > temp->key) {
|
|
|
1769
|
|
|
1770 p = &temp->right;
|
|
|
1771
|
|
|
1772 } else { /* node->key == temp->key */
|
|
|
1773
|
|
|
1774 rn = (ngx_resolver_node_t *) node;
|
|
|
1775 rn_temp = (ngx_resolver_node_t *) temp;
|
|
|
1776
|
|
528
|
1777 p = (ngx_memn2cmp(rn->name, rn_temp->name, rn->nlen, rn_temp->nlen)
|
|
|
1778 < 0) ? &temp->left : &temp->right;
|
|
350
|
1779 }
|
|
|
1780
|
|
|
1781 if (*p == sentinel) {
|
|
|
1782 break;
|
|
|
1783 }
|
|
|
1784
|
|
|
1785 temp = *p;
|
|
|
1786 }
|
|
|
1787
|
|
|
1788 *p = node;
|
|
|
1789 node->parent = temp;
|
|
|
1790 node->left = sentinel;
|
|
|
1791 node->right = sentinel;
|
|
|
1792 ngx_rbt_red(node);
|
|
|
1793 }
|
|
|
1794
|
|
|
1795
|
|
|
1796 static ngx_int_t
|
|
|
1797 ngx_resolver_create_name_query(ngx_resolver_node_t *rn, ngx_resolver_ctx_t *ctx)
|
|
|
1798 {
|
|
|
1799 u_char *p, *s;
|
|
544
|
1800 size_t len, nlen;
|
|
350
|
1801 ngx_uint_t ident;
|
|
|
1802 ngx_resolver_qs_t *qs;
|
|
|
1803 ngx_resolver_query_t *query;
|
|
|
1804
|
|
544
|
1805 nlen = ctx->name.len ? (1 + ctx->name.len + 1) : 1;
|
|
|
1806
|
|
|
1807 len = sizeof(ngx_resolver_query_t) + nlen + sizeof(ngx_resolver_qs_t);
|
|
|
1808
|
|
|
1809 p = ngx_resolver_alloc(ctx->resolver, len);
|
|
350
|
1810 if (p == NULL) {
|
|
|
1811 return NGX_ERROR;
|
|
|
1812 }
|
|
|
1813
|
|
|
1814 rn->qlen = (u_short) len;
|
|
|
1815 rn->query = p;
|
|
|
1816
|
|
|
1817 query = (ngx_resolver_query_t *) p;
|
|
|
1818
|
|
|
1819 ident = ngx_random();
|
|
|
1820
|
|
|
1821 ngx_log_debug2(NGX_LOG_DEBUG_CORE, ctx->resolver->log, 0,
|
|
|
1822 "resolve: \"%V\" %i", &ctx->name, ident & 0xffff);
|
|
|
1823
|
|
|
1824 query->ident_hi = (u_char) ((ident >> 8) & 0xff);
|
|
|
1825 query->ident_lo = (u_char) (ident & 0xff);
|
|
|
1826
|
|
|
1827 /* recursion query */
|
|
|
1828 query->flags_hi = 1; query->flags_lo = 0;
|
|
|
1829
|
|
|
1830 /* one question */
|
|
|
1831 query->nqs_hi = 0; query->nqs_lo = 1;
|
|
|
1832 query->nan_hi = 0; query->nan_lo = 0;
|
|
|
1833 query->nns_hi = 0; query->nns_lo = 0;
|
|
|
1834 query->nar_hi = 0; query->nar_lo = 0;
|
|
|
1835
|
|
544
|
1836 p += sizeof(ngx_resolver_query_t) + nlen;
|
|
350
|
1837
|
|
|
1838 qs = (ngx_resolver_qs_t *) p;
|
|
|
1839
|
|
|
1840 /* query type */
|
|
|
1841 qs->type_hi = 0; qs->type_lo = (u_char) ctx->type;
|
|
|
1842
|
|
|
1843 /* IP query class */
|
|
|
1844 qs->class_hi = 0; qs->class_lo = 1;
|
|
|
1845
|
|
|
1846 /* convert "www.example.com" to "\3www\7example\3com\0" */
|
|
|
1847
|
|
|
1848 len = 0;
|
|
|
1849 p--;
|
|
|
1850 *p-- = '\0';
|
|
|
1851
|
|
674
|
1852 if (ctx->name.len == 0) {
|
|
|
1853 return NGX_DECLINED;
|
|
|
1854 }
|
|
|
1855
|
|
350
|
1856 for (s = ctx->name.data + ctx->name.len - 1; s >= ctx->name.data; s--) {
|
|
|
1857 if (*s != '.') {
|
|
|
1858 *p = *s;
|
|
|
1859 len++;
|
|
|
1860
|
|
|
1861 } else {
|
|
668
|
1862 if (len == 0 || len > 255) {
|
|
372
|
1863 return NGX_DECLINED;
|
|
|
1864 }
|
|
|
1865
|
|
350
|
1866 *p = (u_char) len;
|
|
|
1867 len = 0;
|
|
|
1868 }
|
|
|
1869
|
|
|
1870 p--;
|
|
|
1871 }
|
|
|
1872
|
|
668
|
1873 if (len == 0 || len > 255) {
|
|
|
1874 return NGX_DECLINED;
|
|
|
1875 }
|
|
|
1876
|
|
350
|
1877 *p = (u_char) len;
|
|
|
1878
|
|
|
1879 return NGX_OK;
|
|
|
1880 }
|
|
|
1881
|
|
|
1882
|
|
|
1883 /* AF_INET only */
|
|
|
1884
|
|
|
1885 static ngx_int_t
|
|
|
1886 ngx_resolver_create_addr_query(ngx_resolver_node_t *rn, ngx_resolver_ctx_t *ctx)
|
|
|
1887 {
|
|
|
1888 u_char *p, *d;
|
|
|
1889 size_t len;
|
|
|
1890 ngx_int_t n;
|
|
|
1891 ngx_uint_t ident;
|
|
|
1892 ngx_resolver_query_t *query;
|
|
|
1893
|
|
|
1894 len = sizeof(ngx_resolver_query_t)
|
|
|
1895 + sizeof(".255.255.255.255.in-addr.arpa.") - 1
|
|
|
1896 + sizeof(ngx_resolver_qs_t);
|
|
|
1897
|
|
544
|
1898 p = ngx_resolver_alloc(ctx->resolver, len);
|
|
350
|
1899 if (p == NULL) {
|
|
|
1900 return NGX_ERROR;
|
|
|
1901 }
|
|
|
1902
|
|
|
1903 rn->query = p;
|
|
|
1904 query = (ngx_resolver_query_t *) p;
|
|
|
1905
|
|
|
1906 ident = ngx_random();
|
|
|
1907
|
|
|
1908 query->ident_hi = (u_char) ((ident >> 8) & 0xff);
|
|
|
1909 query->ident_lo = (u_char) (ident & 0xff);
|
|
|
1910
|
|
|
1911 /* recursion query */
|
|
|
1912 query->flags_hi = 1; query->flags_lo = 0;
|
|
|
1913
|
|
|
1914 /* one question */
|
|
|
1915 query->nqs_hi = 0; query->nqs_lo = 1;
|
|
|
1916 query->nan_hi = 0; query->nan_lo = 0;
|
|
|
1917 query->nns_hi = 0; query->nns_lo = 0;
|
|
|
1918 query->nar_hi = 0; query->nar_lo = 0;
|
|
|
1919
|
|
|
1920 p += sizeof(ngx_resolver_query_t);
|
|
|
1921
|
|
582
|
1922 for (n = 0; n < 32; n += 8) {
|
|
350
|
1923 d = ngx_sprintf(&p[1], "%ud", (ctx->addr >> n) & 0xff);
|
|
|
1924 *p = (u_char) (d - &p[1]);
|
|
|
1925 p = d;
|
|
|
1926 }
|
|
|
1927
|
|
|
1928 /* query type "PTR", IP query class */
|
|
|
1929 ngx_memcpy(p, "\7in-addr\4arpa\0\0\14\0\1", 18);
|
|
|
1930
|
|
|
1931 rn->qlen = (u_short)
|
|
|
1932 (p + sizeof("\7in-addr\4arpa") + sizeof(ngx_resolver_qs_t)
|
|
|
1933 - rn->query);
|
|
|
1934
|
|
|
1935 return NGX_OK;
|
|
|
1936 }
|
|
|
1937
|
|
|
1938
|
|
|
1939 static ngx_int_t
|
|
|
1940 ngx_resolver_copy(ngx_resolver_t *r, ngx_str_t *name, u_char *buf, u_char *src,
|
|
|
1941 u_char *last)
|
|
|
1942 {
|
|
|
1943 char *err;
|
|
|
1944 u_char *p, *dst;
|
|
|
1945 ssize_t len;
|
|
|
1946 ngx_uint_t i, n;
|
|
|
1947
|
|
|
1948 p = src;
|
|
|
1949 len = -1;
|
|
|
1950
|
|
|
1951 /*
|
|
|
1952 * compression pointers allow to create endless loop, so we set limit;
|
|
|
1953 * 128 pointers should be enough to store 255-byte name
|
|
|
1954 */
|
|
|
1955
|
|
|
1956 for (i = 0; i < 128; i++) {
|
|
|
1957 n = *p++;
|
|
|
1958
|
|
|
1959 if (n == 0) {
|
|
|
1960 goto done;
|
|
|
1961 }
|
|
|
1962
|
|
|
1963 if (n & 0xc0) {
|
|
420
|
1964 n = ((n & 0x3f) << 8) + *p;
|
|
350
|
1965 p = &buf[n];
|
|
|
1966
|
|
|
1967 } else {
|
|
|
1968 len += 1 + n;
|
|
|
1969 p = &p[n];
|
|
|
1970 }
|
|
|
1971
|
|
|
1972 if (p >= last) {
|
|
|
1973 err = "name is out of response";
|
|
|
1974 goto invalid;
|
|
|
1975 }
|
|
|
1976 }
|
|
|
1977
|
|
|
1978 err = "compression pointers loop";
|
|
|
1979
|
|
|
1980 invalid:
|
|
|
1981
|
|
|
1982 ngx_log_error(r->log_level, r->log, 0, err);
|
|
|
1983
|
|
|
1984 return NGX_ERROR;
|
|
|
1985
|
|
|
1986 done:
|
|
|
1987
|
|
|
1988 if (name == NULL) {
|
|
|
1989 return NGX_OK;
|
|
|
1990 }
|
|
|
1991
|
|
542
|
1992 if (len == -1) {
|
|
|
1993 name->len = 0;
|
|
|
1994 name->data = NULL;
|
|
|
1995 return NGX_OK;
|
|
|
1996 }
|
|
|
1997
|
|
350
|
1998 dst = ngx_resolver_alloc(r, len);
|
|
|
1999 if (dst == NULL) {
|
|
|
2000 return NGX_ERROR;
|
|
|
2001 }
|
|
|
2002
|
|
|
2003 name->data = dst;
|
|
|
2004
|
|
|
2005 n = *src++;
|
|
|
2006
|
|
|
2007 for ( ;; ) {
|
|
648
|
2008 if (n & 0xc0) {
|
|
|
2009 n = ((n & 0x3f) << 8) + *src;
|
|
|
2010 src = &buf[n];
|
|
|
2011
|
|
|
2012 n = *src++;
|
|
|
2013
|
|
|
2014 } else {
|
|
350
|
2015 ngx_memcpy(dst, src, n);
|
|
|
2016 dst += n;
|
|
|
2017 src += n;
|
|
|
2018
|
|
|
2019 n = *src++;
|
|
|
2020
|
|
|
2021 if (n != 0) {
|
|
|
2022 *dst++ = '.';
|
|
|
2023 }
|
|
|
2024 }
|
|
|
2025
|
|
|
2026 if (n == 0) {
|
|
|
2027 name->len = dst - name->data;
|
|
|
2028 return NGX_OK;
|
|
|
2029 }
|
|
|
2030 }
|
|
|
2031 }
|
|
|
2032
|
|
|
2033
|
|
|
2034 static void
|
|
|
2035 ngx_resolver_timeout_handler(ngx_event_t *ev)
|
|
|
2036 {
|
|
|
2037 ngx_resolver_ctx_t *ctx;
|
|
|
2038
|
|
|
2039 ctx = ev->data;
|
|
|
2040
|
|
|
2041 ctx->state = NGX_RESOLVE_TIMEDOUT;
|
|
|
2042
|
|
|
2043 ctx->handler(ctx);
|
|
|
2044 }
|
|
|
2045
|
|
|
2046
|
|
|
2047 static void
|
|
|
2048 ngx_resolver_free_node(ngx_resolver_t *r, ngx_resolver_node_t *rn)
|
|
|
2049 {
|
|
|
2050 /* lock alloc mutex */
|
|
|
2051
|
|
|
2052 if (rn->query) {
|
|
|
2053 ngx_resolver_free_locked(r, rn->query);
|
|
|
2054 }
|
|
|
2055
|
|
|
2056 if (rn->name) {
|
|
|
2057 ngx_resolver_free_locked(r, rn->name);
|
|
|
2058 }
|
|
|
2059
|
|
|
2060 if (rn->cnlen) {
|
|
|
2061 ngx_resolver_free_locked(r, rn->u.cname);
|
|
|
2062 }
|
|
|
2063
|
|
|
2064 if (rn->naddrs > 1) {
|
|
|
2065 ngx_resolver_free_locked(r, rn->u.addrs);
|
|
|
2066 }
|
|
|
2067
|
|
|
2068 ngx_resolver_free_locked(r, rn);
|
|
|
2069
|
|
|
2070 /* unlock alloc mutex */
|
|
|
2071 }
|
|
|
2072
|
|
|
2073
|
|
|
2074 static void *
|
|
|
2075 ngx_resolver_alloc(ngx_resolver_t *r, size_t size)
|
|
|
2076 {
|
|
|
2077 u_char *p;
|
|
|
2078
|
|
|
2079 /* lock alloc mutex */
|
|
|
2080
|
|
|
2081 p = ngx_alloc(size, r->log);
|
|
|
2082
|
|
|
2083 /* unlock alloc mutex */
|
|
|
2084
|
|
|
2085 return p;
|
|
|
2086 }
|
|
|
2087
|
|
|
2088
|
|
366
|
2089 static void *
|
|
350
|
2090 ngx_resolver_calloc(ngx_resolver_t *r, size_t size)
|
|
|
2091 {
|
|
|
2092 u_char *p;
|
|
|
2093
|
|
|
2094 p = ngx_resolver_alloc(r, size);
|
|
|
2095
|
|
|
2096 if (p) {
|
|
|
2097 ngx_memzero(p, size);
|
|
|
2098 }
|
|
|
2099
|
|
|
2100 return p;
|
|
|
2101 }
|
|
|
2102
|
|
|
2103
|
|
|
2104 static void
|
|
|
2105 ngx_resolver_free(ngx_resolver_t *r, void *p)
|
|
|
2106 {
|
|
|
2107 /* lock alloc mutex */
|
|
|
2108
|
|
|
2109 ngx_free(p);
|
|
|
2110
|
|
|
2111 /* unlock alloc mutex */
|
|
|
2112 }
|
|
|
2113
|
|
|
2114
|
|
|
2115 static void
|
|
|
2116 ngx_resolver_free_locked(ngx_resolver_t *r, void *p)
|
|
|
2117 {
|
|
|
2118 ngx_free(p);
|
|
|
2119 }
|
|
|
2120
|
|
|
2121
|
|
|
2122 static void *
|
|
|
2123 ngx_resolver_dup(ngx_resolver_t *r, void *src, size_t size)
|
|
|
2124 {
|
|
|
2125 void *dst;
|
|
|
2126
|
|
|
2127 dst = ngx_resolver_alloc(r, size);
|
|
|
2128
|
|
|
2129 if (dst == NULL) {
|
|
|
2130 return dst;
|
|
|
2131 }
|
|
|
2132
|
|
|
2133 ngx_memcpy(dst, src, size);
|
|
|
2134
|
|
|
2135 return dst;
|
|
|
2136 }
|
|
|
2137
|
|
|
2138
|
|
688
|
2139 static in_addr_t *
|
|
|
2140 ngx_resolver_rotate(ngx_resolver_t *r, in_addr_t *src, ngx_uint_t n)
|
|
|
2141 {
|
|
|
2142 void *dst, *p;
|
|
|
2143 ngx_uint_t j;
|
|
|
2144
|
|
|
2145 dst = ngx_resolver_alloc(r, n * sizeof(in_addr_t));
|
|
|
2146
|
|
690
|
2147 if (dst == NULL) {
|
|
|
2148 return dst;
|
|
|
2149 }
|
|
|
2150
|
|
688
|
2151 j = ngx_random() % n;
|
|
|
2152
|
|
|
2153 if (j == 0) {
|
|
|
2154 ngx_memcpy(dst, src, n * sizeof(in_addr_t));
|
|
|
2155 return dst;
|
|
|
2156 }
|
|
|
2157
|
|
|
2158 p = ngx_cpymem(dst, &src[j], (n - j) * sizeof(in_addr_t));
|
|
|
2159 ngx_memcpy(p, src, j * sizeof(in_addr_t));
|
|
|
2160
|
|
|
2161 return dst;
|
|
|
2162 }
|
|
|
2163
|
|
|
2164
|
|
350
|
2165 char *
|
|
|
2166 ngx_resolver_strerror(ngx_int_t err)
|
|
|
2167 {
|
|
|
2168 static char *errors[] = {
|
|
|
2169 "Format error", /* FORMERR */
|
|
|
2170 "Server failure", /* SERVFAIL */
|
|
|
2171 "Host not found", /* NXDOMAIN */
|
|
|
2172 "Unimplemented", /* NOTIMP */
|
|
|
2173 "Operation refused" /* REFUSED */
|
|
|
2174 };
|
|
|
2175
|
|
|
2176 if (err > 0 && err < 6) {
|
|
|
2177 return errors[err - 1];
|
|
|
2178 }
|
|
|
2179
|
|
|
2180 if (err == NGX_RESOLVE_TIMEDOUT) {
|
|
|
2181 return "Operation timed out";
|
|
|
2182 }
|
|
|
2183
|
|
|
2184 return "Unknown error";
|
|
|
2185 }
|
|
|
2186
|
|
|
2187
|
|
560
|
2188 static u_char *
|
|
|
2189 ngx_resolver_log_error(ngx_log_t *log, u_char *buf, size_t len)
|
|
|
2190 {
|
|
|
2191 u_char *p;
|
|
|
2192 ngx_udp_connection_t *uc;
|
|
|
2193
|
|
|
2194 p = buf;
|
|
|
2195
|
|
|
2196 if (log->action) {
|
|
|
2197 p = ngx_snprintf(buf, len, " while %s", log->action);
|
|
|
2198 len -= p - buf;
|
|
|
2199 }
|
|
|
2200
|
|
|
2201 uc = log->data;
|
|
|
2202
|
|
|
2203 if (uc) {
|
|
|
2204 p = ngx_snprintf(p, len, ", resolver: %V", &uc->server);
|
|
|
2205 }
|
|
|
2206
|
|
|
2207 return p;
|
|
|
2208 }
|
|
|
2209
|
|
|
2210
|
|
350
|
2211 ngx_int_t
|
|
|
2212 ngx_udp_connect(ngx_udp_connection_t *uc)
|
|
|
2213 {
|
|
|
2214 int rc;
|
|
|
2215 ngx_int_t event;
|
|
|
2216 ngx_event_t *rev, *wev;
|
|
|
2217 ngx_socket_t s;
|
|
|
2218 ngx_connection_t *c;
|
|
|
2219
|
|
676
|
2220 s = ngx_socket(uc->sockaddr->sa_family, SOCK_DGRAM, 0);
|
|
350
|
2221
|
|
560
|
2222 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, &uc->log, 0, "UDP socket %d", s);
|
|
350
|
2223
|
|
|
2224 if (s == -1) {
|
|
560
|
2225 ngx_log_error(NGX_LOG_ALERT, &uc->log, ngx_socket_errno,
|
|
350
|
2226 ngx_socket_n " failed");
|
|
|
2227 return NGX_ERROR;
|
|
|
2228 }
|
|
|
2229
|
|
560
|
2230 c = ngx_get_connection(s, &uc->log);
|
|
350
|
2231
|
|
|
2232 if (c == NULL) {
|
|
|
2233 if (ngx_close_socket(s) == -1) {
|
|
560
|
2234 ngx_log_error(NGX_LOG_ALERT, &uc->log, ngx_socket_errno,
|
|
350
|
2235 ngx_close_socket_n "failed");
|
|
|
2236 }
|
|
|
2237
|
|
|
2238 return NGX_ERROR;
|
|
|
2239 }
|
|
|
2240
|
|
|
2241 if (ngx_nonblocking(s) == -1) {
|
|
560
|
2242 ngx_log_error(NGX_LOG_ALERT, &uc->log, ngx_socket_errno,
|
|
350
|
2243 ngx_nonblocking_n " failed");
|
|
|
2244
|
|
|
2245 ngx_free_connection(c);
|
|
|
2246
|
|
|
2247 if (ngx_close_socket(s) == -1) {
|
|
560
|
2248 ngx_log_error(NGX_LOG_ALERT, &uc->log, ngx_socket_errno,
|
|
350
|
2249 ngx_close_socket_n " failed");
|
|
|
2250 }
|
|
|
2251
|
|
|
2252 return NGX_ERROR;
|
|
|
2253 }
|
|
|
2254
|
|
|
2255 rev = c->read;
|
|
|
2256 wev = c->write;
|
|
|
2257
|
|
560
|
2258 rev->log = &uc->log;
|
|
|
2259 wev->log = &uc->log;
|
|
350
|
2260
|
|
|
2261 uc->connection = c;
|
|
|
2262
|
|
|
2263 c->number = ngx_atomic_fetch_add(ngx_connection_counter, 1);
|
|
|
2264
|
|
|
2265 #if (NGX_THREADS)
|
|
|
2266
|
|
|
2267 /* TODO: lock event when call completion handler */
|
|
|
2268
|
|
|
2269 rev->lock = &c->lock;
|
|
|
2270 wev->lock = &c->lock;
|
|
|
2271 rev->own_lock = &c->lock;
|
|
|
2272 wev->own_lock = &c->lock;
|
|
|
2273
|
|
|
2274 #endif
|
|
|
2275
|
|
560
|
2276 ngx_log_debug3(NGX_LOG_DEBUG_EVENT, &uc->log, 0,
|
|
350
|
2277 "connect to %V, fd:%d #%d", &uc->server, s, c->number);
|
|
|
2278
|
|
|
2279 rc = connect(s, uc->sockaddr, uc->socklen);
|
|
|
2280
|
|
|
2281 /* TODO: aio, iocp */
|
|
|
2282
|
|
|
2283 if (rc == -1) {
|
|
560
|
2284 ngx_log_error(NGX_LOG_CRIT, &uc->log, ngx_socket_errno,
|
|
|
2285 "connect() failed");
|
|
350
|
2286
|
|
|
2287 return NGX_ERROR;
|
|
|
2288 }
|
|
|
2289
|
|
|
2290 /* UDP sockets are always ready to write */
|
|
|
2291 wev->ready = 1;
|
|
|
2292
|
|
|
2293 if (ngx_add_event) {
|
|
|
2294
|
|
|
2295 event = (ngx_event_flags & NGX_USE_CLEAR_EVENT) ?
|
|
|
2296 /* kqueue, epoll */ NGX_CLEAR_EVENT:
|
|
|
2297 /* select, poll, /dev/poll */ NGX_LEVEL_EVENT;
|
|
|
2298 /* eventport event type has no meaning: oneshot only */
|
|
|
2299
|
|
|
2300 if (ngx_add_event(rev, NGX_READ_EVENT, event) != NGX_OK) {
|
|
|
2301 return NGX_ERROR;
|
|
|
2302 }
|
|
|
2303
|
|
|
2304 } else {
|
|
|
2305 /* rtsig */
|
|
|
2306
|
|
|
2307 if (ngx_add_conn(c) == NGX_ERROR) {
|
|
|
2308 return NGX_ERROR;
|
|
|
2309 }
|
|
|
2310 }
|
|
|
2311
|
|
|
2312 return NGX_OK;
|
|
|
2313 }
|
