50
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
|
4 */
|
|
5
|
|
6
|
|
7 #include <ngx_config.h>
|
|
8 #include <ngx_core.h>
|
|
9 #include <ngx_http.h>
|
|
10
|
|
11
|
|
12 typedef struct {
|
126
|
13 ngx_array_t *codes; /* uintptr_t */
|
50
|
14
|
126
|
15 ngx_uint_t stack_size;
|
50
|
16
|
126
|
17 ngx_flag_t log;
|
184
|
18 ngx_flag_t uninitialized_variable_warn;
|
50
|
19 } ngx_http_rewrite_loc_conf_t;
|
|
20
|
|
21
|
|
22 static void *ngx_http_rewrite_create_loc_conf(ngx_conf_t *cf);
|
|
23 static char *ngx_http_rewrite_merge_loc_conf(ngx_conf_t *cf,
|
|
24 void *parent, void *child);
|
230
|
25 static ngx_int_t ngx_http_rewrite_init(ngx_conf_t *cf);
|
50
|
26 static char *ngx_http_rewrite(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
|
|
27 static char *ngx_http_rewrite_return(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
28 void *conf);
|
76
|
29 static char *ngx_http_rewrite_break(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
30 void *conf);
|
50
|
31 static char *ngx_http_rewrite_if(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
32 void *conf);
|
|
33 static char * ngx_http_rewrite_if_condition(ngx_conf_t *cf,
|
|
34 ngx_http_rewrite_loc_conf_t *lcf);
|
|
35 static char *ngx_http_rewrite_variable(ngx_conf_t *cf,
|
|
36 ngx_http_rewrite_loc_conf_t *lcf, ngx_str_t *value);
|
58
|
37 static char *ngx_http_rewrite_set(ngx_conf_t *cf, ngx_command_t *cmd,
|
50
|
38 void *conf);
|
126
|
39 static char * ngx_http_rewrite_value(ngx_conf_t *cf,
|
|
40 ngx_http_rewrite_loc_conf_t *lcf, ngx_str_t *value);
|
50
|
41
|
|
42
|
|
43 static ngx_command_t ngx_http_rewrite_commands[] = {
|
|
44
|
|
45 { ngx_string("rewrite"),
|
|
46 NGX_HTTP_SRV_CONF|NGX_HTTP_SIF_CONF|NGX_HTTP_LOC_CONF|NGX_HTTP_LIF_CONF
|
|
47 |NGX_CONF_TAKE23,
|
|
48 ngx_http_rewrite,
|
|
49 NGX_HTTP_LOC_CONF_OFFSET,
|
|
50 0,
|
|
51 NULL },
|
|
52
|
|
53 { ngx_string("return"),
|
|
54 NGX_HTTP_SRV_CONF|NGX_HTTP_SIF_CONF|NGX_HTTP_LOC_CONF|NGX_HTTP_LIF_CONF
|
623
|
55 |NGX_CONF_TAKE12,
|
50
|
56 ngx_http_rewrite_return,
|
|
57 NGX_HTTP_LOC_CONF_OFFSET,
|
|
58 0,
|
|
59 NULL },
|
|
60
|
76
|
61 { ngx_string("break"),
|
|
62 NGX_HTTP_SRV_CONF|NGX_HTTP_SIF_CONF|NGX_HTTP_LOC_CONF|NGX_HTTP_LIF_CONF
|
|
63 |NGX_CONF_NOARGS,
|
|
64 ngx_http_rewrite_break,
|
|
65 NGX_HTTP_LOC_CONF_OFFSET,
|
|
66 0,
|
|
67 NULL },
|
|
68
|
50
|
69 { ngx_string("if"),
|
|
70 NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_BLOCK|NGX_CONF_1MORE,
|
|
71 ngx_http_rewrite_if,
|
|
72 NGX_HTTP_LOC_CONF_OFFSET,
|
|
73 0,
|
|
74 NULL },
|
|
75
|
|
76 { ngx_string("set"),
|
|
77 NGX_HTTP_SRV_CONF|NGX_HTTP_SIF_CONF|NGX_HTTP_LOC_CONF|NGX_HTTP_LIF_CONF
|
|
78 |NGX_CONF_TAKE2,
|
|
79 ngx_http_rewrite_set,
|
|
80 NGX_HTTP_LOC_CONF_OFFSET,
|
|
81 0,
|
|
82 NULL },
|
|
83
|
|
84 { ngx_string("rewrite_log"),
|
184
|
85 NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_SIF_CONF|NGX_HTTP_LOC_CONF
|
|
86 |NGX_HTTP_LIF_CONF|NGX_CONF_FLAG,
|
50
|
87 ngx_conf_set_flag_slot,
|
|
88 NGX_HTTP_LOC_CONF_OFFSET,
|
|
89 offsetof(ngx_http_rewrite_loc_conf_t, log),
|
|
90 NULL },
|
|
91
|
184
|
92 { ngx_string("uninitialized_variable_warn"),
|
|
93 NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_SIF_CONF|NGX_HTTP_LOC_CONF
|
|
94 |NGX_HTTP_LIF_CONF|NGX_CONF_FLAG,
|
|
95 ngx_conf_set_flag_slot,
|
|
96 NGX_HTTP_LOC_CONF_OFFSET,
|
|
97 offsetof(ngx_http_rewrite_loc_conf_t, uninitialized_variable_warn),
|
|
98 NULL },
|
|
99
|
50
|
100 ngx_null_command
|
|
101 };
|
|
102
|
|
103
|
216
|
104 static ngx_http_module_t ngx_http_rewrite_module_ctx = {
|
58
|
105 NULL, /* preconfiguration */
|
230
|
106 ngx_http_rewrite_init, /* postconfiguration */
|
50
|
107
|
|
108 NULL, /* create main configuration */
|
|
109 NULL, /* init main configuration */
|
|
110
|
|
111 NULL, /* create server configuration */
|
|
112 NULL, /* merge server configuration */
|
|
113
|
|
114 ngx_http_rewrite_create_loc_conf, /* create location configration */
|
|
115 ngx_http_rewrite_merge_loc_conf /* merge location configration */
|
|
116 };
|
|
117
|
|
118
|
|
119 ngx_module_t ngx_http_rewrite_module = {
|
58
|
120 NGX_MODULE_V1,
|
126
|
121 &ngx_http_rewrite_module_ctx, /* module context */
|
50
|
122 ngx_http_rewrite_commands, /* module directives */
|
|
123 NGX_HTTP_MODULE, /* module type */
|
90
|
124 NULL, /* init master */
|
230
|
125 NULL, /* init module */
|
90
|
126 NULL, /* init process */
|
|
127 NULL, /* init thread */
|
|
128 NULL, /* exit thread */
|
|
129 NULL, /* exit process */
|
|
130 NULL, /* exit master */
|
|
131 NGX_MODULE_V1_PADDING
|
50
|
132 };
|
|
133
|
|
134
|
|
135 static ngx_int_t
|
|
136 ngx_http_rewrite_handler(ngx_http_request_t *r)
|
|
137 {
|
58
|
138 ngx_http_script_code_pt code;
|
|
139 ngx_http_script_engine_t *e;
|
184
|
140 ngx_http_rewrite_loc_conf_t *rlcf;
|
50
|
141
|
184
|
142 rlcf = ngx_http_get_module_loc_conf(r, ngx_http_rewrite_module);
|
50
|
143
|
184
|
144 if (rlcf->codes == NULL) {
|
50
|
145 return NGX_DECLINED;
|
|
146 }
|
|
147
|
58
|
148 e = ngx_pcalloc(r->pool, sizeof(ngx_http_script_engine_t));
|
50
|
149 if (e == NULL) {
|
|
150 return NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
151 }
|
|
152
|
58
|
153 e->sp = ngx_pcalloc(r->pool,
|
184
|
154 rlcf->stack_size * sizeof(ngx_http_variable_value_t));
|
50
|
155 if (e->sp == NULL) {
|
|
156 return NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
157 }
|
|
158
|
184
|
159 e->ip = rlcf->codes->elts;
|
58
|
160 e->request = r;
|
50
|
161 e->quote = 1;
|
184
|
162 e->log = rlcf->log;
|
50
|
163 e->status = NGX_DECLINED;
|
|
164
|
|
165 while (*(uintptr_t *) e->ip) {
|
58
|
166 code = *(ngx_http_script_code_pt *) e->ip;
|
50
|
167 code(e);
|
|
168 }
|
|
169
|
242
|
170 if (e->status == NGX_DECLINED) {
|
|
171 return NGX_DECLINED;
|
|
172 }
|
|
173
|
238
|
174 if (r->err_status == 0) {
|
|
175 return e->status;
|
|
176 }
|
|
177
|
|
178 return r->err_status;
|
50
|
179 }
|
|
180
|
|
181
|
122
|
182 static ngx_int_t
|
|
183 ngx_http_rewrite_var(ngx_http_request_t *r, ngx_http_variable_value_t *v,
|
|
184 uintptr_t data)
|
50
|
185 {
|
184
|
186 ngx_http_variable_t *var;
|
|
187 ngx_http_core_main_conf_t *cmcf;
|
|
188 ngx_http_rewrite_loc_conf_t *rlcf;
|
|
189
|
|
190 rlcf = ngx_http_get_module_loc_conf(r, ngx_http_rewrite_module);
|
|
191
|
|
192 if (rlcf->uninitialized_variable_warn == 0) {
|
|
193 *v = ngx_http_variable_null_value;
|
|
194 return NGX_OK;
|
|
195 }
|
50
|
196
|
|
197 cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module);
|
|
198
|
|
199 var = cmcf->variables.elts;
|
|
200
|
|
201 /*
|
|
202 * the ngx_http_rewrite_module sets variables directly in r->variables,
|
68
|
203 * and they should be handled by ngx_http_get_indexed_variable(),
|
50
|
204 * so the handler is called only if the variable is not initialized
|
|
205 */
|
|
206
|
184
|
207 ngx_log_error(NGX_LOG_WARN, r->connection->log, 0,
|
50
|
208 "using uninitialized \"%V\" variable", &var[data].name);
|
|
209
|
126
|
210 *v = ngx_http_variable_null_value;
|
122
|
211
|
|
212 return NGX_OK;
|
50
|
213 }
|
|
214
|
|
215
|
|
216 static void *
|
|
217 ngx_http_rewrite_create_loc_conf(ngx_conf_t *cf)
|
|
218 {
|
|
219 ngx_http_rewrite_loc_conf_t *conf;
|
|
220
|
|
221 conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_rewrite_loc_conf_t));
|
|
222 if (conf == NULL) {
|
536
|
223 return NULL;
|
50
|
224 }
|
|
225
|
|
226 conf->stack_size = NGX_CONF_UNSET_UINT;
|
|
227 conf->log = NGX_CONF_UNSET;
|
184
|
228 conf->uninitialized_variable_warn = NGX_CONF_UNSET;
|
50
|
229
|
|
230 return conf;
|
|
231 }
|
|
232
|
|
233
|
|
234 static char *
|
|
235 ngx_http_rewrite_merge_loc_conf(ngx_conf_t *cf, void *parent, void *child)
|
|
236 {
|
|
237 ngx_http_rewrite_loc_conf_t *prev = parent;
|
|
238 ngx_http_rewrite_loc_conf_t *conf = child;
|
|
239
|
58
|
240 uintptr_t *code;
|
50
|
241
|
|
242 ngx_conf_merge_value(conf->log, prev->log, 0);
|
184
|
243 ngx_conf_merge_value(conf->uninitialized_variable_warn,
|
|
244 prev->uninitialized_variable_warn, 1);
|
212
|
245 ngx_conf_merge_uint_value(conf->stack_size, prev->stack_size, 10);
|
50
|
246
|
|
247 if (conf->codes == NULL) {
|
|
248 return NGX_CONF_OK;
|
|
249 }
|
|
250
|
|
251 if (conf->codes == prev->codes) {
|
|
252 return NGX_CONF_OK;
|
|
253 }
|
|
254
|
|
255 code = ngx_array_push_n(conf->codes, sizeof(uintptr_t));
|
|
256 if (code == NULL) {
|
|
257 return NGX_CONF_ERROR;
|
|
258 }
|
|
259
|
|
260 *code = (uintptr_t) NULL;
|
|
261
|
|
262 return NGX_CONF_OK;
|
|
263 }
|
|
264
|
|
265
|
58
|
266 static ngx_int_t
|
230
|
267 ngx_http_rewrite_init(ngx_conf_t *cf)
|
126
|
268 {
|
58
|
269 ngx_http_handler_pt *h;
|
|
270 ngx_http_core_main_conf_t *cmcf;
|
|
271
|
230
|
272 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
|
58
|
273
|
122
|
274 h = ngx_array_push(&cmcf->phases[NGX_HTTP_SERVER_REWRITE_PHASE].handlers);
|
|
275 if (h == NULL) {
|
|
276 return NGX_ERROR;
|
|
277 }
|
|
278
|
|
279 *h = ngx_http_rewrite_handler;
|
|
280
|
58
|
281 h = ngx_array_push(&cmcf->phases[NGX_HTTP_REWRITE_PHASE].handlers);
|
|
282 if (h == NULL) {
|
|
283 return NGX_ERROR;
|
|
284 }
|
|
285
|
|
286 *h = ngx_http_rewrite_handler;
|
|
287
|
|
288 return NGX_OK;
|
126
|
289 }
|
58
|
290
|
|
291
|
50
|
292 static char *
|
|
293 ngx_http_rewrite(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
294 {
|
126
|
295 ngx_http_rewrite_loc_conf_t *lcf = conf;
|
|
296
|
589
|
297 ngx_str_t *value;
|
58
|
298 ngx_uint_t last;
|
589
|
299 ngx_regex_compile_t rc;
|
58
|
300 ngx_http_script_code_pt *code;
|
|
301 ngx_http_script_compile_t sc;
|
|
302 ngx_http_script_regex_code_t *regex;
|
|
303 ngx_http_script_regex_end_code_t *regex_end;
|
|
304 u_char errstr[NGX_MAX_CONF_ERRSTR];
|
50
|
305
|
58
|
306 regex = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
307 sizeof(ngx_http_script_regex_code_t));
|
50
|
308 if (regex == NULL) {
|
|
309 return NGX_CONF_ERROR;
|
|
310 }
|
|
311
|
58
|
312 ngx_memzero(regex, sizeof(ngx_http_script_regex_code_t));
|
|
313
|
50
|
314 value = cf->args->elts;
|
|
315
|
589
|
316 ngx_memzero(&rc, sizeof(ngx_regex_compile_t));
|
|
317
|
|
318 rc.pattern = value[1];
|
|
319 rc.err.len = NGX_MAX_CONF_ERRSTR;
|
|
320 rc.err.data = errstr;
|
50
|
321
|
|
322 /* TODO: NGX_REGEX_CASELESS */
|
|
323
|
589
|
324 regex->regex = ngx_http_regex_compile(cf, &rc);
|
50
|
325 if (regex->regex == NULL) {
|
|
326 return NGX_CONF_ERROR;
|
|
327 }
|
|
328
|
58
|
329 regex->code = ngx_http_script_regex_start_code;
|
50
|
330 regex->uri = 1;
|
|
331 regex->name = value[1];
|
|
332
|
58
|
333 if (value[2].data[value[2].len - 1] == '?') {
|
|
334
|
|
335 /* the last "?" drops the original arguments */
|
|
336 value[2].len--;
|
|
337
|
|
338 } else {
|
|
339 regex->add_args = 1;
|
|
340 }
|
|
341
|
50
|
342 last = 0;
|
|
343
|
615
|
344 if (ngx_strncmp(value[2].data, "http://", sizeof("http://") - 1) == 0
|
|
345 || ngx_strncmp(value[2].data, "https://", sizeof("https://") - 1) == 0
|
|
346 || ngx_strncmp(value[2].data, "$scheme", sizeof("$scheme") - 1) == 0)
|
|
347 {
|
382
|
348 regex->status = NGX_HTTP_MOVED_TEMPORARILY;
|
|
349 regex->redirect = 1;
|
|
350 last = 1;
|
|
351 }
|
|
352
|
50
|
353 if (cf->args->nelts == 4) {
|
|
354 if (ngx_strcmp(value[3].data, "last") == 0) {
|
|
355 last = 1;
|
|
356
|
|
357 } else if (ngx_strcmp(value[3].data, "break") == 0) {
|
|
358 regex->break_cycle = 1;
|
|
359 last = 1;
|
|
360
|
|
361 } else if (ngx_strcmp(value[3].data, "redirect") == 0) {
|
|
362 regex->status = NGX_HTTP_MOVED_TEMPORARILY;
|
|
363 regex->redirect = 1;
|
|
364 last = 1;
|
|
365
|
|
366 } else if (ngx_strcmp(value[3].data, "permanent") == 0) {
|
|
367 regex->status = NGX_HTTP_MOVED_PERMANENTLY;
|
|
368 regex->redirect = 1;
|
|
369 last = 1;
|
|
370
|
|
371 } else {
|
|
372 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
373 "invalid parameter \"%V\"", &value[3]);
|
|
374 return NGX_CONF_ERROR;
|
|
375 }
|
|
376 }
|
|
377
|
58
|
378 ngx_memzero(&sc, sizeof(ngx_http_script_compile_t));
|
50
|
379
|
58
|
380 sc.cf = cf;
|
|
381 sc.source = &value[2];
|
|
382 sc.lengths = ®ex->lengths;
|
|
383 sc.values = &lcf->codes;
|
|
384 sc.variables = ngx_http_script_variables_count(&value[2]);
|
|
385 sc.main = regex;
|
|
386 sc.complete_lengths = 1;
|
|
387 sc.compile_args = !regex->redirect;
|
50
|
388
|
58
|
389 if (ngx_http_script_compile(&sc) != NGX_OK) {
|
|
390 return NGX_CONF_ERROR;
|
|
391 }
|
50
|
392
|
58
|
393 regex = sc.main;
|
50
|
394
|
58
|
395 regex->size = sc.size;
|
|
396 regex->args = sc.args;
|
50
|
397
|
190
|
398 if (sc.variables == 0 && !sc.dup_capture) {
|
58
|
399 regex->lengths = NULL;
|
50
|
400 }
|
|
401
|
58
|
402 regex_end = ngx_http_script_add_code(lcf->codes,
|
|
403 sizeof(ngx_http_script_regex_end_code_t),
|
|
404 ®ex);
|
50
|
405 if (regex_end == NULL) {
|
|
406 return NGX_CONF_ERROR;
|
|
407 }
|
|
408
|
58
|
409 regex_end->code = ngx_http_script_regex_end_code;
|
50
|
410 regex_end->uri = regex->uri;
|
|
411 regex_end->args = regex->args;
|
58
|
412 regex_end->add_args = regex->add_args;
|
50
|
413 regex_end->redirect = regex->redirect;
|
|
414
|
|
415 if (last) {
|
66
|
416 code = ngx_http_script_add_code(lcf->codes, sizeof(uintptr_t), ®ex);
|
50
|
417 if (code == NULL) {
|
|
418 return NGX_CONF_ERROR;
|
|
419 }
|
|
420
|
575
|
421 *code = NULL;
|
50
|
422 }
|
|
423
|
|
424 regex->next = (u_char *) lcf->codes->elts + lcf->codes->nelts
|
|
425 - (u_char *) regex;
|
|
426
|
|
427 return NGX_CONF_OK;
|
|
428 }
|
|
429
|
|
430
|
|
431 static char *
|
|
432 ngx_http_rewrite_return(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
433 {
|
126
|
434 ngx_http_rewrite_loc_conf_t *lcf = conf;
|
50
|
435
|
623
|
436 u_char *p;
|
|
437 ngx_str_t *value, *v;
|
|
438 ngx_http_script_return_code_t *ret;
|
|
439 ngx_http_compile_complex_value_t ccv;
|
50
|
440
|
58
|
441 ret = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
442 sizeof(ngx_http_script_return_code_t));
|
50
|
443 if (ret == NULL) {
|
|
444 return NGX_CONF_ERROR;
|
|
445 }
|
|
446
|
|
447 value = cf->args->elts;
|
|
448
|
623
|
449 ngx_memzero(ret, sizeof(ngx_http_script_return_code_t));
|
|
450
|
58
|
451 ret->code = ngx_http_script_return_code;
|
50
|
452
|
623
|
453 p = value[1].data;
|
|
454
|
|
455 ret->status = ngx_atoi(p, value[1].len);
|
50
|
456
|
|
457 if (ret->status == (uintptr_t) NGX_ERROR) {
|
623
|
458
|
|
459 if (cf->args->nelts == 2
|
|
460 && (ngx_strncmp(p, "http://", sizeof("http://") - 1) == 0
|
|
461 || ngx_strncmp(p, "https://", sizeof("https://") - 1) == 0
|
|
462 || ngx_strncmp(p, "$scheme", sizeof("$scheme") - 1) == 0))
|
|
463 {
|
|
464 ret->status = NGX_HTTP_MOVED_TEMPORARILY;
|
|
465 v = &value[1];
|
|
466
|
|
467 } else {
|
|
468 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
469 "invalid return code \"%V\"", &value[1]);
|
|
470 return NGX_CONF_ERROR;
|
|
471 }
|
|
472
|
|
473 } else {
|
|
474
|
|
475 if (cf->args->nelts == 2) {
|
|
476 return NGX_CONF_OK;
|
|
477 }
|
|
478
|
|
479 v = &value[2];
|
|
480 }
|
|
481
|
|
482 ngx_memzero(&ccv, sizeof(ngx_http_compile_complex_value_t));
|
|
483
|
|
484 ccv.cf = cf;
|
|
485 ccv.value = v;
|
|
486 ccv.complex_value = &ret->text;
|
|
487
|
|
488 if (ngx_http_compile_complex_value(&ccv) != NGX_OK) {
|
50
|
489 return NGX_CONF_ERROR;
|
|
490 }
|
|
491
|
|
492 return NGX_CONF_OK;
|
|
493 }
|
|
494
|
|
495
|
|
496 static char *
|
76
|
497 ngx_http_rewrite_break(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
498 {
|
|
499 ngx_http_rewrite_loc_conf_t *lcf = conf;
|
|
500
|
|
501 ngx_http_script_code_pt *code;
|
|
502
|
|
503 code = ngx_http_script_start_code(cf->pool, &lcf->codes, sizeof(uintptr_t));
|
|
504 if (code == NULL) {
|
|
505 return NGX_CONF_ERROR;
|
|
506 }
|
|
507
|
|
508 *code = ngx_http_script_break_code;
|
|
509
|
|
510 return NGX_CONF_OK;
|
|
511 }
|
|
512
|
|
513
|
|
514 static char *
|
50
|
515 ngx_http_rewrite_if(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
516 {
|
126
|
517 ngx_http_rewrite_loc_conf_t *lcf = conf;
|
50
|
518
|
|
519 void *mconf;
|
|
520 char *rv;
|
|
521 u_char *elts;
|
|
522 ngx_uint_t i;
|
|
523 ngx_conf_t save;
|
|
524 ngx_http_module_t *module;
|
|
525 ngx_http_conf_ctx_t *ctx, *pctx;
|
378
|
526 ngx_http_core_loc_conf_t *clcf, *pclcf;
|
58
|
527 ngx_http_script_if_code_t *if_code;
|
50
|
528 ngx_http_rewrite_loc_conf_t *nlcf;
|
|
529
|
|
530 ctx = ngx_pcalloc(cf->pool, sizeof(ngx_http_conf_ctx_t));
|
|
531 if (ctx == NULL) {
|
|
532 return NGX_CONF_ERROR;
|
|
533 }
|
|
534
|
|
535 pctx = cf->ctx;
|
|
536 ctx->main_conf = pctx->main_conf;
|
126
|
537 ctx->srv_conf = pctx->srv_conf;
|
50
|
538
|
|
539 ctx->loc_conf = ngx_pcalloc(cf->pool, sizeof(void *) * ngx_http_max_module);
|
|
540 if (ctx->loc_conf == NULL) {
|
|
541 return NGX_CONF_ERROR;
|
|
542 }
|
|
543
|
|
544 for (i = 0; ngx_modules[i]; i++) {
|
|
545 if (ngx_modules[i]->type != NGX_HTTP_MODULE) {
|
|
546 continue;
|
|
547 }
|
|
548
|
|
549 module = ngx_modules[i]->ctx;
|
126
|
550
|
50
|
551 if (module->create_loc_conf) {
|
|
552
|
|
553 mconf = module->create_loc_conf(cf);
|
|
554 if (mconf == NULL) {
|
|
555 return NGX_CONF_ERROR;
|
|
556 }
|
|
557
|
|
558 ctx->loc_conf[ngx_modules[i]->ctx_index] = mconf;
|
|
559 }
|
|
560 }
|
|
561
|
|
562 pclcf = pctx->loc_conf[ngx_http_core_module.ctx_index];
|
|
563
|
|
564 clcf = ctx->loc_conf[ngx_http_core_module.ctx_index];
|
|
565 clcf->loc_conf = ctx->loc_conf;
|
|
566 clcf->name = pclcf->name;
|
|
567 clcf->noname = 1;
|
|
568
|
378
|
569 if (ngx_http_add_location(cf, &pclcf->locations, clcf) != NGX_OK) {
|
50
|
570 return NGX_CONF_ERROR;
|
|
571 }
|
|
572
|
|
573 if (ngx_http_rewrite_if_condition(cf, lcf) != NGX_CONF_OK) {
|
|
574 return NGX_CONF_ERROR;
|
|
575 }
|
|
576
|
58
|
577 if_code = ngx_array_push_n(lcf->codes, sizeof(ngx_http_script_if_code_t));
|
50
|
578 if (if_code == NULL) {
|
550
|
579 return NGX_CONF_ERROR;
|
50
|
580 }
|
|
581
|
58
|
582 if_code->code = ngx_http_script_if_code;
|
50
|
583
|
|
584 elts = lcf->codes->elts;
|
|
585
|
|
586
|
180
|
587 /* the inner directives must be compiled to the same code array */
|
50
|
588
|
|
589 nlcf = ctx->loc_conf[ngx_http_rewrite_module.ctx_index];
|
|
590 nlcf->codes = lcf->codes;
|
|
591
|
|
592
|
|
593 save = *cf;
|
|
594 cf->ctx = ctx;
|
|
595
|
|
596 if (pclcf->name.len == 0) {
|
|
597 if_code->loc_conf = NULL;
|
|
598 cf->cmd_type = NGX_HTTP_SIF_CONF;
|
|
599
|
|
600 } else {
|
|
601 if_code->loc_conf = ctx->loc_conf;
|
|
602 cf->cmd_type = NGX_HTTP_LIF_CONF;
|
|
603 }
|
|
604
|
|
605 rv = ngx_conf_parse(cf, NULL);
|
|
606
|
|
607 *cf = save;
|
|
608
|
|
609 if (rv != NGX_CONF_OK) {
|
|
610 return rv;
|
|
611 }
|
|
612
|
|
613
|
|
614 if (elts != lcf->codes->elts) {
|
58
|
615 if_code = (ngx_http_script_if_code_t *)
|
50
|
616 ((u_char *) if_code + ((u_char *) lcf->codes->elts - elts));
|
|
617 }
|
|
618
|
|
619 if_code->next = (u_char *) lcf->codes->elts + lcf->codes->nelts
|
|
620 - (u_char *) if_code;
|
|
621
|
104
|
622 /* the code array belong to parent block */
|
|
623
|
|
624 nlcf->codes = NULL;
|
|
625
|
50
|
626 return NGX_CONF_OK;
|
|
627 }
|
|
628
|
|
629
|
|
630 static char *
|
|
631 ngx_http_rewrite_if_condition(ngx_conf_t *cf, ngx_http_rewrite_loc_conf_t *lcf)
|
|
632 {
|
180
|
633 u_char *p;
|
|
634 size_t len;
|
589
|
635 ngx_str_t *value;
|
|
636 ngx_uint_t cur, last;
|
|
637 ngx_regex_compile_t rc;
|
126
|
638 ngx_http_script_code_pt *code;
|
180
|
639 ngx_http_script_file_code_t *fop;
|
58
|
640 ngx_http_script_regex_code_t *regex;
|
|
641 u_char errstr[NGX_MAX_CONF_ERRSTR];
|
50
|
642
|
|
643 value = cf->args->elts;
|
|
644 last = cf->args->nelts - 1;
|
|
645
|
|
646 if (value[1].len < 1 || value[1].data[0] != '(') {
|
|
647 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
648 "invalid condition \"%V\"", &value[1]);
|
|
649 return NGX_CONF_ERROR;
|
|
650 }
|
|
651
|
|
652 if (value[1].len == 1) {
|
|
653 cur = 2;
|
|
654
|
|
655 } else {
|
|
656 cur = 1;
|
|
657 value[1].len--;
|
|
658 value[1].data++;
|
|
659 }
|
|
660
|
|
661 if (value[last].len < 1 || value[last].data[value[last].len - 1] != ')') {
|
|
662 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
663 "invalid condition \"%V\"", &value[last]);
|
|
664 return NGX_CONF_ERROR;
|
|
665 }
|
|
666
|
|
667 if (value[last].len == 1) {
|
|
668 last--;
|
|
669
|
|
670 } else {
|
|
671 value[last].len--;
|
|
672 value[last].data[value[last].len] = '\0';
|
|
673 }
|
|
674
|
180
|
675 len = value[cur].len;
|
|
676 p = value[cur].data;
|
|
677
|
|
678 if (len > 1 && p[0] == '$') {
|
50
|
679
|
|
680 if (cur != last && cur + 2 != last) {
|
|
681 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
180
|
682 "invalid condition \"%V\"", &value[cur]);
|
50
|
683 return NGX_CONF_ERROR;
|
|
684 }
|
|
685
|
126
|
686 if (ngx_http_rewrite_variable(cf, lcf, &value[cur]) != NGX_CONF_OK) {
|
50
|
687 return NGX_CONF_ERROR;
|
|
688 }
|
|
689
|
|
690 if (cur == last) {
|
|
691 return NGX_CONF_OK;
|
|
692 }
|
|
693
|
|
694 cur++;
|
|
695
|
180
|
696 len = value[cur].len;
|
|
697 p = value[cur].data;
|
|
698
|
|
699 if (len == 1 && p[0] == '=') {
|
126
|
700
|
|
701 if (ngx_http_rewrite_value(cf, lcf, &value[last]) != NGX_CONF_OK) {
|
|
702 return NGX_CONF_ERROR;
|
|
703 }
|
|
704
|
|
705 code = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
706 sizeof(uintptr_t));
|
|
707 if (code == NULL) {
|
|
708 return NGX_CONF_ERROR;
|
|
709 }
|
|
710
|
|
711 *code = ngx_http_script_equal_code;
|
|
712
|
|
713 return NGX_CONF_OK;
|
50
|
714 }
|
|
715
|
180
|
716 if (len == 2 && p[0] == '!' && p[1] == '=') {
|
|
717
|
126
|
718 if (ngx_http_rewrite_value(cf, lcf, &value[last]) != NGX_CONF_OK) {
|
|
719 return NGX_CONF_ERROR;
|
|
720 }
|
|
721
|
|
722 code = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
723 sizeof(uintptr_t));
|
|
724 if (code == NULL) {
|
|
725 return NGX_CONF_ERROR;
|
|
726 }
|
|
727
|
|
728 *code = ngx_http_script_not_equal_code;
|
|
729 return NGX_CONF_OK;
|
50
|
730 }
|
|
731
|
180
|
732 if ((len == 1 && p[0] == '~')
|
|
733 || (len == 2 && p[0] == '~' && p[1] == '*')
|
|
734 || (len == 2 && p[0] == '!' && p[1] == '~')
|
|
735 || (len == 3 && p[0] == '!' && p[1] == '~' && p[2] == '*'))
|
126
|
736 {
|
|
737 regex = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
738 sizeof(ngx_http_script_regex_code_t));
|
|
739 if (regex == NULL) {
|
|
740 return NGX_CONF_ERROR;
|
|
741 }
|
58
|
742
|
126
|
743 ngx_memzero(regex, sizeof(ngx_http_script_regex_code_t));
|
|
744
|
589
|
745 ngx_memzero(&rc, sizeof(ngx_regex_compile_t));
|
126
|
746
|
589
|
747 rc.pattern = value[last];
|
|
748 rc.options = (p[len - 1] == '*') ? NGX_REGEX_CASELESS : 0;
|
|
749 rc.err.len = NGX_MAX_CONF_ERRSTR;
|
|
750 rc.err.data = errstr;
|
50
|
751
|
589
|
752 regex->regex = ngx_http_regex_compile(cf, &rc);
|
126
|
753 if (regex->regex == NULL) {
|
|
754 return NGX_CONF_ERROR;
|
|
755 }
|
|
756
|
|
757 regex->code = ngx_http_script_regex_start_code;
|
|
758 regex->next = sizeof(ngx_http_script_regex_code_t);
|
|
759 regex->test = 1;
|
180
|
760 if (p[0] == '!') {
|
|
761 regex->negative_test = 1;
|
|
762 }
|
126
|
763 regex->name = value[last];
|
50
|
764
|
126
|
765 return NGX_CONF_OK;
|
50
|
766 }
|
|
767
|
126
|
768 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
769 "unexpected \"%V\" in condition", &value[cur]);
|
|
770 return NGX_CONF_ERROR;
|
180
|
771
|
|
772 } else if ((len == 2 && p[0] == '-')
|
|
773 || (len == 3 && p[0] == '!' && p[1] == '-'))
|
|
774 {
|
|
775 if (cur + 1 != last) {
|
|
776 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
777 "invalid condition \"%V\"", &value[cur]);
|
|
778 return NGX_CONF_ERROR;
|
|
779 }
|
|
780
|
|
781 value[last].data[value[last].len] = '\0';
|
|
782 value[last].len++;
|
|
783
|
|
784 if (ngx_http_rewrite_value(cf, lcf, &value[last]) != NGX_CONF_OK) {
|
|
785 return NGX_CONF_ERROR;
|
|
786 }
|
|
787
|
|
788 fop = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
789 sizeof(ngx_http_script_file_code_t));
|
|
790 if (fop == NULL) {
|
|
791 return NGX_CONF_ERROR;
|
|
792 }
|
|
793
|
|
794 fop->code = ngx_http_script_file_code;
|
|
795
|
|
796 if (p[1] == 'f') {
|
|
797 fop->op = ngx_http_script_file_plain;
|
|
798 return NGX_CONF_OK;
|
|
799 }
|
|
800
|
218
|
801 if (p[1] == 'd') {
|
|
802 fop->op = ngx_http_script_file_dir;
|
|
803 return NGX_CONF_OK;
|
|
804 }
|
|
805
|
|
806 if (p[1] == 'e') {
|
|
807 fop->op = ngx_http_script_file_exists;
|
|
808 return NGX_CONF_OK;
|
|
809 }
|
|
810
|
|
811 if (p[1] == 'x') {
|
|
812 fop->op = ngx_http_script_file_exec;
|
|
813 return NGX_CONF_OK;
|
|
814 }
|
|
815
|
180
|
816 if (p[0] == '!') {
|
|
817 if (p[2] == 'f') {
|
|
818 fop->op = ngx_http_script_file_not_plain;
|
|
819 return NGX_CONF_OK;
|
|
820 }
|
218
|
821
|
|
822 if (p[2] == 'd') {
|
|
823 fop->op = ngx_http_script_file_not_dir;
|
|
824 return NGX_CONF_OK;
|
|
825 }
|
|
826
|
|
827 if (p[2] == 'e') {
|
|
828 fop->op = ngx_http_script_file_not_exists;
|
|
829 return NGX_CONF_OK;
|
|
830 }
|
|
831
|
|
832 if (p[2] == 'x') {
|
|
833 fop->op = ngx_http_script_file_not_exec;
|
|
834 return NGX_CONF_OK;
|
|
835 }
|
180
|
836 }
|
|
837
|
|
838 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
839 "invalid condition \"%V\"", &value[cur]);
|
|
840 return NGX_CONF_ERROR;
|
50
|
841 }
|
|
842
|
|
843 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
844 "invalid condition \"%V\"", &value[cur]);
|
|
845
|
|
846 return NGX_CONF_ERROR;
|
|
847 }
|
|
848
|
|
849
|
|
850 static char *
|
|
851 ngx_http_rewrite_variable(ngx_conf_t *cf, ngx_http_rewrite_loc_conf_t *lcf,
|
|
852 ngx_str_t *value)
|
|
853 {
|
58
|
854 ngx_int_t index;
|
|
855 ngx_http_script_var_code_t *var_code;
|
50
|
856
|
|
857 value->len--;
|
|
858 value->data++;
|
|
859
|
126
|
860 index = ngx_http_get_variable_index(cf, value);
|
50
|
861
|
126
|
862 if (index == NGX_ERROR) {
|
|
863 return NGX_CONF_ERROR;
|
50
|
864 }
|
|
865
|
126
|
866 var_code = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
867 sizeof(ngx_http_script_var_code_t));
|
|
868 if (var_code == NULL) {
|
|
869 return NGX_CONF_ERROR;
|
50
|
870 }
|
|
871
|
126
|
872 var_code->code = ngx_http_script_var_code;
|
|
873 var_code->index = index;
|
50
|
874
|
|
875 return NGX_CONF_OK;
|
|
876 }
|
|
877
|
|
878
|
|
879 static char *
|
|
880 ngx_http_rewrite_set(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
881 {
|
126
|
882 ngx_http_rewrite_loc_conf_t *lcf = conf;
|
50
|
883
|
186
|
884 ngx_int_t index;
|
|
885 ngx_str_t *value;
|
|
886 ngx_http_variable_t *v;
|
|
887 ngx_http_script_var_code_t *vcode;
|
|
888 ngx_http_script_var_handler_code_t *vhcode;
|
50
|
889
|
|
890 value = cf->args->elts;
|
|
891
|
|
892 if (value[1].data[0] != '$') {
|
|
893 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
894 "invalid variable name \"%V\"", &value[1]);
|
|
895 return NGX_CONF_ERROR;
|
|
896 }
|
|
897
|
|
898 value[1].len--;
|
|
899 value[1].data++;
|
|
900
|
340
|
901 v = ngx_http_add_variable(cf, &value[1], NGX_HTTP_VAR_CHANGEABLE);
|
50
|
902 if (v == NULL) {
|
|
903 return NGX_CONF_ERROR;
|
|
904 }
|
|
905
|
58
|
906 index = ngx_http_get_variable_index(cf, &value[1]);
|
|
907 if (index == NGX_ERROR) {
|
|
908 return NGX_CONF_ERROR;
|
|
909 }
|
50
|
910
|
300
|
911 if (v->get_handler == NULL
|
|
912 && ngx_strncasecmp(value[1].data, (u_char *) "http_", 5) != 0
|
|
913 && ngx_strncasecmp(value[1].data, (u_char *) "sent_http_", 10) != 0
|
|
914 && ngx_strncasecmp(value[1].data, (u_char *) "upstream_http_", 14) != 0)
|
|
915 {
|
186
|
916 v->get_handler = ngx_http_rewrite_var;
|
104
|
917 v->data = index;
|
|
918 }
|
58
|
919
|
126
|
920 if (ngx_http_rewrite_value(cf, lcf, &value[2]) != NGX_CONF_OK) {
|
|
921 return NGX_CONF_ERROR;
|
64
|
922 }
|
50
|
923
|
186
|
924 if (v->set_handler) {
|
|
925 vhcode = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
926 sizeof(ngx_http_script_var_handler_code_t));
|
|
927 if (vhcode == NULL) {
|
|
928 return NGX_CONF_ERROR;
|
|
929 }
|
|
930
|
|
931 vhcode->code = ngx_http_script_var_set_handler_code;
|
|
932 vhcode->handler = v->set_handler;
|
|
933 vhcode->data = v->data;
|
|
934
|
|
935 return NGX_CONF_OK;
|
|
936 }
|
|
937
|
|
938 vcode = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
939 sizeof(ngx_http_script_var_code_t));
|
|
940 if (vcode == NULL) {
|
50
|
941 return NGX_CONF_ERROR;
|
|
942 }
|
|
943
|
186
|
944 vcode->code = ngx_http_script_set_var_code;
|
|
945 vcode->index = (uintptr_t) index;
|
50
|
946
|
|
947 return NGX_CONF_OK;
|
|
948 }
|
126
|
949
|
|
950
|
|
951 static char *
|
|
952 ngx_http_rewrite_value(ngx_conf_t *cf, ngx_http_rewrite_loc_conf_t *lcf,
|
|
953 ngx_str_t *value)
|
|
954 {
|
|
955 ngx_int_t n;
|
|
956 ngx_http_script_compile_t sc;
|
|
957 ngx_http_script_value_code_t *val;
|
|
958 ngx_http_script_complex_value_code_t *complex;
|
|
959
|
|
960 n = ngx_http_script_variables_count(value);
|
|
961
|
|
962 if (n == 0) {
|
|
963 val = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
964 sizeof(ngx_http_script_value_code_t));
|
|
965 if (val == NULL) {
|
|
966 return NGX_CONF_ERROR;
|
|
967 }
|
|
968
|
|
969 n = ngx_atoi(value->data, value->len);
|
|
970
|
|
971 if (n == NGX_ERROR) {
|
|
972 n = 0;
|
|
973 }
|
|
974
|
|
975 val->code = ngx_http_script_value_code;
|
|
976 val->value = (uintptr_t) n;
|
|
977 val->text_len = (uintptr_t) value->len;
|
|
978 val->text_data = (uintptr_t) value->data;
|
|
979
|
|
980 return NGX_CONF_OK;
|
|
981 }
|
|
982
|
|
983 complex = ngx_http_script_start_code(cf->pool, &lcf->codes,
|
|
984 sizeof(ngx_http_script_complex_value_code_t));
|
|
985 if (complex == NULL) {
|
|
986 return NGX_CONF_ERROR;
|
|
987 }
|
|
988
|
|
989 complex->code = ngx_http_script_complex_value_code;
|
|
990 complex->lengths = NULL;
|
|
991
|
|
992 ngx_memzero(&sc, sizeof(ngx_http_script_compile_t));
|
|
993
|
|
994 sc.cf = cf;
|
|
995 sc.source = value;
|
|
996 sc.lengths = &complex->lengths;
|
|
997 sc.values = &lcf->codes;
|
|
998 sc.variables = n;
|
|
999 sc.complete_lengths = 1;
|
|
1000
|
|
1001 if (ngx_http_script_compile(&sc) != NGX_OK) {
|
|
1002 return NGX_CONF_ERROR;
|
|
1003 }
|
|
1004
|
|
1005 return NGX_CONF_OK;
|
|
1006 }
|