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 ngx_http_rewrite_engine_s ngx_http_rewrite_engine_t;
|
|
13
|
|
14 typedef void (*ngx_http_rewrite_code_pt) (ngx_http_rewrite_engine_t *e);
|
|
15
|
|
16
|
|
17 typedef struct {
|
|
18 ngx_str_t name;
|
|
19 ngx_uint_t wildcard;
|
|
20 } ngx_http_rewrite_referer_t;
|
|
21
|
|
22
|
|
23 typedef struct {
|
|
24 ngx_str_t *name;
|
|
25 ngx_http_variable_value_t *value;
|
|
26 } ngx_http_rewrite_variable_t;
|
|
27
|
|
28
|
|
29 typedef struct {
|
|
30 ngx_array_t *codes; /* uintptr_t */
|
|
31 ngx_array_t *referers; /* ngx_http_rewrite_referer_t */
|
|
32
|
|
33 ngx_uint_t max_captures;
|
|
34 ngx_uint_t stack_size;
|
|
35
|
|
36 ngx_flag_t log;
|
|
37
|
|
38 ngx_flag_t no_referer;
|
|
39 } ngx_http_rewrite_loc_conf_t;
|
|
40
|
|
41
|
|
42 typedef struct {
|
|
43 ngx_http_rewrite_code_pt code;
|
|
44 ngx_regex_t *regex;
|
|
45 uintptr_t size;
|
|
46 uintptr_t ncaptures;
|
|
47 uintptr_t status;
|
|
48 uintptr_t next;
|
|
49
|
|
50 uintptr_t test:1;
|
|
51 uintptr_t uri:1;
|
|
52
|
|
53 /* add the r->args to the new arguments */
|
|
54 uintptr_t args:1;
|
|
55
|
|
56 uintptr_t redirect:1;
|
|
57 uintptr_t break_cycle:1;
|
|
58
|
|
59 ngx_str_t name;
|
|
60 } ngx_http_rewrite_regex_code_t;
|
|
61
|
|
62
|
|
63 typedef struct {
|
|
64 ngx_http_rewrite_code_pt code;
|
|
65
|
|
66 uintptr_t uri:1;
|
|
67
|
|
68 /* add the r->args to the new arguments */
|
|
69 uintptr_t args:1;
|
|
70
|
|
71 uintptr_t redirect:1;
|
|
72 } ngx_http_rewrite_regex_end_code_t;
|
|
73
|
|
74 typedef struct {
|
|
75 ngx_http_rewrite_code_pt code;
|
|
76 uintptr_t n;
|
|
77 } ngx_http_rewrite_copy_capture_code_t;
|
|
78
|
|
79
|
|
80 typedef struct {
|
|
81 ngx_http_rewrite_code_pt code;
|
|
82 uintptr_t len;
|
|
83 } ngx_http_rewrite_copy_code_t;
|
|
84
|
|
85
|
|
86 typedef struct {
|
|
87 ngx_http_rewrite_code_pt code;
|
|
88 uintptr_t status;
|
|
89 uintptr_t null;
|
|
90 } ngx_http_rewrite_return_code_t;
|
|
91
|
|
92
|
|
93 typedef struct {
|
|
94 ngx_http_rewrite_code_pt code;
|
|
95 uintptr_t next;
|
|
96 void **loc_conf;
|
|
97 } ngx_http_rewrite_if_code_t;
|
|
98
|
|
99
|
|
100 typedef struct {
|
|
101 ngx_http_rewrite_code_pt code;
|
|
102 uintptr_t value;
|
|
103 uintptr_t text_len;
|
|
104 uintptr_t text_data;
|
|
105 } ngx_http_rewrite_value_code_t;
|
|
106
|
|
107
|
|
108 typedef struct {
|
|
109 ngx_http_rewrite_code_pt code;
|
|
110 uintptr_t index;
|
|
111 } ngx_http_rewrite_var_code_t;
|
|
112
|
|
113
|
|
114 struct ngx_http_rewrite_engine_s {
|
|
115 u_char *ip;
|
|
116 ngx_http_variable_value_t *sp;
|
|
117
|
|
118 ngx_str_t buf;
|
|
119 ngx_str_t *line;
|
|
120
|
|
121 u_char *pos;
|
|
122
|
|
123 /* the start of the rewritten arguments */
|
|
124 u_char *args;
|
|
125
|
|
126 unsigned quote:1;
|
|
127
|
|
128 ngx_int_t status;
|
|
129
|
|
130 int *captures;
|
|
131
|
|
132 ngx_http_request_t *request;
|
|
133 ngx_http_rewrite_loc_conf_t *conf;
|
|
134 };
|
|
135
|
|
136
|
|
137 static ngx_int_t ngx_http_rewrite_init(ngx_cycle_t *cycle);
|
|
138 static void *ngx_http_rewrite_create_loc_conf(ngx_conf_t *cf);
|
|
139 static char *ngx_http_rewrite_merge_loc_conf(ngx_conf_t *cf,
|
|
140 void *parent, void *child);
|
|
141 static char *ngx_http_rewrite(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
|
|
142 static char *ngx_http_rewrite_return(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
143 void *conf);
|
|
144 static char *ngx_http_rewrite_if(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
145 void *conf);
|
|
146 static char * ngx_http_rewrite_if_condition(ngx_conf_t *cf,
|
|
147 ngx_http_rewrite_loc_conf_t *lcf);
|
|
148 static char *ngx_http_rewrite_variable(ngx_conf_t *cf,
|
|
149 ngx_http_rewrite_loc_conf_t *lcf, ngx_str_t *value);
|
|
150 static char *ngx_http_rewrite_valid_referers(ngx_conf_t *cf,
|
|
151 ngx_command_t *cmd, void *conf);
|
|
152 static char * ngx_http_rewrite_set(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
153 void *conf);
|
|
154 static void *ngx_http_rewrite_start_code(ngx_pool_t *pool,
|
|
155 ngx_array_t **codes, size_t size);
|
|
156 static void *ngx_http_rewrite_add_code(ngx_array_t *codes, size_t size,
|
|
157 void *code);
|
|
158
|
|
159
|
|
160 static ngx_command_t ngx_http_rewrite_commands[] = {
|
|
161
|
|
162 { ngx_string("rewrite"),
|
|
163 NGX_HTTP_SRV_CONF|NGX_HTTP_SIF_CONF|NGX_HTTP_LOC_CONF|NGX_HTTP_LIF_CONF
|
|
164 |NGX_CONF_TAKE23,
|
|
165 ngx_http_rewrite,
|
|
166 NGX_HTTP_LOC_CONF_OFFSET,
|
|
167 0,
|
|
168 NULL },
|
|
169
|
|
170 { ngx_string("return"),
|
|
171 NGX_HTTP_SRV_CONF|NGX_HTTP_SIF_CONF|NGX_HTTP_LOC_CONF|NGX_HTTP_LIF_CONF
|
|
172 |NGX_CONF_TAKE1,
|
|
173 ngx_http_rewrite_return,
|
|
174 NGX_HTTP_LOC_CONF_OFFSET,
|
|
175 0,
|
|
176 NULL },
|
|
177
|
|
178 { ngx_string("if"),
|
|
179 NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_BLOCK|NGX_CONF_1MORE,
|
|
180 ngx_http_rewrite_if,
|
|
181 NGX_HTTP_LOC_CONF_OFFSET,
|
|
182 0,
|
|
183 NULL },
|
|
184
|
|
185 { ngx_string("valid_referers"),
|
|
186 NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_1MORE,
|
|
187 ngx_http_rewrite_valid_referers,
|
|
188 NGX_HTTP_LOC_CONF_OFFSET,
|
|
189 0,
|
|
190 NULL },
|
|
191
|
|
192 { ngx_string("set"),
|
|
193 NGX_HTTP_SRV_CONF|NGX_HTTP_SIF_CONF|NGX_HTTP_LOC_CONF|NGX_HTTP_LIF_CONF
|
|
194 |NGX_CONF_TAKE2,
|
|
195 ngx_http_rewrite_set,
|
|
196 NGX_HTTP_LOC_CONF_OFFSET,
|
|
197 0,
|
|
198 NULL },
|
|
199
|
|
200 { ngx_string("rewrite_log"),
|
|
201 NGX_HTTP_SRV_CONF|NGX_HTTP_SIF_CONF|NGX_HTTP_LOC_CONF|NGX_HTTP_LIF_CONF
|
|
202 |NGX_CONF_TAKE1,
|
|
203 ngx_conf_set_flag_slot,
|
|
204 NGX_HTTP_LOC_CONF_OFFSET,
|
|
205 offsetof(ngx_http_rewrite_loc_conf_t, log),
|
|
206 NULL },
|
|
207
|
|
208 ngx_null_command
|
|
209 };
|
|
210
|
|
211
|
|
212 ngx_http_module_t ngx_http_rewrite_module_ctx = {
|
|
213 NULL, /* pre conf */
|
|
214
|
|
215 NULL, /* create main configuration */
|
|
216 NULL, /* init main configuration */
|
|
217
|
|
218 NULL, /* create server configuration */
|
|
219 NULL, /* merge server configuration */
|
|
220
|
|
221 ngx_http_rewrite_create_loc_conf, /* create location configration */
|
|
222 ngx_http_rewrite_merge_loc_conf /* merge location configration */
|
|
223 };
|
|
224
|
|
225
|
|
226 ngx_module_t ngx_http_rewrite_module = {
|
|
227 NGX_MODULE,
|
|
228 &ngx_http_rewrite_module_ctx, /* module context */
|
|
229 ngx_http_rewrite_commands, /* module directives */
|
|
230 NGX_HTTP_MODULE, /* module type */
|
|
231 ngx_http_rewrite_init, /* init module */
|
|
232 NULL /* init process */
|
|
233 };
|
|
234
|
|
235
|
|
236 #define ngx_http_rewrite_exit (u_char *) &ngx_http_rewrite_exit_code
|
|
237
|
|
238 uintptr_t ngx_http_rewrite_exit_code = (uintptr_t) NULL;
|
|
239
|
|
240 static ngx_http_variable_value_t ngx_http_rewrite_null_value =
|
|
241 { 0, ngx_string("") };
|
|
242
|
|
243
|
|
244 static ngx_int_t
|
|
245 ngx_http_rewrite_handler(ngx_http_request_t *r)
|
|
246 {
|
|
247 ngx_http_rewrite_code_pt code;
|
|
248 ngx_http_rewrite_engine_t *e;
|
|
249 ngx_http_rewrite_loc_conf_t *cf;
|
|
250
|
|
251 cf = ngx_http_get_module_loc_conf(r, ngx_http_rewrite_module);
|
|
252
|
|
253 if (cf->codes == NULL) {
|
|
254 return NGX_DECLINED;
|
|
255 }
|
|
256
|
|
257 e = ngx_palloc(r->pool, sizeof(ngx_http_rewrite_engine_t));
|
|
258 if (e == NULL) {
|
|
259 return NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
260 }
|
|
261
|
|
262 e->sp = ngx_palloc(r->pool,
|
|
263 cf->stack_size * sizeof(ngx_http_variable_value_t));
|
|
264 if (e->sp == NULL) {
|
|
265 return NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
266 }
|
|
267
|
|
268 if (cf->max_captures) {
|
|
269 e->captures = ngx_palloc(r->pool, cf->max_captures * sizeof(int));
|
|
270 if (e->captures == NULL) {
|
|
271 return NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
272 }
|
|
273
|
|
274 } else {
|
|
275 e->captures = NULL;
|
|
276 }
|
|
277
|
|
278 e->ip = cf->codes->elts;
|
|
279 e->buf.len = 0;
|
|
280 e->buf.data = NULL;
|
|
281 e->line = NULL;
|
|
282 e->pos = NULL;
|
|
283 e->args = NULL;
|
|
284 e->quote = 1;
|
|
285 e->status = NGX_DECLINED;
|
|
286 e->request = r;
|
|
287 e->conf = cf;
|
|
288
|
|
289 while (*(uintptr_t *) e->ip) {
|
|
290 code = *(ngx_http_rewrite_code_pt *) e->ip;
|
|
291 code(e);
|
|
292 }
|
|
293
|
|
294 return e->status;
|
|
295 }
|
|
296
|
|
297
|
|
298 static void
|
|
299 ngx_http_rewrite_regex_start_code(ngx_http_rewrite_engine_t *e)
|
|
300 {
|
|
301 ngx_int_t rc;
|
|
302 ngx_uint_t n;
|
|
303 ngx_http_request_t *r;
|
|
304 ngx_http_rewrite_regex_code_t *code;
|
|
305
|
|
306 code = (ngx_http_rewrite_regex_code_t *) e->ip;
|
|
307
|
|
308 r = e->request;
|
|
309
|
|
310 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
311 "http rewrite regex: \"%V\"", &code->name);
|
|
312
|
|
313 if (code->uri) {
|
|
314 e->line = &r->uri;
|
|
315 } else {
|
|
316 e->sp--;
|
|
317 e->line = &e->sp->text;
|
|
318 }
|
|
319
|
|
320 rc = ngx_regex_exec(code->regex, e->line, e->captures, code->ncaptures);
|
|
321
|
|
322 if (rc == NGX_REGEX_NO_MATCHED) {
|
|
323 if (e->conf->log) {
|
|
324 ngx_log_error(NGX_LOG_NOTICE, r->connection->log, 0,
|
|
325 "\"%V\" does not match \"%V\"", &code->name, e->line);
|
|
326 }
|
|
327
|
|
328 if (code->test) {
|
|
329 e->sp->value = 0;
|
|
330 e->sp->text.len = 0;
|
|
331 e->sp->text.data = (u_char *) "";
|
|
332 e->sp++;
|
|
333
|
|
334 e->ip += sizeof(ngx_http_rewrite_regex_code_t);
|
|
335 return;
|
|
336 }
|
|
337
|
|
338 e->ip += code->next;
|
|
339 return;
|
|
340 }
|
|
341
|
|
342 if (rc < 0) {
|
|
343 ngx_log_error(NGX_LOG_ALERT, r->connection->log, 0,
|
|
344 ngx_regex_exec_n " failed: %d on \"%V\" using \"%V\"",
|
|
345 rc, e->line, &code->name);
|
|
346
|
|
347 e->ip = ngx_http_rewrite_exit;
|
|
348 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
349 return;
|
|
350 }
|
|
351
|
|
352 if (e->conf->log) {
|
|
353 ngx_log_error(NGX_LOG_NOTICE, r->connection->log, 0,
|
|
354 "\"%V\" matches \"%V\"", &code->name, e->line);
|
|
355 }
|
|
356
|
|
357 if (code->test) {
|
|
358 e->sp->value = 1;
|
|
359 e->sp->text.len = 1;
|
|
360 e->sp->text.data = (u_char *) "1";
|
|
361 e->sp++;
|
|
362
|
|
363 e->ip += sizeof(ngx_http_rewrite_regex_code_t);
|
|
364 return;
|
|
365 }
|
|
366
|
|
367 if (code->status) {
|
|
368 e->status = code->status;
|
|
369
|
|
370 if (!code->redirect) {
|
|
371 e->ip = ngx_http_rewrite_exit;
|
|
372 return;
|
|
373 }
|
|
374 }
|
|
375
|
|
376 e->buf.len = code->size;
|
|
377
|
|
378 if (code->uri) {
|
|
379 if (!code->break_cycle) {
|
|
380 r->uri_changed = 1;
|
|
381 }
|
|
382
|
|
383 if (rc && (r->quoted_uri || r->plus_in_uri)) {
|
|
384 e->buf.len += 2 * ngx_escape_uri(NULL, r->uri.data, r->uri.len,
|
|
385 NGX_ESCAPE_ARGS);
|
|
386 }
|
|
387 }
|
|
388
|
|
389 for (n = 1; n < (ngx_uint_t) rc; n++) {
|
|
390 e->buf.len += e->captures[2 * n + 1] - e->captures[2 * n];
|
|
391 }
|
|
392
|
|
393 if (code->args && r->args.len) {
|
|
394 e->buf.len += r->args.len + 1;
|
|
395 }
|
|
396
|
|
397 e->buf.data = ngx_palloc(r->pool, e->buf.len);
|
|
398 if (e->buf.data == NULL) {
|
|
399 e->ip = ngx_http_rewrite_exit;
|
|
400 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
401 return;
|
|
402 }
|
|
403
|
|
404 e->quote = code->redirect;
|
|
405
|
|
406 e->pos = e->buf.data;
|
|
407
|
|
408 e->ip += sizeof(ngx_http_rewrite_regex_code_t);
|
|
409 }
|
|
410
|
|
411
|
|
412 static void
|
|
413 ngx_http_rewrite_regex_end_code(ngx_http_rewrite_engine_t *e)
|
|
414 {
|
|
415 ngx_http_request_t *r;
|
|
416 ngx_http_rewrite_regex_end_code_t *code;
|
|
417
|
|
418 code = (ngx_http_rewrite_regex_end_code_t *) e->ip;
|
|
419
|
|
420 r = e->request;
|
|
421
|
|
422 e->quote = 0;
|
|
423
|
|
424 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
425 "http rewrite regex end");
|
|
426
|
|
427 if (e->args) {
|
|
428 e->buf.len = e->args - e->buf.data;
|
|
429
|
|
430 if (code->args && r->args.len) {
|
|
431 *e->pos++ = '&';
|
|
432 e->pos = ngx_cpymem(e->pos, r->args.data, r->args.len);
|
|
433 }
|
|
434
|
|
435 r->args.len = e->pos - e->args;
|
|
436 r->args.data = e->args;
|
|
437
|
|
438 e->args = NULL;
|
|
439
|
|
440 } else {
|
|
441 if (code->args && r->args.len) {
|
|
442 *e->pos++ = '?';
|
|
443 e->pos = ngx_cpymem(e->pos, r->args.data, r->args.len);
|
|
444 }
|
|
445
|
|
446 e->buf.len = e->pos - e->buf.data;
|
|
447 }
|
|
448
|
|
449 if (!code->redirect) {
|
|
450 if (e->conf->log) {
|
|
451 ngx_log_error(NGX_LOG_NOTICE, r->connection->log, 0,
|
|
452 "rewritten data: \"%V\", args: \"%V\"",
|
|
453 &e->buf, &r->args);
|
|
454 }
|
|
455
|
|
456 if (code->uri) {
|
|
457 r->uri = e->buf;
|
|
458
|
|
459 if (ngx_http_set_exten(r) != NGX_OK) {
|
|
460 e->ip = ngx_http_rewrite_exit;
|
|
461 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
462 return;
|
|
463 }
|
|
464 }
|
|
465
|
|
466 e->ip += sizeof(ngx_http_rewrite_regex_end_code_t);
|
|
467 return;
|
|
468 }
|
|
469
|
|
470 ngx_log_error(NGX_LOG_NOTICE, r->connection->log, 0,
|
|
471 "rewritten redirect: \"%V\"", &e->buf);
|
|
472
|
|
473 r->headers_out.location = ngx_list_push(&r->headers_out.headers);
|
|
474 if (r->headers_out.location == NULL) {
|
|
475 e->ip = ngx_http_rewrite_exit;
|
|
476 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
477 return;
|
|
478 }
|
|
479
|
|
480 if (e->buf.data[0] != '/') {
|
|
481 r->headers_out.location->key.len = sizeof("Location") - 1;
|
|
482 r->headers_out.location->key.data = (u_char *) "Location";
|
|
483 }
|
|
484
|
|
485 r->headers_out.location->value = e->buf;
|
|
486
|
|
487 e->ip += sizeof(ngx_http_rewrite_regex_end_code_t);
|
|
488 }
|
|
489
|
|
490
|
|
491 static void
|
|
492 ngx_http_rewrite_copy_capture_code(ngx_http_rewrite_engine_t *e)
|
|
493 {
|
|
494 ngx_http_rewrite_copy_capture_code_t *code;
|
|
495
|
|
496 code = (ngx_http_rewrite_copy_capture_code_t *) e->ip;
|
|
497
|
|
498 e->ip += sizeof(ngx_http_rewrite_copy_capture_code_t);
|
|
499
|
|
500 if ((e->args || e->quote)
|
|
501 && (e->request->quoted_uri || e->request->plus_in_uri))
|
|
502 {
|
|
503 e->pos = (u_char *) ngx_escape_uri(e->pos,
|
|
504 &e->line->data[e->captures[code->n]],
|
|
505 e->captures[code->n + 1] - e->captures[code->n],
|
|
506 NGX_ESCAPE_ARGS);
|
|
507 } else {
|
|
508 e->pos = ngx_cpymem(e->pos, &e->line->data[e->captures[code->n]],
|
|
509 e->captures[code->n + 1] - e->captures[code->n]);
|
|
510 }
|
|
511
|
|
512 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0,
|
|
513 "http rewrite capture: \"%V\"", &e->buf);
|
|
514 }
|
|
515
|
|
516
|
|
517 static void
|
|
518 ngx_http_rewrite_copy_code(ngx_http_rewrite_engine_t *e)
|
|
519 {
|
|
520 ngx_http_rewrite_copy_code_t *code;
|
|
521
|
|
522 code = (ngx_http_rewrite_copy_code_t *) e->ip;
|
|
523
|
|
524 e->pos = ngx_cpymem(e->pos, e->ip + sizeof(ngx_http_rewrite_copy_code_t),
|
|
525 code->len);
|
|
526
|
|
527 e->ip += sizeof(ngx_http_rewrite_copy_code_t)
|
|
528 + ((code->len + sizeof(uintptr_t) - 1) & ~(sizeof(uintptr_t) - 1));
|
|
529
|
|
530 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0,
|
|
531 "http rewrite copy: \"%V\"", &e->buf);
|
|
532 }
|
|
533
|
|
534
|
|
535 static void
|
|
536 ngx_http_rewrite_start_args_code(ngx_http_rewrite_engine_t *e)
|
|
537 {
|
|
538 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0,
|
|
539 "http rewrite args");
|
|
540
|
|
541 e->args = e->pos;
|
|
542 e->ip += sizeof(uintptr_t);
|
|
543 }
|
|
544
|
|
545
|
|
546 static void
|
|
547 ngx_http_rewrite_return_code(ngx_http_rewrite_engine_t *e)
|
|
548 {
|
|
549 ngx_http_rewrite_return_code_t *code;
|
|
550
|
|
551 code = (ngx_http_rewrite_return_code_t *) e->ip;
|
|
552
|
|
553 e->status = code->status;
|
|
554
|
|
555 e->ip += sizeof(ngx_http_rewrite_return_code_t) - sizeof(uintptr_t);
|
|
556 }
|
|
557
|
|
558
|
|
559 static void
|
|
560 ngx_http_rewrite_if_code(ngx_http_rewrite_engine_t *e)
|
|
561 {
|
|
562 ngx_http_rewrite_if_code_t *code;
|
|
563
|
|
564 code = (ngx_http_rewrite_if_code_t *) e->ip;
|
|
565
|
|
566 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0,
|
|
567 "http rewrite if");
|
|
568
|
|
569 e->sp--;
|
|
570
|
|
571 if (e->sp->value) {
|
|
572 if (code->loc_conf) {
|
|
573 e->request->loc_conf = code->loc_conf;
|
|
574 }
|
|
575
|
|
576 e->ip += sizeof(ngx_http_rewrite_if_code_t);
|
|
577 return;
|
|
578 }
|
|
579
|
|
580 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0,
|
|
581 "http rewrite if false");
|
|
582
|
|
583 e->ip += code->next;
|
|
584 }
|
|
585
|
|
586
|
|
587 static void
|
|
588 ngx_http_rewrite_value_code(ngx_http_rewrite_engine_t *e)
|
|
589 {
|
|
590 ngx_http_rewrite_value_code_t *code;
|
|
591
|
|
592 code = (ngx_http_rewrite_value_code_t *) e->ip;
|
|
593
|
|
594 e->ip += sizeof(ngx_http_rewrite_value_code_t);
|
|
595
|
|
596 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0,
|
|
597 "http rewrite value");
|
|
598
|
|
599 e->sp->value = (ngx_uint_t) code->value;
|
|
600 e->sp->text.len = (size_t) code->text_len;
|
|
601 e->sp->text.data = (u_char *) code->text_data;
|
|
602 e->sp++;
|
|
603 }
|
|
604
|
|
605
|
|
606 static void
|
|
607 ngx_http_rewrite_set_var_code(ngx_http_rewrite_engine_t *e)
|
|
608 {
|
|
609 ngx_http_request_t *r;
|
|
610 ngx_http_variable_value_t *value;
|
|
611 ngx_http_core_main_conf_t *cmcf;
|
|
612 ngx_http_rewrite_var_code_t *code;
|
|
613
|
|
614 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0,
|
|
615 "http rewrite set var");
|
|
616
|
|
617 code = (ngx_http_rewrite_var_code_t *) e->ip;
|
|
618
|
|
619 e->ip += sizeof(ngx_http_rewrite_var_code_t);
|
|
620
|
|
621 r = e->request;
|
|
622
|
|
623 if (r->variables == NULL) {
|
|
624 cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module);
|
|
625
|
|
626 r->variables = ngx_pcalloc(r->pool, cmcf->variables.nelts
|
|
627 * sizeof(ngx_http_variable_value_t *));
|
|
628 if (r->variables == NULL) {
|
|
629 e->ip = ngx_http_rewrite_exit;
|
|
630 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
631 return;
|
|
632 }
|
|
633 }
|
|
634
|
|
635 value = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t));
|
|
636 if (value == NULL) {
|
|
637 e->ip = ngx_http_rewrite_exit;
|
|
638 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
639 return;
|
|
640 }
|
|
641
|
|
642 e->sp--;
|
|
643
|
|
644 *value = *e->sp;
|
|
645
|
|
646 r->variables[code->index] = value;
|
|
647 }
|
|
648
|
|
649
|
|
650 static void
|
|
651 ngx_http_rewrite_var_code(ngx_http_rewrite_engine_t *e)
|
|
652 {
|
|
653 ngx_http_variable_value_t *value;
|
|
654 ngx_http_rewrite_var_code_t *code;
|
|
655
|
|
656 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0,
|
|
657 "http rewrite var");
|
|
658
|
|
659 code = (ngx_http_rewrite_var_code_t *) e->ip;
|
|
660
|
|
661 e->ip += sizeof(ngx_http_rewrite_var_code_t);
|
|
662
|
|
663 value = ngx_http_get_indexed_variable(e->request, code->index);
|
|
664
|
|
665 if (value == NULL || value == NGX_HTTP_VARIABLE_NOT_FOUND) {
|
|
666 e->sp->value = 0;
|
|
667 e->sp->text.len = 0;
|
|
668 e->sp->text.data = (u_char *) "";
|
|
669 e->sp++;
|
|
670
|
|
671 return;
|
|
672 }
|
|
673
|
|
674 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0,
|
|
675 "http rewrite var: %ui, \"%V\"", value->value, &value->text);
|
|
676
|
|
677 *e->sp = *value;
|
|
678 e->sp++;
|
|
679 }
|
|
680
|
|
681
|
|
682 static void
|
|
683 ngx_http_rewrite_invalid_referer_code(ngx_http_rewrite_engine_t *e)
|
|
684 {
|
|
685 u_char *ref;
|
|
686 size_t len;
|
|
687 ngx_uint_t i, n;
|
|
688 ngx_http_request_t *r;
|
|
689 ngx_http_rewrite_referer_t *refs;
|
|
690 ngx_http_rewrite_loc_conf_t *cf;
|
|
691
|
|
692 r = e->request;
|
|
693
|
|
694 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
695 "http rewrite invalid referer");
|
|
696
|
|
697 cf = ngx_http_get_module_loc_conf(r, ngx_http_rewrite_module);
|
|
698
|
|
699 e->ip += sizeof(uintptr_t);
|
|
700
|
|
701 if (cf->referers == NULL) {
|
|
702 e->sp->value = 0;
|
|
703 e->sp->text.len = 0;
|
|
704 e->sp->text.data = (u_char *) "";
|
|
705 e->sp++;
|
|
706
|
|
707 return;
|
|
708 }
|
|
709
|
|
710 if (r->headers_in.referer == NULL) {
|
|
711 if (cf->no_referer) {
|
|
712 e->sp->value = 0;
|
|
713 e->sp->text.len = 0;
|
|
714 e->sp->text.data = (u_char *) "";
|
|
715 e->sp++;
|
|
716
|
|
717 return;
|
|
718 } else {
|
|
719 e->sp->value = 1;
|
|
720 e->sp->text.len = 1;
|
|
721 e->sp->text.data = (u_char *) "1";
|
|
722 e->sp++;
|
|
723
|
|
724 return;
|
|
725 }
|
|
726 }
|
|
727
|
|
728 len = r->headers_in.referer->value.len;
|
|
729 ref = r->headers_in.referer->value.data;
|
|
730
|
|
731 if (len < sizeof("http://i.ru") - 1
|
|
732 || (ngx_strncasecmp(ref, "http://", 7) != 0))
|
|
733 {
|
|
734 e->sp->value = 1;
|
|
735 e->sp->text.len = 1;
|
|
736 e->sp->text.data = (u_char *) "1";
|
|
737 e->sp++;
|
|
738
|
|
739 return;
|
|
740 }
|
|
741
|
|
742 len -= 7;
|
|
743 ref += 7;
|
|
744
|
|
745 refs = cf->referers->elts;
|
|
746 for (i = 0; i < cf->referers->nelts; i++ ){
|
|
747
|
|
748 if (refs[i].name.len > len) {
|
|
749 continue;
|
|
750 }
|
|
751
|
|
752 if (refs[i].wildcard) {
|
|
753 for (n = 0; n < len; n++) {
|
|
754 if (ref[n] == '/' || ref[n] == ':') {
|
|
755 break;
|
|
756 }
|
|
757
|
|
758 if (ref[n] != '.') {
|
|
759 continue;
|
|
760 }
|
|
761
|
|
762 if (ngx_strncmp(&ref[n], refs[i].name.data,
|
|
763 refs[i].name.len) == 0)
|
|
764 {
|
|
765 e->sp->value = 0;
|
|
766 e->sp->text.len = 0;
|
|
767 e->sp->text.data = (u_char *) "";
|
|
768 e->sp++;
|
|
769
|
|
770 return;
|
|
771 }
|
|
772 }
|
|
773
|
|
774 } else {
|
|
775 if (ngx_strncasecmp(refs[i].name.data, ref, refs[i].name.len) == 0)
|
|
776 {
|
|
777 e->sp->value = 0;
|
|
778 e->sp->text.len = 0;
|
|
779 e->sp->text.data = (u_char *) "";
|
|
780 e->sp++;
|
|
781
|
|
782 return;
|
|
783 }
|
|
784 }
|
|
785 }
|
|
786
|
|
787 e->sp->value = 1;
|
|
788 e->sp->text.len = 1;
|
|
789 e->sp->text.data = (u_char *) "1";
|
|
790 e->sp++;
|
|
791 }
|
|
792
|
|
793
|
|
794 static void
|
|
795 ngx_http_rewrite_nop_code(ngx_http_rewrite_engine_t *e)
|
|
796 {
|
|
797 e->ip += sizeof(uintptr_t);
|
|
798 }
|
|
799
|
|
800
|
|
801 static ngx_http_variable_value_t *
|
|
802 ngx_http_rewrite_var(ngx_http_request_t *r, uintptr_t data)
|
|
803 {
|
|
804 ngx_http_variable_t *var;
|
|
805 ngx_http_core_main_conf_t *cmcf;
|
|
806
|
|
807 cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module);
|
|
808
|
|
809 var = cmcf->variables.elts;
|
|
810
|
|
811 /*
|
|
812 * the ngx_http_rewrite_module sets variables directly in r->variables,
|
|
813 * and they should be handle by ngx_http_get_indexed_variable(),
|
|
814 * so the handler is called only if the variable is not initialized
|
|
815 */
|
|
816
|
|
817 ngx_log_error(NGX_LOG_ALERT, r->connection->log, 0,
|
|
818 "using uninitialized \"%V\" variable", &var[data].name);
|
|
819
|
|
820 return &ngx_http_rewrite_null_value;
|
|
821 }
|
|
822
|
|
823
|
|
824 static ngx_int_t
|
|
825 ngx_http_rewrite_init(ngx_cycle_t *cycle)
|
|
826 {
|
|
827 ngx_http_handler_pt *h;
|
|
828 ngx_http_core_main_conf_t *cmcf;
|
|
829
|
|
830 cmcf = ngx_http_cycle_get_module_main_conf(cycle, ngx_http_core_module);
|
|
831
|
|
832 h = ngx_array_push(&cmcf->phases[NGX_HTTP_REWRITE_PHASE].handlers);
|
|
833 if (h == NULL) {
|
|
834 return NGX_ERROR;
|
|
835 }
|
|
836
|
|
837 *h = ngx_http_rewrite_handler;
|
|
838
|
|
839 return NGX_OK;
|
|
840 }
|
|
841
|
|
842
|
|
843 static void *
|
|
844 ngx_http_rewrite_create_loc_conf(ngx_conf_t *cf)
|
|
845 {
|
|
846 ngx_http_rewrite_loc_conf_t *conf;
|
|
847
|
|
848 conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_rewrite_loc_conf_t));
|
|
849 if (conf == NULL) {
|
|
850 return NGX_CONF_ERROR;
|
|
851 }
|
|
852
|
|
853 conf->stack_size = NGX_CONF_UNSET_UINT;
|
|
854 conf->log = NGX_CONF_UNSET;
|
|
855 conf->no_referer = NGX_CONF_UNSET;
|
|
856
|
|
857 return conf;
|
|
858 }
|
|
859
|
|
860
|
|
861 static char *
|
|
862 ngx_http_rewrite_merge_loc_conf(ngx_conf_t *cf, void *parent, void *child)
|
|
863 {
|
|
864 ngx_http_rewrite_loc_conf_t *prev = parent;
|
|
865 ngx_http_rewrite_loc_conf_t *conf = child;
|
|
866
|
|
867 uintptr_t *code, *last;
|
|
868 ngx_http_rewrite_regex_code_t *regex;
|
|
869
|
|
870 ngx_conf_merge_value(conf->log, prev->log, 0);
|
|
871 ngx_conf_merge_unsigned_value(conf->stack_size, prev->stack_size, 10);
|
|
872
|
|
873 if (conf->referers == NULL) {
|
|
874 conf->referers = prev->referers;
|
|
875 ngx_conf_merge_value(conf->no_referer, prev->no_referer, 0);
|
|
876 }
|
|
877
|
|
878 if (conf->no_referer == NGX_CONF_UNSET) {
|
|
879 conf->no_referer = 0;
|
|
880 }
|
|
881
|
|
882 if (conf->codes == NULL) {
|
|
883 return NGX_CONF_OK;
|
|
884 }
|
|
885
|
|
886 if (conf->codes == prev->codes) {
|
|
887 return NGX_CONF_OK;
|
|
888 }
|
|
889
|
|
890 code = conf->codes->elts;
|
|
891 last = (uintptr_t *) ((u_char *) code + conf->codes->nelts);
|
|
892
|
|
893 while (code < last) {
|
|
894 if (*code == (uintptr_t) NULL) {
|
|
895 return NGX_CONF_OK;
|
|
896 }
|
|
897
|
|
898 if (*code == (uintptr_t) &ngx_http_rewrite_regex_start_code) {
|
|
899 regex = (ngx_http_rewrite_regex_code_t *) code;
|
|
900 if (conf->max_captures < regex->ncaptures) {
|
|
901 conf->max_captures = regex->ncaptures;
|
|
902 }
|
|
903 code = (uintptr_t *) ((u_char *) code + regex->next);
|
|
904 continue;
|
|
905 }
|
|
906
|
|
907 if (*code == (uintptr_t) &ngx_http_rewrite_if_code) {
|
|
908 code += sizeof(ngx_http_rewrite_if_code_t) / sizeof(uintptr_t);
|
|
909 continue;
|
|
910 }
|
|
911
|
|
912 if (*code == (uintptr_t) &ngx_http_rewrite_return_code) {
|
|
913 code += sizeof(ngx_http_rewrite_return_code_t) / sizeof(uintptr_t);
|
|
914 continue;
|
|
915 }
|
|
916
|
|
917 if (*code == (uintptr_t) &ngx_http_rewrite_set_var_code) {
|
|
918 code += sizeof(ngx_http_rewrite_var_code_t) / sizeof(uintptr_t);
|
|
919 continue;
|
|
920 }
|
|
921
|
|
922 if (*code == (uintptr_t) &ngx_http_rewrite_var_code) {
|
|
923 code += sizeof(ngx_http_rewrite_var_code_t) / sizeof(uintptr_t);
|
|
924 continue;
|
|
925 }
|
|
926
|
|
927 if (*code == (uintptr_t) &ngx_http_rewrite_value_code) {
|
|
928 code += sizeof(ngx_http_rewrite_value_code_t) / sizeof(uintptr_t);
|
|
929 continue;
|
|
930 }
|
|
931
|
|
932 if (*code == (uintptr_t) &ngx_http_rewrite_invalid_referer_code) {
|
|
933 code++;
|
|
934 continue;
|
|
935 }
|
|
936
|
|
937 if (*code == (uintptr_t) &ngx_http_rewrite_nop_code) {
|
|
938 code++;
|
|
939 continue;
|
|
940 }
|
|
941
|
|
942 #if (NGX_DEBUG)
|
|
943 ngx_log_error(NGX_LOG_EMERG, cf->log, 0,
|
|
944 "unknown rewrite code: %p", *code);
|
|
945 return NGX_CONF_ERROR;
|
|
946 #endif
|
|
947 }
|
|
948
|
|
949 code = ngx_array_push_n(conf->codes, sizeof(uintptr_t));
|
|
950 if (code == NULL) {
|
|
951 return NGX_CONF_ERROR;
|
|
952 }
|
|
953
|
|
954 *code = (uintptr_t) NULL;
|
|
955
|
|
956 return NGX_CONF_OK;
|
|
957 }
|
|
958
|
|
959
|
|
960 static char *
|
|
961 ngx_http_rewrite(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
962 {
|
|
963 ngx_http_rewrite_loc_conf_t *lcf = conf;
|
|
964
|
|
965 u_char *data;
|
|
966 size_t len, size;
|
|
967 ngx_str_t *value, err;
|
|
968 ngx_int_t n;
|
|
969 ngx_uint_t i, last;
|
|
970 ngx_http_rewrite_code_pt *code;
|
|
971 ngx_http_rewrite_copy_code_t *copy;
|
|
972 ngx_http_rewrite_regex_code_t *regex;
|
|
973 ngx_http_rewrite_regex_end_code_t *regex_end;
|
|
974 ngx_http_rewrite_copy_capture_code_t *copy_capture;
|
|
975 u_char errstr[NGX_MAX_CONF_ERRSTR];
|
|
976
|
|
977 regex = ngx_http_rewrite_start_code(cf->pool, &lcf->codes,
|
|
978 sizeof(ngx_http_rewrite_regex_code_t));
|
|
979 if (regex == NULL) {
|
|
980 return NGX_CONF_ERROR;
|
|
981 }
|
|
982
|
|
983 value = cf->args->elts;
|
|
984
|
|
985 err.len = NGX_MAX_CONF_ERRSTR;
|
|
986 err.data = errstr;
|
|
987
|
|
988 /* TODO: NGX_REGEX_CASELESS */
|
|
989
|
|
990 regex->regex = ngx_regex_compile(&value[1], 0, cf->pool, &err);
|
|
991
|
|
992 if (regex->regex == NULL) {
|
|
993 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "%s", err.data);
|
|
994 return NGX_CONF_ERROR;
|
|
995 }
|
|
996
|
|
997 regex->code = ngx_http_rewrite_regex_start_code;
|
|
998 regex->size = 0;
|
|
999 regex->ncaptures = 0;
|
|
1000 regex->status = 0;
|
|
1001 regex->test = 0;
|
|
1002 regex->uri = 1;
|
|
1003 regex->args = 1;
|
|
1004 regex->redirect = 0;
|
|
1005 regex->name = value[1];
|
|
1006
|
|
1007 last = 0;
|
|
1008
|
|
1009 if (ngx_strncmp(value[2].data, "http://", sizeof("http://") - 1) == 0) {
|
|
1010 regex->status = NGX_HTTP_MOVED_TEMPORARILY;
|
|
1011 regex->redirect = 1;
|
|
1012 last = 1;
|
|
1013 }
|
|
1014
|
|
1015 if (cf->args->nelts == 4) {
|
|
1016 if (ngx_strcmp(value[3].data, "last") == 0) {
|
|
1017 last = 1;
|
|
1018
|
|
1019 } else if (ngx_strcmp(value[3].data, "break") == 0) {
|
|
1020 regex->break_cycle = 1;
|
|
1021 last = 1;
|
|
1022
|
|
1023 } else if (ngx_strcmp(value[3].data, "redirect") == 0) {
|
|
1024 regex->status = NGX_HTTP_MOVED_TEMPORARILY;
|
|
1025 regex->redirect = 1;
|
|
1026 last = 1;
|
|
1027
|
|
1028 } else if (ngx_strcmp(value[3].data, "permanent") == 0) {
|
|
1029 regex->status = NGX_HTTP_MOVED_PERMANENTLY;
|
|
1030 regex->redirect = 1;
|
|
1031 last = 1;
|
|
1032
|
|
1033 } else {
|
|
1034 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1035 "invalid parameter \"%V\"", &value[3]);
|
|
1036 return NGX_CONF_ERROR;
|
|
1037 }
|
|
1038 }
|
|
1039
|
|
1040 i = 0;
|
|
1041
|
|
1042 while (i < value[2].len) {
|
|
1043
|
|
1044 data = &value[2].data[i];
|
|
1045
|
|
1046 if (value[2].data[i] == '$' && i < value[2].len
|
|
1047 && value[2].data[i + 1] >= '1' && value[2].data[i + 1] <= '9')
|
|
1048 {
|
|
1049
|
|
1050 /* the "$1" - "$9" captures */
|
|
1051
|
|
1052 copy_capture = ngx_http_rewrite_add_code(lcf->codes,
|
|
1053 sizeof(ngx_http_rewrite_copy_capture_code_t),
|
|
1054 ®ex);
|
|
1055 if (copy_capture == NULL) {
|
|
1056 return NGX_CONF_ERROR;
|
|
1057 }
|
|
1058
|
|
1059 i++;
|
|
1060
|
|
1061 copy_capture->code = ngx_http_rewrite_copy_capture_code;
|
|
1062 copy_capture->n = value[2].data[i] - '0';
|
|
1063
|
|
1064 if (regex->ncaptures < copy_capture->n) {
|
|
1065 regex->ncaptures = copy_capture->n;
|
|
1066 }
|
|
1067
|
|
1068 copy_capture->n *= 2;
|
|
1069
|
|
1070 i++;
|
|
1071
|
|
1072 continue;
|
|
1073 }
|
|
1074
|
|
1075 if (value[2].data[i] == '?') {
|
|
1076
|
|
1077 /* the arguments */
|
|
1078
|
|
1079 if (i == value[2].len - 1) {
|
|
1080 /* the last "?" drops the original arguments */
|
|
1081 regex->args = 0;
|
|
1082 break;
|
|
1083 }
|
|
1084
|
|
1085 if (!regex->redirect) {
|
|
1086 code = ngx_http_rewrite_add_code(lcf->codes, sizeof(uintptr_t),
|
|
1087 ®ex);
|
|
1088 if (code == NULL) {
|
|
1089 return NGX_CONF_ERROR;
|
|
1090 }
|
|
1091
|
|
1092 *code = ngx_http_rewrite_start_args_code;
|
|
1093
|
|
1094 i++;
|
|
1095
|
|
1096 continue;
|
|
1097 }
|
|
1098 }
|
|
1099
|
|
1100 i++;
|
|
1101
|
|
1102 /* the substituion strings */
|
|
1103
|
|
1104 while (i < value[2].len && value[2].data[i] != '$') {
|
|
1105
|
|
1106 if (value[2].data[i] == '?') {
|
|
1107
|
|
1108 if (i == value[2].len - 1) {
|
|
1109 /*
|
|
1110 * the last "?" drops the original arguments,
|
|
1111 * and it should not be copied to a substituion
|
|
1112 */
|
|
1113 regex->args = 0;
|
|
1114 break;
|
|
1115 }
|
|
1116
|
|
1117 if (!regex->redirect) {
|
|
1118 break;
|
|
1119 }
|
|
1120 }
|
|
1121
|
|
1122 i++;
|
|
1123 }
|
|
1124
|
|
1125 len = &value[2].data[i] - data;
|
|
1126
|
|
1127 if (len == 0) {
|
|
1128 continue;
|
|
1129 }
|
|
1130
|
|
1131 regex->size += len;
|
|
1132
|
|
1133 size = (len + sizeof(uintptr_t) - 1) & ~(sizeof(uintptr_t) - 1);
|
|
1134
|
|
1135 copy = ngx_http_rewrite_add_code(lcf->codes,
|
|
1136 sizeof(ngx_http_rewrite_copy_code_t) + size,
|
|
1137 ®ex);
|
|
1138 if (copy == NULL) {
|
|
1139 return NGX_CONF_ERROR;
|
|
1140 }
|
|
1141
|
|
1142 copy->code = ngx_http_rewrite_copy_code;
|
|
1143 copy->len = len;
|
|
1144
|
|
1145 ngx_memcpy((u_char *) copy + sizeof(ngx_http_rewrite_copy_code_t),
|
|
1146 data, len);
|
|
1147 }
|
|
1148
|
|
1149 n = ngx_regex_capture_count(regex->regex);
|
|
1150
|
|
1151 if (n < 0) {
|
|
1152 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1153 ngx_regex_capture_count_n " failed for "
|
|
1154 "pattern \"%V\"", &value[1]);
|
|
1155 return NGX_CONF_ERROR;
|
|
1156 }
|
|
1157
|
|
1158 if (regex->ncaptures > (ngx_uint_t) n) {
|
|
1159 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1160 "pattern \"%V\" has less captures "
|
|
1161 "than referrenced in substitution \"%V\"",
|
|
1162 &value[1], &value[2]);
|
|
1163 return NGX_CONF_ERROR;
|
|
1164 }
|
|
1165
|
|
1166 if (regex->ncaptures < (ngx_uint_t) n) {
|
|
1167 regex->ncaptures = (ngx_uint_t) n;
|
|
1168 }
|
|
1169
|
|
1170 if (regex->ncaptures) {
|
|
1171 regex->ncaptures = (regex->ncaptures + 1) * 3;
|
|
1172 }
|
|
1173
|
|
1174 regex_end = ngx_http_rewrite_add_code(lcf->codes,
|
|
1175 sizeof(ngx_http_rewrite_regex_end_code_t),
|
|
1176 ®ex);
|
|
1177 if (regex_end == NULL) {
|
|
1178 return NGX_CONF_ERROR;
|
|
1179 }
|
|
1180
|
|
1181 regex_end->code = ngx_http_rewrite_regex_end_code;
|
|
1182 regex_end->uri = regex->uri;
|
|
1183 regex_end->args = regex->args;
|
|
1184 regex_end->redirect = regex->redirect;
|
|
1185
|
|
1186 if (last) {
|
|
1187 code = ngx_http_rewrite_add_code(lcf->codes, sizeof(uintptr_t),
|
|
1188 ®ex);
|
|
1189 if (code == NULL) {
|
|
1190 return NGX_CONF_ERROR;
|
|
1191 }
|
|
1192
|
|
1193 *code = (uintptr_t) NULL;
|
|
1194 }
|
|
1195
|
|
1196 regex->next = (u_char *) lcf->codes->elts + lcf->codes->nelts
|
|
1197 - (u_char *) regex;
|
|
1198
|
|
1199 return NGX_CONF_OK;
|
|
1200 }
|
|
1201
|
|
1202
|
|
1203 static char *
|
|
1204 ngx_http_rewrite_return(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
1205 {
|
|
1206 ngx_http_rewrite_loc_conf_t *lcf = conf;
|
|
1207
|
|
1208 ngx_str_t *value;
|
|
1209 ngx_http_rewrite_return_code_t *ret;
|
|
1210
|
|
1211 ret = ngx_http_rewrite_start_code(cf->pool, &lcf->codes,
|
|
1212 sizeof(ngx_http_rewrite_return_code_t));
|
|
1213 if (ret == NULL) {
|
|
1214 return NGX_CONF_ERROR;
|
|
1215 }
|
|
1216
|
|
1217 value = cf->args->elts;
|
|
1218
|
|
1219 ret->code = ngx_http_rewrite_return_code;
|
|
1220 ret->null = (uintptr_t) NULL;
|
|
1221
|
|
1222 ret->status = ngx_atoi(value[1].data, value[1].len);
|
|
1223
|
|
1224 if (ret->status == (uintptr_t) NGX_ERROR) {
|
|
1225 return NGX_CONF_ERROR;
|
|
1226 }
|
|
1227
|
|
1228 return NGX_CONF_OK;
|
|
1229 }
|
|
1230
|
|
1231
|
|
1232 static char *
|
|
1233 ngx_http_rewrite_if(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
1234 {
|
|
1235 ngx_http_rewrite_loc_conf_t *lcf = conf;
|
|
1236
|
|
1237 void *mconf;
|
|
1238 char *rv;
|
|
1239 u_char *elts;
|
|
1240 ngx_uint_t i;
|
|
1241 ngx_conf_t save;
|
|
1242 ngx_http_module_t *module;
|
|
1243 ngx_http_conf_ctx_t *ctx, *pctx;
|
|
1244 ngx_http_core_loc_conf_t *clcf, *pclcf, **clcfp;
|
|
1245 ngx_http_rewrite_if_code_t *if_code;
|
|
1246 ngx_http_rewrite_loc_conf_t *nlcf;
|
|
1247
|
|
1248 ctx = ngx_pcalloc(cf->pool, sizeof(ngx_http_conf_ctx_t));
|
|
1249 if (ctx == NULL) {
|
|
1250 return NGX_CONF_ERROR;
|
|
1251 }
|
|
1252
|
|
1253 pctx = cf->ctx;
|
|
1254 ctx->main_conf = pctx->main_conf;
|
|
1255 ctx->srv_conf = pctx->srv_conf;
|
|
1256
|
|
1257 ctx->loc_conf = ngx_pcalloc(cf->pool, sizeof(void *) * ngx_http_max_module);
|
|
1258 if (ctx->loc_conf == NULL) {
|
|
1259 return NGX_CONF_ERROR;
|
|
1260 }
|
|
1261
|
|
1262 for (i = 0; ngx_modules[i]; i++) {
|
|
1263 if (ngx_modules[i]->type != NGX_HTTP_MODULE) {
|
|
1264 continue;
|
|
1265 }
|
|
1266
|
|
1267 module = ngx_modules[i]->ctx;
|
|
1268
|
|
1269 if (module->create_loc_conf) {
|
|
1270
|
|
1271 mconf = module->create_loc_conf(cf);
|
|
1272 if (mconf == NULL) {
|
|
1273 return NGX_CONF_ERROR;
|
|
1274 }
|
|
1275
|
|
1276 ctx->loc_conf[ngx_modules[i]->ctx_index] = mconf;
|
|
1277 }
|
|
1278 }
|
|
1279
|
|
1280 pclcf = pctx->loc_conf[ngx_http_core_module.ctx_index];
|
|
1281
|
|
1282 clcf = ctx->loc_conf[ngx_http_core_module.ctx_index];
|
|
1283 clcf->loc_conf = ctx->loc_conf;
|
|
1284 clcf->name = pclcf->name;
|
|
1285 clcf->noname = 1;
|
|
1286
|
|
1287 if (pclcf->locations.elts == NULL) {
|
|
1288 if (ngx_array_init(&pclcf->locations, cf->pool, 4, sizeof(void *))
|
|
1289 == NGX_ERROR)
|
|
1290 {
|
|
1291 return NGX_CONF_ERROR;
|
|
1292 }
|
|
1293 }
|
|
1294
|
|
1295 clcfp = ngx_array_push(&pclcf->locations);
|
|
1296 if (clcfp == NULL) {
|
|
1297 return NGX_CONF_ERROR;
|
|
1298 }
|
|
1299
|
|
1300 *clcfp = clcf;
|
|
1301
|
|
1302
|
|
1303 if (ngx_http_rewrite_if_condition(cf, lcf) != NGX_CONF_OK) {
|
|
1304 return NGX_CONF_ERROR;
|
|
1305 }
|
|
1306
|
|
1307 if_code = ngx_array_push_n(lcf->codes, sizeof(ngx_http_rewrite_if_code_t));
|
|
1308 if (if_code == NULL) {
|
|
1309 return NULL;
|
|
1310 }
|
|
1311
|
|
1312 if_code->code = ngx_http_rewrite_if_code;
|
|
1313
|
|
1314 elts = lcf->codes->elts;
|
|
1315
|
|
1316
|
|
1317 /* the inside directives must compile to the same code array */
|
|
1318
|
|
1319 nlcf = ctx->loc_conf[ngx_http_rewrite_module.ctx_index];
|
|
1320 nlcf->codes = lcf->codes;
|
|
1321
|
|
1322
|
|
1323 save = *cf;
|
|
1324 cf->ctx = ctx;
|
|
1325
|
|
1326 if (pclcf->name.len == 0) {
|
|
1327 if_code->loc_conf = NULL;
|
|
1328 cf->cmd_type = NGX_HTTP_SIF_CONF;
|
|
1329
|
|
1330 } else {
|
|
1331 if_code->loc_conf = ctx->loc_conf;
|
|
1332 cf->cmd_type = NGX_HTTP_LIF_CONF;
|
|
1333 }
|
|
1334
|
|
1335 rv = ngx_conf_parse(cf, NULL);
|
|
1336
|
|
1337 *cf = save;
|
|
1338
|
|
1339 if (rv != NGX_CONF_OK) {
|
|
1340 return rv;
|
|
1341 }
|
|
1342
|
|
1343
|
|
1344 if (elts != lcf->codes->elts) {
|
|
1345 if_code = (ngx_http_rewrite_if_code_t *)
|
|
1346 ((u_char *) if_code + ((u_char *) lcf->codes->elts - elts));
|
|
1347 }
|
|
1348
|
|
1349 if_code->next = (u_char *) lcf->codes->elts + lcf->codes->nelts
|
|
1350 - (u_char *) if_code;
|
|
1351
|
|
1352 return NGX_CONF_OK;
|
|
1353 }
|
|
1354
|
|
1355
|
|
1356 static char *
|
|
1357 ngx_http_rewrite_if_condition(ngx_conf_t *cf, ngx_http_rewrite_loc_conf_t *lcf)
|
|
1358 {
|
|
1359 ngx_str_t *value, err;
|
|
1360 ngx_uint_t cur, last;
|
|
1361 ngx_http_rewrite_regex_code_t *regex;
|
|
1362 u_char errstr[NGX_MAX_CONF_ERRSTR];
|
|
1363
|
|
1364 value = cf->args->elts;
|
|
1365 last = cf->args->nelts - 1;
|
|
1366
|
|
1367 if (value[1].len < 1 || value[1].data[0] != '(') {
|
|
1368 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1369 "invalid condition \"%V\"", &value[1]);
|
|
1370 return NGX_CONF_ERROR;
|
|
1371 }
|
|
1372
|
|
1373 if (value[1].len == 1) {
|
|
1374 cur = 2;
|
|
1375
|
|
1376 } else {
|
|
1377 cur = 1;
|
|
1378 value[1].len--;
|
|
1379 value[1].data++;
|
|
1380 }
|
|
1381
|
|
1382 if (value[last].len < 1 || value[last].data[value[last].len - 1] != ')') {
|
|
1383 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1384 "invalid condition \"%V\"", &value[last]);
|
|
1385 return NGX_CONF_ERROR;
|
|
1386 }
|
|
1387
|
|
1388 if (value[last].len == 1) {
|
|
1389 last--;
|
|
1390
|
|
1391 } else {
|
|
1392 value[last].len--;
|
|
1393 value[last].data[value[last].len] = '\0';
|
|
1394 }
|
|
1395
|
|
1396 if (value[cur].len > 1 && value[cur].data[0] == '$') {
|
|
1397
|
|
1398 if (cur != last && cur + 2 != last) {
|
|
1399 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1400 "invalid condition \"%V\"", &value[cur]);
|
|
1401 return NGX_CONF_ERROR;
|
|
1402 }
|
|
1403
|
|
1404 if (ngx_http_rewrite_variable(cf, lcf, &value[cur])!= NGX_CONF_OK) {
|
|
1405 return NGX_CONF_ERROR;
|
|
1406 }
|
|
1407
|
|
1408 if (cur == last) {
|
|
1409 return NGX_CONF_OK;
|
|
1410 }
|
|
1411
|
|
1412 cur++;
|
|
1413
|
|
1414 if ((value[cur].len == 1 && value[cur].data[0] != '~')
|
|
1415 || (value[cur].len == 2
|
|
1416 && value[cur].data[0] != '~' && value[cur].data[1] != '*'))
|
|
1417 {
|
|
1418 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1419 "unexpected \"%V\" in condition", &value[cur]);
|
|
1420 return NGX_CONF_ERROR;
|
|
1421 }
|
|
1422
|
|
1423 regex = ngx_http_rewrite_start_code(cf->pool, &lcf->codes,
|
|
1424 sizeof(ngx_http_rewrite_regex_code_t));
|
|
1425 if (regex == NULL) {
|
|
1426 return NGX_CONF_ERROR;
|
|
1427 }
|
|
1428
|
|
1429 err.len = NGX_MAX_CONF_ERRSTR;
|
|
1430 err.data = errstr;
|
|
1431
|
|
1432 regex->regex = ngx_regex_compile(&value[last],
|
|
1433 (value[cur].len == 2) ? NGX_REGEX_CASELESS : 0,
|
|
1434 cf->pool, &err);
|
|
1435
|
|
1436 if (regex->regex == NULL) {
|
|
1437 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "%s", err.data);
|
|
1438 return NGX_CONF_ERROR;
|
|
1439 }
|
|
1440
|
|
1441 regex->code = ngx_http_rewrite_regex_start_code;
|
|
1442 regex->size = 0;
|
|
1443 regex->ncaptures = 0;
|
|
1444 regex->status = 0;
|
|
1445 regex->next = sizeof(ngx_http_rewrite_regex_code_t);
|
|
1446 regex->test = 1;
|
|
1447 regex->uri = 0;
|
|
1448 regex->args = 0;
|
|
1449 regex->redirect = 0;
|
|
1450 regex->name = value[last];
|
|
1451
|
|
1452 return NGX_CONF_OK;
|
|
1453 }
|
|
1454
|
|
1455 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1456 "invalid condition \"%V\"", &value[cur]);
|
|
1457
|
|
1458 return NGX_CONF_ERROR;
|
|
1459 }
|
|
1460
|
|
1461
|
|
1462 static char *
|
|
1463 ngx_http_rewrite_variable(ngx_conf_t *cf, ngx_http_rewrite_loc_conf_t *lcf,
|
|
1464 ngx_str_t *value)
|
|
1465 {
|
|
1466 ngx_http_variable_t *var;
|
|
1467 ngx_http_rewrite_code_pt *code;
|
|
1468 ngx_http_rewrite_var_code_t *var_code;
|
|
1469
|
|
1470 value->len--;
|
|
1471 value->data++;
|
|
1472
|
|
1473 if (value->len == sizeof("invalid_referer") - 1
|
|
1474 && ngx_strncmp(value->data, "invalid_referer",
|
|
1475 sizeof("invalid_referer") - 1) == 0)
|
|
1476 {
|
|
1477 code = ngx_http_rewrite_start_code(cf->pool, &lcf->codes,
|
|
1478 sizeof(ngx_http_rewrite_code_pt));
|
|
1479 if (code == NULL) {
|
|
1480 return NGX_CONF_ERROR;
|
|
1481 }
|
|
1482
|
|
1483 *code = ngx_http_rewrite_invalid_referer_code;
|
|
1484
|
|
1485 } else {
|
|
1486 var = ngx_http_add_variable(cf, value, 0);
|
|
1487
|
|
1488 if (var == NULL) {
|
|
1489 return NGX_CONF_ERROR;
|
|
1490 }
|
|
1491
|
|
1492 var_code = ngx_http_rewrite_start_code(cf->pool, &lcf->codes,
|
|
1493 sizeof(ngx_http_rewrite_var_code_t));
|
|
1494 if (var_code == NULL) {
|
|
1495 return NGX_CONF_ERROR;
|
|
1496 }
|
|
1497
|
|
1498 var_code->code = ngx_http_rewrite_var_code;
|
|
1499 var_code->index = var->index;
|
|
1500 }
|
|
1501
|
|
1502 return NGX_CONF_OK;
|
|
1503 }
|
|
1504
|
|
1505
|
|
1506 static char *
|
|
1507 ngx_http_rewrite_valid_referers(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
1508 {
|
|
1509 ngx_http_rewrite_loc_conf_t *lcf = conf;
|
|
1510
|
|
1511 ngx_uint_t i, server_names;
|
|
1512 ngx_str_t *value;
|
|
1513 ngx_http_server_name_t *sn;
|
|
1514 ngx_http_core_srv_conf_t *cscf;
|
|
1515 ngx_http_rewrite_referer_t *ref;
|
|
1516
|
|
1517 cscf = ngx_http_conf_get_module_srv_conf(cf, ngx_http_core_module);
|
|
1518
|
|
1519 if (lcf->referers == NULL) {
|
|
1520 lcf->referers = ngx_array_create(cf->pool,
|
|
1521 cf->args->nelts + cscf->server_names.nelts,
|
|
1522 sizeof(ngx_http_rewrite_referer_t));
|
|
1523 if (lcf->referers == NULL) {
|
|
1524 return NGX_CONF_ERROR;
|
|
1525 }
|
|
1526 }
|
|
1527
|
|
1528 value = cf->args->elts;
|
|
1529 server_names = 0;
|
|
1530
|
|
1531 for (i = 1; i < cf->args->nelts; i++) {
|
|
1532 if (value[i].len == 0) {
|
|
1533 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1534 "invalid referer \"%V\"", &value[i]);
|
|
1535 return NGX_CONF_ERROR;
|
|
1536 }
|
|
1537
|
|
1538 if (ngx_strcmp(value[i].data, "none") == 0) {
|
|
1539 lcf->no_referer = 1;
|
|
1540 continue;
|
|
1541 }
|
|
1542
|
|
1543 if (ngx_strcmp(value[i].data, "server_names") == 0) {
|
|
1544 server_names = 1;
|
|
1545 continue;
|
|
1546 }
|
|
1547
|
|
1548 ref = ngx_array_push(lcf->referers);
|
|
1549 if (ref == NULL) {
|
|
1550 return NGX_CONF_ERROR;
|
|
1551 }
|
|
1552
|
|
1553 if (value[i].data[0] != '*') {
|
|
1554 ref->name = value[i];
|
|
1555 ref->wildcard = 0;
|
|
1556 continue;
|
|
1557 }
|
|
1558
|
|
1559 if (value[i].data[1] != '.') {
|
|
1560 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1561 "invalid wildcard referer \"%V\"", &value[i]);
|
|
1562 return NGX_CONF_ERROR;
|
|
1563 }
|
|
1564
|
|
1565 ref->name.len = value[i].len - 1;
|
|
1566 ref->name.data = value[i].data + 1;
|
|
1567 ref->wildcard = 1;
|
|
1568 }
|
|
1569
|
|
1570 if (!server_names) {
|
|
1571 return NGX_CONF_OK;
|
|
1572 }
|
|
1573
|
|
1574 sn = cscf->server_names.elts;
|
|
1575 for (i = 0; i < cscf->server_names.nelts; i++) {
|
|
1576 ref = ngx_array_push(lcf->referers);
|
|
1577 if (ref == NULL) {
|
|
1578 return NGX_CONF_ERROR;
|
|
1579 }
|
|
1580
|
|
1581 ref->name.len = sn[i].name.len + 1;
|
|
1582 ref->name.data = ngx_palloc(cf->pool, ref->name.len);
|
|
1583 if (ref->name.data == NULL) {
|
|
1584 return NGX_CONF_ERROR;
|
|
1585 }
|
|
1586
|
|
1587 ngx_memcpy(ref->name.data, sn[i].name.data, sn[i].name.len);
|
|
1588 ref->name.data[sn[i].name.len] = '/';
|
|
1589 ref->wildcard = sn[i].wildcard;
|
|
1590 }
|
|
1591
|
|
1592 return NGX_CONF_OK;
|
|
1593 }
|
|
1594
|
|
1595
|
|
1596 static char *
|
|
1597 ngx_http_rewrite_set(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
1598 {
|
|
1599 ngx_http_rewrite_loc_conf_t *lcf = conf;
|
|
1600
|
|
1601 ngx_int_t n;
|
|
1602 ngx_str_t *value;
|
|
1603 ngx_http_variable_t *v;
|
|
1604 ngx_http_rewrite_var_code_t *var;
|
|
1605 ngx_http_rewrite_value_code_t *val;
|
|
1606
|
|
1607 value = cf->args->elts;
|
|
1608
|
|
1609 if (value[1].data[0] != '$') {
|
|
1610 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1611 "invalid variable name \"%V\"", &value[1]);
|
|
1612 return NGX_CONF_ERROR;
|
|
1613 }
|
|
1614
|
|
1615 value[1].len--;
|
|
1616 value[1].data++;
|
|
1617
|
|
1618 v = ngx_http_add_variable(cf, &value[1], 1);
|
|
1619 if (v == NULL) {
|
|
1620 return NGX_CONF_ERROR;
|
|
1621 }
|
|
1622
|
|
1623 v->handler = ngx_http_rewrite_var;
|
|
1624 v->data = v->index;
|
|
1625
|
|
1626 val = ngx_http_rewrite_start_code(cf->pool, &lcf->codes,
|
|
1627 sizeof(ngx_http_rewrite_value_code_t));
|
|
1628 if (val == NULL) {
|
|
1629 return NGX_CONF_ERROR;
|
|
1630 }
|
|
1631
|
|
1632 n = ngx_atoi(value[2].data, value[2].len);
|
|
1633
|
|
1634 if (n == NGX_ERROR) {
|
|
1635 n = 0;
|
|
1636 }
|
|
1637
|
|
1638 val->code = ngx_http_rewrite_value_code;
|
|
1639 val->value = (uintptr_t) n;
|
|
1640 val->text_len = (uintptr_t) value[2].len;
|
|
1641 val->text_data = (uintptr_t) value[2].data;
|
|
1642
|
|
1643 var = ngx_http_rewrite_start_code(cf->pool, &lcf->codes,
|
|
1644 sizeof(ngx_http_rewrite_var_code_t));
|
|
1645 if (var == NULL) {
|
|
1646 return NGX_CONF_ERROR;
|
|
1647 }
|
|
1648
|
|
1649 var->code = ngx_http_rewrite_set_var_code;
|
|
1650 var->index = (uintptr_t) v->index;
|
|
1651
|
|
1652 return NGX_CONF_OK;
|
|
1653 }
|
|
1654
|
|
1655
|
|
1656 static void *
|
|
1657 ngx_http_rewrite_start_code(ngx_pool_t *pool, ngx_array_t **codes, size_t size)
|
|
1658 {
|
|
1659 if (*codes == NULL) {
|
|
1660 *codes = ngx_array_create(pool, 256, 1);
|
|
1661 if (*codes == NULL) {
|
|
1662 return NULL;
|
|
1663 }
|
|
1664 }
|
|
1665
|
|
1666 return ngx_array_push_n(*codes, size);
|
|
1667 }
|
|
1668
|
|
1669
|
|
1670 static void *
|
|
1671 ngx_http_rewrite_add_code(ngx_array_t *codes, size_t size, void *code)
|
|
1672 {
|
|
1673 u_char *elts, **p;
|
|
1674 void *new;
|
|
1675
|
|
1676 elts = codes->elts;
|
|
1677
|
|
1678 new = ngx_array_push_n(codes, size);
|
|
1679 if (new == NULL) {
|
|
1680 return NGX_CONF_ERROR;
|
|
1681 }
|
|
1682
|
|
1683 if (elts != codes->elts) {
|
|
1684 p = code;
|
|
1685 *p += (u_char *) codes->elts - elts;
|
|
1686 }
|
|
1687
|
|
1688 return new;
|
|
1689 }
|