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