Ticket #464: jbuf.patch

pjmedia/include/pjmedia/jbuf.h

@@ -235 +235 @@
                                       void *frame, 
                                       char *p_frm_type);
 
+/**
+ * Get a frame from the jitter buffer. The jitter buffer will return the
+ * oldest frame from it's buffer, when it is available.
+ *
+ * @param jb            The jitter buffer.
+ * @param frame         Buffer to receive the payload from the jitter buffer.
+ *                      @see pjmedia_jbuf_get_frame().    
+ * @param size          Pointer to receive frame size.
+ * @param p_frm_type    Pointer to receive frame type.
+ *                      @see pjmedia_jbuf_get_frame().    
+ */
+PJ_DECL(void) pjmedia_jbuf_get_frame2(pjmedia_jbuf *jb, 
+                                      void *frame, 
+                                      pj_size_t *size, 
+                                      char *p_frm_type);
 
+
 /**
  * Get jitter buffer current state/settings.
  *

pjmedia/src/pjmedia/jbuf.c

@@ -34 +34 @@
 {
     char        *flist_buffer;
     int         *flist_frame_type;
+    pj_size_t   *flist_content_len;
     unsigned     flist_frame_size;
     unsigned     flist_max_count;
     unsigned     flist_empty;
@@ -109 +110 @@
         pj_pool_zalloc(pool, sizeof(framelist->flist_frame_type[0]) * 
                                 framelist->flist_max_count);
 
+    framelist->flist_content_len = (pj_size_t*)
+        pj_pool_zalloc(pool, sizeof(framelist->flist_content_len[0]) * 
+                                framelist->flist_max_count);
+
     framelist->flist_empty = 1;
 
     return PJ_SUCCESS;
@@ -134 +139 @@
 
 
 static pj_bool_t jb_framelist_get(jb_framelist_t *framelist,
-                                  void *frame,
+                                  void *frame, pj_size_t *size,
                                   pjmedia_jb_frame_type *p_type) 
 {
     if (!framelist->flist_empty) {
@@ -144 +149 @@
                   framelist->flist_frame_size);
         *p_type = (pjmedia_jb_frame_type) 
                   framelist->flist_frame_type[framelist->flist_head];
+        *size   = framelist->flist_content_len[framelist->flist_head];
 
         pj_bzero(framelist->flist_buffer + 
                     framelist->flist_head * framelist->flist_frame_size,
                   framelist->flist_frame_size);
         framelist->flist_frame_type[framelist->flist_head] = 
             PJMEDIA_JB_MISSING_FRAME;
+        framelist->flist_content_len[framelist->flist_head] = 0;
 
         framelist->flist_origin++;
         framelist->flist_head = (framelist->flist_head + 1 ) % 
@@ -194 +201 @@
         pj_memset(framelist->flist_frame_type+framelist->flist_head,
                   PJMEDIA_JB_MISSING_FRAME,
                   step1*sizeof(framelist->flist_frame_type[0]));
+        pj_bzero(framelist->flist_content_len+framelist->flist_head,
+                  step1*sizeof(framelist->flist_content_len[0]));
 
         if (step2) {
             pj_bzero( framelist->flist_buffer,
@@ -201 +210 @@
             pj_memset(framelist->flist_frame_type,
                       PJMEDIA_JB_MISSING_FRAME,
                       step2*sizeof(framelist->flist_frame_type[0]));
+            pj_bzero (framelist->flist_content_len,
+                      step2*sizeof(framelist->flist_content_len[0]));
         }
 
         // update pointers
@@ -257 +268 @@
               frame, frame_size);
 
     framelist->flist_frame_type[where] = PJMEDIA_JB_NORMAL_FRAME;
+    framelist->flist_content_len[where] = frame_size;
 
     return PJ_TRUE;
 }
@@ -510 +522 @@
                                      void *frame, 
                                      char *p_frame_type)
 {
+    pj_size_t size;
+
+    pjmedia_jbuf_get_frame2(jb, frame, &size, p_frame_type);
+}
+
+/*
+ * Get frame from jitter buffer.
+ */
+PJ_DEF(void) pjmedia_jbuf_get_frame2(pjmedia_jbuf *jb, 
+                                     void *frame, 
+                                     pj_size_t *size,
+                                     char *p_frame_type)
+{
     pjmedia_jb_frame_type ftype;
 
     jb->jb_level--;
@@ -530 +555 @@
         else
             *p_frame_type = PJMEDIA_JB_ZERO_PREFETCH_FRAME;
 
+        *size = 0;
+
         return;
     }
 
     /* Retrieve a frame from frame list */
-    if (jb_framelist_get(&jb->jb_framelist,frame,&ftype) == PJ_FALSE) {
+    if (jb_framelist_get(&jb->jb_framelist,frame,size,&ftype) == PJ_FALSE) {
         /* Can't return frame because jitter buffer is empty! */
         pj_bzero(frame, jb->jb_frame_size);
         *p_frame_type = PJMEDIA_JB_ZERO_EMPTY_FRAME;
+        *size = 0;
 
         return;
     }