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