Browse Source

[Fix] Support v2 checks in controller

tags/1.6.0
Vsevolod Stakhov 7 years ago
parent
commit
c36baca068
1 changed files with 17 additions and 0 deletions
  1. 17
    0
      src/controller.c

+ 17
- 0
src/controller.c View File

@@ -53,6 +53,7 @@
#define PATH_SAVE_MAP "/savemap"
#define PATH_SCAN "/scan"
#define PATH_CHECK "/check"
#define PATH_CHECKV2 "/checkv2"
#define PATH_STAT "/stat"
#define PATH_STAT_RESET "/statreset"
#define PATH_COUNTERS "/counters"
@@ -2055,6 +2056,19 @@ rspamd_controller_handle_scan (struct rspamd_http_connection_entry *conn_ent,
task->flags |= RSPAMD_TASK_FLAG_MIME;
task->resolver = ctx->resolver;

if (!rspamd_protocol_handle_request (task, msg)) {
if (task->err) {
rspamd_controller_send_error (conn_ent, task->err->code, "%s",
task->err->message);
}
else {
rspamd_controller_send_error (conn_ent, 500,
"Message load error: unknown error");
}
rspamd_session_destroy (task->s);
return 0;
}

if (!rspamd_task_load_message (task, msg, msg->body_buf.begin, msg->body_buf.len)) {
if (task->err) {
rspamd_controller_send_error (conn_ent, task->err->code, "%s",
@@ -3675,6 +3689,9 @@ start_controller_worker (struct rspamd_worker *worker)
rspamd_http_router_add_path (ctx->http,
PATH_CHECK,
rspamd_controller_handle_scan);
rspamd_http_router_add_path (ctx->http,
PATH_CHECKV2,
rspamd_controller_handle_scan);
rspamd_http_router_add_path (ctx->http,
PATH_STAT,
rspamd_controller_handle_stat);

Loading…
Cancel
Save