/*

  * 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. 

  */

 /**

  * SECTION:element-mythtvsrc

  *

  * <refsect2>

  * <para>

  * MythTVSrc allows to access a remote MythTV backend streaming Video/Audio server,

  * and to render audio and video content through a TCP/IP connection to a specific

  * port on this server, and based on a known MythTV protocol that is based on 

  * some message passing, such as REQUEST_BLOCK on a specified number of bytes, to get

  * some chunk of remote file data.

  * You should pass the information aboute the remote MythTV backend server 

  * through the <link linkend="GstMythTVSrc--location">location</link> property.

  * </para>

  * <title>Examples</title>

  * <para>

  * If you want to get the LiveTV content (set channel, TV tuner, RemoteEncoder, 

  * Recorder),

  * put the following URI:

  * 

  * <programlisting> 

  *  myth://xxx.xxx.xxx.xxx:6543/livetv?channel=BBC

  * </programlisting>

  *

  * This URI will say to the gmyth library to configure the Recorder instance (used to

  * change the channel, start the TV multimedia content transmition, etc.), using

  * the IP address (xxx.xxx.xxx.xxx) and port number (6543) of the MythTV backend 

  * server, and setting the channel name to "BBC". 

  * 

  * To get a already recorded the MythTV NUV file, put the following URI:

  * 

  * <programlisting>

  *  myth://xxx.xxx.xxx.xxx:6543/filename.nuv

  * </programlisting>

  *

  * This URI will say to the gmyth library to configure the Recorder instance (used to

  * change the channel, start the TV multimedia content transmition, etc.), using

  * the IP address (xxx.xxx.xxx.xxx) and port number (6543) of the MythTV backend 

  * server, and setting the channel name to "BBC".

  * 

  * Another possible way to use the LiveTV content, and just in the case you want to 

  * use the mysql database, put the location URI in the following format:

  * 

  * <programlisting> 

  *  myth://mythtv:mythtv@xxx.xxx.xxx.xxx:6543/?mythconverg&channel=9

  * </programlisting>

  * 

  * 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).

  * </para>

  * </refsect2>

  */

 #ifdef HAVE_CONFIG_H

 #include "config.h"

 #endif

 #include "gstmythtvsrc.h"

 #include <gmyth/gmyth_file.h>

 #include <gmyth/gmyth_file_transfer.h>

 #include <gmyth/gmyth_file_local.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_ANY);

     /*

      * 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,

     PROP_GMYTHTV_MAX_TRY

 };

 static void     gst_mythtv_src_clear(GstMythtvSrc * mythtv_src);

 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          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 GMythFileReadResult 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_string("mythtv-channel", "mythtv-channel",

                              "Change MythTV channel number",

                              "", G_PARAM_READWRITE));

     g_object_class_install_property

         (gobject_class, PROP_GMYTHTV_MAX_TRY,

          g_param_spec_int("max-try", "max-try",

                           "Set the max try for get MythTV free recorder",

                           0, G_MAXINT, 10, 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 = 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->update_prog_chain = FALSE;

     this->channel_name = NULL;

     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_clear(GstMythtvSrc * mythtv_src)

 {

     mythtv_src->unique_setup = FALSE;

     if (mythtv_src->spawn_livetv) {

         g_object_unref(mythtv_src->spawn_livetv);

         mythtv_src->spawn_livetv = NULL;

     }

     if (mythtv_src->file) {

         g_object_unref(mythtv_src->file);

         mythtv_src->file = NULL;

     }

     if (mythtv_src->backend_info) {

         g_object_unref(mythtv_src->backend_info);

         mythtv_src->backend_info = NULL;

     }

     if (mythtv_src->bytes_queue) {

         g_byte_array_free(mythtv_src->bytes_queue, TRUE);

         mythtv_src->bytes_queue = NULL;

     }

 }

 static void

 gst_mythtv_src_finalize(GObject * gobject)

 {

     GstMythtvSrc   *this = GST_MYTHTV_SRC(gobject);

     gst_mythtv_src_clear(this);

     if (this->uri_name) {

         g_free(this->uri_name);

         this->uri_name = NULL;

     }

     if (this->user_agent) {

         g_free(this->user_agent);

         this->user_agent = NULL;

     }

     G_OBJECT_CLASS(parent_class)->finalize(gobject);

 }

 static GMythFileReadResult

 do_read_request_response(GstMythtvSrc * src, guint size,

                          GByteArray * data_ptr)

 {

     gint            read = 0;

     guint           sizetoread = size;

     gint            max_iters = GMYTHTV_TRANSFER_MAX_RESENDS;

     GMythFileReadResult result;

     GST_LOG_OBJECT(src, "Starting: Reading %d bytes...", sizetoread);

     /*

      * Loop sending the Myth File Transfer request: Retry whilst

      * authentication fails and we supply it. 

      */

     while (sizetoread == size && --max_iters > 0) {

         /*

          * if ( gmyth_backend_info_is_local_file(src->backend_info) ) 

          */

         if (IS_GMYTH_FILE_LOCAL(src->file))

             result = gmyth_file_local_read(GMYTH_FILE_LOCAL(src->file),

                                            data_ptr, sizetoread,

                                            src->live_tv);

         else if (IS_GMYTH_FILE_TRANSFER(src->file))

             result =

                 gmyth_file_transfer_read(GMYTH_FILE_TRANSFER(src->file),

                                          data_ptr, sizetoread,

                                          src->live_tv);

         if (data_ptr->len > 0) {

             read += data_ptr->len;

             sizetoread -= data_ptr->len;

         } else if (data_ptr->len <= 0) {

             if (src->live_tv == FALSE) {

                 result = GMYTH_FILE_READ_EOF;

                 goto eos;

             } else {

                 if (result == GMYTH_FILE_READ_ERROR) {  /* -314 */

                     GST_INFO_OBJECT(src,

                                     "[LiveTV] FileTransfer READ_ERROR!");

                     goto done;

                 } else if (result == GMYTH_FILE_READ_NEXT_PROG_CHAIN) { /* -315 

                                                                          */

                     GST_INFO_OBJECT(src,

                                     "[LiveTV] FileTransfer - Go to athe next program chain!");

                     src->update_prog_chain = TRUE;

                     continue;

                 }

                 goto done;

             }

         }

         /*

          * else if (data_ptr->len == 0) goto done; 

          */

         if (read == sizetoread)

             goto done;

     }

     if ((read < 0 && !src->live_tv) || max_iters == 0) {

         result = GMYTH_FILE_READ_EOF;

         goto eos;

     }

     goto done;

   eos:

     src->eos = TRUE;

   done:

     return result;

 }

 static GstFlowReturn

 gst_mythtv_src_create(GstPushSrc * psrc, GstBuffer ** outbuf)

 {

     GstMythtvSrc   *src;

     GstFlowReturn   ret = GST_FLOW_OK;

     guint           buffer_size_inter = 0;

     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;

     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);

   program_chain_changed:

     /*

      * just get from the byte array, no network effort... 

      */

     if ((src->buffer_remain = src->bytes_queue->len) < MAX_READ_SIZE) {

         GByteArray     *buffer;

         GMythFileReadResult result = GMYTH_FILE_READ_OK;

         buffer = NULL;

         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();

         result = do_read_request_response(src, buffer_size_inter, buffer);

         /*

          * got the next program info? 

          */

         if (G_UNLIKELY(src->update_prog_chain) || 

             (result == GMYTH_FILE_READ_NEXT_PROG_CHAIN)) {

             GST_DEBUG_OBJECT(src,

                              "Update PROGRAM CHAIN!!! buffer_size = %d.",

                              src->bytes_queue->len);

             gst_pad_push_event(GST_BASE_SRC_PAD(GST_BASE_SRC(psrc)),

                                gst_event_new_custom

                                (GST_EVENT_CUSTOM_DOWNSTREAM, NULL));

             /*

              * gst_pad_push_event (GST_BASE_SRC_PAD (GST_BASE_SRC (psrc)),

              * gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_BYTES, 0,

              * -1, 0)); 

              */

             src->update_prog_chain = FALSE;

             src->eos = FALSE;

             if (result == GMYTH_FILE_READ_NEXT_PROG_CHAIN) {

                 /*

                  * if (buffer != NULL) { g_byte_array_free (buffer, TRUE);

                  * buffer = NULL; } goto program_chain_changed; 

                  */

             } else if (result == GMYTH_FILE_READ_OK) {

                 /*

                  * remove wasteful, NUV file header data 

                  */

                 /*

                  * buffer = g_byte_array_remove_range( buffer, 0, 512 ); 

                  */

                 /*

                  * TODO: need to send a new segment event to NUVDemux? 

                  */

                 // gst_pad_push_event (GST_BASE_SRC_PAD (GST_BASE_SRC

                 // (psrc)),

                 // gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_BYTES, 

                 // 0, -1, 0));

                 /*

                  * goto change_progchain; 

                  */

             }

         }

         /*

          */

         if (G_UNLIKELY(buffer->len < 0)) {

             if (buffer != NULL) {

                 g_byte_array_free(buffer, TRUE);

                 buffer = NULL;

             }

             if (src->live_tv || 

                 (result == GMYTH_FILE_READ_NEXT_PROG_CHAIN))

                 goto change_progchain;

             else

                 goto read_error;

         } else if (G_UNLIKELY(buffer->len == 0)) {

             if (buffer != NULL) {

                 g_byte_array_free(buffer, TRUE);

                 buffer = NULL;

             }

             if (!src->live_tv) {

                 if (src->eos)

                     goto eos;

                 else 

                     goto done;

             }

             else

                 goto program_chain_changed;

         }

         src->bytes_queue =

             g_byte_array_append(src->bytes_queue, buffer->data,

                                 buffer->len);

         if (buffer->len > buffer_size_inter)

             GST_WARNING_OBJECT(src,

                                "INCREASED buffer size! Backend sent more than we ask him... (%d)",

                                abs(buffer->len - buffer_size_inter));

         src->buffer_remain += buffer->len;

         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;

     buffer_size = (src->buffer_remain <  MAX_READ_SIZE) ? 

                    src->buffer_remain : 

                    MAX_READ_SIZE;

     *outbuf = gst_buffer_new();

     /*

      * GST_DEBUG_OBJECT (src, "read from network? %s!, buffer_remain =

      * %d", (buffer_size_inter == 0) ? "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);

     if (src->eos  || 

         (!src->live_tv && (src->bytes_read >= src->content_size))

        )

         goto eos;

   done:

     {

         return ret;

     }

   eos:

     {

         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));

         /*

          * TODO: need to send a new segment event to NUVDemux? 

          */

         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));

         goto program_chain_changed;

     }

 }

 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) {

         /*

          * if ( gmyth_backend_info_is_local_file(src->backend_info) ) 

          */

         if (IS_GMYTH_FILE_LOCAL(src->file))

             new_offset =

                 gmyth_file_local_seek(GMYTH_FILE_LOCAL(src->file),

                                       segment->start, G_SEEK_SET);

         else if (IS_GMYTH_FILE_TRANSFER(src->file))

             new_offset =

                 gmyth_file_transfer_seek(GMYTH_FILE_TRANSFER(src->file),

                                          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 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;

     }

 }

 /*

  * 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;

     GstMessage     *msg;

     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);

     src->live_tv = gmyth_uri_is_livetv(gmyth_uri);

     /*

      * testing UPnP... 

      */

     /*

      * gmyth_backend_info_set_hostname( src->backend_info, NULL ); 

      */

     if (src->live_tv) {

         src->spawn_livetv = gmyth_livetv_new(src->backend_info);

         gchar          *ch = gmyth_uri_get_channel_name(gmyth_uri);

         if (ch != NULL)

             src->channel_name = ch;

         if (src->channel_name != NULL) {

             gboolean        result;

             result =

                 gmyth_livetv_channel_name_setup(src->spawn_livetv,

                                                 src->channel_name);

             if (result == FALSE) {

                 GST_INFO_OBJECT(src, "LiveTV setup felt down on error");

                 ret = FALSE;

                 goto init_failed;

             }

         } else {

             if (gmyth_livetv_setup(src->spawn_livetv) == FALSE) {

                 GST_INFO_OBJECT(src, "LiveTV setup felt down on error");

                 ret = FALSE;

                 goto init_failed;

             }

         }

         /*

          * testing change channel... 

          */

         /*

          * gmyth_recorder_change_channel( src->spawn_livetv->recorder,

          * CHANNEL_DIRECTION_UP ); 

          */

         src->file =

             GMYTH_FILE(gmyth_livetv_create_file_transfer

                        (src->spawn_livetv));

         if (NULL == src->file) {

             GST_INFO_OBJECT(src, "[LiveTV] FileTransfer equals to NULL");

             ret = FALSE;

             goto init_failed;

         }

         /*

          * Check if the file is local to this specific client renderer 

          */

         if (gmyth_uri_is_local_file(gmyth_uri))

             ret = gmyth_file_local_open(GMYTH_FILE_LOCAL(src->file));

         else

             ret =

                 gmyth_file_transfer_open(GMYTH_FILE_TRANSFER(src->file),

                                          src->spawn_livetv->uri !=

                                          NULL ? gmyth_uri_get_path(src->

                                                                    spawn_livetv->

                                                                    uri) :

                                          src->spawn_livetv->proginfo->

                                          pathname->str);

         /*

          * sets the mythtvsrc "location" property 

          */

         g_object_set(src, "location", gmyth_file_get_uri(src->file), NULL);

         if (!ret) {

             GST_INFO_OBJECT(src,

                             "Error: couldn't open the FileTransfer from LiveTV source!");

             g_object_unref(src->file);

             src->file = NULL;

             goto init_failed;

         }

     } else {

         /*

          * Check if the file is local to this specific client renderer,

          * and tries to open a local connection 

          */

         if (gmyth_uri_is_local_file(gmyth_uri)) {

             src->file =

                 GMYTH_FILE(gmyth_file_local_new(src->backend_info));

             ret = gmyth_file_local_open(GMYTH_FILE_LOCAL(src->file));

         } else {

             src->file =

                 GMYTH_FILE(gmyth_file_transfer_new(src->backend_info));

             ret =

                 gmyth_file_transfer_open(GMYTH_FILE_TRANSFER(src->file),

                                          src->uri_name);

         }

     }                           /* if (else) - recorded FileTransfer */

     if (NULL == src->file) {

         GST_INFO_OBJECT(src, "FileTransfer is NULL");

         goto init_failed;

     }

     /*

      * GST_INFO_OBJECT( src, "uri = %s", src->spawn_livetv->file); 

      */

     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!",

                     gmyth_file_get_filesize(src->file), src->content_size);

     src->content_size = gmyth_file_get_filesize(src->file);

     msg =

         gst_message_new_duration(GST_OBJECT(src), GST_FORMAT_BYTES,

                                  src->content_size);

     gst_element_post_message(GST_ELEMENT(src), msg);

     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 (gmyth_uri != NULL) {

         g_object_unref(gmyth_uri);

         gmyth_uri = NULL;

     }

     if (src->spawn_livetv != NULL) {

         g_object_unref(src->spawn_livetv);

         src->spawn_livetv = NULL;

     }

     GST_ELEMENT_ERROR(src, LIBRARY, INIT,

                       (NULL),

                       ("Could not initialize MythTV library (%i, %s)", ret,

                        src->uri_name));

     gst_mythtv_src_clear(src);

     return FALSE;

   begin_req_failed:

     if (gmyth_uri != NULL) {

         g_object_unref(gmyth_uri);

         gmyth_uri = NULL;

     }

     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 = GST_MYTHTV_SRC(bsrc);

     gst_mythtv_src_clear(src);

     /*

      * 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 = TRUE;

     switch (GST_EVENT_TYPE(event)) {

     case GST_EVENT_EOS:

         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);

     }

     GST_DEBUG_OBJECT (src, "HANDLE EVENT %d", ret);

     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:

         {

             gst_query_parse_duration(query, &formt, NULL);

             if (formt == GST_FORMAT_BYTES) {

                 gint64          size = myth->content_size;

                 gst_query_set_duration(query, GST_FORMAT_BYTES, 10);

                 GST_DEBUG_OBJECT(myth, "SIZE %" G_GINT64_FORMAT, 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:

     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:

         gst_mythtv_src_clear(src);

         break;

     case GST_STATE_CHANGE_PLAYING_TO_PAUSED:

     case GST_STATE_CHANGE_PAUSED_TO_READY:

         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_name = g_value_dup_string(value);

             break;

         }

     case PROP_GMYTHTV_MAX_TRY:

         {

             mythtvsrc->max_try = 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:

         {

             g_value_set_string(value, mythtvsrc->uri_name);

             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:

         {

             g_value_set_string(value, mythtvsrc->live_chain_id);

             break;

         }

     case PROP_GMYTHTV_CHANNEL_NUM:

         {

             g_value_set_string(value, mythtvsrc->channel_name);

             break;

         }

     case PROP_GMYTHTV_MAX_TRY:

         {

             g_value_set_int(value, mythtvsrc->max_try);

             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);

 }