0ebc285aa0
*) refactor ngx_palloc() *) introduce ngx_pnalloc() *) additional pool blocks have smaller header
179 lines
3.2 KiB
C
179 lines
3.2 KiB
C
|
|
/*
|
|
* Copyright (C) Igor Sysoev
|
|
*/
|
|
|
|
|
|
#include <ngx_config.h>
|
|
#include <ngx_core.h>
|
|
|
|
|
|
static void * ngx_libc_cdecl ngx_regex_malloc(size_t size);
|
|
static void ngx_libc_cdecl ngx_regex_free(void *p);
|
|
|
|
|
|
static ngx_pool_t *ngx_pcre_pool;
|
|
|
|
|
|
void
|
|
ngx_regex_init(void)
|
|
{
|
|
pcre_malloc = ngx_regex_malloc;
|
|
pcre_free = ngx_regex_free;
|
|
}
|
|
|
|
|
|
ngx_regex_t *
|
|
ngx_regex_compile(ngx_str_t *pattern, ngx_int_t options, ngx_pool_t *pool,
|
|
ngx_str_t *err)
|
|
{
|
|
int erroff;
|
|
const char *errstr;
|
|
ngx_regex_t *re;
|
|
#if (NGX_THREADS)
|
|
ngx_core_tls_t *tls;
|
|
|
|
#if (NGX_SUPPRESS_WARN)
|
|
tls = NULL;
|
|
#endif
|
|
|
|
if (ngx_threaded) {
|
|
tls = ngx_thread_get_tls(ngx_core_tls_key);
|
|
tls->pool = pool;
|
|
} else {
|
|
ngx_pcre_pool = pool;
|
|
}
|
|
|
|
#else
|
|
|
|
ngx_pcre_pool = pool;
|
|
|
|
#endif
|
|
|
|
re = pcre_compile((const char *) pattern->data, (int) options,
|
|
&errstr, &erroff, NULL);
|
|
|
|
if (re == NULL) {
|
|
if ((size_t) erroff == pattern->len) {
|
|
ngx_snprintf(err->data, err->len - 1,
|
|
"pcre_compile() failed: %s in \"%s\"%Z",
|
|
errstr, pattern->data);
|
|
} else {
|
|
ngx_snprintf(err->data, err->len - 1,
|
|
"pcre_compile() failed: %s in \"%s\" at \"%s\"%Z",
|
|
errstr, pattern->data, pattern->data + erroff);
|
|
}
|
|
}
|
|
|
|
/* ensure that there is no current pool */
|
|
|
|
#if (NGX_THREADS)
|
|
if (ngx_threaded) {
|
|
tls->pool = NULL;
|
|
} else {
|
|
ngx_pcre_pool = NULL;
|
|
}
|
|
#else
|
|
ngx_pcre_pool = NULL;
|
|
#endif
|
|
|
|
return re;
|
|
}
|
|
|
|
|
|
ngx_int_t
|
|
ngx_regex_capture_count(ngx_regex_t *re)
|
|
{
|
|
int rc, n;
|
|
|
|
n = 0;
|
|
|
|
rc = pcre_fullinfo(re, NULL, PCRE_INFO_CAPTURECOUNT, &n);
|
|
|
|
if (rc < 0) {
|
|
return (ngx_int_t) rc;
|
|
}
|
|
|
|
return (ngx_int_t) n;
|
|
}
|
|
|
|
|
|
ngx_int_t
|
|
ngx_regex_exec(ngx_regex_t *re, ngx_str_t *s, int *captures, ngx_int_t size)
|
|
{
|
|
int rc;
|
|
|
|
rc = pcre_exec(re, NULL, (const char *) s->data, s->len, 0, 0,
|
|
captures, size);
|
|
|
|
if (rc == -1) {
|
|
return NGX_REGEX_NO_MATCHED;
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
|
|
ngx_int_t
|
|
ngx_regex_exec_array(ngx_array_t *a, ngx_str_t *s, ngx_log_t *log)
|
|
{
|
|
ngx_int_t n;
|
|
ngx_uint_t i;
|
|
ngx_regex_elt_t *re;
|
|
|
|
re = a->elts;
|
|
|
|
for (i = 0; i < a->nelts; i++) {
|
|
|
|
n = ngx_regex_exec(re[i].regex, s, NULL, 0);
|
|
|
|
if (n == NGX_REGEX_NO_MATCHED) {
|
|
continue;
|
|
}
|
|
|
|
if (n < 0) {
|
|
ngx_log_error(NGX_LOG_ALERT, log, 0,
|
|
ngx_regex_exec_n " failed: %d on \"%V\" using \"%s\"",
|
|
n, s, re[i].name);
|
|
return NGX_ERROR;
|
|
}
|
|
|
|
/* match */
|
|
|
|
return NGX_OK;
|
|
}
|
|
|
|
return NGX_DECLINED;
|
|
}
|
|
|
|
|
|
static void * ngx_libc_cdecl
|
|
ngx_regex_malloc(size_t size)
|
|
{
|
|
ngx_pool_t *pool;
|
|
#if (NGX_THREADS)
|
|
ngx_core_tls_t *tls;
|
|
|
|
if (ngx_threaded) {
|
|
tls = ngx_thread_get_tls(ngx_core_tls_key);
|
|
pool = tls->pool;
|
|
} else {
|
|
pool = ngx_pcre_pool;
|
|
}
|
|
#else
|
|
pool = ngx_pcre_pool;
|
|
#endif
|
|
|
|
if (pool) {
|
|
return ngx_palloc(pool, size);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
|
|
static void ngx_libc_cdecl
|
|
ngx_regex_free(void *p)
|
|
{
|
|
return;
|
|
}
|