Changeset 1436 for pjproject/trunk/pjmedia/src/pjmedia/transport_ice.c

Timestamp:

Sep 15, 2007 8:55:00 AM (17 years ago)

Author:

bennylp

Message:

Implemented ticket #373: Packet loss simulation in PJMEDIA ICE transport

File:

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

pjproject/trunk/pjmedia/src/pjmedia/transport_ice.c

@@ -21 +21 @@
 #include <pj/assert.h>
 #include <pj/log.h>
+#include <pj/rand.h>
+
 
 struct transport_ice
@@ -33 +35 @@
     pj_sockaddr_in       remote_rtp;
     pj_sockaddr_in       remote_rtcp;
+
+    unsigned             tx_drop_pct;   /**< Percent of tx pkts to drop.    */
+    unsigned             rx_drop_pct;   /**< Percent of rx pkts to drop.    */
 
     void               (*rtp_cb)(void*,
@@ -651 +656 @@
 {
     struct transport_ice *tp_ice = (struct transport_ice*)tp;
+
+    /* Simulate packet lost on TX direction */
+    if (tp_ice->tx_drop_pct) {
+        if ((pj_rand() % 100) <= (int)tp_ice->tx_drop_pct) {
+            PJ_LOG(5,(tp_ice->ice_st->obj_name, 
+                      "TX RTP packet dropped because of pkt lost "
+                      "simulation"));
+            return PJ_SUCCESS;
+        }
+    }
+
     return pj_ice_strans_sendto(tp_ice->ice_st, 1, 
                                 pkt, size, &tp_ice->remote_rtp,
@@ -679 +695 @@
     struct transport_ice *tp_ice = (struct transport_ice*) ice_st->user_data;
 
-    if (comp_id==1 && tp_ice->rtp_cb)
+    if (comp_id==1 && tp_ice->rtp_cb) {
+
+        /* Simulate packet lost on RX direction */
+        if (tp_ice->rx_drop_pct) {
+            if ((pj_rand() % 100) <= (int)tp_ice->rx_drop_pct) {
+                PJ_LOG(5,(ice_st->obj_name, 
+                          "RX RTP packet dropped because of pkt lost "
+                          "simulation"));
+                return;
+            }
+        }
+
         (*tp_ice->rtp_cb)(tp_ice->stream, pkt, size);
-    else if (comp_id==2 && tp_ice->rtcp_cb)
+
+    } else if (comp_id==2 && tp_ice->rtcp_cb)
         (*tp_ice->rtcp_cb)(tp_ice->stream, pkt, size);
 
@@ -734 +762 @@
 
 
+/* Simulate lost */
+PJ_DEF(pj_status_t) pjmedia_ice_simulate_lost( pjmedia_transport *tp,
+                                               pjmedia_dir dir,
+                                               unsigned pct_lost)
+{
+    struct transport_ice *ice = (struct transport_ice*) tp;
+
+    PJ_ASSERT_RETURN(tp && pct_lost <= 100, PJ_EINVAL);
+
+    if (dir & PJMEDIA_DIR_ENCODING)
+        ice->tx_drop_pct = pct_lost;
+
+    if (dir & PJMEDIA_DIR_DECODING)
+        ice->rx_drop_pct = pct_lost;
+
+    return PJ_SUCCESS;
+}
+