diff options
Diffstat (limited to 'src/controller.c')
| -rw-r--r-- | src/controller.c | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/controller.c b/src/controller.c index a4c0a7bb4..30daa26fc 100644 --- a/src/controller.c +++ b/src/controller.c @@ -466,7 +466,7 @@ controller_read_socket (f_str_t *in, void *arg) if (session->state == STATE_COMMAND) { session->state = STATE_REPLY; } - if (session->state != STATE_LEARN) { + if (session->state != STATE_LEARN && session->state != STATE_OTHER) { rspamd_dispatcher_write (session->dispatcher, END, sizeof (END) - 1, FALSE, TRUE); } @@ -639,7 +639,8 @@ void register_custom_controller_command (const char *name, controller_func_t handler, gboolean privilleged, gboolean require_message) { struct custom_controller_command *cmd; - + + cmd = g_malloc (sizeof (struct custom_controller_command)); cmd->command = name; cmd->handler = handler; cmd->privilleged = privilleged; |