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

Timestamp:

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

Author:

riza

Message:

Close #1019: Support for multiple listeners.

File:

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

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

@@ -30 +30 @@
  */
 #if PJ_HAS_TCP
+
+static pj_status_t multi_listener_test(pjsip_tpfactory *factory[],
+                                       unsigned num_factory,
+                                       pjsip_transport *tp[],
+                                       unsigned *num_tp)
+{
+    pj_status_t status;
+    unsigned i = 0;
+    pj_str_t s;
+    pjsip_transport *tcp;
+    pjsip_tpfactory *tpfactory = NULL;
+    pj_sockaddr_in rem_addr;
+    pjsip_tpselector tp_sel;
+    unsigned ntp = 0;
+
+    for (;i<num_factory;++i)
+    {
+        /* Start TCP listener on arbitrary port. */
+        status = pjsip_tcp_transport_start(endpt, NULL, 1, &tpfactory);
+        if (status != PJ_SUCCESS) {
+            app_perror("   Error: unable to start TCP transport", status);
+            return -10;
+        }
+
+        factory[i] = tpfactory;
+    }
+
+    /* Get the last listener address */
+    status = pj_sockaddr_in_init(&rem_addr, &tpfactory->addr_name.host,
+                                 (pj_uint16_t)tpfactory->addr_name.port);
+    if (status != PJ_SUCCESS) {
+        app_perror("   Error: possibly invalid TCP address name", status);
+        return -11;
+    }
+
+    /* Acquire transport without selector. */
+    status = pjsip_endpt_acquire_transport(endpt, PJSIP_TRANSPORT_TCP,
+                                           &rem_addr, sizeof(rem_addr),
+                                           NULL, &tcp);
+    if (status != PJ_SUCCESS || tcp == NULL) {
+        app_perror("   Error: unable to acquire TCP transport", status);
+        return -12;
+    }
+    tp[ntp++] = tcp;
+
+    /* After pjsip_endpt_acquire_transport, TCP transport must have
+     * reference counter 1.
+     */
+    if (pj_atomic_get(tcp->ref_cnt) != 1)
+        return -13;
+
+    /* Acquire with the same remote address, should return the same tp. */
+    status = pjsip_endpt_acquire_transport(endpt, PJSIP_TRANSPORT_TCP,
+                                           &rem_addr, sizeof(rem_addr),
+                                           NULL, &tcp);
+    if (status != PJ_SUCCESS || tcp == NULL) {
+        app_perror("   Error: unable to acquire TCP transport", status);
+        return -14;
+    }
+
+    /* Should return existing transport. */
+    if (tp[ntp-1] != tcp) {
+        return -15;
+    }
+
+    /* Using the same TCP transport, it must have reference counter 2.
+     */
+    if (pj_atomic_get(tcp->ref_cnt) != 2)
+        return -16;
+
+    /* Decrease the reference. */
+    pjsip_transport_dec_ref(tcp);
+
+    /* Test basic transport attributes */
+    status = generic_transport_test(tcp);
+    if (status != PJ_SUCCESS)
+        return status;
+
+    /* Check again that reference counter is 1. */
+    if (pj_atomic_get(tcp->ref_cnt) != 1)
+        return -17;
+
+    /* Acquire transport test with selector. */
+    pj_bzero(&tp_sel, sizeof(tp_sel));
+    tp_sel.type = PJSIP_TPSELECTOR_LISTENER;
+    tp_sel.u.listener = factory[num_factory/2];
+    pj_sockaddr_in_init(&rem_addr, pj_cstr(&s, "1.1.1.1"), 80);
+    status = pjsip_endpt_acquire_transport(endpt, PJSIP_TRANSPORT_TCP,
+                                           &rem_addr, sizeof(rem_addr),
+                                           &tp_sel, &tcp);
+    if (status != PJ_SUCCESS) {
+        app_perror("   Error: unable to acquire TCP transport", status);
+        return -18;
+    }
+
+    /* The transport should have the same factory set on the selector. */
+    if (tcp->factory != factory[num_factory/2])
+        return -19;
+
+    /* The transport should be newly created. */
+    for (i = 0; i < ntp; ++i) {
+        if (tp[i] == tcp) {
+            break;
+        }
+    }
+    if (i != ntp)
+        return -20;
+
+    tp[ntp++] = tcp;
+
+    for (i = 0; i<ntp; ++i) {
+        if (pj_atomic_get(tp[i]->ref_cnt) != 1)
+            return -21;
+    }
+    *num_tp = ntp;
+
+    return PJ_SUCCESS;
+}
+
 int transport_tcp_test(void)
 {
     enum { SEND_RECV_LOOP = 8 };
-    pjsip_tpfactory *tpfactory;
-    pjsip_transport *tcp;
+    enum { NUM_LISTENER = 4 };
+    enum { NUM_TP = 8 };
+    pjsip_tpfactory *tpfactory[NUM_LISTENER];
+    pjsip_transport *tcp[NUM_TP];
     pj_sockaddr_in rem_addr;
     pj_status_t status;
@@ -40 +161 @@
     char addr[PJ_INET_ADDRSTRLEN];
     int rtt[SEND_RECV_LOOP], min_rtt;
-    int i, pkt_lost;
-
-    /* Start TCP listener on arbitrary port. */
-    status = pjsip_tcp_transport_start(endpt, NULL, 1, &tpfactory);
-    if (status != PJ_SUCCESS) {
-        app_perror("   Error: unable to start TCP transport", status);
-        return -10;
-    }
-
-
-    /* Get the listener address */
-    status = pj_sockaddr_in_init(&rem_addr, &tpfactory->addr_name.host,
-                                 (pj_uint16_t)tpfactory->addr_name.port);
-    if (status != PJ_SUCCESS) {
-        app_perror("   Error: possibly invalid TCP address name", status);
-        return -14;
-    }
+    int pkt_lost;
+    unsigned i;
+    unsigned num_listener = NUM_LISTENER;
+    unsigned num_tp = NUM_TP;
+
+    status = multi_listener_test(tpfactory, num_listener, tcp, &num_tp);
+    if (status != PJ_SUCCESS)
+        return status;
+
+    /* Get the last listener address */
+    status = pj_sockaddr_in_init(&rem_addr, &tpfactory[0]->addr_name.host,
+                                 (pj_uint16_t)tpfactory[0]->addr_name.port);
 
     pj_ansi_sprintf(url, "sip:alice@%s:%d;transport=tcp",
                     pj_inet_ntop2(pj_AF_INET(), &rem_addr.sin_addr, addr,
-                                  sizeof(addr)),
+                                  sizeof(addr)),
                     pj_ntohs(rem_addr.sin_port));
-
-
-    /* Acquire one TCP transport. */
-    status = pjsip_endpt_acquire_transport(endpt, PJSIP_TRANSPORT_TCP, 
-                                           &rem_addr, sizeof(rem_addr),
-                                           NULL, &tcp);
-    if (status != PJ_SUCCESS || tcp == NULL) {
-        app_perror("   Error: unable to acquire TCP transport", status);
-        return -17;
-    }
-
-    /* After pjsip_endpt_acquire_transport, TCP transport must have
-     * reference counter 1. 
-     */
-    if (pj_atomic_get(tcp->ref_cnt) != 1)
-        return -20;
-
-    /* Test basic transport attributes */
-    status = generic_transport_test(tcp);
-    if (status != PJ_SUCCESS)
-        return status;
-
-
-    /* Check again that reference counter is 1. */
-    if (pj_atomic_get(tcp->ref_cnt) != 1)
-        return -40;
 
     /* Load test */
@@ -95 +185 @@
     /* Basic transport's send/receive loopback test. */
     for (i=0; i<SEND_RECV_LOOP; ++i) {
-        status = transport_send_recv_test(PJSIP_TRANSPORT_TCP, tcp, url, &rtt[i]);
+        status = transport_send_recv_test(PJSIP_TRANSPORT_TCP, tcp[0], url,
+                                          &rtt[i]);
 
         if (status != 0) {
-            pjsip_transport_dec_ref(tcp);
+            for (i = 0; i < num_tp ; ++i) {
+                pjsip_transport_dec_ref(tcp[i]);
+            }
             flush_events(500);
             return -72;
@@ -116 +209 @@
 
     /* Multi-threaded round-trip test. */
-    status = transport_rt_test(PJSIP_TRANSPORT_TCP, tcp, url, &pkt_lost);
+    status = transport_rt_test(PJSIP_TRANSPORT_TCP, tcp[0], url, &pkt_lost);
     if (status != 0) {
-        pjsip_transport_dec_ref(tcp);
+        for (i = 0; i < num_tp ; ++i) {
+            pjsip_transport_dec_ref(tcp[i]);
+        }
         return status;
     }
@@ -126 +221 @@
 
     /* Check again that reference counter is still 1. */
-    if (pj_atomic_get(tcp->ref_cnt) != 1)
-        return -80;
-
-    /* Destroy this transport. */
-    pjsip_transport_dec_ref(tcp);
-
-    /* Force destroy this transport. */
-    status = pjsip_transport_destroy(tcp);
-    if (status != PJ_SUCCESS)
-        return -90;
-
-    /* Unregister factory */
-    status = pjsip_tpmgr_unregister_tpfactory(pjsip_endpt_get_tpmgr(endpt), 
-                                              tpfactory);
-    if (status != PJ_SUCCESS)
-        return -95;
+    for (i = 0; i < num_tp; ++i) {
+        if (pj_atomic_get(tcp[i]->ref_cnt) != 1)
+            return -80;
+    }
+
+    for (i = 0; i < num_tp; ++i) {
+        /* Destroy this transport. */
+        pjsip_transport_dec_ref(tcp[i]);
+
+        /* Force destroy this transport. */
+        status = pjsip_transport_destroy(tcp[i]);
+        if (status != PJ_SUCCESS)
+            return -90;
+    }
+
+    for (i = 0; i < num_listener; ++i) {
+        /* Unregister factory */
+        status = pjsip_tpmgr_unregister_tpfactory(pjsip_endpt_get_tpmgr(endpt),
+                                                  tpfactory[i]);
+        if (status != PJ_SUCCESS)
+            return -95;
+    }
 
     /* Flush events. */