stats: Make sure unfinished commands get freed when their session disconnects.

diff --git a/src/stats/mail-command.c b/src/stats/mail-command.c
index cab7a42fa68b063acc8d6133183a3d478a28eca7..440b50586ce741d112cf73c39939af313e901dfc 100644 (file)
--- a/src/stats/mail-command.c
+++ b/src/stats/mail-command.c
@@ -153,9 +153,17 @@ int mail_command_update_parse(const char *const *args, const char **error_r)
 
 void mail_commands_free_memory(void)
 {
-       while (stable_mail_commands != NULL &&
-              stable_mail_commands->refcount == 0) {
-               i_assert(stable_mail_commands->id == 0);
+       while (stable_mail_commands != NULL) {
+               struct mail_command *cmd = stable_mail_commands;
+
+               if (cmd->refcount == 0)
+                       i_assert(cmd->id == 0);
+               else if (cmd->refcount == 1 && cmd->session->disconnected) {
+                       /* session was probably lost */
+                       mail_command_unref(&cmd);
+               } else {
+                       break;
+               }
                mail_command_free(stable_mail_commands);
 
                if (global_used_memory < stats_settings->memory_limit)