mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-09 05:38:58 +00:00
Original commit message from CVS: * gst/base/gsttypefindhelper.c: (helper_find_peek): Fix evil typefind crasher: getrange() might return a short buffer at the end of a file, but gst_type_find_peek() must either return the full data as requested or NULL, but never a short buffer.
158 lines
4.2 KiB
C
158 lines
4.2 KiB
C
/* GStreamer
|
|
* Copyright (C) 1999,2000 Erik Walthinsen <omega@cse.ogi.edu>
|
|
* 2000,2005 Wim Taymans <wim@fluendo.com>
|
|
*
|
|
* gsttypefindhelper.c:
|
|
*
|
|
* 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., 59 Temple Place - Suite 330,
|
|
* Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
# include "config.h"
|
|
#endif
|
|
|
|
#include "gsttypefindhelper.h"
|
|
|
|
/**
|
|
* typefind code here
|
|
*/
|
|
typedef struct
|
|
{
|
|
GstPad *src;
|
|
guint best_probability;
|
|
GstCaps *caps;
|
|
guint64 size;
|
|
GList *buffers;
|
|
GstTypeFindFactory *factory;
|
|
}
|
|
GstTypeFindHelper;
|
|
|
|
static guint8 *
|
|
helper_find_peek (gpointer data, gint64 offset, guint size)
|
|
{
|
|
GstTypeFindHelper *find;
|
|
GstBuffer *buffer;
|
|
GstPad *src;
|
|
GstFlowReturn ret;
|
|
|
|
find = (GstTypeFindHelper *) data;
|
|
src = find->src;
|
|
|
|
GST_LOG_OBJECT (src, "'%s' called peek (%" G_GINT64_FORMAT ", %u)",
|
|
GST_PLUGIN_FEATURE_NAME (find->factory), offset, size);
|
|
|
|
if (size == 0)
|
|
return NULL;
|
|
|
|
if (offset < 0) {
|
|
if (find->size == -1 || find->size < -offset)
|
|
return NULL;
|
|
|
|
offset += find->size;
|
|
}
|
|
|
|
/* TODO: is it worth checking our list of buffers for
|
|
* a match before pulling a new buffer via _getrange()? */
|
|
buffer = NULL;
|
|
ret = GST_PAD_GETRANGEFUNC (src) (src, offset, size, &buffer);
|
|
|
|
if (ret != GST_FLOW_OK)
|
|
goto error;
|
|
|
|
/* getrange might silently return shortened buffers at the end of a file,
|
|
* we must, however, always return either the full requested data or NULL */
|
|
if (GST_BUFFER_OFFSET (buffer) != offset || GST_BUFFER_SIZE (buffer) < size) {
|
|
GST_DEBUG ("droping short buffer: %" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT
|
|
" instead of %" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT,
|
|
GST_BUFFER_OFFSET (buffer), GST_BUFFER_OFFSET (buffer) +
|
|
GST_BUFFER_SIZE (buffer), offset, offset + size);
|
|
gst_buffer_unref (buffer);
|
|
return NULL;
|
|
}
|
|
|
|
find->buffers = g_list_prepend (find->buffers, buffer);
|
|
return GST_BUFFER_DATA (buffer);
|
|
|
|
error:
|
|
{
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
static void
|
|
helper_find_suggest (gpointer data, guint probability, const GstCaps * caps)
|
|
{
|
|
GstTypeFindHelper *find = (GstTypeFindHelper *) data;
|
|
|
|
GST_LOG_OBJECT (find->src,
|
|
"'%s' called called suggest (%u, %" GST_PTR_FORMAT ")",
|
|
GST_PLUGIN_FEATURE_NAME (find->factory), probability, caps);
|
|
|
|
if (probability > find->best_probability) {
|
|
GstCaps *copy = gst_caps_copy (caps);
|
|
|
|
gst_caps_replace (&find->caps, copy);
|
|
gst_caps_unref (copy);
|
|
find->best_probability = probability;
|
|
}
|
|
}
|
|
|
|
GstCaps *
|
|
gst_type_find_helper (GstPad * src, guint64 size)
|
|
{
|
|
GstTypeFind gst_find;
|
|
GstTypeFindHelper find;
|
|
GList *walk, *type_list = NULL;
|
|
GstCaps *result = NULL;
|
|
|
|
g_return_val_if_fail (src != NULL, NULL);
|
|
g_return_val_if_fail (GST_PAD_GETRANGEFUNC (src) != NULL, NULL);
|
|
|
|
walk = type_list = gst_type_find_factory_get_list ();
|
|
|
|
find.src = src;
|
|
find.best_probability = 0;
|
|
find.caps = NULL;
|
|
find.size = size;
|
|
find.buffers = NULL;
|
|
gst_find.data = &find;
|
|
gst_find.peek = helper_find_peek;
|
|
gst_find.suggest = helper_find_suggest;
|
|
gst_find.get_length = NULL;
|
|
|
|
while (walk) {
|
|
GstTypeFindFactory *factory = GST_TYPE_FIND_FACTORY (walk->data);
|
|
|
|
find.factory = factory;
|
|
|
|
gst_type_find_factory_call_function (factory, &gst_find);
|
|
if (find.best_probability >= GST_TYPE_FIND_MAXIMUM)
|
|
break;
|
|
walk = g_list_next (walk);
|
|
}
|
|
|
|
if (find.best_probability > 0)
|
|
result = find.caps;
|
|
|
|
for (walk = find.buffers; walk; walk = walk->next)
|
|
gst_buffer_unref (GST_BUFFER (walk->data));
|
|
g_list_free (find.buffers);
|
|
|
|
return result;
|
|
}
|