filter: Fix a memory leak in non-btr mode.

[paraslash.git] / recv.c

diff --git a/recv.c b/recv.c
index 9d6f028319fc6e1e8f36f7406fc64475fb3576df..ca505561f2a1dbb65f4426ff29fd4fb44888a3b0 100644 (file)
--- a/recv.c
+++ b/recv.c
@@ -9,6 +9,7 @@
 #include <regex.h>
 #include <sys/types.h>
 #include <dirent.h>
+#include <stdbool.h>
 
 #include "para.h"
 #include "list.h"
@@ -20,6 +21,7 @@
 #include "string.h"
 #include "error.h"
 #include "stdout.h"
+#include "buffer_tree.h"
 
 /** The gengetopt args info struct. */
 static struct recv_args_info conf;
@@ -90,11 +92,16 @@ int main(int argc, char *argv[])
        }
        r = &receivers[receiver_num];
        rn.receiver = r;
+       if (conf.buffer_tree_given)
+               rn.btrn = btr_new_node("receiver", NULL, NULL, NULL);
        ret = r->open(&rn);
        if (ret < 0)
                goto out;
        r_opened = 1;
 
+       if (conf.buffer_tree_given)
+               sot.btrn = btr_new_node("stdout", rn.btrn, NULL, NULL);
+
        stdout_set_defaults(&sot);
        sot.bufp = &rn.buf;
        sot.loaded = &rn.loaded;
@@ -110,6 +117,11 @@ int main(int argc, char *argv[])
 out:
        if (r_opened)
                r->close(&rn);
+       if (conf.buffer_tree_given) {
+               btr_free_node(rn.btrn);
+               btr_free_node(sot.btrn);
+       }
+       free(rn.conf);
        if (r)
                r->shutdown();
        if (ret < 0)