]> git.sesse.net Git - vlc/blobdiff - modules/stream_out/rtp.c
* Improve the h264 rtp muxing even more. Should now be able to handle multiple h264...
[vlc] / modules / stream_out / rtp.c
index fb14e45869e4c9b72bfe04b1e61d5d6d3c4e71b2..5e19ffead8a678a2846dad8c3c2484a43cd9eaa2 100644 (file)
 /*****************************************************************************
  * Preamble
  *****************************************************************************/
+#include "config.h"
+#include <stdio.h>
 #include <stdlib.h>
+#ifdef HAVE_UNISTD_H
+#   include <unistd.h>
+#endif
 
 #include <errno.h>
 
@@ -244,6 +249,7 @@ struct sout_stream_id_t
     char        *psz_destination;
     int         i_port;
     int         i_cat;
+    int         i_bitrate;
 
     /* Packetizer specific fields */
     pf_rtp_packetizer_t pf_packetize;
@@ -290,11 +296,12 @@ static int Open( vlc_object_t *p_this )
     sout_stream_t       *p_stream = (sout_stream_t*)p_this;
     sout_instance_t     *p_sout = p_stream->p_sout;
     sout_stream_sys_t   *p_sys = NULL;
-    sout_cfg_t          *p_cfg = NULL;
+    config_chain_t      *p_cfg = NULL;
     vlc_value_t         val;
     vlc_bool_t          b_rtsp = VLC_FALSE;
 
-    sout_CfgParse( p_stream, SOUT_CFG_PREFIX, ppsz_sout_options, p_stream->p_cfg );
+    config_ChainParse( p_stream, SOUT_CFG_PREFIX,
+                       ppsz_sout_options, p_stream->p_cfg );
 
     p_sys = malloc( sizeof( sout_stream_sys_t ) );
 
@@ -520,17 +527,11 @@ static int Open( vlc_object_t *p_this )
         }
 
         /* create the SDP for a muxed stream (only once) */
-        /* FIXME  http://www.faqs.org/rfcs/rfc2327.html
-           All text fields should be UTF-8 encoded. Use global a:charset to announce this.
+        /* FIXME  http://www.faqs.org/rfcs/rfc4566.html
            o= - should be local username (no spaces allowed)
-           o= time should be hashed with some other value to garantue uniqueness
-           o= we need IP6 support?
+           o= time should be hashed with some other value to garantee uniqueness
            o= don't use the localhost address. use fully qualified domain name or IP4 address
-           p= international phone number (pass via vars?)
-           c= IP6 support
-           a= recvonly (missing)
-           a= type:broadcast (missing)
-           a= charset: (normally charset should be UTF-8, this can be used to override s= and i=)
+          a= source-filter: we need our source address
            a= x-plgroup: (missing)
            RTP packets need to get the correct src IP address  */
         if( net_AddressIsMulticast( (vlc_object_t *)p_stream, p_sys->psz_destination ) )
@@ -551,9 +552,11 @@ static int Open( vlc_object_t *p_this )
                   "i=%s\r\n"
                   "u=%s\r\n"
                   "e=%s\r\n"
+                  "c=IN IP%c %s%s\r\n"
                   "t=0 0\r\n" /* permanent stream */ /* when scheduled from vlm, we should set this info correctly */
                   "a=tool:"PACKAGE_STRING"\r\n"
-                  "c=IN IP%c %s%s\r\n"
+                  "a=recvonly\r\n"
+                  "a=type:broadcast\r\n"
                   "m=video %d RTP/AVP %d\r\n"
                   "a=rtpmap:%d %s\r\n",
                   p_sys->i_sdp_id, p_sys->i_sdp_version,
@@ -588,7 +591,7 @@ static int Open( vlc_object_t *p_this )
     var_Get( p_stream, SOUT_CFG_PREFIX "sdp", &val );
     if( *val.psz_string )
     {
-        sout_cfg_t *p_cfg;
+        config_chain_t *p_cfg;
 
         SDPHandleUrl( p_stream, val.psz_string );
 
@@ -689,6 +692,13 @@ static void Close( vlc_object_t * p_this )
         free( p_sys->psz_sdp );
         p_sys->psz_sdp = NULL;
     }
+    if( p_sys->b_export_sdp_file )
+    {
+#ifdef HAVE_UNISTD_H
+        unlink( p_sys->psz_sdp_file );
+#endif
+        free( p_sys->psz_sdp_file );
+    }
     free( p_sys );
 }
 
@@ -806,6 +816,10 @@ static char *SDPGenerate( const sout_stream_t *p_stream,
         {
             i_size += strlen( "a=fmtp:* *\r\n" ) + strlen( id->psz_fmtp ) + 10;
         }
+        if ( id->i_bitrate)
+        {
+            i_size += strlen( "b=AS: *\r\n") + 10;
+        }
         if( b_rtsp )
         {
             i_size += strlen( "a=control:*/trackID=*\r\n" ) + strlen( p_sys->psz_rtsp_control ) + 10;
@@ -876,6 +890,10 @@ static char *SDPGenerate( const sout_stream_t *p_stream,
             p += sprintf( p, "a=fmtp:%d %s\r\n", id->i_payload_type,
                           id->psz_fmtp );
         }
+        if ( id->i_bitrate)
+        {
+            p += sprintf(p,"b=AS:%d\r\n",id->i_bitrate);
+        }
         if( b_rtsp )
         {
             p += sprintf( p, "a=control:trackID=%d\r\n", i );
@@ -902,6 +920,7 @@ static int rtp_packetize_split( sout_stream_t *, sout_stream_id_t *, block_t * )
 static int rtp_packetize_mp4a ( sout_stream_t *, sout_stream_id_t *, block_t * );
 static int rtp_packetize_mp4a_latm ( sout_stream_t *, sout_stream_id_t *, block_t * );
 static int rtp_packetize_h263 ( sout_stream_t *, sout_stream_id_t *, block_t * );
+static int rtp_packetize_h264 ( sout_stream_t *, sout_stream_id_t *, block_t * );
 static int rtp_packetize_amr  ( sout_stream_t *, sout_stream_id_t *, block_t * );
 
 static void sprintf_hexa( char *s, uint8_t *p_data, int i_data )
@@ -917,6 +936,68 @@ static void sprintf_hexa( char *s, uint8_t *p_data, int i_data )
     s[2*i_data] = '\0';
 }
 
+static const char basis_64[] =
+    "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
+
+int ap_base64encode_len(int len)
+{
+    return ((len + 2) / 3 * 4) + 1;
+}
+
+int ap_base64encode_binary(char *encoded,
+                                       const unsigned char *string, int len)
+{
+    int i;
+    char *p;
+
+    p = encoded;
+    for (i = 0; i < len - 2; i += 3) {
+        *p++ = basis_64[(string[i] >> 2) & 0x3F];
+        *p++ = basis_64[((string[i] & 0x3) << 4) |
+                        ((int) (string[i + 1] & 0xF0) >> 4)];
+        *p++ = basis_64[((string[i + 1] & 0xF) << 2) |
+                        ((int) (string[i + 2] & 0xC0) >> 6)];
+        *p++ = basis_64[string[i + 2] & 0x3F];
+    }
+    if (i < len) {
+        *p++ = basis_64[(string[i] >> 2) & 0x3F];
+        if (i == (len - 1)) {
+            *p++ = basis_64[((string[i] & 0x3) << 4)];
+            *p++ = '=';
+        }
+        else {
+            *p++ = basis_64[((string[i] & 0x3) << 4) |
+                            ((int) (string[i + 1] & 0xF0) >> 4)];
+            *p++ = basis_64[((string[i + 1] & 0xF) << 2)];
+        }
+        *p++ = '=';
+    }
+
+    *p++ = '\0';
+    return p - encoded;
+}
+
+int ap_base64encode(char *encoded, const char *string, int len)
+{
+    return ap_base64encode_binary(encoded, (const unsigned char *) string, len);
+}
+
+char *b64_encode(char *buf, int len)
+{
+    int elen;
+    char *out;
+
+    if(len == 0)
+        len = strlen(buf);
+
+    elen = ap_base64encode_len(len);
+    out = (char *) malloc(sizeof(char) * (elen + 1));
+
+    ap_base64encode(out, buf, len);
+
+    return out;
+}
+
 static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
 {
     sout_instance_t   *p_sout = p_stream->p_sout;
@@ -1067,6 +1148,68 @@ static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
             id->psz_rtpmap = strdup( "H263-1998/90000" );
             id->pf_packetize = rtp_packetize_h263;
             break;
+        case VLC_FOURCC( 'h', '2', '6', '4' ):
+            id->i_payload_type = p_sys->i_payload_type++;
+            id->i_clock_rate = 90000;
+            id->psz_rtpmap = strdup( "H264/90000" );
+            id->pf_packetize = rtp_packetize_h264;
+            if( p_fmt->i_extra > 0 )
+            {
+                uint8_t *p_buffer = p_fmt->p_extra;
+                int     i_buffer = p_fmt->i_extra;
+                char    *p_64_sps = NULL;
+                char    *p_64_pps = NULL;
+                char    hexa[6];
+                
+                while( i_buffer > 4 && 
+                    p_buffer[0] == 0 && p_buffer[1] == 0 &&
+                    p_buffer[2] == 0 && p_buffer[3] == 1 )
+                {
+                    const int i_nal_type = p_buffer[4]&0x1f;
+                    int i_offset    = 1;
+                    int i_size      = 0;
+                    int i_startcode = 0;
+                    int i_encoded   = 0;
+                    
+                    msg_Dbg( p_stream, "we found a startcode for NAL with TYPE:%d", i_nal_type );
+                    
+                    for( i_offset = 1; i_offset+3 < i_buffer ; i_offset++)
+                    {
+                        if( p_buffer[i_offset] == 0 && p_buffer[i_offset+1] == 0 && p_buffer[i_offset+2] == 0 && p_buffer[i_offset+3] == 1 )
+                        {
+                            /* we found another startcode */
+                            i_startcode = i_offset;
+                            break;
+                        } 
+                    }
+                    i_size = i_startcode ? i_startcode : i_buffer;
+                    if( i_nal_type == 7 )
+                    {
+                        p_64_sps = (char *)malloc( ap_base64encode_len( i_size - 4) );
+                        i_encoded = ap_base64encode_binary( p_64_sps, &p_buffer[4], i_size - 4 );
+                        p_64_sps[i_encoded] = '\0';
+                        sprintf_hexa( hexa, &p_buffer[5], 3 );
+                        hexa[6] = '\0'; 
+                    }
+                    if( i_nal_type == 8 )
+                    {
+                        p_64_pps = (char *)malloc( ap_base64encode_len( i_size - 4) );
+                        i_encoded = ap_base64encode_binary( p_64_pps, &p_buffer[4], i_size - 4 );
+                        p_64_pps[i_encoded] = '\0';
+                    }
+                    i_buffer -= i_size;
+                    p_buffer += i_size;
+                }
+                /* FIXME: All this is a bit unsafe */
+                asprintf( &id->psz_fmtp, "packetization-mode=1;profile-level-id=%s;sprop-parameter-sets=%s,%s;", hexa, p_64_sps, p_64_pps );
+                free( p_64_sps );
+                free( p_64_pps );
+            }
+            else
+                id->psz_fmtp = strdup( "packetization-mode=1" );
+if( p_fmt->i_extra > 0 )
+msg_Dbg( p_stream, "WE HAVE %d bytes extra data", p_fmt->i_extra );
+            break;
 
         case VLC_FOURCC( 'm', 'p', '4', 'v' ):
         {
@@ -1168,6 +1311,7 @@ static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
     id->ssrc[3] = rand()&0xff;
     id->i_sequence = rand()&0xffff;
     id->i_timestamp_start = rand()&0xffffffff;
+    id->i_bitrate = p_fmt->i_bitrate/1000; /* Stream bitrate in kbps */
 
     id->i_mtu    = config_GetInt( p_stream, "mtu" );  /* XXX beuk */
     if( id->i_mtu <= 16 + MTU_REDUCE )
@@ -1590,7 +1734,7 @@ static int  RtspCallback( httpd_callback_sys_t *p_args,
             answer->i_status = 200;
             answer->psz_status = strdup( "OK" );
             httpd_MsgAdd( answer, "Content-type",  "%s", "application/sdp" );
-
+           httpd_MsgAdd( answer, "Content-Base",  "%s/", p_sys->psz_rtsp_control );
             answer->p_body = (uint8_t *)psz_sdp;
             answer->i_body = strlen( psz_sdp );
             break;
@@ -2334,6 +2478,108 @@ static int rtp_packetize_h263( sout_stream_t *p_stream, sout_stream_id_t *id,
     return VLC_SUCCESS;
 }
 
+/* rfc3984 */
+static int rtp_packetize_h264( sout_stream_t *p_stream, sout_stream_id_t *id,
+                               block_t *in )
+{
+    int     i_max   = id->i_mtu - 12; /* payload max in one packet */
+    uint8_t *p_data = in->p_buffer;
+    int     i_data  = in->i_buffer;
+    block_t *out;
+    int     i_nal_type;
+    int     i_payload;
+
+    while( i_data > 4 )
+    {
+        if( p_data[0] == 0x00 && p_data[1] == 0x00 && p_data[2] == 0x01 && /* startcode */
+            (p_data[3]&0x1f) > 0 && (p_data[3]&0x1f) < 24 ) /* naltype should be between 1 and 23 */
+        {
+            p_data += 3;
+            i_data -= 3;
+            i_nal_type = p_data[0]&0x1f;
+
+            /* Skip global headers */
+            if( i_nal_type == 7 || i_nal_type == 8 )
+                continue;
+            
+            if( i_data <= i_max ) /* The whole pack will fit in one rtp payload */
+            {
+                /* single NAL */
+                i_payload = __MIN( i_max, i_data );
+                out = block_New( p_stream, 12 + i_payload );
+
+                /* rtp common header */
+                rtp_packetize_common( id, out, 1,
+                                      in->i_pts > 0 ? in->i_pts : in->i_dts );
+
+                memcpy( &out->p_buffer[12], p_data, i_payload );
+
+                out->i_buffer+= i_payload;
+                out->i_dts    = in->i_dts;
+                out->i_length = in->i_length;
+
+                rtp_packetize_send( id, out );
+                p_data += i_payload;
+                i_data -= i_payload;
+                /*msg_Dbg( p_stream, "nal-out plain %d %02x", i_payload, out->p_buffer[16] );*/
+            }
+            else
+            {
+                /* FU-A */
+                uint8_t     nalh; /* The nalheader byte */
+                int         start=1, end=0;
+                
+                /* skip but remember the nalh byte */
+                nalh = *p_data;
+                p_data++;
+                i_data--;
+
+                i_max   = id->i_mtu - 14;
+
+                /*msg_Dbg( p_stream, "nal-out fragmented %02x %d", nalh, i_rest);*/
+
+                while( end == 0 )
+                {
+                    i_payload = __MIN( i_max, i_data );
+                    out = block_New( p_stream, 14 + i_payload );
+
+                    if( i_data == i_payload )
+                        end = 1;
+
+                    /* rtp common header */
+                    rtp_packetize_common( id, out, (end)?1:0,
+                                      in->i_pts > 0 ? in->i_pts : in->i_dts );
+
+                    /* FU indicator */
+                    out->p_buffer[12] = (nalh&0x60)|28;
+                    /* FU header */
+                    out->p_buffer[13] = (start<<7)|(end<<6)|(nalh&0x1f);
+
+                    memcpy( &out->p_buffer[14], p_data, i_payload );
+         
+                    out->i_buffer   = 14 + i_payload;
+                    /* The dts should "slide" ? See the i_count trick and adapt */
+                    out->i_dts    = in->i_dts;
+                    
+                    rtp_packetize_send( id, out );
+                    /*msg_Dbg( p_stream, "nal-out fragmented: frag %d %d %02x %02x %d", start,end,
+                    out->p_buffer[12], out->p_buffer[13], i_payload );*/
+
+                    i_data -= i_payload;
+                    p_data +- i_payload;
+                    start = 0;
+                }
+            }
+        }
+        else
+        {
+            p_data++;
+            i_data--;
+        }
+    }
+    return VLC_SUCCESS;
+}
+
 static int rtp_packetize_amr( sout_stream_t *p_stream, sout_stream_id_t *id,
                               block_t *in )
 {