gstreamer/ext/mythtv/gstmythtvsrc.c

1301 lines
35 KiB
C
Raw Normal View History

/* GStreamer MythTV Plug-in
* Copyright (C) <2006> Rosfran Borges <rosfran.borges@indt.org.br>
*
* 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.
*/
/**
* When using the LiveTV content, put the location URI in the following
* format:
*
* myth://mythtv:mythtv@xxx.xxx.xxx.xxx:6543/?mythconverg
*
* Where the first field is the protocol (myth), the second and third are user
* name (mythtv) and password (mythtv), then backend host name and port number,
* and the last field is the database name (mythconverg).
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "gstmythtvsrc.h"
#include <gmyth/gmyth_file_transfer.h>
#include <gmyth/gmyth_livetv.h>
#include <gmyth/gmyth_socket.h>
#include <gmyth/gmyth_tvchain.h>
#include <string.h>
#include <unistd.h>
GST_DEBUG_CATEGORY_STATIC (mythtvsrc_debug);
#define GST_CAT_DEFAULT mythtvsrc_debug
#define GST_GMYTHTV_ID_NUM 1
#define GST_GMYTHTV_CHANNEL_DEFAULT_NUM (-1)
#define GMYTHTV_VERSION_DEFAULT 30
#define GMYTHTV_TRANSFER_MAX_WAITS 100
#define GMYTHTV_TRANSFER_MAX_RESENDS 2
#define GMYTHTV_TRANSFER_MAX_BUFFER (128*1024)
#define MAX_READ_SIZE (4*1024)
#define GST_FLOW_ERROR_NO_DATA (-101)
#define REQUEST_MAX_SIZE (64*1024)
#define INTERNAL_BUFFER_SIZE (90*1024)
static const GstElementDetails gst_mythtv_src_details =
GST_ELEMENT_DETAILS ("MythTV client source",
"Source/Network",
"Control and receive data as a client over the network "
"via raw socket connections using the MythTV protocol",
"Rosfran Borges <rosfran.borges@indt.org.br>");
static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS ("video/x-nuv"));
enum
{
PROP_0,
PROP_LOCATION,
#ifndef GST_DISABLE_GST_DEBUG
PROP_GMYTHTV_DBG,
#endif
PROP_GMYTHTV_VERSION,
PROP_GMYTHTV_LIVE,
PROP_GMYTHTV_LIVEID,
PROP_GMYTHTV_LIVE_CHAINID,
PROP_GMYTHTV_ENABLE_TIMING_POSITION,
PROP_GMYTHTV_CHANNEL_NUM
};
static void gst_mythtv_src_finalize (GObject * gobject);
static GstFlowReturn gst_mythtv_src_create (GstPushSrc * psrc,
GstBuffer ** outbuf);
static gboolean gst_mythtv_src_start (GstBaseSrc * bsrc);
static gboolean gst_mythtv_src_stop (GstBaseSrc * bsrc);
static gboolean gst_mythtv_src_get_size (GstBaseSrc * bsrc, guint64 * size);
static gboolean gst_mythtv_src_is_seekable (GstBaseSrc * push_src);
static gboolean gst_mythtv_src_do_seek (GstBaseSrc * base,
GstSegment * segment);
static gboolean gst_mythtv_src_next_program_chain (GstMythtvSrc * src);
static GstStateChangeReturn
gst_mythtv_src_change_state (GstElement * element, GstStateChange transition);
static void gst_mythtv_src_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec);
static void gst_mythtv_src_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec);
static void gst_mythtv_src_uri_handler_init (gpointer g_iface,
gpointer iface_data);
static gboolean gst_mythtv_src_handle_query (GstPad * pad, GstQuery * query);
static gboolean gst_mythtv_src_handle_event (GstPad * pad, GstEvent * event);
static gint do_read_request_response (GstMythtvSrc * src, guint size,
GByteArray * data_ptr);
static void
_urihandler_init (GType type)
{
static const GInterfaceInfo urihandler_info = {
gst_mythtv_src_uri_handler_init,
NULL,
NULL
};
g_type_add_interface_static (type, GST_TYPE_URI_HANDLER, &urihandler_info);
GST_DEBUG_CATEGORY_INIT (mythtvsrc_debug, "mythtvsrc", 0, "MythTV src");
}
GST_BOILERPLATE_FULL (GstMythtvSrc, gst_mythtv_src, GstPushSrc,
GST_TYPE_PUSH_SRC, _urihandler_init)
static void gst_mythtv_src_base_init (gpointer g_class)
{
GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&srctemplate));
gst_element_class_set_details (element_class, &gst_mythtv_src_details);
element_class->change_state = gst_mythtv_src_change_state;
}
static void
gst_mythtv_src_class_init (GstMythtvSrcClass * klass)
{
GObjectClass *gobject_class;
GstPushSrcClass *gstpushsrc_class;
GstBaseSrcClass *gstbasesrc_class;
gobject_class = (GObjectClass *) klass;
gstbasesrc_class = (GstBaseSrcClass *) klass;
gstpushsrc_class = (GstPushSrcClass *) klass;
gobject_class->set_property = gst_mythtv_src_set_property;
gobject_class->get_property = gst_mythtv_src_get_property;
gobject_class->finalize = gst_mythtv_src_finalize;
g_object_class_install_property
(gobject_class, PROP_LOCATION,
g_param_spec_string ("location", "Location",
"The location. In the form:"
"\n\t\t\tmyth://a.com/file.nuv"
"\n\t\t\tmyth://a.com:23223/file.nuv "
"\n\t\t\ta.com/file.nuv - default scheme 'myth'",
"", G_PARAM_READWRITE));
g_object_class_install_property
(gobject_class, PROP_GMYTHTV_VERSION,
g_param_spec_int ("mythtv-version", "mythtv-version",
"Change MythTV version", 26, 30, 26, G_PARAM_READWRITE));
g_object_class_install_property
(gobject_class, PROP_GMYTHTV_LIVEID,
g_param_spec_int ("mythtv-live-id", "mythtv-live-id",
"Change MythTV version",
0, 200, GST_GMYTHTV_ID_NUM, G_PARAM_READWRITE));
g_object_class_install_property
(gobject_class, PROP_GMYTHTV_LIVE_CHAINID,
g_param_spec_string ("mythtv-live-chainid", "mythtv-live-chainid",
"Sets the MythTV chain ID (from TV Chain)", "", G_PARAM_READWRITE));
g_object_class_install_property
(gobject_class, PROP_GMYTHTV_LIVE,
g_param_spec_boolean ("mythtv-live", "mythtv-live",
"Enable MythTV Live TV content streaming", FALSE, G_PARAM_READWRITE));
g_object_class_install_property
(gobject_class, PROP_GMYTHTV_ENABLE_TIMING_POSITION,
g_param_spec_boolean ("mythtv-enable-timing-position",
"mythtv-enable-timing-position",
"Enable MythTV Live TV content size continuous updating", FALSE,
G_PARAM_READWRITE));
g_object_class_install_property
(gobject_class, PROP_GMYTHTV_CHANNEL_NUM,
g_param_spec_int ("mythtv-channel", "mythtv-channel",
"Change MythTV channel number",
-1, 99999, GST_GMYTHTV_CHANNEL_DEFAULT_NUM, G_PARAM_READWRITE));
#ifndef GST_DISABLE_GST_DEBUG
g_object_class_install_property
(gobject_class, PROP_GMYTHTV_DBG,
g_param_spec_boolean ("mythtv-debug", "mythtv-debug",
"Enable MythTV debug messages", FALSE, G_PARAM_READWRITE));
#endif
gstbasesrc_class->start = gst_mythtv_src_start;
gstbasesrc_class->stop = gst_mythtv_src_stop;
gstbasesrc_class->get_size = gst_mythtv_src_get_size;
gstbasesrc_class->is_seekable = gst_mythtv_src_is_seekable;
gstbasesrc_class->do_seek = gst_mythtv_src_do_seek;
gstpushsrc_class->create = gst_mythtv_src_create;
GST_DEBUG_CATEGORY_INIT (mythtvsrc_debug, "mythtvsrc", 0,
"MythTV Client Source");
}
static void
gst_mythtv_src_init (GstMythtvSrc * this, GstMythtvSrcClass * g_class)
{
this->file_transfer = NULL;
this->unique_setup = FALSE;
this->mythtv_version = GMYTHTV_VERSION_DEFAULT;
this->state = GST_MYTHTV_SRC_FILE_TRANSFER;
this->bytes_read = 0;
this->prev_content_size = 0;
this->content_size = 0;
this->read_offset = 0;
this->content_size_last = 0;
this->live_tv = FALSE;
this->enable_timing_position = FALSE;
this->update_prog_chain = FALSE;
this->user_agent = g_strdup ("mythtvsrc");
this->mythtv_caps = NULL;
this->update_prog_chain = FALSE;
this->channel_num = GST_GMYTHTV_CHANNEL_DEFAULT_NUM;
this->eos = FALSE;
this->bytes_queue = NULL;
this->wait_to_transfer = 0;
gst_base_src_set_format (GST_BASE_SRC (this), GST_FORMAT_BYTES);
gst_pad_set_event_function (GST_BASE_SRC_PAD (GST_BASE_SRC (this)),
gst_mythtv_src_handle_event);
gst_pad_set_query_function (GST_BASE_SRC_PAD (GST_BASE_SRC (this)),
gst_mythtv_src_handle_query);
}
static void
gst_mythtv_src_finalize (GObject * gobject)
{
GstMythtvSrc *this = GST_MYTHTV_SRC (gobject);
if (this->mythtv_caps) {
gst_caps_unref (this->mythtv_caps);
this->mythtv_caps = NULL;
}
if (this->file_transfer) {
g_object_unref (this->file_transfer);
this->file_transfer = NULL;
}
if (this->spawn_livetv) {
g_object_unref (this->spawn_livetv);
this->spawn_livetv = NULL;
}
if (this->backend_info) {
g_object_unref (this->backend_info);
this->backend_info = NULL;
}
if (this->uri_name) {
g_free (this->uri_name);
}
if (this->user_agent) {
g_free (this->user_agent);
}
if (this->bytes_queue) {
g_byte_array_free (this->bytes_queue, TRUE);
this->bytes_queue = NULL;
}
G_OBJECT_CLASS (parent_class)->finalize (gobject);
}
static gint
do_read_request_response (GstMythtvSrc * src, guint size, GByteArray * data_ptr)
{
gint read = 0;
guint sizetoread = size;
gint max_iters = GMYTHTV_TRANSFER_MAX_RESENDS;
GST_LOG_OBJECT (src, "Starting: Reading %d bytes...", sizetoread);
/* Loop sending the Myth File Transfer request:
* Retry whilst authentication fails and we supply it. */
gint len = 0;
while (sizetoread == size && --max_iters > 0) {
len = gmyth_file_transfer_read (src->file_transfer,
data_ptr, sizetoread, TRUE);
if (len > 0) {
read += len;
sizetoread -= len;
} else if (len < 0) {
if (src->live_tv == FALSE) {
read = -1;
goto eos;
} else {
if (len == GMYTHTV_FILE_TRANSFER_READ_ERROR) { /* -314 */
GST_INFO_OBJECT (src, "[LiveTV] FileTransfer READ_ERROR!");
goto done;
} else if (len == GMYTHTV_FILE_TRANSFER_NEXT_PROG_CHAIN) { /* -315 */
GST_INFO_OBJECT (src,
"[LiveTV] FileTransfer - Go to the next program chain!");
continue;
}
goto done;
}
} else if (len == 0)
goto done;
if (read == sizetoread)
goto done;
}
if ((read < 0 && !src->live_tv) || max_iters == 0)
goto eos;
goto done;
eos:
src->eos = TRUE;
done:
return read;
}
static GstFlowReturn
gst_mythtv_src_create (GstPushSrc * psrc, GstBuffer ** outbuf)
{
GstMythtvSrc *src;
GstFlowReturn ret = GST_FLOW_OK;
gint read = -1;
src = GST_MYTHTV_SRC (psrc);
/* The caller should know the number of bytes and not read beyond EOS. */
if (G_UNLIKELY (src->eos))
goto eos;
if (G_UNLIKELY (src->update_prog_chain))
goto change_progchain;
GST_DEBUG_OBJECT (src, "offset = %" G_GUINT64_FORMAT ", size = %d...",
src->read_offset, MAX_READ_SIZE);
GST_DEBUG_OBJECT (src, "Create: buffer_remain: %d, buffer_size = %d.",
(gint) src->buffer_remain, src->bytes_queue->len);
/* just get from the byte array, no network effort... */
if ((src->buffer_remain = src->bytes_queue->len) < MAX_READ_SIZE) {
GByteArray *buffer = NULL;
guint buffer_size_inter = (INTERNAL_BUFFER_SIZE - src->buffer_remain);
if (buffer_size_inter > REQUEST_MAX_SIZE)
buffer_size_inter = REQUEST_MAX_SIZE;
buffer = g_byte_array_new ();
read = do_read_request_response (src, buffer_size_inter, buffer);
if (G_UNLIKELY (read < 0)) {
if (src->live_tv)
goto change_progchain;
else
goto read_error;
} else if (G_UNLIKELY (read == 0)) {
if (!src->live_tv)
goto eos;
else
goto done;
}
if (G_UNLIKELY (src->update_prog_chain))
goto change_progchain;
src->bytes_queue =
g_byte_array_append (src->bytes_queue, buffer->data, read);
if (read > buffer_size_inter)
GST_WARNING_OBJECT (src,
"INCREASED buffer size! Backend sent more than we ask him... (%d)",
abs (read - buffer_size_inter));
src->buffer_remain += read;
if (buffer != NULL) {
g_byte_array_free (buffer, TRUE);
buffer = NULL;
}
GST_DEBUG_OBJECT (src,
"BYTES READ (actual) = %d, BYTES READ (cumulative) = %llu, "
"OFFSET = %llu, CONTENT SIZE = %llu.", read,
src->bytes_read, src->read_offset, src->content_size);
}
guint buffer_size =
(src->buffer_remain < MAX_READ_SIZE) ? src->buffer_remain : MAX_READ_SIZE;
*outbuf = gst_buffer_new ();
/* gets the first buffer_size bytes from the byte array buffer variable */
/* guint8 *buf = g_memdup( src->bytes_queue->data, buffer_size ); */
GST_DEBUG_OBJECT (src, "read from network? %s!, buffer_remain = %d",
(read ==
-1) ? "NO, got from buffer" : "YES, go see the backend's log file",
src->buffer_remain);
GST_BUFFER_SIZE (*outbuf) = buffer_size;
GST_BUFFER_MALLOCDATA (*outbuf) = g_malloc0 (GST_BUFFER_SIZE (*outbuf));
GST_BUFFER_DATA (*outbuf) = GST_BUFFER_MALLOCDATA (*outbuf);
g_memmove (GST_BUFFER_DATA ((*outbuf)), src->bytes_queue->data,
GST_BUFFER_SIZE (*outbuf));
GST_BUFFER_OFFSET (*outbuf) = src->read_offset;
GST_BUFFER_OFFSET_END (*outbuf) =
src->read_offset + GST_BUFFER_SIZE (*outbuf);
src->buffer_remain -= GST_BUFFER_SIZE (*outbuf);
src->read_offset += GST_BUFFER_SIZE (*outbuf);
src->bytes_read += GST_BUFFER_SIZE (*outbuf);
GST_DEBUG_OBJECT (src, "Buffer output with size: %d",
GST_BUFFER_SIZE (*outbuf));
/* flushs the newly buffer got from byte array */
src->bytes_queue =
g_byte_array_remove_range (src->bytes_queue, 0, buffer_size);
GST_DEBUG_OBJECT (src, "Got buffer: BUFFER --->SIZE = %d, OFFSET = %llu, "
"OFFSET_END = %llu.", GST_BUFFER_SIZE (*outbuf),
GST_BUFFER_OFFSET (*outbuf), GST_BUFFER_OFFSET_END (*outbuf));
GST_DEBUG_OBJECT (src, "CONTENT_SIZE = %llu, BYTES_READ = %llu.",
src->content_size, src->bytes_read);
if (G_UNLIKELY (src->eos) || (!src->live_tv
&& (src->bytes_read >= src->content_size)))
goto eos;
done:
{
const gchar *reason = gst_flow_get_name (ret);
GST_DEBUG_OBJECT (src, "DONE task, reason %s", reason);
return ret;
}
eos:
{
const gchar *reason = gst_flow_get_name (ret);
GST_DEBUG_OBJECT (src, "pausing task, reason %s", reason);
return GST_FLOW_UNEXPECTED;
}
/* ERRORS */
read_error:
{
GST_ELEMENT_ERROR (src, RESOURCE, READ,
(NULL), ("Could not read any bytes (%i, %s)", read, src->uri_name));
return GST_FLOW_ERROR;
}
change_progchain:
{
GST_ELEMENT_ERROR (src, RESOURCE, READ,
(NULL), ("Seek failed, go to the next program info... (%i, %s)", read,
src->uri_name));
gst_pad_push_event (GST_BASE_SRC_PAD (GST_BASE_SRC (psrc)),
gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0, -1, 0));
/* go to the next program chain */
src->unique_setup = FALSE;
src->update_prog_chain = TRUE;
gst_mythtv_src_next_program_chain (src);
return GST_FLOW_ERROR_NO_DATA;
}
}
gint64
gst_mythtv_src_get_position (GstMythtvSrc * src)
{
gint64 size_tmp = 0;
guint max_tries = 2;
if (src->live_tv == TRUE && (abs (src->content_size - src->bytes_read) <
GMYTHTV_TRANSFER_MAX_BUFFER)) {
get_file_pos:
g_usleep (10);
size_tmp = gmyth_recorder_get_file_position (src->spawn_livetv->recorder);
if (size_tmp > (src->content_size + GMYTHTV_TRANSFER_MAX_BUFFER))
src->content_size = size_tmp;
else if (size_tmp > 0 && --max_tries > 0)
goto get_file_pos;
GST_LOG_OBJECT (src, "GET_POSITION: file_position = %lld", size_tmp);
/* sets the last content size amount before it can be updated */
src->prev_content_size = src->content_size;
}
return src->content_size;
}
static gboolean
gst_mythtv_src_do_seek (GstBaseSrc * base, GstSegment * segment)
{
GstMythtvSrc *src = GST_MYTHTV_SRC (base);
gint64 new_offset = -1;
gint64 actual_seek = segment->start;
gboolean ret = TRUE;
GST_LOG_OBJECT (src, "seek, segment: %" GST_SEGMENT_FORMAT, segment);
if (segment->format == GST_FORMAT_TIME) {
goto done;
}
GST_LOG_OBJECT (src,
"Trying to seek at the value (actual_seek = %lld, read_offset = %lld)",
actual_seek, src->read_offset);
/* verify if it needs to seek */
if (src->read_offset != actual_seek) {
new_offset =
gmyth_file_transfer_seek (src->file_transfer, segment->start, SEEK_SET);
GST_LOG_OBJECT (src,
"Segment offset start = %lld, SRC Offset = %lld, NEW actual backend SEEK Offset = %lld.",
segment->start, src->read_offset, new_offset);
if (G_UNLIKELY (new_offset < 0)) {
ret = FALSE;
if (src->live_tv)
goto change_progchain;
else
goto eos;
}
src->read_offset = new_offset;
if (ret == FALSE) {
GST_INFO_OBJECT (src, "Failed to set the SEEK on segment!");
}
}
done:
return ret;
eos:
{
GST_DEBUG_OBJECT (src, "EOS found on seeking!!!");
return FALSE;
}
change_progchain:
{
GST_ELEMENT_ERROR (src, RESOURCE, READ,
(NULL), ("Seek failed, go to the next program info... (%i, %s)", read,
src->uri_name));
gst_pad_push_event (GST_BASE_SRC_PAD (base),
gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0, -1, 0));
/* go to the next program chain */
src->unique_setup = FALSE;
src->update_prog_chain = TRUE;
gst_mythtv_src_next_program_chain (src);
return TRUE;
}
}
/* create a socket for connecting to remote server */
static gboolean
gst_mythtv_src_start (GstBaseSrc * bsrc)
{
GstMythtvSrc *src = GST_MYTHTV_SRC (bsrc);
GString *chain_id_local = NULL;
GMythURI *gmyth_uri = NULL;
gboolean ret = TRUE;
if (G_UNLIKELY (src->update_prog_chain))
goto change_progchain;
if (src->unique_setup == FALSE) {
src->unique_setup = TRUE;
} else {
goto done;
}
gmyth_uri = gmyth_uri_new_with_value (src->uri_name);
src->backend_info = gmyth_backend_info_new_with_uri (src->uri_name);
/* testing UPnP... */
/* gmyth_backend_info_set_hostname( src->backend_info, NULL ); */
if (src->live_tv || gmyth_uri_is_livetv (gmyth_uri)) {
gint ch;
src->spawn_livetv = gmyth_livetv_new ();
ch = gmyth_uri_get_channel_num (gmyth_uri);
if (ch != -1)
src->channel_num = ch;
if (src->channel_num != GST_GMYTHTV_CHANNEL_DEFAULT_NUM) {
if (gmyth_livetv_channel_setup (src->spawn_livetv, src->channel_num,
src->backend_info) == FALSE) {
GST_INFO_OBJECT (src, "LiveTV setup felt down on error");
ret = FALSE;
goto init_failed;
}
} else {
if (gmyth_livetv_setup (src->spawn_livetv, src->backend_info) == FALSE) {
GST_INFO_OBJECT (src, "LiveTV setup felt down on error");
ret = FALSE;
goto init_failed;
}
}
src->file_transfer = gmyth_livetv_create_file_transfer (src->spawn_livetv);
if (NULL == src->file_transfer) {
GST_INFO_OBJECT (src, "[LiveTV] FileTransfer equals to NULL");
ret = FALSE;
goto init_failed;
}
} else {
src->file_transfer = gmyth_file_transfer_new (src->backend_info);
ret = gmyth_file_transfer_open (src->file_transfer, src->uri_name);
}
if (NULL == src->file_transfer) {
GST_INFO_OBJECT (src, "FileTransfer is NULL");
goto init_failed;
}
/*GST_INFO_OBJECT( src, "uri = %s", src->spawn_livetv->file_transfer); */
if (ret == FALSE) {
#ifndef GST_DISABLE_GST_DEBUG
if (src->mythtv_msgs_dbg)
GST_INFO_OBJECT (src,
"MythTV FileTransfer request failed when setting up socket connection!");
#endif
goto begin_req_failed;
}
GST_INFO_OBJECT (src,
"MythTV FileTransfer filesize = %lld, content_size = %lld!",
src->file_transfer->filesize, src->content_size);
src->content_size = src->file_transfer->filesize;
src->do_start = FALSE;
/* this is used for the buffer cache */
src->bytes_queue = g_byte_array_sized_new (INTERNAL_BUFFER_SIZE);
src->buffer_remain = 0;
gst_pad_push_event (GST_BASE_SRC_PAD (GST_BASE_SRC (src)),
gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0,
src->content_size, 0));
done:
/*if ( gmyth_uri != NULL )
{
g_object_unref( gmyth_uri );
gmyth_uri = NULL;
} */
if (chain_id_local != NULL) {
g_string_free (chain_id_local, TRUE);
chain_id_local = NULL;
}
return TRUE;
/* ERRORS */
init_failed:
{
if (src->spawn_livetv != NULL)
g_object_unref (src->spawn_livetv);
GST_ELEMENT_ERROR (src, LIBRARY, INIT,
(NULL), ("Could not initialize MythTV library (%i, %s)", ret,
src->uri_name));
return FALSE;
}
begin_req_failed:
{
GST_ELEMENT_ERROR (src, LIBRARY, INIT,
(NULL), ("Could not begin request sent to MythTV server (%i, %s)", ret,
src->uri_name));
return FALSE;
}
change_progchain:
{
GST_ELEMENT_ERROR (src, RESOURCE, READ,
(NULL), ("Seek failed, go to the next program info... (%s)",
src->uri_name));
gst_pad_push_event (GST_BASE_SRC_PAD (GST_BASE_SRC (src)),
gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0, -1, 0));
/* go to the next program chain */
src->unique_setup = FALSE;
src->update_prog_chain = TRUE;
gst_mythtv_src_next_program_chain (src);
return TRUE;
}
}
/* create a new socket for connecting to the next program chain */
static gboolean
gst_mythtv_src_next_program_chain (GstMythtvSrc * src)
{
GString *chain_id_local = NULL;
gboolean ret = TRUE;
if (!src->live_tv)
goto init_failed;
if (src->unique_setup == FALSE) {
src->unique_setup = TRUE;
} else {
goto done;
}
GST_PAD_STREAM_LOCK (GST_BASE_SRC_PAD (GST_BASE_SRC (src)));
if (src->file_transfer) {
g_object_unref (src->file_transfer);
src->file_transfer = NULL;
}
if (src->uri_name) {
g_free (src->uri_name);
}
if (src->backend_info == NULL)
src->backend_info = gmyth_backend_info_new_with_uri (src->uri_name);
if (src->live_tv) {
if (gmyth_livetv_next_program_chain (src->spawn_livetv) == FALSE) {
GST_INFO_OBJECT (src, "Failed to go to the next program chain!");
ret = FALSE;
goto init_failed;
}
/* set up the uri variable */
src->uri_name = g_strdup (src->spawn_livetv->proginfo->pathname->str);
chain_id_local = gmyth_tvchain_get_id (src->spawn_livetv->tvchain);
if (chain_id_local != NULL) {
src->live_chain_id = g_strdup (chain_id_local->str);
GST_DEBUG_OBJECT (src, "Local chain ID = %s", src->live_chain_id);
}
src->live_tv_id = src->spawn_livetv->recorder->recorder_num;
GST_LOG_OBJECT (src, "LiveTV id = %d, URI path = %s",
src->live_tv_id, src->uri_name);
}
src->file_transfer = gmyth_file_transfer_new (src->backend_info);
if (src->file_transfer == NULL) {
goto init_failed;
}
ret = gmyth_file_transfer_open (src->file_transfer, src->uri_name);
/* sets the Playback monitor connection */
/* sets the FileTransfer instance connection (video/audio download) */
if (ret == FALSE) {
#ifndef GST_DISABLE_GST_DEBUG
if (src->mythtv_msgs_dbg)
GST_ERROR_OBJECT (src,
"MythTV FileTransfer request failed when setting up socket connection!");
#endif
goto begin_req_failed;
}
src->content_size_last = src->content_size;
src->content_size = src->file_transfer->filesize;
if (src->live_tv) {
src->wait_to_transfer = 0;
while (src->wait_to_transfer++ < GMYTHTV_TRANSFER_MAX_WAITS &&
src->content_size < GMYTHTV_TRANSFER_MAX_BUFFER)
src->content_size = gst_mythtv_src_get_position (src);
}
src->read_offset = 0;
if (src->bytes_queue != NULL) {
g_byte_array_free (src->bytes_queue, TRUE);
}
src->bytes_queue = g_byte_array_sized_new (INTERNAL_BUFFER_SIZE);
done:
src->update_prog_chain = FALSE;
GST_PAD_STREAM_UNLOCK (GST_BASE_SRC_PAD (GST_BASE_SRC (src)));
return TRUE;
/* ERRORS */
init_failed:
{
if (src->spawn_livetv != NULL)
g_object_unref (src->spawn_livetv);
GST_ELEMENT_ERROR (src, LIBRARY, INIT,
(NULL), ("Could not initialize MythTV library (%i, %s)", ret,
src->uri_name));
return FALSE;
}
begin_req_failed:
{
GST_ELEMENT_ERROR (src, LIBRARY, INIT,
(NULL), ("Could not begin request sent to MythTV server (%i, %s)", ret,
src->uri_name));
return FALSE;
}
}
static gboolean
gst_mythtv_src_get_size (GstBaseSrc * bsrc, guint64 * size)
{
GstMythtvSrc *src = GST_MYTHTV_SRC (bsrc);
gboolean ret = TRUE;
GST_LOG_OBJECT (src, "Differs from previous content size: %d (max.: %d)",
abs (src->content_size - src->prev_content_size),
GMYTHTV_TRANSFER_MAX_BUFFER);
if (src->live_tv) {
ret = FALSE;
} else if (src->live_tv && src->enable_timing_position
&& (abs (src->content_size - src->bytes_read) <
GMYTHTV_TRANSFER_MAX_BUFFER)) {
gint64 new_offset =
gmyth_recorder_get_file_position (src->spawn_livetv->recorder);
if (new_offset > 0 && new_offset > src->content_size) {
src->content_size = new_offset;
} else if (new_offset < src->content_size) {
src->update_prog_chain = TRUE;
}
}
*size = src->content_size;
GST_LOG_OBJECT (src, "Content size = %lld", src->content_size);
return ret;
}
/* close the socket and associated resources
* used both to recover from errors and go to NULL state */
static gboolean
gst_mythtv_src_stop (GstBaseSrc * bsrc)
{
GstMythtvSrc *src;
src = GST_MYTHTV_SRC (bsrc);
if (src->uri_name) {
g_free (src->uri_name);
src->uri_name = NULL;
}
if (src->mythtv_caps) {
gst_caps_unref (src->mythtv_caps);
src->mythtv_caps = NULL;
}
src->eos = FALSE;
return TRUE;
}
static gboolean
gst_mythtv_src_handle_event (GstPad * pad, GstEvent * event)
{
GstMythtvSrc *src = GST_MYTHTV_SRC (GST_PAD_PARENT (pad));
gint64 cont_size = 0;
gboolean ret = FALSE;
switch (GST_EVENT_TYPE (event)) {
case GST_EVENT_EOS:
GST_WARNING_OBJECT (src, "Got EOS event");
if (src->live_tv) {
cont_size = gst_mythtv_src_get_position (src);
if (cont_size > src->content_size) {
src->content_size = cont_size;
src->eos = FALSE;
} else {
src->eos = TRUE;
gst_element_set_state (GST_ELEMENT (src), GST_STATE_NULL);
gst_element_set_locked_state (GST_ELEMENT (src), FALSE);
}
}
break;
default:
ret = gst_pad_event_default (pad, event);
}
return ret;
}
static gboolean
gst_mythtv_src_is_seekable (GstBaseSrc * push_src)
{
return TRUE;
}
static gboolean
gst_mythtv_src_handle_query (GstPad * pad, GstQuery * query)
{
gboolean res = FALSE;
GstMythtvSrc *myth = GST_MYTHTV_SRC (gst_pad_get_parent (pad));
GstFormat formt;
switch (GST_QUERY_TYPE (query)) {
case GST_QUERY_POSITION:
{
gst_query_parse_position (query, &formt, NULL);
if (formt == GST_FORMAT_BYTES) {
gst_query_set_position (query, formt, myth->read_offset);
GST_DEBUG_OBJECT (myth, "POS %" G_GINT64_FORMAT, myth->read_offset);
res = TRUE;
} else if (formt == GST_FORMAT_TIME) {
res = gst_pad_query_default (pad, query);
}
break;
}
case GST_QUERY_DURATION:
{
#if 0
if (myth->duration != 0) {
gint64 total;
gint64 fps;
fps = nuv->h->i_fpsn / nuv->h->i_fpsd;
total =
gst_util_uint64_scale_int (GST_SECOND, nuv->h->i_video_blocks, fps);
}
#endif
gst_query_parse_duration (query, &formt, NULL);
if (formt == GST_FORMAT_BYTES) {
gst_query_set_duration (query, formt, myth->content_size);
GST_DEBUG_OBJECT (myth, "SIZE %" G_GINT64_FORMAT, myth->content_size);
res = TRUE;
} else if (formt == GST_FORMAT_TIME) {
res = gst_pad_query_default (pad, query);
}
break;
}
default:
{
res = gst_pad_query_default (pad, query);
break;
}
}
gst_object_unref (myth);
return res;
}
static GstStateChangeReturn
gst_mythtv_src_change_state (GstElement * element, GstStateChange transition)
{
GstStateChangeReturn ret = GST_STATE_CHANGE_FAILURE;
GstMythtvSrc *src = GST_MYTHTV_SRC (element);
switch (transition) {
case GST_STATE_CHANGE_NULL_TO_READY:
break;
case GST_STATE_CHANGE_READY_TO_PAUSED:
break;
case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
if (src->live_tv) {
if (!gmyth_recorder_send_frontend_ready_command (src->spawn_livetv->
recorder))
GST_WARNING_OBJECT (src,
"Couldn't send the FRONTEND_READY message to the backend!");
else
GST_DEBUG_OBJECT (src, "FRONTEND_READY was sent to the backend");
}
break;
default:
break;
}
ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
if (ret == GST_STATE_CHANGE_FAILURE)
return ret;
switch (transition) {
case GST_STATE_CHANGE_READY_TO_NULL:
break;
case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
case GST_STATE_CHANGE_PAUSED_TO_READY:
if (src->live_tv) {
if (!gmyth_recorder_send_frontend_ready_command (src->spawn_livetv->
recorder))
GST_WARNING_OBJECT (src,
"Couldn't send the FRONTEND_READY message to the backend!");
else
GST_DEBUG_OBJECT (src, "FRONTEND_READY was sent to the backend");
}
break;
default:
break;
}
return ret;
}
static void
gst_mythtv_src_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec)
{
GstMythtvSrc *mythtvsrc = GST_MYTHTV_SRC (object);
GST_OBJECT_LOCK (mythtvsrc);
switch (prop_id) {
case PROP_LOCATION:
{
if (!g_value_get_string (value)) {
GST_WARNING ("location property cannot be NULL");
break;
}
if (mythtvsrc->uri_name != NULL) {
g_free (mythtvsrc->uri_name);
mythtvsrc->uri_name = NULL;
}
mythtvsrc->uri_name = g_value_dup_string (value);
break;
}
#ifndef GST_DISABLE_GST_DEBUG
case PROP_GMYTHTV_DBG:
{
mythtvsrc->mythtv_msgs_dbg = g_value_get_boolean (value);
break;
}
#endif
case PROP_GMYTHTV_VERSION:
{
mythtvsrc->mythtv_version = g_value_get_int (value);
break;
}
case PROP_GMYTHTV_LIVEID:
{
mythtvsrc->live_tv_id = g_value_get_int (value);
break;
}
case PROP_GMYTHTV_LIVE:
{
mythtvsrc->live_tv = g_value_get_boolean (value);
break;
}
case PROP_GMYTHTV_ENABLE_TIMING_POSITION:
{
mythtvsrc->enable_timing_position = g_value_get_boolean (value);
break;
}
case PROP_GMYTHTV_LIVE_CHAINID:
{
if (!g_value_get_string (value)) {
GST_WARNING ("MythTV Live chainid property cannot be NULL");
break;
}
if (mythtvsrc->live_chain_id != NULL) {
g_free (mythtvsrc->live_chain_id);
mythtvsrc->live_chain_id = NULL;
}
mythtvsrc->live_chain_id = g_value_dup_string (value);
break;
}
case PROP_GMYTHTV_CHANNEL_NUM:
{
mythtvsrc->channel_num = g_value_get_int (value);
break;
}
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
GST_OBJECT_UNLOCK (mythtvsrc);
}
static void
gst_mythtv_src_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec)
{
GstMythtvSrc *mythtvsrc = GST_MYTHTV_SRC (object);
GST_OBJECT_LOCK (mythtvsrc);
switch (prop_id) {
case PROP_LOCATION:
{
gchar *str = g_strdup ("");
if (mythtvsrc->uri_name == NULL) {
g_free (mythtvsrc->uri_name);
mythtvsrc->uri_name = NULL;
} else {
str = g_strdup (mythtvsrc->uri_name);
}
g_value_set_string (value, str);
break;
}
#ifndef GST_DISABLE_GST_DEBUG
case PROP_GMYTHTV_DBG:
g_value_set_boolean (value, mythtvsrc->mythtv_msgs_dbg);
break;
#endif
case PROP_GMYTHTV_VERSION:
{
g_value_set_int (value, mythtvsrc->mythtv_version);
break;
}
case PROP_GMYTHTV_LIVEID:
{
g_value_set_int (value, mythtvsrc->live_tv_id);
break;
}
case PROP_GMYTHTV_LIVE:
g_value_set_boolean (value, mythtvsrc->live_tv);
break;
case PROP_GMYTHTV_ENABLE_TIMING_POSITION:
g_value_set_boolean (value, mythtvsrc->enable_timing_position);
break;
case PROP_GMYTHTV_LIVE_CHAINID:
{
gchar *str = g_strdup ("");
if (mythtvsrc->live_chain_id == NULL) {
g_free (mythtvsrc->live_chain_id);
mythtvsrc->live_chain_id = NULL;
} else {
str = g_strdup (mythtvsrc->live_chain_id);
}
g_value_set_string (value, str);
break;
}
case PROP_GMYTHTV_CHANNEL_NUM:
{
g_value_set_int (value, mythtvsrc->channel_num);
break;
}
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
GST_OBJECT_UNLOCK (mythtvsrc);
}
static gboolean
plugin_init (GstPlugin * plugin)
{
return gst_element_register (plugin, "mythtvsrc", GST_RANK_NONE,
GST_TYPE_MYTHTV_SRC);
}
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
GST_VERSION_MINOR,
"mythtv",
"lib MythTV src",
plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN);
/*** GSTURIHANDLER INTERFACE *************************************************/
static guint
gst_mythtv_src_uri_get_type (void)
{
return GST_URI_SRC;
}
static gchar **
gst_mythtv_src_uri_get_protocols (void)
{
static gchar *protocols[] = { "myth", "myths", NULL };
return protocols;
}
static const gchar *
gst_mythtv_src_uri_get_uri (GstURIHandler * handler)
{
GstMythtvSrc *src = GST_MYTHTV_SRC (handler);
return src->uri_name;
}
static gboolean
gst_mythtv_src_uri_set_uri (GstURIHandler * handler, const gchar * uri)
{
GstMythtvSrc *src = GST_MYTHTV_SRC (handler);
gchar *protocol;
protocol = gst_uri_get_protocol (uri);
if ((strcmp (protocol, "myth") != 0) && (strcmp (protocol, "myths") != 0)) {
g_free (protocol);
return FALSE;
}
g_free (protocol);
g_object_set (src, "location", uri, NULL);
return TRUE;
}
static void
gst_mythtv_src_uri_handler_init (gpointer g_iface, gpointer iface_data)
{
GstURIHandlerInterface *iface = (GstURIHandlerInterface *) g_iface;
iface->get_type = gst_mythtv_src_uri_get_type;
iface->get_protocols = gst_mythtv_src_uri_get_protocols;
iface->get_uri = gst_mythtv_src_uri_get_uri;
iface->set_uri = gst_mythtv_src_uri_set_uri;
}
void
size_header_handler (void *userdata, const char *value)
{
GstMythtvSrc *src = GST_MYTHTV_SRC (userdata);
GST_DEBUG_OBJECT (src, "content size = %lld bytes", src->content_size);
}