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
|
|
15 * "ngx_sprintf(text, "%ud.%ud.%ud.%ud", p[0], p[1], p[2], p[3])",
|
|
16 * however, they were implemented long before the ngx_sprintf() appeared
|
|
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
|
58
|
24 static
|
|
25 ngx_inline size_t 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;
|
|
228 in_addr_t in_addr;
|
|
229 ngx_uint_t i, len;
|
|
230 ngx_peers_t *peers;
|
|
231 struct hostent *h;
|
|
232 struct sockaddr_in *sin;
|
|
233
|
|
234 err = ngx_inet_parse_host_port(u);
|
|
235
|
|
236 if (err) {
|
|
237 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
238 "%s in upstream \"%V\"", err, &u->name);
|
|
239 return NULL;
|
|
240 }
|
|
241
|
|
242 if (u->default_port) {
|
|
243 if (u->default_port_value == 0) {
|
|
244 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
245 "no port in upstream \"%V\"", &u->name);
|
|
246 return NULL;
|
|
247 }
|
|
248
|
|
249 u->port = u->default_port_value;
|
|
250
|
50
|
251 u->port_text.data = ngx_palloc(cf->pool, sizeof("65536") - 1);
|
|
252 if (u->port_text.data == NULL) {
|
28
|
253 return NULL;
|
|
254 }
|
|
255
|
|
256 u->port_text.len = ngx_sprintf(u->port_text.data, "%d",
|
|
257 u->default_port_value)
|
|
258 - u->port_text.data;
|
|
259
|
|
260 } else if (u->port) {
|
|
261 if (u->port == u->default_port_value) {
|
|
262 u->default_port = 1;
|
|
263 }
|
|
264
|
|
265 } else {
|
|
266 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
267 "no port in upstream \"%V\"", &u->name);
|
|
268 return NULL;
|
|
269 }
|
|
270
|
|
271 if (u->host.len == 0) {
|
|
272 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
273 "no host in upstream \"%V\"", &u->name);
|
|
274 return NULL;
|
|
275 }
|
|
276
|
|
277 u->port = htons(u->port);
|
|
278
|
50
|
279 host = ngx_palloc(cf->pool, u->host.len + 1);
|
|
280 if (host == NULL) {
|
28
|
281 return NULL;
|
|
282 }
|
|
283
|
|
284 ngx_cpystrn(host, u->host.data, u->host.len + 1);
|
|
285
|
|
286 /* AF_INET only */
|
|
287
|
|
288 in_addr = inet_addr((char *) host);
|
|
289
|
|
290 if (in_addr == INADDR_NONE) {
|
|
291 h = gethostbyname((char *) host);
|
|
292
|
|
293 if (h == NULL || h->h_addr_list[0] == NULL) {
|
|
294 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
295 "host %s is not found in upstream \"%V\"",
|
|
296 host, &u->name);
|
|
297 return NULL;
|
|
298 }
|
|
299
|
|
300 for (i = 0; h->h_addr_list[i] != NULL; i++) { /* void */ }
|
|
301
|
|
302 /* MP: ngx_shared_palloc() */
|
|
303
|
|
304 peers = ngx_pcalloc(cf->pool,
|
|
305 sizeof(ngx_peers_t) + sizeof(ngx_peer_t) * (i - 1));
|
|
306 if (peers == NULL) {
|
|
307 return NULL;
|
|
308 }
|
|
309
|
|
310 peers->number = i;
|
|
311 peers->weight = 1;
|
|
312
|
|
313 for (i = 0; h->h_addr_list[i] != NULL; i++) {
|
|
314
|
50
|
315 sin = ngx_pcalloc(cf->pool, sizeof(struct sockaddr_in));
|
|
316 if (sin == NULL) {
|
28
|
317 return NULL;
|
|
318 }
|
|
319
|
|
320 sin->sin_family = AF_INET;
|
|
321 sin->sin_port = u->port;
|
36
|
322 sin->sin_addr.s_addr = *(in_addr_t *) (h->h_addr_list[i]);
|
28
|
323
|
|
324 peers->peer[i].sockaddr = (struct sockaddr *) sin;
|
|
325 peers->peer[i].socklen = sizeof(struct sockaddr_in);
|
|
326
|
|
327 len = INET_ADDRSTRLEN - 1 + 1 + u->port_text.len;
|
|
328
|
50
|
329 peers->peer[i].name.data = ngx_palloc(cf->pool, len);
|
|
330 if (peers->peer[i].name.data == NULL) {
|
28
|
331 return NULL;
|
|
332 }
|
|
333
|
|
334 len = ngx_sock_ntop(AF_INET, (struct sockaddr *) sin,
|
|
335 peers->peer[i].name.data, len);
|
|
336
|
|
337 peers->peer[i].name.data[len++] = ':';
|
0
|
338
|
28
|
339 ngx_memcpy(peers->peer[i].name.data + len,
|
|
340 u->port_text.data, u->port_text.len);
|
|
341
|
|
342 peers->peer[i].name.len = len + u->port_text.len;
|
|
343
|
|
344 peers->peer[i].uri_separator = "";
|
|
345
|
|
346 peers->peer[i].weight = 1;
|
|
347 peers->peer[i].max_fails = 1;
|
|
348 peers->peer[i].fail_timeout = 60;
|
|
349 }
|
|
350
|
|
351 } else {
|
|
352
|
|
353 /* MP: ngx_shared_palloc() */
|
|
354
|
50
|
355 peers = ngx_pcalloc(cf->pool, sizeof(ngx_peers_t));
|
|
356 if (peers == NULL) {
|
28
|
357 return NULL;
|
|
358 }
|
|
359
|
50
|
360 sin = ngx_pcalloc(cf->pool, sizeof(struct sockaddr_in));
|
|
361 if (sin == NULL) {
|
28
|
362 return NULL;
|
|
363 }
|
|
364
|
|
365 peers->number = 1;
|
|
366
|
|
367 sin->sin_family = AF_INET;
|
|
368 sin->sin_port = u->port;
|
|
369 sin->sin_addr.s_addr = in_addr;
|
|
370
|
|
371 peers->peer[0].sockaddr = (struct sockaddr *) sin;
|
|
372 peers->peer[0].socklen = sizeof(struct sockaddr_in);
|
|
373
|
|
374 len = u->host.len + 1 + u->port_text.len;
|
|
375
|
|
376 peers->peer[0].name.len = len;
|
|
377
|
50
|
378 peers->peer[0].name.data = ngx_palloc(cf->pool, len);
|
|
379 if (peers->peer[0].name.data == NULL) {
|
28
|
380 return NULL;
|
|
381 }
|
|
382
|
|
383 len = u->host.len;
|
|
384
|
|
385 ngx_memcpy(peers->peer[0].name.data, u->host.data, len);
|
|
386
|
|
387 peers->peer[0].name.data[len++] = ':';
|
|
388
|
|
389 ngx_memcpy(peers->peer[0].name.data + len,
|
|
390 u->port_text.data, u->port_text.len);
|
|
391
|
|
392 peers->peer[0].uri_separator = "";
|
|
393 }
|
|
394
|
|
395 return peers;
|
|
396 }
|
|
397
|
|
398
|
58
|
399 char *
|
|
400 ngx_inet_parse_host_port(ngx_inet_upstream_t *u)
|
28
|
401 {
|
|
402 size_t i;
|
|
403 ngx_int_t port;
|
|
404 ngx_str_t *url;
|
|
405
|
|
406 url = &u->url;
|
|
407
|
|
408 if (u->port_only) {
|
|
409 i = 0;
|
|
410
|
|
411 } else {
|
|
412 if (url->data[0] == ':' || url->data[0] == '/') {
|
|
413 return "invalid host";
|
|
414 }
|
|
415
|
|
416 i = 1;
|
|
417 }
|
|
418
|
|
419 u->host.data = url->data;
|
|
420 u->host_header = *url;
|
|
421
|
64
|
422 for ( /* void */ ; i < url->len; i++) {
|
28
|
423
|
|
424 if (url->data[i] == ':') {
|
|
425 u->port_text.data = &url->data[i] + 1;
|
|
426 u->host.len = i;
|
|
427
|
|
428 if (!u->uri_part) {
|
|
429 u->port_text.len = &url->data[url->len] - u->port_text.data;
|
|
430 break;
|
|
431 }
|
|
432 }
|
|
433
|
|
434 if (url->data[i] == '/') {
|
|
435 u->uri.data = &url->data[i];
|
|
436 u->uri.len = url->len - i;
|
|
437 u->host_header.len = i;
|
|
438
|
|
439 if (u->host.len == 0) {
|
|
440 u->host.len = i;
|
|
441 }
|
|
442
|
|
443 if (u->port_text.data == NULL) {
|
|
444 u->default_port = 1;
|
|
445 return NULL;
|
|
446 }
|
|
447
|
|
448 u->port_text.len = &url->data[i] - u->port_text.data;
|
|
449
|
|
450 if (u->port_text.len == 0) {
|
|
451 return "invalid port";
|
|
452 }
|
|
453
|
0
|
454 break;
|
|
455 }
|
|
456 }
|
|
457
|
28
|
458 if (u->port_text.data == NULL) {
|
|
459 port = ngx_atoi(url->data, url->len);
|
|
460
|
|
461 if (port == NGX_ERROR) {
|
|
462 u->default_port = 1;
|
|
463 u->host.len = url->len;
|
|
464
|
|
465 return NULL;
|
|
466 }
|
0
|
467
|
28
|
468 u->port_text = *url;
|
|
469 u->wildcard = 1;
|
0
|
470
|
28
|
471 } else {
|
|
472 if (u->port_text.len == 0) {
|
|
473 return "no URI";
|
|
474 }
|
|
475
|
|
476 port = ngx_atoi(u->port_text.data, u->port_text.len);
|
|
477
|
|
478 if (port == NGX_ERROR || port < 1 || port > 65536) {
|
|
479 return "invalid port";
|
|
480 }
|
0
|
481 }
|
|
482
|
28
|
483 u->port = (in_port_t) port;
|
0
|
484
|
28
|
485 return NULL;
|
0
|
486 }
|