png: fix video state leaks

diff --git a/ext/libpng/gstpngdec.c b/ext/libpng/gstpngdec.c
index 7c4d558..a9c3a32 100644 (file)
--- a/ext/libpng/gstpngdec.c
+++ b/ext/libpng/gstpngdec.c
@@ -508,5 +508,14 @@ gst_pngdec_stop (GstVideoDecoder * decoder)
 
   pngdec->color_type = -1;
 
+  if (pngdec->input_state) {
+    gst_video_codec_state_unref (pngdec->input_state);
+    pngdec->input_state = NULL;
+  }
+  if (pngdec->output_state) {
+    gst_video_codec_state_unref (pngdec->output_state);
+    pngdec->output_state = NULL;
+  }
+
   return TRUE;
 }

diff --git a/ext/libpng/gstpngenc.c b/ext/libpng/gstpngenc.c
index f8dcff4..9d3e1d0 100644 (file)
--- a/ext/libpng/gstpngenc.c
+++ b/ext/libpng/gstpngenc.c
@@ -85,6 +85,8 @@ static gboolean gst_pngenc_set_format (GstVideoEncoder * encoder,
 static gboolean gst_pngenc_propose_allocation (GstVideoEncoder * encoder,
     GstQuery * query);
 
+static void gst_pngenc_finalize (GObject * object);
+
 static void
 user_error_fn (png_structp png_ptr, png_const_charp error_msg)
 {
@@ -137,6 +139,7 @@ gst_pngenc_class_init (GstPngEncClass * klass)
   venc_class->set_format = gst_pngenc_set_format;
   venc_class->handle_frame = gst_pngenc_handle_frame;
   venc_class->propose_allocation = gst_pngenc_propose_allocation;
+  gobject_class->finalize = gst_pngenc_finalize;
 
   GST_DEBUG_CATEGORY_INIT (pngenc_debug, "pngenc", 0, "PNG image encoder");
 }
@@ -204,6 +207,17 @@ gst_pngenc_init (GstPngEnc * pngenc)
 }
 
 static void
+gst_pngenc_finalize (GObject * object)
+{
+  GstPngEnc *pngenc = GST_PNGENC (object);
+
+  if (pngenc->input_state)
+    gst_video_codec_state_unref (pngenc->input_state);
+
+  G_OBJECT_CLASS (parent_class)->finalize (object);
+}
+
+static void
 user_flush_data (png_structp png_ptr G_GNUC_UNUSED)
 {
 }