diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-10 21:30:40 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-10 21:30:40 +0000 |
commit | 133a45c109da5310add55824db21af5239951f93 (patch) | |
tree | ba6ac4c0a950a0dda56451944315d66409923918 /src/libutil/regexp.c | |
parent | Initial commit. (diff) | |
download | rspamd-133a45c109da5310add55824db21af5239951f93.tar.xz rspamd-133a45c109da5310add55824db21af5239951f93.zip |
Adding upstream version 3.8.1.upstream/3.8.1upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/libutil/regexp.c')
-rw-r--r-- | src/libutil/regexp.c | 1359 |
1 files changed, 1359 insertions, 0 deletions
diff --git a/src/libutil/regexp.c b/src/libutil/regexp.c new file mode 100644 index 0000000..9f143ac --- /dev/null +++ b/src/libutil/regexp.c @@ -0,0 +1,1359 @@ +/* + * Copyright 2023 Vsevolod Stakhov + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +#include "config.h" +#include "regexp.h" +#include "cryptobox.h" +#include "ref.h" +#include "util.h" +#include "rspamd.h" +#include "contrib/fastutf8/fastutf8.h" + +#ifndef WITH_PCRE2 +/* Normal pcre path */ +#include <pcre.h> +#define PCRE_T pcre +#define PCRE_EXTRA_T pcre_extra +#define PCRE_JIT_T pcre_jit_stack +#define PCRE_FREE pcre_free +#define PCRE_JIT_STACK_FREE pcre_jit_stack_free +#define PCRE_FLAG(x) G_PASTE(PCRE_, x) +#else +/* PCRE 2 path */ +#ifndef PCRE2_CODE_UNIT_WIDTH +#define PCRE2_CODE_UNIT_WIDTH 8 +#endif + +#include <pcre2.h> +#define PCRE_T pcre2_code +#define PCRE_JIT_T pcre2_jit_stack +#define PCRE_FREE pcre2_code_free +#define PCRE_JIT_STACK_FREE pcre2_jit_stack_free + +#define PCRE_FLAG(x) G_PASTE(PCRE2_, x) +#endif + +typedef guchar regexp_id_t[rspamd_cryptobox_HASHBYTES]; + +#undef DISABLE_JIT_FAST + +struct rspamd_regexp_s { + gdouble exec_time; + gchar *pattern; + PCRE_T *re; + PCRE_T *raw_re; +#ifndef WITH_PCRE2 + PCRE_EXTRA_T *extra; + PCRE_EXTRA_T *raw_extra; +#else + pcre2_match_context *mcontext; + pcre2_match_context *raw_mcontext; +#endif + regexp_id_t id; + ref_entry_t ref; + gpointer ud; + gpointer re_class; + guint64 cache_id; + gsize match_limit; + guint max_hits; + gint flags; + gint pcre_flags; + gint ncaptures; +}; + +struct rspamd_regexp_cache { + GHashTable *tbl; +#ifdef HAVE_PCRE_JIT + PCRE_JIT_T *jstack; +#endif +}; + +static struct rspamd_regexp_cache *global_re_cache = NULL; +static gboolean can_jit = FALSE; +static gboolean check_jit = TRUE; +static const int max_re_cache_size = 8192; + +#ifdef WITH_PCRE2 +static pcre2_compile_context *pcre2_ctx = NULL; +#endif + +static GQuark +rspamd_regexp_quark(void) +{ + return g_quark_from_static_string("rspamd-regexp"); +} + +static void +rspamd_regexp_generate_id(const gchar *pattern, const gchar *flags, + regexp_id_t out) +{ + rspamd_cryptobox_hash_state_t st; + + rspamd_cryptobox_hash_init(&st, NULL, 0); + + if (flags) { + rspamd_cryptobox_hash_update(&st, flags, strlen(flags)); + } + + rspamd_cryptobox_hash_update(&st, pattern, strlen(pattern)); + rspamd_cryptobox_hash_final(&st, out); +} + +static void +rspamd_regexp_dtor(rspamd_regexp_t *re) +{ + if (re) { + if (re->raw_re && re->raw_re != re->re) { +#ifndef WITH_PCRE2 + /* PCRE1 version */ +#ifdef HAVE_PCRE_JIT + if (re->raw_extra) { + pcre_free_study(re->raw_extra); + } +#endif +#else + /* PCRE 2 version */ + if (re->raw_mcontext) { + pcre2_match_context_free(re->raw_mcontext); + } +#endif + PCRE_FREE(re->raw_re); + } + + if (re->re) { +#ifndef WITH_PCRE2 + /* PCRE1 version */ +#ifdef HAVE_PCRE_JIT + if (re->extra) { + pcre_free_study(re->extra); + } +#endif +#else + /* PCRE 2 version */ + if (re->mcontext) { + pcre2_match_context_free(re->mcontext); + } +#endif + PCRE_FREE(re->re); + } + + if (re->pattern) { + g_free(re->pattern); + } + + g_free(re); + } +} + +static void +rspamd_regexp_post_process(rspamd_regexp_t *r) +{ + if (global_re_cache == NULL) { + rspamd_regexp_library_init(NULL); + } +#if defined(WITH_PCRE2) + static const guint max_recursion_depth = 100000, max_backtrack = 1000000; + + /* Create match context */ + r->mcontext = pcre2_match_context_create(NULL); + g_assert(r->mcontext != NULL); + pcre2_set_recursion_limit(r->mcontext, max_recursion_depth); + pcre2_set_match_limit(r->mcontext, max_backtrack); + + if (r->raw_re && r->re != r->raw_re) { + r->raw_mcontext = pcre2_match_context_create(NULL); + g_assert(r->raw_mcontext != NULL); + pcre2_set_recursion_limit(r->raw_mcontext, max_recursion_depth); + pcre2_set_match_limit(r->raw_mcontext, max_backtrack); + } + else if (r->raw_re) { + r->raw_mcontext = r->mcontext; + } + else { + r->raw_mcontext = NULL; + } + +#ifdef HAVE_PCRE_JIT + guint jit_flags = can_jit ? PCRE2_JIT_COMPLETE : 0; + gsize jsz; + PCRE2_UCHAR errstr[128]; + int errcode; + + if (can_jit) { + if ((errcode = pcre2_jit_compile(r->re, jit_flags)) < 0) { + pcre2_get_error_message(errcode, errstr, G_N_ELEMENTS(errstr)); + msg_err("jit compilation is not supported: %s; pattern: \"%s\"", errstr, r->pattern); + r->flags |= RSPAMD_REGEXP_FLAG_DISABLE_JIT; + } + else { + if (!(pcre2_pattern_info(r->re, PCRE2_INFO_JITSIZE, &jsz) >= 0 && jsz > 0)) { + msg_err("cannot exec pcre2_pattern_info(PCRE2_INFO_JITSIZE) on \"%s\"", r->pattern); + r->flags |= RSPAMD_REGEXP_FLAG_DISABLE_JIT; + } + } + } + else { + r->flags |= RSPAMD_REGEXP_FLAG_DISABLE_JIT; + } + + if (!(r->flags & RSPAMD_REGEXP_FLAG_DISABLE_JIT)) { + pcre2_jit_stack_assign(r->mcontext, NULL, global_re_cache->jstack); + } + + if (r->raw_re && r->re != r->raw_re && !(r->flags & RSPAMD_REGEXP_FLAG_DISABLE_JIT)) { + if ((errcode = pcre2_jit_compile(r->raw_re, jit_flags)) < 0) { + pcre2_get_error_message(errcode, errstr, G_N_ELEMENTS(errstr)); + msg_debug("jit compilation is not supported for raw regexp: %s; pattern: \"%s\"", errstr, r->pattern); + r->flags |= RSPAMD_REGEXP_FLAG_DISABLE_JIT; + } + else { + if (!(pcre2_pattern_info(r->raw_re, PCRE2_INFO_JITSIZE, &jsz) >= 0 && jsz > 0)) { + msg_err("cannot exec pcre2_pattern_info(PCRE2_INFO_JITSIZE) on \"%s\"", r->pattern); + } + else if (!(r->flags & RSPAMD_REGEXP_FLAG_DISABLE_JIT)) { + g_assert(r->raw_mcontext != NULL); + pcre2_jit_stack_assign(r->raw_mcontext, NULL, global_re_cache->jstack); + } + } + } +#endif + +#else + const gchar *err_str = "unknown"; + gboolean try_jit = TRUE, try_raw_jit = TRUE; + gint study_flags = 0; + +#if defined(HAVE_PCRE_JIT) + study_flags |= PCRE_STUDY_JIT_COMPILE; +#endif + + /* Pcre 1 needs study */ + if (r->re) { + r->extra = pcre_study(r->re, study_flags, &err_str); + + if (r->extra == NULL) { + msg_debug("cannot optimize regexp pattern: '%s': %s", + r->pattern, err_str); + try_jit = FALSE; + r->flags |= RSPAMD_REGEXP_FLAG_DISABLE_JIT; + } + } + else { + g_assert_not_reached(); + } + + if (r->raw_re && r->raw_re != r->re) { + r->raw_extra = pcre_study(r->re, study_flags, &err_str); + } + else if (r->raw_re == r->re) { + r->raw_extra = r->extra; + } + + if (r->raw_extra == NULL) { + + msg_debug("cannot optimize raw regexp pattern: '%s': %s", + r->pattern, err_str); + try_raw_jit = FALSE; + } + /* JIT path */ + if (try_jit) { +#ifdef HAVE_PCRE_JIT + gint jit, n; + + if (can_jit) { + jit = 0; + n = pcre_fullinfo(r->re, r->extra, + PCRE_INFO_JIT, &jit); + + if (n != 0 || jit != 1) { + msg_debug("jit compilation of %s is not supported", r->pattern); + r->flags |= RSPAMD_REGEXP_FLAG_DISABLE_JIT; + } + else { + pcre_assign_jit_stack(r->extra, NULL, global_re_cache->jstack); + } + } +#endif + } + else { + msg_debug("cannot optimize regexp pattern: '%s': %s", + r->pattern, err_str); + r->flags |= RSPAMD_REGEXP_FLAG_DISABLE_JIT; + } + + if (try_raw_jit) { +#ifdef HAVE_PCRE_JIT + gint jit, n; + + if (can_jit) { + + if (r->raw_re != r->re) { + jit = 0; + n = pcre_fullinfo(r->raw_re, r->raw_extra, + PCRE_INFO_JIT, &jit); + + if (n != 0 || jit != 1) { + msg_debug("jit compilation of %s is not supported", r->pattern); + r->flags |= RSPAMD_REGEXP_FLAG_DISABLE_JIT; + } + else { + pcre_assign_jit_stack(r->raw_extra, NULL, + global_re_cache->jstack); + } + } + } +#endif + } +#endif /* WITH_PCRE2 */ +} + +rspamd_regexp_t * +rspamd_regexp_new_len(const gchar *pattern, gsize len, const gchar *flags, + GError **err) +{ + const gchar *start = pattern, *end = start + len, *flags_str = NULL, *flags_end = NULL; + gchar *err_str; + rspamd_regexp_t *res; + gboolean explicit_utf = FALSE; + PCRE_T *r; + gchar sep = 0, *real_pattern; +#ifndef WITH_PCRE2 + gint err_off; +#else + gsize err_off; +#endif + gint regexp_flags = 0, rspamd_flags = 0, err_code, ncaptures; + gboolean strict_flags = FALSE; + + rspamd_regexp_library_init(NULL); + + if (pattern == NULL) { + g_set_error(err, rspamd_regexp_quark(), EINVAL, + "cannot create regexp from a NULL pattern"); + return NULL; + } + + if (flags == NULL && start + 1 < end) { + /* We need to parse pattern and detect flags set */ + if (*start == '/') { + sep = '/'; + } + else if (*start == 'm' && start[1] != '\\' && g_ascii_ispunct(start[1])) { + start++; + sep = *start; + + /* Paired braces */ + if (sep == '{') { + sep = '}'; + } + + rspamd_flags |= RSPAMD_REGEXP_FLAG_FULL_MATCH; + } + if (sep == 0) { + /* We have no flags, no separators and just use all line as expr */ + start = pattern; + rspamd_flags &= ~RSPAMD_REGEXP_FLAG_FULL_MATCH; + } + else { + gchar *last_sep = rspamd_memrchr(pattern, sep, len); + + if (last_sep == NULL || last_sep <= start) { + g_set_error(err, rspamd_regexp_quark(), EINVAL, + "pattern is not enclosed with %c: %s", + sep, pattern); + return NULL; + } + flags_str = last_sep + 1; + flags_end = end; + end = last_sep; + start++; + } + } + else { + /* Strictly check all flags */ + strict_flags = TRUE; + start = pattern; + flags_str = flags; + if (flags) { + flags_end = flags + strlen(flags); + } + } + + rspamd_flags |= RSPAMD_REGEXP_FLAG_RAW; + +#ifndef WITH_PCRE2 + regexp_flags &= ~PCRE_FLAG(UTF8); + regexp_flags |= PCRE_FLAG(NEWLINE_ANYCRLF); +#else + regexp_flags &= ~PCRE_FLAG(UTF); +#endif + + if (flags_str != NULL) { + while (flags_str < flags_end) { + switch (*flags_str) { + case 'i': + regexp_flags |= PCRE_FLAG(CASELESS); + break; + case 'm': + regexp_flags |= PCRE_FLAG(MULTILINE); + break; + case 's': + regexp_flags |= PCRE_FLAG(DOTALL); + break; + case 'x': + regexp_flags |= PCRE_FLAG(EXTENDED); + break; + case 'u': + rspamd_flags &= ~RSPAMD_REGEXP_FLAG_RAW; + rspamd_flags |= RSPAMD_REGEXP_FLAG_UTF; +#ifndef WITH_PCRE2 + regexp_flags |= PCRE_FLAG(UTF8); +#else + regexp_flags |= PCRE_FLAG(UTF); +#endif + explicit_utf = TRUE; + break; + case 'O': + /* We optimize all regexps by default */ + rspamd_flags |= RSPAMD_REGEXP_FLAG_NOOPT; + break; + case 'L': + /* SOM_LEFTMOST hyperscan flag */ + rspamd_flags |= RSPAMD_REGEXP_FLAG_LEFTMOST; + break; + case 'r': + rspamd_flags |= RSPAMD_REGEXP_FLAG_RAW; + rspamd_flags &= ~RSPAMD_REGEXP_FLAG_UTF; +#ifndef WITH_PCRE2 + regexp_flags &= ~PCRE_FLAG(UTF8); +#else + regexp_flags &= ~PCRE_FLAG(UTF); +#endif + break; + default: + if (strict_flags) { + g_set_error(err, rspamd_regexp_quark(), EINVAL, + "invalid regexp flag: %c in pattern %s", + *flags_str, pattern); + return NULL; + } + msg_warn("invalid flag '%c' in pattern %s", *flags_str, pattern); + goto fin; + break; + } + flags_str++; + } + } +fin: + + real_pattern = g_malloc(end - start + 1); + rspamd_strlcpy(real_pattern, start, end - start + 1); + +#ifndef WITH_PCRE2 + r = pcre_compile(real_pattern, regexp_flags, + (const char **) &err_str, &err_off, NULL); + (void) err_code; +#else + r = pcre2_compile(real_pattern, PCRE2_ZERO_TERMINATED, + regexp_flags, + &err_code, &err_off, pcre2_ctx); + + if (r == NULL) { + err_str = g_alloca(1024); + memset(err_str, 0, 1024); + pcre2_get_error_message(err_code, err_str, 1024); + } +#endif + + if (r == NULL) { + g_set_error(err, rspamd_regexp_quark(), EINVAL, + "regexp parsing error: '%s' at position %d; pattern: %s", + err_str, (gint) err_off, real_pattern); + g_free(real_pattern); + + return NULL; + } + + /* Now allocate the target structure */ + res = g_malloc0(sizeof(*res)); + REF_INIT_RETAIN(res, rspamd_regexp_dtor); + res->flags = rspamd_flags; + res->pattern = real_pattern; + res->cache_id = RSPAMD_INVALID_ID; + res->pcre_flags = regexp_flags; + res->max_hits = 0; + res->re = r; + + if (rspamd_flags & RSPAMD_REGEXP_FLAG_RAW) { + res->raw_re = r; + } + else if (!explicit_utf) { +#ifndef WITH_PCRE2 + res->raw_re = pcre_compile(real_pattern, regexp_flags & ~PCRE_FLAG(UTF8), + (const char **) &err_str, &err_off, NULL); + (void) err_code; +#else + res->raw_re = pcre2_compile(real_pattern, PCRE2_ZERO_TERMINATED, + regexp_flags & ~PCRE_FLAG(UTF), + &err_code, &err_off, pcre2_ctx); + if (res->raw_re == NULL) { + err_str = g_alloca(1024); + memset(err_str, 0, 1024); + pcre2_get_error_message(err_code, err_str, 1024); + } +#endif + if (res->raw_re == NULL) { + msg_warn("raw regexp parsing error: '%s': '%s' at position %d", + err_str, real_pattern, (gint) err_off); + } + } + + rspamd_regexp_post_process(res); + rspamd_regexp_generate_id(pattern, flags, res->id); + +#ifndef WITH_PCRE2 + /* Check number of captures */ + if (pcre_fullinfo(res->raw_re, res->extra, PCRE_INFO_CAPTURECOUNT, + &ncaptures) == 0) { + res->ncaptures = ncaptures; + } +#else + /* Check number of captures */ + if (pcre2_pattern_info(res->raw_re, PCRE2_INFO_CAPTURECOUNT, + &ncaptures) == 0) { + res->ncaptures = ncaptures; + } +#endif + + return res; +} + +rspamd_regexp_t * +rspamd_regexp_new(const gchar *pattern, const gchar *flags, + GError **err) +{ + return rspamd_regexp_new_len(pattern, strlen(pattern), flags, err); +} + +#ifndef WITH_PCRE2 +gboolean +rspamd_regexp_search(const rspamd_regexp_t *re, const gchar *text, gsize len, + const gchar **start, const gchar **end, gboolean raw, + GArray *captures) +{ + pcre *r; + pcre_extra *ext; +#if defined(HAVE_PCRE_JIT) && defined(HAVE_PCRE_JIT_FAST) && !defined(DISABLE_JIT_FAST) + pcre_jit_stack *st = NULL; +#endif + const gchar *mt; + gsize remain = 0; + gint rc, match_flags = 0, *ovec, ncaptures, i; + const int junk = 0xdeadbabe; + + g_assert(re != NULL); + g_assert(text != NULL); + + if (len == 0) { + len = strlen(text); + } + + if (re->match_limit > 0 && len > re->match_limit) { + len = re->match_limit; + } + + if (end != NULL && *end != NULL) { + /* Incremental search */ + mt = (*end); + + if ((gint) len > (mt - text)) { + remain = len - (mt - text); + } + } + else { + mt = text; + remain = len; + } + + if (remain == 0) { + return FALSE; + } + + match_flags = PCRE_NEWLINE_ANYCRLF; + + if ((re->flags & RSPAMD_REGEXP_FLAG_RAW) || raw) { + r = re->raw_re; + ext = re->raw_extra; +#if defined(HAVE_PCRE_JIT) && defined(HAVE_PCRE_JIT_FAST) && !defined(DISABLE_JIT_FAST) + st = global_re_cache->jstack; +#endif + } + else { + r = re->re; + ext = re->extra; +#if defined(HAVE_PCRE_JIT) && defined(HAVE_PCRE_JIT_FAST) && !defined(DISABLE_JIT_FAST) + if (rspamd_fast_utf8_validate(mt, remain) == 0) { + st = global_re_cache->jstack; + } + else { + msg_err("bad utf8 input for JIT re '%s'", re->pattern); + return FALSE; + } +#endif + } + + if (r == NULL) { + /* Invalid regexp type for the specified input */ + return FALSE; + } + + ncaptures = (re->ncaptures + 1) * 3; + ovec = g_alloca(sizeof(gint) * ncaptures); + + + for (i = 0; i < ncaptures; i++) { + ovec[i] = junk; + } + + if (!(re->flags & RSPAMD_REGEXP_FLAG_NOOPT)) { +#ifdef HAVE_PCRE_JIT +#if defined(HAVE_PCRE_JIT_FAST) && !defined(DISABLE_JIT_FAST) + /* XXX: flags seems to be broken with jit fast path */ + g_assert(remain > 0); + g_assert(mt != NULL); + + if (st != NULL && !(re->flags & RSPAMD_REGEXP_FLAG_DISABLE_JIT) && can_jit) { + rc = pcre_jit_exec(r, ext, mt, remain, 0, 0, ovec, + ncaptures, st); + } + else { + rc = pcre_exec(r, ext, mt, remain, 0, match_flags, ovec, + ncaptures); + } +#else + rc = pcre_exec(r, ext, mt, remain, 0, match_flags, ovec, + ncaptures); +#endif +#else + rc = pcre_exec(r, ext, mt, remain, 0, match_flags, ovec, + ncaptures); +#endif + } + else { + rc = pcre_exec(r, ext, mt, remain, 0, match_flags, ovec, + ncaptures); + } + + if (rc >= 0) { + if (rc > 0) { + if (start) { + *start = mt + ovec[0]; + } + if (end) { + *end = mt + ovec[1]; + } + } + else { + if (start) { + *start = mt; + } + if (end) { + *end = mt + remain; + } + } + + if (captures != NULL && rc >= 1) { + struct rspamd_re_capture *elt; + + g_assert(g_array_get_element_size(captures) == + sizeof(struct rspamd_re_capture)); + g_array_set_size(captures, rc); + + for (i = 0; i < rc; i++) { + if (ovec[i * 2] != junk && ovec[i * 2] >= 0) { + elt = &g_array_index(captures, struct rspamd_re_capture, i); + elt->p = mt + ovec[i * 2]; + elt->len = (mt + ovec[i * 2 + 1]) - elt->p; + } + else { + /* Runtime match returned fewer captures than expected */ + g_array_set_size(captures, i); + break; + } + } + } + + if (re->flags & RSPAMD_REGEXP_FLAG_FULL_MATCH) { + /* We also ensure that the match is full */ + if (ovec[0] != 0 || (guint) ovec[1] < len) { + return FALSE; + } + } + + return TRUE; + } + + return FALSE; +} +#else +/* PCRE 2 version */ +gboolean +rspamd_regexp_search(const rspamd_regexp_t *re, const gchar *text, gsize len, + const gchar **start, const gchar **end, gboolean raw, + GArray *captures) +{ + pcre2_match_data *match_data; + pcre2_match_context *mcontext; + PCRE_T *r; + const gchar *mt; + PCRE2_SIZE remain = 0, *ovec; + const PCRE2_SIZE junk = 0xdeadbabeeeeeeeeULL; + gint rc, match_flags, novec, i; + gboolean ret = FALSE; + + g_assert(re != NULL); + g_assert(text != NULL); + + if (len == 0) { + len = strlen(text); + } + + if (re->match_limit > 0 && len > re->match_limit) { + len = re->match_limit; + } + + if (end != NULL && *end != NULL) { + /* Incremental search */ + mt = (*end); + + if ((gint) len > (mt - text)) { + remain = len - (mt - text); + } + } + else { + mt = text; + remain = len; + } + + if (remain == 0) { + return FALSE; + } + + match_flags = 0; + + if (raw || re->re == re->raw_re) { + r = re->raw_re; + mcontext = re->raw_mcontext; + } + else { + r = re->re; + mcontext = re->mcontext; + } + + if (r == NULL) { + /* Invalid regexp type for the specified input */ + return FALSE; + } + + match_data = pcre2_match_data_create(re->ncaptures + 1, NULL); + novec = pcre2_get_ovector_count(match_data); + ovec = pcre2_get_ovector_pointer(match_data); + + /* Fill ovec with crap, so we can stop if actual matches is less than announced */ + for (i = 0; i < novec; i++) { + ovec[i * 2] = junk; + ovec[i * 2 + 1] = junk; + } + +#ifdef HAVE_PCRE_JIT + if (!(re->flags & RSPAMD_REGEXP_FLAG_DISABLE_JIT) && can_jit) { + if (re->re != re->raw_re && rspamd_fast_utf8_validate(mt, remain) != 0) { + msg_err("bad utf8 input for JIT re '%s'", re->pattern); + return FALSE; + } + + rc = pcre2_jit_match(r, mt, remain, 0, match_flags, match_data, + mcontext); + } + else { + rc = pcre2_match(r, mt, remain, 0, match_flags, match_data, + mcontext); + } +#else + rc = pcre2_match(r, mt, remain, 0, match_flags, match_data, + mcontext); +#endif + + if (rc >= 0) { + if (novec > 0) { + if (start) { + *start = mt + ovec[0]; + } + if (end) { + *end = mt + ovec[1]; + } + } + else { + if (start) { + *start = mt; + } + if (end) { + *end = mt + remain; + } + } + + if (captures != NULL && novec >= 1) { + struct rspamd_re_capture *elt; + + g_assert(g_array_get_element_size(captures) == + sizeof(struct rspamd_re_capture)); + g_array_set_size(captures, novec); + + for (i = 0; i < novec; i++) { + if (ovec[i * 2] != junk && ovec[i * 2] != PCRE2_UNSET) { + elt = &g_array_index(captures, struct rspamd_re_capture, i); + elt->p = mt + ovec[i * 2]; + elt->len = (mt + ovec[i * 2 + 1]) - elt->p; + } + else { + g_array_set_size(captures, i); + break; + } + } + } + + ret = TRUE; + + if (re->flags & RSPAMD_REGEXP_FLAG_FULL_MATCH) { + /* We also ensure that the match is full */ + if (ovec[0] != 0 || (guint) ovec[1] < len) { + ret = FALSE; + } + } + } + + pcre2_match_data_free(match_data); + + return ret; +} +#endif + +const char * +rspamd_regexp_get_pattern(const rspamd_regexp_t *re) +{ + g_assert(re != NULL); + + return re->pattern; +} + +guint rspamd_regexp_set_flags(rspamd_regexp_t *re, guint new_flags) +{ + guint old_flags; + + g_assert(re != NULL); + old_flags = re->flags; + re->flags = new_flags; + + return old_flags; +} + +guint rspamd_regexp_get_flags(const rspamd_regexp_t *re) +{ + g_assert(re != NULL); + + return re->flags; +} + +guint rspamd_regexp_get_pcre_flags(const rspamd_regexp_t *re) +{ + g_assert(re != NULL); + + return re->pcre_flags; +} + +guint rspamd_regexp_get_maxhits(const rspamd_regexp_t *re) +{ + g_assert(re != NULL); + + return re->max_hits; +} + +guint rspamd_regexp_set_maxhits(rspamd_regexp_t *re, guint new_maxhits) +{ + guint old_hits; + + g_assert(re != NULL); + old_hits = re->max_hits; + re->max_hits = new_maxhits; + + return old_hits; +} + +guint64 +rspamd_regexp_get_cache_id(const rspamd_regexp_t *re) +{ + g_assert(re != NULL); + + return re->cache_id; +} + +guint64 +rspamd_regexp_set_cache_id(rspamd_regexp_t *re, guint64 id) +{ + guint64 old; + + g_assert(re != NULL); + old = re->cache_id; + re->cache_id = id; + + return old; +} + +gsize rspamd_regexp_get_match_limit(const rspamd_regexp_t *re) +{ + g_assert(re != NULL); + + return re->match_limit; +} + +gsize rspamd_regexp_set_match_limit(rspamd_regexp_t *re, gsize lim) +{ + gsize old; + + g_assert(re != NULL); + old = re->match_limit; + re->match_limit = lim; + + return old; +} + +gboolean +rspamd_regexp_match(const rspamd_regexp_t *re, const gchar *text, gsize len, + gboolean raw) +{ + const gchar *start = NULL, *end = NULL; + + g_assert(re != NULL); + g_assert(text != NULL); + + if (len == 0) { + len = strlen(text); + } + + if (rspamd_regexp_search(re, text, len, &start, &end, raw, NULL)) { + if (start == text && end == text + len) { + return TRUE; + } + } + + return FALSE; +} + +void rspamd_regexp_unref(rspamd_regexp_t *re) +{ + REF_RELEASE(re); +} + +rspamd_regexp_t * +rspamd_regexp_ref(rspamd_regexp_t *re) +{ + g_assert(re != NULL); + + REF_RETAIN(re); + + return re; +} + +void rspamd_regexp_set_ud(rspamd_regexp_t *re, gpointer ud) +{ + g_assert(re != NULL); + + re->ud = ud; +} + +gpointer +rspamd_regexp_get_ud(const rspamd_regexp_t *re) +{ + g_assert(re != NULL); + + return re->ud; +} + +gboolean +rspamd_regexp_equal(gconstpointer a, gconstpointer b) +{ + const guchar *ia = a, *ib = b; + + return (memcmp(ia, ib, sizeof(regexp_id_t)) == 0); +} + +guint32 +rspamd_regexp_hash(gconstpointer a) +{ + const guchar *ia = a; + guint32 res; + + memcpy(&res, ia, sizeof(res)); + + return res; +} + +gboolean +rspamd_regexp_cmp(gconstpointer a, gconstpointer b) +{ + const guchar *ia = a, *ib = b; + + return memcmp(ia, ib, sizeof(regexp_id_t)); +} + +struct rspamd_regexp_cache * +rspamd_regexp_cache_new(void) +{ + struct rspamd_regexp_cache *ncache; + + ncache = g_malloc0(sizeof(*ncache)); + ncache->tbl = g_hash_table_new_full(rspamd_regexp_hash, rspamd_regexp_equal, + NULL, (GDestroyNotify) rspamd_regexp_unref); +#ifdef HAVE_PCRE_JIT +#ifdef WITH_PCRE2 + ncache->jstack = pcre2_jit_stack_create(32 * 1024, 1024 * 1024, NULL); +#else + ncache->jstack = pcre_jit_stack_alloc(32 * 1024, 1024 * 1024); +#endif +#endif + return ncache; +} + + +rspamd_regexp_t * +rspamd_regexp_cache_query(struct rspamd_regexp_cache *cache, + const gchar *pattern, + const gchar *flags) +{ + rspamd_regexp_t *res = NULL; + regexp_id_t id; + + if (cache == NULL) { + rspamd_regexp_library_init(NULL); + cache = global_re_cache; + } + + g_assert(cache != NULL); + rspamd_regexp_generate_id(pattern, flags, id); + + res = g_hash_table_lookup(cache->tbl, id); + + return res; +} + + +rspamd_regexp_t * +rspamd_regexp_cache_create(struct rspamd_regexp_cache *cache, + const gchar *pattern, + const gchar *flags, GError **err) +{ + rspamd_regexp_t *res; + + if (cache == NULL) { + rspamd_regexp_library_init(NULL); + cache = global_re_cache; + } + + g_assert(cache != NULL); + res = rspamd_regexp_cache_query(cache, pattern, flags); + + if (res != NULL) { + return res; + } + + res = rspamd_regexp_new(pattern, flags, err); + + if (res) { + /* REF_RETAIN (res); */ + if (g_hash_table_size(cache->tbl) < max_re_cache_size) { + g_hash_table_insert(cache->tbl, res->id, res); + } + else { + msg_warn("cannot insert regexp to the cache: maximum size is reached (%d expressions); " + "it might be cached regexp misuse; regexp pattern: %s", + max_re_cache_size, pattern); + } + } + + return res; +} + +gboolean +rspamd_regexp_cache_remove(struct rspamd_regexp_cache *cache, + rspamd_regexp_t *re) +{ + if (cache == NULL) { + cache = global_re_cache; + } + + g_assert(cache != NULL); + g_assert(re != NULL); + + return g_hash_table_remove(cache->tbl, re->id); +} + +void rspamd_regexp_cache_destroy(struct rspamd_regexp_cache *cache) +{ + if (cache != NULL) { + g_hash_table_destroy(cache->tbl); +#ifdef HAVE_PCRE_JIT +#ifdef WITH_PCRE2 + if (cache->jstack) { + pcre2_jit_stack_free(cache->jstack); + } +#else + if (cache->jstack) { + pcre_jit_stack_free(cache->jstack); + } +#endif +#endif + g_free(cache); + } +} + +RSPAMD_CONSTRUCTOR(rspamd_re_static_pool_ctor) +{ + global_re_cache = rspamd_regexp_cache_new(); +#ifdef WITH_PCRE2 + pcre2_ctx = pcre2_compile_context_create(NULL); + pcre2_set_newline(pcre2_ctx, PCRE_FLAG(NEWLINE_ANY)); +#endif +} + +RSPAMD_DESTRUCTOR(rspamd_re_static_pool_dtor) +{ + rspamd_regexp_cache_destroy(global_re_cache); +#ifdef WITH_PCRE2 + pcre2_compile_context_free(pcre2_ctx); +#endif +} + + +void rspamd_regexp_library_init(struct rspamd_config *cfg) +{ + if (cfg) { + if (cfg->disable_pcre_jit) { + can_jit = FALSE; + check_jit = FALSE; + } + else if (!can_jit) { + check_jit = TRUE; + } + } + + if (check_jit) { +#ifdef HAVE_PCRE_JIT + gint jit, rc; + gchar *str; + +#ifndef WITH_PCRE2 + rc = pcre_config(PCRE_CONFIG_JIT, &jit); +#else + rc = pcre2_config(PCRE2_CONFIG_JIT, &jit); +#endif + + if (rc == 0 && jit == 1) { +#ifndef WITH_PCRE2 +#ifdef PCRE_CONFIG_JITTARGET + pcre_config(PCRE_CONFIG_JITTARGET, &str); + msg_info("pcre is compiled with JIT for %s", str); +#else + msg_info("pcre is compiled with JIT for unknown target"); +#endif +#else + rc = pcre2_config(PCRE2_CONFIG_JITTARGET, NULL); + + if (rc > 0) { + str = g_alloca(rc); + pcre2_config(PCRE2_CONFIG_JITTARGET, str); + msg_info("pcre2 is compiled with JIT for %s", str); + } + else { + msg_info("pcre2 is compiled with JIT for unknown"); + } + +#endif /* WITH_PCRE2 */ + + if (getenv("VALGRIND") == NULL) { + can_jit = TRUE; + } + else { + msg_info("disabling PCRE jit as it does not play well with valgrind"); + can_jit = FALSE; + } + } + else { + msg_info("pcre is compiled without JIT support, so many optimizations" + " are impossible"); + can_jit = FALSE; + } +#else + msg_info("pcre is too old and has no JIT support, so many optimizations" + " are impossible"); + can_jit = FALSE; +#endif + check_jit = FALSE; + } +} + +gpointer +rspamd_regexp_get_id(const rspamd_regexp_t *re) +{ + g_assert(re != NULL); + + return (gpointer) re->id; +} + +gpointer +rspamd_regexp_get_class(const rspamd_regexp_t *re) +{ + g_assert(re != NULL); + + return re->re_class; +} + +gpointer +rspamd_regexp_set_class(rspamd_regexp_t *re, gpointer re_class) +{ + gpointer old_class; + + g_assert(re != NULL); + + old_class = re->re_class; + re->re_class = re_class; + + return old_class; +} + +rspamd_regexp_t * +rspamd_regexp_from_glob(const gchar *gl, gsize sz, GError **err) +{ + GString *out; + rspamd_regexp_t *re; + const gchar *end; + gboolean escaping = FALSE; + gint nbraces = 0; + + g_assert(gl != NULL); + + if (sz == 0) { + sz = strlen(gl); + } + + end = gl + sz; + out = g_string_sized_new(sz + 2); + g_string_append_c(out, '^'); + + while (gl < end) { + switch (*gl) { + case '*': + if (escaping) { + g_string_append(out, "\\*"); + } + else { + g_string_append(out, ".*"); + } + + escaping = FALSE; + break; + case '?': + if (escaping) { + g_string_append(out, "\\?"); + } + else { + g_string_append(out, "."); + } + + escaping = FALSE; + break; + case '.': + case '(': + case ')': + case '+': + case '|': + case '^': + case '$': + case '@': + case '%': + g_string_append_c(out, '\\'); + g_string_append_c(out, *gl); + escaping = FALSE; + break; + case '\\': + if (escaping) { + g_string_append(out, "\\\\"); + escaping = FALSE; + } + else { + escaping = TRUE; + } + break; + case '{': + if (escaping) { + g_string_append(out, "\\{"); + } + else { + g_string_append_c(out, '('); + nbraces++; + } + + escaping = FALSE; + break; + case '}': + if (nbraces > 0 && !escaping) { + g_string_append_c(out, ')'); + nbraces--; + } + else if (escaping) { + g_string_append(out, "\\}"); + } + else { + g_string_append(out, "}"); + } + + escaping = FALSE; + break; + case ',': + if (nbraces > 0 && !escaping) { + g_string_append_c(out, '|'); + } + else if (escaping) { + g_string_append(out, "\\,"); + } + else { + g_string_append_c(out, ','); + } + + break; + default: + escaping = FALSE; + g_string_append_c(out, *gl); + break; + } + + gl++; + } + + g_string_append_c(out, '$'); + re = rspamd_regexp_new(out->str, "i", err); + g_string_free(out, TRUE); + + return re; +} |