client: make sure the watch exists while sending data

Protect the send_func with a lock. This allows us to wait for sending
to complete before changing the send_func and user_data. We add an
extra ref to the watch to make sure that it remains valid during
sending.
When closing the connection, set the send_func to NULL

Fixes https://bugzilla.gnome.org/show_bug.cgi?id=692433
This commit is contained in:
Wim Taymans 2013-01-28 10:31:50 +01:00
parent 114442bdb8
commit 6db0dbc76c

View file

@ -30,6 +30,7 @@
struct _GstRTSPClientPrivate struct _GstRTSPClientPrivate
{ {
GMutex lock; GMutex lock;
GMutex send_lock;
GstRTSPConnection *connection; GstRTSPConnection *connection;
GstRTSPWatch *watch; GstRTSPWatch *watch;
guint close_seq; guint close_seq;
@ -208,6 +209,7 @@ gst_rtsp_client_init (GstRTSPClient * client)
client->priv = priv; client->priv = priv;
g_mutex_init (&priv->lock); g_mutex_init (&priv->lock);
g_mutex_init (&priv->send_lock);
priv->use_client_settings = DEFAULT_USE_CLIENT_SETTINGS; priv->use_client_settings = DEFAULT_USE_CLIENT_SETTINGS;
priv->close_seq = 0; priv->close_seq = 0;
} }
@ -258,12 +260,11 @@ gst_rtsp_client_finalize (GObject * obj)
GST_INFO ("finalize client %p", client); GST_INFO ("finalize client %p", client);
gst_rtsp_client_set_send_func (client, NULL, NULL, NULL);
if (priv->watch) if (priv->watch)
g_source_destroy ((GSource *) priv->watch); g_source_destroy ((GSource *) priv->watch);
if (priv->send_notify)
priv->send_notify (priv->send_data);
client_cleanup_sessions (client); client_cleanup_sessions (client);
if (priv->connection) if (priv->connection)
@ -284,6 +285,7 @@ gst_rtsp_client_finalize (GObject * obj)
g_free (priv->server_ip); g_free (priv->server_ip);
g_mutex_clear (&priv->lock); g_mutex_clear (&priv->lock);
g_mutex_clear (&priv->send_lock);
G_OBJECT_CLASS (gst_rtsp_client_parent_class)->finalize (obj); G_OBJECT_CLASS (gst_rtsp_client_parent_class)->finalize (obj);
} }
@ -374,8 +376,10 @@ send_response (GstRTSPClient * client, GstRTSPSession * session,
if (close) if (close)
gst_rtsp_message_add_header (response, GST_RTSP_HDR_CONNECTION, "close"); gst_rtsp_message_add_header (response, GST_RTSP_HDR_CONNECTION, "close");
g_mutex_lock (&priv->send_lock);
if (priv->send_func) if (priv->send_func)
priv->send_func (client, response, close, priv->send_data); priv->send_func (client, response, close, priv->send_data);
g_mutex_unlock (&priv->send_lock);
gst_rtsp_message_unset (response); gst_rtsp_message_unset (response);
} }
@ -543,8 +547,10 @@ do_send_data (GstBuffer * buffer, guint8 channel, GstRTSPClient * client)
gst_rtsp_message_take_body (&message, map_info.data, map_info.size); gst_rtsp_message_take_body (&message, map_info.data, map_info.size);
g_mutex_lock (&priv->send_lock);
if (priv->send_func) if (priv->send_func)
priv->send_func (client, &message, FALSE, priv->send_data); priv->send_func (client, &message, FALSE, priv->send_data);
g_mutex_unlock (&priv->send_lock);
gst_rtsp_message_steal_body (&message, &data, &usize); gst_rtsp_message_steal_body (&message, &data, &usize);
gst_buffer_unmap (buffer, &map_info); gst_buffer_unmap (buffer, &map_info);
@ -2026,13 +2032,13 @@ gst_rtsp_client_set_send_func (GstRTSPClient * client,
priv = client->priv; priv = client->priv;
g_mutex_lock (&priv->lock); g_mutex_lock (&priv->send_lock);
priv->send_func = func; priv->send_func = func;
old_notify = priv->send_notify; old_notify = priv->send_notify;
old_data = priv->send_data; old_data = priv->send_data;
priv->send_notify = notify; priv->send_notify = notify;
priv->send_data = user_data; priv->send_data = user_data;
g_mutex_unlock (&priv->lock); g_mutex_unlock (&priv->send_lock);
if (old_notify) if (old_notify)
old_notify (old_data); old_notify (old_data);
@ -2118,6 +2124,8 @@ closed (GstRTSPWatch * watch, gpointer user_data)
g_mutex_unlock (&tunnels_lock); g_mutex_unlock (&tunnels_lock);
} }
gst_rtsp_client_set_send_func (client, NULL, NULL, NULL);
return GST_RTSP_OK; return GST_RTSP_OK;
} }
@ -2461,7 +2469,8 @@ gst_rtsp_client_attach (GstRTSPClient * client, GMainContext * context)
/* create watch for the connection and attach */ /* create watch for the connection and attach */
priv->watch = gst_rtsp_watch_new (priv->connection, &watch_funcs, priv->watch = gst_rtsp_watch_new (priv->connection, &watch_funcs,
g_object_ref (client), (GDestroyNotify) client_watch_notify); g_object_ref (client), (GDestroyNotify) client_watch_notify);
gst_rtsp_client_set_send_func (client, do_send_message, NULL, NULL); gst_rtsp_client_set_send_func (client, do_send_message, priv->watch,
(GDestroyNotify) gst_rtsp_watch_unref);
/* FIXME make this configurable. We don't want to do this yet because it will /* FIXME make this configurable. We don't want to do this yet because it will
* be superceeded by a cache object later */ * be superceeded by a cache object later */
@ -2469,7 +2478,6 @@ gst_rtsp_client_attach (GstRTSPClient * client, GMainContext * context)
GST_INFO ("attaching to context %p", context); GST_INFO ("attaching to context %p", context);
res = gst_rtsp_watch_attach (priv->watch, context); res = gst_rtsp_watch_attach (priv->watch, context);
gst_rtsp_watch_unref (priv->watch);
return res; return res;
} }