Mercurial > hg > nginx
annotate src/stream/ngx_stream_core_module.c @ 6184:fa663739e115
Stream: client-side PROXY protocol.
The new directive "proxy_protocol" toggles sending out PROXY protocol header
to upstream once connection is established.
author | Roman Arutyunyan <arut@nginx.com> |
---|---|
date | Tue, 16 Jun 2015 13:45:16 +0300 |
parents | 68c106e6fa0a |
children | 7565e056fad6 |
rev | line source |
---|---|
6115 | 1 |
2 /* | |
3 * Copyright (C) Roman Arutyunyan | |
4 * Copyright (C) Nginx, Inc. | |
5 */ | |
6 | |
7 | |
8 #include <ngx_config.h> | |
9 #include <ngx_core.h> | |
10 #include <ngx_stream.h> | |
11 | |
12 | |
13 static void *ngx_stream_core_create_main_conf(ngx_conf_t *cf); | |
14 static void *ngx_stream_core_create_srv_conf(ngx_conf_t *cf); | |
15 static char *ngx_stream_core_merge_srv_conf(ngx_conf_t *cf, void *parent, | |
16 void *child); | |
17 static char *ngx_stream_core_error_log(ngx_conf_t *cf, ngx_command_t *cmd, | |
18 void *conf); | |
19 static char *ngx_stream_core_server(ngx_conf_t *cf, ngx_command_t *cmd, | |
20 void *conf); | |
21 static char *ngx_stream_core_listen(ngx_conf_t *cf, ngx_command_t *cmd, | |
22 void *conf); | |
23 | |
24 | |
25 static ngx_command_t ngx_stream_core_commands[] = { | |
26 | |
27 { ngx_string("server"), | |
28 NGX_STREAM_MAIN_CONF|NGX_CONF_BLOCK|NGX_CONF_NOARGS, | |
29 ngx_stream_core_server, | |
30 0, | |
31 0, | |
32 NULL }, | |
33 | |
34 { ngx_string("listen"), | |
35 NGX_STREAM_SRV_CONF|NGX_CONF_1MORE, | |
36 ngx_stream_core_listen, | |
37 NGX_STREAM_SRV_CONF_OFFSET, | |
38 0, | |
39 NULL }, | |
40 | |
41 { ngx_string("error_log"), | |
42 NGX_STREAM_MAIN_CONF|NGX_STREAM_SRV_CONF|NGX_CONF_1MORE, | |
43 ngx_stream_core_error_log, | |
44 NGX_STREAM_SRV_CONF_OFFSET, | |
45 0, | |
46 NULL }, | |
47 | |
48 ngx_null_command | |
49 }; | |
50 | |
51 | |
52 static ngx_stream_module_t ngx_stream_core_module_ctx = { | |
6174
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
53 NULL, /* postconfiguration */ |
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
54 |
6115 | 55 ngx_stream_core_create_main_conf, /* create main configuration */ |
56 NULL, /* init main configuration */ | |
57 | |
58 ngx_stream_core_create_srv_conf, /* create server configuration */ | |
59 ngx_stream_core_merge_srv_conf /* merge server configuration */ | |
60 }; | |
61 | |
62 | |
63 ngx_module_t ngx_stream_core_module = { | |
64 NGX_MODULE_V1, | |
65 &ngx_stream_core_module_ctx, /* module context */ | |
66 ngx_stream_core_commands, /* module directives */ | |
67 NGX_STREAM_MODULE, /* module type */ | |
68 NULL, /* init master */ | |
69 NULL, /* init module */ | |
70 NULL, /* init process */ | |
71 NULL, /* init thread */ | |
72 NULL, /* exit thread */ | |
73 NULL, /* exit process */ | |
74 NULL, /* exit master */ | |
75 NGX_MODULE_V1_PADDING | |
76 }; | |
77 | |
78 | |
79 static void * | |
80 ngx_stream_core_create_main_conf(ngx_conf_t *cf) | |
81 { | |
82 ngx_stream_core_main_conf_t *cmcf; | |
83 | |
84 cmcf = ngx_pcalloc(cf->pool, sizeof(ngx_stream_core_main_conf_t)); | |
85 if (cmcf == NULL) { | |
86 return NULL; | |
87 } | |
88 | |
89 if (ngx_array_init(&cmcf->servers, cf->pool, 4, | |
90 sizeof(ngx_stream_core_srv_conf_t *)) | |
91 != NGX_OK) | |
92 { | |
93 return NULL; | |
94 } | |
95 | |
96 if (ngx_array_init(&cmcf->listen, cf->pool, 4, sizeof(ngx_stream_listen_t)) | |
97 != NGX_OK) | |
98 { | |
99 return NULL; | |
100 } | |
101 | |
102 return cmcf; | |
103 } | |
104 | |
105 | |
106 static void * | |
107 ngx_stream_core_create_srv_conf(ngx_conf_t *cf) | |
108 { | |
109 ngx_stream_core_srv_conf_t *cscf; | |
110 | |
111 cscf = ngx_pcalloc(cf->pool, sizeof(ngx_stream_core_srv_conf_t)); | |
112 if (cscf == NULL) { | |
113 return NULL; | |
114 } | |
115 | |
116 /* | |
117 * set by ngx_pcalloc(): | |
118 * | |
119 * cscf->handler = NULL; | |
120 * cscf->error_log = NULL; | |
121 */ | |
122 | |
123 cscf->file_name = cf->conf_file->file.name.data; | |
124 cscf->line = cf->conf_file->line; | |
125 | |
126 return cscf; | |
127 } | |
128 | |
129 | |
130 static char * | |
131 ngx_stream_core_merge_srv_conf(ngx_conf_t *cf, void *parent, void *child) | |
132 { | |
133 ngx_stream_core_srv_conf_t *prev = parent; | |
134 ngx_stream_core_srv_conf_t *conf = child; | |
135 | |
136 if (conf->handler == NULL) { | |
137 ngx_log_error(NGX_LOG_EMERG, cf->log, 0, | |
138 "no handler for server in %s:%ui", | |
139 conf->file_name, conf->line); | |
140 return NGX_CONF_ERROR; | |
141 } | |
142 | |
143 if (conf->error_log == NULL) { | |
144 if (prev->error_log) { | |
145 conf->error_log = prev->error_log; | |
146 } else { | |
147 conf->error_log = &cf->cycle->new_log; | |
148 } | |
149 } | |
150 | |
151 return NGX_CONF_OK; | |
152 } | |
153 | |
154 | |
155 static char * | |
156 ngx_stream_core_error_log(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
157 { | |
158 ngx_stream_core_srv_conf_t *cscf = conf; | |
159 | |
160 return ngx_log_set_log(cf, &cscf->error_log); | |
161 } | |
162 | |
163 | |
164 static char * | |
165 ngx_stream_core_server(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
166 { | |
167 char *rv; | |
168 void *mconf; | |
169 ngx_uint_t m; | |
170 ngx_conf_t pcf; | |
171 ngx_stream_module_t *module; | |
172 ngx_stream_conf_ctx_t *ctx, *stream_ctx; | |
173 ngx_stream_core_srv_conf_t *cscf, **cscfp; | |
174 ngx_stream_core_main_conf_t *cmcf; | |
175 | |
176 ctx = ngx_pcalloc(cf->pool, sizeof(ngx_stream_conf_ctx_t)); | |
177 if (ctx == NULL) { | |
178 return NGX_CONF_ERROR; | |
179 } | |
180 | |
181 stream_ctx = cf->ctx; | |
182 ctx->main_conf = stream_ctx->main_conf; | |
183 | |
184 /* the server{}'s srv_conf */ | |
185 | |
186 ctx->srv_conf = ngx_pcalloc(cf->pool, | |
187 sizeof(void *) * ngx_stream_max_module); | |
188 if (ctx->srv_conf == NULL) { | |
189 return NGX_CONF_ERROR; | |
190 } | |
191 | |
192 for (m = 0; ngx_modules[m]; m++) { | |
193 if (ngx_modules[m]->type != NGX_STREAM_MODULE) { | |
194 continue; | |
195 } | |
196 | |
197 module = ngx_modules[m]->ctx; | |
198 | |
199 if (module->create_srv_conf) { | |
200 mconf = module->create_srv_conf(cf); | |
201 if (mconf == NULL) { | |
202 return NGX_CONF_ERROR; | |
203 } | |
204 | |
205 ctx->srv_conf[ngx_modules[m]->ctx_index] = mconf; | |
206 } | |
207 } | |
208 | |
209 /* the server configuration context */ | |
210 | |
211 cscf = ctx->srv_conf[ngx_stream_core_module.ctx_index]; | |
212 cscf->ctx = ctx; | |
213 | |
214 cmcf = ctx->main_conf[ngx_stream_core_module.ctx_index]; | |
215 | |
216 cscfp = ngx_array_push(&cmcf->servers); | |
217 if (cscfp == NULL) { | |
218 return NGX_CONF_ERROR; | |
219 } | |
220 | |
221 *cscfp = cscf; | |
222 | |
223 | |
224 /* parse inside server{} */ | |
225 | |
226 pcf = *cf; | |
227 cf->ctx = ctx; | |
228 cf->cmd_type = NGX_STREAM_SRV_CONF; | |
229 | |
230 rv = ngx_conf_parse(cf, NULL); | |
231 | |
232 *cf = pcf; | |
233 | |
234 return rv; | |
235 } | |
236 | |
237 | |
238 static char * | |
239 ngx_stream_core_listen(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
240 { | |
241 size_t len, off; | |
242 in_port_t port; | |
243 ngx_str_t *value; | |
244 ngx_url_t u; | |
245 ngx_uint_t i; | |
246 struct sockaddr *sa; | |
247 struct sockaddr_in *sin; | |
248 ngx_stream_listen_t *ls; | |
249 ngx_stream_core_main_conf_t *cmcf; | |
250 #if (NGX_HAVE_INET6) | |
251 struct sockaddr_in6 *sin6; | |
252 #endif | |
253 | |
254 value = cf->args->elts; | |
255 | |
256 ngx_memzero(&u, sizeof(ngx_url_t)); | |
257 | |
258 u.url = value[1]; | |
259 u.listen = 1; | |
260 | |
261 if (ngx_parse_url(cf->pool, &u) != NGX_OK) { | |
262 if (u.err) { | |
263 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
264 "%s in \"%V\" of the \"listen\" directive", | |
265 u.err, &u.url); | |
266 } | |
267 | |
268 return NGX_CONF_ERROR; | |
269 } | |
270 | |
271 cmcf = ngx_stream_conf_get_module_main_conf(cf, ngx_stream_core_module); | |
272 | |
273 ls = cmcf->listen.elts; | |
274 | |
275 for (i = 0; i < cmcf->listen.nelts; i++) { | |
276 | |
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
277 sa = &ls[i].u.sockaddr; |
6115 | 278 |
279 if (sa->sa_family != u.family) { | |
280 continue; | |
281 } | |
282 | |
283 switch (sa->sa_family) { | |
284 | |
285 #if (NGX_HAVE_INET6) | |
286 case AF_INET6: | |
287 off = offsetof(struct sockaddr_in6, sin6_addr); | |
288 len = 16; | |
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
289 sin6 = &ls[i].u.sockaddr_in6; |
6115 | 290 port = sin6->sin6_port; |
291 break; | |
292 #endif | |
293 | |
294 #if (NGX_HAVE_UNIX_DOMAIN) | |
295 case AF_UNIX: | |
296 off = offsetof(struct sockaddr_un, sun_path); | |
297 len = sizeof(((struct sockaddr_un *) sa)->sun_path); | |
298 port = 0; | |
299 break; | |
300 #endif | |
301 | |
302 default: /* AF_INET */ | |
303 off = offsetof(struct sockaddr_in, sin_addr); | |
304 len = 4; | |
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
305 sin = &ls[i].u.sockaddr_in; |
6115 | 306 port = sin->sin_port; |
307 break; | |
308 } | |
309 | |
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
310 if (ngx_memcmp(ls[i].u.sockaddr_data + off, u.sockaddr + off, len) |
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
311 != 0) |
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
312 { |
6115 | 313 continue; |
314 } | |
315 | |
316 if (port != u.port) { | |
317 continue; | |
318 } | |
319 | |
320 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
321 "duplicate \"%V\" address and port pair", &u.url); | |
322 return NGX_CONF_ERROR; | |
323 } | |
324 | |
325 ls = ngx_array_push(&cmcf->listen); | |
326 if (ls == NULL) { | |
327 return NGX_CONF_ERROR; | |
328 } | |
329 | |
330 ngx_memzero(ls, sizeof(ngx_stream_listen_t)); | |
331 | |
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
332 ngx_memcpy(&ls->u.sockaddr, u.sockaddr, u.socklen); |
6115 | 333 |
334 ls->socklen = u.socklen; | |
6172 | 335 ls->backlog = NGX_LISTEN_BACKLOG; |
6115 | 336 ls->wildcard = u.wildcard; |
337 ls->ctx = cf->ctx; | |
338 | |
339 #if (NGX_HAVE_INET6 && defined IPV6_V6ONLY) | |
340 ls->ipv6only = 1; | |
341 #endif | |
342 | |
343 for (i = 2; i < cf->args->nelts; i++) { | |
344 | |
345 if (ngx_strcmp(value[i].data, "bind") == 0) { | |
346 ls->bind = 1; | |
347 continue; | |
348 } | |
349 | |
6172 | 350 if (ngx_strncmp(value[i].data, "backlog=", 8) == 0) { |
351 ls->backlog = ngx_atoi(value[i].data + 8, value[i].len - 8); | |
352 ls->bind = 1; | |
353 | |
354 if (ls->backlog == NGX_ERROR || ls->backlog == 0) { | |
355 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
356 "invalid backlog \"%V\"", &value[i]); | |
357 return NGX_CONF_ERROR; | |
358 } | |
359 | |
360 continue; | |
361 } | |
362 | |
6115 | 363 if (ngx_strncmp(value[i].data, "ipv6only=o", 10) == 0) { |
364 #if (NGX_HAVE_INET6 && defined IPV6_V6ONLY) | |
365 struct sockaddr *sa; | |
366 u_char buf[NGX_SOCKADDR_STRLEN]; | |
367 | |
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
368 sa = &ls->u.sockaddr; |
6115 | 369 |
370 if (sa->sa_family == AF_INET6) { | |
371 | |
372 if (ngx_strcmp(&value[i].data[10], "n") == 0) { | |
373 ls->ipv6only = 1; | |
374 | |
375 } else if (ngx_strcmp(&value[i].data[10], "ff") == 0) { | |
376 ls->ipv6only = 0; | |
377 | |
378 } else { | |
379 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
380 "invalid ipv6only flags \"%s\"", | |
381 &value[i].data[9]); | |
382 return NGX_CONF_ERROR; | |
383 } | |
384 | |
385 ls->bind = 1; | |
386 | |
387 } else { | |
388 len = ngx_sock_ntop(sa, ls->socklen, buf, | |
389 NGX_SOCKADDR_STRLEN, 1); | |
390 | |
391 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
392 "ipv6only is not supported " | |
393 "on addr \"%*s\", ignored", len, buf); | |
394 } | |
395 | |
396 continue; | |
397 #else | |
398 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
399 "bind ipv6only is not supported " | |
400 "on this platform"); | |
401 return NGX_CONF_ERROR; | |
402 #endif | |
403 } | |
404 | |
6153
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
405 if (ngx_strcmp(value[i].data, "reuseport") == 0) { |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
406 #if (NGX_HAVE_REUSEPORT) |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
407 ls->reuseport = 1; |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
408 ls->bind = 1; |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
409 #else |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
410 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
411 "reuseport is not supported " |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
412 "on this platform, ignored"); |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
413 #endif |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
414 continue; |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
415 } |
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6115
diff
changeset
|
416 |
6115 | 417 if (ngx_strcmp(value[i].data, "ssl") == 0) { |
418 #if (NGX_STREAM_SSL) | |
419 ls->ssl = 1; | |
420 continue; | |
421 #else | |
422 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
423 "the \"ssl\" parameter requires " | |
424 "ngx_stream_ssl_module"); | |
425 return NGX_CONF_ERROR; | |
426 #endif | |
427 } | |
428 | |
429 if (ngx_strncmp(value[i].data, "so_keepalive=", 13) == 0) { | |
430 | |
431 if (ngx_strcmp(&value[i].data[13], "on") == 0) { | |
432 ls->so_keepalive = 1; | |
433 | |
434 } else if (ngx_strcmp(&value[i].data[13], "off") == 0) { | |
435 ls->so_keepalive = 2; | |
436 | |
437 } else { | |
438 | |
439 #if (NGX_HAVE_KEEPALIVE_TUNABLE) | |
440 u_char *p, *end; | |
441 ngx_str_t s; | |
442 | |
443 end = value[i].data + value[i].len; | |
444 s.data = value[i].data + 13; | |
445 | |
446 p = ngx_strlchr(s.data, end, ':'); | |
447 if (p == NULL) { | |
448 p = end; | |
449 } | |
450 | |
451 if (p > s.data) { | |
452 s.len = p - s.data; | |
453 | |
454 ls->tcp_keepidle = ngx_parse_time(&s, 1); | |
455 if (ls->tcp_keepidle == (time_t) NGX_ERROR) { | |
456 goto invalid_so_keepalive; | |
457 } | |
458 } | |
459 | |
460 s.data = (p < end) ? (p + 1) : end; | |
461 | |
462 p = ngx_strlchr(s.data, end, ':'); | |
463 if (p == NULL) { | |
464 p = end; | |
465 } | |
466 | |
467 if (p > s.data) { | |
468 s.len = p - s.data; | |
469 | |
470 ls->tcp_keepintvl = ngx_parse_time(&s, 1); | |
471 if (ls->tcp_keepintvl == (time_t) NGX_ERROR) { | |
472 goto invalid_so_keepalive; | |
473 } | |
474 } | |
475 | |
476 s.data = (p < end) ? (p + 1) : end; | |
477 | |
478 if (s.data < end) { | |
479 s.len = end - s.data; | |
480 | |
481 ls->tcp_keepcnt = ngx_atoi(s.data, s.len); | |
482 if (ls->tcp_keepcnt == NGX_ERROR) { | |
483 goto invalid_so_keepalive; | |
484 } | |
485 } | |
486 | |
487 if (ls->tcp_keepidle == 0 && ls->tcp_keepintvl == 0 | |
488 && ls->tcp_keepcnt == 0) | |
489 { | |
490 goto invalid_so_keepalive; | |
491 } | |
492 | |
493 ls->so_keepalive = 1; | |
494 | |
495 #else | |
496 | |
497 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
498 "the \"so_keepalive\" parameter accepts " | |
499 "only \"on\" or \"off\" on this platform"); | |
500 return NGX_CONF_ERROR; | |
501 | |
502 #endif | |
503 } | |
504 | |
505 ls->bind = 1; | |
506 | |
507 continue; | |
508 | |
509 #if (NGX_HAVE_KEEPALIVE_TUNABLE) | |
510 invalid_so_keepalive: | |
511 | |
512 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
513 "invalid so_keepalive value: \"%s\"", | |
514 &value[i].data[13]); | |
515 return NGX_CONF_ERROR; | |
516 #endif | |
517 } | |
518 | |
519 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
520 "the invalid \"%V\" parameter", &value[i]); | |
521 return NGX_CONF_ERROR; | |
522 } | |
523 | |
524 return NGX_CONF_OK; | |
525 } |