138 lines
3.8 KiB
C
138 lines
3.8 KiB
C
/*
|
|
* Copyright 2016 Google Inc.
|
|
* author: Edward Hervey <bilboed@bilboed.com>
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <locale.h>
|
|
|
|
#include <stdlib.h>
|
|
#include <glib.h>
|
|
#include <gst/gst.h>
|
|
#include <gst/pbutils/pbutils.h>
|
|
|
|
#ifndef LOCAL_FUZZ_BUILD
|
|
GST_PLUGIN_STATIC_DECLARE (coreelements);
|
|
GST_PLUGIN_STATIC_DECLARE (playback);
|
|
GST_PLUGIN_STATIC_DECLARE (typefindfunctions);
|
|
GST_PLUGIN_STATIC_DECLARE (app);
|
|
GST_PLUGIN_STATIC_DECLARE (ogg);
|
|
GST_PLUGIN_STATIC_DECLARE (theora);
|
|
GST_PLUGIN_STATIC_DECLARE (vorbis);
|
|
#endif
|
|
|
|
/* push-based discoverer fuzzing target
|
|
*
|
|
* This application can be compiled with libFuzzer to simulate
|
|
* a push-based discoverer execution.
|
|
*
|
|
* To reproduce the failing behaviour, use:
|
|
* $ gst-discoverer-1.0 pushfile:///...
|
|
*
|
|
* The goal is to cover basic usage of demuxers, parsers and
|
|
* base decoder elements.
|
|
*
|
|
* When compiling, only link the required demuxer/parser/decoder
|
|
* plugins and keep it to a limited range (ex: ogg/theora/vorbis)
|
|
*
|
|
**/
|
|
|
|
const guint8 *fuzztesting_data;
|
|
size_t fuzztesting_size;
|
|
|
|
static void
|
|
appsrc_configuration (GstDiscoverer * dc, GstElement * source, gpointer data)
|
|
{
|
|
GstBuffer *buf;
|
|
GstFlowReturn ret;
|
|
|
|
/* Create buffer from fuzztesting_data which shouldn't be freed */
|
|
buf =
|
|
gst_buffer_new_wrapped_full (0, (gpointer) fuzztesting_data,
|
|
fuzztesting_size, 0, fuzztesting_size, NULL, NULL);
|
|
g_object_set (G_OBJECT (source), "size", fuzztesting_size, NULL);
|
|
g_signal_emit_by_name (G_OBJECT (source), "push-buffer", buf, &ret);
|
|
gst_buffer_unref (buf);
|
|
}
|
|
|
|
static void
|
|
custom_logger (const gchar * log_domain,
|
|
GLogLevelFlags log_level, const gchar * message, gpointer unused_data)
|
|
{
|
|
if (log_level & G_LOG_LEVEL_CRITICAL) {
|
|
g_printerr ("CRITICAL ERROR : %s\n", message);
|
|
abort ();
|
|
} else if (log_level & G_LOG_LEVEL_WARNING) {
|
|
g_printerr ("WARNING : %s\n", message);
|
|
}
|
|
}
|
|
|
|
int
|
|
LLVMFuzzerTestOneInput (const guint8 * data, size_t size)
|
|
{
|
|
GError *err = NULL;
|
|
GstDiscoverer *dc;
|
|
gint timeout = 10;
|
|
GstDiscovererInfo *info;
|
|
static gboolean initialized = FALSE;
|
|
|
|
if (!initialized) {
|
|
/* We want critical warnings to assert so we can fix them */
|
|
g_log_set_always_fatal (G_LOG_LEVEL_CRITICAL);
|
|
g_log_set_default_handler (custom_logger, NULL);
|
|
|
|
/* Only initialize and register plugins once */
|
|
gst_init (NULL, NULL);
|
|
|
|
#ifndef LOCAL_FUZZ_BUILD
|
|
GST_PLUGIN_STATIC_REGISTER (coreelements);
|
|
GST_PLUGIN_STATIC_REGISTER (playback);
|
|
GST_PLUGIN_STATIC_REGISTER (typefindfunctions);
|
|
GST_PLUGIN_STATIC_REGISTER (app);
|
|
GST_PLUGIN_STATIC_REGISTER (ogg);
|
|
GST_PLUGIN_STATIC_REGISTER (theora);
|
|
GST_PLUGIN_STATIC_REGISTER (vorbis);
|
|
#endif
|
|
|
|
initialized = TRUE;
|
|
}
|
|
|
|
dc = gst_discoverer_new (timeout * GST_SECOND, &err);
|
|
if (G_UNLIKELY (dc == NULL)) {
|
|
g_print ("Error initializing: %s\n", err->message);
|
|
g_clear_error (&err);
|
|
exit (1);
|
|
}
|
|
|
|
fuzztesting_data = data;
|
|
fuzztesting_size = size;
|
|
|
|
/* Connect to source-setup signal to give the data */
|
|
g_signal_connect (dc, "source-setup", (GCallback) appsrc_configuration, NULL);
|
|
|
|
info = gst_discoverer_discover_uri (dc, "appsrc://", &err);
|
|
g_clear_error (&err);
|
|
if (info)
|
|
gst_discoverer_info_unref (info);
|
|
|
|
g_object_unref (dc);
|
|
|
|
return 0;
|
|
}
|