git.sesse.net Git - vlc/blob - modules/access/udp.c

   1 /*****************************************************************************
   2  * udp.c: raw UDP input module
   3  *****************************************************************************
   4  * Copyright (C) 2001-2005 the VideoLAN team
   5  * Copyright (C) 2007 Remi Denis-Courmont
   6  * $Id$
   7  *
   8  * Authors: Christophe Massiot <massiot@via.ecp.fr>
   9  *          Tristan Leteurtre <tooney@via.ecp.fr>
  10  *          Laurent Aimar <fenrir@via.ecp.fr>
  11  *          Jean-Paul Saman <jpsaman #_at_# m2x dot nl>
  12  *          Remi Denis-Courmont
  13  *
  14  * Reviewed: 23 October 2003, Jean-Paul Saman <jpsaman _at_ videolan _dot_ org>
  15  *
  16  * This program is free software; you can redistribute it and/or modify
  17  * it under the terms of the GNU General Public License as published by
  18  * the Free Software Foundation; either version 2 of the License, or
  19  * (at your option) any later version.
  20  *
  21  * This program is distributed in the hope that it will be useful,
  22  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  23  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  24  * GNU General Public License for more details.
  25  *
  26  * You should have received a copy of the GNU General Public License
  27  * along with this program; if not, write to the Free Software
  28  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  29  *****************************************************************************/
  30
  31 /*****************************************************************************
  32  * Preamble
  33  *****************************************************************************/
  34
  35 #ifdef HAVE_CONFIG_H
  36 # include "config.h"
  37 #endif
  38
  39 #include <vlc_common.h>
  40 #include <vlc_plugin.h>
  41 #include <vlc_access.h>
  42 #include <vlc_network.h>
  43
  44 #define MTU 65535
  45
  46 /*****************************************************************************
  47  * Module descriptor
  48  *****************************************************************************/
  49 #define CACHING_TEXT N_("Caching value in ms")
  50 #define CACHING_LONGTEXT N_( \
  51     "Caching value for UDP streams. This " \
  52     "value should be set in milliseconds." )
  53
  54 static int  Open ( vlc_object_t * );
  55 static void Close( vlc_object_t * );
  56
  57 vlc_module_begin();
  58     set_shortname( N_("UDP" ) );
  59     set_description( N_("UDP input") );
  60     set_category( CAT_INPUT );
  61     set_subcategory( SUBCAT_INPUT_ACCESS );
  62
  63     add_integer( "udp-caching", DEFAULT_PTS_DELAY / 1000, NULL, CACHING_TEXT,
  64                  CACHING_LONGTEXT, true );
  65     add_obsolete_integer( "rtp-late" );
  66     add_obsolete_bool( "udp-auto-mtu" );
  67
  68     set_capability( "access", 0 );
  69     add_shortcut( "udp" );
  70     add_shortcut( "udpstream" );
  71     add_shortcut( "udp4" );
  72     add_shortcut( "udp6" );
  73
  74     set_callbacks( Open, Close );
  75 vlc_module_end();
  76
  77 /*****************************************************************************
  78  * Local prototypes
  79  *****************************************************************************/
  80 #define RTP_HEADER_LEN 12
  81
  82 static block_t *BlockUDP( access_t * );
  83 static int Control( access_t *, int, va_list );
  84
  85 /*****************************************************************************
  86  * Open: open the socket
  87  *****************************************************************************/
  88 static int Open( vlc_object_t *p_this )
  89 {
  90     access_t     *p_access = (access_t*)p_this;
  91
  92     char *psz_name = strdup( p_access->psz_path );
  93     char *psz_parser;
  94     const char *psz_server_addr, *psz_bind_addr = "";
  95     int  i_bind_port, i_server_port = 0;
  96     int fam = AF_UNSPEC;
  97     int fd;
  98
  99     /* Set up p_access */
 100     access_InitFields( p_access );
 101     ACCESS_SET_CALLBACKS( NULL, BlockUDP, Control, NULL );
 102     p_access->info.b_prebuffered = false;
 103
 104     if (strlen (p_access->psz_access) > 0)
 105     {
 106         switch (p_access->psz_access[strlen (p_access->psz_access) - 1])
 107         {
 108             case '4':
 109                 fam = AF_INET;
 110                 break;
 111
 112             case '6':
 113                 fam = AF_INET6;
 114                 break;
 115         }
 116     }
 117
 118     i_bind_port = var_CreateGetInteger( p_access, "server-port" );
 119
 120     /* Parse psz_name syntax :
 121      * [serveraddr[:serverport]][@[bindaddr]:[bindport]] */
 122     psz_parser = strchr( psz_name, '@' );
 123     if( psz_parser != NULL )
 124     {
 125         /* Found bind address and/or bind port */
 126         *psz_parser++ = '\0';
 127         psz_bind_addr = psz_parser;
 128
 129         if( psz_bind_addr[0] == '[' )
 130             /* skips bracket'd IPv6 address */
 131             psz_parser = strchr( psz_parser, ']' );
 132
 133         if( psz_parser != NULL )
 134         {
 135             psz_parser = strchr( psz_parser, ':' );
 136             if( psz_parser != NULL )
 137             {
 138                 *psz_parser++ = '\0';
 139                 i_bind_port = atoi( psz_parser );
 140             }
 141         }
 142     }
 143
 144     psz_server_addr = psz_name;
 145     psz_parser = ( psz_server_addr[0] == '[' )
 146         ? strchr( psz_name, ']' ) /* skips bracket'd IPv6 address */
 147         : psz_name;
 148
 149     if( psz_parser != NULL )
 150     {
 151         psz_parser = strchr( psz_parser, ':' );
 152         if( psz_parser != NULL )
 153         {
 154             *psz_parser++ = '\0';
 155             i_server_port = atoi( psz_parser );
 156         }
 157     }
 158
 159     msg_Dbg( p_access, "opening server=%s:%d local=%s:%d",
 160              psz_server_addr, i_server_port, psz_bind_addr, i_bind_port );
 161
 162     fd = net_OpenDgram( p_access, psz_bind_addr, i_bind_port,
 163                         psz_server_addr, i_server_port, fam, IPPROTO_UDP );
 164     free (psz_name);
 165     if( fd == -1 )
 166     {
 167         msg_Err( p_access, "cannot open socket" );
 168         return VLC_EGENERIC;
 169     }
 170     p_access->p_sys = (void *)(intptr_t)fd;
 171
 172     /* Update default_pts to a suitable value for udp access */
 173     var_Create( p_access, "udp-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
 174     return VLC_SUCCESS;
 175 }
 176
 177 /*****************************************************************************
 178  * Close: free unused data structures
 179  *****************************************************************************/
 180 static void Close( vlc_object_t *p_this )
 181 {
 182     access_t     *p_access = (access_t*)p_this;
 183
 184     net_Close( (intptr_t)p_access->p_sys );
 185 }
 186
 187 /*****************************************************************************
 188  * Control:
 189  *****************************************************************************/
 190 static int Control( access_t *p_access, int i_query, va_list args )
 191 {
 192     bool   *pb_bool;
 193     int          *pi_int;
 194     int64_t      *pi_64;
 195
 196     switch( i_query )
 197     {
 198         /* */
 199         case ACCESS_CAN_SEEK:
 200         case ACCESS_CAN_FASTSEEK:
 201         case ACCESS_CAN_PAUSE:
 202         case ACCESS_CAN_CONTROL_PACE:
 203             pb_bool = (bool*)va_arg( args, bool* );
 204             *pb_bool = false;
 205             break;
 206         /* */
 207         case ACCESS_GET_MTU:
 208             pi_int = (int*)va_arg( args, int * );
 209             *pi_int = MTU;
 210             break;
 211
 212         case ACCESS_GET_PTS_DELAY:
 213             pi_64 = (int64_t*)va_arg( args, int64_t * );
 214             *pi_64 = var_GetInteger( p_access, "udp-caching" ) * 1000;
 215             break;
 216
 217         /* */
 218         case ACCESS_SET_PAUSE_STATE:
 219         case ACCESS_GET_TITLE_INFO:
 220         case ACCESS_SET_TITLE:
 221         case ACCESS_SET_SEEKPOINT:
 222         case ACCESS_SET_PRIVATE_ID_STATE:
 223         case ACCESS_GET_CONTENT_TYPE:
 224             return VLC_EGENERIC;
 225
 226         default:
 227             msg_Warn( p_access, "unimplemented query in control" );
 228             return VLC_EGENERIC;
 229
 230     }
 231     return VLC_SUCCESS;
 232 }
 233
 234 /*****************************************************************************
 235  * BlockUDP:
 236  *****************************************************************************/
 237 static block_t *BlockUDP( access_t *p_access )
 238 {
 239     access_sys_t *p_sys = p_access->p_sys;
 240     block_t      *p_block;
 241     ssize_t len;
 242
 243     if( p_access->info.b_eof )
 244         return NULL;
 245
 246     /* Read data */
 247     p_block = block_New( p_access, MTU );
 248     len = net_Read( p_access, (intptr_t)p_sys, NULL,
 249                     p_block->p_buffer, MTU, false );
 250     if( len < 0 )
 251     {
 252         block_Release( p_block );
 253         return NULL;
 254     }
 255
 256     return block_Realloc( p_block, 0, len );
 257 }