diff options
Diffstat (limited to 'src/libstat/classifiers/classifiers.h')
| -rw-r--r-- | src/libstat/classifiers/classifiers.h | 75 |
1 files changed, 45 insertions, 30 deletions
diff --git a/src/libstat/classifiers/classifiers.h b/src/libstat/classifiers/classifiers.h index 738a5e8c9..4e159fb13 100644 --- a/src/libstat/classifiers/classifiers.h +++ b/src/libstat/classifiers/classifiers.h @@ -9,6 +9,10 @@ /* Consider this value as 0 */ #define ALPHA 0.0001 +#ifdef __cplusplus +extern "C" { +#endif + struct rspamd_classifier_config; struct rspamd_task; struct rspamd_config; @@ -18,49 +22,58 @@ struct token_node_s; struct rspamd_stat_classifier { char *name; - gboolean (*init_func)(struct rspamd_config *cfg, - struct ev_loop *ev_base, - struct rspamd_classifier *cl); - gboolean (*classify_func)(struct rspamd_classifier * ctx, - GPtrArray *tokens, - struct rspamd_task *task); - gboolean (*learn_spam_func)(struct rspamd_classifier * ctx, - GPtrArray *input, - struct rspamd_task *task, - gboolean is_spam, - gboolean unlearn, - GError **err); - void (*fin_func)(struct rspamd_classifier *cl); + + gboolean (*init_func) (struct rspamd_config *cfg, + struct ev_loop *ev_base, + struct rspamd_classifier *cl); + + gboolean (*classify_func) (struct rspamd_classifier *ctx, + GPtrArray *tokens, + struct rspamd_task *task); + + gboolean (*learn_spam_func) (struct rspamd_classifier *ctx, + GPtrArray *input, + struct rspamd_task *task, + gboolean is_spam, + gboolean unlearn, + GError **err); + + void (*fin_func) (struct rspamd_classifier *cl); }; /* Bayes algorithm */ gboolean bayes_init (struct rspamd_config *cfg, struct ev_loop *ev_base, struct rspamd_classifier *); + gboolean bayes_classify (struct rspamd_classifier *ctx, - GPtrArray *tokens, - struct rspamd_task *task); + GPtrArray *tokens, + struct rspamd_task *task); + gboolean bayes_learn_spam (struct rspamd_classifier *ctx, - GPtrArray *tokens, - struct rspamd_task *task, - gboolean is_spam, - gboolean unlearn, - GError **err); + GPtrArray *tokens, + struct rspamd_task *task, + gboolean is_spam, + gboolean unlearn, + GError **err); + void bayes_fin (struct rspamd_classifier *); /* Generic lua classifier */ gboolean lua_classifier_init (struct rspamd_config *cfg, struct ev_loop *ev_base, struct rspamd_classifier *); + gboolean lua_classifier_classify (struct rspamd_classifier *ctx, - GPtrArray *tokens, - struct rspamd_task *task); + GPtrArray *tokens, + struct rspamd_task *task); + gboolean lua_classifier_learn_spam (struct rspamd_classifier *ctx, - GPtrArray *tokens, - struct rspamd_task *task, - gboolean is_spam, - gboolean unlearn, - GError **err); + GPtrArray *tokens, + struct rspamd_task *task, + gboolean is_spam, + gboolean unlearn, + GError **err); extern guint rspamd_bayes_log_id; #define msg_debug_bayes(...) rspamd_conditional_debug_fast (NULL, task->from_addr, \ @@ -68,7 +81,9 @@ extern guint rspamd_bayes_log_id; G_STRFUNC, \ __VA_ARGS__) + +#ifdef __cplusplus +} +#endif + #endif -/* - * vi:ts=4 - */ |