Changeset 2242 for pjproject

Timestamp:

Aug 26, 2008 4:59:10 PM (16 years ago)

Author:

nanang

Message:

Ticket #600: Added probation state in media transport SRTP and also capability to auto-restart when packets received in probation state are 'invalid'.

File:

• pjproject/trunk/pjmedia/src/pjmedia/transport_srtp.c (modified) (6 diffs)

pjproject/trunk/pjmedia/src/pjmedia/transport_srtp.c

@@ -37 +37 @@
 #define MAX_RTCP_BUFFER_LEN         1500
 #define MAX_KEY_LEN                 32
+
+/* Initial value of probation counter. When probation counter > 0, 
+ * it means SRTP is in probation state, and it may restart when
+ * srtp_unprotect() returns err_status_replay_*
+ */
+#define PROBATION_CNT_INIT          100
+
 #define DEACTIVATE_MEDIA(pool, m)   pjmedia_sdp_media_deactivate(pool, m)
 
@@ -77 +84 @@
 typedef struct transport_srtp
 {
-    pjmedia_transport    base;      /**< Base transport interface. */
-    pj_pool_t           *pool;
-    pj_lock_t           *mutex;
+    pjmedia_transport    base;              /**< Base transport interface.  */
+    pj_pool_t           *pool;              /**< Pool for transport SRTP.   */
+    pj_lock_t           *mutex;             /**< Mutex for libsrtp contexts.*/
     char                 rtp_tx_buffer[MAX_RTP_BUFFER_LEN];
     char                 rtcp_tx_buffer[MAX_RTCP_BUFFER_LEN];
@@ -121 +128 @@
     pjmedia_srtp_use     peer_use;
 
+    /* When probation counter > 0, it means SRTP is in probation state, 
+     * and it may restart when srtp_unprotect() returns err_status_replay_*
+     */
+    unsigned             probation_cnt;
 } transport_srtp;
 
@@ -350 +361 @@
     srtp->session_inited = PJ_FALSE;
     srtp->bypass_srtp = PJ_FALSE;
+    srtp->probation_cnt = PROBATION_CNT_INIT;
     srtp->peer_use = opt->use;
 
@@ -776 +788 @@
     PJ_ASSERT_ON_FAIL( (((long)pkt) & 0x03)==0, return );
 
+    if (srtp->probation_cnt > 0)
+        --srtp->probation_cnt;
+
     pj_lock_acquire(srtp->mutex);
 
     err = srtp_unprotect(srtp->srtp_rx_ctx, (pj_uint8_t*)pkt, &len);
-    
+
+    if (srtp->probation_cnt > 0 && 
+        (err == err_status_replay_old || err == err_status_replay_fail)) 
+    {
+        /* Handle such condition that stream is updated (RTP seq is reinited
+        * & SRTP is restarted), but some old packets are still coming 
+        * so SRTP is learning wrong RTP seq. While the newly inited RTP seq
+        * comes, SRTP thinks the RTP seq is replayed, so srtp_unprotect() 
+        * will returning err_status_replay_*. Restarting SRTP can resolve 
+        * this.
+        */
+        if (pjmedia_transport_srtp_start((pjmedia_transport*)srtp, 
+                                         &srtp->tx_policy, &srtp->rx_policy) 
+                                         != PJ_SUCCESS)
+        {
+            PJ_LOG(5,(srtp->pool->obj_name, "Failed to restart SRTP, err=%s", 
+                      get_libsrtp_errstr(err)));
+        } else {
+            err = srtp_unprotect(srtp->srtp_rx_ctx, (pj_uint8_t*)pkt, &len);
+        }
+    }
+
     if (err == err_status_ok) {
         srtp->rtp_cb(srtp->user_data, pkt, len);
@@ -1379 +1415 @@
     }
 
+    /* Reset probation counts */
+    srtp->probation_cnt = PROBATION_CNT_INIT;
+
     /* Got policy_local & policy_remote, let's initalize the SRTP */
     status = pjmedia_transport_srtp_start(tp, &srtp->tx_policy_neg, &srtp->rx_policy_neg);