Fix writer: leak of the event classes

[babeltrace.git] / plugins / writer / write.c

diff --git a/plugins/writer/write.c b/plugins/writer/write.c
index 970eaea0c777a2275cbfbb6aaf792a7b8c069919..5ee5d0662a2c8d4eb8167d26c4a9efc5813a7d28 100644 (file)
--- a/plugins/writer/write.c
+++ b/plugins/writer/write.c
@@ -333,6 +333,7 @@ enum bt_component_status copy_event_classes(FILE *err,
                        goto end;
                }
                bt_put(event_class);
+               bt_put(writer_event_class);
        }
 
 end:
@@ -732,6 +733,7 @@ struct bt_ctf_stream *get_writer_stream(
        } else {
                writer_stream = insert_new_stream(writer_component, ctf_writer,
                                stream_class, stream);
+               bt_get(writer_stream);
        }
 
        bt_put(ctf_writer);