Changeset 6002 for pjproject/trunk/pjsip/src/test/transport_udp_test.c

Timestamp:

May 23, 2019 5:21:59 PM (4 years ago)

Author:

riza

Message:

Close #1019: Support for multiple listeners.

File:

• pjproject/trunk/pjsip/src/test/transport_udp_test.c (modified) (5 diffs)

pjproject/trunk/pjsip/src/test/transport_udp_test.c

@@ -25 +25 @@
 #define THIS_FILE   "transport_udp_test.c"
 
+static pj_status_t multi_transport_test(pjsip_transport *tp[], unsigned num_tp)
+{
+    pj_status_t status;
+    pj_uint16_t i = 0;
+    pj_str_t s;
+    pjsip_transport *udp_tp;
+    pj_sockaddr_in rem_addr;    
+    pjsip_tpselector tp_sel;
+
+    for (;i<num_tp;++i)
+    {
+        pj_sockaddr_in addr;
+
+        pj_sockaddr_in_init(&addr, NULL, TEST_UDP_PORT+i);
+
+        /* Start UDP transport. */
+        status = pjsip_udp_transport_start( endpt, &addr, NULL, 1, &udp_tp);
+        if (status != PJ_SUCCESS) {
+            app_perror("   Error: unable to start UDP transport", status);
+            return -110;
+        }
+
+        /* UDP transport must have initial reference counter set to 1. */
+        if (pj_atomic_get(udp_tp->ref_cnt) != 1)
+            return -120;
+
+        /* Test basic transport attributes */
+        status = generic_transport_test(udp_tp);
+        if (status != PJ_SUCCESS)
+            return status;
+
+        tp[i] = udp_tp;
+    }
+
+    for (i = 0; i < num_tp; ++i) {
+        udp_tp = tp[i];
+        if (pj_atomic_get(udp_tp->ref_cnt) != 1)
+            return -130;
+    }
+
+    /* Acquire transport test without selector. */
+    pj_sockaddr_in_init(&rem_addr, pj_cstr(&s, "1.1.1.1"), 80);
+    status = pjsip_endpt_acquire_transport(endpt, PJSIP_TRANSPORT_UDP,
+                                           &rem_addr, sizeof(rem_addr),
+                                           NULL, &udp_tp);
+    if (status != PJ_SUCCESS)
+        return -140;
+
+    for (i = 0; i < num_tp; ++i) {
+        if (udp_tp == tp[i]) {
+            break;
+        }
+    }
+    if (i == num_tp)
+        return -150;
+
+    pjsip_transport_dec_ref(udp_tp);
+
+    if (pj_atomic_get(udp_tp->ref_cnt) != 1)
+        return -160;
+
+    /* Acquire transport test with selector. */
+    pj_bzero(&tp_sel, sizeof(tp_sel));
+    tp_sel.type = PJSIP_TPSELECTOR_TRANSPORT;
+    tp_sel.u.transport = tp[num_tp-1];
+    pj_sockaddr_in_init(&rem_addr, pj_cstr(&s, "1.1.1.1"), 80);
+    status = pjsip_endpt_acquire_transport(endpt, PJSIP_TRANSPORT_UDP,
+                                           &rem_addr, sizeof(rem_addr),
+                                           &tp_sel, &udp_tp);
+    if (status != PJ_SUCCESS)
+        return -170;
+
+    if (udp_tp != tp[num_tp-1])
+        return -180;
+
+    pjsip_transport_dec_ref(udp_tp);
+
+    if (pj_atomic_get(udp_tp->ref_cnt) != 1)
+        return -190;
+
+    return PJ_SUCCESS;
+}
 
 /*
@@ -32 +114 @@
 {
     enum { SEND_RECV_LOOP = 8 };
-    pjsip_transport *udp_tp, *tp;
-    pj_sockaddr_in addr, rem_addr;
+    enum { NUM_TP = 4 };
+    pjsip_transport *tp[NUM_TP], *udp_tp;
+    pj_sockaddr_in rem_addr;
     pj_str_t s;
     pj_status_t status;
@@ -39 +122 @@
     int i, pkt_lost;
 
-    pj_sockaddr_in_init(&addr, NULL, TEST_UDP_PORT);
-
-    /* Start UDP transport. */
-    status = pjsip_udp_transport_start( endpt, &addr, NULL, 1, &udp_tp);
-    if (status != PJ_SUCCESS) {
-        app_perror("   Error: unable to start UDP transport", status);
-        return -10;
-    }
-
-    /* UDP transport must have initial reference counter set to 1. */
-    if (pj_atomic_get(udp_tp->ref_cnt) != 1)
-        return -20;
-
-    /* Test basic transport attributes */
-    status = generic_transport_test(udp_tp);
+    status = multi_transport_test(&tp[0], NUM_TP);
     if (status != PJ_SUCCESS)
         return status;
-
-    /* Test that transport manager is returning the correct
-     * transport.
-     */
-    pj_sockaddr_in_init(&rem_addr, pj_cstr(&s, "1.1.1.1"), 80);
-    status = pjsip_endpt_acquire_transport(endpt, PJSIP_TRANSPORT_UDP, 
-                                           &rem_addr, sizeof(rem_addr),
-                                           NULL, &tp);
-    if (status != PJ_SUCCESS)
-        return -50;
-    if (tp != udp_tp)
-        return -60;
-
-    /* pjsip_endpt_acquire_transport() adds reference, so we need
-     * to decrement it.
-     */
-    pjsip_transport_dec_ref(tp);
-
-    /* Check again that reference counter is 1. */
-    if (pj_atomic_get(udp_tp->ref_cnt) != 1)
-        return -70;
 
     /* Basic transport's send/receive loopback test. */
     pj_sockaddr_in_init(&rem_addr, pj_cstr(&s, "127.0.0.1"), TEST_UDP_PORT);
     for (i=0; i<SEND_RECV_LOOP; ++i) {
-        status = transport_send_recv_test(PJSIP_TRANSPORT_UDP, tp, 
+        status = transport_send_recv_test(PJSIP_TRANSPORT_UDP, tp[0], 
                                           "sip:alice@127.0.0.1:"TEST_UDP_PORT_STR,
                                           &rtt[i]);
@@ -99 +147 @@
 
     /* Multi-threaded round-trip test. */
-    status = transport_rt_test(PJSIP_TRANSPORT_UDP, tp, 
+    status = transport_rt_test(PJSIP_TRANSPORT_UDP, tp[0], 
                                "sip:alice@127.0.0.1:"TEST_UDP_PORT_STR, 
                                &pkt_lost);
@@ -108 +156 @@
         PJ_LOG(3,(THIS_FILE, "   note: %d packet(s) was lost", pkt_lost));
 
-    /* Check again that reference counter is 1. */
-    if (pj_atomic_get(udp_tp->ref_cnt) != 1)
-        return -80;
+    for (i = 0; i < NUM_TP; ++i) {
+        udp_tp = tp[i];
 
-    /* Destroy this transport. */
-    pjsip_transport_dec_ref(udp_tp);
+        /* Check again that reference counter is 1. */
+        if (pj_atomic_get(udp_tp->ref_cnt) != 1)
+            return -80;
 
-    /* Force destroy this transport. */
-    status = pjsip_transport_destroy(udp_tp);
-    if (status != PJ_SUCCESS)
-        return -90;
+        /* Destroy this transport. */
+        pjsip_transport_dec_ref(udp_tp);
+        status = pjsip_transport_destroy(udp_tp);
+        if (status != PJ_SUCCESS)
+            return -90;
+    }
 
     /* Flush events. */