Mercurial > hg > nginx
annotate src/http/ngx_http_script.c @ 623:0216fd1471f3 release-0.3.33
nginx-0.3.33-RELEASE import
*) Feature: the "http_503" parameter of the "proxy_next_upstream" or
"fastcgi_next_upstream" directives.
*) Bugfix: ngx_http_perl_module did not work with inlined in the
configuration code, if it was not started with the "sub" word.
*) Bugfix: in the "post_action" directive.
| author | Igor Sysoev <igor@sysoev.ru> |
|---|---|
| date | Wed, 15 Mar 2006 09:53:04 +0000 |
| parents | 77f77f53214a |
| children | 5d2b8078c1c2 |
| rev | line source |
|---|---|
|
441
da8c5707af39
nginx-0.1.0-2004-09-28-12:34:51 import; set copyright and remove unused files
Igor Sysoev <igor@sysoev.ru>
parents:
99
diff
changeset
|
1 |
|
da8c5707af39
nginx-0.1.0-2004-09-28-12:34:51 import; set copyright and remove unused files
Igor Sysoev <igor@sysoev.ru>
parents:
99
diff
changeset
|
2 /* |
|
444
42d11f017717
nginx-0.1.0-2004-09-29-20:00:49 import; remove years from copyright
Igor Sysoev <igor@sysoev.ru>
parents:
441
diff
changeset
|
3 * Copyright (C) Igor Sysoev |
|
441
da8c5707af39
nginx-0.1.0-2004-09-28-12:34:51 import; set copyright and remove unused files
Igor Sysoev <igor@sysoev.ru>
parents:
99
diff
changeset
|
4 */ |
|
da8c5707af39
nginx-0.1.0-2004-09-28-12:34:51 import; set copyright and remove unused files
Igor Sysoev <igor@sysoev.ru>
parents:
99
diff
changeset
|
5 |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
6 |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
7 #include <ngx_config.h> |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
8 #include <ngx_core.h> |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
9 #include <ngx_http.h> |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
10 |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
11 |
| 509 | 12 #define ngx_http_script_exit (u_char *) &ngx_http_script_exit_code |
| 13 | |
| 14 static uintptr_t ngx_http_script_exit_code = (uintptr_t) NULL; | |
| 15 | |
| 16 | |
| 17 ngx_uint_t | |
| 18 ngx_http_script_variables_count(ngx_str_t *value) | |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
19 { |
| 509 | 20 ngx_uint_t i, n; |
| 507 | 21 |
| 509 | 22 for (n = 0, i = 0; i < value->len; i++) { |
| 23 if (value->data[i] == '$') { | |
| 24 n++; | |
| 25 } | |
| 507 | 26 } |
| 27 | |
| 509 | 28 return n; |
| 29 } | |
| 30 | |
| 31 | |
| 32 ngx_int_t | |
| 33 ngx_http_script_compile(ngx_http_script_compile_t *sc) | |
| 34 { | |
| 35 u_char ch; | |
| 36 size_t size; | |
| 573 | 37 ngx_int_t index, *p; |
| 509 | 38 ngx_str_t name; |
| 39 uintptr_t *code; | |
| 40 ngx_uint_t i, n, bracket; | |
| 41 ngx_http_script_var_code_t *var_code; | |
| 42 ngx_http_script_copy_code_t *copy; | |
| 43 ngx_http_script_copy_capture_code_t *copy_capture; | |
| 44 | |
| 573 | 45 if (sc->flushes && *sc->flushes == NULL) { |
| 46 n = sc->variables ? sc->variables : 1; | |
| 47 *sc->flushes = ngx_array_create(sc->cf->pool, n, sizeof(ngx_uint_t)); | |
| 48 if (*sc->flushes == NULL) { | |
| 49 return NGX_ERROR; | |
| 50 } | |
| 51 } | |
| 52 | |
| 53 | |
| 509 | 54 if (*sc->lengths == NULL) { |
| 55 n = sc->variables * (2 * sizeof(ngx_http_script_copy_code_t) | |
| 56 + sizeof(ngx_http_script_var_code_t)) | |
| 57 + sizeof(uintptr_t); | |
| 58 | |
| 59 *sc->lengths = ngx_array_create(sc->cf->pool, n, 1); | |
| 60 if (*sc->lengths == NULL) { | |
| 61 return NGX_ERROR; | |
| 62 } | |
| 63 } | |
| 64 | |
| 65 | |
| 66 if (*sc->values == NULL) { | |
| 67 n = (sc->variables * (2 * sizeof(ngx_http_script_copy_code_t) | |
| 68 + sizeof(ngx_http_script_var_code_t)) | |
| 69 + sizeof(uintptr_t) | |
| 70 + sc->source->len | |
| 71 + sizeof(uintptr_t) - 1) | |
| 72 & ~(sizeof(uintptr_t) - 1); | |
| 73 | |
| 74 *sc->values = ngx_array_create(sc->cf->pool, n, 1); | |
| 75 if (*sc->values == NULL) { | |
| 507 | 76 return NGX_ERROR; |
| 77 } | |
| 78 } | |
| 79 | |
| 509 | 80 sc->variables = 0; |
| 81 | |
| 82 for (i = 0; i < sc->source->len; /* void */ ) { | |
| 83 | |
| 84 name.len = 0; | |
| 85 | |
| 86 if (sc->source->data[i] == '$') { | |
| 87 | |
| 88 if (++i == sc->source->len) { | |
| 89 goto invalid_variable; | |
| 90 } | |
| 91 | |
| 92 if (sc->source->data[i] >= '1' && sc->source->data[i] <= '9') { | |
| 93 | |
| 94 copy_capture = ngx_http_script_add_code(*sc->lengths, | |
| 95 sizeof(ngx_http_script_copy_capture_code_t), | |
| 96 NULL); | |
| 97 if (copy_capture == NULL) { | |
| 98 return NGX_ERROR; | |
| 99 } | |
| 100 | |
| 101 copy_capture->code = (ngx_http_script_code_pt) | |
| 102 ngx_http_script_copy_capture_len_code; | |
| 103 copy_capture->n = 2 * (sc->source->data[i] - '0'); | |
| 104 | |
| 105 copy_capture = ngx_http_script_add_code(*sc->values, | |
| 106 sizeof(ngx_http_script_copy_capture_code_t), | |
| 107 &sc->main); | |
| 108 if (copy_capture == NULL) { | |
| 109 return NGX_ERROR; | |
| 110 } | |
| 111 | |
| 112 copy_capture->code = ngx_http_script_copy_capture_code; | |
| 113 copy_capture->n = sc->source->data[i] - '0'; | |
| 114 | |
| 115 if (sc->ncaptures < copy_capture->n) { | |
| 116 sc->ncaptures = copy_capture->n; | |
| 117 } | |
| 118 | |
| 119 copy_capture->n *= 2; | |
| 507 | 120 |
| 509 | 121 i++; |
| 122 | |
| 123 continue; | |
| 124 } | |
| 125 | |
| 126 if (sc->source->data[i] == '{') { | |
| 127 bracket = 1; | |
| 128 | |
| 129 if (++i == sc->source->len) { | |
| 130 goto invalid_variable; | |
| 131 } | |
| 132 | |
| 133 name.data = &sc->source->data[i]; | |
| 134 | |
| 135 } else { | |
| 136 bracket = 0; | |
| 137 name.data = &sc->source->data[i]; | |
| 138 } | |
| 139 | |
| 140 for ( /* void */ ; i < sc->source->len; i++, name.len++) { | |
| 141 ch = sc->source->data[i]; | |
| 507 | 142 |
| 509 | 143 if (ch == '}' && bracket) { |
| 144 i++; | |
| 145 bracket = 0; | |
| 146 break; | |
| 147 } | |
| 148 | |
| 149 if ((ch >= 'A' && ch <= 'Z') | |
| 150 || (ch >= 'a' && ch <= 'z') | |
| 151 || (ch >= '0' && ch <= '9') | |
| 152 || ch == '_') | |
| 153 { | |
| 154 continue; | |
| 155 } | |
| 156 | |
| 157 break; | |
| 158 } | |
| 159 | |
| 160 if (bracket) { | |
| 161 ngx_conf_log_error(NGX_LOG_EMERG, sc->cf, 0, | |
| 162 "the closing bracket in \"%V\" " | |
| 163 "variable is missing", &name); | |
| 507 | 164 return NGX_ERROR; |
| 165 } | |
| 166 | |
| 509 | 167 if (name.len == 0) { |
| 168 goto invalid_variable; | |
| 169 } | |
| 507 | 170 |
| 509 | 171 sc->variables++; |
| 507 | 172 |
| 509 | 173 index = ngx_http_get_variable_index(sc->cf, &name); |
| 174 | |
| 175 if (index == NGX_ERROR) { | |
| 507 | 176 return NGX_ERROR; |
| 177 } | |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
178 |
| 573 | 179 if (sc->flushes) { |
| 180 p = ngx_array_push(*sc->flushes); | |
| 181 if (p == NULL) { | |
| 182 return NGX_ERROR; | |
| 183 } | |
| 184 | |
| 185 *p = index; | |
| 186 } | |
| 187 | |
| 509 | 188 var_code = ngx_http_script_add_code(*sc->lengths, |
| 189 sizeof(ngx_http_script_var_code_t), | |
| 190 NULL); | |
| 507 | 191 if (var_code == NULL) { |
| 192 return NGX_ERROR; | |
| 193 } | |
| 194 | |
| 195 var_code->code = (ngx_http_script_code_pt) | |
| 509 | 196 ngx_http_script_copy_var_len_code; |
| 197 var_code->index = (uintptr_t) index; | |
| 507 | 198 |
| 199 | |
| 509 | 200 var_code = ngx_http_script_add_code(*sc->values, |
| 201 sizeof(ngx_http_script_var_code_t), | |
| 202 &sc->main); | |
| 507 | 203 if (var_code == NULL) { |
| 204 return NGX_ERROR; | |
| 205 } | |
| 206 | |
| 509 | 207 var_code->code = ngx_http_script_copy_var_code; |
| 208 var_code->index = (uintptr_t) index; | |
| 209 | |
| 210 continue; | |
| 211 } | |
| 507 | 212 |
| 509 | 213 if (sc->source->data[i] == '?' && sc->compile_args) { |
| 214 sc->args = 1; | |
| 215 sc->compile_args = 0; | |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
216 |
| 509 | 217 code = ngx_http_script_add_code(*sc->values, sizeof(uintptr_t), |
| 218 &sc->main); | |
| 219 if (code == NULL) { | |
| 507 | 220 return NGX_ERROR; |
| 221 } | |
| 222 | |
| 509 | 223 *code = (uintptr_t) ngx_http_script_start_args_code; |
| 224 | |
| 225 i++; | |
| 226 | |
| 507 | 227 continue; |
| 228 } | |
| 229 | |
| 509 | 230 name.data = &sc->source->data[i]; |
| 231 | |
| 232 while (i < sc->source->len | |
| 233 && sc->source->data[i] != '$' | |
| 234 && !(sc->source->data[i] == '?' && sc->compile_args)) | |
| 235 { | |
| 236 i++; | |
| 237 name.len++; | |
| 238 } | |
| 239 | |
| 240 sc->size += name.len; | |
| 241 | |
| 242 copy = ngx_http_script_add_code(*sc->lengths, | |
| 243 sizeof(ngx_http_script_copy_code_t), | |
| 244 NULL); | |
| 245 if (copy == NULL) { | |
| 507 | 246 return NGX_ERROR; |
| 247 } | |
| 509 | 248 |
| 249 copy->code = (ngx_http_script_code_pt) ngx_http_script_copy_len_code; | |
| 250 copy->len = name.len; | |
| 251 | |
| 252 size = (sizeof(ngx_http_script_copy_code_t) + name.len | |
| 253 + sizeof(uintptr_t) - 1) | |
| 254 & ~(sizeof(uintptr_t) - 1); | |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
255 |
| 509 | 256 copy = ngx_http_script_add_code(*sc->values, size, &sc->main); |
| 257 if (copy == NULL) { | |
| 258 return NGX_ERROR; | |
| 259 } | |
| 260 | |
| 261 copy->code = ngx_http_script_copy_code; | |
| 262 copy->len = name.len; | |
| 263 | |
| 264 ngx_memcpy((u_char *) copy + sizeof(ngx_http_script_copy_code_t), | |
| 265 name.data, name.len); | |
| 507 | 266 } |
| 267 | |
| 509 | 268 if (sc->complete_lengths) { |
| 269 code = ngx_http_script_add_code(*sc->lengths, sizeof(uintptr_t), NULL); | |
| 270 if (code == NULL) { | |
| 271 return NGX_ERROR; | |
| 272 } | |
| 507 | 273 |
| 509 | 274 *code = (uintptr_t) NULL; |
| 507 | 275 } |
| 276 | |
| 509 | 277 if (sc->complete_values) { |
| 278 code = ngx_http_script_add_code(*sc->values, sizeof(uintptr_t), | |
| 279 &sc->main); | |
| 280 if (code == NULL) { | |
| 281 return NGX_ERROR; | |
| 282 } | |
| 283 | |
| 284 *code = (uintptr_t) NULL; | |
| 285 } | |
| 507 | 286 |
| 287 return NGX_OK; | |
| 509 | 288 |
| 289 invalid_variable: | |
| 290 | |
| 291 ngx_conf_log_error(NGX_LOG_EMERG, sc->cf, 0, "invalid variable name"); | |
| 292 | |
| 293 return NGX_ERROR; | |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
294 } |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
295 |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
296 |
| 623 | 297 u_char * |
| 298 ngx_http_script_run(ngx_http_request_t *r, ngx_str_t *value, | |
| 299 void *code_lengths, size_t len, void *code_values) | |
| 300 { | |
| 301 ngx_http_script_code_pt code; | |
| 302 ngx_http_script_len_code_pt lcode; | |
| 303 ngx_http_script_engine_t e; | |
| 304 | |
| 305 ngx_memzero(&e, sizeof(ngx_http_script_engine_t)); | |
| 306 | |
| 307 e.ip = code_lengths; | |
| 308 e.request = r; | |
| 309 e.flushed = 1; | |
| 310 | |
| 311 while (*(uintptr_t *) e.ip) { | |
| 312 lcode = *(ngx_http_script_len_code_pt *) e.ip; | |
| 313 len += lcode(&e); | |
| 314 } | |
| 315 | |
| 316 | |
| 317 value->len = len; | |
| 318 value->data = ngx_palloc(r->pool, len); | |
| 319 if (value->data == NULL) { | |
| 320 return NULL; | |
| 321 } | |
| 322 | |
| 323 e.ip = code_values; | |
| 324 e.pos = value->data; | |
| 325 | |
| 326 while (*(uintptr_t *) e.ip) { | |
| 327 code = *(ngx_http_script_code_pt *) e.ip; | |
| 328 code((ngx_http_script_engine_t *) &e); | |
| 329 } | |
| 330 | |
| 331 return e.pos; | |
| 332 } | |
| 333 | |
| 334 | |
| 573 | 335 void |
| 336 ngx_http_script_flush_no_cachable_variables(ngx_http_request_t *r, | |
| 337 ngx_array_t *indices) | |
| 338 { | |
| 339 ngx_uint_t n, *index; | |
| 340 | |
| 341 if (indices) { | |
| 342 index = indices->elts; | |
| 343 for (n = 0; n < indices->nelts; n++) { | |
| 344 if (r->variables[index[n]].no_cachable) { | |
| 345 r->variables[index[n]].valid = 0; | |
| 346 r->variables[index[n]].not_found = 0; | |
| 347 } | |
| 348 } | |
| 349 } | |
| 350 } | |
| 351 | |
| 352 | |
| 509 | 353 void * |
| 507 | 354 ngx_http_script_start_code(ngx_pool_t *pool, ngx_array_t **codes, size_t size) |
| 479 | 355 { |
| 507 | 356 if (*codes == NULL) { |
| 357 *codes = ngx_array_create(pool, 256, 1); | |
| 358 if (*codes == NULL) { | |
| 359 return NULL; | |
| 360 } | |
| 361 } | |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
362 |
| 507 | 363 return ngx_array_push_n(*codes, size); |
| 364 } | |
| 365 | |
| 509 | 366 |
| 367 void * | |
| 368 ngx_http_script_add_code(ngx_array_t *codes, size_t size, void *code) | |
| 369 { | |
| 370 u_char *elts, **p; | |
| 371 void *new; | |
| 372 | |
| 373 elts = codes->elts; | |
| 374 | |
| 375 new = ngx_array_push_n(codes, size); | |
| 376 if (new == NULL) { | |
| 377 return NGX_CONF_ERROR; | |
| 378 } | |
| 379 | |
| 380 if (code) { | |
| 381 if (elts != codes->elts) { | |
| 382 p = code; | |
| 383 *p += (u_char *) codes->elts - elts; | |
| 384 } | |
| 385 } | |
| 386 | |
| 387 return new; | |
| 388 } | |
| 507 | 389 |
| 479 | 390 |
| 507 | 391 size_t |
| 509 | 392 ngx_http_script_copy_len_code(ngx_http_script_engine_t *e) |
| 507 | 393 { |
| 394 ngx_http_script_copy_code_t *code; | |
| 479 | 395 |
| 509 | 396 code = (ngx_http_script_copy_code_t *) e->ip; |
| 507 | 397 |
| 509 | 398 e->ip += sizeof(ngx_http_script_copy_code_t); |
| 507 | 399 |
| 400 return code->len; | |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
401 } |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
402 |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
403 |
| 507 | 404 void |
| 509 | 405 ngx_http_script_copy_code(ngx_http_script_engine_t *e) |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
406 { |
| 507 | 407 ngx_http_script_copy_code_t *code; |
| 408 | |
| 509 | 409 code = (ngx_http_script_copy_code_t *) e->ip; |
| 410 | |
| 411 if (!e->skip) { | |
| 573 | 412 e->pos = ngx_copy(e->pos, e->ip + sizeof(ngx_http_script_copy_code_t), |
| 413 code->len); | |
| 509 | 414 } |
| 507 | 415 |
| 509 | 416 e->ip += sizeof(ngx_http_script_copy_code_t) |
| 417 + ((code->len + sizeof(uintptr_t) - 1) & ~(sizeof(uintptr_t) - 1)); | |
| 507 | 418 |
| 509 | 419 if (e->log) { |
| 420 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 421 "http script copy: \"%V\"", &e->buf); | |
| 422 } | |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
423 } |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
424 |
|
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
425 |
| 507 | 426 size_t |
| 509 | 427 ngx_http_script_copy_var_len_code(ngx_http_script_engine_t *e) |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
428 { |
| 507 | 429 ngx_http_variable_value_t *value; |
| 430 ngx_http_script_var_code_t *code; | |
| 431 | |
| 509 | 432 code = (ngx_http_script_var_code_t *) e->ip; |
| 507 | 433 |
| 509 | 434 e->ip += sizeof(ngx_http_script_var_code_t); |
| 507 | 435 |
| 573 | 436 if (e->flushed) { |
| 437 value = ngx_http_get_indexed_variable(e->request, code->index); | |
| 507 | 438 |
| 573 | 439 } else { |
| 440 value = ngx_http_get_flushed_variable(e->request, code->index); | |
| 441 } | |
| 442 | |
| 443 if (value && !value->not_found) { | |
| 444 return value->len; | |
| 507 | 445 } |
| 446 | |
| 553 | 447 return 0; |
|
99
a059e1aa65d4
nginx-0.0.1-2003-06-02-19:24:30 import
Igor Sysoev <igor@sysoev.ru>
parents:
diff
changeset
|
448 } |
| 479 | 449 |
| 450 | |
| 507 | 451 void |
| 509 | 452 ngx_http_script_copy_var_code(ngx_http_script_engine_t *e) |
| 479 | 453 { |
| 507 | 454 ngx_http_variable_value_t *value; |
| 455 ngx_http_script_var_code_t *code; | |
| 456 | |
| 509 | 457 code = (ngx_http_script_var_code_t *) e->ip; |
| 458 | |
| 459 e->ip += sizeof(ngx_http_script_var_code_t); | |
| 460 | |
| 461 if (!e->skip) { | |
| 573 | 462 |
| 463 if (e->flushed) { | |
| 464 value = ngx_http_get_indexed_variable(e->request, code->index); | |
| 509 | 465 |
| 573 | 466 } else { |
| 467 value = ngx_http_get_flushed_variable(e->request, code->index); | |
| 468 } | |
| 469 | |
| 470 if (value && !value->not_found) { | |
| 471 e->pos = ngx_copy(e->pos, value->data, value->len); | |
| 509 | 472 |
| 553 | 473 if (e->log) { |
| 474 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, | |
| 475 e->request->connection->log, 0, | |
| 476 "http script var: \"%V\"", &e->buf); | |
| 477 } | |
| 509 | 478 } |
| 479 } | |
| 480 } | |
| 481 | |
| 482 | |
| 483 size_t | |
| 484 ngx_http_script_copy_capture_len_code(ngx_http_script_engine_t *e) | |
| 485 { | |
| 486 ngx_http_script_copy_capture_code_t *code; | |
| 487 | |
| 488 code = (ngx_http_script_copy_capture_code_t *) e->ip; | |
| 489 | |
| 490 e->ip += sizeof(ngx_http_script_copy_capture_code_t); | |
| 491 | |
| 515 | 492 if (code->n < e->ncaptures) { |
| 493 if ((e->args || e->quote) | |
| 494 && (e->request->quoted_uri || e->request->plus_in_uri)) | |
| 495 { | |
| 496 return e->captures[code->n + 1] - e->captures[code->n] | |
| 497 + ngx_escape_uri(NULL, | |
| 498 &e->line.data[e->captures[code->n]], | |
| 509 | 499 e->captures[code->n + 1] - e->captures[code->n], |
| 500 NGX_ESCAPE_ARGS); | |
| 515 | 501 } else { |
| 502 return e->captures[code->n + 1] - e->captures[code->n]; | |
| 503 } | |
| 509 | 504 } |
| 515 | 505 |
| 506 return 0; | |
| 509 | 507 } |
| 508 | |
| 509 | |
| 510 void | |
| 511 ngx_http_script_copy_capture_code(ngx_http_script_engine_t *e) | |
| 512 { | |
| 513 ngx_http_script_copy_capture_code_t *code; | |
| 514 | |
| 515 code = (ngx_http_script_copy_capture_code_t *) e->ip; | |
| 516 | |
| 517 e->ip += sizeof(ngx_http_script_copy_capture_code_t); | |
| 507 | 518 |
| 515 | 519 if (code->n < e->ncaptures) { |
| 520 if ((e->args || e->quote) | |
| 521 && (e->request->quoted_uri || e->request->plus_in_uri)) | |
| 522 { | |
| 523 e->pos = (u_char *) ngx_escape_uri(e->pos, | |
| 524 &e->line.data[e->captures[code->n]], | |
| 509 | 525 e->captures[code->n + 1] - e->captures[code->n], |
| 526 NGX_ESCAPE_ARGS); | |
| 515 | 527 } else { |
| 573 | 528 e->pos = ngx_copy(e->pos, |
| 529 &e->line.data[e->captures[code->n]], | |
| 530 e->captures[code->n + 1] - e->captures[code->n]); | |
| 515 | 531 } |
| 509 | 532 } |
| 533 | |
| 534 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 535 "http script capture: \"%V\"", &e->buf); | |
| 536 } | |
| 537 | |
| 538 | |
| 539 void | |
| 540 ngx_http_script_start_args_code(ngx_http_script_engine_t *e) | |
| 541 { | |
| 542 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 543 "http script args"); | |
| 577 | 544 |
| 509 | 545 e->args = e->pos; |
| 546 e->ip += sizeof(uintptr_t); | |
| 547 } | |
| 548 | |
| 549 | |
| 517 | 550 |
| 551 #if (NGX_PCRE) | |
| 552 | |
| 509 | 553 void |
| 554 ngx_http_script_regex_start_code(ngx_http_script_engine_t *e) | |
| 555 { | |
| 556 size_t len; | |
| 557 ngx_int_t rc; | |
| 558 ngx_uint_t n; | |
| 559 ngx_http_request_t *r; | |
| 560 ngx_http_script_engine_t le; | |
| 561 ngx_http_script_len_code_pt lcode; | |
| 562 ngx_http_script_regex_code_t *code; | |
| 563 | |
| 564 code = (ngx_http_script_regex_code_t *) e->ip; | |
| 565 | |
| 566 r = e->request; | |
| 567 | |
| 568 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, | |
| 569 "http script regex: \"%V\"", &code->name); | |
| 570 | |
| 571 if (code->uri) { | |
| 515 | 572 e->line = r->uri; |
| 509 | 573 } else { |
| 574 e->sp--; | |
| 573 | 575 e->line.len = e->sp->len; |
| 576 e->line.data = e->sp->data; | |
| 509 | 577 } |
| 507 | 578 |
| 515 | 579 rc = ngx_regex_exec(code->regex, &e->line, e->captures, code->ncaptures); |
| 509 | 580 |
| 581 if (rc == NGX_REGEX_NO_MATCHED) { | |
| 582 if (e->log) { | |
| 583 ngx_log_error(NGX_LOG_NOTICE, r->connection->log, 0, | |
| 515 | 584 "\"%V\" does not match \"%V\"", |
| 585 &code->name, &e->line); | |
| 509 | 586 } |
| 587 | |
| 515 | 588 e->ncaptures = 0; |
| 589 | |
| 509 | 590 if (code->test) { |
| 573 | 591 e->sp->len = 0; |
| 592 e->sp->data = (u_char *) ""; | |
| 509 | 593 e->sp++; |
| 594 | |
| 595 e->ip += sizeof(ngx_http_script_regex_code_t); | |
| 596 return; | |
| 597 } | |
| 598 | |
| 599 e->ip += code->next; | |
| 600 return; | |
| 601 } | |
| 602 | |
| 603 if (rc < 0) { | |
| 604 ngx_log_error(NGX_LOG_ALERT, r->connection->log, 0, | |
| 605 ngx_regex_exec_n " failed: %d on \"%V\" using \"%V\"", | |
| 515 | 606 rc, &e->line, &code->name); |
| 509 | 607 |
| 608 e->ip = ngx_http_script_exit; | |
| 609 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR; | |
| 610 return; | |
| 611 } | |
| 612 | |
| 613 if (e->log) { | |
| 614 ngx_log_error(NGX_LOG_NOTICE, r->connection->log, 0, | |
| 515 | 615 "\"%V\" matches \"%V\"", &code->name, &e->line); |
| 509 | 616 } |
| 617 | |
| 515 | 618 e->ncaptures = code->ncaptures; |
| 619 | |
| 509 | 620 if (code->test) { |
| 573 | 621 e->sp->len = 1; |
| 622 e->sp->data = (u_char *) "1"; | |
| 509 | 623 e->sp++; |
| 624 | |
| 625 e->ip += sizeof(ngx_http_script_regex_code_t); | |
| 626 return; | |
| 627 } | |
| 628 | |
| 629 if (code->status) { | |
| 630 e->status = code->status; | |
| 507 | 631 |
| 509 | 632 if (!code->redirect) { |
| 633 e->ip = ngx_http_script_exit; | |
| 634 return; | |
| 635 } | |
| 636 } | |
| 637 | |
| 638 if (code->uri) { | |
| 639 r->internal = 1; | |
| 640 r->valid_unparsed_uri = 0; | |
| 641 | |
| 642 if (code->break_cycle) { | |
| 643 r->valid_location = 0; | |
| 527 | 644 r->uri_changed = 0; |
| 509 | 645 |
| 646 } else { | |
| 647 r->uri_changed = 1; | |
| 648 } | |
| 649 } | |
| 650 | |
| 651 if (code->lengths == NULL) { | |
| 652 e->buf.len = code->size; | |
| 653 | |
| 654 if (code->uri) { | |
| 655 if (rc && (r->quoted_uri || r->plus_in_uri)) { | |
| 656 e->buf.len += 2 * ngx_escape_uri(NULL, r->uri.data, r->uri.len, | |
| 657 NGX_ESCAPE_ARGS); | |
| 658 } | |
| 659 } | |
| 660 | |
| 661 for (n = 1; n < (ngx_uint_t) rc; n++) { | |
| 662 e->buf.len += e->captures[2 * n + 1] - e->captures[2 * n]; | |
| 663 } | |
| 664 | |
| 665 } else { | |
| 666 ngx_memzero(&le, sizeof(ngx_http_script_engine_t)); | |
| 667 | |
| 668 le.ip = code->lengths->elts; | |
| 669 le.request = r; | |
| 670 le.captures = e->captures; | |
| 515 | 671 le.ncaptures = e->ncaptures; |
| 509 | 672 |
| 673 len = 1; /* reserve 1 byte for possible "?" */ | |
| 674 | |
| 675 while (*(uintptr_t *) le.ip) { | |
| 676 lcode = *(ngx_http_script_len_code_pt *) le.ip; | |
| 677 len += lcode(&le); | |
| 678 } | |
| 679 | |
| 680 e->buf.len = len; | |
| 681 } | |
| 682 | |
| 515 | 683 if (code->add_args && r->args.len) { |
| 509 | 684 e->buf.len += r->args.len + 1; |
| 685 } | |
| 686 | |
| 687 e->buf.data = ngx_palloc(r->pool, e->buf.len); | |
| 688 if (e->buf.data == NULL) { | |
| 689 e->ip = ngx_http_script_exit; | |
| 690 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR; | |
| 507 | 691 return; |
| 692 } | |
| 693 | |
| 509 | 694 e->quote = code->redirect; |
| 695 | |
| 696 e->pos = e->buf.data; | |
| 697 | |
| 698 e->ip += sizeof(ngx_http_script_regex_code_t); | |
| 699 } | |
| 700 | |
| 701 | |
| 702 void | |
| 703 ngx_http_script_regex_end_code(ngx_http_script_engine_t *e) | |
| 704 { | |
| 577 | 705 u_char *dst, *src; |
| 509 | 706 ngx_http_request_t *r; |
| 707 ngx_http_script_regex_end_code_t *code; | |
| 708 | |
| 709 code = (ngx_http_script_regex_end_code_t *) e->ip; | |
| 710 | |
| 711 r = e->request; | |
| 712 | |
| 713 e->quote = 0; | |
| 714 | |
| 715 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, | |
| 716 "http script regex end"); | |
| 717 | |
| 718 if (code->redirect) { | |
| 719 | |
| 577 | 720 dst = e->buf.data; |
| 721 src = e->buf.data; | |
| 722 | |
| 601 | 723 ngx_unescape_uri(&dst, &src, e->pos - e->buf.data, NGX_UNESCAPE_URI); |
| 577 | 724 |
| 725 if (src < e->pos) { | |
| 726 dst = ngx_copy(dst, src, e->pos - src); | |
| 727 } | |
| 728 | |
| 729 e->pos = dst; | |
| 730 | |
| 731 if (code->add_args && r->args.len) { | |
| 509 | 732 *e->pos++ = (u_char) (code->args ? '&' : '?'); |
| 573 | 733 e->pos = ngx_copy(e->pos, r->args.data, r->args.len); |
| 509 | 734 } |
| 735 | |
| 736 e->buf.len = e->pos - e->buf.data; | |
| 737 | |
| 738 if (e->log) { | |
| 739 ngx_log_error(NGX_LOG_NOTICE, r->connection->log, 0, | |
| 740 "rewritten redirect: \"%V\"", &e->buf); | |
| 741 } | |
| 742 | |
| 743 r->headers_out.location = ngx_list_push(&r->headers_out.headers); | |
| 744 if (r->headers_out.location == NULL) { | |
| 745 e->ip = ngx_http_script_exit; | |
| 746 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR; | |
| 747 return; | |
| 748 } | |
| 749 | |
| 577 | 750 r->headers_out.location->hash = 1; |
| 509 | 751 r->headers_out.location->key.len = sizeof("Location") - 1; |
| 752 r->headers_out.location->key.data = (u_char *) "Location"; | |
| 753 r->headers_out.location->value = e->buf; | |
| 754 | |
| 755 e->ip += sizeof(ngx_http_script_regex_end_code_t); | |
| 756 return; | |
| 757 } | |
| 758 | |
| 577 | 759 if (e->args) { |
| 509 | 760 e->buf.len = e->args - e->buf.data; |
| 761 | |
| 762 if (code->add_args && r->args.len) { | |
| 763 *e->pos++ = '&'; | |
| 573 | 764 e->pos = ngx_copy(e->pos, r->args.data, r->args.len); |
| 509 | 765 } |
| 766 | |
| 767 r->args.len = e->pos - e->args; | |
| 768 r->args.data = e->args; | |
| 769 | |
| 770 e->args = NULL; | |
| 771 | |
| 772 } else { | |
| 773 e->buf.len = e->pos - e->buf.data; | |
| 774 } | |
| 775 | |
| 776 if (e->log) { | |
| 777 ngx_log_error(NGX_LOG_NOTICE, r->connection->log, 0, | |
| 778 "rewritten data: \"%V\", args: \"%V\"", | |
| 779 &e->buf, &r->args); | |
| 780 } | |
| 781 | |
| 782 if (code->uri) { | |
| 783 r->uri = e->buf; | |
| 784 | |
| 535 | 785 if (r->uri.len == 0) { |
| 786 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, | |
| 787 "the rewritten URI has a zero length"); | |
| 788 e->ip = ngx_http_script_exit; | |
| 789 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR; | |
| 790 return; | |
| 791 } | |
| 792 | |
| 509 | 793 if (ngx_http_set_exten(r) != NGX_OK) { |
| 794 e->ip = ngx_http_script_exit; | |
| 795 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR; | |
| 796 return; | |
| 797 } | |
| 798 } | |
| 799 | |
| 800 e->ip += sizeof(ngx_http_script_regex_end_code_t); | |
| 801 } | |
| 802 | |
| 517 | 803 #endif |
| 804 | |
| 509 | 805 |
| 806 void | |
| 807 ngx_http_script_return_code(ngx_http_script_engine_t *e) | |
| 577 | 808 { |
| 509 | 809 ngx_http_script_return_code_t *code; |
| 577 | 810 |
| 509 | 811 code = (ngx_http_script_return_code_t *) e->ip; |
| 812 | |
| 813 e->status = code->status; | |
| 814 | |
| 815 e->ip += sizeof(ngx_http_script_return_code_t) - sizeof(uintptr_t); | |
| 479 | 816 } |
| 509 | 817 |
| 818 | |
| 819 void | |
| 527 | 820 ngx_http_script_break_code(ngx_http_script_engine_t *e) |
| 821 { | |
| 822 e->request->uri_changed = 0; | |
| 823 | |
| 824 e->ip = ngx_http_script_exit; | |
| 825 } | |
| 826 | |
| 827 | |
| 828 void | |
| 509 | 829 ngx_http_script_if_code(ngx_http_script_engine_t *e) |
| 830 { | |
| 831 ngx_http_script_if_code_t *code; | |
| 832 | |
| 833 code = (ngx_http_script_if_code_t *) e->ip; | |
| 834 | |
| 835 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 836 "http script if"); | |
| 837 | |
| 838 e->sp--; | |
| 839 | |
| 573 | 840 if (e->sp->len && e->sp->data[0] != '0') { |
| 509 | 841 if (code->loc_conf) { |
| 842 e->request->loc_conf = code->loc_conf; | |
| 535 | 843 ngx_http_update_location_config(e->request); |
| 509 | 844 } |
| 845 | |
| 846 e->ip += sizeof(ngx_http_script_if_code_t); | |
| 847 return; | |
| 848 } | |
| 849 | |
| 850 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 577 | 851 "http script if: false"); |
| 509 | 852 |
| 853 e->ip += code->next; | |
| 854 } | |
| 855 | |
| 856 | |
| 857 void | |
| 577 | 858 ngx_http_script_equal_code(ngx_http_script_engine_t *e) |
| 859 { | |
| 860 ngx_http_variable_value_t *val, *res; | |
| 861 | |
| 862 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 863 "http script equal"); | |
| 864 | |
| 865 e->sp--; | |
| 866 val = e->sp; | |
| 867 res = e->sp - 1; | |
| 868 | |
| 869 e->ip += sizeof(uintptr_t); | |
| 870 | |
| 871 if (val->len == res->len && ngx_strncmp(val->data, res->data, res->len) | |
| 872 == 0) | |
| 873 { | |
| 874 *res = ngx_http_variable_true_value; | |
| 875 return; | |
| 876 } | |
| 877 | |
| 878 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 879 "http script equal: no"); | |
| 880 | |
| 881 *res = ngx_http_variable_null_value; | |
| 882 } | |
| 883 | |
| 884 | |
| 885 void | |
| 886 ngx_http_script_not_equal_code(ngx_http_script_engine_t *e) | |
| 887 { | |
| 888 ngx_http_variable_value_t *val, *res; | |
| 889 | |
| 890 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 891 "http script not equal"); | |
| 892 | |
| 893 e->sp--; | |
| 894 val = e->sp; | |
| 895 res = e->sp - 1; | |
| 896 | |
| 897 e->ip += sizeof(uintptr_t); | |
| 898 | |
| 899 if (val->len == res->len && ngx_strncmp(val->data, res->data, res->len) | |
| 900 == 0) | |
| 901 { | |
| 902 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 903 "http script not equal: no"); | |
| 904 | |
| 905 *res = ngx_http_variable_null_value; | |
| 906 return; | |
| 907 } | |
| 908 | |
| 909 *res = ngx_http_variable_true_value; | |
| 910 } | |
| 911 | |
| 912 | |
| 913 void | |
| 515 | 914 ngx_http_script_complex_value_code(ngx_http_script_engine_t *e) |
| 915 { | |
| 916 size_t len; | |
| 917 ngx_http_script_engine_t le; | |
| 918 ngx_http_script_len_code_pt lcode; | |
| 919 ngx_http_script_complex_value_code_t *code; | |
| 920 | |
| 921 code = (ngx_http_script_complex_value_code_t *) e->ip; | |
| 922 | |
| 923 e->ip += sizeof(ngx_http_script_complex_value_code_t); | |
| 924 | |
| 925 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 926 "http script complex value"); | |
| 927 | |
| 928 ngx_memzero(&le, sizeof(ngx_http_script_engine_t)); | |
| 929 | |
| 930 le.ip = code->lengths->elts; | |
| 931 le.request = e->request; | |
| 932 le.captures = e->captures; | |
| 933 le.ncaptures = e->ncaptures; | |
| 934 | |
| 935 for (len = 0; *(uintptr_t *) le.ip; len += lcode(&le)) { | |
| 936 lcode = *(ngx_http_script_len_code_pt *) le.ip; | |
| 937 } | |
| 938 | |
| 939 e->buf.len = len; | |
| 940 e->buf.data = ngx_palloc(e->request->pool, len); | |
| 941 if (e->buf.data == NULL) { | |
| 942 e->ip = ngx_http_script_exit; | |
| 943 e->status = NGX_HTTP_INTERNAL_SERVER_ERROR; | |
| 944 return; | |
| 945 } | |
| 946 | |
| 947 e->pos = e->buf.data; | |
| 948 | |
| 577 | 949 e->sp->len = e->buf.len; |
| 573 | 950 e->sp->data = e->buf.data; |
| 515 | 951 e->sp++; |
| 952 } | |
| 953 | |
| 954 | |
| 955 void | |
| 509 | 956 ngx_http_script_value_code(ngx_http_script_engine_t *e) |
| 957 { | |
| 958 ngx_http_script_value_code_t *code; | |
| 959 | |
| 960 code = (ngx_http_script_value_code_t *) e->ip; | |
| 961 | |
| 962 e->ip += sizeof(ngx_http_script_value_code_t); | |
| 963 | |
| 573 | 964 e->sp->len = code->text_len; |
| 965 e->sp->data = (u_char *) code->text_data; | |
| 577 | 966 |
| 967 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 968 "http script value: \"%V\"", e->sp); | |
| 969 | |
| 509 | 970 e->sp++; |
| 971 } | |
| 972 | |
| 973 | |
| 974 void | |
| 975 ngx_http_script_set_var_code(ngx_http_script_engine_t *e) | |
| 976 { | |
| 977 ngx_http_request_t *r; | |
| 978 ngx_http_script_var_code_t *code; | |
| 979 | |
| 980 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 981 "http script set var"); | |
| 982 | |
| 983 code = (ngx_http_script_var_code_t *) e->ip; | |
| 984 | |
| 985 e->ip += sizeof(ngx_http_script_var_code_t); | |
| 986 | |
| 987 r = e->request; | |
| 988 | |
| 989 e->sp--; | |
| 990 | |
| 573 | 991 r->variables[code->index].len = e->sp->len; |
| 992 r->variables[code->index].valid = 1; | |
| 993 r->variables[code->index].no_cachable = 0; | |
| 994 r->variables[code->index].not_found = 0; | |
| 995 r->variables[code->index].data = e->sp->data; | |
| 509 | 996 } |
| 997 | |
| 998 | |
| 999 void | |
| 1000 ngx_http_script_var_code(ngx_http_script_engine_t *e) | |
| 1001 { | |
| 1002 ngx_http_variable_value_t *value; | |
| 1003 ngx_http_script_var_code_t *code; | |
| 1004 | |
| 1005 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 1006 "http script var"); | |
| 1007 | |
| 1008 code = (ngx_http_script_var_code_t *) e->ip; | |
| 1009 | |
| 1010 e->ip += sizeof(ngx_http_script_var_code_t); | |
| 1011 | |
| 573 | 1012 value = ngx_http_get_flushed_variable(e->request, code->index); |
| 509 | 1013 |
| 573 | 1014 if (value && !value->not_found) { |
| 1015 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, e->request->connection->log, 0, | |
| 1016 "http script var: \"%V\"", value); | |
| 1017 | |
| 553 | 1018 *e->sp = *value; |
| 509 | 1019 e->sp++; |
| 1020 | |
| 1021 return; | |
| 1022 } | |
| 1023 | |
| 577 | 1024 *e->sp = ngx_http_variable_null_value; |
| 509 | 1025 e->sp++; |
| 1026 } | |
| 1027 | |
| 1028 | |
| 1029 void | |
| 1030 ngx_http_script_nop_code(ngx_http_script_engine_t *e) | |
| 1031 { | |
| 1032 e->ip += sizeof(uintptr_t); | |
| 1033 } |