Mercurial > hg > nginx
annotate src/stream/ngx_stream.c @ 6736:f41c2530dd17
Modules compatibility: removed dependencies on NGX_STREAM_SSL.
External structures are now identical regardless of stream SSL module
compiled in or not.
| author | Maxim Dounin <mdounin@mdounin.ru> |
|---|---|
| date | Mon, 10 Oct 2016 18:44:17 +0300 |
| parents | cebf5fed00bf |
| children | d7ce41bdf050 |
| 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_event.h> | |
| 11 #include <ngx_stream.h> | |
| 12 | |
| 13 | |
| 14 static char *ngx_stream_block(ngx_conf_t *cf, ngx_command_t *cmd, void *conf); | |
| 6693 | 15 static ngx_int_t ngx_stream_init_phases(ngx_conf_t *cf, |
| 16 ngx_stream_core_main_conf_t *cmcf); | |
| 17 static ngx_int_t ngx_stream_init_phase_handlers(ngx_conf_t *cf, | |
| 18 ngx_stream_core_main_conf_t *cmcf); | |
| 6115 | 19 static ngx_int_t ngx_stream_add_ports(ngx_conf_t *cf, ngx_array_t *ports, |
| 20 ngx_stream_listen_t *listen); | |
| 21 static char *ngx_stream_optimize_servers(ngx_conf_t *cf, ngx_array_t *ports); | |
| 22 static ngx_int_t ngx_stream_add_addrs(ngx_conf_t *cf, ngx_stream_port_t *stport, | |
| 23 ngx_stream_conf_addr_t *addr); | |
| 24 #if (NGX_HAVE_INET6) | |
| 25 static ngx_int_t ngx_stream_add_addrs6(ngx_conf_t *cf, | |
| 26 ngx_stream_port_t *stport, ngx_stream_conf_addr_t *addr); | |
| 27 #endif | |
| 28 static ngx_int_t ngx_stream_cmp_conf_addrs(const void *one, const void *two); | |
| 29 | |
| 30 | |
| 31 ngx_uint_t ngx_stream_max_module; | |
| 32 | |
| 33 | |
| 6692 | 34 ngx_stream_filter_pt ngx_stream_top_filter; |
| 35 | |
| 36 | |
| 6115 | 37 static ngx_command_t ngx_stream_commands[] = { |
| 38 | |
| 39 { ngx_string("stream"), | |
| 40 NGX_MAIN_CONF|NGX_CONF_BLOCK|NGX_CONF_NOARGS, | |
| 41 ngx_stream_block, | |
| 42 0, | |
| 43 0, | |
| 44 NULL }, | |
| 45 | |
| 46 ngx_null_command | |
| 47 }; | |
| 48 | |
| 49 | |
| 50 static ngx_core_module_t ngx_stream_module_ctx = { | |
| 51 ngx_string("stream"), | |
| 52 NULL, | |
| 53 NULL | |
| 54 }; | |
| 55 | |
| 56 | |
| 57 ngx_module_t ngx_stream_module = { | |
| 58 NGX_MODULE_V1, | |
| 59 &ngx_stream_module_ctx, /* module context */ | |
| 60 ngx_stream_commands, /* module directives */ | |
| 61 NGX_CORE_MODULE, /* module type */ | |
| 62 NULL, /* init master */ | |
| 63 NULL, /* init module */ | |
| 64 NULL, /* init process */ | |
| 65 NULL, /* init thread */ | |
| 66 NULL, /* exit thread */ | |
| 67 NULL, /* exit process */ | |
| 68 NULL, /* exit master */ | |
| 69 NGX_MODULE_V1_PADDING | |
| 70 }; | |
| 71 | |
| 72 | |
| 73 static char * | |
| 74 ngx_stream_block(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
| 75 { | |
| 76 char *rv; | |
| 77 ngx_uint_t i, m, mi, s; | |
| 78 ngx_conf_t pcf; | |
| 79 ngx_array_t ports; | |
| 80 ngx_stream_listen_t *listen; | |
| 81 ngx_stream_module_t *module; | |
| 82 ngx_stream_conf_ctx_t *ctx; | |
| 83 ngx_stream_core_srv_conf_t **cscfp; | |
| 84 ngx_stream_core_main_conf_t *cmcf; | |
| 85 | |
|
6193
78c06e5e1d76
Disabled duplicate http, mail, and stream blocks.
Vladimir Homutov <vl@nginx.com>
parents:
6174
diff
changeset
|
86 if (*(ngx_stream_conf_ctx_t **) conf) { |
|
78c06e5e1d76
Disabled duplicate http, mail, and stream blocks.
Vladimir Homutov <vl@nginx.com>
parents:
6174
diff
changeset
|
87 return "is duplicate"; |
|
78c06e5e1d76
Disabled duplicate http, mail, and stream blocks.
Vladimir Homutov <vl@nginx.com>
parents:
6174
diff
changeset
|
88 } |
|
78c06e5e1d76
Disabled duplicate http, mail, and stream blocks.
Vladimir Homutov <vl@nginx.com>
parents:
6174
diff
changeset
|
89 |
| 6115 | 90 /* the main stream context */ |
| 91 | |
| 92 ctx = ngx_pcalloc(cf->pool, sizeof(ngx_stream_conf_ctx_t)); | |
| 93 if (ctx == NULL) { | |
| 94 return NGX_CONF_ERROR; | |
| 95 } | |
| 96 | |
| 97 *(ngx_stream_conf_ctx_t **) conf = ctx; | |
| 98 | |
| 99 /* count the number of the stream modules and set up their indices */ | |
| 100 | |
|
6378
0f203a2af17c
Dynamic modules: moved module-related stuff to separate files.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6193
diff
changeset
|
101 ngx_stream_max_module = ngx_count_modules(cf->cycle, NGX_STREAM_MODULE); |
| 6115 | 102 |
| 103 | |
| 104 /* the stream main_conf context, it's the same in the all stream contexts */ | |
| 105 | |
| 106 ctx->main_conf = ngx_pcalloc(cf->pool, | |
| 107 sizeof(void *) * ngx_stream_max_module); | |
| 108 if (ctx->main_conf == NULL) { | |
| 109 return NGX_CONF_ERROR; | |
| 110 } | |
| 111 | |
| 112 | |
| 113 /* | |
| 114 * the stream null srv_conf context, it is used to merge | |
| 115 * the server{}s' srv_conf's | |
| 116 */ | |
| 117 | |
| 118 ctx->srv_conf = ngx_pcalloc(cf->pool, | |
| 119 sizeof(void *) * ngx_stream_max_module); | |
| 120 if (ctx->srv_conf == NULL) { | |
| 121 return NGX_CONF_ERROR; | |
| 122 } | |
| 123 | |
| 124 | |
| 125 /* | |
| 126 * create the main_conf's and the null srv_conf's of the all stream modules | |
| 127 */ | |
| 128 | |
|
6379
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
129 for (m = 0; cf->cycle->modules[m]; m++) { |
|
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
130 if (cf->cycle->modules[m]->type != NGX_STREAM_MODULE) { |
| 6115 | 131 continue; |
| 132 } | |
| 133 | |
|
6379
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
134 module = cf->cycle->modules[m]->ctx; |
|
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
135 mi = cf->cycle->modules[m]->ctx_index; |
| 6115 | 136 |
| 137 if (module->create_main_conf) { | |
| 138 ctx->main_conf[mi] = module->create_main_conf(cf); | |
| 139 if (ctx->main_conf[mi] == NULL) { | |
| 140 return NGX_CONF_ERROR; | |
| 141 } | |
| 142 } | |
| 143 | |
| 144 if (module->create_srv_conf) { | |
| 145 ctx->srv_conf[mi] = module->create_srv_conf(cf); | |
| 146 if (ctx->srv_conf[mi] == NULL) { | |
| 147 return NGX_CONF_ERROR; | |
| 148 } | |
| 149 } | |
| 150 } | |
| 151 | |
| 152 | |
| 153 pcf = *cf; | |
| 154 cf->ctx = ctx; | |
| 155 | |
|
6606
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
156 for (m = 0; cf->cycle->modules[m]; m++) { |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
157 if (cf->cycle->modules[m]->type != NGX_STREAM_MODULE) { |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
158 continue; |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
159 } |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
160 |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
161 module = cf->cycle->modules[m]->ctx; |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
162 |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
163 if (module->preconfiguration) { |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
164 if (module->preconfiguration(cf) != NGX_OK) { |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
165 return NGX_CONF_ERROR; |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
166 } |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
167 } |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
168 } |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
169 |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
170 |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
171 /* parse inside the stream{} block */ |
|
2f41d383c9c7
Stream: added preconfiguration step.
Vladimir Homutov <vl@nginx.com>
parents:
6593
diff
changeset
|
172 |
| 6115 | 173 cf->module_type = NGX_STREAM_MODULE; |
| 174 cf->cmd_type = NGX_STREAM_MAIN_CONF; | |
| 175 rv = ngx_conf_parse(cf, NULL); | |
| 176 | |
| 177 if (rv != NGX_CONF_OK) { | |
| 178 *cf = pcf; | |
| 179 return rv; | |
| 180 } | |
| 181 | |
| 182 | |
| 183 /* init stream{} main_conf's, merge the server{}s' srv_conf's */ | |
| 184 | |
| 185 cmcf = ctx->main_conf[ngx_stream_core_module.ctx_index]; | |
| 186 cscfp = cmcf->servers.elts; | |
| 187 | |
|
6379
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
188 for (m = 0; cf->cycle->modules[m]; m++) { |
|
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
189 if (cf->cycle->modules[m]->type != NGX_STREAM_MODULE) { |
| 6115 | 190 continue; |
| 191 } | |
| 192 | |
|
6379
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
193 module = cf->cycle->modules[m]->ctx; |
|
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
194 mi = cf->cycle->modules[m]->ctx_index; |
| 6115 | 195 |
| 196 /* init stream{} main_conf's */ | |
| 197 | |
| 198 cf->ctx = ctx; | |
| 199 | |
| 200 if (module->init_main_conf) { | |
| 201 rv = module->init_main_conf(cf, ctx->main_conf[mi]); | |
| 202 if (rv != NGX_CONF_OK) { | |
| 203 *cf = pcf; | |
| 204 return rv; | |
| 205 } | |
| 206 } | |
| 207 | |
| 208 for (s = 0; s < cmcf->servers.nelts; s++) { | |
| 209 | |
| 210 /* merge the server{}s' srv_conf's */ | |
| 211 | |
| 212 cf->ctx = cscfp[s]->ctx; | |
| 213 | |
| 214 if (module->merge_srv_conf) { | |
| 215 rv = module->merge_srv_conf(cf, | |
| 216 ctx->srv_conf[mi], | |
| 217 cscfp[s]->ctx->srv_conf[mi]); | |
| 218 if (rv != NGX_CONF_OK) { | |
| 219 *cf = pcf; | |
| 220 return rv; | |
| 221 } | |
| 222 } | |
| 223 } | |
| 224 } | |
| 225 | |
| 6693 | 226 if (ngx_stream_init_phases(cf, cmcf) != NGX_OK) { |
| 227 return NGX_CONF_ERROR; | |
| 228 } | |
| 229 | |
|
6379
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
230 for (m = 0; cf->cycle->modules[m]; m++) { |
|
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
231 if (cf->cycle->modules[m]->type != NGX_STREAM_MODULE) { |
|
6174
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
232 continue; |
|
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
233 } |
|
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
234 |
|
6379
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6378
diff
changeset
|
235 module = cf->cycle->modules[m]->ctx; |
|
6174
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
236 |
|
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
237 if (module->postconfiguration) { |
|
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
238 if (module->postconfiguration(cf) != NGX_OK) { |
|
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
239 return NGX_CONF_ERROR; |
|
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
240 } |
|
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
241 } |
|
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
242 } |
|
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6172
diff
changeset
|
243 |
|
6607
c70b7f4537e1
Stream: variables and script.
Vladimir Homutov <vl@nginx.com>
parents:
6606
diff
changeset
|
244 if (ngx_stream_variables_init_vars(cf) != NGX_OK) { |
|
c70b7f4537e1
Stream: variables and script.
Vladimir Homutov <vl@nginx.com>
parents:
6606
diff
changeset
|
245 return NGX_CONF_ERROR; |
|
c70b7f4537e1
Stream: variables and script.
Vladimir Homutov <vl@nginx.com>
parents:
6606
diff
changeset
|
246 } |
|
c70b7f4537e1
Stream: variables and script.
Vladimir Homutov <vl@nginx.com>
parents:
6606
diff
changeset
|
247 |
| 6115 | 248 *cf = pcf; |
| 249 | |
| 6693 | 250 if (ngx_stream_init_phase_handlers(cf, cmcf) != NGX_OK) { |
| 251 return NGX_CONF_ERROR; | |
| 252 } | |
| 6115 | 253 |
| 254 if (ngx_array_init(&ports, cf->temp_pool, 4, sizeof(ngx_stream_conf_port_t)) | |
| 255 != NGX_OK) | |
| 256 { | |
| 257 return NGX_CONF_ERROR; | |
| 258 } | |
| 259 | |
| 260 listen = cmcf->listen.elts; | |
| 261 | |
| 262 for (i = 0; i < cmcf->listen.nelts; i++) { | |
| 263 if (ngx_stream_add_ports(cf, &ports, &listen[i]) != NGX_OK) { | |
| 264 return NGX_CONF_ERROR; | |
| 265 } | |
| 266 } | |
| 267 | |
| 268 return ngx_stream_optimize_servers(cf, &ports); | |
| 269 } | |
| 270 | |
| 271 | |
| 272 static ngx_int_t | |
| 6693 | 273 ngx_stream_init_phases(ngx_conf_t *cf, ngx_stream_core_main_conf_t *cmcf) |
| 274 { | |
| 275 if (ngx_array_init(&cmcf->phases[NGX_STREAM_POST_ACCEPT_PHASE].handlers, | |
| 276 cf->pool, 1, sizeof(ngx_stream_handler_pt)) | |
| 277 != NGX_OK) | |
| 278 { | |
| 279 return NGX_ERROR; | |
| 280 } | |
| 281 | |
| 282 if (ngx_array_init(&cmcf->phases[NGX_STREAM_PREACCESS_PHASE].handlers, | |
| 283 cf->pool, 1, sizeof(ngx_stream_handler_pt)) | |
| 284 != NGX_OK) | |
| 285 { | |
| 286 return NGX_ERROR; | |
| 287 } | |
| 288 | |
| 289 if (ngx_array_init(&cmcf->phases[NGX_STREAM_ACCESS_PHASE].handlers, | |
| 290 cf->pool, 1, sizeof(ngx_stream_handler_pt)) | |
| 291 != NGX_OK) | |
| 292 { | |
| 293 return NGX_ERROR; | |
| 294 } | |
| 295 | |
| 296 if (ngx_array_init(&cmcf->phases[NGX_STREAM_SSL_PHASE].handlers, | |
| 297 cf->pool, 1, sizeof(ngx_stream_handler_pt)) | |
| 298 != NGX_OK) | |
| 299 { | |
| 300 return NGX_ERROR; | |
| 301 } | |
| 302 | |
| 6694 | 303 if (ngx_array_init(&cmcf->phases[NGX_STREAM_PREREAD_PHASE].handlers, |
| 304 cf->pool, 1, sizeof(ngx_stream_handler_pt)) | |
| 305 != NGX_OK) | |
| 306 { | |
| 307 return NGX_ERROR; | |
| 308 } | |
| 309 | |
| 6693 | 310 if (ngx_array_init(&cmcf->phases[NGX_STREAM_LOG_PHASE].handlers, |
| 311 cf->pool, 1, sizeof(ngx_stream_handler_pt)) | |
| 312 != NGX_OK) | |
| 313 { | |
| 314 return NGX_ERROR; | |
| 315 } | |
| 316 | |
| 317 return NGX_OK; | |
| 318 } | |
| 319 | |
| 320 | |
| 321 static ngx_int_t | |
| 322 ngx_stream_init_phase_handlers(ngx_conf_t *cf, | |
| 323 ngx_stream_core_main_conf_t *cmcf) | |
| 324 { | |
| 325 ngx_int_t j; | |
| 326 ngx_uint_t i, n; | |
| 327 ngx_stream_handler_pt *h; | |
| 328 ngx_stream_phase_handler_t *ph; | |
| 329 ngx_stream_phase_handler_pt checker; | |
| 330 | |
| 331 n = 1 /* content phase */; | |
| 332 | |
| 333 for (i = 0; i < NGX_STREAM_LOG_PHASE; i++) { | |
| 334 n += cmcf->phases[i].handlers.nelts; | |
| 335 } | |
| 336 | |
| 337 ph = ngx_pcalloc(cf->pool, | |
| 338 n * sizeof(ngx_stream_phase_handler_t) + sizeof(void *)); | |
| 339 if (ph == NULL) { | |
| 340 return NGX_ERROR; | |
| 341 } | |
| 342 | |
| 343 cmcf->phase_engine.handlers = ph; | |
| 344 n = 0; | |
| 345 | |
| 346 for (i = 0; i < NGX_STREAM_LOG_PHASE; i++) { | |
| 347 h = cmcf->phases[i].handlers.elts; | |
| 348 | |
| 349 switch (i) { | |
| 350 | |
| 6694 | 351 case NGX_STREAM_PREREAD_PHASE: |
| 352 checker = ngx_stream_core_preread_phase; | |
| 353 break; | |
| 354 | |
| 6693 | 355 case NGX_STREAM_CONTENT_PHASE: |
| 356 ph->checker = ngx_stream_core_content_phase; | |
| 357 n++; | |
| 358 ph++; | |
| 359 | |
| 360 continue; | |
| 361 | |
| 362 default: | |
| 363 checker = ngx_stream_core_generic_phase; | |
| 364 } | |
| 365 | |
| 366 n += cmcf->phases[i].handlers.nelts; | |
| 367 | |
| 368 for (j = cmcf->phases[i].handlers.nelts - 1; j >= 0; j--) { | |
| 369 ph->checker = checker; | |
| 370 ph->handler = h[j]; | |
| 371 ph->next = n; | |
| 372 ph++; | |
| 373 } | |
| 374 } | |
| 375 | |
| 376 return NGX_OK; | |
| 377 } | |
| 378 | |
| 379 | |
| 380 static ngx_int_t | |
| 6115 | 381 ngx_stream_add_ports(ngx_conf_t *cf, ngx_array_t *ports, |
| 382 ngx_stream_listen_t *listen) | |
| 383 { | |
| 384 in_port_t p; | |
| 385 ngx_uint_t i; | |
| 386 struct sockaddr *sa; | |
| 387 ngx_stream_conf_port_t *port; | |
| 388 ngx_stream_conf_addr_t *addr; | |
| 389 | |
|
6560
c90cf79d0e1d
Renamed "u" to "sockaddr" in listen options types.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6436
diff
changeset
|
390 sa = &listen->sockaddr.sockaddr; |
|
6593
b3b7e33083ac
Introduced ngx_inet_get_port() and ngx_inet_set_port() functions.
Roman Arutyunyan <arut@nginx.com>
parents:
6560
diff
changeset
|
391 p = ngx_inet_get_port(sa); |
| 6115 | 392 |
| 393 port = ports->elts; | |
| 394 for (i = 0; i < ports->nelts; i++) { | |
| 395 | |
| 6436 | 396 if (p == port[i].port |
| 397 && listen->type == port[i].type | |
| 398 && sa->sa_family == port[i].family) | |
| 399 { | |
| 6115 | 400 /* a port is already in the port list */ |
| 401 | |
| 402 port = &port[i]; | |
| 403 goto found; | |
| 404 } | |
| 405 } | |
| 406 | |
| 407 /* add a port to the port list */ | |
| 408 | |
| 409 port = ngx_array_push(ports); | |
| 410 if (port == NULL) { | |
| 411 return NGX_ERROR; | |
| 412 } | |
| 413 | |
| 414 port->family = sa->sa_family; | |
| 6436 | 415 port->type = listen->type; |
| 6115 | 416 port->port = p; |
| 417 | |
| 418 if (ngx_array_init(&port->addrs, cf->temp_pool, 2, | |
| 419 sizeof(ngx_stream_conf_addr_t)) | |
| 420 != NGX_OK) | |
| 421 { | |
| 422 return NGX_ERROR; | |
| 423 } | |
| 424 | |
| 425 found: | |
| 426 | |
| 427 addr = ngx_array_push(&port->addrs); | |
| 428 if (addr == NULL) { | |
| 429 return NGX_ERROR; | |
| 430 } | |
| 431 | |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
432 addr->opt = *listen; |
| 6115 | 433 |
| 434 return NGX_OK; | |
| 435 } | |
| 436 | |
| 437 | |
| 438 static char * | |
| 439 ngx_stream_optimize_servers(ngx_conf_t *cf, ngx_array_t *ports) | |
| 440 { | |
| 441 ngx_uint_t i, p, last, bind_wildcard; | |
| 442 ngx_listening_t *ls; | |
| 443 ngx_stream_port_t *stport; | |
| 444 ngx_stream_conf_port_t *port; | |
| 445 ngx_stream_conf_addr_t *addr; | |
| 446 ngx_stream_core_srv_conf_t *cscf; | |
| 447 | |
| 448 port = ports->elts; | |
| 449 for (p = 0; p < ports->nelts; p++) { | |
| 450 | |
| 451 ngx_sort(port[p].addrs.elts, (size_t) port[p].addrs.nelts, | |
| 452 sizeof(ngx_stream_conf_addr_t), ngx_stream_cmp_conf_addrs); | |
| 453 | |
| 454 addr = port[p].addrs.elts; | |
| 455 last = port[p].addrs.nelts; | |
| 456 | |
| 457 /* | |
| 458 * if there is the binding to the "*:port" then we need to bind() | |
| 459 * to the "*:port" only and ignore the other bindings | |
| 460 */ | |
| 461 | |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
462 if (addr[last - 1].opt.wildcard) { |
|
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
463 addr[last - 1].opt.bind = 1; |
| 6115 | 464 bind_wildcard = 1; |
| 465 | |
| 466 } else { | |
| 467 bind_wildcard = 0; | |
| 468 } | |
| 469 | |
| 470 i = 0; | |
| 471 | |
| 472 while (i < last) { | |
| 473 | |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
474 if (bind_wildcard && !addr[i].opt.bind) { |
| 6115 | 475 i++; |
| 476 continue; | |
| 477 } | |
| 478 | |
|
6560
c90cf79d0e1d
Renamed "u" to "sockaddr" in listen options types.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6436
diff
changeset
|
479 ls = ngx_create_listening(cf, &addr[i].opt.sockaddr.sockaddr, |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
480 addr[i].opt.socklen); |
| 6115 | 481 if (ls == NULL) { |
| 482 return NGX_CONF_ERROR; | |
| 483 } | |
| 484 | |
| 485 ls->addr_ntop = 1; | |
| 486 ls->handler = ngx_stream_init_connection; | |
| 487 ls->pool_size = 256; | |
| 6436 | 488 ls->type = addr[i].opt.type; |
| 6115 | 489 |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
490 cscf = addr->opt.ctx->srv_conf[ngx_stream_core_module.ctx_index]; |
|
6129
187aa751ad62
Core: the ngx_set_connection_log() macro.
Vladimir Homutov <vl@nginx.com>
parents:
6115
diff
changeset
|
491 |
| 6115 | 492 ls->logp = cscf->error_log; |
| 493 ls->log.data = &ls->addr_text; | |
| 494 ls->log.handler = ngx_accept_log_error; | |
| 495 | |
| 6172 | 496 ls->backlog = addr[i].opt.backlog; |
| 497 | |
| 6436 | 498 ls->wildcard = addr[i].opt.wildcard; |
| 499 | |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
500 ls->keepalive = addr[i].opt.so_keepalive; |
| 6115 | 501 #if (NGX_HAVE_KEEPALIVE_TUNABLE) |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
502 ls->keepidle = addr[i].opt.tcp_keepidle; |
|
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
503 ls->keepintvl = addr[i].opt.tcp_keepintvl; |
|
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
504 ls->keepcnt = addr[i].opt.tcp_keepcnt; |
| 6115 | 505 #endif |
| 506 | |
|
6719
cebf5fed00bf
Modules compatibility: removed unneeded IPV6_V6ONLY checks.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6694
diff
changeset
|
507 #if (NGX_HAVE_INET6) |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
508 ls->ipv6only = addr[i].opt.ipv6only; |
| 6115 | 509 #endif |
| 510 | |
|
6169
f654addf0eea
Stream: fixed "reuseport" to actually work.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
511 #if (NGX_HAVE_REUSEPORT) |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
512 ls->reuseport = addr[i].opt.reuseport; |
|
6169
f654addf0eea
Stream: fixed "reuseport" to actually work.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
513 #endif |
|
f654addf0eea
Stream: fixed "reuseport" to actually work.
Ruslan Ermilov <ru@nginx.com>
parents:
6153
diff
changeset
|
514 |
| 6115 | 515 stport = ngx_palloc(cf->pool, sizeof(ngx_stream_port_t)); |
| 516 if (stport == NULL) { | |
| 517 return NGX_CONF_ERROR; | |
| 518 } | |
| 519 | |
| 520 ls->servers = stport; | |
| 521 | |
|
6152
3c344ea7d88b
Simplified ngx_http_init_listening().
Maxim Dounin <mdounin@mdounin.ru>
parents:
6129
diff
changeset
|
522 stport->naddrs = i + 1; |
| 6115 | 523 |
| 524 switch (ls->sockaddr->sa_family) { | |
| 525 #if (NGX_HAVE_INET6) | |
| 526 case AF_INET6: | |
| 527 if (ngx_stream_add_addrs6(cf, stport, addr) != NGX_OK) { | |
| 528 return NGX_CONF_ERROR; | |
| 529 } | |
| 530 break; | |
| 531 #endif | |
| 532 default: /* AF_INET */ | |
| 533 if (ngx_stream_add_addrs(cf, stport, addr) != NGX_OK) { | |
| 534 return NGX_CONF_ERROR; | |
| 535 } | |
| 536 break; | |
| 537 } | |
| 538 | |
|
6153
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6152
diff
changeset
|
539 if (ngx_clone_listening(cf, ls) != NGX_OK) { |
|
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6152
diff
changeset
|
540 return NGX_CONF_ERROR; |
|
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6152
diff
changeset
|
541 } |
|
4f6efabcb09b
The "reuseport" option of the "listen" directive.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6152
diff
changeset
|
542 |
| 6115 | 543 addr++; |
| 544 last--; | |
| 545 } | |
| 546 } | |
| 547 | |
| 548 return NGX_CONF_OK; | |
| 549 } | |
| 550 | |
| 551 | |
| 552 static ngx_int_t | |
| 553 ngx_stream_add_addrs(ngx_conf_t *cf, ngx_stream_port_t *stport, | |
| 554 ngx_stream_conf_addr_t *addr) | |
| 555 { | |
| 556 u_char *p; | |
| 557 size_t len; | |
| 558 ngx_uint_t i; | |
| 559 struct sockaddr_in *sin; | |
| 560 ngx_stream_in_addr_t *addrs; | |
| 561 u_char buf[NGX_SOCKADDR_STRLEN]; | |
| 562 | |
| 563 stport->addrs = ngx_pcalloc(cf->pool, | |
| 564 stport->naddrs * sizeof(ngx_stream_in_addr_t)); | |
| 565 if (stport->addrs == NULL) { | |
| 566 return NGX_ERROR; | |
| 567 } | |
| 568 | |
| 569 addrs = stport->addrs; | |
| 570 | |
| 571 for (i = 0; i < stport->naddrs; i++) { | |
| 572 | |
|
6560
c90cf79d0e1d
Renamed "u" to "sockaddr" in listen options types.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6436
diff
changeset
|
573 sin = &addr[i].opt.sockaddr.sockaddr_in; |
| 6115 | 574 addrs[i].addr = sin->sin_addr.s_addr; |
| 575 | |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
576 addrs[i].conf.ctx = addr[i].opt.ctx; |
| 6115 | 577 #if (NGX_STREAM_SSL) |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
578 addrs[i].conf.ssl = addr[i].opt.ssl; |
| 6115 | 579 #endif |
|
6680
7357abd1fa8c
Stream: the "proxy_protocol" parameter of the "listen" directive.
Dmitry Volyntsev <xeioex@nginx.com>
parents:
6607
diff
changeset
|
580 addrs[i].conf.proxy_protocol = addr[i].opt.proxy_protocol; |
| 6115 | 581 |
|
6560
c90cf79d0e1d
Renamed "u" to "sockaddr" in listen options types.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6436
diff
changeset
|
582 len = ngx_sock_ntop(&addr[i].opt.sockaddr.sockaddr, addr[i].opt.socklen, |
|
c90cf79d0e1d
Renamed "u" to "sockaddr" in listen options types.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6436
diff
changeset
|
583 buf, NGX_SOCKADDR_STRLEN, 1); |
| 6115 | 584 |
| 585 p = ngx_pnalloc(cf->pool, len); | |
| 586 if (p == NULL) { | |
| 587 return NGX_ERROR; | |
| 588 } | |
| 589 | |
| 590 ngx_memcpy(p, buf, len); | |
| 591 | |
| 592 addrs[i].conf.addr_text.len = len; | |
| 593 addrs[i].conf.addr_text.data = p; | |
| 594 } | |
| 595 | |
| 596 return NGX_OK; | |
| 597 } | |
| 598 | |
| 599 | |
| 600 #if (NGX_HAVE_INET6) | |
| 601 | |
| 602 static ngx_int_t | |
| 603 ngx_stream_add_addrs6(ngx_conf_t *cf, ngx_stream_port_t *stport, | |
| 604 ngx_stream_conf_addr_t *addr) | |
| 605 { | |
| 606 u_char *p; | |
| 607 size_t len; | |
| 608 ngx_uint_t i; | |
| 609 struct sockaddr_in6 *sin6; | |
| 610 ngx_stream_in6_addr_t *addrs6; | |
| 611 u_char buf[NGX_SOCKADDR_STRLEN]; | |
| 612 | |
| 613 stport->addrs = ngx_pcalloc(cf->pool, | |
| 614 stport->naddrs * sizeof(ngx_stream_in6_addr_t)); | |
| 615 if (stport->addrs == NULL) { | |
| 616 return NGX_ERROR; | |
| 617 } | |
| 618 | |
| 619 addrs6 = stport->addrs; | |
| 620 | |
| 621 for (i = 0; i < stport->naddrs; i++) { | |
| 622 | |
|
6560
c90cf79d0e1d
Renamed "u" to "sockaddr" in listen options types.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6436
diff
changeset
|
623 sin6 = &addr[i].opt.sockaddr.sockaddr_in6; |
| 6115 | 624 addrs6[i].addr6 = sin6->sin6_addr; |
| 625 | |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
626 addrs6[i].conf.ctx = addr[i].opt.ctx; |
| 6115 | 627 #if (NGX_STREAM_SSL) |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
628 addrs6[i].conf.ssl = addr[i].opt.ssl; |
| 6115 | 629 #endif |
|
6680
7357abd1fa8c
Stream: the "proxy_protocol" parameter of the "listen" directive.
Dmitry Volyntsev <xeioex@nginx.com>
parents:
6607
diff
changeset
|
630 addrs6[i].conf.proxy_protocol = addr[i].opt.proxy_protocol; |
| 6115 | 631 |
|
6560
c90cf79d0e1d
Renamed "u" to "sockaddr" in listen options types.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6436
diff
changeset
|
632 len = ngx_sock_ntop(&addr[i].opt.sockaddr.sockaddr, addr[i].opt.socklen, |
|
c90cf79d0e1d
Renamed "u" to "sockaddr" in listen options types.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6436
diff
changeset
|
633 buf, NGX_SOCKADDR_STRLEN, 1); |
| 6115 | 634 |
| 635 p = ngx_pnalloc(cf->pool, len); | |
| 636 if (p == NULL) { | |
| 637 return NGX_ERROR; | |
| 638 } | |
| 639 | |
| 640 ngx_memcpy(p, buf, len); | |
| 641 | |
| 642 addrs6[i].conf.addr_text.len = len; | |
| 643 addrs6[i].conf.addr_text.data = p; | |
| 644 } | |
| 645 | |
| 646 return NGX_OK; | |
| 647 } | |
| 648 | |
| 649 #endif | |
| 650 | |
| 651 | |
| 652 static ngx_int_t | |
| 653 ngx_stream_cmp_conf_addrs(const void *one, const void *two) | |
| 654 { | |
| 655 ngx_stream_conf_addr_t *first, *second; | |
| 656 | |
| 657 first = (ngx_stream_conf_addr_t *) one; | |
| 658 second = (ngx_stream_conf_addr_t *) two; | |
| 659 | |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
660 if (first->opt.wildcard) { |
| 6115 | 661 /* a wildcard must be the last resort, shift it to the end */ |
| 662 return 1; | |
| 663 } | |
| 664 | |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
665 if (second->opt.wildcard) { |
| 6115 | 666 /* a wildcard must be the last resort, shift it to the end */ |
| 667 return -1; | |
| 668 } | |
| 669 | |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
670 if (first->opt.bind && !second->opt.bind) { |
| 6115 | 671 /* shift explicit bind()ed addresses to the start */ |
| 672 return -1; | |
| 673 } | |
| 674 | |
|
6170
c13091e6292c
Stream: embed ngx_stream_listen_t into ngx_stream_conf_addr_t.
Ruslan Ermilov <ru@nginx.com>
parents:
6169
diff
changeset
|
675 if (!first->opt.bind && second->opt.bind) { |
| 6115 | 676 /* shift explicit bind()ed addresses to the start */ |
| 677 return 1; | |
| 678 } | |
| 679 | |
| 680 /* do not sort by default */ | |
| 681 | |
| 682 return 0; | |
| 683 } |