X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Fstream_out%2Frtp.c;h=36efc8f959bc6a84d148ed7f8d0ba9da6c758f30;hb=1071c1bc2829435ed5d42de68568fe05a108fe44;hp=fe9a8f10da9d9b6c405e4c2900f07e2734702300;hpb=4fa484f5461bcccaaf37957b0803efc0932fce81;p=vlc diff --git a/modules/stream_out/rtp.c b/modules/stream_out/rtp.c index fe9a8f10da..36efc8f959 100644 --- a/modules/stream_out/rtp.c +++ b/modules/stream_out/rtp.c @@ -2,8 +2,7 @@ * rtp.c: rtp stream output module ***************************************************************************** * Copyright (C) 2003-2004 the VideoLAN team - * Copyright © 2007 Rémi Denis-Courmont - * $Id$ + * Copyright © 2007-2008 Rémi Denis-Courmont * * Authors: Laurent Aimar * @@ -40,6 +39,7 @@ #include #include #include +#include #include "rtp.h" @@ -130,6 +130,15 @@ #define PROTO_LONGTEXT N_( \ "This selects which transport protocol to use for RTP." ) +#define SRTP_KEY_TEXT N_("SRTP key (hexadecimal)") +#define SRTP_KEY_LONGTEXT N_( \ + "RTP packets will be integrity-protected and ciphered "\ + "with this Secure RTP master shared secret key.") + +#define SRTP_SALT_TEXT N_("SRTP salt (hexadecimal)") +#define SRTP_SALT_LONGTEXT N_( \ + "Secure RTP requires a (non-secret) master salt value.") + static const char *const ppsz_protos[] = { "dccp", "sctp", "tcp", "udp", "udplite", }; @@ -192,6 +201,11 @@ vlc_module_begin(); add_bool( SOUT_CFG_PREFIX "rtcp-mux", false, NULL, RTCP_MUX_TEXT, RTCP_MUX_LONGTEXT, false ); + add_string( SOUT_CFG_PREFIX "key", "", NULL, + SRTP_KEY_TEXT, SRTP_KEY_LONGTEXT, false ); + add_string( SOUT_CFG_PREFIX "salt", "", NULL, + SRTP_SALT_TEXT, SRTP_SALT_LONGTEXT, false ); + add_bool( SOUT_CFG_PREFIX "mp4a-latm", 0, NULL, RFC3016_TEXT, RFC3016_LONGTEXT, false ); @@ -204,7 +218,7 @@ vlc_module_end(); static const char *const ppsz_sout_options[] = { "dst", "name", "port", "port-audio", "port-video", "*sdp", "ttl", "mux", "sap", "description", "url", "email", "phone", - "proto", "rtcp-mux", + "proto", "rtcp-mux", "key", "salt", "mp4a-latm", NULL }; @@ -220,7 +234,7 @@ static int MuxSend( sout_stream_t *, sout_stream_id_t *, static sout_access_out_t *GrabberCreate( sout_stream_t *p_sout ); static void ThreadSend( vlc_object_t *p_this ); -static void SDPHandleUrl( sout_stream_t *, char * ); +static void SDPHandleUrl( sout_stream_t *, const char * ); static int SapSetup( sout_stream_t *p_stream ); static int FileSetup( sout_stream_t *p_stream ); @@ -299,8 +313,9 @@ struct sout_stream_id_t int i_bitrate; /* Packetizer specific fields */ + int i_mtu; + srtp_session_t *srtp; pf_rtp_packetizer_t pf_packetize; - int i_mtu; /* Packets sinks */ vlc_mutex_t lock_sink; @@ -592,7 +607,7 @@ static void Close( vlc_object_t * p_this ) /***************************************************************************** * SDPHandleUrl: *****************************************************************************/ -static void SDPHandleUrl( sout_stream_t *p_stream, char *psz_url ) +static void SDPHandleUrl( sout_stream_t *p_stream, const char *psz_url ) { sout_stream_sys_t *p_sys = p_stream->p_sys; vlc_url_t url; @@ -902,13 +917,37 @@ static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt ) id->i_bitrate = 0; } - id->pf_packetize = NULL; id->i_mtu = config_GetInt( p_stream, "mtu" ); if( id->i_mtu <= 12 + 16 ) id->i_mtu = 576 - 20 - 8; /* pessimistic */ - msg_Dbg( p_stream, "maximum RTP packet size: %d bytes", id->i_mtu ); + id->srtp = NULL; + id->pf_packetize = NULL; + + char *key = var_CreateGetNonEmptyString (p_stream, SOUT_CFG_PREFIX"key"); + if (key) + { + id->srtp = srtp_create (SRTP_ENCR_AES_CM, SRTP_AUTH_HMAC_SHA1, 10, + SRTP_PRF_AES_CM, SRTP_RCC_MODE1); + if (id->srtp == NULL) + { + free (key); + goto error; + } + + char *salt = var_CreateGetNonEmptyString (p_stream, SOUT_CFG_PREFIX"salt"); + errno = srtp_setkeystring (id->srtp, key, salt ? salt : ""); + free (salt); + free (key); + if (errno) + { + msg_Err (p_stream, "bad SRTP key/salt combination (%m)"); + goto error; + } + id->i_sequence = 0; /* FIXME: awful hack for libvlc_srtp */ + } + vlc_mutex_init( &id->lock_sink ); id->sinkc = 0; id->sinkv = NULL; @@ -1016,6 +1055,7 @@ static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt ) case VLC_FOURCC( 'm', 'p', '3', ' ' ): id->i_payload_type = 14; id->psz_enc = "MPA"; + id->i_clock_rate = 90000; /* not 44100 */ id->pf_packetize = rtp_packetize_mpa; break; case VLC_FOURCC( 'm', 'p', 'g', 'v' ): @@ -1251,6 +1291,8 @@ static int Del( sout_stream_t *p_stream, sout_stream_id_t *id ) rtp_del_sink( id, id->sinkv[0].rtp_fd ); /* sink for explicit dst= */ if( id->listen_fd != NULL ) net_ListenClose( id->listen_fd ); + if( id->srtp != NULL ) + srtp_destroy( id->srtp ); vlc_mutex_destroy( &id->lock_sink ); @@ -1391,12 +1433,29 @@ 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; - while( !id->b_die ) + while( vlc_object_alive (id) ) { block_t *out = block_FifoGet( id->p_fifo ); if( out == NULL ) continue; /* Forced wakeup */ + if( id->srtp ) + { /* FIXME: this is awfully inefficient */ + size_t len = out->i_buffer; + out = block_Realloc( out, 0, len + 10 ); + out->i_buffer = len; + + int val = srtp_send( id->srtp, out->p_buffer, &len, len + 10 ); + if( val ) + { + errno = val; + msg_Dbg( id, "SRTP sending error: %m" ); + block_Release( out ); + continue; + } + out->i_buffer = len; + } + mtime_t i_date = out->i_dts + i_caching; ssize_t len = out->i_buffer; @@ -1408,7 +1467,8 @@ static void ThreadSend( vlc_object_t *p_this ) for( int i = 0; i < id->sinkc; i++ ) { - SendRTCP( id->sinkv[i].rtcp, out ); + if( !id->srtp ) /* FIXME: SRTCP support */ + SendRTCP( id->sinkv[i].rtcp, out ); if( send( id->sinkv[i].rtp_fd, out->p_buffer, len, 0 ) >= 0 ) continue; @@ -1578,8 +1638,8 @@ static int MuxDel( sout_stream_t *p_stream, sout_stream_id_t *id ) } -static int AccessOutGrabberWriteBuffer( sout_stream_t *p_stream, - const block_t *p_buffer ) +static ssize_t AccessOutGrabberWriteBuffer( sout_stream_t *p_stream, + const block_t *p_buffer ) { sout_stream_sys_t *p_sys = p_stream->p_sys; sout_stream_id_t *id = p_sys->es[0]; @@ -1587,14 +1647,14 @@ static int AccessOutGrabberWriteBuffer( sout_stream_t *p_stream, int64_t i_dts = p_buffer->i_dts; uint8_t *p_data = p_buffer->p_buffer; - unsigned int i_data = p_buffer->i_buffer; - unsigned int i_max = id->i_mtu - 12; + size_t i_data = p_buffer->i_buffer; + size_t i_max = id->i_mtu - 12; - unsigned i_packet = ( p_buffer->i_buffer + i_max - 1 ) / i_max; + size_t i_packet = ( p_buffer->i_buffer + i_max - 1 ) / i_max; while( i_data > 0 ) { - unsigned int i_size; + size_t i_size; /* output complete packet */ if( p_sys->packet && @@ -1629,8 +1689,8 @@ static int AccessOutGrabberWriteBuffer( sout_stream_t *p_stream, } -static int AccessOutGrabberWrite( sout_access_out_t *p_access, - block_t *p_buffer ) +static ssize_t AccessOutGrabberWrite( sout_access_out_t *p_access, + block_t *p_buffer ) { sout_stream_t *p_stream = (sout_stream_t*)p_access->p_sys;