diff options
Diffstat (limited to 'src/libstat/backends')
-rw-r--r-- | src/libstat/backends/backends.h | 41 | ||||
-rw-r--r-- | src/libstat/backends/mmaped_file.c | 12 | ||||
-rw-r--r-- | src/libstat/backends/sqlite3_backend.c | 14 |
3 files changed, 39 insertions, 28 deletions
diff --git a/src/libstat/backends/backends.h b/src/libstat/backends/backends.h index f970a297b..e40bb6c9c 100644 --- a/src/libstat/backends/backends.h +++ b/src/libstat/backends/backends.h @@ -45,15 +45,19 @@ struct rspamd_stat_backend { gpointer (*init)(struct rspamd_stat_ctx *ctx, struct rspamd_config *cfg); gpointer (*runtime)(struct rspamd_task *task, struct rspamd_statfile_config *stcf, gboolean learn, gpointer ctx); - gboolean (*process_token)(struct token_node_s *tok, + gboolean (*process_token)(struct rspamd_task *task, struct token_node_s *tok, struct rspamd_token_result *res, gpointer ctx); - gboolean (*learn_token)(struct token_node_s *tok, + gboolean (*learn_token)(struct rspamd_task *task, struct token_node_s *tok, struct rspamd_token_result *res, gpointer ctx); - gulong (*total_learns)(struct rspamd_statfile_runtime *runtime, gpointer ctx); - void (*finalize_learn)(struct rspamd_statfile_runtime *runtime, gpointer ctx); - gulong (*inc_learns)(struct rspamd_statfile_runtime *runtime, gpointer ctx); - gulong (*dec_learns)(struct rspamd_statfile_runtime *runtime, gpointer ctx); - ucl_object_t* (*get_stat)(struct rspamd_statfile_runtime *runtime, gpointer ctx); + gulong (*total_learns)(struct rspamd_task *task, + gpointer runtime, gpointer ctx); + void (*finalize_learn)(struct rspamd_task *task, + gpointer runtime, gpointer ctx); + gulong (*inc_learns)(struct rspamd_task *task, + gpointer runtime, gpointer ctx); + gulong (*dec_learns)(struct rspamd_task *task, + gpointer runtime, gpointer ctx); + ucl_object_t* (*get_stat)(gpointer runtime, gpointer ctx); void (*close)(gpointer ctx); gpointer ctx; }; @@ -63,21 +67,28 @@ struct rspamd_stat_backend { gpointer rspamd_##name##_runtime (struct rspamd_task *task, \ struct rspamd_statfile_config *stcf, \ gboolean learn, gpointer ctx); \ - gboolean rspamd_##name##_process_token (struct token_node_s *tok, \ + gboolean rspamd_##name##_process_token (struct rspamd_task *task, \ + struct token_node_s *tok, \ struct rspamd_token_result *res, \ gpointer ctx); \ - gboolean rspamd_##name##_learn_token (struct token_node_s *tok, \ + gboolean rspamd_##name##_learn_token (struct rspamd_task *task, \ + struct token_node_s *tok, \ struct rspamd_token_result *res, \ gpointer ctx); \ - void rspamd_##name##_finalize_learn (gpointer runtime, \ + void rspamd_##name##_finalize_learn (struct rspamd_task *task, \ + gpointer runtime, \ gpointer ctx); \ - gulong rspamd_##name##_total_learns (gpointer runtime, \ + gulong rspamd_##name##_total_learns (struct rspamd_task *task, \ + gpointer runtime, \ gpointer ctx); \ - gulong rspamd_##name##_inc_learns (gpointer runtime, \ + gulong rspamd_##name##_inc_learns (struct rspamd_task *task, \ + gpointer runtime, \ gpointer ctx); \ - gulong rspamd_##name##_dec_learns (gpointer runtime, \ - gpointer ctx); \ - gulong rspamd_##name##_learns (gpointer runtime, \ + gulong rspamd_##name##_dec_learns (struct rspamd_task *task, \ + gpointer runtime, \ + gpointer ctx); \ + gulong rspamd_##name##_learns (struct rspamd_task *task, \ + gpointer runtime, \ gpointer ctx); \ ucl_object_t * rspamd_##name##_get_stat (gpointer runtime, \ gpointer ctx); \ diff --git a/src/libstat/backends/mmaped_file.c b/src/libstat/backends/mmaped_file.c index 7c12e092d..b7e5650d3 100644 --- a/src/libstat/backends/mmaped_file.c +++ b/src/libstat/backends/mmaped_file.c @@ -937,7 +937,7 @@ rspamd_mmaped_file_runtime (struct rspamd_task *task, } gboolean -rspamd_mmaped_file_process_token (rspamd_token_t *tok, +rspamd_mmaped_file_process_token (struct rspamd_task *task, rspamd_token_t *tok, struct rspamd_token_result *res, gpointer p) { @@ -971,7 +971,7 @@ rspamd_mmaped_file_process_token (rspamd_token_t *tok, } gboolean -rspamd_mmaped_file_learn_token (rspamd_token_t *tok, +rspamd_mmaped_file_learn_token (struct rspamd_task *task, rspamd_token_t *tok, struct rspamd_token_result *res, gpointer p) { @@ -1001,7 +1001,7 @@ rspamd_mmaped_file_learn_token (rspamd_token_t *tok, } gulong -rspamd_mmaped_file_total_learns (gpointer runtime, +rspamd_mmaped_file_total_learns (struct rspamd_task *task, gpointer runtime, gpointer ctx) { rspamd_mmaped_file_t *mf = (rspamd_mmaped_file_t *)runtime; @@ -1016,7 +1016,7 @@ rspamd_mmaped_file_total_learns (gpointer runtime, } gulong -rspamd_mmaped_file_inc_learns (gpointer runtime, +rspamd_mmaped_file_inc_learns (struct rspamd_task *task, gpointer runtime, gpointer ctx) { rspamd_mmaped_file_t *mf = (rspamd_mmaped_file_t *)runtime; @@ -1032,7 +1032,7 @@ rspamd_mmaped_file_inc_learns (gpointer runtime, } gulong -rspamd_mmaped_file_dec_learns (gpointer runtime, +rspamd_mmaped_file_dec_learns (struct rspamd_task *task, gpointer runtime, gpointer ctx) { rspamd_mmaped_file_t *mf = (rspamd_mmaped_file_t *)runtime; @@ -1080,7 +1080,7 @@ rspamd_mmaped_file_get_stat (gpointer runtime, } void -rspamd_mmaped_file_finalize_learn (gpointer runtime, +rspamd_mmaped_file_finalize_learn (struct rspamd_task *task, gpointer runtime, gpointer ctx) { rspamd_mmaped_file_t *mf = (rspamd_mmaped_file_t *)runtime; diff --git a/src/libstat/backends/sqlite3_backend.c b/src/libstat/backends/sqlite3_backend.c index ea52314fc..9eee2c614 100644 --- a/src/libstat/backends/sqlite3_backend.c +++ b/src/libstat/backends/sqlite3_backend.c @@ -476,7 +476,7 @@ rspamd_sqlite3_runtime (struct rspamd_task *task, } gboolean -rspamd_sqlite3_process_token (struct token_node_s *tok, +rspamd_sqlite3_process_token (struct rspamd_task *task, struct token_node_s *tok, struct rspamd_token_result *res, gpointer p) { struct rspamd_stat_sqlite3_db *bk; @@ -513,7 +513,7 @@ rspamd_sqlite3_process_token (struct token_node_s *tok, } gboolean -rspamd_sqlite3_learn_token (struct token_node_s *tok, +rspamd_sqlite3_learn_token (struct rspamd_task *task, struct token_node_s *tok, struct rspamd_token_result *res, gpointer p) { struct rspamd_stat_sqlite3_db *bk; @@ -549,7 +549,7 @@ rspamd_sqlite3_learn_token (struct token_node_s *tok, } void -rspamd_sqlite3_finalize_learn (gpointer runtime, +rspamd_sqlite3_finalize_learn (struct rspamd_task *task, gpointer runtime, gpointer ctx) { struct rspamd_stat_sqlite3_db *bk = runtime; @@ -565,7 +565,7 @@ rspamd_sqlite3_finalize_learn (gpointer runtime, } gulong -rspamd_sqlite3_total_learns (gpointer runtime, +rspamd_sqlite3_total_learns (struct rspamd_task *task, gpointer runtime, gpointer ctx) { struct rspamd_stat_sqlite3_db *bk = runtime; @@ -579,7 +579,7 @@ rspamd_sqlite3_total_learns (gpointer runtime, } gulong -rspamd_sqlite3_inc_learns (gpointer runtime, +rspamd_sqlite3_inc_learns (struct rspamd_task *task, gpointer runtime, gpointer ctx) { struct rspamd_stat_sqlite3_db *bk = runtime; @@ -594,7 +594,7 @@ rspamd_sqlite3_inc_learns (gpointer runtime, } gulong -rspamd_sqlite3_dec_learns (gpointer runtime, +rspamd_sqlite3_dec_learns (struct rspamd_task *task, gpointer runtime, gpointer ctx) { struct rspamd_stat_sqlite3_db *bk = runtime; @@ -609,7 +609,7 @@ rspamd_sqlite3_dec_learns (gpointer runtime, } gulong -rspamd_sqlite3_learns (gpointer runtime, +rspamd_sqlite3_learns (struct rspamd_task *task, gpointer runtime, gpointer ctx) { struct rspamd_stat_sqlite3_db *bk = runtime; |