--- /dev/null
+/* GStreamer
+ *
+ * Unit test for leakstracer
+ *
+ * Copyright (C) <2019> Nirbheek Chauhan <nirbheek@centricular.com>
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Library General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Library General Public License for more details.
+ *
+ * You should have received a copy of the GNU Library General Public
+ * License along with this library; if not, write to the
+ * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
+ * Boston, MA 02110-1301, USA.
+ */
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <gst/gst.h>
+#include <gst/check/gstcheck.h>
+
+#define PROBE_TYPE GST_PAD_PROBE_TYPE_BUFFER | GST_PAD_PROBE_TYPE_BLOCK
+
+#define NUM_BUFFERS 2
+
+struct RetBufferCtx
+{
+ GstBuffer *bufs[NUM_BUFFERS];
+ guint idx;
+};
+
+static void
+ret_buffer_ctx_free (struct RetBufferCtx *ctx, gboolean free_bufs)
+{
+ guint ii;
+ if (free_bufs)
+ for (ii = 0; ii < ctx->idx; ii++)
+ gst_buffer_unref (ctx->bufs[ii]);
+ g_free (ctx);
+}
+
+static GstPadProbeReturn
+ref_buffer (GstPad * srcpad, GstPadProbeInfo * info, gpointer user_data)
+{
+ GstBuffer *buffer;
+ struct RetBufferCtx *ctx = user_data;
+
+ if (!(GST_PAD_PROBE_INFO_TYPE (info) & GST_PAD_PROBE_TYPE_BUFFER))
+ return GST_PAD_PROBE_PASS;
+
+ buffer = GST_PAD_PROBE_INFO_BUFFER (info);
+ /* ref buffer so it leaks */
+ gst_buffer_ref (buffer);
+
+ if (ctx) {
+ /* we can only store NUM_BUFFERS buffers */
+ fail_unless (ctx->idx < NUM_BUFFERS);
+ /* return the buffer so it can be freed later to avoid triggering valgrind
+ * in gst-validate */
+ ctx->bufs[ctx->idx] = buffer;
+ ctx->idx++;
+ }
+
+ return GST_PAD_PROBE_PASS;
+}
+
+static GstTracer *
+get_tracer_by_name (const gchar * name)
+{
+ GList *tracers, *l;
+ GstTracer *tracer = NULL;
+
+ tracers = gst_tracing_get_active_tracers ();
+ for (l = tracers; l; l = l->next)
+ if (g_strcmp0 (GST_OBJECT_NAME (l->data), name) == 0)
+ tracer = l->data;
+
+ g_list_free (tracers);
+ return tracer;
+}
+
+/* Test logging of live objects to debug logs */
+GST_START_TEST (test_log_live_objects)
+{
+ GstElement *pipe, *src, *sink;
+ GstPad *srcpad;
+ GstMessage *m;
+ struct RetBufferCtx *ctx = g_new0 (struct RetBufferCtx, 1);
+
+ pipe = gst_pipeline_new ("pipeline");
+ fail_unless (pipe);
+ src = gst_element_factory_make ("fakesrc", NULL);
+ fail_unless (src);
+ g_object_set (src, "num-buffers", NUM_BUFFERS, NULL);
+
+ sink = gst_element_factory_make ("fakesink", NULL);
+ fail_unless (sink);
+
+ gst_bin_add_many (GST_BIN (pipe), src, sink, NULL);
+ fail_unless (gst_element_link (src, sink));
+
+ srcpad = gst_element_get_static_pad (src, "src");
+ gst_pad_add_probe (srcpad, PROBE_TYPE, ref_buffer, ctx, NULL);
+ gst_object_unref (srcpad);
+
+ GST_DEBUG ("Setting pipeline to PLAYING");
+ fail_unless_equals_int (gst_element_set_state (pipe, GST_STATE_PLAYING),
+ GST_STATE_CHANGE_ASYNC);
+
+ m = gst_bus_timed_pop_filtered (GST_ELEMENT_BUS (pipe), -1, GST_MESSAGE_EOS);
+ gst_message_unref (m);
+
+ fail_unless_equals_int (gst_element_set_state (pipe, GST_STATE_NULL),
+ GST_STATE_CHANGE_SUCCESS);
+ gst_object_unref (pipe);
+
+ /* Check the live-objects data returned by the tracer */
+ {
+ GstTracer *tracer = get_tracer_by_name ("plain");
+ fail_unless (tracer);
+ g_signal_emit_by_name (tracer, "log-live-objects");
+ gst_object_unref (tracer);
+ }
+
+ ret_buffer_ctx_free (ctx, TRUE);
+}
+
+GST_END_TEST;
+
+/* Test fetching of live objects with no detail */
+GST_START_TEST (test_get_live_objects)
+{
+ GstElement *pipe, *src, *sink;
+ GstPad *srcpad;
+ GstMessage *m;
+ struct RetBufferCtx *ctx = g_new0 (struct RetBufferCtx, 1);
+
+ pipe = gst_pipeline_new ("pipeline");
+ fail_unless (pipe);
+ src = gst_element_factory_make ("fakesrc", NULL);
+ fail_unless (src);
+ g_object_set (src, "num-buffers", NUM_BUFFERS, NULL);
+
+ sink = gst_element_factory_make ("fakesink", NULL);
+ fail_unless (sink);
+
+ gst_bin_add_many (GST_BIN (pipe), src, sink, NULL);
+ fail_unless (gst_element_link (src, sink));
+
+ srcpad = gst_element_get_static_pad (src, "src");
+ gst_pad_add_probe (srcpad, PROBE_TYPE, ref_buffer, ctx, NULL);
+ gst_object_unref (srcpad);
+
+ GST_DEBUG ("Setting pipeline to PLAYING");
+ fail_unless_equals_int (gst_element_set_state (pipe, GST_STATE_PLAYING),
+ GST_STATE_CHANGE_ASYNC);
+
+ m = gst_bus_timed_pop_filtered (GST_ELEMENT_BUS (pipe), -1, GST_MESSAGE_EOS);
+ gst_message_unref (m);
+
+ fail_unless_equals_int (gst_element_set_state (pipe, GST_STATE_NULL),
+ GST_STATE_CHANGE_SUCCESS);
+ gst_object_unref (pipe);
+
+ /* Check the live-objects data returned by the tracer */
+ {
+ guint ii, size;
+ GstStructure *info;
+ const GValue *leaks;
+ GstTracer *tracer = get_tracer_by_name ("plain");
+ fail_unless (tracer);
+ g_signal_emit_by_name (tracer, "get-live-objects", &info);
+ fail_unless_equals_int (gst_structure_n_fields (info), 1);
+ leaks = gst_structure_get_value (info, "live-objects-list");
+ fail_unless (G_VALUE_HOLDS (leaks, GST_TYPE_LIST));
+ size = gst_value_list_get_size (leaks);
+ fail_unless_equals_int (size, NUM_BUFFERS);
+ for (ii = 0; ii < size; ii++) {
+ const GValue *v;
+ const GstStructure *s;
+ guint ref_count;
+
+ v = gst_value_list_get_value (leaks, ii);
+ fail_unless (G_VALUE_HOLDS (v, GST_TYPE_STRUCTURE));
+
+ s = gst_value_get_structure (v);
+ fail_unless (gst_structure_has_field_typed (s, "object",
+ GST_TYPE_BUFFER));
+
+ fail_unless (gst_structure_has_field_typed (s, "ref-count", G_TYPE_UINT));
+ fail_unless (gst_structure_get_uint (s, "ref-count", &ref_count));
+ fail_unless_equals_int (ref_count, 1);
+
+ fail_unless (gst_structure_has_field_typed (s, "trace", G_TYPE_STRING));
+ fail_unless_equals_string (gst_structure_get_string (s, "trace"), NULL);
+
+ fail_unless (!gst_structure_has_field (s, "ref-infos"));
+ fail_unless_equals_int (gst_structure_n_fields (s), 3);
+ }
+ gst_structure_free (info);
+ gst_object_unref (tracer);
+ }
+
+ /* leaked buffers were freed above with @info */
+ ret_buffer_ctx_free (ctx, FALSE);
+}
+
+GST_END_TEST;
+
+/* Test fetching of filtered live objects with full detail */
+GST_START_TEST (test_get_live_objects_filtered_detailed)
+{
+ GstElement *pipe, *src, *sink;
+ GstPad *srcpad;
+ GstMessage *m;
+ struct RetBufferCtx *ctx = g_new0 (struct RetBufferCtx, 1);
+
+ pipe = gst_pipeline_new ("pipeline");
+ fail_unless (pipe);
+ src = gst_element_factory_make ("fakesrc", NULL);
+ fail_unless (src);
+ g_object_set (src, "num-buffers", NUM_BUFFERS, NULL);
+
+ sink = gst_element_factory_make ("fakesink", NULL);
+ fail_unless (sink);
+
+ gst_bin_add_many (GST_BIN (pipe), src, sink, NULL);
+ fail_unless (gst_element_link (src, sink));
+
+ srcpad = gst_element_get_static_pad (src, "src");
+ gst_pad_add_probe (srcpad, PROBE_TYPE, ref_buffer, ctx, NULL);
+ /* leak srcpad on purpose */
+ gst_element_get_static_pad (sink, "sink");
+ /* leak sinkpad on purpose */
+
+ GST_DEBUG ("Setting pipeline to PLAYING");
+ fail_unless_equals_int (gst_element_set_state (pipe, GST_STATE_PLAYING),
+ GST_STATE_CHANGE_ASYNC);
+
+ m = gst_bus_timed_pop_filtered (GST_ELEMENT_BUS (pipe), -1, GST_MESSAGE_EOS);
+ gst_message_unref (m);
+
+ fail_unless_equals_int (gst_element_set_state (pipe, GST_STATE_NULL),
+ GST_STATE_CHANGE_SUCCESS);
+ gst_object_unref (pipe);
+
+ /* Check the live-objects data returned by the tracer */
+ {
+ guint ii, jj, isize, jsize;
+ GstStructure *info;
+ const GValue *leaks;
+ GstTracer *tracer = get_tracer_by_name ("more");
+ fail_unless (tracer);
+ g_signal_emit_by_name (tracer, "get-live-objects", &info);
+ fail_unless_equals_int (gst_structure_n_fields (info), 1);
+ leaks = gst_structure_get_value (info, "live-objects-list");
+ fail_unless (G_VALUE_HOLDS (leaks, GST_TYPE_LIST));
+ isize = gst_value_list_get_size (leaks);
+ fail_unless_equals_int (isize, NUM_BUFFERS);
+ for (ii = 0; ii < isize; ii++) {
+ const GValue *v;
+ const GstStructure *s;
+ guint ref_count;
+
+ v = gst_value_list_get_value (leaks, ii);
+ fail_unless (G_VALUE_HOLDS (v, GST_TYPE_STRUCTURE));
+
+ s = gst_value_get_structure (v);
+ fail_unless (gst_structure_has_field_typed (s, "object", GST_TYPE_PAD));
+
+ fail_unless (gst_structure_has_field_typed (s, "ref-count", G_TYPE_UINT));
+ fail_unless (gst_structure_get_uint (s, "ref-count", &ref_count));
+ fail_unless_equals_int (ref_count, 1);
+
+ fail_unless (gst_structure_has_field_typed (s, "trace", G_TYPE_STRING));
+ fail_unless (gst_structure_get_string (s, "trace"));
+
+ fail_unless (gst_structure_has_field_typed (s, "ref-infos",
+ GST_TYPE_LIST));
+ fail_unless_equals_int (gst_structure_n_fields (s), 4);
+
+ v = gst_structure_get_value (s, "ref-infos");
+ jsize = gst_value_list_get_size (v);
+ for (jj = 0; jj < jsize; jj++) {
+ const GValue *rv;
+ const GstStructure *r;
+
+ rv = gst_value_list_get_value (v, jj);
+ fail_unless (G_VALUE_HOLDS (rv, GST_TYPE_STRUCTURE));
+
+ r = gst_value_get_structure (rv);
+ fail_unless (gst_structure_has_field_typed (r, "ts",
+ GST_TYPE_CLOCK_TIME));
+
+ fail_unless (gst_structure_has_field_typed (r, "desc", G_TYPE_STRING));
+ fail_unless (gst_structure_get_string (r, "desc"));
+
+ fail_unless (gst_structure_get_uint (r, "ref-count", &ref_count));
+ fail_unless (ref_count > 0);
+
+ fail_unless (gst_structure_has_field_typed (r, "trace", G_TYPE_STRING));
+ fail_unless (gst_structure_get_string (r, "trace"));
+
+ fail_unless_equals_int (gst_structure_n_fields (r), 4);
+ }
+ }
+ gst_structure_free (info);
+ gst_object_unref (tracer);
+ }
+
+ ret_buffer_ctx_free (ctx, TRUE);
+ /* leaked pads were freed above with @info */
+}
+
+GST_END_TEST;
+
+static Suite *
+leakstracer_suite (void)
+{
+ Suite *s = suite_create ("leakstracer");
+ TCase *tc_chain_1 = tcase_create ("live-objects");
+
+ suite_add_tcase (s, tc_chain_1);
+ tcase_add_test (tc_chain_1, test_log_live_objects);
+ tcase_add_test (tc_chain_1, test_get_live_objects);
+ tcase_add_test (tc_chain_1, test_get_live_objects_filtered_detailed);
+
+ return s;
+}
+
+/* Replacement for GST_CHECK_MAIN (leakstracer); because we need to set the
+ * env before gst_init() is called */
+int
+main (int argc, char **argv)
+{
+ Suite *s;
+ g_setenv ("GST_TRACERS", "leaks(name=plain,log-leaks-on-deinit=false);"
+ "leaks(name=more,filters=GstPad,check-refs=true,stack-traces-flags=full,log-leaks-on-deinit=false);",
+ TRUE);
+ gst_check_init (&argc, &argv);
+ s = leakstracer_suite ();
+ return gst_check_run_suite (s, "leakstracer", __FILE__);
+}