]> git.sesse.net Git - vlc/blobdiff - modules/stream_out/rtp.c
Removes trailing spaces. Removes tabs.
[vlc] / modules / stream_out / rtp.c
index 40ba98d59f8f0b99e0d729129a2aeaa8170dd8f9..878e5fb9f387dc873abd9aee580df8d6fbe73e95 100644 (file)
 #include "rtp.h"
 
 #ifdef HAVE_UNISTD_H
+#   include <sys/types.h>
 #   include <unistd.h>
+#   include <fcntl.h>
+#   include <sys/stat.h>
 #endif
 
 #include <errno.h>
     "the multicast packets sent by the stream output (0 = use operating " \
     "system built-in default).")
 
+#define DCCP_TEXT N_("DCCP transport")
+#define DCCP_LONGTEXT N_( \
+    "This enables DCCP instead of UDP as a transport for RTP." )
+#define TCP_TEXT N_("TCP transport")
+#define TCP_LONGTEXT N_( \
+    "This enables TCP instead of UDP as a transport for RTP." )
+#define UDP_LITE_TEXT N_("UDP-Lite transport")
+#define UDP_LITE_LONGTEXT N_( \
+    "This enables UDP-Lite instead of UDP as a transport for RTP." )
+
 #define RFC3016_TEXT N_("MP4A LATM")
 #define RFC3016_LONGTEXT N_( \
     "This allows you to stream MPEG4 LATM audio streams (see RFC3016)." )
@@ -139,6 +152,13 @@ vlc_module_begin();
     add_integer( SOUT_CFG_PREFIX "ttl", 0, NULL, TTL_TEXT,
                  TTL_LONGTEXT, VLC_TRUE );
 
+    add_bool( SOUT_CFG_PREFIX "dccp", 0, NULL,
+              DCCP_TEXT, DCCP_LONGTEXT, VLC_FALSE );
+    add_bool( SOUT_CFG_PREFIX "tcp", 0, NULL,
+              TCP_TEXT, TCP_LONGTEXT, VLC_FALSE );
+    add_bool( SOUT_CFG_PREFIX "udplite", 0, NULL,
+              UDP_LITE_TEXT, UDP_LITE_LONGTEXT, VLC_FALSE );
+
     add_bool( SOUT_CFG_PREFIX "mp4a-latm", 0, NULL, RFC3016_TEXT,
                  RFC3016_LONGTEXT, VLC_FALSE );
 
@@ -150,7 +170,9 @@ vlc_module_end();
  *****************************************************************************/
 static const char *ppsz_sout_options[] = {
     "dst", "name", "port", "port-audio", "port-video", "*sdp", "ttl", "mux",
-    "description", "url","email", "mp4a-latm", NULL
+    "description", "url", "email",
+    "dccp", "tcp", "udplite",
+    "mp4a-latm", NULL
 };
 
 static sout_stream_id_t *Add ( sout_stream_t *, es_format_t * );
@@ -173,7 +195,7 @@ static int HttpSetup( sout_stream_t *p_stream, vlc_url_t * );
 
 struct sout_stream_sys_t
 {
-    /* sdp */
+    /* SDP */
     int64_t i_sdp_id;
     int     i_sdp_version;
     char    *psz_sdp;
@@ -184,24 +206,28 @@ struct sout_stream_sys_t
     char        *psz_session_url;
     char        *psz_session_email;
 
-    /* */
+    /* SDP to disk */
     vlc_bool_t b_export_sdp_file;
     char *psz_sdp_file;
-    /* sap */
+
+    /* SDP via SAP */
     vlc_bool_t b_export_sap;
     session_descriptor_t *p_session;
 
+    /* SDP via HTTP */
     httpd_host_t *p_httpd_host;
     httpd_file_t *p_httpd_file;
 
+    /* RTSP */
     rtsp_stream_t *rtsp;
 
     /* */
-    char *psz_destination;
-    int  i_port;
-    int  i_port_audio;
-    int  i_port_video;
-    int  i_ttl;
+    char     *psz_destination;
+    uint8_t   proto;
+    uint8_t   i_ttl;
+    uint16_t  i_port;
+    uint16_t  i_port_audio;
+    uint16_t  i_port_video;
     vlc_bool_t b_latm;
 
     /* when need to use a private one or when using muxer */
@@ -224,7 +250,7 @@ typedef int (*pf_rtp_packetizer_t)( sout_stream_t *, sout_stream_id_t *,
 typedef struct rtp_sink_t
 {
     int rtp_fd;
-    int rtcp_fd;
+    rtcp_sender_t *rtcp;
 } rtp_sink_t;
 
 struct sout_stream_id_t
@@ -309,13 +335,12 @@ static int Open( vlc_object_t *p_this )
 
     for( p_cfg = p_stream->p_cfg; p_cfg != NULL; p_cfg = p_cfg->p_next )
     {
-        if( !strcmp( p_cfg->psz_name, "sdp" ) )
+        if( !strcmp( p_cfg->psz_name, "sdp" )
+         && ( p_cfg->psz_value != NULL )
+         && !strncasecmp( p_cfg->psz_value, "rtsp:", 5 ) )
         {
-            if( p_cfg->psz_value && !strncasecmp( p_cfg->psz_value, "rtsp", 4 ) )
-            {
-                b_rtsp = VLC_TRUE;
-                break;
-            }
+            b_rtsp = VLC_TRUE;
+            break;
         }
     }
     if( !b_rtsp )
@@ -323,24 +348,35 @@ static int Open( vlc_object_t *p_this )
         psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "sdp" );
         if( psz != NULL )
         {
-            if( !strncasecmp( psz, "rtsp", 4 ) )
+            if( !strncasecmp( psz, "rtsp:", 5 ) )
                 b_rtsp = VLC_TRUE;
             free( psz );
         }
     }
 
-    if( p_sys->psz_destination == NULL )
+    /* Transport protocol */
+    p_sys->proto = IPPROTO_UDP;
+
+#if 0
+    if( var_GetBool( p_stream, SOUT_CFG_PREFIX "dccp" ) )
     {
-        if( !b_rtsp )
-        {
-            msg_Err( p_stream, "missing destination and not in RTSP mode" );
-            free( p_sys );
-            return VLC_EGENERIC;
-        }
+        p_sys->sotype = SOCK_DCCP;
+        p_sys->proto = 33 /*IPPROTO_DCCP*/;
     }
-    else if( p_sys->i_port <= 0 )
+    else
+    if( var_GetBool( p_stream, SOUT_CFG_PREFIX "tcp" ) )
+    {
+        p_sys->sotype = SOCK_STREAM;
+        p_sys->proto = IPPROTO_TCP;
+    }
+    else
+#endif
+    if( var_GetBool( p_stream, SOUT_CFG_PREFIX "udplite" ) )
+        p_sys->proto = 136 /*IPPROTO_UDPLITE*/;
+
+    if( ( p_sys->psz_destination == NULL ) && !b_rtsp )
     {
-        msg_Err( p_stream, "invalid port" );
+        msg_Err( p_stream, "missing destination and not in RTSP mode" );
         free( p_sys );
         return VLC_EGENERIC;
     }
@@ -354,9 +390,6 @@ static int Open( vlc_object_t *p_this )
          * ttl are set. */
         p_sys->i_ttl = config_GetInt( p_stream, "ttl" );
     }
-    if( p_sys->i_ttl > 255 )
-        p_sys->i_ttl = 255;
-    /* must not exceed 999 once formatted */
 
     p_sys->b_latm = var_GetBool( p_stream, SOUT_CFG_PREFIX "mp4a-latm" );
 
@@ -469,7 +502,7 @@ static int Open( vlc_object_t *p_this )
                     continue;
 
                 /* needed both :sout-rtp-sdp= and rtp{sdp=} can be used */
-                if( !strcmp( p_cfg->psz_value, psz ) ) 
+                if( !strcmp( p_cfg->psz_value, psz ) )
                     continue;
 
                 SDPHandleUrl( p_stream, p_cfg->psz_value );
@@ -587,7 +620,7 @@ static void SDPHandleUrl( sout_stream_t *p_stream, char *psz_url )
                                      id->i_port, id->i_port + 1 );
         }
     }
-    else if( ( url.psz_protocol && !strcasecmp( url.psz_protocol, "sap" ) ) || 
+    else if( ( url.psz_protocol && !strcasecmp( url.psz_protocol, "sap" ) ) ||
              ( url.psz_host && !strcasecmp( url.psz_host, "sap" ) ) )
     {
         p_sys->b_export_sap = VLC_TRUE;
@@ -805,7 +838,7 @@ static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
      * mux (TS/PS), then p_fmt is NULL. */
     sout_stream_sys_t *p_sys = p_stream->p_sys;
     sout_stream_id_t  *id;
-    int               i_port;
+    int               i_port, cscov = -1;
     char              *psz_sdp;
 
     id = vlc_object_create( p_stream, sizeof( sout_stream_id_t ) );
@@ -894,7 +927,7 @@ static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
     {
         int ttl = (p_sys->i_ttl > 0) ? p_sys->i_ttl : -1;
         int fd = net_ConnectDgram( p_stream, p_sys->psz_destination,
-                                   i_port, ttl, IPPROTO_UDP );
+                                   i_port, ttl, p_sys->proto );
 
         if( fd == -1 )
         {
@@ -966,7 +999,7 @@ static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
                 char    *p_64_pps = NULL;
                 char    hexa[6+1];
 
-                while( i_buffer > 4 && 
+                while( i_buffer > 4 &&
                        p_buffer[0] == 0 && p_buffer[1] == 0 &&
                        p_buffer[2] == 0 && p_buffer[3] == 1 )
                 {
@@ -1105,6 +1138,11 @@ static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
             return NULL;
     }
 
+    if( cscov != -1 )
+        cscov += 8 /* UDP */ + 12 /* RTP */;
+    if( id->sinkc > 0 )
+        net_SetCSCov( id->sinkv[0].rtp_fd, cscov, -1 );
+
     if( id->i_payload_type == p_sys->i_payload_type )
         p_sys->i_payload_type++;
 
@@ -1304,21 +1342,60 @@ static void ThreadSend( vlc_object_t *p_this )
 {
     sout_stream_id_t *id = (sout_stream_id_t *)p_this;
     unsigned i_caching = id->i_caching;
+#ifdef HAVE_TEE
+    int fd[5] = { -1, -1, -1, -1, -1 };
+
+    if( pipe( fd ) )
+        fd[0] = fd[1] = -1;
+    else
+    if( pipe( fd ) )
+        fd[2] = fd[3] = -1;
+    else
+        fd[4] = open( "/dev/null", O_WRONLY );
+#endif
 
     while( !id->b_die )
     {
         block_t *out = block_FifoGet( id->p_fifo );
         mtime_t  i_date = out->i_dts + i_caching;
+        ssize_t  len = out->i_buffer;
 
+#ifdef HAVE_TEE
+        if( fd[4] != -1 )
+            len = write( fd[1], out->p_buffer, len);
+        if( len == -1 )
+            continue; /* Uho - should not happen */
+#endif
         mwait( i_date );
 
         vlc_mutex_lock( &id->lock_sink );
         for( int i = 0; i < id->sinkc; i++ )
-            send( id->sinkv[i].rtp_fd, out->p_buffer, out->i_buffer, 0 );
+        {
+            SendRTCP( id->sinkv[i].rtcp, out );
+
+#ifdef HAVE_TEE
+            tee( fd[0], fd[3], len, 0 );
+            if( splice( fd[2], NULL, id->sinkv[i].rtp_fd, NULL, len,
+                        SPLICE_F_NONBLOCK ) >= 0 )
+                continue;
+
+            /* splice failed */
+            splice( fd[2], NULL, fd[4], NULL, len, 0 );
+#endif
+            send( id->sinkv[i].rtp_fd, out->p_buffer, len, 0 );
+        }
         vlc_mutex_unlock( &id->lock_sink );
 
         block_Release( out );
+#ifdef HAVE_TEE
+        splice( fd[0], NULL, fd[4], NULL, len, 0 );
+#endif
     }
+
+#ifdef HAVE_TEE
+    for( unsigned i = 0; i < 5; i++ )
+        close( fd[i] );
+#endif
 }
 
 static inline void rtp_packetize_send( sout_stream_id_t *id, block_t *out )
@@ -1328,7 +1405,10 @@ static inline void rtp_packetize_send( sout_stream_id_t *id, block_t *out )
 
 int rtp_add_sink( sout_stream_id_t *id, int fd )
 {
-    rtp_sink_t sink = { fd, -1 };
+    rtp_sink_t sink = { fd, NULL };
+    sink.rtcp = OpenRTCP( VLC_OBJECT( id->p_stream ), fd, IPPROTO_UDP );
+    if( sink.rtcp == NULL )
+        msg_Err( id, "RTCP failed!" );
 
     vlc_mutex_lock( &id->lock_sink );
     INSERT_ELEM( id->sinkv, id->sinkc, id->sinkc, sink );
@@ -1338,18 +1418,23 @@ int rtp_add_sink( sout_stream_id_t *id, int fd )
 
 void rtp_del_sink( sout_stream_id_t *id, int fd )
 {
+    rtp_sink_t sink = { fd, NULL };
+
     /* NOTE: must be safe to use if fd is not included */
     vlc_mutex_lock( &id->lock_sink );
     for( int i = 0; i < id->sinkc; i++ )
     {
         if (id->sinkv[i].rtp_fd == fd)
         {
+            sink = id->sinkv[i];
             REMOVE_ELEM( id->sinkv, id->sinkc, i );
             break;
         }
     }
     vlc_mutex_unlock( &id->lock_sink );
-    net_Close( fd );
+
+    CloseRTCP( sink.rtcp );
+    net_Close( sink.rtp_fd );
 }
 
 /****************************************************************************
@@ -1914,7 +1999,7 @@ static int rtp_packetize_h264( sout_stream_t *p_stream, sout_stream_id_t *id,
                 i_size = i_offset - ( p_buffer[i_offset-1] == 0 ? 1 : 0);
                 i_skip = i_offset;
                 break;
-            } 
+            }
         }
         /* TODO add STAP-A to remove a lot of overhead with small slice/sei/... */
         rtp_packetize_h264_nal( p_stream, id, p_buffer, i_size,