* Boston, MA 02111-1307, USA.
*/
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
+
#include <gst/gst.h>
+#include "gstfilesrc.h"
-#include <sys/types.h>
+#include <stdio.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
+#ifdef HAVE_MMAP
#include <sys/mman.h>
+#endif
+#include <errno.h>
+#include <string.h>
+
+#include "../gst-i18n-lib.h"
/**********************************************************************
*/
-GstElementDetails gst_filesrc_details = {
- "File Source",
- "Source/File",
- "Read from arbitrary point in a file",
- VERSION,
- "Erik Walthinsen <omega@cse.ogi.edu>",
- "(C) 1999",
-};
-
-
-#define GST_TYPE_FILESRC \
- (gst_filesrc_get_type())
-#define GST_FILESRC(obj) \
- (GTK_CHECK_CAST((obj),GST_TYPE_FILESRC,GstFileSrc))
-#define GST_FILESRC_CLASS(klass) \
- (GTK_CHECK_CLASS_CAST((klass),GST_TYPE_FILESRC,GstFileSrcClass))
-#define GST_IS_FILESRC(obj) \
- (GTK_CHECK_TYPE((obj),GST_TYPE_FILESRC))
-#define GST_IS_FILESRC_CLASS(obj) \
- (GTK_CHECK_CLASS_TYPE((klass),GST_TYPE_FILESRC))
-
-typedef enum {
- GST_FILESRC_OPEN = GST_ELEMENT_FLAG_LAST,
-
- GST_FILESRC_FLAG_LAST = GST_ELEMENT_FLAG_LAST + 2,
-} GstFileSrcFlags;
+GST_DEBUG_CATEGORY_STATIC (gst_filesrc_debug);
+#define GST_CAT_DEFAULT gst_filesrc_debug
-typedef struct _GstFileSrc GstFileSrc;
-typedef struct _GstFileSrcClass GstFileSrcClass;
-
-struct _GstFileSrc {
- GstElement element;
- GstPad *srcpad;
-
- gchar *filename; // filename
- gint fd; // open file descriptor
- off_t filelen; // what's the file length?
-
- off_t curoffset; // current offset in file
- off_t block_size; // bytes per read
-
- GstBuffer *mapbuf;
- off_t mapsize;
-
- GTree *map_regions;
- GMutex *map_regions_lock;
-};
-
-struct _GstFileSrcClass {
- GstElementClass parent_class;
-};
+GstElementDetails gst_filesrc_details = GST_ELEMENT_DETAILS ("File Source",
+ "Source/File",
+ "Read from arbitrary point in a file",
+ "Erik Walthinsen <omega@cse.ogi.edu>");
+#define DEFAULT_BLOCKSIZE 4*1024
+#define DEFAULT_MMAPSIZE 4*1024*1024
/* FileSrc signals and args */
-enum {
+enum
+{
/* FILL ME */
LAST_SIGNAL
};
-enum {
+enum
+{
ARG_0,
ARG_LOCATION,
+ ARG_FD,
ARG_BLOCKSIZE,
- ARG_OFFSET,
- ARG_SIZE,
+ ARG_MMAPSIZE,
+ ARG_TOUCH
};
+static const GstEventMask *
+gst_filesrc_get_event_mask (GstPad * pad)
+{
+ static const GstEventMask masks[] = {
+ {GST_EVENT_SEEK, GST_SEEK_METHOD_CUR |
+ GST_SEEK_METHOD_SET | GST_SEEK_METHOD_END | GST_SEEK_FLAG_FLUSH},
+ {GST_EVENT_FLUSH, 0},
+ {GST_EVENT_SIZE, 0},
+ {0, 0}
+ };
+
+ return masks;
+}
-static void gst_filesrc_class_init (GstFileSrcClass *klass);
-static void gst_filesrc_init (GstFileSrc *filesrc);
+static const GstQueryType *
+gst_filesrc_get_query_types (GstPad * pad)
+{
+ static const GstQueryType types[] = {
+ GST_QUERY_TOTAL,
+ GST_QUERY_POSITION,
+ 0
+ };
-static void gst_filesrc_set_arg (GtkObject *object, GtkArg *arg, guint id);
-static void gst_filesrc_get_arg (GtkObject *object, GtkArg *arg, guint id);
+ return types;
+}
-static GstBuffer * gst_filesrc_get (GstPad *pad);
+static const GstFormat *
+gst_filesrc_get_formats (GstPad * pad)
+{
+ static const GstFormat formats[] = {
+ GST_FORMAT_BYTES,
+ 0,
+ };
-static GstElementStateReturn gst_filesrc_change_state (GstElement *element);
+ return formats;
+}
+static void gst_filesrc_dispose (GObject * object);
-static GstElementClass *parent_class = NULL;
-//static guint gst_filesrc_signals[LAST_SIGNAL] = { 0 };
+static void gst_filesrc_set_property (GObject * object, guint prop_id,
+ const GValue * value, GParamSpec * pspec);
+static void gst_filesrc_get_property (GObject * object, guint prop_id,
+ GValue * value, GParamSpec * pspec);
-GtkType
-gst_filesrc_get_type(void)
-{
- static GtkType filesrc_type = 0;
-
- if (!filesrc_type) {
- static const GtkTypeInfo filesrc_info = {
- "GstFileSrc",
- sizeof(GstFileSrc),
- sizeof(GstFileSrcClass),
- (GtkClassInitFunc)gst_filesrc_class_init,
- (GtkObjectInitFunc)gst_filesrc_init,
- (GtkArgSetFunc)gst_filesrc_set_arg,
- (GtkArgGetFunc)gst_filesrc_get_arg,
- (GtkClassInitFunc)NULL,
- };
- filesrc_type = gtk_type_unique (GST_TYPE_ELEMENT, &filesrc_info);
- }
- return filesrc_type;
-}
+static gboolean gst_filesrc_check_filesize (GstFileSrc * src);
+static GstData *gst_filesrc_get (GstPad * pad);
+static gboolean gst_filesrc_srcpad_event (GstPad * pad, GstEvent * event);
+static gboolean gst_filesrc_srcpad_query (GstPad * pad, GstQueryType type,
+ GstFormat * format, gint64 * value);
-static void
-gst_filesrc_class_init (GstFileSrcClass *klass)
-{
- GtkObjectClass *gtkobject_class;
- GstElementClass *gstelement_class;
+static GstElementStateReturn gst_filesrc_change_state (GstElement * element);
- gtkobject_class = (GtkObjectClass*)klass;
- gstelement_class = (GstElementClass*)klass;
+static void gst_filesrc_uri_handler_init (gpointer g_iface,
+ gpointer iface_data);
- parent_class = gtk_type_class (GST_TYPE_ELEMENT);
+static void
+_do_init (GType filesrc_type)
+{
+ static const GInterfaceInfo urihandler_info = {
+ gst_filesrc_uri_handler_init,
+ NULL,
+ NULL
+ };
+
+ g_type_add_interface_static (filesrc_type, GST_TYPE_URI_HANDLER,
+ &urihandler_info);
+ GST_DEBUG_CATEGORY_INIT (gst_filesrc_debug, "filesrc", 0, "filesrc element");
+}
- gtk_object_add_arg_type ("GstFileSrc::location", GST_TYPE_FILENAME,
- GTK_ARG_READWRITE, ARG_LOCATION);
- gtk_object_add_arg_type ("GstFileSrc::blocksize", GTK_TYPE_INT,
- GTK_ARG_READWRITE, ARG_BLOCKSIZE);
- gtk_object_add_arg_type ("GstFileSrc::offset", GTK_TYPE_LONG,
- GTK_ARG_READWRITE, ARG_OFFSET);
- gtk_object_add_arg_type ("GstFileSrc::size", GTK_TYPE_LONG,
- GTK_ARG_READABLE, ARG_SIZE);
+GST_BOILERPLATE_FULL (GstFileSrc, gst_filesrc, GstElement, GST_TYPE_ELEMENT,
+ _do_init);
- gtkobject_class->set_arg = gst_filesrc_set_arg;
- gtkobject_class->get_arg = gst_filesrc_get_arg;
+static void
+gst_filesrc_base_init (gpointer g_class)
+{
+ GstElementClass *gstelement_class = GST_ELEMENT_CLASS (g_class);
- gstelement_class->change_state = gst_filesrc_change_state;
+ gst_element_class_set_details (gstelement_class, &gst_filesrc_details);
}
-
-static gint
-gst_filesrc_bufcmp (gconstpointer a, gconstpointer b)
+static void
+gst_filesrc_class_init (GstFileSrcClass * klass)
{
-// GstBuffer *bufa = (GstBuffer *)a, *bufb = (GstBuffer *)b;
-
- // sort first by offset, then in reverse by size
- if (GST_BUFFER_OFFSET(a) < GST_BUFFER_OFFSET(b)) return -1;
- else if (GST_BUFFER_OFFSET(a) > GST_BUFFER_OFFSET(b)) return 1;
- else if (GST_BUFFER_SIZE(a) > GST_BUFFER_SIZE(b)) return -1;
- else if (GST_BUFFER_SIZE(a) < GST_BUFFER_SIZE(b)) return 1;
- else return 0;
+ GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
+ GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);
+
+ gobject_class = (GObjectClass *) klass;
+
+
+ g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_FD,
+ g_param_spec_int ("fd", "File-descriptor",
+ "File-descriptor for the file being mmap()d", 0, G_MAXINT, 0,
+ G_PARAM_READABLE));
+ g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_LOCATION,
+ g_param_spec_string ("location", "File Location",
+ "Location of the file to read", NULL, G_PARAM_READWRITE));
+ g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_BLOCKSIZE,
+ g_param_spec_ulong ("blocksize", "Block size",
+ "Size in bytes to read per buffer", 1, G_MAXULONG, DEFAULT_BLOCKSIZE,
+ G_PARAM_READWRITE));
+ g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_MMAPSIZE,
+ g_param_spec_ulong ("mmapsize", "mmap() Block Size",
+ "Size in bytes of mmap()d regions", 0, G_MAXULONG, DEFAULT_MMAPSIZE,
+ G_PARAM_READWRITE));
+ g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_TOUCH,
+ g_param_spec_boolean ("touch", "Touch read data",
+ "Touch data to force disk read", FALSE, G_PARAM_READWRITE));
+
+ gobject_class->dispose = gst_filesrc_dispose;
+ gobject_class->set_property = gst_filesrc_set_property;
+ gobject_class->get_property = gst_filesrc_get_property;
+
+ gstelement_class->change_state = gst_filesrc_change_state;
}
static void
-gst_filesrc_init (GstFileSrc *src)
+gst_filesrc_init (GstFileSrc * src)
{
src->srcpad = gst_pad_new ("src", GST_PAD_SRC);
- gst_pad_set_get_function (src->srcpad,gst_filesrc_get);
+ gst_pad_set_get_function (src->srcpad, gst_filesrc_get);
+ gst_pad_set_event_function (src->srcpad, gst_filesrc_srcpad_event);
+ gst_pad_set_event_mask_function (src->srcpad, gst_filesrc_get_event_mask);
+ gst_pad_set_query_function (src->srcpad, gst_filesrc_srcpad_query);
+ gst_pad_set_query_type_function (src->srcpad, gst_filesrc_get_query_types);
+ gst_pad_set_formats_function (src->srcpad, gst_filesrc_get_formats);
gst_element_add_pad (GST_ELEMENT (src), src->srcpad);
+#ifdef HAVE_MMAP
+ src->pagesize = getpagesize ();
+#endif
+
src->filename = NULL;
src->fd = 0;
src->filelen = 0;
+ src->uri = NULL;
src->curoffset = 0;
- src->block_size = 9216;
+ src->block_size = DEFAULT_BLOCKSIZE;
+ src->touch = FALSE;
src->mapbuf = NULL;
- src->mapsize = 1 * 1024 * 1024; // default is 1MB
+ src->mapsize = DEFAULT_MMAPSIZE; /* default is 4MB */
+}
+
+static void
+gst_filesrc_dispose (GObject * object)
+{
+ GstFileSrc *src;
- src->map_regions = g_tree_new(gst_filesrc_bufcmp);
- src->map_regions_lock = g_mutex_new();
+ src = GST_FILESRC (object);
+
+ g_free (src->filename);
+ g_free (src->uri);
+
+ /* dispose may be called multiple times */
+ src->filename = NULL;
+ src->uri = NULL;
+
+ G_OBJECT_CLASS (parent_class)->dispose (object);
}
+static gboolean
+gst_filesrc_set_location (GstFileSrc * src, const gchar * location)
+{
+ /* the element must be stopped in order to do this */
+ if (GST_STATE (src) != GST_STATE_READY && GST_STATE (src) != GST_STATE_NULL)
+ return FALSE;
+
+ g_free (src->filename);
+ g_free (src->uri);
+
+ /* clear the filename if we get a NULL (is that possible?) */
+ if (location == NULL) {
+ src->filename = NULL;
+ src->uri = NULL;
+ } else {
+ src->filename = g_strdup (location);
+ src->uri = gst_uri_construct ("file", src->filename);
+ }
+ g_object_notify (G_OBJECT (src), "location");
+ gst_uri_handler_new_uri (GST_URI_HANDLER (src), src->uri);
+
+ return TRUE;
+}
static void
-gst_filesrc_set_arg (GtkObject *object, GtkArg *arg, guint id)
+gst_filesrc_set_property (GObject * object, guint prop_id, const GValue * value,
+ GParamSpec * pspec)
{
GstFileSrc *src;
src = GST_FILESRC (object);
- switch(id) {
+ switch (prop_id) {
case ARG_LOCATION:
- /* the element must be stopped in order to do this */
- g_return_if_fail (GST_STATE (src) < GST_STATE_PLAYING);
-
- if (src->filename) g_free (src->filename);
- /* clear the filename if we get a NULL (is that possible?) */
- if (GTK_VALUE_STRING (*arg) == NULL) {
- gst_element_set_state (GST_ELEMENT (object), GST_STATE_NULL);
- src->filename = NULL;
- /* otherwise set the new filename */
- } else {
- src->filename = g_strdup (GTK_VALUE_STRING (*arg));
- }
+ gst_filesrc_set_location (src, g_value_get_string (value));
break;
case ARG_BLOCKSIZE:
- src->block_size = GTK_VALUE_INT (*arg);
+ src->block_size = g_value_get_ulong (value);
+ g_object_notify (G_OBJECT (src), "blocksize");
+ break;
+ case ARG_MMAPSIZE:
+ if ((src->mapsize % src->pagesize) == 0) {
+ src->mapsize = g_value_get_ulong (value);
+ g_object_notify (G_OBJECT (src), "mmapsize");
+ } else {
+ GST_INFO_OBJECT (src,
+ "invalid mapsize, must be a multiple of pagesize, which is %d",
+ src->pagesize);
+ }
break;
- case ARG_OFFSET:
- src->curoffset = GTK_VALUE_LONG (*arg);
+ case ARG_TOUCH:
+ src->touch = g_value_get_boolean (value);
+ g_object_notify (G_OBJECT (src), "touch");
break;
default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static void
-gst_filesrc_get_arg (GtkObject *object, GtkArg *arg, guint id)
+gst_filesrc_get_property (GObject * object, guint prop_id, GValue * value,
+ GParamSpec * pspec)
{
GstFileSrc *src;
src = GST_FILESRC (object);
- switch (id) {
+ switch (prop_id) {
case ARG_LOCATION:
- GTK_VALUE_STRING (*arg) = src->filename;
+ g_value_set_string (value, src->filename);
+ break;
+ case ARG_FD:
+ g_value_set_int (value, src->fd);
break;
case ARG_BLOCKSIZE:
- GTK_VALUE_INT (*arg) = src->block_size;
+ g_value_set_ulong (value, src->block_size);
break;
- case ARG_OFFSET:
- GTK_VALUE_LONG (*arg) = src->curoffset;
+ case ARG_MMAPSIZE:
+ g_value_set_ulong (value, src->mapsize);
break;
- case ARG_SIZE:
- GTK_VALUE_LONG (*arg) = src->filelen;
+ case ARG_TOUCH:
+ g_value_set_boolean (value, src->touch);
break;
default:
- arg->type = GTK_TYPE_INVALID;
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
+#ifdef HAVE_MMAP
static void
-gst_filesrc_free_parent_mmap (GstBuffer *buf)
+gst_filesrc_free_parent_mmap (GstBuffer * buf)
{
- GstFileSrc *src = GST_FILESRC(GST_BUFFER_POOL_PRIVATE(buf));
-
- // remove the buffer from the list of available mmap'd regions
- g_mutex_lock(src->map_regions_lock);
- g_tree_remove(src->map_regions,buf);
- // check to see if the tree is empty
- if (g_tree_nnodes(src->map_regions) == 0) {
- // we have to free the bufferpool we don't have yet
- }
- g_mutex_unlock(src->map_regions_lock);
-
- // now unmap the memory
- munmap(GST_BUFFER_DATA(buf),GST_BUFFER_MAXSIZE(buf));
+ GST_LOG ("freeing mmap()d buffer at %" G_GUINT64_FORMAT "+%u",
+ GST_BUFFER_OFFSET (buf), GST_BUFFER_SIZE (buf));
+
+#ifdef MADV_DONTNEED
+ /* madvise to tell the kernel what to do with it */
+ madvise (GST_BUFFER_DATA (buf), GST_BUFFER_SIZE (buf), MADV_DONTNEED);
+#endif
+ /* now unmap the memory */
+ munmap (GST_BUFFER_DATA (buf), GST_BUFFER_MAXSIZE (buf));
+ /* cast to unsigned long, since there's no gportable way to print
+ * guint64 as hex */
+ GST_LOG ("unmapped region %08lx+%08lx at %p",
+ (unsigned long) GST_BUFFER_OFFSET (buf),
+ (unsigned long) GST_BUFFER_MAXSIZE (buf), GST_BUFFER_DATA (buf));
+
+ GST_BUFFER_DATA (buf) = NULL;
}
+#endif
+#ifdef HAVE_MMAP
static GstBuffer *
-gst_filesrc_map_region (GstFileSrc *src, off_t offset, off_t size)
+gst_filesrc_map_region (GstFileSrc * src, off_t offset, size_t size)
{
GstBuffer *buf;
-
- GST_DEBUG(0, "mapping region %d+%d from file into memory\n",offset,size);
-
- // time to allocate a new mapbuf
- buf = gst_buffer_new();
- // mmap() the data into this new buffer
- GST_BUFFER_DATA(buf) = mmap (NULL, size, PROT_READ, MAP_SHARED, src->fd, offset);
- if (GST_BUFFER_DATA(buf) == NULL) {
- fprintf(stderr, "ERROR: gstfilesrc couldn't map file!\n");
- } else if (GST_BUFFER_DATA(buf) == -1) {
- perror("gstfilesrc:mmap()");
+ gint retval;
+ void *mmapregion;
+
+ g_return_val_if_fail (offset >= 0, NULL);
+
+ GST_LOG_OBJECT (src, "mapping region %08llx+%08lx from file into memory",
+ offset, (unsigned long) size);
+ mmapregion = mmap (NULL, size, PROT_READ, MAP_SHARED, src->fd, offset);
+
+ if (mmapregion == NULL) {
+ GST_ELEMENT_ERROR (src, RESOURCE, TOO_LAZY, (NULL), ("mmap call failed."));
+ return NULL;
+ } else if (mmapregion == MAP_FAILED) {
+ GST_WARNING_OBJECT (src, "mmap (0x%08lx, %d, 0x%llx) failed: %s",
+ (unsigned long) size, src->fd, offset, strerror (errno));
+ return NULL;
}
- // fill in the rest of the fields
- GST_BUFFER_FLAGS(buf) = GST_BUFFER_READONLY | GST_BUFFER_ORIGINAL;
- GST_BUFFER_SIZE(buf) = size;
- GST_BUFFER_MAXSIZE(buf) = size;
- GST_BUFFER_OFFSET(buf) = offset;
- GST_BUFFER_TIMESTAMP(buf) = -1LL;
- GST_BUFFER_POOL_PRIVATE(buf) = src;
- GST_BUFFER_FREE_FUNC(buf) = gst_filesrc_free_parent_mmap;
-
- g_mutex_lock(src->map_regions_lock);
- g_tree_insert(src->map_regions,buf,buf);
- g_mutex_unlock(src->map_regions_lock);
+ GST_LOG_OBJECT (src, "mapped region %08lx+%08lx from file into memory at %p",
+ (unsigned long) offset, (unsigned long) size, mmapregion);
+
+ /* time to allocate a new mapbuf */
+ buf = gst_buffer_new ();
+ /* mmap() the data into this new buffer */
+ GST_BUFFER_DATA (buf) = mmapregion;
+
+#ifdef MADV_SEQUENTIAL
+ /* madvise to tell the kernel what to do with it */
+ retval =
+ madvise (GST_BUFFER_DATA (buf), GST_BUFFER_SIZE (buf), MADV_SEQUENTIAL);
+#endif
+ /* fill in the rest of the fields */
+ GST_BUFFER_FLAG_SET (buf, GST_BUFFER_READONLY);
+ GST_BUFFER_FLAG_SET (buf, GST_BUFFER_ORIGINAL);
+ GST_BUFFER_SIZE (buf) = size;
+ GST_BUFFER_MAXSIZE (buf) = size;
+ GST_BUFFER_OFFSET (buf) = offset;
+ GST_BUFFER_OFFSET_END (buf) = offset + size;
+ GST_BUFFER_TIMESTAMP (buf) = GST_CLOCK_TIME_NONE;
+ GST_BUFFER_PRIVATE (buf) = src;
+ GST_BUFFER_FREE_DATA_FUNC (buf) = gst_filesrc_free_parent_mmap;
return buf;
}
+#endif
+#ifdef HAVE_MMAP
static GstBuffer *
-gst_filesrc_map_small_region (GstFileSrc *src, off_t offset, off_t size)
+gst_filesrc_map_small_region (GstFileSrc * src, off_t offset, size_t size)
{
- gint pagesize = getpagesize();
- int mod, mapbase, mapsize;
+ size_t mapsize;
+ off_t mod, mapbase;
GstBuffer *map;
-// printf("attempting to map a small buffer at %d+%d\n",offset,size);
+/* printf("attempting to map a small buffer at %d+%d\n",offset,size); */
- // if the offset starts at a non-page boundary, we have to special case
- if ((mod = offset % pagesize)) {
- mapbase = offset - mod;
- mapsize = (((mapbase + size + mod) + (pagesize - 1)) / pagesize) * pagesize - mapbase;
-// printf("not on page boundaries, resizing map to %d+%d\n",mapbase,mapsize);
- map = gst_filesrc_map_region(src, mapbase, mapsize);
- return gst_buffer_create_sub (map, offset - mapbase, size);
- }
+ /* if the offset starts at a non-page boundary, we have to special case */
+ if ((mod = offset % src->pagesize)) {
+ GstBuffer *ret;
- return gst_filesrc_map_region(src,offset,size);
-}
+ mapbase = offset - mod;
+ mapsize =
+ ((size + mod + src->pagesize - 1) / src->pagesize) * src->pagesize;
+/* printf("not on page boundaries, resizing map to %d+%d\n",mapbase,mapsize);*/
+ map = gst_filesrc_map_region (src, mapbase, mapsize);
+ if (map == NULL)
+ return NULL;
-typedef struct {
- off_t offset;
- off_t size;
-} GstFileSrcRegion;
+ ret = gst_buffer_create_sub (map, offset - mapbase, size);
+ GST_BUFFER_OFFSET (ret) = GST_BUFFER_OFFSET (map) + offset - mapbase;
-// This allows us to search for a potential mmap region.
-static gint
-gst_filesrc_search_region_match (gpointer a, gpointer b)
-{
- GstFileSrcRegion *r = (GstFileSrcRegion *)b;
+ gst_buffer_unref (map);
- // trying to walk b down the tree, current node is a
- if (r->offset < GST_BUFFER_OFFSET(a)) return -1;
- else if (r->offset >= (GST_BUFFER_OFFSET(a) + GST_BUFFER_SIZE(a))) return 1;
- else if ((r->offset + r->size) <= (GST_BUFFER_OFFSET(a) + GST_BUFFER_SIZE(a))) return 0;
+ return ret;
+ }
- return -2;
+ return gst_filesrc_map_region (src, offset, size);
}
+#endif
+#ifdef HAVE_MMAP
/**
- * gst_filesrc_get:
+ * gst_filesrc_get_mmap:
* @pad: #GstPad to push a buffer from
*
* Push a new buffer from the filesrc at the current offset.
*/
static GstBuffer *
-gst_filesrc_get (GstPad *pad)
+gst_filesrc_get_mmap (GstFileSrc * src)
{
- GstFileSrc *src;
- GstBuffer *buf = NULL, *map;
- off_t readend,mapstart,mapend;
- GstFileSrcRegion region;
-
- g_return_val_if_fail (pad != NULL, NULL);
- src = GST_FILESRC (gst_pad_get_parent (pad));
- g_return_val_if_fail (GST_FLAG_IS_SET (src, GST_FILESRC_OPEN), NULL);
+ GstBuffer *buf = NULL;
+ size_t readsize, mapsize;
+ off_t readend, mapstart, mapend;
+ int i;
+
+ /* calculate end pointers so we don't have to do so repeatedly later */
+ readsize = src->block_size;
+ readend = src->curoffset + src->block_size; /* note this is the byte *after* the read */
+ mapstart = GST_BUFFER_OFFSET (src->mapbuf);
+ mapsize = GST_BUFFER_SIZE (src->mapbuf);
+ mapend = mapstart + mapsize; /* note this is the byte *after* the map */
+
+ /* check to see if we're going to overflow the end of the file */
+ if (readend > src->filelen) {
+ if (!gst_filesrc_check_filesize (src) || readend > src->filelen) {
+ readsize = src->filelen - src->curoffset;
+ readend = src->curoffset + readsize;
+ }
+ }
- // calculate end poiters so we don't have to do so repeatedly later
- readend = src->curoffset + src->block_size; // note this is the byte *after* the read
- mapstart = GST_BUFFER_OFFSET(src->mapbuf);
- mapend = mapstart + GST_BUFFER_SIZE(src->mapbuf); // note this is the byte *after* the map
+ GST_LOG ("attempting to read %08lx, %08lx, %08lx, %08lx",
+ (unsigned long) readsize, (unsigned long) readend,
+ (unsigned long) mapstart, (unsigned long) mapend);
- // if the start is past the mapstart
+ /* if the start is past the mapstart */
if (src->curoffset >= mapstart) {
- // if the end is before the mapend, the buffer is in current mmap region...
- // ('cause by definition if readend is in the buffer, so's readstart)
+ /* if the end is before the mapend, the buffer is in current mmap region... */
+ /* ('cause by definition if readend is in the buffer, so's readstart) */
if (readend <= mapend) {
-// printf("read buf %d+%d lives in current mapbuf %d+%d, creating subbuffer of mapbuf\n",
-// src->curoffset,src->block_size,GST_BUFFER_OFFSET(src->mapbuf),GST_BUFFER_SIZE(src->mapbuf));
- buf = gst_buffer_create_sub (src->mapbuf, src->curoffset - GST_BUFFER_OFFSET(src->mapbuf),
- src->block_size);
-
- // if the start actually is within the current mmap region, map an overlap buffer
+ GST_LOG_OBJECT (src,
+ "read buf %llu+%d lives in current mapbuf %lld+%d, creating subbuffer of mapbuf",
+ src->curoffset, (int) readsize, mapstart, mapsize);
+ buf =
+ gst_buffer_create_sub (src->mapbuf, src->curoffset - mapstart,
+ readsize);
+ GST_BUFFER_OFFSET (buf) = src->curoffset;
+
+ /* if the start actually is within the current mmap region, map an overlap buffer */
} else if (src->curoffset < mapend) {
-// printf("read buf %d+%d starts in mapbuf %d+%d but ends outside, creating new mmap\n",
-// src->curoffset,src->block_size,GST_BUFFER_OFFSET(src->mapbuf),GST_BUFFER_SIZE(src->mapbuf));
- buf = gst_filesrc_map_small_region (src, src->curoffset, src->block_size);
+ GST_LOG_OBJECT (src,
+ "read buf %llu+%d starts in mapbuf %d+%d but ends outside, creating new mmap",
+ (unsigned long long) src->curoffset, (gint) readsize, (gint) mapstart,
+ (gint) mapsize);
+ buf = gst_filesrc_map_small_region (src, src->curoffset, readsize);
+ if (buf == NULL)
+ return NULL;
}
- // the only other option is that buffer is totally outside, which means we search for it
+ /* the only other option is that buffer is totally outside, which means we search for it */
- // now we can assume that the start is *before* the current mmap region
- // if the readend is past mapstart, we have two options
+ /* now we can assume that the start is *before* the current mmap region */
+ /* if the readend is past mapstart, we have two options */
} else if (readend >= mapstart) {
- // either the read buffer overlaps the start of the mmap region
- // or the read buffer fully contains the current mmap region
- // either way, it's really not relevant, we just create a new region anyway
-// printf("read buf %d+%d starts before mapbuf %d+%d, but overlaps it\n",
-// src->curoffset,src->block_size,GST_BUFFER_OFFSET(src->mapbuf),GST_BUFFER_SIZE(src->mapbuf));
- buf = gst_filesrc_map_small_region (src, src->curoffset, src->block_size);
+ /* either the read buffer overlaps the start of the mmap region */
+ /* or the read buffer fully contains the current mmap region */
+ /* either way, it's really not relevant, we just create a new region anyway */
+ GST_LOG_OBJECT (src,
+ "read buf %llu+%d starts before mapbuf %d+%d, but overlaps it",
+ (unsigned long long) src->curoffset, (gint) readsize, (gint) mapstart,
+ (gint) mapsize);
+ buf = gst_filesrc_map_small_region (src, src->curoffset, readsize);
+ if (buf == NULL)
+ return NULL;
}
- // then deal with the case where the read buffer is totally outside
+ /* then deal with the case where the read buffer is totally outside */
if (buf == NULL) {
- // first check to see if there's a map that covers the right region already
-// printf("searching for mapbuf to cover %d+%d\n",src->curoffset,src->block_size);
- region.offset = src->curoffset;
- region.size = src->block_size;
- map = g_tree_search(src->map_regions,gst_filesrc_search_region_match,®ion);
-
- // if we found an exact match, subbuffer it
- if (map != NULL) {
-// printf("found mapbuf at %d+%d, creating subbuffer\n",GST_BUFFER_OFFSET(map),GST_BUFFER_SIZE(map));
- buf = gst_buffer_create_sub (map, src->curoffset - GST_BUFFER_OFFSET(map), src->block_size);
-
- // otherwise we need to create something out of thin air
+ /* first check to see if there's a map that covers the right region already */
+ GST_LOG_OBJECT (src, "searching for mapbuf to cover %llu+%d",
+ src->curoffset, (int) readsize);
+
+ /* if the read buffer crosses a mmap region boundary, create a one-off region */
+ if ((src->curoffset / src->mapsize) != (readend / src->mapsize)) {
+ GST_LOG_OBJECT (src,
+ "read buf %llu+%d crosses a %d-byte boundary, creating a one-off",
+ src->curoffset, (int) readsize, (int) src->mapsize);
+ buf = gst_filesrc_map_small_region (src, src->curoffset, readsize);
+ if (buf == NULL)
+ return NULL;
+
+ /* otherwise we will create a new mmap region and set it to the default */
} else {
- // if the read buffer crosses a mmap region boundary, create a one-off region
- if ((src->curoffset / src->mapsize) != ((src->curoffset + src->block_size) / src->mapsize)) {
-// printf("read buf %d+%d crosses a %d-byte boundary, creating a one-off\n",
-// src->curoffset,src->block_size,src->mapsize);
- buf = gst_filesrc_map_small_region (src, src->curoffset, src->block_size);
-
- // otherwise we will create a new mmap region and set it to the default
- } else {
- off_t nextmap = src->curoffset - (src->curoffset % src->mapsize);
-// printf("read buf %d+%d in new mapbuf at %d+%d, mapping and subbuffering\n",
-// src->curoffset,src->block_size,nextmap,src->mapsize);
- // first, we're done with the old mapbuf
- gst_buffer_unref(src->mapbuf);
- // create a new one
- src->mapbuf = gst_filesrc_map_region (src, nextmap, src->mapsize);
- // subbuffer it
- buf = gst_buffer_create_sub (src->mapbuf, src->curoffset - GST_BUFFER_OFFSET(src->mapbuf), src->block_size);
+ size_t mapsize;
+
+ off_t nextmap = src->curoffset - (src->curoffset % src->mapsize);
+
+ GST_LOG_OBJECT (src,
+ "read buf %llu+%d in new mapbuf at %llu+%d, mapping and subbuffering",
+ src->curoffset, readsize, nextmap, src->mapsize);
+ /* first, we're done with the old mapbuf */
+ gst_buffer_unref (src->mapbuf);
+ mapsize = src->mapsize;
+
+ /* double the mapsize as long as the readsize is smaller */
+ while (readsize - (src->curoffset - nextmap) > mapsize) {
+ GST_LOG_OBJECT (src, "readsize smaller then mapsize %08x %d",
+ readsize, (int) mapsize);
+ mapsize <<= 1;
}
+ /* create a new one */
+ src->mapbuf = gst_filesrc_map_region (src, nextmap, mapsize);
+ if (src->mapbuf == NULL)
+ return NULL;
+
+ /* subbuffer it */
+ buf =
+ gst_buffer_create_sub (src->mapbuf, src->curoffset - nextmap,
+ readsize);
+ GST_BUFFER_OFFSET (buf) =
+ GST_BUFFER_OFFSET (src->mapbuf) + src->curoffset - nextmap;
}
}
+ /* if we need to touch the buffer (to bring it into memory), do so */
+ if (src->touch) {
+ volatile guchar *p = GST_BUFFER_DATA (buf), c;
+
+ for (i = 0; i < GST_BUFFER_SIZE (buf); i += src->pagesize)
+ c = p[i];
+ }
+
/* we're done, return the buffer */
- src->curoffset += GST_BUFFER_SIZE(buf);
+ g_assert (src->curoffset == GST_BUFFER_OFFSET (buf));
+ src->curoffset += GST_BUFFER_SIZE (buf);
+ return buf;
+}
+#endif
+
+static GstBuffer *
+gst_filesrc_get_read (GstFileSrc * src)
+{
+ GstBuffer *buf = NULL;
+ size_t readsize;
+ int ret;
+
+ readsize = src->block_size;
+ if (src->curoffset + readsize > src->filelen) {
+ if (!gst_filesrc_check_filesize (src)
+ || src->curoffset + readsize > src->filelen) {
+ readsize = src->filelen - src->curoffset;
+ }
+ }
+
+ buf = gst_buffer_new_and_alloc (readsize);
+ g_return_val_if_fail (buf != NULL, NULL);
+
+ ret = read (src->fd, GST_BUFFER_DATA (buf), readsize);
+ if (ret < 0) {
+ GST_ELEMENT_ERROR (src, RESOURCE, READ, (NULL), GST_ERROR_SYSTEM);
+ return NULL;
+ }
+ if (ret < readsize) {
+ GST_ELEMENT_ERROR (src, RESOURCE, READ, (NULL),
+ ("unexpected end of file."));
+ return NULL;
+ }
+
+ GST_BUFFER_SIZE (buf) = readsize;
+ GST_BUFFER_MAXSIZE (buf) = readsize;
+ GST_BUFFER_OFFSET (buf) = src->curoffset;
+ GST_BUFFER_OFFSET_END (buf) = src->curoffset + readsize;
+ src->curoffset += readsize;
+
return buf;
}
+static GstData *
+gst_filesrc_get (GstPad * pad)
+{
+ GstFileSrc *src;
+
+ g_return_val_if_fail (pad != NULL, NULL);
+ src = GST_FILESRC (gst_pad_get_parent (pad));
+ g_return_val_if_fail (GST_FLAG_IS_SET (src, GST_FILESRC_OPEN), NULL);
+
+ /* check for flush */
+ if (src->need_flush) {
+ src->need_flush = FALSE;
+ GST_DEBUG_OBJECT (src, "sending flush");
+ return GST_DATA (gst_event_new_flush ());
+ }
+ /* check for seek */
+ if (src->need_discont) {
+ GstEvent *event;
+
+ src->need_discont = FALSE;
+ GST_DEBUG_OBJECT (src, "sending discont");
+ event =
+ gst_event_new_discontinuous (FALSE, GST_FORMAT_BYTES, src->curoffset,
+ NULL);
+ return GST_DATA (event);
+ }
+
+ /* check for EOF */
+ g_assert (src->curoffset <= src->filelen);
+ if (src->curoffset == src->filelen) {
+ if (!gst_filesrc_check_filesize (src) || src->curoffset >= src->filelen) {
+ GST_DEBUG_OBJECT (src, "eos %" G_GINT64_FORMAT " %" G_GINT64_FORMAT,
+ src->curoffset, src->filelen);
+ gst_element_set_eos (GST_ELEMENT (src));
+ return GST_DATA (gst_event_new (GST_EVENT_EOS));
+ }
+ }
+#ifdef HAVE_MMAP
+ if (src->using_mmap) {
+ return GST_DATA (gst_filesrc_get_mmap (src));
+ } else {
+ return GST_DATA (gst_filesrc_get_read (src));
+ }
+#else
+ return GST_DATA (gst_filesrc_get_read (src));
+#endif
+}
+
+/* TRUE if the filesize of the file was updated */
+static gboolean
+gst_filesrc_check_filesize (GstFileSrc * src)
+{
+ struct stat stat_results;
+
+ g_return_val_if_fail (GST_FLAG_IS_SET (src, GST_FILESRC_OPEN), FALSE);
+
+ fstat (src->fd, &stat_results);
+ GST_DEBUG_OBJECT (src,
+ "checked filesize on %s (was %" G_GUINT64_FORMAT ", is %" G_GUINT64_FORMAT
+ ")", src->filename, src->filelen, (guint64) stat_results.st_size);
+ if (src->filelen == (guint64) stat_results.st_size)
+ return FALSE;
+ src->filelen = stat_results.st_size;
+ return TRUE;
+}
+
/* open the file and mmap it, necessary to go to READY state */
-static gboolean
-gst_filesrc_open_file (GstFileSrc *src)
+static gboolean
+gst_filesrc_open_file (GstFileSrc * src)
{
- g_return_val_if_fail (!GST_FLAG_IS_SET (src ,GST_FILESRC_OPEN), FALSE);
+ g_return_val_if_fail (!GST_FLAG_IS_SET (src, GST_FILESRC_OPEN), FALSE);
- GST_DEBUG(0, "opening file %s\n",src->filename);
+ if (src->filename == NULL || src->filename[0] == '\0') {
+ GST_ELEMENT_ERROR (src, RESOURCE, NOT_FOUND,
+ (_("No file name specified for reading.")), (NULL));
+ return FALSE;
+ }
+
+
+ GST_INFO_OBJECT (src, "opening file %s", src->filename);
/* open the file */
src->fd = open (src->filename, O_RDONLY);
if (src->fd < 0) {
- perror ("open");
- gst_element_error (GST_ELEMENT (src), g_strconcat("opening file \"", src->filename, "\"", NULL));
+ if (errno == ENOENT)
+ GST_ELEMENT_ERROR (src, RESOURCE, NOT_FOUND, (NULL), (NULL));
+ else
+ GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
+ (_("Could not open file \"%s\" for reading."), src->filename),
+ GST_ERROR_SYSTEM);
return FALSE;
} else {
+ /* check if it is a regular file, otherwise bail out */
+ struct stat stat_results;
+
+ fstat (src->fd, &stat_results);
+
+ if (!S_ISREG (stat_results.st_mode)) {
+ GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
+ (_("File \"%s\" isn't a regular file."), src->filename), (NULL));
+ close (src->fd);
+ return FALSE;
+ }
+
/* find the file length */
- src->filelen = lseek (src->fd, 0, SEEK_END);
- lseek (src->fd, 0, SEEK_SET);
+ src->filelen = stat_results.st_size;
- // allocate the first mmap'd region
+ src->using_mmap = FALSE;
+#ifdef HAVE_MMAP
+ /* allocate the first mmap'd region */
src->mapbuf = gst_filesrc_map_region (src, 0, src->mapsize);
+ if (src->mapbuf != NULL) {
+ src->using_mmap = TRUE;
+ }
+#endif
+
src->curoffset = 0;
/* unmap and close the file */
static void
-gst_filesrc_close_file (GstFileSrc *src)
+gst_filesrc_close_file (GstFileSrc * src)
{
g_return_if_fail (GST_FLAG_IS_SET (src, GST_FILESRC_OPEN));
src->filelen = 0;
src->curoffset = 0;
+ if (src->mapbuf) {
+ gst_buffer_unref (src->mapbuf);
+ src->mapbuf = NULL;
+ }
+
GST_FLAG_UNSET (src, GST_FILESRC_OPEN);
}
static GstElementStateReturn
-gst_filesrc_change_state (GstElement *element)
+gst_filesrc_change_state (GstElement * element)
{
- g_return_val_if_fail (GST_IS_FILESRC (element), GST_STATE_FAILURE);
+ GstFileSrc *src = GST_FILESRC (element);
- if (GST_STATE_PENDING (element) == GST_STATE_NULL) {
- if (GST_FLAG_IS_SET (element, GST_FILESRC_OPEN))
- gst_filesrc_close_file (GST_FILESRC (element));
- } else {
- if (!GST_FLAG_IS_SET (element, GST_FILESRC_OPEN)) {
- if (!gst_filesrc_open_file (GST_FILESRC (element)))
- return GST_STATE_FAILURE;
- }
+ switch (GST_STATE_TRANSITION (element)) {
+ case GST_STATE_NULL_TO_READY:
+ break;
+ case GST_STATE_READY_TO_NULL:
+ break;
+ case GST_STATE_READY_TO_PAUSED:
+ if (!GST_FLAG_IS_SET (element, GST_FILESRC_OPEN)) {
+ if (!gst_filesrc_open_file (GST_FILESRC (element)))
+ return GST_STATE_FAILURE;
+ }
+ src->need_discont = TRUE;
+ break;
+ case GST_STATE_PAUSED_TO_READY:
+ if (GST_FLAG_IS_SET (element, GST_FILESRC_OPEN))
+ gst_filesrc_close_file (GST_FILESRC (element));
+ src->need_discont = TRUE;
+ break;
+ default:
+ break;
}
if (GST_ELEMENT_CLASS (parent_class)->change_state)
return GST_STATE_SUCCESS;
}
+
+static gboolean
+gst_filesrc_srcpad_query (GstPad * pad, GstQueryType type,
+ GstFormat * format, gint64 * value)
+{
+ GstFileSrc *src = GST_FILESRC (GST_PAD_PARENT (pad));
+
+ switch (type) {
+ case GST_QUERY_TOTAL:
+ if (*format != GST_FORMAT_BYTES) {
+ return FALSE;
+ }
+ gst_filesrc_check_filesize (src);
+ *value = src->filelen;
+ break;
+ case GST_QUERY_POSITION:
+ switch (*format) {
+ case GST_FORMAT_BYTES:
+ *value = src->curoffset;
+ break;
+ case GST_FORMAT_PERCENT:
+ if (src->filelen == 0)
+ return FALSE;
+ *value = src->curoffset * GST_FORMAT_PERCENT_MAX / src->filelen;
+ break;
+ default:
+ return FALSE;
+ }
+ break;
+ default:
+ return FALSE;
+ break;
+ }
+ return TRUE;
+}
+
+static gboolean
+gst_filesrc_srcpad_event (GstPad * pad, GstEvent * event)
+{
+ GstFileSrc *src = GST_FILESRC (GST_PAD_PARENT (pad));
+
+ GST_DEBUG_OBJECT (src, "event %d", GST_EVENT_TYPE (event));
+
+ switch (GST_EVENT_TYPE (event)) {
+ case GST_EVENT_SEEK:
+ {
+ gint64 offset;
+
+ if (GST_EVENT_SEEK_FORMAT (event) != GST_FORMAT_BYTES) {
+ goto error;
+ }
+
+ offset = GST_EVENT_SEEK_OFFSET (event);
+
+ switch (GST_EVENT_SEEK_METHOD (event)) {
+ case GST_SEEK_METHOD_SET:
+ if (offset < 0 ||
+ (offset > src->filelen && (!gst_filesrc_check_filesize (src)
+ || offset > src->filelen))) {
+ goto error;
+ }
+ src->curoffset = offset;
+ GST_DEBUG_OBJECT (src, "seek set pending to %" G_GINT64_FORMAT,
+ src->curoffset);
+ break;
+ case GST_SEEK_METHOD_CUR:
+ if (offset + src->curoffset > src->filelen)
+ if (!gst_filesrc_check_filesize (src)
+ || offset + src->curoffset > src->filelen)
+ goto error;
+ src->curoffset += offset;
+ GST_DEBUG_OBJECT (src, "seek cur pending to %" G_GINT64_FORMAT,
+ src->curoffset);
+ break;
+ case GST_SEEK_METHOD_END:
+ if (ABS (offset) > src->filelen) {
+ if (!gst_filesrc_check_filesize (src)
+ || ABS (offset) > src->filelen)
+ goto error;
+ goto error;
+ }
+ src->curoffset = src->filelen - ABS (offset);
+ GST_DEBUG_OBJECT (src, "seek end pending to %" G_GINT64_FORMAT,
+ src->curoffset);
+ break;
+ default:
+ goto error;
+ break;
+ }
+ src->need_discont = TRUE;
+ src->need_flush = GST_EVENT_SEEK_FLAGS (event) & GST_SEEK_FLAG_FLUSH;
+ break;
+ }
+ case GST_EVENT_SIZE:
+ if (GST_EVENT_SIZE_FORMAT (event) != GST_FORMAT_BYTES) {
+ goto error;
+ }
+ src->block_size = GST_EVENT_SIZE_VALUE (event);
+ g_object_notify (G_OBJECT (src), "blocksize");
+ break;
+ case GST_EVENT_FLUSH:
+ src->need_flush = TRUE;
+ break;
+ default:
+ goto error;
+ break;
+ }
+ gst_event_unref (event);
+ return TRUE;
+
+error:
+ gst_event_unref (event);
+ return FALSE;
+}
+
+/*** GSTURIHANDLER INTERFACE *************************************************/
+
+static guint
+gst_filesrc_uri_get_type (void)
+{
+ return GST_URI_SRC;
+}
+static gchar **
+gst_filesrc_uri_get_protocols (void)
+{
+ static gchar *protocols[] = { "file", NULL };
+
+ return protocols;
+}
+static const gchar *
+gst_filesrc_uri_get_uri (GstURIHandler * handler)
+{
+ GstFileSrc *src = GST_FILESRC (handler);
+
+ return src->uri;
+}
+
+static gboolean
+gst_filesrc_uri_set_uri (GstURIHandler * handler, const gchar * uri)
+{
+ gchar *protocol, *location;
+ gboolean ret;
+ GstFileSrc *src = GST_FILESRC (handler);
+
+ protocol = gst_uri_get_protocol (uri);
+ if (strcmp (protocol, "file") != 0) {
+ g_free (protocol);
+ return FALSE;
+ }
+ g_free (protocol);
+ location = gst_uri_get_location (uri);
+ ret = gst_filesrc_set_location (src, location);
+ g_free (location);
+
+ return ret;
+}
+
+static void
+gst_filesrc_uri_handler_init (gpointer g_iface, gpointer iface_data)
+{
+ GstURIHandlerInterface *iface = (GstURIHandlerInterface *) g_iface;
+
+ iface->get_type = gst_filesrc_uri_get_type;
+ iface->get_protocols = gst_filesrc_uri_get_protocols;
+ iface->get_uri = gst_filesrc_uri_get_uri;
+ iface->set_uri = gst_filesrc_uri_set_uri;
+}
projects / platform / upstream / gstreamer.git / blobdiff