|
0
|
1
|
|
|
2 /*
|
|
|
3 * Copyright (C) Igor Sysoev
|
|
|
4 */
|
|
|
5
|
|
|
6
|
|
|
7 #include <ngx_config.h>
|
|
|
8 #include <ngx_core.h>
|
|
28
|
9 #include <ngx_event.h>
|
|
|
10 #include <ngx_event_connect.h>
|
|
0
|
11
|
|
|
12
|
|
10
|
13 /*
|
|
|
14 * ngx_sock_ntop() and ngx_inet_ntop() may be implemented as
|
|
130
|
15 * "ngx_sprintf(text, "%ud.%ud.%ud.%ud", p[0], p[1], p[2], p[3])", however,
|
|
|
16 * they had been implemented long before the ngx_sprintf() had appeared
|
|
10
|
17 * and they are faster by 1.5-2.5 times, so it is worth to keep them.
|
|
|
18 *
|
|
|
19 * By the way, the implementation using ngx_sprintf() is faster by 2.5-3 times
|
|
28
|
20 * than using FreeBSD libc's snprintf().
|
|
10
|
21 */
|
|
|
22
|
|
|
23
|
|
180
|
24 static ngx_inline size_t
|
|
|
25 ngx_sprint_uchar(u_char *text, u_char c, size_t len)
|
|
0
|
26 {
|
|
|
27 size_t n;
|
|
|
28 ngx_uint_t c1, c2;
|
|
|
29
|
|
|
30 n = 0;
|
|
|
31
|
|
|
32 if (len == n) {
|
|
|
33 return n;
|
|
|
34 }
|
|
|
35
|
|
|
36 c1 = c / 100;
|
|
|
37
|
|
|
38 if (c1) {
|
|
|
39 *text++ = (u_char) (c1 + '0');
|
|
|
40 n++;
|
|
|
41
|
|
|
42 if (len == n) {
|
|
|
43 return n;
|
|
|
44 }
|
|
|
45 }
|
|
|
46
|
|
|
47 c2 = (c % 100) / 10;
|
|
|
48
|
|
|
49 if (c1 || c2) {
|
|
|
50 *text++ = (u_char) (c2 + '0');
|
|
|
51 n++;
|
|
|
52
|
|
|
53 if (len == n) {
|
|
|
54 return n;
|
|
|
55 }
|
|
|
56 }
|
|
|
57
|
|
|
58 c2 = c % 10;
|
|
|
59
|
|
|
60 *text++ = (u_char) (c2 + '0');
|
|
|
61 n++;
|
|
|
62
|
|
|
63 return n;
|
|
|
64 }
|
|
|
65
|
|
|
66
|
|
|
67 /* AF_INET only */
|
|
|
68
|
|
58
|
69 size_t
|
|
|
70 ngx_sock_ntop(int family, struct sockaddr *sa, u_char *text, size_t len)
|
|
0
|
71 {
|
|
|
72 u_char *p;
|
|
|
73 size_t n;
|
|
|
74 ngx_uint_t i;
|
|
28
|
75 struct sockaddr_in *sin;
|
|
0
|
76
|
|
|
77 if (len == 0) {
|
|
|
78 return 0;
|
|
|
79 }
|
|
|
80
|
|
|
81 if (family != AF_INET) {
|
|
|
82 return 0;
|
|
|
83 }
|
|
|
84
|
|
28
|
85 sin = (struct sockaddr_in *) sa;
|
|
|
86 p = (u_char *) &sin->sin_addr;
|
|
0
|
87
|
|
|
88 if (len > INET_ADDRSTRLEN) {
|
|
|
89 len = INET_ADDRSTRLEN;
|
|
|
90 }
|
|
|
91
|
|
|
92 n = ngx_sprint_uchar(text, p[0], len);
|
|
|
93
|
|
|
94 i = 1;
|
|
|
95
|
|
|
96 do {
|
|
|
97 if (len == n) {
|
|
|
98 text[n - 1] = '\0';
|
|
|
99 return n;
|
|
|
100 }
|
|
|
101
|
|
|
102 text[n++] = '.';
|
|
|
103
|
|
|
104 if (len == n) {
|
|
|
105 text[n - 1] = '\0';
|
|
|
106 return n;
|
|
|
107 }
|
|
|
108
|
|
|
109 n += ngx_sprint_uchar(&text[n], p[i++], len - n);
|
|
|
110
|
|
|
111 } while (i < 4);
|
|
|
112
|
|
|
113 if (len == n) {
|
|
|
114 text[n] = '\0';
|
|
|
115 return n;
|
|
|
116 }
|
|
|
117
|
|
|
118 text[n] = '\0';
|
|
|
119
|
|
|
120 return n;
|
|
|
121 }
|
|
|
122
|
|
58
|
123 size_t
|
|
|
124 ngx_inet_ntop(int family, void *addr, u_char *text, size_t len)
|
|
0
|
125 {
|
|
|
126 u_char *p;
|
|
|
127 size_t n;
|
|
|
128 ngx_uint_t i;
|
|
|
129
|
|
|
130 if (len == 0) {
|
|
|
131 return 0;
|
|
|
132 }
|
|
|
133
|
|
|
134 if (family != AF_INET) {
|
|
|
135 return 0;
|
|
|
136 }
|
|
|
137
|
|
|
138 p = (u_char *) addr;
|
|
|
139
|
|
|
140 if (len > INET_ADDRSTRLEN) {
|
|
|
141 len = INET_ADDRSTRLEN;
|
|
|
142 }
|
|
|
143
|
|
|
144 n = ngx_sprint_uchar(text, p[0], len);
|
|
|
145
|
|
|
146 i = 1;
|
|
|
147
|
|
|
148 do {
|
|
|
149 if (len == n) {
|
|
|
150 text[n - 1] = '\0';
|
|
|
151 return n;
|
|
|
152 }
|
|
|
153
|
|
|
154 text[n++] = '.';
|
|
|
155
|
|
|
156 if (len == n) {
|
|
|
157 text[n - 1] = '\0';
|
|
|
158 return n;
|
|
|
159 }
|
|
|
160
|
|
|
161 n += ngx_sprint_uchar(&text[n], p[i++], len - n);
|
|
|
162
|
|
|
163 } while (i < 4);
|
|
|
164
|
|
|
165 if (len == n) {
|
|
|
166 text[n] = '\0';
|
|
|
167 return n;
|
|
|
168 }
|
|
|
169
|
|
|
170 text[n] = '\0';
|
|
|
171
|
|
|
172 return n;
|
|
|
173 }
|
|
|
174
|
|
|
175
|
|
|
176 /* AF_INET only */
|
|
|
177
|
|
58
|
178 ngx_int_t
|
|
|
179 ngx_ptocidr(ngx_str_t *text, void *cidr)
|
|
0
|
180 {
|
|
|
181 ngx_int_t m;
|
|
|
182 ngx_uint_t i;
|
|
|
183 ngx_inet_cidr_t *in_cidr;
|
|
|
184
|
|
|
185 in_cidr = cidr;
|
|
|
186
|
|
|
187 for (i = 0; i < text->len; i++) {
|
|
|
188 if (text->data[i] == '/') {
|
|
|
189 break;
|
|
|
190 }
|
|
|
191 }
|
|
|
192
|
|
|
193 if (i == text->len) {
|
|
|
194 return NGX_ERROR;
|
|
|
195 }
|
|
|
196
|
|
|
197 text->data[i] = '\0';
|
|
|
198 in_cidr->addr = inet_addr((char *) text->data);
|
|
|
199 text->data[i] = '/';
|
|
|
200 if (in_cidr->addr == INADDR_NONE) {
|
|
|
201 return NGX_ERROR;
|
|
|
202 }
|
|
|
203
|
|
|
204 m = ngx_atoi(&text->data[i + 1], text->len - (i + 1));
|
|
|
205 if (m == NGX_ERROR) {
|
|
|
206 return NGX_ERROR;
|
|
|
207 }
|
|
|
208
|
|
|
209 if (m == 0) {
|
|
|
210
|
|
|
211 /* the x86 compilers use the shl instruction that shifts by modulo 32 */
|
|
|
212
|
|
|
213 in_cidr->mask = 0;
|
|
|
214 return NGX_OK;
|
|
|
215 }
|
|
|
216
|
|
|
217 in_cidr->mask = htonl((ngx_uint_t) (0 - (1 << (32 - m))));
|
|
|
218
|
|
|
219 return NGX_OK;
|
|
|
220 }
|
|
|
221
|
|
|
222
|
|
58
|
223 ngx_peers_t *
|
|
|
224 ngx_inet_upstream_parse(ngx_conf_t *cf, ngx_inet_upstream_t *u)
|
|
0
|
225 {
|
|
28
|
226 char *err;
|
|
|
227 u_char *host;
|
|
66
|
228 size_t len;
|
|
28
|
229 in_addr_t in_addr;
|
|
66
|
230 ngx_uint_t i;
|
|
28
|
231 ngx_peers_t *peers;
|
|
|
232 struct hostent *h;
|
|
|
233 struct sockaddr_in *sin;
|
|
|
234
|
|
|
235 err = ngx_inet_parse_host_port(u);
|
|
|
236
|
|
|
237 if (err) {
|
|
|
238 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
|
239 "%s in upstream \"%V\"", err, &u->name);
|
|
|
240 return NULL;
|
|
|
241 }
|
|
|
242
|
|
|
243 if (u->default_port) {
|
|
|
244 if (u->default_port_value == 0) {
|
|
|
245 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
|
246 "no port in upstream \"%V\"", &u->name);
|
|
|
247 return NULL;
|
|
|
248 }
|
|
|
249
|
|
|
250 u->port = u->default_port_value;
|
|
|
251
|
|
50
|
252 u->port_text.data = ngx_palloc(cf->pool, sizeof("65536") - 1);
|
|
|
253 if (u->port_text.data == NULL) {
|
|
28
|
254 return NULL;
|
|
|
255 }
|
|
|
256
|
|
|
257 u->port_text.len = ngx_sprintf(u->port_text.data, "%d",
|
|
|
258 u->default_port_value)
|
|
|
259 - u->port_text.data;
|
|
|
260
|
|
|
261 } else if (u->port) {
|
|
|
262 if (u->port == u->default_port_value) {
|
|
|
263 u->default_port = 1;
|
|
|
264 }
|
|
|
265
|
|
|
266 } else {
|
|
|
267 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
|
268 "no port in upstream \"%V\"", &u->name);
|
|
|
269 return NULL;
|
|
|
270 }
|
|
|
271
|
|
|
272 if (u->host.len == 0) {
|
|
|
273 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
|
274 "no host in upstream \"%V\"", &u->name);
|
|
|
275 return NULL;
|
|
|
276 }
|
|
|
277
|
|
|
278 u->port = htons(u->port);
|
|
|
279
|
|
50
|
280 host = ngx_palloc(cf->pool, u->host.len + 1);
|
|
|
281 if (host == NULL) {
|
|
28
|
282 return NULL;
|
|
|
283 }
|
|
|
284
|
|
66
|
285 (void) ngx_cpystrn(host, u->host.data, u->host.len + 1);
|
|
28
|
286
|
|
|
287 /* AF_INET only */
|
|
|
288
|
|
|
289 in_addr = inet_addr((char *) host);
|
|
|
290
|
|
|
291 if (in_addr == INADDR_NONE) {
|
|
|
292 h = gethostbyname((char *) host);
|
|
|
293
|
|
|
294 if (h == NULL || h->h_addr_list[0] == NULL) {
|
|
|
295 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
|
296 "host %s is not found in upstream \"%V\"",
|
|
|
297 host, &u->name);
|
|
|
298 return NULL;
|
|
|
299 }
|
|
|
300
|
|
|
301 for (i = 0; h->h_addr_list[i] != NULL; i++) { /* void */ }
|
|
|
302
|
|
|
303 /* MP: ngx_shared_palloc() */
|
|
|
304
|
|
|
305 peers = ngx_pcalloc(cf->pool,
|
|
|
306 sizeof(ngx_peers_t) + sizeof(ngx_peer_t) * (i - 1));
|
|
|
307 if (peers == NULL) {
|
|
|
308 return NULL;
|
|
|
309 }
|
|
|
310
|
|
|
311 peers->number = i;
|
|
|
312 peers->weight = 1;
|
|
|
313
|
|
|
314 for (i = 0; h->h_addr_list[i] != NULL; i++) {
|
|
|
315
|
|
50
|
316 sin = ngx_pcalloc(cf->pool, sizeof(struct sockaddr_in));
|
|
|
317 if (sin == NULL) {
|
|
28
|
318 return NULL;
|
|
|
319 }
|
|
|
320
|
|
|
321 sin->sin_family = AF_INET;
|
|
|
322 sin->sin_port = u->port;
|
|
36
|
323 sin->sin_addr.s_addr = *(in_addr_t *) (h->h_addr_list[i]);
|
|
28
|
324
|
|
|
325 peers->peer[i].sockaddr = (struct sockaddr *) sin;
|
|
|
326 peers->peer[i].socklen = sizeof(struct sockaddr_in);
|
|
|
327
|
|
|
328 len = INET_ADDRSTRLEN - 1 + 1 + u->port_text.len;
|
|
126
|
329
|
|
50
|
330 peers->peer[i].name.data = ngx_palloc(cf->pool, len);
|
|
|
331 if (peers->peer[i].name.data == NULL) {
|
|
28
|
332 return NULL;
|
|
|
333 }
|
|
|
334
|
|
|
335 len = ngx_sock_ntop(AF_INET, (struct sockaddr *) sin,
|
|
|
336 peers->peer[i].name.data, len);
|
|
|
337
|
|
|
338 peers->peer[i].name.data[len++] = ':';
|
|
0
|
339
|
|
28
|
340 ngx_memcpy(peers->peer[i].name.data + len,
|
|
|
341 u->port_text.data, u->port_text.len);
|
|
|
342
|
|
|
343 peers->peer[i].name.len = len + u->port_text.len;
|
|
|
344
|
|
|
345 peers->peer[i].uri_separator = "";
|
|
|
346 }
|
|
|
347
|
|
|
348 } else {
|
|
|
349
|
|
|
350 /* MP: ngx_shared_palloc() */
|
|
|
351
|
|
50
|
352 peers = ngx_pcalloc(cf->pool, sizeof(ngx_peers_t));
|
|
|
353 if (peers == NULL) {
|
|
28
|
354 return NULL;
|
|
|
355 }
|
|
|
356
|
|
50
|
357 sin = ngx_pcalloc(cf->pool, sizeof(struct sockaddr_in));
|
|
|
358 if (sin == NULL) {
|
|
28
|
359 return NULL;
|
|
|
360 }
|
|
|
361
|
|
|
362 peers->number = 1;
|
|
|
363
|
|
|
364 sin->sin_family = AF_INET;
|
|
|
365 sin->sin_port = u->port;
|
|
|
366 sin->sin_addr.s_addr = in_addr;
|
|
|
367
|
|
|
368 peers->peer[0].sockaddr = (struct sockaddr *) sin;
|
|
|
369 peers->peer[0].socklen = sizeof(struct sockaddr_in);
|
|
|
370
|
|
|
371 len = u->host.len + 1 + u->port_text.len;
|
|
|
372
|
|
|
373 peers->peer[0].name.len = len;
|
|
|
374
|
|
50
|
375 peers->peer[0].name.data = ngx_palloc(cf->pool, len);
|
|
|
376 if (peers->peer[0].name.data == NULL) {
|
|
28
|
377 return NULL;
|
|
|
378 }
|
|
|
379
|
|
|
380 len = u->host.len;
|
|
|
381
|
|
|
382 ngx_memcpy(peers->peer[0].name.data, u->host.data, len);
|
|
|
383
|
|
|
384 peers->peer[0].name.data[len++] = ':';
|
|
|
385
|
|
|
386 ngx_memcpy(peers->peer[0].name.data + len,
|
|
|
387 u->port_text.data, u->port_text.len);
|
|
|
388
|
|
|
389 peers->peer[0].uri_separator = "";
|
|
|
390 }
|
|
|
391
|
|
|
392 return peers;
|
|
|
393 }
|
|
|
394
|
|
|
395
|
|
58
|
396 char *
|
|
|
397 ngx_inet_parse_host_port(ngx_inet_upstream_t *u)
|
|
28
|
398 {
|
|
|
399 size_t i;
|
|
|
400 ngx_int_t port;
|
|
|
401 ngx_str_t *url;
|
|
|
402
|
|
|
403 url = &u->url;
|
|
|
404
|
|
|
405 if (u->port_only) {
|
|
|
406 i = 0;
|
|
|
407
|
|
|
408 } else {
|
|
|
409 if (url->data[0] == ':' || url->data[0] == '/') {
|
|
|
410 return "invalid host";
|
|
|
411 }
|
|
|
412
|
|
|
413 i = 1;
|
|
|
414 }
|
|
|
415
|
|
|
416 u->host.data = url->data;
|
|
|
417 u->host_header = *url;
|
|
|
418
|
|
64
|
419 for ( /* void */ ; i < url->len; i++) {
|
|
28
|
420
|
|
|
421 if (url->data[i] == ':') {
|
|
|
422 u->port_text.data = &url->data[i] + 1;
|
|
|
423 u->host.len = i;
|
|
|
424
|
|
|
425 if (!u->uri_part) {
|
|
|
426 u->port_text.len = &url->data[url->len] - u->port_text.data;
|
|
|
427 break;
|
|
|
428 }
|
|
|
429 }
|
|
|
430
|
|
|
431 if (url->data[i] == '/') {
|
|
|
432 u->uri.data = &url->data[i];
|
|
|
433 u->uri.len = url->len - i;
|
|
|
434 u->host_header.len = i;
|
|
|
435
|
|
|
436 if (u->host.len == 0) {
|
|
|
437 u->host.len = i;
|
|
|
438 }
|
|
|
439
|
|
|
440 if (u->port_text.data == NULL) {
|
|
|
441 u->default_port = 1;
|
|
|
442 return NULL;
|
|
|
443 }
|
|
|
444
|
|
|
445 u->port_text.len = &url->data[i] - u->port_text.data;
|
|
|
446
|
|
|
447 if (u->port_text.len == 0) {
|
|
|
448 return "invalid port";
|
|
|
449 }
|
|
|
450
|
|
0
|
451 break;
|
|
|
452 }
|
|
|
453 }
|
|
|
454
|
|
118
|
455 if (u->port_text.data) {
|
|
|
456
|
|
|
457 if (u->port_text.len == 0) {
|
|
|
458 u->port_text.len = &url->data[i] - u->port_text.data;
|
|
|
459
|
|
|
460 if (u->port_text.len == 0) {
|
|
|
461 return "invalid port";
|
|
|
462 }
|
|
|
463 }
|
|
|
464
|
|
|
465 port = ngx_atoi(u->port_text.data, u->port_text.len);
|
|
|
466
|
|
|
467 if (port == NGX_ERROR || port < 1 || port > 65536) {
|
|
|
468 return "invalid port";
|
|
|
469 }
|
|
|
470
|
|
|
471 } else {
|
|
28
|
472 port = ngx_atoi(url->data, url->len);
|
|
|
473
|
|
|
474 if (port == NGX_ERROR) {
|
|
|
475 u->default_port = 1;
|
|
|
476 u->host.len = url->len;
|
|
|
477
|
|
|
478 return NULL;
|
|
|
479 }
|
|
0
|
480
|
|
28
|
481 u->port_text = *url;
|
|
|
482 u->wildcard = 1;
|
|
0
|
483 }
|
|
|
484
|
|
28
|
485 u->port = (in_port_t) port;
|
|
0
|
486
|
|
28
|
487 return NULL;
|
|
0
|
488 }
|
