Add method to set re class for any rspamd regexp

diff --git a/src/libutil/regexp.c b/src/libutil/regexp.c
index 00c60a3376dcfdcc9d984a7bb7a1dc99c6ec0a25..530574100ebce9601851c3bb5fada01b3db66b24 100644 (file)
--- a/src/libutil/regexp.c
+++ b/src/libutil/regexp.c
@@ -51,6 +51,7 @@ struct rspamd_regexp_s {
        regexp_id_t id;
        ref_entry_t ref;
        gpointer ud;
+       gpointer re_class;
        guint64 cache_id;
        gint flags;
        gint ncaptures;
@@ -747,3 +748,24 @@ rspamd_regexp_get_id (rspamd_regexp_t *re)
 
        return re->id;
 }
+
+gpointer
+rspamd_regexp_get_class (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;
+}

diff --git a/src/libutil/regexp.h b/src/libutil/regexp.h
index cf012668a2762ced9b4659a99f806b41c4f6b6c0..a67bef9906689593212e6bc1c5be4827a4dc7498 100644 (file)
--- a/src/libutil/regexp.h
+++ b/src/libutil/regexp.h
@@ -132,6 +132,17 @@ guint64 rspamd_regexp_get_cache_id (rspamd_regexp_t *re);
  */
 guint64 rspamd_regexp_set_cache_id (rspamd_regexp_t *re, guint64 id);
 
+/**
+ * Get regexp class for the re object
+ */
+gpointer rspamd_regexp_get_class (rspamd_regexp_t *re);
+
+/**
+ * Set regexp class for the re object
+ * @return old re class value
+ */
+gpointer rspamd_regexp_set_class (rspamd_regexp_t *re, gpointer re_class);
+
 /**
  * Create new regexp cache
  * @return