shout.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:20k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * shout.c: This module forwards vorbis streams to an icecast server
  3.  *****************************************************************************
  4.  * Copyright (C) 2005 the VideoLAN team
  5.  * $Id: 12a13d210608c41f5620254f0e2858e42e0e6f47 $
  6.  *
  7.  * Authors: Daniel Fischer <dan at subsignal dot org>
  8.  *          Derk-Jan Hartman <hartman at videolan dot org>
  9.  *
  10.  * This program is free software; you can redistribute it and/or modify
  11.  * it under the terms of the GNU General Public License as published by
  12.  * the Free Software Foundation; either version 2 of the License, or
  13.  * (at your option) any later version.
  14.  *
  15.  * This program is distributed in the hope that it will be useful,
  16.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18.  * GNU General Public License for more details.
  19.  *
  20.  * You should have received a copy of the GNU General Public License
  21.  * along with this program; if not, write to the Free Software
  22.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23.  *****************************************************************************/
  24. /*****************************************************************************
  25.  * Some Comments:
  26.  *
  27.  * - this only works for ogg and/or mp3, and we don't check this yet.
  28.  * - MP3 metadata is not passed along, since metadata is only available after
  29.  *   this module is opened.
  30.  *
  31.  * Typical usage:
  32.  *
  33.  * vlc v4l:/dev/video:input=2:norm=pal:size=192x144 
  34.  * --sout '#transcode{vcodec=theora,vb=300,acodec=vorb,ab=96}
  35.  * :std{access=shout,mux=ogg,dst=localhost:8005}'
  36.  *
  37.  *****************************************************************************/
  38. /*****************************************************************************
  39.  * Preamble
  40.  *****************************************************************************/
  41. #ifdef HAVE_CONFIG_H
  42. # include "config.h"
  43. #endif
  44. #include <vlc_common.h>
  45. #include <vlc_plugin.h>
  46. #include <vlc_sout.h>
  47. #include <vlc_block.h>
  48. #include <shout/shout.h>
  49. /*****************************************************************************
  50.  * Module descriptor
  51.  *****************************************************************************/
  52. static int  Open ( vlc_object_t * );
  53. static void Close( vlc_object_t * );
  54. #define SOUT_CFG_PREFIX "sout-shout-"
  55. #define NAME_TEXT N_("Stream name")
  56. #define NAME_LONGTEXT N_("Name to give to this stream/channel on the " 
  57.                          "shoutcast/icecast server." )
  58. #define DESCRIPTION_TEXT N_("Stream description")
  59. #define DESCRIPTION_LONGTEXT N_("Description of the stream content or " 
  60.                                 "information about your channel." )
  61. #define MP3_TEXT N_("Stream MP3")
  62. #define MP3_LONGTEXT N_("You normally have to feed the shoutcast module " 
  63.                         "with Ogg streams. It is also possible to stream " 
  64.                         "MP3 instead, so you can forward MP3 streams to " 
  65.                         "the shoutcast/icecast server." )
  66. /* To be listed properly as a public stream on the Yellow Pages of shoutcast/icecast
  67.    the genres should match those used on the corresponding sites. Several examples
  68.    are Alternative, Classical, Comedy, Country etc. */
  69. #define GENRE_TEXT N_("Genre description")
  70. #define GENRE_LONGTEXT N_("Genre of the content. " )
  71. #define URL_TEXT N_("URL description")
  72. #define URL_LONGTEXT N_("URL with information about the stream or your channel. " )
  73. /* The shout module only "transmits" data. It does not have direct access to
  74.    "codec level" information. Stream information such as bitrate, samplerate,
  75.    channel numbers and quality (in case of Ogg streaming) need to be set manually */
  76. #define BITRATE_TEXT N_("Bitrate")
  77. #define BITRATE_LONGTEXT N_("Bitrate information of the transcoded stream. " )
  78. #define SAMPLERATE_TEXT N_("Samplerate")
  79. #define SAMPLERATE_LONGTEXT N_("Samplerate information of the transcoded stream. " )
  80. #define CHANNELS_TEXT N_("Number of channels")
  81. #define CHANNELS_LONGTEXT N_("Number of channels information of the transcoded stream. " )
  82. #define QUALITY_TEXT N_("Ogg Vorbis Quality")
  83. #define QUALITY_LONGTEXT N_("Ogg Vorbis Quality information of the transcoded stream. " )
  84. #define PUBLIC_TEXT N_("Stream public")
  85. #define PUBLIC_LONGTEXT N_("Make the server publicly available on the 'Yellow Pages' " 
  86.                            "(directory listing of streams) on the icecast/shoutcast " 
  87.                            "website. Requires the bitrate information specified for " 
  88.                            "shoutcast. Requires Ogg streaming for icecast." )
  89. vlc_module_begin ()
  90.     set_description( N_("IceCAST output") )
  91.     set_shortname( "Shoutcast" )
  92.     set_capability( "sout access", 0 )
  93.     set_category( CAT_SOUT )
  94.     set_subcategory( SUBCAT_SOUT_ACO )
  95.     add_shortcut( "shout" )
  96.     add_string( SOUT_CFG_PREFIX "name", "VLC media player - Live stream", NULL,
  97.                 NAME_TEXT, NAME_LONGTEXT, false )
  98.     add_string( SOUT_CFG_PREFIX "description",
  99.                  "Live stream from VLC media player", NULL,
  100.                 DESCRIPTION_TEXT, DESCRIPTION_LONGTEXT, false )
  101.     add_bool(   SOUT_CFG_PREFIX "mp3", false, NULL,
  102.                 MP3_TEXT, MP3_LONGTEXT, true )
  103.     add_string( SOUT_CFG_PREFIX "genre", "Alternative", NULL,
  104.                 GENRE_TEXT, GENRE_LONGTEXT, false )
  105.     add_string( SOUT_CFG_PREFIX "url", "http://www.videolan.org/vlc", NULL,
  106.                 URL_TEXT, URL_LONGTEXT, false )
  107.     add_string( SOUT_CFG_PREFIX "bitrate", "", NULL,
  108.                 BITRATE_TEXT, BITRATE_LONGTEXT, false )
  109.     add_string( SOUT_CFG_PREFIX "samplerate", "", NULL,
  110.                 SAMPLERATE_TEXT, SAMPLERATE_LONGTEXT, false )
  111.     add_string( SOUT_CFG_PREFIX "channels", "", NULL,
  112.                 CHANNELS_TEXT, CHANNELS_LONGTEXT, false )
  113.     add_string( SOUT_CFG_PREFIX "quality", "", NULL,
  114.                 QUALITY_TEXT, QUALITY_LONGTEXT, false )
  115.     add_bool(   SOUT_CFG_PREFIX "public", false, NULL,
  116.                 PUBLIC_TEXT, PUBLIC_LONGTEXT, true )
  117.     set_callbacks( Open, Close )
  118. vlc_module_end ()
  119. /*****************************************************************************
  120.  * Exported prototypes
  121.  *****************************************************************************/
  122. static const char *const ppsz_sout_options[] = {
  123.     "name", "description", "mp3", "genre", "url", "bitrate", "samplerate",
  124.     "channels", "quality", "public", NULL
  125. };
  126. /*****************************************************************************
  127.  * Exported prototypes
  128.  *****************************************************************************/
  129. static ssize_t Write( sout_access_out_t *, block_t * );
  130. static int Seek ( sout_access_out_t *, off_t  );
  131. static int Control( sout_access_out_t *, int, va_list );
  132. struct sout_access_out_sys_t
  133. {
  134.     shout_t *p_shout;
  135. };
  136. /*****************************************************************************
  137.  * Open: open the shout connection
  138.  *****************************************************************************/
  139. static int Open( vlc_object_t *p_this )
  140. {
  141.     sout_access_out_t *p_access = (sout_access_out_t*)p_this;
  142.     sout_access_out_sys_t *p_sys;
  143.     shout_t *p_shout;
  144.     long i_ret;
  145.     unsigned int i_port;
  146.     vlc_value_t val;
  147.     char *psz_accessname = NULL;
  148.     char *psz_parser = NULL;
  149.     char *psz_user = NULL;
  150.     char *psz_pass = NULL;
  151.     char *psz_host = NULL;
  152.     char *psz_mount = NULL;
  153.     char *psz_name = NULL;
  154.     char *psz_description = NULL;
  155.     char *tmp_port = NULL;
  156.     char *psz_genre = NULL;
  157.     char *psz_url = NULL;
  158.     config_ChainParse( p_access, SOUT_CFG_PREFIX, ppsz_sout_options, p_access->p_cfg );
  159.     if( !p_access->psz_path )
  160.     {
  161.         msg_Err( p_access,
  162.                  "please specify url=user:password@host:port/mountpoint" );
  163.         return VLC_EGENERIC;
  164.     }
  165.     psz_accessname = psz_parser = strdup( p_access->psz_path );
  166.     if( !psz_parser )
  167.         return VLC_ENOMEM;
  168.     /* Parse connection data user:pwd@host:port/mountpoint */
  169.     psz_user = psz_parser;
  170.     while( psz_parser[0] && psz_parser[0] != ':' ) psz_parser++;
  171.     if( psz_parser[0] ) { psz_parser[0] = 0; psz_parser++; }
  172.     psz_pass = psz_parser;
  173.     while( psz_parser[0] && psz_parser[0] != '@' ) psz_parser++;
  174.     if( psz_parser[0] ) { psz_parser[0] = 0; psz_parser++; }
  175.     psz_host = psz_parser;
  176.     while( psz_parser[0] && psz_parser[0] != ':' ) psz_parser++;
  177.     if( psz_parser[0] ) { psz_parser[0] = 0; psz_parser++; }
  178.     tmp_port = psz_parser;
  179.     while( psz_parser[0] && psz_parser[0] != '/' ) psz_parser++;
  180.     if( psz_parser[0] ) { psz_parser[0] = 0; psz_parser++; }
  181.     psz_mount = psz_parser;
  182.     i_port = atoi( tmp_port );
  183.     p_sys = p_access->p_sys = malloc( sizeof( sout_access_out_sys_t ) );
  184.     if( !p_sys )
  185.     {
  186.         free( psz_accessname );
  187.         return VLC_ENOMEM;
  188.     }
  189.     var_Get( p_access, SOUT_CFG_PREFIX "name", &val );
  190.     if( *val.psz_string )
  191.         psz_name = val.psz_string;
  192.     else
  193.         free( val.psz_string );
  194.     var_Get( p_access, SOUT_CFG_PREFIX "description", &val );
  195.     if( *val.psz_string )
  196.         psz_description = val.psz_string;
  197.     else
  198.         free( val.psz_string );
  199.     var_Get( p_access, SOUT_CFG_PREFIX "genre", &val );
  200.     if( *val.psz_string )
  201.         psz_genre = val.psz_string;
  202.     else
  203.         free( val.psz_string );
  204.     var_Get( p_access, SOUT_CFG_PREFIX "url", &val );
  205.     if( *val.psz_string )
  206.         psz_url = val.psz_string;
  207.     else
  208.         free( val.psz_string );
  209.     p_shout = p_sys->p_shout = shout_new();
  210.     if( !p_shout
  211.          || shout_set_host( p_shout, psz_host ) != SHOUTERR_SUCCESS
  212.          || shout_set_protocol( p_shout, SHOUT_PROTOCOL_ICY ) != SHOUTERR_SUCCESS
  213.          || shout_set_port( p_shout, i_port ) != SHOUTERR_SUCCESS
  214.          || shout_set_password( p_shout, psz_pass ) != SHOUTERR_SUCCESS
  215.          || shout_set_mount( p_shout, psz_mount ) != SHOUTERR_SUCCESS
  216.          || shout_set_user( p_shout, psz_user ) != SHOUTERR_SUCCESS
  217.          || shout_set_agent( p_shout, "VLC media player " VERSION ) != SHOUTERR_SUCCESS
  218.          || shout_set_name( p_shout, psz_name ) != SHOUTERR_SUCCESS
  219.          || shout_set_description( p_shout, psz_description ) != SHOUTERR_SUCCESS
  220.          || shout_set_genre( p_shout, psz_genre ) != SHOUTERR_SUCCESS
  221.          || shout_set_url( p_shout, psz_url ) != SHOUTERR_SUCCESS
  222.          /* || shout_set_nonblocking( p_shout, 1 ) != SHOUTERR_SUCCESS */
  223.       )
  224.     {
  225.         msg_Err( p_access, "failed to initialize shout streaming to %s:%i/%s",
  226.                  psz_host, i_port, psz_mount );
  227.         free( p_access->p_sys );
  228.         free( psz_accessname );
  229.         free( psz_name );
  230.         free( psz_description );
  231.         free( psz_genre );
  232.         free( psz_url );
  233.         return VLC_EGENERIC;
  234.     }
  235.     free( psz_name );
  236.     free( psz_description );
  237.     free( psz_genre );
  238.     free( psz_url );
  239.     var_Get( p_access, SOUT_CFG_PREFIX "mp3", &val );
  240.     if( val.b_bool == true )
  241.         i_ret = shout_set_format( p_shout, SHOUT_FORMAT_MP3 );
  242.     else
  243.         i_ret = shout_set_format( p_shout, SHOUT_FORMAT_OGG );
  244.     if( i_ret != SHOUTERR_SUCCESS )
  245.     {
  246.         msg_Err( p_access, "failed to set the shoutcast streaming format" );
  247.         free( p_access->p_sys );
  248.         free( psz_accessname );
  249.         return VLC_EGENERIC;
  250.     }
  251.     /* Don't force bitrate to 0 but only use when specified. This will otherwise
  252.        show an empty field on icecast directory listing instead of NA */
  253.     var_Get( p_access, SOUT_CFG_PREFIX "bitrate", &val );
  254.     if( *val.psz_string )
  255.     {
  256.         i_ret = shout_set_audio_info( p_shout, SHOUT_AI_BITRATE, val.psz_string );
  257.         if( i_ret != SHOUTERR_SUCCESS )
  258.         {
  259.             msg_Err( p_access, "failed to set the information about the bitrate" );
  260.             free( val.psz_string );
  261.             free( p_access->p_sys );
  262.             free( psz_accessname );
  263.             return VLC_EGENERIC;
  264.         }
  265.     }
  266.     else
  267.     {
  268.         /* Bitrate information is used for icecast/shoutcast servers directory
  269.            listings (sorting, stream info etc.) */
  270.         msg_Warn( p_access, "no bitrate information specified (required for listing " 
  271.                             "the server as public on the shoutcast website)" );
  272.         free( val.psz_string );
  273.     }
  274.     /* Information about samplerate, channels and quality will not be propagated
  275.        through the YP protocol for icecast to the public directory listing when
  276.        the icecast server is operating in shoutcast compatibility mode */
  277.     var_Get( p_access, SOUT_CFG_PREFIX "samplerate", &val );
  278.     if( *val.psz_string )
  279.     {
  280.         i_ret = shout_set_audio_info( p_shout, SHOUT_AI_SAMPLERATE, val.psz_string );
  281.         if( i_ret != SHOUTERR_SUCCESS )
  282.         {
  283.             msg_Err( p_access, "failed to set the information about the samplerate" );
  284.             free( val.psz_string );
  285.             free( p_access->p_sys );
  286.             free( psz_accessname );
  287.             return VLC_EGENERIC;
  288.         }
  289.     }
  290.     else
  291.         free( val.psz_string );
  292.     var_Get( p_access, SOUT_CFG_PREFIX "channels", &val );
  293.     if( *val.psz_string )
  294.     {
  295.         i_ret = shout_set_audio_info( p_shout, SHOUT_AI_CHANNELS, val.psz_string );
  296.         if( i_ret != SHOUTERR_SUCCESS )
  297.         {
  298.             msg_Err( p_access, "failed to set the information about the number of channels" );
  299.             free( val.psz_string );
  300.             free( p_access->p_sys );
  301.             free( psz_accessname );
  302.             return VLC_EGENERIC;
  303.         }
  304.     }
  305.     else
  306.         free( val.psz_string );
  307.     var_Get( p_access, SOUT_CFG_PREFIX "quality", &val );
  308.     if( *val.psz_string )
  309.     {
  310.         i_ret = shout_set_audio_info( p_shout, SHOUT_AI_QUALITY, val.psz_string );
  311.         if( i_ret != SHOUTERR_SUCCESS )
  312.         {
  313.             msg_Err( p_access, "failed to set the information about Ogg Vorbis quality" );
  314.             free( val.psz_string );
  315.             free( p_access->p_sys );
  316.             free( psz_accessname );
  317.             return VLC_EGENERIC;
  318.         }
  319.     }
  320.     else
  321.         free( val.psz_string );
  322.     var_Get( p_access, SOUT_CFG_PREFIX "public", &val );
  323.     if( val.b_bool == true )
  324.     {
  325.         i_ret = shout_set_public( p_shout, 1 );
  326.         if( i_ret != SHOUTERR_SUCCESS )
  327.         {
  328.             msg_Err( p_access, "failed to set the server status setting to public" );
  329.             free( p_access->p_sys );
  330.             free( psz_accessname );
  331.             return VLC_EGENERIC;
  332.         }
  333.     }
  334.     /* Connect at startup. Cycle through the possible protocols. */
  335.     i_ret = shout_get_connected( p_shout );
  336.     while ( i_ret != SHOUTERR_CONNECTED )
  337.     {
  338.         /* Shout parameters cannot be changed on an open connection */
  339.         i_ret = shout_close( p_shout );
  340.         if( i_ret == SHOUTERR_SUCCESS )
  341.         {
  342.             i_ret = SHOUTERR_UNCONNECTED;
  343.         }
  344.         /* Re-initialize for Shoutcast using ICY protocol. Not needed for initial connection
  345.            but it is when we are reconnecting after other protocol was tried. */
  346.         i_ret = shout_set_protocol( p_shout, SHOUT_PROTOCOL_ICY );
  347.         if( i_ret != SHOUTERR_SUCCESS )
  348.         {
  349.             msg_Err( p_access, "failed to set the protocol to 'icy'" );
  350.             free( p_access->p_sys );
  351.             free( psz_accessname );
  352.             return VLC_EGENERIC;
  353.         }
  354.         i_ret = shout_open( p_shout );
  355.         if( i_ret == SHOUTERR_SUCCESS )
  356.         {
  357.             i_ret = SHOUTERR_CONNECTED;
  358.             msg_Dbg( p_access, "connected using 'icy' (shoutcast) protocol" );
  359.         }
  360.         else
  361.         {
  362.             msg_Warn( p_access, "failed to connect using 'icy' (shoutcast) protocol" );
  363.             /* Shout parameters cannot be changed on an open connection */
  364.             i_ret = shout_close( p_shout );
  365.             if( i_ret == SHOUTERR_SUCCESS )
  366.             {
  367.                 i_ret = SHOUTERR_UNCONNECTED;
  368.             }
  369.             /* IceCAST using HTTP protocol */
  370.             i_ret = shout_set_protocol( p_shout, SHOUT_PROTOCOL_HTTP );
  371.             if( i_ret != SHOUTERR_SUCCESS )
  372.             {
  373.                 msg_Err( p_access, "failed to set the protocol to 'http'" );
  374.                 free( p_access->p_sys );
  375.                 free( psz_accessname );
  376.                 return VLC_EGENERIC;
  377.             }
  378.             i_ret = shout_open( p_shout );
  379.             if( i_ret == SHOUTERR_SUCCESS )
  380.             {
  381.                 i_ret = SHOUTERR_CONNECTED;
  382.                 msg_Dbg( p_access, "connected using 'http' (icecast 2.x) protocol" );
  383.             }
  384.             else
  385.                 msg_Warn( p_access, "failed to connect using 'http' (icecast 2.x) protocol " );
  386.         }
  387. /*
  388.         for non-blocking, use:
  389.         while( i_ret == SHOUTERR_BUSY )
  390.         {
  391.             sleep( 1 );
  392.             i_ret = shout_get_connected( p_shout );
  393.         }
  394. */
  395.         if ( i_ret != SHOUTERR_CONNECTED )
  396.      {
  397.          msg_Warn( p_access, "unable to establish connection, retrying..." );
  398.             msleep( 30000000 );
  399.         }
  400.     }
  401.     if( i_ret != SHOUTERR_CONNECTED )
  402.     {
  403.         msg_Err( p_access, "failed to open shout stream to %s:%i/%s: %s",
  404.                  psz_host, i_port, psz_mount, shout_get_error(p_shout) );
  405.         free( p_access->p_sys );
  406.         free( psz_accessname );
  407.         return VLC_EGENERIC;
  408.     }
  409.     p_access->pf_write = Write;
  410.     p_access->pf_seek  = Seek;
  411.     p_access->pf_control = Control;
  412.     msg_Dbg( p_access, "shout access output opened (%s@%s:%i/%s)",
  413.              psz_user, psz_host, i_port, psz_mount );
  414.     free( psz_accessname );
  415.     return VLC_SUCCESS;
  416. }
  417. /*****************************************************************************
  418.  * Close: close the target
  419.  *****************************************************************************/
  420. static void Close( vlc_object_t * p_this )
  421. {
  422.     sout_access_out_t *p_access = (sout_access_out_t*)p_this;
  423.     if( p_access->p_sys && p_access->p_sys->p_shout )
  424.     {
  425.         shout_close( p_access->p_sys->p_shout );
  426.         shout_shutdown();
  427.     }
  428.     free( p_access->p_sys );
  429.     msg_Dbg( p_access, "shout access output closed" );
  430. }
  431. static int Control( sout_access_out_t *p_access, int i_query, va_list args )
  432. {
  433.     switch( i_query )
  434.     {
  435.         case ACCESS_OUT_CONTROLS_PACE:
  436.         {
  437.             bool *pb = va_arg( args, bool * );
  438.             *pb = strcmp( p_access->psz_access, "stream" );
  439.             break;
  440.         }
  441.         default:
  442.             return VLC_EGENERIC;
  443.     }
  444.     return VLC_SUCCESS;
  445. }
  446. /*****************************************************************************
  447.  * Write: standard write
  448.  *****************************************************************************/
  449. static ssize_t Write( sout_access_out_t *p_access, block_t *p_buffer )
  450. {
  451.     size_t i_write = 0;
  452.     shout_sync( p_access->p_sys->p_shout );
  453.     while( p_buffer )
  454.     {
  455.         block_t *p_next = p_buffer->p_next;
  456.         if( shout_send( p_access->p_sys->p_shout,
  457.                         p_buffer->p_buffer, p_buffer->i_buffer )
  458.              == SHOUTERR_SUCCESS )
  459.         {
  460.             i_write += p_buffer->i_buffer;
  461.         }
  462.         else
  463.         {
  464.             msg_Err( p_access, "cannot write to stream: %s",
  465.                      shout_get_error(p_access->p_sys->p_shout) );
  466.             /* The most common cause seems to be a server disconnect, resulting in a
  467.                Socket Error which can only be fixed by closing and reconnecting.
  468.                Since we already began with a working connection, the most feasable
  469.                approach to get out of this error status is a (timed) reconnect approach. */
  470.             shout_close( p_access->p_sys->p_shout );
  471.             msg_Warn( p_access, "server unavailable? trying to reconnect..." );
  472.             /* Re-open the connection (protocol params have already been set) and re-sync */
  473.             if( shout_open( p_access->p_sys->p_shout ) == SHOUTERR_SUCCESS )
  474.             {
  475.                 shout_sync( p_access->p_sys->p_shout );
  476.                 msg_Warn( p_access, "reconnected to server" );
  477.             }
  478.             else
  479.             {
  480.                 msg_Err( p_access, "failed to reconnect to server" );
  481.                 block_ChainRelease (p_buffer);
  482.                 return VLC_EGENERIC;
  483.             }
  484.         }
  485.         block_Release( p_buffer );
  486.         /* XXX: Unsure if that's the cause for some audio trouble... */
  487.         p_buffer = p_next;
  488.     }
  489.     return i_write;
  490. }
  491. /*****************************************************************************
  492.  * Seek: seek to a specific location -- not supported
  493.  *****************************************************************************/
  494. static int Seek( sout_access_out_t *p_access, off_t i_pos )
  495. {
  496.     msg_Err( p_access, "cannot seek on shout" );
  497.     return VLC_EGENERIC;
  498. }