]> git.sesse.net Git - vlc/blobdiff - modules/stream_out/rtpfmt.c
avcodec: set rc-buffer-size to 0, as it is default in ffmpeg/avconv
[vlc] / modules / stream_out / rtpfmt.c
index c31fd3286a8a108745481e198976764a3f73ccd6..79ae441226552ef486d410b82513bf8767596edb 100644 (file)
@@ -32,6 +32,7 @@
 #include <vlc_strings.h>
 
 #include "rtp.h"
+#include "../demux/xiph.h"
 
 #include <assert.h>
 
@@ -51,7 +52,110 @@ static int rtp_packetize_g726_16 (sout_stream_id_t *, block_t *);
 static int rtp_packetize_g726_24 (sout_stream_id_t *, block_t *);
 static int rtp_packetize_g726_32 (sout_stream_id_t *, block_t *);
 static int rtp_packetize_g726_40 (sout_stream_id_t *, block_t *);
-static int rtp_packetize_vorbis (sout_stream_id_t *, block_t *);
+static int rtp_packetize_xiph (sout_stream_id_t *, block_t *);
+
+#define XIPH_IDENT (0)
+
+/* Helpers common to xiph codecs (vorbis and theora) */
+
+static int rtp_xiph_pack_headers(size_t room, void *p_extra, size_t i_extra,
+                                 uint8_t **p_buffer, size_t *i_buffer,
+                                 uint8_t *theora_pixel_fmt)
+{
+    unsigned packet_size[XIPH_MAX_HEADER_COUNT];
+    void *packet[XIPH_MAX_HEADER_COUNT];
+    unsigned packet_count;
+    int val = xiph_SplitHeaders(packet_size, packet, &packet_count,
+                                i_extra, p_extra);
+    if (val != VLC_SUCCESS)
+        return val;
+    if (packet_count < 3)
+    {
+        val = VLC_EGENERIC;
+        goto free;
+    }
+
+    if (theora_pixel_fmt != NULL)
+    {
+        if (packet_size[0] < 42)
+        {
+            val = VLC_EGENERIC;
+            goto free;
+        }
+        *theora_pixel_fmt = (((uint8_t *)packet[0])[41] >> 3) & 0x03;
+    }
+
+    unsigned length_size[2] = { 0, 0 };
+    for (int i = 0; i < 2; i++)
+    {
+        unsigned size = packet_size[i];
+        while (size > 0)
+        {
+            length_size[i]++;
+            size >>= 7;
+        }
+    }
+
+    *i_buffer = room + 1 + length_size[0] + length_size[1]
+                + packet_size[0] + packet_size[1] + packet_size[2];
+    *p_buffer = malloc(*i_buffer);
+    if (*p_buffer == NULL)
+    {
+        val = VLC_ENOMEM;
+        goto free;
+    }
+
+    uint8_t *p = *p_buffer + room;
+    /* Number of headers */
+    *p++ = 2;
+
+    for (int i = 0; i < 2; i++)
+    {
+        unsigned size = length_size[i];
+        while (size > 0)
+        {
+            *p = (packet_size[i] >> (7 * (size - 1))) & 0x7f;
+            if (--size > 0)
+                *p |= 0x80;
+            p++;
+        }
+    }
+    for (int i = 0; i < 3; i++)
+    {
+        memcpy(p, packet[i], packet_size[i]);
+        p += packet_size[i];
+    }
+
+    val = VLC_SUCCESS;
+free:
+    for (unsigned i = 0; i < packet_count; i++)
+        free(packet[i]);
+
+    return val;
+}
+
+static char *rtp_xiph_b64_oob_config(void *p_extra, size_t i_extra,
+                                     uint8_t *theora_pixel_fmt)
+{
+    uint8_t *p_buffer;
+    size_t i_buffer;
+    if (rtp_xiph_pack_headers(9, p_extra, i_extra, &p_buffer, &i_buffer,
+                              theora_pixel_fmt) != VLC_SUCCESS)
+        return NULL;
+
+    /* Number of packed headers */
+    SetDWBE(p_buffer, 1);
+    /* Ident */
+    uint32_t ident = XIPH_IDENT;
+    SetWBE(p_buffer + 4, ident >> 8);
+    p_buffer[6] = ident & 0xff;
+    /* Length field */
+    SetWBE(p_buffer + 7, i_buffer);
+
+    char *config = vlc_b64_encode_binary(p_buffer, i_buffer);
+    free(p_buffer);
+    return config;
+}
 
 static void sprintf_hexa( char *s, uint8_t *p_data, int i_data )
 {
@@ -134,6 +238,10 @@ int rtp_get_fmt( vlc_object_t *obj, es_format_t *p_fmt, const char *mux,
             rtp_fmt->ptname = "L8";
             rtp_fmt->pf_packetize = rtp_packetize_split;
             break;
+        case VLC_CODEC_S24B:
+            rtp_fmt->ptname = "L24";
+            rtp_fmt->pf_packetize = rtp_packetize_split;
+            break;
         case VLC_CODEC_MPGA:
             rtp_fmt->payload_type = 14;
             rtp_fmt->ptname = "MPA";
@@ -337,6 +445,67 @@ int rtp_get_fmt( vlc_object_t *obj, es_format_t *p_fmt, const char *mux,
             rtp_fmt->ptname = "SPEEX";
             rtp_fmt->pf_packetize = rtp_packetize_spx;
             break;
+        case VLC_CODEC_VORBIS:
+            rtp_fmt->ptname = "vorbis";
+            rtp_fmt->pf_packetize = rtp_packetize_xiph;
+            if( p_fmt->i_extra > 0 )
+            {
+                rtp_fmt->fmtp = NULL;
+                char *config = rtp_xiph_b64_oob_config(p_fmt->p_extra,
+                                                       p_fmt->i_extra, NULL);
+                if (config == NULL)
+                    break;
+                if( asprintf( &rtp_fmt->fmtp,
+                              "configuration=%s;", config ) == -1 )
+                    rtp_fmt->fmtp = NULL;
+                free(config);
+            }
+            break;
+        case VLC_CODEC_THEORA:
+            rtp_fmt->ptname = "theora";
+            rtp_fmt->pf_packetize = rtp_packetize_xiph;
+            if( p_fmt->i_extra > 0 )
+            {
+                rtp_fmt->fmtp = NULL;
+                uint8_t pixel_fmt, c1, c2;
+                char *config = rtp_xiph_b64_oob_config(p_fmt->p_extra,
+                                                       p_fmt->i_extra,
+                                                       &pixel_fmt);
+                if (config == NULL)
+                    break;
+
+                if (pixel_fmt == 1)
+                {
+                    /* reserved */
+                    free(config);
+                    break;
+                }
+                switch (pixel_fmt)
+                {
+                    case 0:
+                        c1 = 2;
+                        c2 = 0;
+                        break;
+                    case 2:
+                        c1 = c2 = 2;
+                        break;
+                    case 3:
+                        c1 = c2 = 4;
+                        break;
+                    default:
+                        assert(0);
+                }
+
+                if( asprintf( &rtp_fmt->fmtp,
+                              "sampling=YCbCr-4:%d:%d; width=%d; height=%d; "
+                              "delivery-method=inline; configuration=%s; "
+                              "delivery-method=in_band;", c1, c2,
+                              p_fmt->video.i_width, p_fmt->video.i_height,
+                              config ) == -1 )
+                    rtp_fmt->fmtp = NULL;
+                free(config);
+            }
+            break;
         case VLC_CODEC_ITU_T140:
             rtp_fmt->ptname = "t140" ;
             rtp_fmt->clock_rate = 1000;
@@ -358,6 +527,139 @@ rtp_packetize_h264_nal( sout_stream_id_t *id,
                         const uint8_t *p_data, int i_data, int64_t i_pts,
                         int64_t i_dts, bool b_last, int64_t i_length );
 
+int rtp_packetize_xiph_config( sout_stream_id_t *id, const char *fmtp,
+                               int64_t i_pts )
+{
+    if (fmtp == NULL)
+        return VLC_EGENERIC;
+
+    /* extract base64 configuration from fmtp */
+    char *start = strstr(fmtp, "configuration=");
+    assert(start != NULL);
+    start += sizeof("configuration=") - 1;
+    char *end = strchr(start, ';');
+    assert(end != NULL);
+    size_t len = end - start;
+    char b64[len + 1];
+    memcpy(b64, start, len);
+    b64[len] = '\0';
+
+    int     i_max   = rtp_mtu (id) - 6; /* payload max in one packet */
+
+    uint8_t *p_orig, *p_data;
+    int i_data;
+
+    i_data = vlc_b64_decode_binary(&p_orig, b64);
+    if (i_data == 0)
+        return VLC_EGENERIC;
+    assert(i_data > 9);
+    p_data = p_orig + 9;
+    i_data -= 9;
+
+    int i_count = ( i_data + i_max - 1 ) / i_max;
+
+    for( int i = 0; i < i_count; i++ )
+    {
+        int           i_payload = __MIN( i_max, i_data );
+        block_t *out = block_Alloc( 18 + i_payload );
+
+        unsigned fragtype, numpkts;
+        if (i_count == 1)
+        {
+            fragtype = 0;
+            numpkts = 1;
+        }
+        else
+        {
+            numpkts = 0;
+            if (i == 0)
+                fragtype = 1;
+            else if (i == i_count - 1)
+                fragtype = 3;
+            else
+                fragtype = 2;
+        }
+        /* Ident:24, Fragment type:2, Vorbis/Theora Data Type:2, # of pkts:4 */
+        uint32_t header = ((XIPH_IDENT & 0xffffff) << 8) |
+                          (fragtype << 6) | (1 << 4) | numpkts;
+
+        /* rtp common header */
+        rtp_packetize_common( id, out, 0, i_pts );
+
+        SetDWBE( out->p_buffer + 12, header);
+        SetWBE( out->p_buffer + 16, i_payload);
+        memcpy( &out->p_buffer[18], p_data, i_payload );
+
+        out->i_buffer   = 18 + i_payload;
+        out->i_dts    = i_pts;
+
+        rtp_packetize_send( id, out );
+
+        p_data += i_payload;
+        i_data -= i_payload;
+    }
+
+    free(p_orig);
+
+    return VLC_SUCCESS;
+}
+
+/* rfc5215 */
+static int rtp_packetize_xiph( sout_stream_id_t *id, block_t *in )
+{
+    int     i_max   = rtp_mtu (id) - 6; /* payload max in one packet */
+    int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
+
+    uint8_t *p_data = in->p_buffer;
+    int     i_data  = in->i_buffer;
+
+    for( int i = 0; i < i_count; i++ )
+    {
+        int           i_payload = __MIN( i_max, i_data );
+        block_t *out = block_Alloc( 18 + i_payload );
+
+        unsigned fragtype, numpkts;
+        if (i_count == 1)
+        {
+            /* No fragmentation */
+            fragtype = 0;
+            numpkts = 1;
+        }
+        else
+        {
+            /* Fragmentation */
+            numpkts = 0;
+            if (i == 0)
+                fragtype = 1;
+            else if (i == i_count - 1)
+                fragtype = 3;
+            else
+                fragtype = 2;
+        }
+        /* Ident:24, Fragment type:2, Vorbis/Theora Data Type:2, # of pkts:4 */
+        uint32_t header = ((XIPH_IDENT & 0xffffff) << 8) |
+                          (fragtype << 6) | (0 << 4) | numpkts;
+
+        /* rtp common header */
+        rtp_packetize_common( id, out, 0, in->i_pts);
+
+        SetDWBE( out->p_buffer + 12, header);
+        SetWBE( out->p_buffer + 16, i_payload);
+        memcpy( &out->p_buffer[18], p_data, i_payload );
+
+        out->i_buffer   = 18 + i_payload;
+        out->i_dts    = in->i_dts + i * in->i_length / i_count;
+        out->i_length = in->i_length / i_count;
+
+        rtp_packetize_send( id, out );
+
+        p_data += i_payload;
+        i_data -= i_payload;
+    }
+
+    return VLC_SUCCESS;
+}
+
 static int rtp_packetize_mpa( sout_stream_id_t *id, block_t *in )
 {
     int     i_max   = rtp_mtu (id) - 4; /* payload max in one packet */
@@ -930,7 +1232,7 @@ static int rtp_packetize_t140( sout_stream_id_t *id, block_t *in )
         memcpy( out->p_buffer + 12, p_data, i_payload );
 
         out->i_buffer = 12 + i_payload;
-        out->i_dts    = out->i_pts;
+        out->i_dts    = in->i_pts;
         out->i_length = 0;
 
         rtp_packetize_send( id, out );