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 } |