Fix: consumerd: add missing put_subbuf for ust and kernel errors

[lttng-tools.git] / src / common / ust-consumer / ust-consumer.c

diff --git a/src/common/ust-consumer/ust-consumer.c b/src/common/ust-consumer/ust-consumer.c
index 7bbfa60a26fc6f90aed740869b0773994cd86ca2..97f0497cb12b14a71da7289c5030b256a8848ebd 100644 (file)
--- a/src/common/ust-consumer/ust-consumer.c
+++ b/src/common/ust-consumer/ust-consumer.c
@@ -2496,6 +2496,8 @@ retry:
                index.offset = htobe64(stream->out_fd_offset);
                ret = get_index_values(&index, ustream);
                if (ret < 0) {
                index.offset = htobe64(stream->out_fd_offset);
                ret = get_index_values(&index, ustream);
                if (ret < 0) {

+                       err = ustctl_put_subbuf(ustream);
+                       assert(err == 0);

                        goto end;
                }
 
                        goto end;
                }
 


@@ -2503,6 +2505,8 @@ retry:
                ret = update_stream_stats(stream);
                if (ret < 0) {
                        PERROR("kernctl_get_events_discarded");
                ret = update_stream_stats(stream);
                if (ret < 0) {
                        PERROR("kernctl_get_events_discarded");

+                       err = ustctl_put_subbuf(ustream);
+                       assert(err == 0);

                        goto end;
                }
        } else {
                        goto end;
                }
        } else {