Mercurial > hg > nginx
annotate src/stream/ngx_stream_upstream.c @ 6675:ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Keeps the full address of the upstream server. If several servers were
contacted during proxying, their addresses are separated by commas,
e.g. "192.168.1.1:80, 192.168.1.2:80".
| author | Vladimir Homutov <vl@nginx.com> |
|---|---|
| date | Fri, 02 Sep 2016 18:27:05 +0300 |
| parents | 8ed51b02f655 |
| children | df3a7c029dec |
| rev | line source |
|---|---|
| 6115 | 1 |
| 2 /* | |
| 3 * Copyright (C) Igor Sysoev | |
| 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 | |
|
6675
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
13 static ngx_int_t ngx_stream_upstream_add_variables(ngx_conf_t *cf); |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
14 static ngx_int_t ngx_stream_upstream_addr_variable(ngx_stream_session_t *s, |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
15 ngx_stream_variable_value_t *v, uintptr_t data); |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
16 |
| 6115 | 17 static char *ngx_stream_upstream(ngx_conf_t *cf, ngx_command_t *cmd, |
| 18 void *dummy); | |
| 19 static char *ngx_stream_upstream_server(ngx_conf_t *cf, ngx_command_t *cmd, | |
| 20 void *conf); | |
| 21 static void *ngx_stream_upstream_create_main_conf(ngx_conf_t *cf); | |
| 22 static char *ngx_stream_upstream_init_main_conf(ngx_conf_t *cf, void *conf); | |
| 23 | |
| 24 | |
| 25 static ngx_command_t ngx_stream_upstream_commands[] = { | |
| 26 | |
| 27 { ngx_string("upstream"), | |
| 28 NGX_STREAM_MAIN_CONF|NGX_CONF_BLOCK|NGX_CONF_TAKE1, | |
| 29 ngx_stream_upstream, | |
| 30 0, | |
| 31 0, | |
| 32 NULL }, | |
| 33 | |
| 34 { ngx_string("server"), | |
| 35 NGX_STREAM_UPS_CONF|NGX_CONF_1MORE, | |
| 36 ngx_stream_upstream_server, | |
| 37 NGX_STREAM_SRV_CONF_OFFSET, | |
| 38 0, | |
| 39 NULL }, | |
| 40 | |
| 41 ngx_null_command | |
| 42 }; | |
| 43 | |
| 44 | |
| 45 static ngx_stream_module_t ngx_stream_upstream_module_ctx = { | |
|
6675
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
46 ngx_stream_upstream_add_variables, /* preconfiguration */ |
|
6174
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6115
diff
changeset
|
47 NULL, /* postconfiguration */ |
|
68c106e6fa0a
Stream: added postconfiguration method to stream modules.
Vladimir Homutov <vl@nginx.com>
parents:
6115
diff
changeset
|
48 |
| 6115 | 49 ngx_stream_upstream_create_main_conf, /* create main configuration */ |
| 50 ngx_stream_upstream_init_main_conf, /* init main configuration */ | |
| 51 | |
| 52 NULL, /* create server configuration */ | |
| 6629 | 53 NULL /* merge server configuration */ |
| 6115 | 54 }; |
| 55 | |
| 56 | |
| 57 ngx_module_t ngx_stream_upstream_module = { | |
| 58 NGX_MODULE_V1, | |
| 59 &ngx_stream_upstream_module_ctx, /* module context */ | |
| 60 ngx_stream_upstream_commands, /* module directives */ | |
| 61 NGX_STREAM_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 | |
|
6675
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
73 static ngx_stream_variable_t ngx_stream_upstream_vars[] = { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
74 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
75 { ngx_string("upstream_addr"), NULL, |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
76 ngx_stream_upstream_addr_variable, 0, |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
77 NGX_STREAM_VAR_NOCACHEABLE, 0 }, |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
78 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
79 { ngx_null_string, NULL, NULL, 0, 0, 0 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
80 }; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
81 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
82 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
83 static ngx_int_t |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
84 ngx_stream_upstream_add_variables(ngx_conf_t *cf) |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
85 { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
86 ngx_stream_variable_t *var, *v; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
87 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
88 for (v = ngx_stream_upstream_vars; v->name.len; v++) { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
89 var = ngx_stream_add_variable(cf, &v->name, v->flags); |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
90 if (var == NULL) { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
91 return NGX_ERROR; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
92 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
93 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
94 var->get_handler = v->get_handler; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
95 var->data = v->data; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
96 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
97 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
98 return NGX_OK; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
99 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
100 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
101 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
102 static ngx_int_t |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
103 ngx_stream_upstream_addr_variable(ngx_stream_session_t *s, |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
104 ngx_stream_variable_value_t *v, uintptr_t data) |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
105 { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
106 u_char *p; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
107 size_t len; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
108 ngx_uint_t i; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
109 ngx_stream_upstream_state_t *state; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
110 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
111 v->valid = 1; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
112 v->no_cacheable = 0; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
113 v->not_found = 0; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
114 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
115 if (s->upstream_states == NULL || s->upstream_states->nelts == 0) { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
116 v->not_found = 1; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
117 return NGX_OK; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
118 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
119 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
120 len = 0; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
121 state = s->upstream_states->elts; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
122 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
123 for (i = 0; i < s->upstream_states->nelts; i++) { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
124 if (state[i].peer) { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
125 len += state[i].peer->len; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
126 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
127 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
128 len += 2; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
129 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
130 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
131 p = ngx_pnalloc(s->connection->pool, len); |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
132 if (p == NULL) { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
133 return NGX_ERROR; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
134 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
135 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
136 v->data = p; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
137 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
138 i = 0; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
139 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
140 for ( ;; ) { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
141 if (state[i].peer) { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
142 p = ngx_cpymem(p, state[i].peer->data, state[i].peer->len); |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
143 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
144 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
145 if (++i == s->upstream_states->nelts) { |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
146 break; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
147 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
148 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
149 *p++ = ','; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
150 *p++ = ' '; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
151 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
152 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
153 v->len = p - v->data; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
154 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
155 return NGX_OK; |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
156 } |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
157 |
|
ab9b4fd8c5b7
Stream: the $upstream_addr variable.
Vladimir Homutov <vl@nginx.com>
parents:
6629
diff
changeset
|
158 |
| 6115 | 159 static char * |
| 160 ngx_stream_upstream(ngx_conf_t *cf, ngx_command_t *cmd, void *dummy) | |
| 161 { | |
| 162 char *rv; | |
| 163 void *mconf; | |
| 164 ngx_str_t *value; | |
| 165 ngx_url_t u; | |
| 166 ngx_uint_t m; | |
| 167 ngx_conf_t pcf; | |
| 168 ngx_stream_module_t *module; | |
| 169 ngx_stream_conf_ctx_t *ctx, *stream_ctx; | |
| 170 ngx_stream_upstream_srv_conf_t *uscf; | |
| 171 | |
| 172 ngx_memzero(&u, sizeof(ngx_url_t)); | |
| 173 | |
| 174 value = cf->args->elts; | |
| 175 u.host = value[1]; | |
| 176 u.no_resolve = 1; | |
| 177 u.no_port = 1; | |
| 178 | |
| 179 uscf = ngx_stream_upstream_add(cf, &u, NGX_STREAM_UPSTREAM_CREATE | |
| 180 |NGX_STREAM_UPSTREAM_WEIGHT | |
| 181 |NGX_STREAM_UPSTREAM_MAX_FAILS | |
| 182 |NGX_STREAM_UPSTREAM_FAIL_TIMEOUT | |
| 183 |NGX_STREAM_UPSTREAM_DOWN | |
| 184 |NGX_STREAM_UPSTREAM_BACKUP); | |
| 185 if (uscf == NULL) { | |
| 186 return NGX_CONF_ERROR; | |
| 187 } | |
| 188 | |
| 189 | |
| 190 ctx = ngx_pcalloc(cf->pool, sizeof(ngx_stream_conf_ctx_t)); | |
| 191 if (ctx == NULL) { | |
| 192 return NGX_CONF_ERROR; | |
| 193 } | |
| 194 | |
| 195 stream_ctx = cf->ctx; | |
| 196 ctx->main_conf = stream_ctx->main_conf; | |
| 197 | |
| 198 /* the upstream{}'s srv_conf */ | |
| 199 | |
| 200 ctx->srv_conf = ngx_pcalloc(cf->pool, | |
| 201 sizeof(void *) * ngx_stream_max_module); | |
| 202 if (ctx->srv_conf == NULL) { | |
| 203 return NGX_CONF_ERROR; | |
| 204 } | |
| 205 | |
| 206 ctx->srv_conf[ngx_stream_upstream_module.ctx_index] = uscf; | |
| 207 | |
| 208 uscf->srv_conf = ctx->srv_conf; | |
| 209 | |
|
6379
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6174
diff
changeset
|
210 for (m = 0; cf->cycle->modules[m]; m++) { |
|
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6174
diff
changeset
|
211 if (cf->cycle->modules[m]->type != NGX_STREAM_MODULE) { |
| 6115 | 212 continue; |
| 213 } | |
| 214 | |
|
6379
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6174
diff
changeset
|
215 module = cf->cycle->modules[m]->ctx; |
| 6115 | 216 |
| 217 if (module->create_srv_conf) { | |
| 218 mconf = module->create_srv_conf(cf); | |
| 219 if (mconf == NULL) { | |
| 220 return NGX_CONF_ERROR; | |
| 221 } | |
| 222 | |
|
6379
cf5e822cf470
Dynamic modules: changed ngx_modules to cycle->modules.
Maxim Dounin <mdounin@mdounin.ru>
parents:
6174
diff
changeset
|
223 ctx->srv_conf[cf->cycle->modules[m]->ctx_index] = mconf; |
| 6115 | 224 } |
| 225 } | |
| 226 | |
| 227 uscf->servers = ngx_array_create(cf->pool, 4, | |
| 228 sizeof(ngx_stream_upstream_server_t)); | |
| 229 if (uscf->servers == NULL) { | |
| 230 return NGX_CONF_ERROR; | |
| 231 } | |
| 232 | |
| 233 | |
| 234 /* parse inside upstream{} */ | |
| 235 | |
| 236 pcf = *cf; | |
| 237 cf->ctx = ctx; | |
| 238 cf->cmd_type = NGX_STREAM_UPS_CONF; | |
| 239 | |
| 240 rv = ngx_conf_parse(cf, NULL); | |
| 241 | |
| 242 *cf = pcf; | |
| 243 | |
| 244 if (rv != NGX_CONF_OK) { | |
| 245 return rv; | |
| 246 } | |
| 247 | |
| 248 if (uscf->servers->nelts == 0) { | |
| 249 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 250 "no servers are inside upstream"); | |
| 251 return NGX_CONF_ERROR; | |
| 252 } | |
| 253 | |
| 254 return rv; | |
| 255 } | |
| 256 | |
| 257 | |
| 258 static char * | |
| 259 ngx_stream_upstream_server(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
| 260 { | |
| 261 ngx_stream_upstream_srv_conf_t *uscf = conf; | |
| 262 | |
| 263 time_t fail_timeout; | |
| 264 ngx_str_t *value, s; | |
| 265 ngx_url_t u; | |
| 266 ngx_int_t weight, max_fails; | |
| 267 ngx_uint_t i; | |
| 268 ngx_stream_upstream_server_t *us; | |
| 269 | |
| 270 us = ngx_array_push(uscf->servers); | |
| 271 if (us == NULL) { | |
| 272 return NGX_CONF_ERROR; | |
| 273 } | |
| 274 | |
| 275 ngx_memzero(us, sizeof(ngx_stream_upstream_server_t)); | |
| 276 | |
| 277 value = cf->args->elts; | |
| 278 | |
| 279 weight = 1; | |
| 280 max_fails = 1; | |
| 281 fail_timeout = 10; | |
| 282 | |
| 283 for (i = 2; i < cf->args->nelts; i++) { | |
| 284 | |
| 285 if (ngx_strncmp(value[i].data, "weight=", 7) == 0) { | |
| 286 | |
| 287 if (!(uscf->flags & NGX_STREAM_UPSTREAM_WEIGHT)) { | |
| 288 goto not_supported; | |
| 289 } | |
| 290 | |
| 291 weight = ngx_atoi(&value[i].data[7], value[i].len - 7); | |
| 292 | |
| 293 if (weight == NGX_ERROR || weight == 0) { | |
| 294 goto invalid; | |
| 295 } | |
| 296 | |
| 297 continue; | |
| 298 } | |
| 299 | |
| 300 if (ngx_strncmp(value[i].data, "max_fails=", 10) == 0) { | |
| 301 | |
| 302 if (!(uscf->flags & NGX_STREAM_UPSTREAM_MAX_FAILS)) { | |
| 303 goto not_supported; | |
| 304 } | |
| 305 | |
| 306 max_fails = ngx_atoi(&value[i].data[10], value[i].len - 10); | |
| 307 | |
| 308 if (max_fails == NGX_ERROR) { | |
| 309 goto invalid; | |
| 310 } | |
| 311 | |
| 312 continue; | |
| 313 } | |
| 314 | |
| 315 if (ngx_strncmp(value[i].data, "fail_timeout=", 13) == 0) { | |
| 316 | |
| 317 if (!(uscf->flags & NGX_STREAM_UPSTREAM_FAIL_TIMEOUT)) { | |
| 318 goto not_supported; | |
| 319 } | |
| 320 | |
| 321 s.len = value[i].len - 13; | |
| 322 s.data = &value[i].data[13]; | |
| 323 | |
| 324 fail_timeout = ngx_parse_time(&s, 1); | |
| 325 | |
| 326 if (fail_timeout == (time_t) NGX_ERROR) { | |
| 327 goto invalid; | |
| 328 } | |
| 329 | |
| 330 continue; | |
| 331 } | |
| 332 | |
| 333 if (ngx_strcmp(value[i].data, "backup") == 0) { | |
| 334 | |
| 335 if (!(uscf->flags & NGX_STREAM_UPSTREAM_BACKUP)) { | |
| 336 goto not_supported; | |
| 337 } | |
| 338 | |
| 339 us->backup = 1; | |
| 340 | |
| 341 continue; | |
| 342 } | |
| 343 | |
| 344 if (ngx_strcmp(value[i].data, "down") == 0) { | |
| 345 | |
| 346 if (!(uscf->flags & NGX_STREAM_UPSTREAM_DOWN)) { | |
| 347 goto not_supported; | |
| 348 } | |
| 349 | |
| 350 us->down = 1; | |
| 351 | |
| 352 continue; | |
| 353 } | |
| 354 | |
| 355 goto invalid; | |
| 356 } | |
| 357 | |
| 358 ngx_memzero(&u, sizeof(ngx_url_t)); | |
| 359 | |
| 360 u.url = value[1]; | |
| 361 | |
| 362 if (ngx_parse_url(cf->pool, &u) != NGX_OK) { | |
| 363 if (u.err) { | |
| 364 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 365 "%s in upstream \"%V\"", u.err, &u.url); | |
| 366 } | |
| 367 | |
| 368 return NGX_CONF_ERROR; | |
| 369 } | |
| 370 | |
| 371 if (u.no_port) { | |
| 372 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 373 "no port in upstream \"%V\"", &u.url); | |
| 374 return NGX_CONF_ERROR; | |
| 375 } | |
| 376 | |
| 377 us->name = u.url; | |
| 378 us->addrs = u.addrs; | |
| 379 us->naddrs = u.naddrs; | |
| 380 us->weight = weight; | |
| 381 us->max_fails = max_fails; | |
| 382 us->fail_timeout = fail_timeout; | |
| 383 | |
| 384 return NGX_CONF_OK; | |
| 385 | |
| 386 invalid: | |
| 387 | |
| 388 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 389 "invalid parameter \"%V\"", &value[i]); | |
| 390 | |
| 391 return NGX_CONF_ERROR; | |
| 392 | |
| 393 not_supported: | |
| 394 | |
| 395 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 396 "balancing method does not support parameter \"%V\"", | |
| 397 &value[i]); | |
| 398 | |
| 399 return NGX_CONF_ERROR; | |
| 400 } | |
| 401 | |
| 402 | |
| 403 ngx_stream_upstream_srv_conf_t * | |
| 404 ngx_stream_upstream_add(ngx_conf_t *cf, ngx_url_t *u, ngx_uint_t flags) | |
| 405 { | |
| 406 ngx_uint_t i; | |
| 407 ngx_stream_upstream_server_t *us; | |
| 408 ngx_stream_upstream_srv_conf_t *uscf, **uscfp; | |
| 409 ngx_stream_upstream_main_conf_t *umcf; | |
| 410 | |
| 411 if (!(flags & NGX_STREAM_UPSTREAM_CREATE)) { | |
| 412 | |
| 413 if (ngx_parse_url(cf->pool, u) != NGX_OK) { | |
| 414 if (u->err) { | |
| 415 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 416 "%s in upstream \"%V\"", u->err, &u->url); | |
| 417 } | |
| 418 | |
| 419 return NULL; | |
| 420 } | |
| 421 } | |
| 422 | |
| 423 umcf = ngx_stream_conf_get_module_main_conf(cf, ngx_stream_upstream_module); | |
| 424 | |
| 425 uscfp = umcf->upstreams.elts; | |
| 426 | |
| 427 for (i = 0; i < umcf->upstreams.nelts; i++) { | |
| 428 | |
| 429 if (uscfp[i]->host.len != u->host.len | |
| 430 || ngx_strncasecmp(uscfp[i]->host.data, u->host.data, u->host.len) | |
| 431 != 0) | |
| 432 { | |
| 433 continue; | |
| 434 } | |
| 435 | |
| 436 if ((flags & NGX_STREAM_UPSTREAM_CREATE) | |
| 437 && (uscfp[i]->flags & NGX_STREAM_UPSTREAM_CREATE)) | |
| 438 { | |
| 439 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 440 "duplicate upstream \"%V\"", &u->host); | |
| 441 return NULL; | |
| 442 } | |
| 443 | |
| 444 if ((uscfp[i]->flags & NGX_STREAM_UPSTREAM_CREATE) && !u->no_port) { | |
| 445 ngx_conf_log_error(NGX_LOG_WARN, cf, 0, | |
| 446 "upstream \"%V\" may not have port %d", | |
| 447 &u->host, u->port); | |
| 448 return NULL; | |
| 449 } | |
| 450 | |
| 451 if ((flags & NGX_STREAM_UPSTREAM_CREATE) && !uscfp[i]->no_port) { | |
| 452 ngx_log_error(NGX_LOG_WARN, cf->log, 0, | |
| 453 "upstream \"%V\" may not have port %d in %s:%ui", | |
| 454 &u->host, uscfp[i]->port, | |
| 455 uscfp[i]->file_name, uscfp[i]->line); | |
| 456 return NULL; | |
| 457 } | |
| 458 | |
| 459 if (uscfp[i]->port != u->port) { | |
| 460 continue; | |
| 461 } | |
| 462 | |
| 463 if (flags & NGX_STREAM_UPSTREAM_CREATE) { | |
| 464 uscfp[i]->flags = flags; | |
| 465 } | |
| 466 | |
| 467 return uscfp[i]; | |
| 468 } | |
| 469 | |
| 470 uscf = ngx_pcalloc(cf->pool, sizeof(ngx_stream_upstream_srv_conf_t)); | |
| 471 if (uscf == NULL) { | |
| 472 return NULL; | |
| 473 } | |
| 474 | |
| 475 uscf->flags = flags; | |
| 476 uscf->host = u->host; | |
| 477 uscf->file_name = cf->conf_file->file.name.data; | |
| 478 uscf->line = cf->conf_file->line; | |
| 479 uscf->port = u->port; | |
| 480 uscf->no_port = u->no_port; | |
| 481 | |
|
6459
78fc2dce69e7
Stream: detect port absence in proxy_pass with IP literal.
Roman Arutyunyan <arut@nginx.com>
parents:
6379
diff
changeset
|
482 if (u->naddrs == 1 && (u->port || u->family == AF_UNIX)) { |
| 6115 | 483 uscf->servers = ngx_array_create(cf->pool, 1, |
| 484 sizeof(ngx_stream_upstream_server_t)); | |
| 485 if (uscf->servers == NULL) { | |
| 486 return NULL; | |
| 487 } | |
| 488 | |
| 489 us = ngx_array_push(uscf->servers); | |
| 490 if (us == NULL) { | |
| 491 return NULL; | |
| 492 } | |
| 493 | |
| 494 ngx_memzero(us, sizeof(ngx_stream_upstream_server_t)); | |
| 495 | |
| 496 us->addrs = u->addrs; | |
| 497 us->naddrs = 1; | |
| 498 } | |
| 499 | |
| 500 uscfp = ngx_array_push(&umcf->upstreams); | |
| 501 if (uscfp == NULL) { | |
| 502 return NULL; | |
| 503 } | |
| 504 | |
| 505 *uscfp = uscf; | |
| 506 | |
| 507 return uscf; | |
| 508 } | |
| 509 | |
| 510 | |
| 511 static void * | |
| 512 ngx_stream_upstream_create_main_conf(ngx_conf_t *cf) | |
| 513 { | |
| 514 ngx_stream_upstream_main_conf_t *umcf; | |
| 515 | |
| 516 umcf = ngx_pcalloc(cf->pool, sizeof(ngx_stream_upstream_main_conf_t)); | |
| 517 if (umcf == NULL) { | |
| 518 return NULL; | |
| 519 } | |
| 520 | |
| 521 if (ngx_array_init(&umcf->upstreams, cf->pool, 4, | |
| 522 sizeof(ngx_stream_upstream_srv_conf_t *)) | |
| 523 != NGX_OK) | |
| 524 { | |
| 525 return NULL; | |
| 526 } | |
| 527 | |
| 528 return umcf; | |
| 529 } | |
| 530 | |
| 531 | |
| 532 static char * | |
| 533 ngx_stream_upstream_init_main_conf(ngx_conf_t *cf, void *conf) | |
| 534 { | |
| 535 ngx_stream_upstream_main_conf_t *umcf = conf; | |
| 536 | |
| 537 ngx_uint_t i; | |
| 538 ngx_stream_upstream_init_pt init; | |
| 539 ngx_stream_upstream_srv_conf_t **uscfp; | |
| 540 | |
| 541 uscfp = umcf->upstreams.elts; | |
| 542 | |
| 543 for (i = 0; i < umcf->upstreams.nelts; i++) { | |
| 544 | |
| 545 init = uscfp[i]->peer.init_upstream | |
| 546 ? uscfp[i]->peer.init_upstream | |
| 547 : ngx_stream_upstream_init_round_robin; | |
| 548 | |
| 549 if (init(cf, uscfp[i]) != NGX_OK) { | |
| 550 return NGX_CONF_ERROR; | |
| 551 } | |
| 552 } | |
| 553 | |
| 554 return NGX_CONF_OK; | |
| 555 } |