Changeset 1444 for pjproject/trunk/pjmedia/src/pjmedia/rtcp.c

Timestamp:

Sep 20, 2007 11:30:59 AM (17 years ago)

Author:

bennylp

Message:

Ticket #377: Support for sending RTCP RR

File:

• pjproject/trunk/pjmedia/src/pjmedia/rtcp.c (modified) (9 diffs)

pjproject/trunk/pjmedia/src/pjmedia/rtcp.c

@@ -131 +131 @@
                                pj_uint32_t ssrc)
 {
-    pjmedia_rtcp_pkt *rtcp_pkt = &sess->rtcp_pkt;
+    pjmedia_rtcp_sr_pkt *sr_pkt = &sess->rtcp_sr_pkt;
     pj_time_val now;
     
@@ -147 +147 @@
     sess->pkt_size = samples_per_frame;
 
-    /* Init common RTCP header */
-    rtcp_pkt->common.version = 2;
-    rtcp_pkt->common.count = 1;
-    rtcp_pkt->common.pt = RTCP_SR;
-    rtcp_pkt->common.length = pj_htons(12);
-    
-    /* Init SR */
-    rtcp_pkt->sr.ssrc = pj_htonl(ssrc);
-    
+    /* Init common RTCP SR header */
+    sr_pkt->common.version = 2;
+    sr_pkt->common.count = 1;
+    sr_pkt->common.pt = RTCP_SR;
+    sr_pkt->common.length = pj_htons(12);
+    sr_pkt->common.ssrc = pj_htonl(ssrc);
+    
+    /* Copy to RTCP RR header */
+    pj_memcpy(&sess->rtcp_rr_pkt.common, &sr_pkt->common, 
+              sizeof(pjmedia_rtcp_common));
+    sess->rtcp_rr_pkt.common.pt = RTCP_RR;
+    sess->rtcp_rr_pkt.common.length = pj_htons(7);
+
     /* Get time and timestamp base and frequency */
     pj_gettimeofday(&now);
@@ -341 +345 @@
     if (common->pt == RTCP_SR) {
         sr = (pjmedia_rtcp_sr*) (((char*)pkt) + sizeof(pjmedia_rtcp_common));
-        if (common->count > 0 && size >= (sizeof(pjmedia_rtcp_pkt))) {
+        if (common->count > 0 && size >= (sizeof(pjmedia_rtcp_sr_pkt))) {
             rr = (pjmedia_rtcp_rr*)(((char*)pkt) + (sizeof(pjmedia_rtcp_common)
                                     + sizeof(pjmedia_rtcp_sr)));
@@ -530 +534 @@
 
 PJ_DEF(void) pjmedia_rtcp_build_rtcp(pjmedia_rtcp_session *sess, 
-                                     pjmedia_rtcp_pkt **ret_p_pkt, 
-                                     int *len)
+                                     void **ret_p_pkt, int *len)
 {
     pj_uint32_t expected, expected_interval, received_interval, lost_interval;
-    pjmedia_rtcp_pkt *rtcp_pkt = &sess->rtcp_pkt;
+    pjmedia_rtcp_common *common;
+    pjmedia_rtcp_sr *sr;
+    pjmedia_rtcp_rr *rr;
     pj_timestamp ts_now;
     pjmedia_rtcp_ntp_rec ntp;
-    
-    /* Packet count */
-    rtcp_pkt->sr.sender_pcount = pj_htonl(sess->stat.tx.pkt);
-
-    /* Octets count */
-    rtcp_pkt->sr.sender_bcount = pj_htonl(sess->stat.tx.bytes);
+
+    /* Get current NTP time. */
+    pj_get_timestamp(&ts_now);
+    pjmedia_rtcp_get_ntp_time(sess, &ntp);
+
+
+    /* See if we have transmitted RTP packets since last time we
+     * sent RTCP SR.
+     */
+    if (sess->stat.tx.pkt != pj_ntohl(sess->rtcp_sr_pkt.sr.sender_pcount)) {
+
+        /* So we should send RTCP SR */
+        *ret_p_pkt = (void*) &sess->rtcp_sr_pkt;
+        *len = sizeof(pjmedia_rtcp_sr_pkt);
+        common = &sess->rtcp_sr_pkt.common;
+        rr = &sess->rtcp_sr_pkt.rr;
+        sr = &sess->rtcp_sr_pkt.sr;
+
+        /* Update packet count */
+        sr->sender_pcount = pj_htonl(sess->stat.tx.pkt);
+
+        /* Update octets count */
+        sr->sender_bcount = pj_htonl(sess->stat.tx.bytes);
+
+        /* Fill in NTP timestamp in SR. */
+        sr->ntp_sec = pj_htonl(ntp.hi);
+        sr->ntp_frac = pj_htonl(ntp.lo);
+
+        TRACE_((sess->name, "TX RTCP SR: ntp_ts=%p", 
+                           ((ntp.hi & 0xFFFF) << 16) + ((ntp.lo & 0xFFFF0000) 
+                                >> 16)));
+
+
+    } else {
+        /* We should send RTCP RR then */
+        *ret_p_pkt = (void*) &sess->rtcp_rr_pkt;
+        *len = sizeof(pjmedia_rtcp_rr_pkt);
+        common = &sess->rtcp_rr_pkt.common;
+        rr = &sess->rtcp_rr_pkt.rr;
+        sr = NULL;
+    }
     
     /* SSRC and last_seq */
-    rtcp_pkt->rr.ssrc = pj_htonl(sess->peer_ssrc);
-    rtcp_pkt->rr.last_seq = (sess->seq_ctrl.cycles & 0xFFFF0000L);
-    rtcp_pkt->rr.last_seq += sess->seq_ctrl.max_seq;
-    rtcp_pkt->rr.last_seq = pj_htonl(rtcp_pkt->rr.last_seq);
+    rr->ssrc = pj_htonl(sess->peer_ssrc);
+    rr->last_seq = (sess->seq_ctrl.cycles & 0xFFFF0000L);
+    /* Since this is an "+=" operation, make sure we update last_seq on
+     * both RR and SR.
+     */
+    sess->rtcp_sr_pkt.rr.last_seq += sess->seq_ctrl.max_seq;
+    sess->rtcp_rr_pkt.rr.last_seq += sess->seq_ctrl.max_seq;
+    rr->last_seq = pj_htonl(rr->last_seq);
 
 
     /* Jitter */
-    rtcp_pkt->rr.jitter = pj_htonl(sess->jitter >> 4);
+    rr->jitter = pj_htonl(sess->jitter >> 4);
     
     
     /* Total lost. */
-    expected = pj_ntohl(rtcp_pkt->rr.last_seq) - sess->seq_ctrl.base_seq;
+    expected = pj_ntohl(rr->last_seq) - sess->seq_ctrl.base_seq;
 
     /* This is bug: total lost already calculated on each incoming RTP!
@@ -565 +609 @@
     */
 
-    rtcp_pkt->rr.total_lost_2 = (sess->stat.rx.loss >> 16) & 0xFF;
-    rtcp_pkt->rr.total_lost_1 = (sess->stat.rx.loss >> 8) & 0xFF;
-    rtcp_pkt->rr.total_lost_0 = (sess->stat.rx.loss & 0xFF);
+    rr->total_lost_2 = (sess->stat.rx.loss >> 16) & 0xFF;
+    rr->total_lost_1 = (sess->stat.rx.loss >> 8) & 0xFF;
+    rr->total_lost_0 = (sess->stat.rx.loss & 0xFF);
 
     /* Fraction lost calculation */
@@ -579 +623 @@
     
     if (expected_interval==0 || lost_interval == 0) {
-        rtcp_pkt->rr.fract_lost = 0;
+        rr->fract_lost = 0;
     } else {
-        rtcp_pkt->rr.fract_lost = (lost_interval << 8) / expected_interval;
-    }
-    
-    /* Get current NTP time. */
-    pj_get_timestamp(&ts_now);
-    pjmedia_rtcp_get_ntp_time(sess, &ntp);
-    
-    /* Fill in NTP timestamp in SR. */
-    rtcp_pkt->sr.ntp_sec = pj_htonl(ntp.hi);
-    rtcp_pkt->sr.ntp_frac = pj_htonl(ntp.lo);
-
-    TRACE_((sess->name, "TX RTCP SR: ntp_ts=%p", 
-                       ((ntp.hi & 0xFFFF) << 16) + ((ntp.lo & 0xFFFF0000) 
-                            >> 16)));
-
+        rr->fract_lost = (lost_interval << 8) / expected_interval;
+    }
+    
     if (sess->rx_lsr_time.u64 == 0 || sess->rx_lsr == 0) {
-        rtcp_pkt->rr.lsr = 0;
-        rtcp_pkt->rr.dlsr = 0;
+        rr->lsr = 0;
+        rr->dlsr = 0;
     } else {
         pj_timestamp ts;
@@ -611 +643 @@
            LSR is the middle 32bit of the last SR NTP time received.
          */
-        rtcp_pkt->rr.lsr = pj_htonl(lsr);
+        rr->lsr = pj_htonl(lsr);
         
         /* Fill in DLSR.
@@ -623 +655 @@
         /* Get DLSR */
         dlsr = (pj_uint32_t)(ts.u64 - lsr_time);
-        rtcp_pkt->rr.dlsr = pj_htonl(dlsr);
+        rr->dlsr = pj_htonl(dlsr);
 
         TRACE_((sess->name,"Tx RTCP RR: lsr=%p, lsr_time=%p, now=%p, dlsr=%p"
@@ -638 +670 @@
     pj_gettimeofday(&sess->stat.rx.update);
     sess->stat.rx.update_cnt++;
-
-
-    /* Return pointer. */
-    *ret_p_pkt = rtcp_pkt;
-    *len = sizeof(pjmedia_rtcp_pkt);
 }