[vlc-devel] [PATCH] udp: pull and buffer received packets in udp access module

Tzu-Jung Lee roylee17 at gmail.com
Fri Aug 16 17:15:20 CEST 2013


Signed-off-by: Tzu-Jung Lee <tjlee at ambarella.com>
---
 modules/access/udp.c | 80 +++++++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 63 insertions(+), 17 deletions(-)

diff --git a/modules/access/udp.c b/modules/access/udp.c
index 6613ebe..c8bc519 100644
--- a/modules/access/udp.c
+++ b/modules/access/udp.c
@@ -40,6 +40,7 @@
 #include <vlc_plugin.h>
 #include <vlc_access.h>
 #include <vlc_network.h>
+#include <vlc_block.h>
 
 #define MTU 65535
 
@@ -63,11 +64,21 @@ vlc_module_begin ()
     set_callbacks( Open, Close )
 vlc_module_end ()
 
+struct access_sys_t
+{
+    int          i_handle;
+
+    block_fifo_t *p_fifo;
+
+    vlc_thread_t thread;
+};
+
 /*****************************************************************************
  * Local prototypes
  *****************************************************************************/
 static block_t *BlockUDP( access_t * );
 static int Control( access_t *, int, va_list );
+static void* ThreadRead( void *data );
 
 /*****************************************************************************
  * Open: open the socket
@@ -75,12 +86,18 @@ static int Control( access_t *, int, va_list );
 static int Open( vlc_object_t *p_this )
 {
     access_t     *p_access = (access_t*)p_this;
+    access_sys_t *p_sys;
 
     char *psz_name = strdup( p_access->psz_location );
     char *psz_parser;
     const char *psz_server_addr, *psz_bind_addr = "";
     int  i_bind_port = 1234, i_server_port = 0;
-    int fd;
+    int  i_handle;
+
+    if( !( p_sys = malloc ( sizeof( *p_sys ) ) ) )
+        return VLC_ENOMEM;
+
+    p_access->p_sys = p_sys;
 
     /* Set up p_access */
     access_InitFields( p_access );
@@ -128,15 +145,27 @@ static int Open( vlc_object_t *p_this )
     msg_Dbg( p_access, "opening server=%s:%d local=%s:%d",
              psz_server_addr, i_server_port, psz_bind_addr, i_bind_port );
 
-    fd = net_OpenDgram( p_access, psz_bind_addr, i_bind_port,
+    i_handle = net_OpenDgram( p_access, psz_bind_addr, i_bind_port,
                         psz_server_addr, i_server_port, IPPROTO_UDP );
     free (psz_name);
-    if( fd == -1 )
+    if( i_handle == -1 )
     {
         msg_Err( p_access, "cannot open socket" );
         return VLC_EGENERIC;
     }
-    p_access->p_sys = (void *)(intptr_t)fd;
+
+    p_sys->i_handle = i_handle;
+    p_sys->p_fifo = block_FifoNew();
+
+    if( vlc_clone( &p_sys->thread, ThreadRead, p_access,
+                           VLC_THREAD_PRIORITY_HIGHEST ) )
+    {
+        msg_Err( p_access, "cannot spawn UDP access thread" );
+        block_FifoRelease( p_sys->p_fifo );
+        net_Close (i_handle);
+        free (p_sys);
+        return VLC_EGENERIC;
+    }
 
     return VLC_SUCCESS;
 }
@@ -147,8 +176,11 @@ static int Open( vlc_object_t *p_this )
 static void Close( vlc_object_t *p_this )
 {
     access_t     *p_access = (access_t*)p_this;
+    access_sys_t *p_sys = p_access->p_sys;
 
-    net_Close( (intptr_t)p_access->p_sys );
+    block_FifoRelease( p_sys->p_fifo );
+    net_Close( p_sys->i_handle );
+    free (p_sys);
 }
 
 /*****************************************************************************
@@ -198,20 +230,34 @@ static int Control( access_t *p_access, int i_query, va_list args )
  *****************************************************************************/
 static block_t *BlockUDP( access_t *p_access )
 {
-    int fd = (intptr_t)p_access->p_sys;
+    access_sys_t *p_sys = p_access->p_sys;
 
-    /* Read data */
-    block_t *p_block = block_Alloc( MTU );
-    if( unlikely(p_block == NULL) )
-        return NULL;
+    return block_FifoGet( p_sys->p_fifo );
+}
+/*****************************************************************************
+ * ThreadRead: Pull packets from socket as soon as possible.
+ *****************************************************************************/
+static void* ThreadRead( void *data )
+{
+    access_t     *p_access = (access_t*)data;
+    access_sys_t *p_sys = p_access->p_sys;
 
-    ssize_t len = net_Read( p_access, fd, NULL,
-                            p_block->p_buffer, MTU, false );
-    if( len < 0 )
+    for (;;)
     {
-        block_Release( p_block );
-        return NULL;
-    }
+        block_t *p_pk = block_Alloc( MTU );
+        if( unlikely(p_pk == NULL) )
+            return NULL;
 
-    return block_Realloc( p_block, 0, len );
+        ssize_t len = net_Read( p_access, p_sys->i_handle, NULL,
+                                p_pk->p_buffer, MTU, false );
+        if( len < 0 )
+        {
+            block_Release( p_pk );
+            return NULL;
+        }
+
+        p_pk = block_Realloc( p_pk, 0, len );
+        block_FifoPut( p_sys->p_fifo, p_pk );
+    }
+    return NULL;
 }
-- 
1.8.3.2




More information about the vlc-devel mailing list