From d2891054094ec9e11c413757d8d7c2e95f9ead0d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Dr=C3=B6ge?= <sebastian.droege@collabora.co.uk>
Date: Fri, 4 Jun 2010 22:36:16 +0200
Subject: [PATCH] rtmpsrc: Add properties for setting the swfUrl and pageUrl
 properties

These are required for some streams unfortunately.
---
 ext/rtmp/gstrtmpsrc.c | 59 ++++++++++++++++++++++++++++++++++++++++---
 ext/rtmp/gstrtmpsrc.h |  2 ++
 2 files changed, 58 insertions(+), 3 deletions(-)

diff --git a/ext/rtmp/gstrtmpsrc.c b/ext/rtmp/gstrtmpsrc.c
index a8d559bb61..410237084a 100644
--- a/ext/rtmp/gstrtmpsrc.c
+++ b/ext/rtmp/gstrtmpsrc.c
@@ -65,8 +65,13 @@ enum
 {
   PROP_0,
   PROP_LOCATION,
+  PROP_SWF_URL,
+  PROP_PAGE_URL
 };
 
+static const AVal av_page_url = { (char *) "pageUrl", 7 };
+static const AVal av_swf_url = { (char *) "swfUrl", 6 };
+
 static void gst_rtmp_src_uri_handler_init (gpointer g_iface,
     gpointer iface_data);
 
@@ -132,6 +137,14 @@ gst_rtmp_src_class_init (GstRTMPSrcClass * klass)
   /* properties */
   gst_element_class_install_std_props (GST_ELEMENT_CLASS (klass),
       "location", PROP_LOCATION, G_PARAM_READWRITE, NULL);
+  g_object_class_install_property (gobject_class, PROP_SWF_URL,
+      g_param_spec_string ("swf-url", "SWF URL",
+          "URL of the corresponding SWF file",
+          NULL, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
+  g_object_class_install_property (gobject_class, PROP_PAGE_URL,
+      g_param_spec_string ("page-url", "Page URL",
+          "URL of the originating page",
+          NULL, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 
   gstbasesrc_class->start = GST_DEBUG_FUNCPTR (gst_rtmp_src_start);
   gstbasesrc_class->stop = GST_DEBUG_FUNCPTR (gst_rtmp_src_stop);
@@ -191,6 +204,11 @@ gst_rtmp_src_uri_get_uri (GstURIHandler * handler)
   return src->uri;
 }
 
+#define STR2AVAL(av,str) G_STMT_START { \
+  av.av_val = str; \
+  av.av_len = strlen(av.av_val); \
+} G_STMT_END;
+
 static gboolean
 gst_rtmp_src_uri_set_uri (GstURIHandler * handler, const gchar * uri)
 {
@@ -210,6 +228,7 @@ gst_rtmp_src_uri_set_uri (GstURIHandler * handler, const gchar * uri)
   }
 
   if (uri != NULL) {
+    AVal val;
 
     new_location = g_strdup (uri);
 
@@ -222,9 +241,19 @@ gst_rtmp_src_uri_set_uri (GstURIHandler * handler, const gchar * uri)
       RTMP_Free (src->rtmp);
       src->rtmp = NULL;
       return FALSE;
-    } else {
-      src->uri = g_strdup (uri);
-      GST_DEBUG_OBJECT (src, "parsed uri '%s' properly", src->uri);
+    }
+    src->uri = g_strdup (uri);
+    GST_DEBUG_OBJECT (src, "parsed uri '%s' properly", src->uri);
+
+
+    if (src->page_url) {
+      STR2AVAL (val, src->page_url);
+      RTMP_SetOpt (src->rtmp, &av_page_url, &val);
+    }
+
+    if (src->swf_url) {
+      STR2AVAL (val, src->swf_url);
+      RTMP_SetOpt (src->rtmp, &av_swf_url, &val);
     }
   }
 
@@ -258,6 +287,30 @@ gst_rtmp_src_set_property (GObject * object, guint prop_id,
           g_value_get_string (value));
       break;
     }
+    case PROP_SWF_URL:{
+      g_free (src->swf_url);
+      src->swf_url = g_value_dup_string (value);
+
+      if (src->rtmp && src->swf_url) {
+        AVal val;
+
+        STR2AVAL (val, src->swf_url);
+        RTMP_SetOpt (src->rtmp, &av_swf_url, &val);
+      }
+      break;
+    }
+    case PROP_PAGE_URL:{
+      g_free (src->page_url);
+      src->page_url = g_value_dup_string (value);
+
+      if (src->rtmp && src->page_url) {
+        AVal val;
+
+        STR2AVAL (val, src->page_url);
+        RTMP_SetOpt (src->rtmp, &av_page_url, &val);
+      }
+      break;
+    }
     default:
       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
       break;
diff --git a/ext/rtmp/gstrtmpsrc.h b/ext/rtmp/gstrtmpsrc.h
index 19d1710646..f7224b9db8 100644
--- a/ext/rtmp/gstrtmpsrc.h
+++ b/ext/rtmp/gstrtmpsrc.h
@@ -59,6 +59,8 @@ struct _GstRTMPSrc
   
   /* < private > */
   gchar *uri;
+  gchar *swf_url;
+  gchar *page_url;
 
   RTMP *rtmp;