Changeset 1790 for pjproject/trunk/pjmedia/src/pjmedia/transport_udp.c

Timestamp:

Feb 13, 2008 4:59:29 PM (16 years ago)

Author:

bennylp

Message:

Ticket #460: Concurrency problem when destroying stream (thanks Michael Broughton)

File:

• pjproject/trunk/pjmedia/src/pjmedia/transport_udp.c (modified) (5 diffs)

pjproject/trunk/pjmedia/src/pjmedia/transport_udp.c

@@ -320 +320 @@
         goto on_error;
     
+    /* Disallow concurrency so that detach() and destroy() are
+     * synchronized with the callback.
+     */
+    status = pj_ioqueue_set_concurrency(tp->rtp_key, PJ_FALSE);
+    if (status != PJ_SUCCESS)
+        goto on_error;
+
     pj_ioqueue_op_key_init(&tp->rtp_read_op, sizeof(tp->rtp_read_op));
     for (i=0; i<PJ_ARRAY_SIZE(tp->rtp_pending_write); ++i)
@@ -344 +351 @@
         goto on_error;
 
+    status = pj_ioqueue_set_concurrency(tp->rtcp_key, PJ_FALSE);
+    if (status != PJ_SUCCESS)
+        goto on_error;
+
     pj_ioqueue_op_key_init(&tp->rtcp_read_op, sizeof(tp->rtcp_read_op));
     pj_ioqueue_op_key_init(&tp->rtcp_write_op, sizeof(tp->rtcp_write_op));
@@ -384 +395 @@
 
     if (udp->rtp_key) {
+        /* This will block the execution if callback is still
+         * being called.
+         */
         pj_ioqueue_unregister(udp->rtp_key);
         udp->rtp_key = NULL;
@@ -646 +660 @@
 
     if (udp->attached) {
+        /* Lock the ioqueue keys to make sure that callbacks are
+         * not executed. See ticket #460 for details.
+         */
+        pj_ioqueue_lock_key(udp->rtp_key);
+        pj_ioqueue_lock_key(udp->rtcp_key);
+
         /* User data is unreferenced on Release build */
         PJ_UNUSED_ARG(user_data);
@@ -659 +679 @@
         udp->rtcp_cb = NULL;
         udp->user_data = NULL;
+
+        /* Unlock keys */
+        pj_ioqueue_unlock_key(udp->rtcp_key);
+        pj_ioqueue_unlock_key(udp->rtp_key);
     }
 }