Changeset 5513 for pjproject/branches/projects/uwp/pjsip/src/pjsip/sip_transport.c

Timestamp:

Dec 28, 2016 3:40:07 AM (7 years ago)

Author:

nanang

Message:

Re #1900: More merged from trunk (r5512 mistakenly contains merged changes in third-party dir only).

Location:

pjproject/branches/projects/uwp

Files:

• . (modified) (1 prop)
• pjsip/src/pjsip/sip_transport.c (modified) (7 diffs)

pjproject/branches/projects/uwp/pjsip/src/pjsip/sip_transport.c

@@ -492 +492 @@
 PJ_DEF(pj_status_t) pjsip_tx_data_dec_ref( pjsip_tx_data *tdata )
 {
-    pj_assert( pj_atomic_get(tdata->ref_cnt) > 0);
-    if (pj_atomic_dec_and_get(tdata->ref_cnt) <= 0) {
+    pj_atomic_value_t ref_cnt;
+    
+    PJ_ASSERT_RETURN(tdata && tdata->ref_cnt, PJ_EINVAL);
+
+    ref_cnt = pj_atomic_dec_and_get(tdata->ref_cnt);
+    pj_assert( ref_cnt >= 0);
+    if (ref_cnt == 0) {
         tx_data_destroy(tdata);
         return PJSIP_EBUFDESTROYED;
@@ -1294 +1299 @@
 {
     pj_bzero(prm, sizeof(*prm));
+}
+
+static pj_bool_t pjsip_tpmgr_is_tpfactory_valid(pjsip_tpmgr *mgr,
+                                                pjsip_tpfactory *tpf)
+{
+    pjsip_tpfactory *p;
+
+    pj_lock_acquire(mgr->lock);
+    for (p=mgr->factory_list.next; p!=&mgr->factory_list; p=p->next) {
+        if (p == tpf) {
+            pj_lock_release(mgr->lock);
+            return PJ_TRUE;
+        }
+    }
+    pj_lock_release(mgr->lock);
+
+    return PJ_FALSE;
 }
 
@@ -2001 +2023 @@
         return PJ_SUCCESS;
 
-
-    } else if (sel && sel->type == PJSIP_TPSELECTOR_LISTENER &&
-               sel->u.listener)
-    {
-        /* Application has requested that a specific listener is to
-         * be used. In this case, skip transport hash table lookup.
-         */
-
-        /* Verify that the listener type matches the destination type */
-        if (sel->u.listener->type != type) {
-            pj_lock_release(mgr->lock);
-            return PJSIP_ETPNOTSUITABLE;
-        }
-
-        /* We'll use this listener to create transport */
-        factory = sel->u.listener;
-
     } else {
 
         /*
          * This is the "normal" flow, where application doesn't specify
-         * specific transport/listener to be used to send message to.
+         * specific transport to be used to send message to.
          * In this case, lookup the transport from the hash table.
          */
@@ -2028 +2033 @@
         int key_len;
         pjsip_transport *transport;
+
+        /* If listener is specified, verify that the listener type matches
+         * the destination type.
+         */
+        if (sel && sel->type == PJSIP_TPSELECTOR_LISTENER && sel->u.listener)
+        {
+            if (sel->u.listener->type != type) {
+                pj_lock_release(mgr->lock);
+                return PJSIP_ETPNOTSUITABLE;
+            }
+        }
 
         pj_bzero(&key, sizeof(key));
@@ -2070 +2086 @@
         }
 
+        /* If transport is found and listener is specified, verify listener */
+        else if (sel && sel->type == PJSIP_TPSELECTOR_LISTENER &&
+                 sel->u.listener && transport->factory != sel->u.listener)
+        {
+            transport = NULL;
+            /* This will cause a new transport to be created which will be a
+             * 'duplicate' of the existing transport (same type & remote addr,
+             * but different factory). Any future hash lookup will return
+             * the new one, and eventually the old one will still be freed
+             * (by application or #1774).
+             */
+        }
+
         if (transport!=NULL && !transport->is_shutdown) {
             /*
@@ -2082 +2111 @@
         }
 
+
         /*
          * Transport not found!
-         * Find factory that can create such transport.
+         * So we need to create one, find factory that can create
+         * such transport.
          */
-        factory = mgr->factory_list.next;
-        while (factory != &mgr->factory_list) {
-            if (factory->type == type)
-                break;
-            factory = factory->next;
-        }
-
-        if (factory == &mgr->factory_list) {
-            /* No factory can create the transport! */
-            pj_lock_release(mgr->lock);
-            TRACE_((THIS_FILE, "No suitable factory was found either"));
-            return PJSIP_EUNSUPTRANSPORT;
+        if (sel && sel->type == PJSIP_TPSELECTOR_LISTENER && sel->u.listener)
+        {
+            /* Application has requested that a specific listener is to
+             * be used.
+             */
+
+            /* Verify that the listener type matches the destination type */
+            if (sel->u.listener->type != type) {
+                pj_lock_release(mgr->lock);
+                return PJSIP_ETPNOTSUITABLE;
+            }
+
+            /* We'll use this listener to create transport */
+            factory = sel->u.listener;
+
+            /* Verify if listener is still valid */
+            if (!pjsip_tpmgr_is_tpfactory_valid(mgr, factory)) {
+                pj_lock_release(mgr->lock);
+                PJ_LOG(3,(THIS_FILE, "Specified factory for creating "
+                                     "transport is not found"));
+                return PJ_ENOTFOUND;
+            }
+
+        } else {
+
+            /* Find factory with type matches the destination type */
+            factory = mgr->factory_list.next;
+            while (factory != &mgr->factory_list) {
+                if (factory->type == type)
+                    break;
+                factory = factory->next;
+            }
+
+            if (factory == &mgr->factory_list) {
+                /* No factory can create the transport! */
+                pj_lock_release(mgr->lock);
+                TRACE_((THIS_FILE, "No suitable factory was found either"));
+                return PJSIP_EUNSUPTRANSPORT;
+            }
         }
     }
@@ -2117 +2175 @@
             {pj_lock_release(mgr->lock); return PJ_EBUG;});
         pjsip_transport_add_ref(*tp);
+        (*tp)->factory = factory;
     }
     pj_lock_release(mgr->lock);