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;
|
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;
|
|
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 peers->peer[i].weight = 1;
|
|
348 peers->peer[i].max_fails = 1;
|
|
349 peers->peer[i].fail_timeout = 60;
|
|
350 }
|
|
351
|
|
352 } else {
|
|
353
|
|
354 /* MP: ngx_shared_palloc() */
|
|
355
|
50
|
356 peers = ngx_pcalloc(cf->pool, sizeof(ngx_peers_t));
|
|
357 if (peers == NULL) {
|
28
|
358 return NULL;
|
|
359 }
|
|
360
|
50
|
361 sin = ngx_pcalloc(cf->pool, sizeof(struct sockaddr_in));
|
|
362 if (sin == NULL) {
|
28
|
363 return NULL;
|
|
364 }
|
|
365
|
|
366 peers->number = 1;
|
|
367
|
|
368 sin->sin_family = AF_INET;
|
|
369 sin->sin_port = u->port;
|
|
370 sin->sin_addr.s_addr = in_addr;
|
|
371
|
|
372 peers->peer[0].sockaddr = (struct sockaddr *) sin;
|
|
373 peers->peer[0].socklen = sizeof(struct sockaddr_in);
|
|
374
|
|
375 len = u->host.len + 1 + u->port_text.len;
|
|
376
|
|
377 peers->peer[0].name.len = len;
|
|
378
|
50
|
379 peers->peer[0].name.data = ngx_palloc(cf->pool, len);
|
|
380 if (peers->peer[0].name.data == NULL) {
|
28
|
381 return NULL;
|
|
382 }
|
|
383
|
|
384 len = u->host.len;
|
|
385
|
|
386 ngx_memcpy(peers->peer[0].name.data, u->host.data, len);
|
|
387
|
|
388 peers->peer[0].name.data[len++] = ':';
|
|
389
|
|
390 ngx_memcpy(peers->peer[0].name.data + len,
|
|
391 u->port_text.data, u->port_text.len);
|
|
392
|
|
393 peers->peer[0].uri_separator = "";
|
|
394 }
|
|
395
|
|
396 return peers;
|
|
397 }
|
|
398
|
|
399
|
58
|
400 char *
|
|
401 ngx_inet_parse_host_port(ngx_inet_upstream_t *u)
|
28
|
402 {
|
|
403 size_t i;
|
|
404 ngx_int_t port;
|
|
405 ngx_str_t *url;
|
|
406
|
|
407 url = &u->url;
|
|
408
|
|
409 if (u->port_only) {
|
|
410 i = 0;
|
|
411
|
|
412 } else {
|
|
413 if (url->data[0] == ':' || url->data[0] == '/') {
|
|
414 return "invalid host";
|
|
415 }
|
|
416
|
|
417 i = 1;
|
|
418 }
|
|
419
|
|
420 u->host.data = url->data;
|
|
421 u->host_header = *url;
|
|
422
|
64
|
423 for ( /* void */ ; i < url->len; i++) {
|
28
|
424
|
|
425 if (url->data[i] == ':') {
|
|
426 u->port_text.data = &url->data[i] + 1;
|
|
427 u->host.len = i;
|
|
428
|
|
429 if (!u->uri_part) {
|
|
430 u->port_text.len = &url->data[url->len] - u->port_text.data;
|
|
431 break;
|
|
432 }
|
|
433 }
|
|
434
|
|
435 if (url->data[i] == '/') {
|
|
436 u->uri.data = &url->data[i];
|
|
437 u->uri.len = url->len - i;
|
|
438 u->host_header.len = i;
|
|
439
|
|
440 if (u->host.len == 0) {
|
|
441 u->host.len = i;
|
|
442 }
|
|
443
|
|
444 if (u->port_text.data == NULL) {
|
|
445 u->default_port = 1;
|
|
446 return NULL;
|
|
447 }
|
|
448
|
|
449 u->port_text.len = &url->data[i] - u->port_text.data;
|
|
450
|
|
451 if (u->port_text.len == 0) {
|
|
452 return "invalid port";
|
|
453 }
|
|
454
|
0
|
455 break;
|
|
456 }
|
|
457 }
|
|
458
|
28
|
459 if (u->port_text.data == NULL) {
|
|
460 port = ngx_atoi(url->data, url->len);
|
|
461
|
|
462 if (port == NGX_ERROR) {
|
|
463 u->default_port = 1;
|
|
464 u->host.len = url->len;
|
|
465
|
|
466 return NULL;
|
|
467 }
|
0
|
468
|
28
|
469 u->port_text = *url;
|
|
470 u->wildcard = 1;
|
0
|
471
|
28
|
472 } else {
|
|
473 if (u->port_text.len == 0) {
|
|
474 return "no URI";
|
|
475 }
|
|
476
|
|
477 port = ngx_atoi(u->port_text.data, u->port_text.len);
|
|
478
|
|
479 if (port == NGX_ERROR || port < 1 || port > 65536) {
|
|
480 return "invalid port";
|
|
481 }
|
0
|
482 }
|
|
483
|
28
|
484 u->port = (in_port_t) port;
|
0
|
485
|
28
|
486 return NULL;
|
0
|
487 }
|