static const gchar * gst_fd_src_uri_get_uri (GstURIHandler * handler) { GstFdSrc *src = GST_FD_SRC (handler); return src->uri; }
static gboolean gst_fd_src_uri_set_uri (GstURIHandler * handler, const gchar * uri) { gchar *protocol; GstFdSrc *src = GST_FD_SRC (handler); gint fd; protocol = gst_uri_get_protocol (uri); if (strcmp (protocol, "fd") != 0) { g_free (protocol); return FALSE; } g_free (protocol); if (sscanf (uri, "fd://%d", &fd) != 1 || fd < 0) return FALSE; src->new_fd = fd; GST_OBJECT_LOCK (src); if (GST_STATE (GST_ELEMENT (src)) <= GST_STATE_READY) { gst_fd_src_update_fd (src); } GST_OBJECT_UNLOCK (src); return TRUE; }
static gboolean gst_fd_src_get_size (GstBaseSrc * bsrc, guint64 * size) { GstFdSrc *src = GST_FD_SRC (bsrc); struct stat stat_results; if (!src->seekable_fd) { /* If it isn't seekable, we won't know the length (but fstat will still * succeed, and wrongly say our length is zero. */ return FALSE; } if (fstat (src->fd, &stat_results) < 0) goto could_not_stat; *size = stat_results.st_size; return TRUE; /* ERROR */ could_not_stat: { return FALSE; } }
static gboolean gst_fd_src_do_seek (GstBaseSrc * bsrc, GstSegment * segment) { gint res; gint64 offset; GstFdSrc *src = GST_FD_SRC (bsrc); offset = segment->start; /* No need to seek to the current position */ if (offset == src->curoffset) return TRUE; res = lseek (src->fd, offset, SEEK_SET); if (G_UNLIKELY (res < 0 || res != offset)) goto seek_failed; segment->last_stop = segment->start; segment->time = segment->start; return TRUE; seek_failed: GST_DEBUG_OBJECT (src, "lseek returned %" G_GINT64_FORMAT, offset); return FALSE; }
static void gst_fd_src_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec) { GstFdSrc *src = GST_FD_SRC (object); switch (prop_id) { case PROP_FD: src->new_fd = g_value_get_int (value); /* If state is ready or below, update the current fd immediately * so it is reflected in get_properties and uri */ GST_OBJECT_LOCK (object); if (GST_STATE (GST_ELEMENT (src)) <= GST_STATE_READY) { GST_DEBUG_OBJECT (src, "state ready or lower, updating to use new fd"); gst_fd_src_update_fd (src); } else { GST_DEBUG_OBJECT (src, "state above ready, not updating to new fd yet"); } GST_OBJECT_UNLOCK (object); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } }
static gboolean gst_fd_src_is_seekable (GstBaseSrc * bsrc) { GstFdSrc *src = GST_FD_SRC (bsrc); return src->seekable_fd; }
static void gst_fd_src_dispose (GObject * obj) { GstFdSrc *src = GST_FD_SRC (obj); g_free (src->uri); src->uri = NULL; G_OBJECT_CLASS (parent_class)->dispose (obj); }
static gboolean gst_fd_src_unlock_stop (GstBaseSrc * bsrc) { GstFdSrc *src = GST_FD_SRC (bsrc); GST_LOG_OBJECT (src, "No longer flushing"); GST_OBJECT_LOCK (src); gst_poll_set_flushing (src->fdset, FALSE); GST_OBJECT_UNLOCK (src); return TRUE; }
static gboolean gst_fd_src_stop (GstBaseSrc * bsrc) { GstFdSrc *src = GST_FD_SRC (bsrc); if (src->fdset) { gst_poll_free (src->fdset); src->fdset = NULL; } return TRUE; }
static void gst_fd_src_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec) { GstFdSrc *src = GST_FD_SRC (object); switch (prop_id) { case PROP_FD: g_value_set_int (value, src->fd); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } }
static gboolean gst_fd_src_uri_set_uri (GstURIHandler * handler, const gchar * uri) { gchar *protocol, *q; GstFdSrc *src = GST_FD_SRC (handler); gint fd; guint64 size = -1; GST_INFO_OBJECT (src, "checking uri %s", uri); protocol = gst_uri_get_protocol (uri); if (strcmp (protocol, "fd") != 0) { g_free (protocol); return FALSE; } g_free (protocol); if (sscanf (uri, "fd://%d", &fd) != 1 || fd < 0) return FALSE; if ((q = g_strstr_len (uri, -1, "?"))) { gchar *sp; GST_INFO_OBJECT (src, "found ?"); if ((sp = g_strstr_len (q, -1, "size="))) { if (sscanf (sp, "size=%" G_GUINT64_FORMAT, &size) != 1) { GST_INFO_OBJECT (src, "parsing size failed"); size = -1; } else { GST_INFO_OBJECT (src, "found size %" G_GUINT64_FORMAT, size); } } } src->new_fd = fd; GST_OBJECT_LOCK (src); if (GST_STATE (GST_ELEMENT (src)) <= GST_STATE_READY) { gst_fd_src_update_fd (src, size); } GST_OBJECT_UNLOCK (src); return TRUE; }
static gboolean gst_fd_src_query (GstBaseSrc * basesrc, GstQuery * query) { gboolean ret = FALSE; GstFdSrc *src = GST_FD_SRC (basesrc); switch (GST_QUERY_TYPE (query)) { case GST_QUERY_URI: gst_query_set_uri (query, src->uri); ret = TRUE; break; default: ret = FALSE; break; } if (!ret) ret = GST_BASE_SRC_CLASS (parent_class)->query (basesrc, query); return ret; }
static gboolean gst_fd_src_start (GstBaseSrc * bsrc) { GstFdSrc *src = GST_FD_SRC (bsrc); src->curoffset = 0; if ((src->fdset = gst_poll_new (TRUE)) == NULL) goto socket_pair; gst_fd_src_update_fd (src); return TRUE; /* ERRORS */ socket_pair: { GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ_WRITE, (NULL), GST_ERROR_SYSTEM); return FALSE; } }
static GstFlowReturn gst_fd_src_create (GstPushSrc * psrc, GstBuffer ** outbuf) { GstFdSrc *src; GstBuffer *buf; gssize readbytes; guint blocksize; #ifndef HAVE_WIN32 gint retval; #endif src = GST_FD_SRC (psrc); #ifndef HAVE_WIN32 do { retval = gst_poll_wait (src->fdset, GST_CLOCK_TIME_NONE); } while (retval == -1 && (errno == EINTR || errno == EAGAIN)); /* retry if interrupted */ if (retval == -1) { if (errno == EBUSY) goto stopped; else goto select_error; } #endif blocksize = GST_BASE_SRC (src)->blocksize; /* create the buffer */ buf = gst_buffer_new_and_alloc (blocksize); do { readbytes = read (src->fd, GST_BUFFER_DATA (buf), blocksize); GST_LOG_OBJECT (src, "read %" G_GSSIZE_FORMAT, readbytes); } while (readbytes == -1 && errno == EINTR); /* retry if interrupted */ if (readbytes < 0) goto read_error; if (readbytes == 0) goto eos; GST_BUFFER_OFFSET (buf) = src->curoffset; GST_BUFFER_SIZE (buf) = readbytes; GST_BUFFER_TIMESTAMP (buf) = GST_CLOCK_TIME_NONE; src->curoffset += readbytes; GST_LOG_OBJECT (psrc, "Read buffer of size %" G_GSSIZE_FORMAT, readbytes); /* we're done, return the buffer */ *outbuf = buf; return GST_FLOW_OK; /* ERRORS */ #ifndef HAVE_WIN32 select_error: { GST_ELEMENT_ERROR (src, RESOURCE, READ, (NULL), ("select on file descriptor: %s.", g_strerror (errno))); GST_DEBUG_OBJECT (psrc, "Error during select"); return GST_FLOW_ERROR; } stopped: { GST_DEBUG_OBJECT (psrc, "Select stopped"); return GST_FLOW_WRONG_STATE; } #endif eos: { GST_DEBUG_OBJECT (psrc, "Read 0 bytes. EOS."); gst_buffer_unref (buf); return GST_FLOW_UNEXPECTED; } read_error: { GST_ELEMENT_ERROR (src, RESOURCE, READ, (NULL), ("read on file descriptor: %s.", g_strerror (errno))); GST_DEBUG_OBJECT (psrc, "Error reading from fd"); gst_buffer_unref (buf); return GST_FLOW_ERROR; } }
static GstFlowReturn gst_fd_src_create (GstPushSrc * psrc, GstBuffer ** outbuf) { GstFdSrc *src; GstBuffer *buf; gssize readbytes; guint blocksize; GstClockTime timeout; #ifndef HAVE_WIN32 gboolean try_again; gint retval; #endif src = GST_FD_SRC (psrc); if (src->timeout > 0) { timeout = src->timeout * GST_USECOND; } else { timeout = GST_CLOCK_TIME_NONE; } #ifndef HAVE_WIN32 do { try_again = FALSE; GST_LOG_OBJECT (src, "doing poll, timeout %" GST_TIME_FORMAT, GST_TIME_ARGS (src->timeout)); retval = gst_poll_wait (src->fdset, timeout); GST_LOG_OBJECT (src, "poll returned %d", retval); if (G_UNLIKELY (retval == -1)) { if (errno == EINTR || errno == EAGAIN) { /* retry if interrupted */ try_again = TRUE; } else if (errno == EBUSY) { goto stopped; } else { goto poll_error; } } else if (G_UNLIKELY (retval == 0)) { try_again = TRUE; /* timeout, post element message */ gst_element_post_message (GST_ELEMENT_CAST (src), gst_message_new_element (GST_OBJECT_CAST (src), gst_structure_new ("GstFdSrcTimeout", "timeout", G_TYPE_UINT64, src->timeout, NULL))); } } while (G_UNLIKELY (try_again)); /* retry if interrupted or timeout */ #endif blocksize = GST_BASE_SRC (src)->blocksize; /* create the buffer */ buf = gst_buffer_try_new_and_alloc (blocksize); if (G_UNLIKELY (buf == NULL)) { GST_ERROR_OBJECT (src, "Failed to allocate %u bytes", blocksize); return GST_FLOW_ERROR; } do { readbytes = read (src->fd, GST_BUFFER_DATA (buf), blocksize); GST_LOG_OBJECT (src, "read %" G_GSSIZE_FORMAT, readbytes); } while (readbytes == -1 && errno == EINTR); /* retry if interrupted */ if (readbytes < 0) goto read_error; if (readbytes == 0) goto eos; GST_BUFFER_OFFSET (buf) = src->curoffset; GST_BUFFER_SIZE (buf) = readbytes; GST_BUFFER_TIMESTAMP (buf) = GST_CLOCK_TIME_NONE; src->curoffset += readbytes; GST_LOG_OBJECT (psrc, "Read buffer of size %" G_GSSIZE_FORMAT, readbytes); /* we're done, return the buffer */ *outbuf = buf; return GST_FLOW_OK; /* ERRORS */ #ifndef HAVE_WIN32 poll_error: { GST_ELEMENT_ERROR (src, RESOURCE, READ, (NULL), ("poll on file descriptor: %s.", g_strerror (errno))); GST_DEBUG_OBJECT (psrc, "Error during poll"); return GST_FLOW_ERROR; } stopped: { GST_DEBUG_OBJECT (psrc, "Poll stopped"); return GST_FLOW_WRONG_STATE; } #endif eos: { GST_DEBUG_OBJECT (psrc, "Read 0 bytes. EOS."); gst_buffer_unref (buf); return GST_FLOW_UNEXPECTED; } read_error: { GST_ELEMENT_ERROR (src, RESOURCE, READ, (NULL), ("read on file descriptor: %s.", g_strerror (errno))); GST_DEBUG_OBJECT (psrc, "Error reading from fd"); gst_buffer_unref (buf); return GST_FLOW_ERROR; } }