FaSTTUBe/GST-Tensordecoder-ov_ep
37
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

pnmdec: Partially fix negotiation issues and refcount leaks

Browse Source
This commit is contained in:
Sebastian Dröge 2009-09-10 08:53:46 +02:00
parent cfb04acde5
commit 58a5e422b6
1 changed files with 26 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
gst/pnm/gstpnmdec.c
View File

@ -37,6 +37,7 @@
#include "gstpnmutils.h" #include "gstpnmutils.h"
#include <gst/gstutils.h> #include <gst/gstutils.h>
#include <gst/video/video.h>
#include <string.h> #include <string.h>
@ -48,8 +49,7 @@ static GstElementClass *parent_class;
static GstStaticPadTemplate gst_pnmdec_src_pad_template = static GstStaticPadTemplate gst_pnmdec_src_pad_template =
GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS, GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS,
GST_STATIC_CAPS ("video/x-raw-rgb, bpp = (int) 24, " GST_STATIC_CAPS (GST_VIDEO_CAPS_BGRx));
"width = (int) [ 1, MAX ], height = (int) [ 1, MAX ]"));
static GstStaticPadTemplate gst_pnmdec_sink_pad_template = static GstStaticPadTemplate gst_pnmdec_sink_pad_template =
GST_STATIC_PAD_TEMPLATE ("sink", GST_PAD_SINK, GST_PAD_ALWAYS, GST_STATIC_PAD_TEMPLATE ("sink", GST_PAD_SINK, GST_PAD_ALWAYS,
@ -70,19 +70,25 @@ gst_pnmdec_chain (GstPad * pad, GstBuffer * data)
GST_BUFFER_SIZE (data))) { GST_BUFFER_SIZE (data))) {
case GST_PNM_INFO_MNGR_RESULT_FAILED: case GST_PNM_INFO_MNGR_RESULT_FAILED:
gst_buffer_unref (data); gst_buffer_unref (data);
return GST_FLOW_ERROR; r = GST_FLOW_ERROR;
goto out;
case GST_PNM_INFO_MNGR_RESULT_READING: case GST_PNM_INFO_MNGR_RESULT_READING:
gst_buffer_unref (data); gst_buffer_unref (data);
return GST_FLOW_OK; r = GST_FLOW_OK;
goto out;
case GST_PNM_INFO_MNGR_RESULT_FINISHED: case GST_PNM_INFO_MNGR_RESULT_FINISHED:
offset = s->mngr.data_offset; offset = s->mngr.data_offset;
caps = gst_pad_get_caps (src); caps = gst_pad_get_caps (src);
caps = gst_caps_make_writable (caps);
gst_caps_truncate (caps);
gst_caps_set_simple (caps, gst_caps_set_simple (caps,
"width", G_TYPE_INT, s->mngr.info.width, "width", G_TYPE_INT, s->mngr.info.width,
"height", G_TYPE_INT, s->mngr.info.height, NULL); "height", G_TYPE_INT, s->mngr.info.height, "framerate",
GST_TYPE_FRACTION, 0, 1, NULL);
if (!gst_pad_set_caps (src, caps)) { if (!gst_pad_set_caps (src, caps)) {
gst_caps_unref (caps); gst_caps_unref (caps);
return GST_FLOW_ERROR; r = GST_FLOW_ERROR;
goto out;
} }
gst_caps_unref (caps); gst_caps_unref (caps);
switch (s->mngr.info.type) { switch (s->mngr.info.type) {
@ -100,8 +106,10 @@ gst_pnmdec_chain (GstPad * pad, GstBuffer * data)
} }
} }
if (offset == GST_BUFFER_SIZE (data)) if (offset == GST_BUFFER_SIZE (data)) {
return GST_FLOW_OK; r = GST_FLOW_OK;
goto out;
}
/* If we got the whole image, just push the buffer. */ /* If we got the whole image, just push the buffer. */
if (GST_BUFFER_SIZE (data) - offset == s->size) { if (GST_BUFFER_SIZE (data) - offset == s->size) {
@ -110,7 +118,8 @@ gst_pnmdec_chain (GstPad * pad, GstBuffer * data)
memset (&s->mngr, 0, sizeof (GstPnmInfoMngr)); memset (&s->mngr, 0, sizeof (GstPnmInfoMngr));
s->size = 0; s->size = 0;
gst_buffer_set_caps (buf, GST_PAD_CAPS (src)); gst_buffer_set_caps (buf, GST_PAD_CAPS (src));
return gst_pad_push (src, buf); r = gst_pad_push (src, buf);
goto out;
} }
/* We didn't get the whole image. */ /* We didn't get the whole image. */
@ -123,8 +132,10 @@ gst_pnmdec_chain (GstPad * pad, GstBuffer * data)
gst_buffer_unref (s->buf); gst_buffer_unref (s->buf);
s->buf = buf; s->buf = buf;
} }
if (!s->buf) if (!s->buf) {
return GST_FLOW_ERROR; r = GST_FLOW_ERROR;
goto out;
}
/* Do we now have the full image? If yes, push. */ /* Do we now have the full image? If yes, push. */
if (GST_BUFFER_SIZE (s->buf) == s->size) { if (GST_BUFFER_SIZE (s->buf) == s->size) {
@ -135,6 +146,10 @@ gst_pnmdec_chain (GstPad * pad, GstBuffer * data)
s->size = 0; s->size = 0;
} }
out:
gst_object_unref (src);
gst_object_unref (s);
return r; return r;
} }