]> git.sesse.net Git - vlc/commitdiff
* udp: converted to access2 (using pf_block, so for now it may hurt a
authorLaurent Aimar <fenrir@videolan.org>
Fri, 4 Jun 2004 21:51:34 +0000 (21:51 +0000)
committerLaurent Aimar <fenrir@videolan.org>
Fri, 4 Jun 2004 21:51:34 +0000 (21:51 +0000)
bit the perfs as we do a memcpy in access2.c)

modules/access/access2.c
modules/access/udp.c

index 99d74343032a312fc4bef3bb969782c65ec6786d..4d32e5348043b9203352fb1cb1afdce2f65105d4 100644 (file)
@@ -44,12 +44,18 @@ vlc_module_begin();
     add_shortcut( "ftp" );
     add_shortcut( "tcp" );
     add_shortcut( "pvr" );
+
     add_shortcut( "file" );
     add_shortcut( "stream" );
     add_shortcut( "kfir" );
 
-    /* Hack */
-    //add_shortcut( "file" );
+    add_shortcut( "udp" );
+    add_shortcut( "udp4" );
+    add_shortcut( "udp6" );
+
+    add_shortcut( "rtp" );
+    add_shortcut( "rtp4" );
+    add_shortcut( "rtp6" );
 vlc_module_end();
 
 /*****************************************************************************
@@ -68,6 +74,8 @@ typedef struct
     int i_title;
     input_title_t **title;
 
+    vlc_bool_t b_first_read;
+
 } access2_sys_t;
 
 /*****************************************************************************
@@ -183,6 +191,7 @@ static int Access2Open( vlc_object_t * p_this )
 
     p_sys->p_access = p_access;
     p_sys->p_block = NULL;
+    p_sys->b_first_read = VLC_TRUE;
 
     return VLC_SUCCESS;
 }
@@ -252,10 +261,14 @@ static int  Access2Read( input_thread_t *p_input, byte_t *p_buffer, size_t i_len
         }
 
         i_copy = __MIN( i_len - i_total, p_sys->p_block->i_buffer );
-        memcpy( &p_buffer[i_total], p_sys->p_block->p_buffer, i_copy );
+        if( i_copy > 0 )
+        {
+            memcpy( &p_buffer[i_total], p_sys->p_block->p_buffer, i_copy );
+
+            p_sys->p_block->i_buffer -= i_copy;
+            p_sys->p_block->p_buffer += i_copy;
+        }
 
-        p_sys->p_block->i_buffer -= i_copy;
-        p_sys->p_block->p_buffer += i_copy;
         if( p_sys->p_block->i_buffer <= 0 )
         {
             block_Release( p_sys->p_block );
@@ -265,6 +278,16 @@ static int  Access2Read( input_thread_t *p_input, byte_t *p_buffer, size_t i_len
         i_total += i_copy;
     }
 update:
+    if( p_sys->b_first_read )
+    {
+        /* Some access update it only after first read (like udp) */
+        if( p_access->psz_demux && *p_access->psz_demux )
+        {
+            if( !p_input->psz_demux || *p_input->psz_demux == '\0' )
+                p_input->psz_demux = strdup( p_access->psz_demux );
+        }
+        p_sys->b_first_read = VLC_FALSE;
+    }
     if( p_access->info.i_update & INPUT_UPDATE_SIZE )
     {
         vlc_mutex_lock( &p_input->stream.stream_lock );
index 9352f63659bb75269b980761ffdc815471cbe804..e076299d102b28e79ca42b596b4c9a93bbc8b15d 100644 (file)
     "Allows you to modify the default caching value for UDP streams. This " \
     "value should be set in millisecond units." )
 
+#define AUTO_MTU_TEXT N_("Autodetection of MTU")
+#define AUTO_MTU_LONGTEXT N_( \
+    "Allows growing the MTU if truncated packets are found" )
+
 static int  Open ( vlc_object_t * );
 static void Close( vlc_object_t * );
 
@@ -51,8 +55,10 @@ vlc_module_begin();
 
     add_integer( "udp-caching", DEFAULT_PTS_DELAY / 1000, NULL, CACHING_TEXT,
                  CACHING_LONGTEXT, VLC_TRUE );
+    add_bool( "udp-auto-mtu", 0, NULL,
+              AUTO_MTU_TEXT, AUTO_MTU_LONGTEXT, VLC_TRUE );
 
-    set_capability( "access", 0 );
+    set_capability( "access2", 0 );
     add_shortcut( "udp" );
     add_shortcut( "udpstream" );
     add_shortcut( "udp4" );
@@ -68,13 +74,17 @@ vlc_module_end();
  *****************************************************************************/
 #define RTP_HEADER_LEN 12
 
-static ssize_t Read    ( input_thread_t *, byte_t *, size_t );
-static ssize_t RTPRead ( input_thread_t *, byte_t *, size_t );
-static ssize_t RTPChoose( input_thread_t *, byte_t *, size_t );
+static block_t *BlockUDP( access_t * );
+static block_t *BlockRTP( access_t * );
+static block_t *BlockChoose( access_t * );
+static int Control( access_t *, int, va_list );
 
 struct access_sys_t
 {
     int fd;
+
+    int i_mtu;
+    vlc_bool_t b_auto_mtu;
 };
 
 /*****************************************************************************
@@ -82,43 +92,44 @@ struct access_sys_t
  *****************************************************************************/
 static int Open( vlc_object_t *p_this )
 {
-    input_thread_t     *p_input = (input_thread_t *)p_this;
-    access_sys_t       *p_sys;
+    access_t     *p_access = (access_t*)p_this;
+    access_sys_t *p_sys;
 
-    char *              psz_name = strdup(p_input->psz_name);
-    char *              psz_parser = psz_name;
-    char *              psz_server_addr = "";
-    char *              psz_server_port = "";
-    char *              psz_bind_addr = "";
-    char *              psz_bind_port = "";
-    int                 i_bind_port = 0, i_server_port = 0;
-    vlc_value_t         val;
+    char *psz_name = strdup( p_access->psz_path );
+    char *psz_parser = psz_name;
+    char *psz_server_addr = "";
+    char *psz_server_port = "";
+    char *psz_bind_addr = "";
+    char *psz_bind_port = "";
+    int  i_bind_port = 0;
+    int  i_server_port = 0;
+    vlc_value_t val;
 
 
     /* First set ipv4/ipv6 */
-    var_Create( p_input, "ipv4", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
-    var_Create( p_input, "ipv6", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
+    var_Create( p_access, "ipv4", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
+    var_Create( p_access, "ipv6", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
 
-    if( *p_input->psz_access )
+    if( *p_access->psz_access )
     {
         /* Find out which shortcut was used */
-        if( !strncmp( p_input->psz_access, "udp4", 6 ) ||
-            !strncmp( p_input->psz_access, "rtp4", 6 ))
+        if( !strncmp( p_access->psz_access, "udp4", 6 ) ||
+            !strncmp( p_access->psz_access, "rtp4", 6 ))
         {
             val.b_bool = VLC_TRUE;
-            var_Set( p_input, "ipv4", val );
+            var_Set( p_access, "ipv4", val );
 
             val.b_bool = VLC_FALSE;
-            var_Set( p_input, "ipv6", val );
+            var_Set( p_access, "ipv6", val );
         }
-        else if( !strncmp( p_input->psz_access, "udp6", 6 ) ||
-                 !strncmp( p_input->psz_access, "rtp6", 6 ) )
+        else if( !strncmp( p_access->psz_access, "udp6", 6 ) ||
+                 !strncmp( p_access->psz_access, "rtp6", 6 ) )
         {
             val.b_bool = VLC_TRUE;
-            var_Set( p_input, "ipv6", val );
+            var_Set( p_access, "ipv6", val );
 
             val.b_bool = VLC_FALSE;
-            var_Set( p_input, "ipv4", val );
+            var_Set( p_access, "ipv4", val );
         }
     }
 
@@ -190,24 +201,32 @@ static int Open( vlc_object_t *p_this )
     i_server_port = strtol( psz_server_port, NULL, 10 );
     if( ( i_bind_port   = strtol( psz_bind_port,   NULL, 10 ) ) == 0 )
     {
-        i_bind_port = config_GetInt( p_this, "server-port" );
+        var_Create( p_access, "server-port", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
+        var_Get( p_access, "server-port", &val  );
+        i_bind_port = val.i_int;
     }
 
-    msg_Dbg( p_input, "opening server=%s:%d local=%s:%d",
+    msg_Dbg( p_access, "opening server=%s:%d local=%s:%d",
              psz_server_addr, i_server_port, psz_bind_addr, i_bind_port );
 
-    p_sys = p_input->p_access_data = malloc( sizeof( access_sys_t ) );
-    if( p_sys == NULL )
-    {
-        msg_Err( p_input, "out of memory" );
-        return VLC_EGENERIC;
-    }
-
-    p_sys->fd = net_OpenUDP( p_input, psz_bind_addr, i_bind_port,
+    /* Set up p_access */
+    p_access->pf_read = NULL;
+    p_access->pf_block = BlockChoose;
+    p_access->pf_control = Control;
+    p_access->pf_seek = NULL;
+    p_access->info.i_update = 0;
+    p_access->info.i_size = 0;
+    p_access->info.i_pos = 0;
+    p_access->info.b_eof = VLC_FALSE;
+    p_access->info.i_title = 0;
+    p_access->info.i_seekpoint = 0;
+
+    p_access->p_sys = p_sys = malloc( sizeof( access_sys_t ) );
+    p_sys->fd = net_OpenUDP( p_access, psz_bind_addr, i_bind_port,
                                       psz_server_addr, i_server_port );
     if( p_sys->fd < 0 )
     {
-        msg_Err( p_input, "cannot open socket" );
+        msg_Err( p_access, "cannot open socket" );
         free( psz_name );
         free( p_sys );
         return VLC_EGENERIC;
@@ -215,27 +234,16 @@ static int Open( vlc_object_t *p_this )
     free( psz_name );
 
     /* FIXME */
-    var_Create( p_input, "mtu", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
-    var_Get( p_input, "mtu", &val);
-    p_input->i_mtu = val.i_int;
-
-    /* fill p_input fields */
-    p_input->pf_read = RTPChoose;
-    p_input->pf_set_program = input_SetProgram;
-    p_input->pf_set_area = NULL;
-    p_input->pf_seek = NULL;
-
-    vlc_mutex_lock( &p_input->stream.stream_lock );
-    p_input->stream.b_pace_control = VLC_FALSE;
-    p_input->stream.b_seekable = VLC_FALSE;
-    p_input->stream.p_selected_area->i_tell = 0;
-    p_input->stream.i_method = INPUT_METHOD_NETWORK;
-    vlc_mutex_unlock( &p_input->stream.stream_lock );
+    var_Create( p_access, "mtu", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
+    var_Get( p_access, "mtu", &val);
+    p_sys->i_mtu = val.i_int > 0 ? val.i_int : 1500;    /* avoid problem */
+
+    var_Create( p_access, "udp-auto-mtu", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
+    var_Get( p_access, "udp-auto-mtu", &val);
+    p_sys->b_auto_mtu = val.b_bool;
 
     /* Update default_pts to a suitable value for udp access */
-    var_Create( p_input, "udp-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
-    var_Get( p_input, "udp-caching", &val );
-    p_input->i_pts_delay = val.i_int * 1000;
+    var_Create( p_access, "udp-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
 
     return VLC_SUCCESS;
 }
@@ -245,183 +253,199 @@ static int Open( vlc_object_t *p_this )
  *****************************************************************************/
 static void Close( vlc_object_t *p_this )
 {
-    input_thread_t *p_input = (input_thread_t *)p_this;
-    access_sys_t   *p_sys = p_input->p_access_data;
-
-    msg_Info( p_input, "closing UDP target `%s'", p_input->psz_source );
+    access_t     *p_access = (access_t*)p_this;
+    access_sys_t *p_sys = p_access->p_sys;
 
     net_Close( p_sys->fd );
-
     free( p_sys );
 }
 
 /*****************************************************************************
- * Read: read on a file descriptor, checking b_die periodically
+ * Control:
  *****************************************************************************/
-static ssize_t Read( input_thread_t * p_input, byte_t * p_buffer, size_t i_len )
+static int Control( access_t *p_access, int i_query, va_list args )
 {
-    access_sys_t   *p_sys = p_input->p_access_data;
+    access_sys_t *p_sys = p_access->p_sys;
+    vlc_bool_t   *pb_bool;
+    int          *pi_int;
+    int64_t      *pi_64;
+    vlc_value_t  val;
+
+    switch( i_query )
+    {
+        /* */
+        case ACCESS_CAN_SEEK:
+        case ACCESS_CAN_FASTSEEK:
+        case ACCESS_CAN_PAUSE:
+        case ACCESS_CAN_CONTROL_PACE:
+            pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
+            *pb_bool = VLC_FALSE;
+            break;
+        /* */
+        case ACCESS_GET_MTU:
+            pi_int = (int*)va_arg( args, int * );
+            *pi_int = p_sys->i_mtu;
+            break;
+
+        case ACCESS_GET_PTS_DELAY:
+            pi_64 = (int64_t*)va_arg( args, int64_t * );
+            var_Get( p_access, "udp-caching", &val );
+            *pi_64 = val.i_int * 1000;
+            break;
+
+        /* */
+        case ACCESS_SET_PAUSE_STATE:
+        case ACCESS_GET_TITLE_INFO:
+        case ACCESS_SET_TITLE:
+        case ACCESS_SET_SEEKPOINT:
+            return VLC_EGENERIC;
+
+        default:
+            msg_Err( p_access, "unimplemented query in control" );
+            return VLC_EGENERIC;
 
-    return net_Read( p_input, p_sys->fd, p_buffer, i_len, VLC_FALSE );
+    }
+    return VLC_SUCCESS;
 }
 
 /*****************************************************************************
- * RTPRead : read from the network, and parse the RTP header
+ * BlockUDP:
  *****************************************************************************/
-static ssize_t RTPRead( input_thread_t * p_input, byte_t * p_buffer,
-                        size_t i_len )
+static block_t *BlockUDP( access_t *p_access )
 {
-    int         i_rtp_version;
-    int         i_CSRC_count;
-    int         i_payload_type;
-    int         i_skip = 0;
+    access_sys_t *p_sys = p_access->p_sys;
+    block_t      *p_block;
 
-    byte_t *    p_tmp_buffer = alloca( p_input->i_mtu );
+    /* Read data */
+    p_block = block_New( p_access, p_sys->i_mtu );
+    p_block->i_buffer = net_Read( p_access, p_sys->fd, p_block->p_buffer, p_sys->i_mtu, VLC_FALSE );
+    if( p_block->i_buffer <= 0 )
+    {
+        block_Release( p_block );
+        return NULL;
+    }
 
-    /* Get the raw data from the socket.
-     * We first assume that RTP header size is the classic RTP_HEADER_LEN. */
-    ssize_t i_ret = Read( p_input, p_tmp_buffer, p_input->i_mtu );
+    if( p_block->i_buffer >= p_sys->i_mtu && p_sys->b_auto_mtu )
+    {
+        /* Increase by 10% */
+        p_sys->i_mtu += ( p_sys->i_mtu + 9 ) / 10;
+        msg_Dbg( p_access, "increasing MTU to %d", p_sys->i_mtu );
+    }
 
-    if ( i_ret <= 0 ) return 0; /* i_ret is at least 1 */
+    return p_block;
+}
+
+/*****************************************************************************
+ * BlockParseRTP/BlockRTP:
+ *****************************************************************************/
+static block_t *BlockParseRTP( access_t *p_access, block_t *p_block )
+{
+    int     i_rtp_version;
+    int     i_CSRC_count;
+    int     i_payload_type;
+    int     i_skip = 0;
+
+    if( p_block->i_buffer < RTP_HEADER_LEN )
+        goto trash;
 
     /* Parse the header and make some verifications.
      * See RFC 1889 & RFC 2250. */
-
-    i_rtp_version  = ( p_tmp_buffer[0] & 0xC0 ) >> 6;
-    i_CSRC_count   = ( p_tmp_buffer[0] & 0x0F );
-    i_payload_type = ( p_tmp_buffer[1] & 0x7F );
+    i_rtp_version  = ( p_block->p_buffer[0] & 0xC0 ) >> 6;
+    i_CSRC_count   = ( p_block->p_buffer[0] & 0x0F );
+    i_payload_type = ( p_block->p_buffer[1] & 0x7F );
 
     if ( i_rtp_version != 2 )
-        msg_Dbg( p_input, "RTP version is %u, should be 2", i_rtp_version );
+        msg_Dbg( p_access, "RTP version is %u, should be 2", i_rtp_version );
 
     if( i_payload_type == 14 )
-    {
         i_skip = 4;
-    }
     else if( i_payload_type !=  33 && i_payload_type != 32 )
-    {
-        msg_Dbg( p_input, "unsupported RTP payload type (%u)",
-                 i_payload_type );
-    }
+        msg_Dbg( p_access, "unsupported RTP payload type (%u)", i_payload_type );
+
     i_skip += RTP_HEADER_LEN + 4*i_CSRC_count;
 
     /* A CSRC extension field is 32 bits in size (4 bytes) */
-    if ( i_ret < i_skip )
-    {
-        /* Packet is not big enough to hold the complete RTP_HEADER with
-         * CSRC extensions.
-         */
-        msg_Warn( p_input, "RTP input trashing %d bytes", i_ret - i_len );
-        return 0;
-    }
+    if( i_skip >= p_block->i_buffer )
+        goto trash;
 
     /* Return the packet without the RTP header. */
-    i_ret -= i_skip;
+    p_block->i_buffer -= i_skip;
+    p_block->p_buffer += i_skip;
 
-    if ( (size_t)i_ret > i_len )
-    {
-        /* This should NOT happen. */
-        msg_Warn( p_input, "RTP input trashing %d bytes", i_ret - i_len );
-        i_ret = i_len;
-    }
+    return p_block;
 
-    p_input->p_vlc->pf_memcpy( p_buffer, &p_tmp_buffer[i_skip], i_ret );
+trash:
+    msg_Warn( p_access, "received a too short packet for RTP" );
+    block_Release( p_block );
+    return NULL;
+}
 
-    return i_ret;
+static block_t *BlockRTP( access_t *p_access )
+{
+    return BlockParseRTP( p_access, BlockUDP( p_access ) );
 }
 
 /*****************************************************************************
- * RTPChoose : read from the network, and decide whether it's UDP or RTP
+ * BlockChoose: decide between RTP and UDP
  *****************************************************************************/
-static ssize_t RTPChoose( input_thread_t * p_input, byte_t * p_buffer,
-                          size_t i_len )
+static block_t *BlockChoose( access_t *p_access )
 {
-    int         i_rtp_version;
-    int         i_CSRC_count;
-    int         i_payload_type;
-
-    byte_t *    p_tmp_buffer = alloca( p_input->i_mtu );
+    block_t *p_block;
+    int     i_rtp_version;
+    int     i_CSRC_count;
+    int     i_payload_type;
 
-    /* Get the raw data from the socket.
-     * We first assume that RTP header size is the classic RTP_HEADER_LEN. */
-    ssize_t i_ret = Read( p_input, p_tmp_buffer, p_input->i_mtu );
+    if( ( p_block = BlockUDP( p_access ) ) == NULL )
+        return NULL;
 
-    if ( i_ret <= 0 ) return 0; /* i_ret is at least 1 */
-
-    /* Check that it's not TS. */
-    if ( p_tmp_buffer[0] == 0x47 )
+    if( p_block->p_buffer[0] == 0x47 )
     {
-        msg_Dbg( p_input, "detected TS over raw UDP" );
-        p_input->pf_read = Read;
-        p_input->p_vlc->pf_memcpy( p_buffer, p_tmp_buffer, i_ret );
-        return i_ret;
+        msg_Dbg( p_access, "detected TS over raw UDP" );
+        p_access->pf_block = BlockUDP;
+        return p_block;
     }
 
+    if( p_block->i_buffer < RTP_HEADER_LEN )
+        return p_block;
+
     /* Parse the header and make some verifications.
      * See RFC 1889 & RFC 2250. */
 
-    i_rtp_version  = ( p_tmp_buffer[0] & 0xC0 ) >> 6;
-    i_CSRC_count   = ( p_tmp_buffer[0] & 0x0F );
-    i_payload_type = ( p_tmp_buffer[1] & 0x7F );
+    i_rtp_version  = ( p_block->p_buffer[0] & 0xC0 ) >> 6;
+    i_CSRC_count   = ( p_block->p_buffer[0] & 0x0F );
+    i_payload_type = ( p_block->p_buffer[1] & 0x7F );
 
-    if ( i_rtp_version != 2 )
+    if( i_rtp_version != 2 )
     {
-        msg_Dbg( p_input, "no supported RTP header detected" );
-        p_input->pf_read = Read;
-        p_input->p_vlc->pf_memcpy( p_buffer, p_tmp_buffer, i_ret );
-        return i_ret;
+        msg_Dbg( p_access, "no supported RTP header detected" );
+        p_access->pf_block = BlockUDP;
+        return p_block;
     }
 
-    switch ( i_payload_type )
-    {
-    case 33:
-        msg_Dbg( p_input, "detected TS over RTP" );
-        break;
-
-    case 14:
-        msg_Dbg( p_input, "detected MPEG audio over RTP" );
-        if( !p_input->psz_demux || *p_input->psz_demux == '\0' )
-        {
-            p_input->psz_demux = "mp3";
-        }
-        break;
-
-    case 32:
-        msg_Dbg( p_input, "detected MPEG video over RTP" );
-        break;
-
-    default:
-        msg_Dbg( p_input, "no RTP header detected" );
-        p_input->pf_read = Read;
-        p_input->p_vlc->pf_memcpy( p_buffer, p_tmp_buffer, i_ret );
-        return i_ret;
-    }
-
-    p_input->pf_read = RTPRead;
-
-    /* A CSRC extension field is 32 bits in size (4 bytes) */
-    if( i_ret < RTP_HEADER_LEN + 4*i_CSRC_count )
-    {
-        /* Packet is not big enough to hold the complete RTP_HEADER with
-         * CSRC extensions.
-         */
-        msg_Warn( p_input, "RTP input trashing %d bytes", i_ret - i_len );
-        return 0;
-    }
-
-    /* Return the packet without the RTP header. */
-    i_ret -= RTP_HEADER_LEN + 4*i_CSRC_count;
-
-    if ( (size_t)i_ret > i_len )
+    switch( i_payload_type )
     {
-        /* This should NOT happen. */
-        msg_Warn( p_input, "RTP input trashing %d bytes", i_ret - i_len );
-        i_ret = i_len;
+        case 33:
+            msg_Dbg( p_access, "detected TS over RTP" );
+            p_access->psz_demux = strdup( "ts2" );
+            break;
+
+        case 14:
+            msg_Dbg( p_access, "detected MPEG audio over RTP" );
+            p_access->psz_demux = strdup( "mp3" );
+            break;
+
+        case 32:
+            msg_Dbg( p_access, "detected MPEG video over RTP" );
+            p_access->psz_demux = strdup( "es" );
+            break;
+
+        default:
+            msg_Dbg( p_access, "no RTP header detected" );
+            p_access->pf_block = BlockUDP;
+            return p_block;
     }
 
-    p_input->p_vlc->pf_memcpy( p_buffer,
-                               &p_tmp_buffer[RTP_HEADER_LEN + 4*i_CSRC_count],
-                               i_ret );
+    p_access->pf_block = BlockRTP;
 
-    return i_ret;
+    return BlockParseRTP( p_access, p_block );
 }