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