# include "config.h"
#endif
-#include <vlc/vlc.h>
+#include <vlc_common.h>
+#include <vlc_plugin.h>
#include <vlc_access.h>
#include <vlc_network.h> /* DOWN: #include <network.h> */
static void Close( vlc_object_t * );
vlc_module_begin();
- set_description( _("RTMP input") );
- set_shortname( _("RTMP") );
+ set_description( N_("RTMP input") );
+ set_shortname( N_("RTMP") );
set_category( CAT_INPUT );
set_subcategory( SUBCAT_INPUT_ACCESS );
STANDARD_READ_ACCESS_INIT
+ p_sys->p_thread =
+ vlc_object_create( p_access, sizeof( rtmp_control_thread_t ) );
+ if( !p_sys->p_thread )
+ return VLC_ENOMEM;
+ vlc_object_attach( p_sys->p_thread, p_access );
+
/* Parse URI - remove spaces */
p = psz = strdup( p_access->psz_path );
while( (p = strchr( p, ' ' )) != NULL )
*p = '+';
- vlc_UrlParse( &p_sys->url, psz, 0 );
+ vlc_UrlParse( &p_sys->p_thread->url, psz, 0 );
free( psz );
- if( !p_access->psz_access ||
- strncmp( p_access->psz_access, "rtmp", 4 ))
- {
- msg_Warn( p_access, "invalid protocol" );
- vlc_UrlClean( &p_sys->url );
- free( p_sys );
- return VLC_EGENERIC;
- }
-
- if( p_sys->url.psz_host == NULL || *p_sys->url.psz_host == '\0' )
+ if( p_sys->p_thread->url.psz_host == NULL
+ || *p_sys->p_thread->url.psz_host == '\0' )
{
- msg_Warn( p_access, "invalid host" );
- vlc_UrlClean( &p_sys->url );
- free( p_sys );
- return VLC_EGENERIC;
+ msg_Warn( p_access, "invalid host" );
+ goto error;
}
- if( p_sys->url.i_port <= 0 )
- p_sys->url.i_port = 1935;
+ if( p_sys->p_thread->url.i_port <= 0 )
+ p_sys->p_thread->url.i_port = 1935;
- if ( p_sys->url.psz_path == NULL ) {
+ if( p_sys->p_thread->url.psz_path == NULL )
+ {
msg_Warn( p_access, "invalid path" );
- vlc_UrlClean( &p_sys->url );
- free( p_sys );
- return VLC_EGENERIC;
+ goto error;
}
- length_path = strlen( p_sys->url.psz_path );
- length_media_name = strlen( strrchr( p_sys->url.psz_path, '/' ) ) - 1;
+ length_path = strlen( p_sys->p_thread->url.psz_path );
+ length_media_name = strlen( strrchr( p_sys->p_thread->url.psz_path, '/' ) ) - 1;
- p_sys->psz_application = strndup( p_sys->url.psz_path + 1, length_path - length_media_name - 2 );
- p_sys->psz_media = strdup( p_sys->url.psz_path + ( length_path - length_media_name ) );
+ p_sys->p_thread->psz_application = strndup( p_sys->p_thread->url.psz_path + 1, length_path - length_media_name - 2 );
+ p_sys->p_thread->psz_media = strdup( p_sys->p_thread->url.psz_path + ( length_path - length_media_name ) );
msg_Dbg( p_access, "rtmp: host='%s' port=%d path='%s'",
- p_sys->url.psz_host, p_sys->url.i_port, p_sys->url.psz_path );
+ p_sys->p_thread->url.psz_host, p_sys->p_thread->url.i_port, p_sys->p_thread->url.psz_path );
- if( p_sys->url.psz_username && *p_sys->url.psz_username )
+ if( p_sys->p_thread->url.psz_username && *p_sys->p_thread->url.psz_username )
{
msg_Dbg( p_access, " user='%s', pwd='%s'",
- p_sys->url.psz_username, p_sys->url.psz_password );
- }
-
- p_sys->p_thread =
- vlc_object_create( p_access, sizeof( rtmp_control_thread_t ) );
- if( !p_sys->p_thread )
- {
- msg_Err( p_access, "out of memory" );
- vlc_UrlClean( &p_sys->url );
- free( p_sys );
- return VLC_EGENERIC;
+ p_sys->p_thread->url.psz_username, p_sys->p_thread->url.psz_password );
}
- vlc_object_attach( p_sys->p_thread, p_access );
+ /* Initialize thread variables */
p_sys->p_thread->b_die = 0;
p_sys->p_thread->b_error= 0;
- p_sys->p_thread->p_fifo_media = block_FifoNew();
+ p_sys->p_thread->p_fifo_input = block_FifoNew();
p_sys->p_thread->p_empty_blocks = block_FifoNew();
p_sys->p_thread->has_audio = 0;
p_sys->p_thread->has_video = 0;
p_sys->p_thread->metadata_received = 0;
p_sys->p_thread->first_media_packet = 1;
p_sys->p_thread->flv_tag_previous_tag_size = 0x00000000; /* FLV_TAG_FIRST_PREVIOUS_TAG_SIZE */
+ p_sys->p_thread->chunk_size_recv = 128; /* RTMP_DEFAULT_CHUNK_SIZE */
+ p_sys->p_thread->chunk_size_send = 128; /* RTMP_DEFAULT_CHUNK_SIZE */
for(i = 0; i < 64; i++)
{
memset( &p_sys->p_thread->rtmp_headers_recv[i], 0, sizeof( rtmp_packet_t ) );
p_sys->p_thread->rtmp_headers_send[i].body = NULL;
}
+ p_sys->p_thread->p_base_object = p_this;
+
vlc_cond_init( p_sys->p_thread, &p_sys->p_thread->wait );
+
vlc_mutex_init( &p_sys->p_thread->lock );
p_sys->p_thread->result_connect = 1;
p_sys->p_thread->result_play = 1;
+ p_sys->p_thread->result_stop = 0;
/* Open connection */
- p_sys->fd = net_ConnectTCP( p_access, p_sys->url.psz_host, p_sys->url.i_port );
- p_sys->p_thread->fd = p_sys->fd;
- if( p_sys->fd == -1 )
+ p_sys->p_thread->fd = net_ConnectTCP( p_access, p_sys->p_thread->url.psz_host, p_sys->p_thread->url.i_port );
+ if( p_sys->p_thread->fd == -1 )
{
int *p_fd_listen;
- msg_Warn( p_access, "cannot connect to %s:%d", p_sys->url.psz_host, p_sys->url.i_port );
+ msg_Warn( p_access, "cannot connect to %s:%d", p_sys->p_thread->url.psz_host, p_sys->p_thread->url.i_port );
msg_Dbg( p_access, "switching to passive mode" );
p_sys->active = 0;
- p_fd_listen = net_ListenTCP( p_access, p_sys->url.psz_host, p_sys->url.i_port );
+ p_fd_listen = net_ListenTCP( p_access, p_sys->p_thread->url.psz_host, p_sys->p_thread->url.i_port );
if( p_fd_listen == NULL )
{
- msg_Warn( p_access, "cannot listen to %s port %i", p_sys->url.psz_host, p_sys->url.i_port );
- vlc_UrlClean( &p_sys->url );
- net_Close( p_sys-> fd );
- free( p_sys );
- return VLC_EGENERIC;
+ msg_Err( p_access, "cannot listen to %s port %i", p_sys->p_thread->url.psz_host, p_sys->p_thread->url.i_port );
+ goto error2;
}
- p_sys->fd = net_Accept( p_access, p_fd_listen, -1 );
+ p_sys->p_thread->fd = net_Accept( p_access, p_fd_listen, -1 );
net_ListenClose( p_fd_listen );
- if( rtmp_handshake_passive( p_this ) < 0 )
+ if( rtmp_handshake_passive( p_this, p_sys->p_thread->fd ) < 0 )
{
- msg_Err( p_access, "Passive handshake failed");
- vlc_UrlClean( &p_sys->url );
- net_Close( p_sys-> fd );
- free( p_sys );
- return VLC_EGENERIC;
+ msg_Err( p_access, "handshake passive failed");
+ goto error2;
}
p_sys->p_thread->result_publish = 1;
-
}
else
{
- msg_Dbg( p_access, "using active connection");
p_sys->active = 1;
- if( rtmp_handshake_active( p_this ) < 0 )
+ if( rtmp_handshake_active( p_this, p_sys->p_thread->fd ) < 0 )
{
- msg_Err( p_access, "Active handshake failed");
- vlc_UrlClean( &p_sys->url );
- net_Close( p_sys-> fd );
- free( p_sys );
- return VLC_EGENERIC;
+ msg_Err( p_access, "handshake active failed");
+ goto error2;
}
p_sys->p_thread->result_publish = 0;
VLC_THREAD_PRIORITY_INPUT, false ) )
{
msg_Err( p_access, "cannot spawn rtmp control thread" );
- vlc_UrlClean( &p_sys->url );
- net_Close( p_sys-> fd );
- free( p_sys );
- return VLC_EGENERIC;
+ goto error2;
}
- if( p_sys->active )
+ if( p_sys->active )
{
- msg_Dbg( p_access, "Activation active connection");
- if( rtmp_connect_active( p_this ) < 0)
+ if( rtmp_connect_active( p_sys->p_thread ) < 0 )
{
- msg_Err( p_access, "Active connection failed");
- vlc_UrlClean( &p_sys->url );
- net_Close( p_sys-> fd );
- free( p_sys );
- return VLC_EGENERIC;
+ msg_Err( p_access, "connect active failed");
+ goto error2;
}
}
p_access->p_sys->flv_packet = NULL;
p_access->p_sys->read_packet = 1;
- msg_Dbg( p_access, "waiting for buffer to fill");
- /* Wait until enough data is received for extracting metadata */
- while( block_FifoCount( p_access->p_sys->p_thread->p_fifo_media ) < 10 )
- {
- msg_Dbg( p_access, "waiting for buffer to fill");
- msleep(1000);
- continue;
- }
-
/* Update default_pts to a suitable value for rtmp access */
var_Create( p_access, "rtmp-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
return VLC_SUCCESS;
+
+error2:
+ vlc_cond_destroy( &p_sys->p_thread->wait );
+ vlc_mutex_destroy( &p_sys->p_thread->lock );
+
+ free( p_sys->p_thread->psz_application );
+ free( p_sys->p_thread->psz_media );
+
+ net_Close( p_sys->p_thread->fd );
+error:
+ vlc_object_detach( p_sys->p_thread );
+ vlc_object_release( p_sys->p_thread );
+
+ vlc_UrlClean( &p_sys->p_thread->url );
+ free( p_sys );
+
+ return VLC_EGENERIC;
}
/*****************************************************************************
{
access_t *p_access = (access_t *) p_this;
access_sys_t *p_sys = p_access->p_sys;
+ int i;
- msg_Warn(p_access, "Close");
-
+/* p_sys->p_thread->b_die = true;*/
vlc_object_kill( p_sys->p_thread );
- block_FifoWake( p_sys->p_thread->p_fifo_media );
+ block_FifoWake( p_sys->p_thread->p_fifo_input );
block_FifoWake( p_sys->p_thread->p_empty_blocks );
- /*
- for( i = 0; i < 5; i++ )
- {
- block_t *p_dummy = block_New( p_access, 256 );
- p_dummy->i_dts = 0;
- p_dummy->i_pts = 0;
- p_dummy->i_length = 0;
- memset( p_dummy->p_buffer, 0, p_dummy->i_buffer );
- block_FifoPut( p_sys->p_thread->p_fifo_media, p_dummy );
- }
- for( i = 0; i < 5; i++ )
- {
- block_t *p_dummy = block_New( p_access, 256 );
- p_dummy->i_dts = 0;
- p_dummy->i_pts = 0;
- p_dummy->i_length = 0;
- memset( p_dummy->p_buffer, 0, p_dummy->i_buffer );
- block_FifoPut( p_sys->p_thread->p_empty_blocks, p_dummy );
- }*/
+
vlc_thread_join( p_sys->p_thread );
vlc_cond_destroy( &p_sys->p_thread->wait );
vlc_mutex_destroy( &p_sys->p_thread->lock );
- block_FifoRelease( p_sys->p_thread->p_fifo_media );
+ block_FifoRelease( p_sys->p_thread->p_fifo_input );
block_FifoRelease( p_sys->p_thread->p_empty_blocks );
- net_Close( p_sys->fd );
+ for( i = 0; i < 64; i++ ) /* RTMP_HEADER_STREAM_INDEX_MASK */
+ {
+ if( p_sys->p_thread->rtmp_headers_recv[i].body != NULL )
+ {
+ free( p_sys->p_thread->rtmp_headers_recv[i].body->body );
+ free( p_sys->p_thread->rtmp_headers_recv[i].body );
+ }
+ }
+
+ net_Close( p_sys->p_thread->fd );
var_Destroy( p_access, "rtmp-caching" );
+ vlc_object_detach( p_sys->p_thread );
+ vlc_object_release( p_sys->p_thread );
- vlc_UrlClean( &p_sys->url );
- free( p_sys->psz_application );
- free( p_sys->psz_media );
+ vlc_UrlClean( &p_sys->p_thread->url );
+ free( p_sys->p_thread->psz_application );
+ free( p_sys->p_thread->psz_media );
free( p_sys );
}
static int Read( access_t *p_access, uint8_t *p_buffer, size_t i_len )
{
access_sys_t *p_sys = p_access->p_sys;
- int i_len_tmp;
-
- if( p_sys->fd < 0 )
- {
- p_access->info.b_eof = true;
- return 0;
- }
+ rtmp_packet_t *rtmp_packet;
+ uint8_t *tmp_buffer;
+ ssize_t i_ret;
+ size_t i_len_tmp;
i_len_tmp = 0;
while( i_len_tmp < i_len )
{
+ if( p_sys->p_thread->result_stop || p_access->info.b_eof || p_access->b_die )
+ {
+ p_access->info.b_eof = true;
+ return 0;
+ }
+
if( p_sys->read_packet )
{
if( !p_sys->p_thread->metadata_received )
{
+ /* Wait until enough data is received for extracting metadata */
+ if( block_FifoCount( p_sys->p_thread->p_fifo_input ) < 10 )
+ {
+ msleep(100000);
+ continue;
+ }
+
p_sys->flv_packet = flv_get_metadata( p_access );
p_sys->p_thread->metadata_received = 1;
}
else
{
- if( p_sys->active && block_FifoCount( p_sys->p_thread->p_fifo_media ) == 0 )
- {
- p_access->info.b_eof = true;
- break;
- }
-
- p_sys->flv_packet = block_FifoGet( p_sys->p_thread->p_fifo_media );
+ p_sys->flv_packet = block_FifoGet( p_sys->p_thread->p_fifo_input );
if( p_sys->flv_packet == NULL )
continue; /* Forced wake-up */
}
i_len_tmp += i_len - i_len_tmp;
}
}
-/*int i;
-for(i = 0; i < i_len_tmp; i += 16)
-{
- msg_Warn(p_access,"%.2x%.2x %.2x%.2x %.2x%.2x %.2x%.2x %.2x%.2x %.2x%.2x %.2x%.2x %.2x%.2x",
-p_buffer[i], p_buffer[i+1], p_buffer[i+2], p_buffer[i+3], p_buffer[i+4], p_buffer[i+5], p_buffer[i+6], p_buffer[i+7],
-p_buffer[i+8], p_buffer[i+9], p_buffer[i+10], p_buffer[i+11], p_buffer[i+12], p_buffer[i+13], p_buffer[i+14], p_buffer[i+15]);
-}*/
- if( i_len_tmp > 0 ) {
+ if( i_len_tmp > 0 )
+ {
if( p_sys->p_thread->result_publish )
{
/* Send publish onStatus event only once */
p_sys->p_thread->result_publish = 0;
- rtmp_send_publish_start( p_access );
+ rtmp_packet = rtmp_build_publish_start( p_sys->p_thread );
+
+ tmp_buffer = rtmp_encode_packet( p_sys->p_thread, rtmp_packet );
+
+ i_ret = net_Write( p_sys->p_thread, p_sys->p_thread->fd, NULL, tmp_buffer, rtmp_packet->length_encoded );
+ if( i_ret != rtmp_packet->length_encoded )
+ {
+ free( rtmp_packet->body->body );
+ free( rtmp_packet->body );
+ free( rtmp_packet );
+ free( tmp_buffer );
+ msg_Err( p_access, "failed send publish start" );
+ return -1;
+ }
+ free( rtmp_packet->body->body );
+ free( rtmp_packet->body );
+ free( rtmp_packet );
+ free( tmp_buffer );
}
p_access->info.i_pos += i_len_tmp;
- rtmp_send_bytes_read( p_access, p_access->info.i_pos );
+ rtmp_packet = rtmp_build_bytes_read( p_sys->p_thread, p_access->info.i_pos );
+
+ tmp_buffer = rtmp_encode_packet( p_sys->p_thread, rtmp_packet );
+
+ i_ret = net_Write( p_sys->p_thread, p_sys->p_thread->fd, NULL, tmp_buffer, rtmp_packet->length_encoded );
+ if( i_ret != rtmp_packet->length_encoded )
+ {
+ free( rtmp_packet->body->body );
+ free( rtmp_packet->body );
+ free( rtmp_packet );
+ free( tmp_buffer );
+ msg_Err( p_access, "failed send bytes read" );
+ return -1;
+ }
+ free( rtmp_packet->body->body );
+ free( rtmp_packet->body );
+ free( rtmp_packet );
+ free( tmp_buffer );
}
return i_len_tmp;
*****************************************************************************/
static int Seek( access_t *p_access, int64_t i_pos )
{
+ VLC_UNUSED( p_access );
+ VLC_UNUSED( i_pos );
/*msg_Warn ( p_access, "Seek to %lld", i_pos);
switch( rtmp_seek( p_access, i_pos ) )
{
*****************************************************************************/
static int Control( access_t *p_access, int i_query, va_list args )
{
- bool *pb_bool;
- int *pi_int;
- int64_t *pi_64;
+ bool *pb_bool;
+ int *pi_int;
+ int64_t *pi_64;
switch( i_query )
{
/* */
case ACCESS_CAN_SEEK:
case ACCESS_CAN_FASTSEEK:
- pb_bool = (bool*)va_arg( args, bool* );
+ pb_bool = (bool*) va_arg( args, bool* );
*pb_bool = false; /* TODO */
break;
case ACCESS_CAN_PAUSE:
- pb_bool = (bool*)va_arg( args, bool* );
+ pb_bool = (bool*) va_arg( args, bool* );
*pb_bool = false; /* TODO */
break;
case ACCESS_CAN_CONTROL_PACE:
- pb_bool = (bool*)va_arg( args, bool* );
+ pb_bool = (bool*) va_arg( args, bool* );
*pb_bool = true;
break;
/* */
case ACCESS_GET_MTU:
- pi_int = (int*)va_arg( args, int * );
+ pi_int = (int*) va_arg( args, int * );
*pi_int = 0;
break;
case ACCESS_GET_PTS_DELAY:
- pi_64 = (int64_t*)va_arg( args, int64_t * );
+ pi_64 = (int64_t*) va_arg( args, int64_t * );
*pi_64 = var_GetInteger( p_access, "rtmp-caching" ) * INT64_C(1000);
break;
while( !p_thread->b_die )
{
-
rtmp_packet = rtmp_read_net_packet( p_thread );
if( rtmp_packet != NULL )
{
if( rtmp_packet->content_type < 0x01 /* RTMP_CONTENT_TYPE_CHUNK_SIZE */
|| rtmp_packet->content_type > 0x14 ) /* RTMP_CONTENT_TYPE_INVOKE */
+ {
+ free( rtmp_packet->body->body );
+ free( rtmp_packet->body );
+ free( rtmp_packet );
+
msg_Warn( p_thread, "unknown content type received" );
+ }
else
p_thread->rtmp_handler[rtmp_packet->content_type]( p_thread, rtmp_packet );
}
}
p_thread->b_die = 1;
+ ((access_t *) p_thread->p_base_object)->info.b_eof = true;
+
+ block_FifoWake( p_thread->p_fifo_input );
}
}
}