Commit aa5fe86a authored by Miguel Angel Cabrera Moya's avatar Miguel Angel Cabrera Moya Committed by Ilkka Ollakka

RTMP bugfix and reusability enhacenment

Signed-off-by: default avatarIlkka Ollakka <ileoo@videolan.org>
parent fb19b264
...@@ -82,77 +82,87 @@ static int Open( vlc_object_t *p_this ) ...@@ -82,77 +82,87 @@ static int Open( vlc_object_t *p_this )
int length_path, length_media_name; int length_path, length_media_name;
int i; int i;
/*DOWN:
p_access->info.i_update = 0;
p_access->info.i_size = 0;
p_access->info.i_pos = 0;
p_access->info.b_eof = false;
p_access->info.i_title = 0;
p_access->info.i_seekpoint = 0;
p_access->pf_read = Read;
p_access->pf_block = Block;
p_access->pf_control = Control;
p_access->pf_seek = Seek;
do
{
p_access->p_sys = (access_sys_t *) malloc( sizeof( access_sys_t ) );
if( !p_access->p_sys )
return VLC_ENOMEM;
} while(0);
p_sys = p_access->p_sys;
memset( p_sys, 0, sizeof( access_sys_t ) );
*/
STANDARD_READ_ACCESS_INIT STANDARD_READ_ACCESS_INIT
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" );
return VLC_ENOMEM;
}
vlc_object_attach( p_sys->p_thread, p_access );
/* Parse URI - remove spaces */ /* Parse URI - remove spaces */
p = psz = strdup( p_access->psz_path ); p = psz = strdup( p_access->psz_path );
while( (p = strchr( p, ' ' )) != NULL ) while( (p = strchr( p, ' ' )) != NULL )
*p = '+'; *p = '+';
vlc_UrlParse( &p_sys->url, psz, 0 ); vlc_UrlParse( &p_sys->p_thread->url, psz, 0 );
free( psz ); free( psz );
if( !p_access->psz_access || if( p_sys->p_thread->url.psz_host == NULL
strncmp( p_access->psz_access, "rtmp", 4 )) || *p_sys->p_thread->url.psz_host == '\0' )
{
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' )
{ {
msg_Warn( p_access, "invalid host" ); msg_Warn( p_access, "invalid host" );
vlc_UrlClean( &p_sys->url ); goto error;
free( p_sys );
return VLC_EGENERIC;
} }
if( p_sys->url.i_port <= 0 ) if( p_sys->p_thread->url.i_port <= 0 )
p_sys->url.i_port = 1935; 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" ); msg_Warn( p_access, "invalid path" );
vlc_UrlClean( &p_sys->url ); goto error;
free( p_sys );
return VLC_EGENERIC;
} }
length_path = strlen( p_sys->url.psz_path ); length_path = strlen( p_sys->p_thread->url.psz_path );
length_media_name = strlen( strrchr( p_sys->url.psz_path, '/' ) ) - 1; 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->p_thread->psz_application = strndup( p_sys->p_thread->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_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'", 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'", msg_Dbg( p_access, " user='%s', pwd='%s'",
p_sys->url.psz_username, p_sys->url.psz_password ); p_sys->p_thread->url.psz_username, p_sys->p_thread->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;
} }
vlc_object_attach( p_sys->p_thread, p_access ); /* Initialize thread variables */
p_sys->p_thread->b_die = 0; p_sys->p_thread->b_die = 0;
p_sys->p_thread->b_error= 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->p_empty_blocks = block_FifoNew();
p_sys->p_thread->has_audio = 0; p_sys->p_thread->has_audio = 0;
p_sys->p_thread->has_video = 0; p_sys->p_thread->has_video = 0;
p_sys->p_thread->metadata_received = 0; p_sys->p_thread->metadata_received = 0;
p_sys->p_thread->first_media_packet = 1; 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->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++) for(i = 0; i < 64; i++)
{ {
memset( &p_sys->p_thread->rtmp_headers_recv[i], 0, sizeof( rtmp_packet_t ) ); memset( &p_sys->p_thread->rtmp_headers_recv[i], 0, sizeof( rtmp_packet_t ) );
...@@ -167,62 +177,54 @@ static int Open( vlc_object_t *p_this ) ...@@ -167,62 +177,54 @@ static int Open( vlc_object_t *p_this )
p_sys->p_thread->rtmp_headers_send[i].body = NULL; 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_cond_init( p_sys->p_thread, &p_sys->p_thread->wait );
vlc_mutex_init( &p_sys->p_thread->lock ); vlc_mutex_init( &p_sys->p_thread->lock );
p_sys->p_thread->result_connect = 1; p_sys->p_thread->result_connect = 1;
p_sys->p_thread->result_play = 1; p_sys->p_thread->result_play = 1;
p_sys->p_thread->result_stop = 0;
/* Open connection */ /* Open connection */
p_sys->fd = net_ConnectTCP( p_access, p_sys->url.psz_host, p_sys->url.i_port ); p_sys->p_thread->fd = net_ConnectTCP( p_access, p_sys->p_thread->url.psz_host, p_sys->p_thread->url.i_port );
p_sys->p_thread->fd = p_sys->fd; if( p_sys->p_thread->fd == -1 )
if( p_sys->fd == -1 )
{ {
int *p_fd_listen; 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" ); msg_Dbg( p_access, "switching to passive mode" );
p_sys->active = 0; 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 ) 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 ); msg_Err( p_access, "cannot listen to %s port %i", p_sys->p_thread->url.psz_host, p_sys->p_thread->url.i_port );
vlc_UrlClean( &p_sys->url ); goto error2;
net_Close( p_sys-> fd );
free( p_sys );
return VLC_EGENERIC;
} }
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 ); 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"); msg_Err( p_access, "handshake passive failed");
vlc_UrlClean( &p_sys->url ); goto error2;
net_Close( p_sys-> fd );
free( p_sys );
return VLC_EGENERIC;
} }
p_sys->p_thread->result_publish = 1; p_sys->p_thread->result_publish = 1;
} }
else else
{ {
msg_Dbg( p_access, "using active connection");
p_sys->active = 1; 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"); msg_Err( p_access, "handshake active failed");
vlc_UrlClean( &p_sys->url ); goto error2;
net_Close( p_sys-> fd );
free( p_sys );
return VLC_EGENERIC;
} }
p_sys->p_thread->result_publish = 0; p_sys->p_thread->result_publish = 0;
...@@ -232,22 +234,15 @@ static int Open( vlc_object_t *p_this ) ...@@ -232,22 +234,15 @@ static int Open( vlc_object_t *p_this )
VLC_THREAD_PRIORITY_INPUT, false ) ) VLC_THREAD_PRIORITY_INPUT, false ) )
{ {
msg_Err( p_access, "cannot spawn rtmp control thread" ); msg_Err( p_access, "cannot spawn rtmp control thread" );
vlc_UrlClean( &p_sys->url ); goto error2;
net_Close( p_sys-> fd );
free( p_sys );
return VLC_EGENERIC;
} }
if( p_sys->active ) if( p_sys->active )
{ {
msg_Dbg( p_access, "Activation active connection"); if( rtmp_connect_active( p_sys->p_thread ) < 0 )
if( rtmp_connect_active( p_this ) < 0)
{ {
msg_Err( p_access, "Active connection failed"); msg_Err( p_access, "connect active failed");
vlc_UrlClean( &p_sys->url ); goto error2;
net_Close( p_sys-> fd );
free( p_sys );
return VLC_EGENERIC;
} }
} }
...@@ -255,19 +250,27 @@ static int Open( vlc_object_t *p_this ) ...@@ -255,19 +250,27 @@ static int Open( vlc_object_t *p_this )
p_access->p_sys->flv_packet = NULL; p_access->p_sys->flv_packet = NULL;
p_access->p_sys->read_packet = 1; 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 */ /* Update default_pts to a suitable value for rtmp access */
var_Create( p_access, "rtmp-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT ); var_Create( p_access, "rtmp-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
return VLC_SUCCESS; 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;
} }
/***************************************************************************** /*****************************************************************************
...@@ -277,47 +280,40 @@ static void Close( vlc_object_t * p_this ) ...@@ -277,47 +280,40 @@ static void Close( vlc_object_t * p_this )
{ {
access_t *p_access = (access_t *) p_this; access_t *p_access = (access_t *) p_this;
access_sys_t *p_sys = p_access->p_sys; 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 ); 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 ); 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_thread_join( p_sys->p_thread );
vlc_cond_destroy( &p_sys->p_thread->wait ); vlc_cond_destroy( &p_sys->p_thread->wait );
vlc_mutex_destroy( &p_sys->p_thread->lock ); 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 ); 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" ); 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 ); vlc_UrlClean( &p_sys->p_thread->url );
free( p_sys->psz_application ); free( p_sys->p_thread->psz_application );
free( p_sys->psz_media ); free( p_sys->p_thread->psz_media );
free( p_sys ); free( p_sys );
} }
...@@ -327,35 +323,39 @@ static void Close( vlc_object_t * p_this ) ...@@ -327,35 +323,39 @@ static void Close( vlc_object_t * p_this )
static int Read( access_t *p_access, uint8_t *p_buffer, size_t i_len ) static int Read( access_t *p_access, uint8_t *p_buffer, size_t i_len )
{ {
access_sys_t *p_sys = p_access->p_sys; access_sys_t *p_sys = p_access->p_sys;
rtmp_packet_t *rtmp_packet;
uint8_t *tmp_buffer;
ssize_t i_ret;
int i_len_tmp; int i_len_tmp;
if( p_sys->fd < 0 )
{
p_access->info.b_eof = true;
return 0;
}
i_len_tmp = 0; i_len_tmp = 0;
while( i_len_tmp < i_len ) 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->read_packet )
{ {
if( !p_sys->p_thread->metadata_received ) 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->flv_packet = flv_get_metadata( p_access );
p_sys->p_thread->metadata_received = 1; p_sys->p_thread->metadata_received = 1;
} }
else else
{ {
if( p_sys->active && block_FifoCount( p_sys->p_thread->p_fifo_media ) == 0 ) p_sys->flv_packet = block_FifoGet( p_sys->p_thread->p_fifo_input );
{
p_access->info.b_eof = true;
break;
}
p_sys->flv_packet = block_FifoGet( p_sys->p_thread->p_fifo_media );
if( p_sys->flv_packet == NULL ) if( p_sys->flv_packet == NULL )
continue; /* Forced wake-up */ continue; /* Forced wake-up */
} }
...@@ -400,12 +400,44 @@ p_buffer[i+8], p_buffer[i+9], p_buffer[i+10], p_buffer[i+11], p_buffer[i+12], p_ ...@@ -400,12 +400,44 @@ p_buffer[i+8], p_buffer[i+9], p_buffer[i+10], p_buffer[i+11], p_buffer[i+12], p_
/* Send publish onStatus event only once */ /* Send publish onStatus event only once */
p_sys->p_thread->result_publish = 0; 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" );
}
free( rtmp_packet->body->body );
free( rtmp_packet->body );
free( rtmp_packet );
free( tmp_buffer );
} }
p_access->info.i_pos += i_len_tmp; 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" );
}
free( rtmp_packet->body->body );
free( rtmp_packet->body );
free( rtmp_packet );
free( tmp_buffer );
} }
return i_len_tmp; return i_len_tmp;
...@@ -436,37 +468,37 @@ static int Seek( access_t *p_access, int64_t i_pos ) ...@@ -436,37 +468,37 @@ static int Seek( access_t *p_access, int64_t i_pos )
*****************************************************************************/ *****************************************************************************/
static int Control( access_t *p_access, int i_query, va_list args ) static int Control( access_t *p_access, int i_query, va_list args )
{ {
bool *pb_bool; bool *pb_bool;
int *pi_int; int *pi_int;
int64_t *pi_64; int64_t *pi_64;
switch( i_query ) switch( i_query )
{ {
/* */ /* */
case ACCESS_CAN_SEEK: case ACCESS_CAN_SEEK:
case ACCESS_CAN_FASTSEEK: case ACCESS_CAN_FASTSEEK:
pb_bool = (bool*)va_arg( args, bool* ); pb_bool = (bool*) va_arg( args, bool* );
*pb_bool = false; /* TODO */ *pb_bool = false; /* TODO */
break; break;
case ACCESS_CAN_PAUSE: case ACCESS_CAN_PAUSE:
pb_bool = (bool*)va_arg( args, bool* ); pb_bool = (bool*) va_arg( args, bool* );
*pb_bool = false; /* TODO */ *pb_bool = false; /* TODO */
break; break;
case ACCESS_CAN_CONTROL_PACE: case ACCESS_CAN_CONTROL_PACE:
pb_bool = (bool*)va_arg( args, bool* ); pb_bool = (bool*) va_arg( args, bool* );
*pb_bool = true; *pb_bool = true;
break; break;
/* */ /* */
case ACCESS_GET_MTU: case ACCESS_GET_MTU:
pi_int = (int*)va_arg( args, int * ); pi_int = (int*) va_arg( args, int * );
*pi_int = 0; *pi_int = 0;
break; break;
case ACCESS_GET_PTS_DELAY: 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); *pi_64 = var_GetInteger( p_access, "rtmp-caching" ) * INT64_C(1000);
break; break;
...@@ -503,13 +535,18 @@ static void ThreadControl( vlc_object_t *p_this ) ...@@ -503,13 +535,18 @@ static void ThreadControl( vlc_object_t *p_this )
while( !p_thread->b_die ) while( !p_thread->b_die )
{ {
rtmp_packet = rtmp_read_net_packet( p_thread ); rtmp_packet = rtmp_read_net_packet( p_thread );
if( rtmp_packet != NULL ) if( rtmp_packet != NULL )
{ {
if( rtmp_packet->content_type < 0x01 /* RTMP_CONTENT_TYPE_CHUNK_SIZE */ if( rtmp_packet->content_type < 0x01 /* RTMP_CONTENT_TYPE_CHUNK_SIZE */
|| rtmp_packet->content_type > 0x14 ) /* RTMP_CONTENT_TYPE_INVOKE */ || 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" ); msg_Warn( p_thread, "unknown content type received" );
}
else else
p_thread->rtmp_handler[rtmp_packet->content_type]( p_thread, rtmp_packet ); p_thread->rtmp_handler[rtmp_packet->content_type]( p_thread, rtmp_packet );
} }
...@@ -524,6 +561,9 @@ static void ThreadControl( vlc_object_t *p_this ) ...@@ -524,6 +561,9 @@ static void ThreadControl( vlc_object_t *p_this )
} }
p_thread->b_die = 1; p_thread->b_die = 1;
((access_t *) p_thread->p_base_object)->info.b_eof = true;
block_FifoWake( p_thread->p_fifo_input );
} }
} }
} }
...@@ -98,32 +98,36 @@ const uint16_t RTMP_PING_PONG_FROM_CLIENT = 0x0007; ...@@ -98,32 +98,36 @@ const uint16_t RTMP_PING_PONG_FROM_CLIENT = 0x0007;
const uint8_t RTMP_PING_SIZE_CLEAR_STREAM = 6; const uint8_t RTMP_PING_SIZE_CLEAR_STREAM = 6;
const uint8_t RTMP_PING_SIZE_CLEAR_PLAYING_BUFFER = 6; const uint8_t RTMP_PING_SIZE_CLEAR_PLAYING_BUFFER = 6;
const uint8_t RTMP_PING_SIZE_BUFFER_TIME_CLIENT = 10; const uint8_t RTMP_PING_SIZE_BUFFER_TIME_CLIENT = 10;
/*const uint8_t RTMP_PING_SIZE_RESET_STREAM = 0x0004; TODO const uint8_t RTMP_PING_SIZE_RESET_STREAM = 6;
const uint8_t RTMP_PING_SIZE_CLIENT_FROM_SERVER = 0x0006; /*const uint8_t RTMP_PING_SIZE_CLIENT_FROM_SERVER = 0x0006; TODO
const uint8_t RTMP_PING_SIZE_PONG_FROM_CLIENT = 0x0007; const uint8_t RTMP_PING_SIZE_PONG_FROM_CLIENT = 0x0007;
*/ */
/* default values */
const uint8_t RTMP_DEFAULT_STREAM_INDEX_CONTROL = 0x02;
const uint8_t RTMP_DEFAULT_STREAM_INDEX_INVOKE = 0x03;
const uint8_t RTMP_DEFAULT_STREAM_INDEX_NOTIFY = 0x04;
const uint8_t RTMP_DEFAULT_STREAM_INDEX_VIDEO_DATA = 0x05;
const uint8_t RTMP_DEFAULT_STREAM_INDEX_AUDIO_DATA = 0x06;
const uint32_t RTMP_DEFAULT_CHUNK_SIZE = 128;
const double RTMP_DEFAULT_STREAM_CLIENT_ID = 1.0;
const double RTMP_DEFAULT_STREAM_SERVER_ID = 1.0;
/* misc */ /* misc */
const uint16_t MAX_EMPTY_BLOCKS = 200; /* empty_blocks in fifo for media*/ const uint16_t MAX_EMPTY_BLOCKS = 200; /* empty blocks in fifo for media*/
const uint16_t RTMP_BODY_SIZE_ALLOC = 1024; const uint16_t RTMP_BODY_SIZE_ALLOC = 1024;
const uint32_t RTMP_TIME_CLIENT_BUFFER = 5000; /* miliseconds */ const uint32_t RTMP_TIME_CLIENT_BUFFER = 2000; /* miliseconds */
const uint8_t RTMP_CONTROL_STREAM_INDEX_DEFAULT = 0x02; const uint32_t RTMP_SERVER_BW = 0x00000200;
const uint8_t RTMP_AMF_STREAM_INDEX_DEFAULT = 0x03;
const uint32_t RTMP_SRC_DST_CONNECT_OBJECT = 0x00000000; const uint32_t RTMP_SRC_DST_CONNECT_OBJECT = 0x00000000;
const uint32_t RTMP_SRC_DST_CONNECT_OBJECT2 = 0x00000001; const uint32_t RTMP_SRC_DST_CONNECT_OBJECT2 = 0x00000001;
const uint32_t RTMP_SRC_DST_DEFAULT = 0x01000000; const uint32_t RTMP_SRC_DST_DEFAULT = 0x01000000;
const uint64_t RTMP_AUDIOCODECS = 0x4083380000000000; const uint64_t RTMP_AUDIOCODECS = 0x4083380000000000;
const uint64_t RTMP_VIDEOCODECS = 0x405f000000000000; const uint64_t RTMP_VIDEOCODECS = 0x405f000000000000;
const uint64_t RTMP_VIDEOFUNCTION = 0x3ff0000000000000; const uint64_t RTMP_VIDEOFUNCTION = 0x3ff0000000000000;
/***************************************************************************** /*****************************************************************************
* AMF header: * AMF header:
******************************************************************************/ ******************************************************************************/
/* packet sizes */
const uint8_t AMF_PACKET_SIZE_VIDEO = 128;
const uint8_t AMF_PACKET_SIZE_AUDIO = 64;
/* boolean constants */ /* boolean constants */
const uint8_t AMF_BOOLEAN_FALSE = 0x00; const uint8_t AMF_BOOLEAN_FALSE = 0x00;
const uint8_t AMF_BOOLEAN_TRUE = 0x01; const uint8_t AMF_BOOLEAN_TRUE = 0x01;
...@@ -165,6 +169,7 @@ const uint64_t AMF_CALL_NETCONNECTION_CONNECT_VIDEOCODECS = 0x405F000000000000; ...@@ -165,6 +169,7 @@ const uint64_t AMF_CALL_NETCONNECTION_CONNECT_VIDEOCODECS = 0x405F000000000000;
const uint64_t AMF_CALL_NETCONNECTION_CONNECT_VIDEOFUNCTION = 0x3FF0000000000000; const uint64_t AMF_CALL_NETCONNECTION_CONNECT_VIDEOFUNCTION = 0x3FF0000000000000;
const uint64_t AMF_CALL_NETCONNECTION_CONNECT_OBJECTENCODING = 0x0; const uint64_t AMF_CALL_NETCONNECTION_CONNECT_OBJECTENCODING = 0x0;
const double AMF_CALL_STREAM_CLIENT_NUMBER = 3.0; const double AMF_CALL_STREAM_CLIENT_NUMBER = 3.0;
const double AMF_CALL_ONBWDONE = 2.0;
const uint64_t AMF_CALL_NETSTREAM_PLAY = 0x0; const uint64_t AMF_CALL_NETSTREAM_PLAY = 0x0;
/***************************************************************************** /*****************************************************************************
...@@ -221,24 +226,31 @@ const uint8_t FLV_VIDEO_FRAME_TYPE_DISPOSABLE_INTER_FRAME = 0x30; ...@@ -221,24 +226,31 @@ const uint8_t FLV_VIDEO_FRAME_TYPE_DISPOSABLE_INTER_FRAME = 0x30;
/***************************************************************************** /*****************************************************************************
* static RTMP functions: * static RTMP functions:
******************************************************************************/ ******************************************************************************/
static void rtmp_handler_null ( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ); static void rtmp_handler_null ( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet );
static void rtmp_handler_invoke ( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ); static void rtmp_handler_chunk_size ( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet );
static void rtmp_handler_audio_data ( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ); static void rtmp_handler_invoke ( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet );
static void rtmp_handler_video_data ( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ); static void rtmp_handler_audio_data ( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet );
static void rtmp_handler_notify ( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ); static void rtmp_handler_video_data ( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet );
static void rtmp_handler_notify ( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet );
static rtmp_packet_t *rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t timestamp, uint8_t content_type, uint32_t src_dst, rtmp_body_t *body ); static rtmp_packet_t *rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t timestamp, uint8_t content_type, uint32_t src_dst, rtmp_body_t *body );
static block_t *rtmp_new_block( rtmp_control_thread_t *p_thread, uint8_t *buffer, int32_t length_buffer );
static uint8_t *rtmp_encode_packet( access_t *p_access, rtmp_packet_t *rtmp_packet );
static rtmp_packet_t *rtmp_encode_NetConnection_Connect_result( rtmp_control_thread_t *p_thread, double number ); static rtmp_packet_t *rtmp_encode_onBWDone( rtmp_control_thread_t *p_thread, double number );
static rtmp_packet_t *rtmp_encode_createStream_result( rtmp_control_thread_t *p_thread, double number ); static rtmp_packet_t *rtmp_encode_server_bw( rtmp_control_thread_t *p_thread, uint32_t number );
static uint8_t rtmp_encode_header_size( access_t *p_access, uint8_t header_size ); static rtmp_packet_t *rtmp_encode_NetConnection_connect_result( rtmp_control_thread_t *p_thread, double number );
static rtmp_packet_t *rtmp_encode_createStream_result( rtmp_control_thread_t *p_thread, double stream_client, double stream_server );
static rtmp_packet_t *rtmp_encode_ping_reset_stream( rtmp_control_thread_t *p_thread );
static rtmp_packet_t *rtmp_encode_ping_clear_stream( rtmp_control_thread_t *p_thread, uint32_t src_dst );
static rtmp_packet_t *rtmp_encode_NetStream_play_reset_onStatus( rtmp_control_thread_t *p_thread, char *psz_media );
static rtmp_packet_t *rtmp_encode_NetStream_play_start_onStatus( rtmp_control_thread_t *p_thread, char *psz_media );
static uint8_t rtmp_encode_header_size( vlc_object_t *p_this, uint8_t header_size );
static uint8_t rtmp_decode_header_size( vlc_object_t *p_this, uint8_t header_size ); static uint8_t rtmp_decode_header_size( vlc_object_t *p_this, uint8_t header_size );
static uint8_t rtmp_get_stream_index( uint8_t content_type );
static rtmp_body_t *rtmp_body_new( void ); static void rtmp_body_append( rtmp_body_t *rtmp_body, uint8_t *buffer, uint32_t length );
static void rtmp_body_append( rtmp_body_t *rtmp_body, uint8_t *buffer, uint8_t length );
static uint8_t *ping_encode( uint16_t type, uint32_t src_dst, uint32_t third_arg, uint32_t fourth_arg );/*TODO: change function name*/ static uint8_t *rtmp_encode_ping( uint16_t type, uint32_t src_dst, uint32_t third_arg, uint32_t fourth_arg );
/***************************************************************************** /*****************************************************************************
* static AMF functions: * static AMF functions:
...@@ -253,40 +265,34 @@ static char *amf_decode_object( uint8_t **buffer ); ...@@ -253,40 +265,34 @@ static char *amf_decode_object( uint8_t **buffer );
/***************************************************************************** /*****************************************************************************
* static FLV functions: * static FLV functions:
******************************************************************************/ ******************************************************************************/
static block_t *flv_new_packet( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet ); static void flv_rebuild( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet );
static void flv_rebuild( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ); static void flv_get_metadata_audio( rtmp_control_thread_t *p_thread, rtmp_packet_t *packet_audio, uint8_t *stereo, uint8_t *audiosamplesize, uint32_t *audiosamplerate, uint8_t *audiocodecid );
static void flv_get_metadata_audio( rtmp_packet_t *packet_audio, uint8_t *stereo, uint8_t *audiosamplesize, uint8_t *audiosamplerate, uint8_t *audiocodecid ); static void flv_get_metadata_video( rtmp_control_thread_t *p_thread, rtmp_packet_t *packet_video, uint8_t *videocodecid, uint8_t *frametype );
static void flv_get_metadata_video( rtmp_packet_t *packet_video, uint8_t *videocodecid, uint8_t *frametype ); static rtmp_packet_t *flv_build_onMetaData( access_t *p_access, uint64_t duration, uint8_t stereo, uint8_t audiosamplesize, uint32_t audiosamplerate, uint8_t audiocodecid, uint8_t videocodecid );
static rtmp_packet_t *flv_build_onMetaData( access_t *p_access, uint64_t duration, uint8_t stereo, uint8_t audiosamplesize, uint8_t audiosamplerate, uint8_t audiocodecid, uint8_t videocodecid );
/***************************************************************************** /*****************************************************************************
* RTMP implementation: * RTMP implementation:
******************************************************************************/ ******************************************************************************/
int
/***************************************************************************** rtmp_handshake_passive( vlc_object_t *p_this, int fd )
* rtmp_handshake_passive:
*******************************************************************************/
int rtmp_handshake_passive( vlc_object_t *p_this )
{ {
access_t *p_access = (access_t *) p_this;
access_sys_t *p_sys = p_access->p_sys;
uint8_t p_read[RTMP_HANDSHAKE_BODY_SIZE + 1]; uint8_t p_read[RTMP_HANDSHAKE_BODY_SIZE + 1];
uint8_t p_write[RTMP_HANDSHAKE_BODY_SIZE * 2 + 1]; uint8_t p_write[RTMP_HANDSHAKE_BODY_SIZE * 2 + 1];
ssize_t i_ret; ssize_t i_ret;
int i; int i;
/* Receive handshake */ /* Receive handshake */
i_ret = net_Read( p_access, p_sys->fd, NULL, p_read, RTMP_HANDSHAKE_BODY_SIZE + 1, true ); i_ret = net_Read( p_this, fd, NULL, p_read, RTMP_HANDSHAKE_BODY_SIZE + 1, true );
if( i_ret != RTMP_HANDSHAKE_BODY_SIZE + 1 ) if( i_ret != RTMP_HANDSHAKE_BODY_SIZE + 1 )
{ {
msg_Err( p_access, "failed to receive handshake" ); msg_Err( p_this, "failed to receive handshake" );
return -1; return -1;
} }
/* Check handshake */ /* Check handshake */
if ( p_read[0] != RTMP_HANDSHAKE ) if ( p_read[0] != RTMP_HANDSHAKE )
{ {
msg_Err( p_access, "first byte in handshake corrupt" ); msg_Err( p_this, "first byte in handshake corrupt" );
return -1; return -1;
} }
...@@ -296,18 +302,18 @@ int rtmp_handshake_passive( vlc_object_t *p_this ) ...@@ -296,18 +302,18 @@ int rtmp_handshake_passive( vlc_object_t *p_this )
memcpy( p_write + 1 + RTMP_HANDSHAKE_BODY_SIZE, p_read + 1, RTMP_HANDSHAKE_BODY_SIZE ); memcpy( p_write + 1 + RTMP_HANDSHAKE_BODY_SIZE, p_read + 1, RTMP_HANDSHAKE_BODY_SIZE );
/* Send handshake*/ /* Send handshake*/
i_ret = net_Write( p_access, p_sys->fd, NULL, p_write, RTMP_HANDSHAKE_BODY_SIZE * 2 + 1 ); i_ret = net_Write( p_this, fd, NULL, p_write, RTMP_HANDSHAKE_BODY_SIZE * 2 + 1 );
if( i_ret != RTMP_HANDSHAKE_BODY_SIZE * 2 + 1 ) if( i_ret != RTMP_HANDSHAKE_BODY_SIZE * 2 + 1 )
{ {
msg_Err( p_access, "failed to send handshake" ); msg_Err( p_this, "failed to send handshake" );
return -1; return -1;
} }
/* Receive acknowledge */ /* Receive acknowledge */
i_ret = net_Read( p_access, p_sys->fd, NULL, p_read, RTMP_HANDSHAKE_BODY_SIZE, true ); i_ret = net_Read( p_this, fd, NULL, p_read, RTMP_HANDSHAKE_BODY_SIZE, true );
if( i_ret != RTMP_HANDSHAKE_BODY_SIZE ) if( i_ret != RTMP_HANDSHAKE_BODY_SIZE )
{ {
msg_Err( p_access, "failed to receive acknowledge" ); msg_Err( p_this, "failed to receive acknowledge" );
return -1; return -1;
} }
...@@ -315,19 +321,16 @@ int rtmp_handshake_passive( vlc_object_t *p_this ) ...@@ -315,19 +321,16 @@ int rtmp_handshake_passive( vlc_object_t *p_this )
for(i = 8; i < RTMP_HANDSHAKE_BODY_SIZE; i++ ) for(i = 8; i < RTMP_HANDSHAKE_BODY_SIZE; i++ )
if( p_write[i + 1] != p_read[i] ) if( p_write[i + 1] != p_read[i] )
{ {
msg_Err( p_access, "body acknowledge received corrupt" ); msg_Err( p_this, "body acknowledge received corrupt" );
return -1; return -1;
} }
return 0; return 0;
} }
/*****************************************************************************
* rtmp_handshake_active: int
*******************************************************************************/ rtmp_handshake_active( vlc_object_t *p_this, int fd )
int rtmp_handshake_active( vlc_object_t *p_this )
{ {
access_t *p_access = (access_t *) p_this;
access_sys_t *p_sys = p_access->p_sys;
uint8_t p_read[RTMP_HANDSHAKE_BODY_SIZE * 2 + 1]; uint8_t p_read[RTMP_HANDSHAKE_BODY_SIZE * 2 + 1];
uint8_t p_write[RTMP_HANDSHAKE_BODY_SIZE + 1]; uint8_t p_write[RTMP_HANDSHAKE_BODY_SIZE + 1];
ssize_t i_ret; ssize_t i_ret;
...@@ -338,53 +341,49 @@ int rtmp_handshake_active( vlc_object_t *p_this ) ...@@ -338,53 +341,49 @@ int rtmp_handshake_active( vlc_object_t *p_this )
p_write[i + 1] = i & 0xFF; p_write[i + 1] = i & 0xFF;
/* Send handshake*/ /* Send handshake*/
i_ret = net_Write( p_access, p_sys->fd, NULL, p_write, RTMP_HANDSHAKE_BODY_SIZE + 1 ); i_ret = net_Write( p_this, fd, NULL, p_write, RTMP_HANDSHAKE_BODY_SIZE + 1 );
if( i_ret != RTMP_HANDSHAKE_BODY_SIZE + 1 ) if( i_ret != RTMP_HANDSHAKE_BODY_SIZE + 1 )
{ {
msg_Err( p_access, "failed to send handshake" ); msg_Err( p_this, "failed to send handshake" );
return -1; return -1;
} }
/* Receive handshake */ /* Receive handshake */
i_ret = net_Read( p_access, p_sys->fd, NULL, p_read, RTMP_HANDSHAKE_BODY_SIZE * 2 + 1, true ); i_ret = net_Read( p_this, fd, NULL, p_read, RTMP_HANDSHAKE_BODY_SIZE * 2 + 1, true );
if( i_ret != RTMP_HANDSHAKE_BODY_SIZE * 2 + 1 ) if( i_ret != RTMP_HANDSHAKE_BODY_SIZE * 2 + 1 )
{ {
msg_Err( p_access, "failed to receive handshake" ); msg_Err( p_this, "failed to receive handshake" );
return -1; return -1;
} }
/* Check handshake */ /* Check handshake */
if( p_read[0] != RTMP_HANDSHAKE ) if( p_read[0] != RTMP_HANDSHAKE )
{ {
msg_Err( p_access, "first byte in handshake received corrupt" ); msg_Err( p_this, "first byte in handshake received corrupt" );
return -1; return -1;
} }
for(i = 0; i < RTMP_HANDSHAKE_BODY_SIZE; i++ ) for(i = 8; i < RTMP_HANDSHAKE_BODY_SIZE; i++ )
if( p_write[i + 1] != p_read[i + 1 + RTMP_HANDSHAKE_BODY_SIZE] ) if( p_write[i + 1] != p_read[i + 1 + RTMP_HANDSHAKE_BODY_SIZE] )
{ {
msg_Err( p_access, "body handshake received corrupt" ); msg_Err( p_this, "body handshake received corrupt" );
return -1; return -1;
} }
/* Acknowledge handshake */ /* Acknowledge handshake */
i_ret = net_Write( p_access, p_sys->fd, NULL, p_read + 1, RTMP_HANDSHAKE_BODY_SIZE ); i_ret = net_Write( p_this, fd, NULL, p_read + 1, RTMP_HANDSHAKE_BODY_SIZE );
if( i_ret != RTMP_HANDSHAKE_BODY_SIZE ) if( i_ret != RTMP_HANDSHAKE_BODY_SIZE )
{ {
msg_Err( p_access, "failed to acknowledge handshake" ); msg_Err( p_this, "failed to acknowledge handshake" );
return -1; return -1;
} }
return 0; return 0;
} }
/***************************************************************************** int
* rtmp_connect_active: rtmp_connect_active( rtmp_control_thread_t *p_thread )
******************************************************************************/
int rtmp_connect_active( vlc_object_t *p_this )
{ {
access_t *p_access = (access_t *) p_this;
access_sys_t *p_sys = p_access->p_sys;
rtmp_packet_t *rtmp_packet; rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body; rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer; uint8_t *tmp_buffer;
...@@ -392,7 +391,7 @@ int rtmp_connect_active( vlc_object_t *p_this ) ...@@ -392,7 +391,7 @@ int rtmp_connect_active( vlc_object_t *p_this )
ssize_t i_ret; ssize_t i_ret;
/* Build NetConnection.connect call */ /* Build NetConnection.connect call */
rtmp_body = rtmp_body_new(); rtmp_body = rtmp_body_new( -1 );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "connect" ); tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "connect" );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
...@@ -409,28 +408,28 @@ int rtmp_connect_active( vlc_object_t *p_this ) ...@@ -409,28 +408,28 @@ int rtmp_connect_active( vlc_object_t *p_this )
free( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "app", tmp_buffer = amf_encode_object_variable( "app",
AMF_DATATYPE_STRING, p_sys->psz_application ); AMF_DATATYPE_STRING, p_thread->psz_application );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "app" ) + AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "app" ) +
AMF_DATATYPE_SIZE_STRING + strlen( p_sys->psz_application ) ); AMF_DATATYPE_SIZE_STRING + strlen( p_thread->psz_application ) );
free ( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "flashVer", tmp_buffer = amf_encode_object_variable( "flashVer",
AMF_DATATYPE_STRING, "LNX 9,0,48,0" ); AMF_DATATYPE_STRING, "LNX 9,0,48,0" );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "flashVer" ) + AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "flashVer" ) +
AMF_DATATYPE_SIZE_STRING + strlen( "LNX 9,0,48,0" ) ); AMF_DATATYPE_SIZE_STRING + strlen( "LNX 9,0,48,0" ) );
free ( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "swfUrl", tmp_buffer = amf_encode_object_variable( "swfUrl",
AMF_DATATYPE_STRING, "file:///mac.flv" ); AMF_DATATYPE_STRING, "file:///mac.flv" );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "swfUrl" ) + AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "swfUrl" ) +
AMF_DATATYPE_SIZE_STRING + strlen( "file:///mac.flv" ) ); AMF_DATATYPE_SIZE_STRING + strlen( "file:///mac.flv" ) );
free ( tmp_buffer ); free( tmp_buffer );
tmp_url = (char *) malloc( strlen( "rtmp://") + strlen( p_sys->url.psz_buffer ) + 1 ); tmp_url = (char *) malloc( strlen( "rtmp://") + strlen( p_thread->url.psz_buffer ) + 1 );
sprintf( tmp_url, "rtmp://%s", p_sys->url.psz_buffer ); sprintf( tmp_url, "rtmp://%s", p_thread->url.psz_buffer );
tmp_buffer = amf_encode_object_variable( "tcUrl", tmp_buffer = amf_encode_object_variable( "tcUrl",
AMF_DATATYPE_STRING, tmp_url ); AMF_DATATYPE_STRING, tmp_url );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
...@@ -485,22 +484,22 @@ int rtmp_connect_active( vlc_object_t *p_this ) ...@@ -485,22 +484,22 @@ int rtmp_connect_active( vlc_object_t *p_this )
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_END_OF_OBJECT ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_END_OF_OBJECT );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_sys->p_thread, RTMP_AMF_STREAM_INDEX_DEFAULT, rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_INVOKE,
0, RTMP_CONTENT_TYPE_INVOKE, 0, rtmp_body ); 0, RTMP_CONTENT_TYPE_INVOKE, 0, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
tmp_buffer = rtmp_encode_packet( p_access, rtmp_packet ); tmp_buffer = rtmp_encode_packet( p_thread, rtmp_packet );
/* Call NetConnection.connect */ /* Call NetConnection.connect */
i_ret = net_Write( p_access, p_sys->fd, NULL, tmp_buffer, rtmp_packet->length_encoded ); i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, rtmp_packet->length_encoded );
if( i_ret != rtmp_packet->length_encoded ) if( i_ret != rtmp_packet->length_encoded )
{ {
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
free( rtmp_packet->body ); free( rtmp_packet->body );
free( rtmp_packet ); free( rtmp_packet );
free( tmp_buffer ); free( tmp_buffer );
msg_Err( p_access, "failed send call NetConnection.connect" ); msg_Err( p_thread, "failed send call NetConnection.connect" );
return -1; return -1;
} }
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
...@@ -509,28 +508,28 @@ int rtmp_connect_active( vlc_object_t *p_this ) ...@@ -509,28 +508,28 @@ int rtmp_connect_active( vlc_object_t *p_this )
free( tmp_buffer ); free( tmp_buffer );
/* Wait for NetConnection.connect result */ /* Wait for NetConnection.connect result */
vlc_mutex_lock( &p_sys->p_thread->lock ); vlc_mutex_lock( &p_thread->lock );
vlc_cond_wait( &p_sys->p_thread->wait, &p_sys->p_thread->lock ); vlc_cond_wait( &p_thread->wait, &p_thread->lock );
vlc_mutex_unlock( &p_sys->p_thread->lock ); vlc_mutex_unlock( &p_thread->lock );
if( p_sys->p_thread->result_connect ) if( p_thread->result_connect )
{ {
msg_Err( p_access, "failed call NetConnection.connect" ); msg_Err( p_thread, "failed call NetConnection.connect" );
return -1; return -1;
} }
/* Force control thread to stop if receive NetStream.play call and wait is not ready */ /* Force control thread to stop if receive NetStream.play call and wait is not ready */
vlc_mutex_lock( &p_sys->p_thread->lock ); vlc_mutex_lock( &p_thread->lock );
/* Build NetStream.createStream call */ /* Build NetStream.createStream call */
rtmp_body = rtmp_body_new(); rtmp_body = rtmp_body_new( -1 );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "createStream" ); tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "createStream" );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_STRING + strlen( "createStream" ) ); AMF_DATATYPE_SIZE_STRING + strlen( "createStream" ) );
free( tmp_buffer ); free( tmp_buffer );
p_sys->p_thread->stream_client = AMF_CALL_STREAM_CLIENT_NUMBER; p_thread->stream_client_id = RTMP_DEFAULT_STREAM_CLIENT_ID;
tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER, tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER,
&AMF_CALL_STREAM_CLIENT_NUMBER ); &AMF_CALL_STREAM_CLIENT_NUMBER );
...@@ -541,22 +540,22 @@ int rtmp_connect_active( vlc_object_t *p_this ) ...@@ -541,22 +540,22 @@ int rtmp_connect_active( vlc_object_t *p_this )
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NULL ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NULL );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_sys->p_thread, RTMP_AMF_STREAM_INDEX_DEFAULT, rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_INVOKE,
0, RTMP_CONTENT_TYPE_INVOKE, 0, rtmp_body ); 0, RTMP_CONTENT_TYPE_INVOKE, 0, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
tmp_buffer = rtmp_encode_packet( p_access, rtmp_packet ); tmp_buffer = rtmp_encode_packet( p_thread, rtmp_packet );
/* Call NetStream.createStream */ /* Call NetStream.createStream */
i_ret = net_Write( p_access, p_sys->fd, NULL, tmp_buffer, rtmp_packet->length_encoded ); i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, rtmp_packet->length_encoded );
if( i_ret != rtmp_packet->length_encoded ) if( i_ret != rtmp_packet->length_encoded )
{ {
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
free( rtmp_packet->body ); free( rtmp_packet->body );
free( rtmp_packet ); free( rtmp_packet );
free( tmp_buffer ); free( tmp_buffer );
msg_Err( p_access, "failed send call NetStream.createStream" ); msg_Err( p_thread, "failed send call NetStream.createStream" );
return -1; return -1;
} }
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
...@@ -565,28 +564,28 @@ int rtmp_connect_active( vlc_object_t *p_this ) ...@@ -565,28 +564,28 @@ int rtmp_connect_active( vlc_object_t *p_this )
free( tmp_buffer ); free( tmp_buffer );
/*TODO: read server stream number*/ /*TODO: read server stream number*/
/* Build ping packet */ /* Build ping packet */
rtmp_body = rtmp_body_new(); rtmp_body = rtmp_body_new( -1 );
tmp_buffer = ping_encode( RTMP_PING_BUFFER_TIME_CLIENT, RTMP_SRC_DST_CONNECT_OBJECT, RTMP_TIME_CLIENT_BUFFER, 0 ); tmp_buffer = rtmp_encode_ping( RTMP_PING_BUFFER_TIME_CLIENT, RTMP_SRC_DST_CONNECT_OBJECT, RTMP_TIME_CLIENT_BUFFER, 0 );
rtmp_body_append( rtmp_body, tmp_buffer, RTMP_PING_SIZE_BUFFER_TIME_CLIENT ); rtmp_body_append( rtmp_body, tmp_buffer, RTMP_PING_SIZE_BUFFER_TIME_CLIENT );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_sys->p_thread, RTMP_CONTROL_STREAM_INDEX_DEFAULT, rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_CONTROL,
0, RTMP_CONTENT_TYPE_PING, 0, rtmp_body ); 0, RTMP_CONTENT_TYPE_PING, 0, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
tmp_buffer = rtmp_encode_packet( p_access, rtmp_packet ); tmp_buffer = rtmp_encode_packet( p_thread, rtmp_packet );
/* Send ping packet */ /* Send ping packet */
i_ret = net_Write( p_access, p_sys->fd, NULL, tmp_buffer, rtmp_packet->length_encoded ); i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, rtmp_packet->length_encoded );
if( i_ret != rtmp_packet->length_encoded ) if( i_ret != rtmp_packet->length_encoded )
{ {
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
free( rtmp_packet->body ); free( rtmp_packet->body );
free( rtmp_packet ); free( rtmp_packet );
free( tmp_buffer ); free( tmp_buffer );
msg_Err( p_access, "failed send ping BUFFER_TIME_CLIENT" ); msg_Err( p_thread, "failed send ping BUFFER_TIME_CLIENT" );
return -1; return -1;
} }
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
...@@ -595,7 +594,7 @@ int rtmp_connect_active( vlc_object_t *p_this ) ...@@ -595,7 +594,7 @@ int rtmp_connect_active( vlc_object_t *p_this )
free( tmp_buffer ); free( tmp_buffer );
/* Build NetStream.play call */ /* Build NetStream.play call */
rtmp_body = rtmp_body_new(); rtmp_body = rtmp_body_new( -1 );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "play" ); tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "play" );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
...@@ -611,27 +610,27 @@ int rtmp_connect_active( vlc_object_t *p_this ) ...@@ -611,27 +610,27 @@ int rtmp_connect_active( vlc_object_t *p_this )
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NULL ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NULL );
free( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, p_sys->psz_media ); tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, p_thread->psz_media );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_STRING + strlen( p_sys->psz_media ) ); AMF_DATATYPE_SIZE_STRING + strlen( p_thread->psz_media ) );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_sys->p_thread, RTMP_AMF_STREAM_INDEX_DEFAULT, rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_INVOKE,
0, RTMP_CONTENT_TYPE_INVOKE, RTMP_SRC_DST_DEFAULT, rtmp_body ); 0, RTMP_CONTENT_TYPE_INVOKE, RTMP_SRC_DST_DEFAULT, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
tmp_buffer = rtmp_encode_packet( p_access, rtmp_packet ); tmp_buffer = rtmp_encode_packet( p_thread, rtmp_packet );
/* Call NetStream.play */ /* Call NetStream.play */
i_ret = net_Write( p_access, p_sys->fd, NULL, tmp_buffer, rtmp_packet->length_encoded ); i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, rtmp_packet->length_encoded );
if( i_ret != rtmp_packet->length_encoded ) if( i_ret != rtmp_packet->length_encoded )
{ {
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
free( rtmp_packet->body ); free( rtmp_packet->body );
free( rtmp_packet ); free( rtmp_packet );
free( tmp_buffer ); free( tmp_buffer );
msg_Err( p_access, "failed send call NetStream.play" ); msg_Err( p_thread, "failed send call NetStream.play" );
return -1; return -1;
} }
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
...@@ -640,28 +639,28 @@ int rtmp_connect_active( vlc_object_t *p_this ) ...@@ -640,28 +639,28 @@ int rtmp_connect_active( vlc_object_t *p_this )
free( tmp_buffer ); free( tmp_buffer );
/* Build ping packet */ /* Build ping packet */
rtmp_body = rtmp_body_new(); rtmp_body = rtmp_body_new( -1 );
tmp_buffer = ping_encode( RTMP_PING_BUFFER_TIME_CLIENT, RTMP_SRC_DST_CONNECT_OBJECT2, RTMP_TIME_CLIENT_BUFFER, 0 ); tmp_buffer = rtmp_encode_ping( RTMP_PING_BUFFER_TIME_CLIENT, RTMP_SRC_DST_CONNECT_OBJECT2, RTMP_TIME_CLIENT_BUFFER, 0 );
rtmp_body_append( rtmp_body, tmp_buffer, RTMP_PING_SIZE_BUFFER_TIME_CLIENT ); rtmp_body_append( rtmp_body, tmp_buffer, RTMP_PING_SIZE_BUFFER_TIME_CLIENT );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_sys->p_thread, RTMP_CONTROL_STREAM_INDEX_DEFAULT, rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_CONTROL,
0, RTMP_CONTENT_TYPE_PING, 0, rtmp_body ); 0, RTMP_CONTENT_TYPE_PING, 0, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
tmp_buffer = rtmp_encode_packet( p_access, rtmp_packet ); tmp_buffer = rtmp_encode_packet( p_thread, rtmp_packet );
/* Send ping packet */ /* Send ping packet */
i_ret = net_Write( p_access, p_sys->fd, NULL, tmp_buffer, rtmp_packet->length_encoded ); i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, rtmp_packet->length_encoded );
if( i_ret != rtmp_packet->length_encoded ) if( i_ret != rtmp_packet->length_encoded )
{ {
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
free( rtmp_packet->body ); free( rtmp_packet->body );
free( rtmp_packet ); free( rtmp_packet );
free( tmp_buffer ); free( tmp_buffer );
msg_Err( p_access, "failed send ping BUFFER_TIME_CLIENT" ); msg_Err( p_thread, "failed send ping BUFFER_TIME_CLIENT" );
return -1; return -1;
} }
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
...@@ -670,20 +669,40 @@ int rtmp_connect_active( vlc_object_t *p_this ) ...@@ -670,20 +669,40 @@ int rtmp_connect_active( vlc_object_t *p_this )
free( tmp_buffer ); free( tmp_buffer );
/* Wait for NetStream.play.start result */ /* Wait for NetStream.play.start result */
vlc_cond_wait( &p_sys->p_thread->wait, &p_sys->p_thread->lock ); vlc_cond_wait( &p_thread->wait, &p_thread->lock );
vlc_mutex_unlock( &p_sys->p_thread->lock ); vlc_mutex_unlock( &p_thread->lock );
if( p_sys->p_thread->result_play ) if( p_thread->result_play )
{ {
msg_Err( p_access, "failed call NetStream.play" ); msg_Err( p_thread, "failed call NetStream.play" );
return -1; return -1;
} }
/* Next packet is the beginning of flv stream */ /* Next packet is the beginning of flv stream */
msg_Dbg(p_access, "Next packet is the beginning of flv stream"); msg_Dbg( p_thread, "next packet is the beginning of flv stream" );
return 0;
}
int
rtmp_connect_passive( rtmp_control_thread_t *p_thread )
{
/* Force control thread to stop if receive NetStream.play call and wait is not ready */
vlc_mutex_lock( &p_thread->lock );
/* Wait for NetStream.play.start result */
vlc_cond_wait( &p_thread->wait, &p_thread->lock );
vlc_mutex_unlock( &p_thread->lock );
if( p_thread->result_play )
{
msg_Err( p_thread, "failed call NetStream.play" );
return -1;
}
return 0; return 0;
} }
/* TODO /* TODO
int int
rtmp_seek( access_t *p_access, int64_t i_pos ) rtmp_seek( access_t *p_access, int64_t i_pos )
...@@ -718,7 +737,7 @@ msg_Warn(p_access, "i_pos %lld", i_pos); ...@@ -718,7 +737,7 @@ msg_Warn(p_access, "i_pos %lld", i_pos);
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_sys->p_thread, RTMP_AMF_STREAM_INDEX_DEFAULT, rtmp_packet = rtmp_new_packet( p_sys->p_thread, RTMP_DEFAULT_STREAM_INDEX_INVOKE,
0, RTMP_DATATYPE_INVOKE, RTMP_SRC_DST_DEFAULT, rtmp_body ); 0, RTMP_DATATYPE_INVOKE, RTMP_SRC_DST_DEFAULT, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
...@@ -750,17 +769,16 @@ msg_Warn(p_access, "i_pos %lld", i_pos); ...@@ -750,17 +769,16 @@ msg_Warn(p_access, "i_pos %lld", i_pos);
return 0; return 0;
} }
*/ */
int
rtmp_send_bytes_read( access_t *p_access, uint32_t reply ) rtmp_packet_t *
rtmp_build_bytes_read( rtmp_control_thread_t *p_thread, uint32_t reply )
{ {
access_sys_t *p_sys = p_access->p_sys;
rtmp_packet_t *rtmp_packet; rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body; rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer; uint8_t *tmp_buffer;
ssize_t i_ret;
/* Build bytes read packet */ /* Build bytes read packet */
rtmp_body = rtmp_body_new(); rtmp_body = rtmp_body_new( -1 );
tmp_buffer = (uint8_t *) malloc( sizeof( uint32_t ) * sizeof( uint8_t ) ); tmp_buffer = (uint8_t *) malloc( sizeof( uint32_t ) * sizeof( uint8_t ) );
reply = hton32( reply ); reply = hton32( reply );
...@@ -769,43 +787,23 @@ rtmp_send_bytes_read( access_t *p_access, uint32_t reply ) ...@@ -769,43 +787,23 @@ rtmp_send_bytes_read( access_t *p_access, uint32_t reply )
rtmp_body_append( rtmp_body, tmp_buffer, sizeof( uint32_t ) ); rtmp_body_append( rtmp_body, tmp_buffer, sizeof( uint32_t ) );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_sys->p_thread, RTMP_CONTROL_STREAM_INDEX_DEFAULT, rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_CONTROL,
0, RTMP_CONTENT_TYPE_BYTES_READ, 0, rtmp_body ); 0, RTMP_CONTENT_TYPE_BYTES_READ, 0, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
tmp_buffer = rtmp_encode_packet( p_access, rtmp_packet ); return rtmp_packet;
/* Send bytes read packet */
i_ret = net_Write( p_access, p_sys->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 0;
} }
int rtmp_packet_t *
rtmp_send_publish_start( access_t *p_access ) rtmp_build_publish_start( rtmp_control_thread_t *p_thread )
{ {
access_sys_t *p_sys = p_access->p_sys;
rtmp_packet_t *rtmp_packet; rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body; rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer; uint8_t *tmp_buffer;
ssize_t i_ret;
/* Build publish start event */ /* Build publish start event */
rtmp_body = rtmp_body_new(); rtmp_body = rtmp_body_new( -1 );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "onStatus" ); tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "onStatus" );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
...@@ -813,7 +811,7 @@ rtmp_send_publish_start( access_t *p_access ) ...@@ -813,7 +811,7 @@ rtmp_send_publish_start( access_t *p_access )
free( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER, tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER,
&p_sys->p_thread->stream_server ); &p_thread->stream_server_id );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer ); free( tmp_buffer );
...@@ -847,16 +845,16 @@ rtmp_send_publish_start( access_t *p_access ) ...@@ -847,16 +845,16 @@ rtmp_send_publish_start( access_t *p_access )
free ( tmp_buffer ); free ( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "details", tmp_buffer = amf_encode_object_variable( "details",
AMF_DATATYPE_STRING, p_sys->p_thread->psz_publish ); AMF_DATATYPE_STRING, p_thread->psz_publish );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "details" ) + AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "details" ) +
AMF_DATATYPE_SIZE_STRING + strlen( p_sys->p_thread->psz_publish ) ); AMF_DATATYPE_SIZE_STRING + strlen( p_thread->psz_publish ) );
free ( tmp_buffer ); free ( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "clientid", tmp_buffer = amf_encode_object_variable( "clientid",
AMF_DATATYPE_NUMBER, &p_sys->p_thread->stream_client ); AMF_DATATYPE_NUMBER, &p_thread->stream_client_id );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "videoCodecs" ) + AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "clientid" ) +
AMF_DATATYPE_SIZE_NUMBER ); AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer ); free( tmp_buffer );
...@@ -864,145 +862,157 @@ rtmp_send_publish_start( access_t *p_access ) ...@@ -864,145 +862,157 @@ rtmp_send_publish_start( access_t *p_access )
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_END_OF_OBJECT ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_END_OF_OBJECT );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_sys->p_thread, RTMP_AMF_STREAM_INDEX_DEFAULT, rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_INVOKE,
0, RTMP_CONTENT_TYPE_INVOKE, 0, rtmp_body ); 0, RTMP_CONTENT_TYPE_INVOKE, 0, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
tmp_buffer = rtmp_encode_packet( p_access, rtmp_packet ); return rtmp_packet;
/* Send publish start event */
i_ret = net_Write( p_access, p_sys->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 event" );
return -1;
}
free( rtmp_packet->body->body );
free( rtmp_packet->body );
free( rtmp_packet );
free( tmp_buffer );
return 0;
} }
rtmp_packet_t * rtmp_packet_t *
rtmp_read_net_packet( rtmp_control_thread_t *p_thread ) rtmp_build_flv_over_rtmp( rtmp_control_thread_t *p_thread, block_t *p_buffer )
{ {
int interchunk_headers;
uint8_t p_read[12], trash;
rtmp_packet_t *rtmp_packet; rtmp_packet_t *rtmp_packet;
ssize_t i_ret;
int i;
i_ret = net_Read( p_thread, p_thread->fd, NULL, p_read, 1, true );
if( i_ret != 1 )
{
msg_Err( p_thread, "rtmp_read_net_packet: net_Read error");
return NULL;
}
rtmp_packet = (rtmp_packet_t *) malloc( sizeof( rtmp_packet_t ) );
rtmp_packet->body = (rtmp_body_t *) malloc( sizeof( rtmp_body_t ) );
rtmp_packet->length_header = rtmp_decode_header_size( (vlc_object_t *) p_thread, p_read[0] & RTMP_HEADER_SIZE_MASK ); if( p_thread->flv_length_body > 0 )
rtmp_packet->stream_index = p_read[0] & RTMP_HEADER_STREAM_INDEX_MASK;
i_ret = net_Read( p_thread, p_thread->fd, NULL, p_read + 1, rtmp_packet->length_header - 1, true );
if( i_ret != rtmp_packet->length_header - 1 )
goto error;
if( rtmp_packet->length_header >= 4 )
{ {
p_read[0] = 0; p_thread->flv_length_body -= p_buffer->i_buffer;
rtmp_body_append( p_thread->flv_body, p_buffer->p_buffer, p_buffer->i_buffer );
rtmp_packet->timestamp_relative = ntoh32( *(uint32_t *) p_read ); if( p_thread->flv_length_body > 0 )
rtmp_packet->timestamp = p_thread->rtmp_headers_recv[rtmp_packet->stream_index].timestamp + rtmp_packet->timestamp_relative; return NULL;
p_thread->rtmp_headers_recv[rtmp_packet->stream_index].timestamp = rtmp_packet->timestamp;
p_thread->rtmp_headers_recv[rtmp_packet->stream_index].timestamp_relative = rtmp_packet->timestamp_relative;
} }
else else
{ {
rtmp_packet->timestamp_relative = p_thread->rtmp_headers_recv[rtmp_packet->stream_index].timestamp_relative; p_thread->flv_content_type = *p_buffer->p_buffer;
rtmp_packet->timestamp = p_thread->rtmp_headers_recv[rtmp_packet->stream_index].timestamp + rtmp_packet->timestamp_relative;
p_thread->rtmp_headers_recv[rtmp_packet->stream_index].timestamp = rtmp_packet->timestamp; p_buffer->p_buffer[0] = 0;
p_thread->flv_length_body = ntoh32( *(uint32_t *) (p_buffer->p_buffer) );
p_buffer->p_buffer[3] = 0;
p_thread->flv_timestamp = ntoh32( *(uint32_t *) (p_buffer->p_buffer + 3) );
} }
if( rtmp_packet->length_header >= 8 ) if( p_thread->flv_length_body > p_buffer->i_buffer - FLV_TAG_SIZE - FLV_TAG_PREVIOUS_TAG_SIZE )
{ {
p_read[3] = 0; p_thread->flv_length_body -= p_buffer->i_buffer - FLV_TAG_SIZE - FLV_TAG_PREVIOUS_TAG_SIZE;
rtmp_packet->body->length_body = ntoh32( *(uint32_t *) (p_read + 3) ); rtmp_body_append( p_thread->flv_body, p_buffer->p_buffer + FLV_TAG_SIZE, p_buffer->i_buffer - FLV_TAG_SIZE );
rtmp_packet->content_type = p_read[7];
interchunk_headers = rtmp_packet->body->length_body / AMF_PACKET_SIZE_VIDEO; return NULL;
rtmp_packet->length_encoded = rtmp_packet->length_header + rtmp_packet->body->length_body + interchunk_headers;
p_thread->rtmp_headers_recv[rtmp_packet->stream_index].length_header = rtmp_packet->length_header;
p_thread->rtmp_headers_recv[rtmp_packet->stream_index].content_type = rtmp_packet->content_type;
p_thread->rtmp_headers_recv[rtmp_packet->stream_index].length_encoded = rtmp_packet->length_encoded;
} }
else else
{ {
rtmp_packet->content_type = p_thread->rtmp_headers_recv[rtmp_packet->stream_index].content_type; rtmp_body_append( p_thread->flv_body, p_buffer->p_buffer + FLV_TAG_SIZE, p_thread->flv_length_body );
}
rtmp_packet->body->length_body = rtmp_packet = rtmp_new_packet( p_thread, rtmp_get_stream_index( p_thread->flv_content_type ),
p_thread->rtmp_headers_recv[rtmp_packet->stream_index].length_encoded - p_thread->flv_timestamp, p_thread->flv_content_type, RTMP_SRC_DST_DEFAULT, p_thread->flv_body );
p_thread->rtmp_headers_recv[rtmp_packet->stream_index].length_header - p_thread->flv_length_body = 0;
(p_thread->rtmp_headers_recv[rtmp_packet->stream_index].length_encoded - rtmp_body_reset( p_thread->flv_body );
p_thread->rtmp_headers_recv[rtmp_packet->stream_index].length_header) / AMF_PACKET_SIZE_VIDEO;
interchunk_headers = rtmp_packet->body->length_body / AMF_PACKET_SIZE_VIDEO; return rtmp_packet;
rtmp_packet->length_encoded = rtmp_packet->length_header + rtmp_packet->body->length_body + interchunk_headers; }
}
if( rtmp_packet->length_header >= 12 ) rtmp_packet_t *
{ rtmp_read_net_packet( rtmp_control_thread_t *p_thread )
rtmp_packet->src_dst = ntoh32( *(uint32_t *) (p_read + 8) ); {
int length_header;
int stream_index;
int bytes_left;
uint8_t p_read[12];
rtmp_packet_t *rtmp_packet;
ssize_t i_ret;
p_thread->rtmp_headers_recv[rtmp_packet->stream_index].src_dst = rtmp_packet->src_dst;
} for(;;)
else
{ {
rtmp_packet->src_dst = p_thread->rtmp_headers_recv[rtmp_packet->stream_index].src_dst; i_ret = net_Read( p_thread, p_thread->fd, NULL, p_read, 1, true );
} if( i_ret != 1 )
goto error;
rtmp_packet->body->length_buffer = rtmp_packet->body->length_body; length_header = rtmp_decode_header_size( (vlc_object_t *) p_thread, p_read[0] & RTMP_HEADER_SIZE_MASK );
rtmp_packet->body->body = (uint8_t *) malloc( rtmp_packet->body->length_buffer * sizeof( uint8_t ) ); stream_index = p_read[0] & RTMP_HEADER_STREAM_INDEX_MASK;
if( rtmp_packet->body->length_body % AMF_PACKET_SIZE_VIDEO == 0 ) i_ret = net_Read( p_thread, p_thread->fd, NULL, p_read + 1, length_header - 1, true );
interchunk_headers--; if( i_ret != length_header - 1 )
goto error;
/* Update timestamp if not is an interchunk packet */
if( length_header == 1 && p_thread->rtmp_headers_recv[stream_index].body == NULL )
{
p_thread->rtmp_headers_recv[stream_index].timestamp +=
p_thread->rtmp_headers_recv[stream_index].timestamp_relative;
}
/* Length 4 and 8 headers have relative timestamp */
if( length_header == 4 || length_header == 8 )
{
p_read[0] = 0;
p_thread->rtmp_headers_recv[stream_index].timestamp_relative = ntoh32( *(uint32_t *) p_read );
p_thread->rtmp_headers_recv[stream_index].timestamp +=
p_thread->rtmp_headers_recv[stream_index].timestamp_relative;
}
if( length_header >= 8 )
{
p_read[3] = 0;
p_thread->rtmp_headers_recv[stream_index].length_body = ntoh32( *(uint32_t *) (p_read + 3) );
p_thread->rtmp_headers_recv[stream_index].content_type = p_read[7];
}
/* Length 12 headers have absolute timestamp */
if( length_header >= 12 )
{
p_read[0] = 0;
p_thread->rtmp_headers_recv[stream_index].timestamp = ntoh32( *(uint32_t *) p_read );
p_thread->rtmp_headers_recv[stream_index].src_dst = ntoh32( *(uint32_t *) (p_read + 8) );
}
if( p_thread->rtmp_headers_recv[stream_index].body == NULL )
{
p_thread->rtmp_headers_recv[stream_index].body =
rtmp_body_new( p_thread->rtmp_headers_recv[stream_index].length_body );
}
bytes_left = p_thread->rtmp_headers_recv[stream_index].body->length_buffer -
p_thread->rtmp_headers_recv[stream_index].body->length_body;
if( bytes_left > p_thread->chunk_size_recv )
bytes_left = p_thread->chunk_size_recv;
for(i = 0; i < interchunk_headers; i++)
{
i_ret = net_Read( p_thread, p_thread->fd, NULL, i_ret = net_Read( p_thread, p_thread->fd, NULL,
rtmp_packet->body->body + (i * AMF_PACKET_SIZE_VIDEO), AMF_PACKET_SIZE_VIDEO, true ); p_thread->rtmp_headers_recv[stream_index].body->body +
if( i_ret != AMF_PACKET_SIZE_VIDEO ) p_thread->rtmp_headers_recv[stream_index].body->length_body,
goto error2; bytes_left, true );
i_ret = net_Read( p_thread, p_thread->fd, NULL, &trash, 1, true ); if( i_ret != bytes_left )
if( i_ret != 1 ) goto error;
goto error2;
}
i_ret = net_Read( p_thread, p_thread->fd, NULL, p_thread->rtmp_headers_recv[stream_index].body->length_body += bytes_left;
rtmp_packet->body->body + (i * AMF_PACKET_SIZE_VIDEO),
rtmp_packet->body->length_body - (i * AMF_PACKET_SIZE_VIDEO), true );
if( i_ret != rtmp_packet->body->length_body - (i * AMF_PACKET_SIZE_VIDEO) )
goto error2;
return rtmp_packet; if( p_thread->rtmp_headers_recv[stream_index].length_body == p_thread->rtmp_headers_recv[stream_index].body->length_body )
{
rtmp_packet = (rtmp_packet_t *) malloc( sizeof( rtmp_packet_t ) );
error2: rtmp_packet->stream_index = stream_index;
free( rtmp_packet->body->body ); rtmp_packet->timestamp = p_thread->rtmp_headers_recv[stream_index].timestamp;
error: rtmp_packet->timestamp_relative = p_thread->rtmp_headers_recv[stream_index].timestamp_relative;
free( rtmp_packet->body ); rtmp_packet->content_type = p_thread->rtmp_headers_recv[stream_index].content_type;
free( rtmp_packet ); rtmp_packet->src_dst = p_thread->rtmp_headers_recv[stream_index].src_dst;
rtmp_packet->length_body = p_thread->rtmp_headers_recv[stream_index].length_body;
rtmp_packet->body = p_thread->rtmp_headers_recv[stream_index].body;
p_thread->rtmp_headers_recv[stream_index].body = NULL;
return rtmp_packet;
}
}
error:
msg_Err( p_thread, "rtmp_read_net_packet: net_Read error"); msg_Err( p_thread, "rtmp_read_net_packet: net_Read error");
return NULL; return NULL;
...@@ -1011,7 +1021,7 @@ error: ...@@ -1011,7 +1021,7 @@ error:
void void
rtmp_init_handler( rtmp_handler_t *rtmp_handler ) rtmp_init_handler( rtmp_handler_t *rtmp_handler )
{ {
rtmp_handler[RTMP_CONTENT_TYPE_CHUNK_SIZE] = rtmp_handler_null; rtmp_handler[RTMP_CONTENT_TYPE_CHUNK_SIZE] = rtmp_handler_chunk_size;
rtmp_handler[RTMP_CONTENT_TYPE_UNKNOWN_02] = rtmp_handler_null; rtmp_handler[RTMP_CONTENT_TYPE_UNKNOWN_02] = rtmp_handler_null;
rtmp_handler[RTMP_CONTENT_TYPE_BYTES_READ] = rtmp_handler_null; rtmp_handler[RTMP_CONTENT_TYPE_BYTES_READ] = rtmp_handler_null;
rtmp_handler[RTMP_CONTENT_TYPE_PING] = rtmp_handler_null; rtmp_handler[RTMP_CONTENT_TYPE_PING] = rtmp_handler_null;
...@@ -1030,30 +1040,40 @@ rtmp_init_handler( rtmp_handler_t *rtmp_handler ) ...@@ -1030,30 +1040,40 @@ rtmp_init_handler( rtmp_handler_t *rtmp_handler )
} }
static void static void
rtmp_handler_null( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ) rtmp_handler_null( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet )
{
free( rtmp_packet->body->body );
free( rtmp_packet->body );
free( rtmp_packet );
}
static void
rtmp_handler_chunk_size( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet )
{ {
p_thread->chunk_size_recv = ntoh32( *(uint32_t *) (rtmp_packet->body->body) );
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
free( rtmp_packet->body ); free( rtmp_packet->body );
free( rtmp_packet ); free( rtmp_packet );
} }
static void static void
rtmp_handler_audio_data( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ) rtmp_handler_audio_data( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet )
{ {
block_t *p_buffer; block_t *p_buffer;
if( !rtmp_control_thread->has_audio ) if( !p_thread->has_audio )
{ {
rtmp_control_thread->has_audio = 1; p_thread->has_audio = 1;
flv_get_metadata_audio( rtmp_packet, flv_get_metadata_audio( p_thread, rtmp_packet,
&rtmp_control_thread->metadata_stereo, &rtmp_control_thread->metadata_samplesize, &p_thread->metadata_stereo, &p_thread->metadata_samplesize,
&rtmp_control_thread->metadata_samplerate, &rtmp_control_thread->metadata_audiocodecid ); &p_thread->metadata_samplerate, &p_thread->metadata_audiocodecid );
} }
flv_rebuild( rtmp_control_thread, rtmp_packet ); flv_rebuild( p_thread, rtmp_packet );
p_buffer = flv_new_packet( rtmp_control_thread, rtmp_packet ); p_buffer = rtmp_new_block( p_thread, rtmp_packet->body->body, rtmp_packet->body->length_body );
block_FifoPut( rtmp_control_thread->p_fifo_media, p_buffer ); block_FifoPut( p_thread->p_fifo_input, p_buffer );
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
free( rtmp_packet->body ); free( rtmp_packet->body );
...@@ -1061,20 +1081,21 @@ rtmp_handler_audio_data( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet ...@@ -1061,20 +1081,21 @@ rtmp_handler_audio_data( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet
} }
static void static void
rtmp_handler_video_data( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ) rtmp_handler_video_data( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet )
{ {
block_t *p_buffer; block_t *p_buffer;
if( !rtmp_control_thread->has_video ) if( !p_thread->has_video )
{ {
rtmp_control_thread->has_video = 1; p_thread->has_video = 1;
flv_get_metadata_video( rtmp_packet, flv_get_metadata_video( p_thread, rtmp_packet,
&rtmp_control_thread->metadata_videocodecid, &rtmp_control_thread->metadata_frametype ); &p_thread->metadata_videocodecid, &p_thread->metadata_frametype );
} }
flv_rebuild( rtmp_control_thread, rtmp_packet );
p_buffer = flv_new_packet( rtmp_control_thread, rtmp_packet ); flv_rebuild( p_thread, rtmp_packet );
block_FifoPut( rtmp_control_thread->p_fifo_media, p_buffer ); p_buffer = rtmp_new_block( p_thread, rtmp_packet->body->body, rtmp_packet->body->length_body );
block_FifoPut( p_thread->p_fifo_input, p_buffer );
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
free( rtmp_packet->body ); free( rtmp_packet->body );
...@@ -1082,15 +1103,15 @@ rtmp_handler_video_data( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet ...@@ -1082,15 +1103,15 @@ rtmp_handler_video_data( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet
} }
static void static void
rtmp_handler_notify( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ) rtmp_handler_notify( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet )
{ {
block_t *p_buffer; block_t *p_buffer;
rtmp_control_thread->metadata_received = 1; p_thread->metadata_received = 1;
flv_rebuild( rtmp_control_thread, rtmp_packet ); flv_rebuild( p_thread, rtmp_packet );
p_buffer = flv_new_packet( rtmp_control_thread, rtmp_packet ); p_buffer = rtmp_new_block( p_thread, rtmp_packet->body->body, rtmp_packet->body->length_body );
block_FifoPut( rtmp_control_thread->p_fifo_media, p_buffer ); block_FifoPut( p_thread->p_fifo_input, p_buffer );
free( rtmp_packet->body->body ); free( rtmp_packet->body->body );
free( rtmp_packet->body ); free( rtmp_packet->body );
...@@ -1098,9 +1119,8 @@ rtmp_handler_notify( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t * ...@@ -1098,9 +1119,8 @@ rtmp_handler_notify( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *
} }
static void static void
rtmp_handler_invoke( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ) rtmp_handler_invoke( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet )
{ {
access_t *p_access = (access_t *) rtmp_control_thread;
rtmp_packet_t *tmp_rtmp_packet; rtmp_packet_t *tmp_rtmp_packet;
uint8_t *i, *end, *tmp_buffer; uint8_t *i, *end, *tmp_buffer;
double number; double number;
...@@ -1116,23 +1136,80 @@ rtmp_handler_invoke( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t * ...@@ -1116,23 +1136,80 @@ rtmp_handler_invoke( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *
i++; /* Pass over AMF_DATATYPE_NUMBER */ i++; /* Pass over AMF_DATATYPE_NUMBER */
number = amf_decode_number( &i ); number = amf_decode_number( &i );
msg_Dbg( rtmp_control_thread, "%s %.1f", string, number ); msg_Dbg( p_thread, "%s %.1f", string, number );
if( strcmp( "connect", string ) == 0 ) if( strcmp( "connect", string ) == 0 )
{ {
tmp_rtmp_packet = rtmp_encode_NetConnection_Connect_result( rtmp_control_thread, number ); /* Connection bandwith */
tmp_rtmp_packet = rtmp_encode_onBWDone( p_thread, AMF_CALL_ONBWDONE );
tmp_buffer = rtmp_encode_packet( p_thread, tmp_rtmp_packet );
i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded );
if( i_ret != tmp_rtmp_packet->length_encoded )
{
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
msg_Err( p_thread, "failed send connection bandwith" );
}
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
/* Server bandwith */
tmp_rtmp_packet = rtmp_encode_server_bw( p_thread, RTMP_SERVER_BW );
tmp_buffer = rtmp_encode_packet( p_thread, tmp_rtmp_packet );
i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded );
if( i_ret != tmp_rtmp_packet->length_encoded )
{
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
msg_Err( p_thread, "failed send server bandwith" );
}
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
/* Clear stream */
tmp_rtmp_packet = rtmp_encode_ping_clear_stream( p_thread, RTMP_SRC_DST_CONNECT_OBJECT );
tmp_buffer = rtmp_encode_packet( p_thread, tmp_rtmp_packet );
tmp_buffer = rtmp_encode_packet( p_access, tmp_rtmp_packet ); i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded );
if( i_ret != tmp_rtmp_packet->length_encoded )
{
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
msg_Err( p_thread, "failed send clear stream" );
}
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
/* Reply NetConnection.connect */ /* Reply NetConnection.connect */
i_ret = net_Write( rtmp_control_thread, rtmp_control_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded ); tmp_rtmp_packet = rtmp_encode_NetConnection_connect_result( p_thread, number );
tmp_buffer = rtmp_encode_packet( p_thread, tmp_rtmp_packet );
i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded );
if( i_ret != tmp_rtmp_packet->length_encoded ) if( i_ret != tmp_rtmp_packet->length_encoded )
{ {
free( tmp_rtmp_packet->body->body ); free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body ); free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet ); free( tmp_rtmp_packet );
free( tmp_buffer ); free( tmp_buffer );
msg_Err( rtmp_control_thread, "failed send reply NetConnection.connect" ); msg_Err( p_thread, "failed send reply NetConnection.connect" );
} }
free( tmp_rtmp_packet->body->body ); free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body ); free( tmp_rtmp_packet->body );
...@@ -1141,22 +1218,60 @@ rtmp_handler_invoke( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t * ...@@ -1141,22 +1218,60 @@ rtmp_handler_invoke( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *
} }
else if( strcmp( "createStream", string ) == 0 ) else if( strcmp( "createStream", string ) == 0 )
{ {
rtmp_control_thread->stream_client = number; p_thread->stream_client_id = number;
rtmp_control_thread->stream_server = number; p_thread->stream_server_id = RTMP_DEFAULT_STREAM_SERVER_ID;
tmp_rtmp_packet = rtmp_encode_createStream_result( rtmp_control_thread, number ); /* Reply createStream */
tmp_rtmp_packet = rtmp_encode_createStream_result( p_thread, p_thread->stream_client_id, p_thread->stream_server_id );
tmp_buffer = rtmp_encode_packet( p_access, tmp_rtmp_packet ); tmp_buffer = rtmp_encode_packet( p_thread, tmp_rtmp_packet );
/* Reply createStream */ i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded );
i_ret = net_Write( rtmp_control_thread, rtmp_control_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded ); if( i_ret != tmp_rtmp_packet->length_encoded )
{
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
msg_Err( p_thread, "failed send reply createStream" );
}
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
/* Reset stream */
tmp_rtmp_packet = rtmp_encode_ping_reset_stream( p_thread );
tmp_buffer = rtmp_encode_packet( p_thread, tmp_rtmp_packet );
i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded );
if( i_ret != tmp_rtmp_packet->length_encoded )
{
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
msg_Err( p_thread, "failed send reset stream" );
}
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
/* Clear stream */
tmp_rtmp_packet = rtmp_encode_ping_clear_stream( p_thread, RTMP_SRC_DST_CONNECT_OBJECT2 );
tmp_buffer = rtmp_encode_packet( p_thread, tmp_rtmp_packet );
i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded );
if( i_ret != tmp_rtmp_packet->length_encoded ) if( i_ret != tmp_rtmp_packet->length_encoded )
{ {
free( tmp_rtmp_packet->body->body ); free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body ); free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet ); free( tmp_rtmp_packet );
free( tmp_buffer ); free( tmp_buffer );
msg_Err( rtmp_control_thread, "failed send reply createStream" ); msg_Err( p_thread, "failed send clear stream" );
} }
free( tmp_rtmp_packet->body->body ); free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body ); free( tmp_rtmp_packet->body );
...@@ -1166,101 +1281,182 @@ rtmp_handler_invoke( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t * ...@@ -1166,101 +1281,182 @@ rtmp_handler_invoke( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *
else if( strcmp( "publish", string ) == 0 ) else if( strcmp( "publish", string ) == 0 )
{ {
i++; i++;
msg_Dbg( rtmp_control_thread, "Null" ); msg_Dbg( p_thread, "null" );
i++; i++;
string2 = amf_decode_string( &i ); string2 = amf_decode_string( &i );
msg_Dbg( rtmp_control_thread, "String: %s", string2 ); msg_Dbg( p_thread, "string: %s", string2 );
rtmp_control_thread->psz_publish = strdup( string2 ); p_thread->psz_publish = strdup( string2 );
free( string2 ); free( string2 );
} }
else if( strcmp( "play", string ) == 0 )
{
i++;
msg_Dbg( p_thread, "null" );
free( string ); i++;
string2 = amf_decode_string( &i );
msg_Dbg( p_thread, "string: %s", string2 );
while( i < end ) /* Reply NetStream.play.reset */
{ tmp_rtmp_packet = rtmp_encode_NetStream_play_reset_onStatus( p_thread, string2 );
if( *i == AMF_DATATYPE_NUMBER )
tmp_buffer = rtmp_encode_packet( p_thread, tmp_rtmp_packet );
i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded );
if( i_ret != tmp_rtmp_packet->length_encoded )
{ {
i++; free( tmp_rtmp_packet->body->body );
msg_Dbg( rtmp_control_thread, "Number: %le", amf_decode_number( &i ) ); free( tmp_rtmp_packet->body );
} free( tmp_rtmp_packet );
free( tmp_buffer );
msg_Err( p_thread, "failed send reply NetStream.play.reset" );
}
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
/* Reply NetStream.play.start */
tmp_rtmp_packet = rtmp_encode_NetStream_play_start_onStatus( p_thread, string2 );
tmp_buffer = rtmp_encode_packet( p_thread, tmp_rtmp_packet );
i_ret = net_Write( p_thread, p_thread->fd, NULL, tmp_buffer, tmp_rtmp_packet->length_encoded );
if( i_ret != tmp_rtmp_packet->length_encoded )
{
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
msg_Err( p_thread, "failed send reply NetStream.play.start" );
}
free( tmp_rtmp_packet->body->body );
free( tmp_rtmp_packet->body );
free( tmp_rtmp_packet );
free( tmp_buffer );
free( string2 );
p_thread->result_play = 0;
vlc_mutex_lock( &p_thread->lock );
vlc_cond_signal( &p_thread->wait );
vlc_mutex_unlock( &p_thread->lock );
}
free( string );
while( i < end )
{
if( *i == AMF_DATATYPE_NUMBER )
{
i++;
msg_Dbg( p_thread, "number: %le", amf_decode_number( &i ) );
}
else if( *i == AMF_DATATYPE_BOOLEAN ) else if( *i == AMF_DATATYPE_BOOLEAN )
{ {
i++; i++;
msg_Dbg( rtmp_control_thread, "Boolean: %s", amf_decode_boolean( &i ) ? "True" : "False" ); msg_Dbg( p_thread, "boolean: %s", amf_decode_boolean( &i ) ? "true" : "false" );
} }
else if( *i == AMF_DATATYPE_STRING ) else if( *i == AMF_DATATYPE_STRING )
{ {
i++; i++;
string = amf_decode_string( &i ); string = amf_decode_string( &i );
msg_Dbg( rtmp_control_thread, "String: %s", string ); msg_Dbg( p_thread, "string: %s", string );
free( string ); free( string );
} }
else if( *i == AMF_DATATYPE_OBJECT ) else if( *i == AMF_DATATYPE_OBJECT )
{ {
i++; i++;
msg_Dbg( rtmp_control_thread, "Object" ); msg_Dbg( p_thread, "object" );
while( ( string = amf_decode_object( &i ) ) != NULL ) while( ( string = amf_decode_object( &i ) ) != NULL )
{ {
if( *i == AMF_DATATYPE_NUMBER ) if( *i == AMF_DATATYPE_NUMBER )
{ {
i++; i++;
msg_Dbg( rtmp_control_thread, "Key: %s Value: %le", string, amf_decode_number( &i ) ); msg_Dbg( p_thread, "key: %s value: %le", string, amf_decode_number( &i ) );
} }
else if( *i == AMF_DATATYPE_BOOLEAN ) else if( *i == AMF_DATATYPE_BOOLEAN )
{ {
i++; i++;
msg_Dbg( rtmp_control_thread, "Key: %s Value: %s", string, amf_decode_boolean( &i ) ? "True" : "False" ); msg_Dbg( p_thread, "key: %s value: %s", string, amf_decode_boolean( &i ) ? "true" : "false" );
} }
else if( *i == AMF_DATATYPE_STRING ) else if( *i == AMF_DATATYPE_STRING )
{ {
i++; i++;
string2 = amf_decode_string( &i ); string2 = amf_decode_string( &i );
msg_Dbg( rtmp_control_thread, "Key: %s Value: %s", string, string2 ); msg_Dbg( p_thread, "key: %s value: %s", string, string2 );
if( strcmp( "code", string ) == 0 ) if( strcmp( "code", string ) == 0 )
{ {
if( strcmp( "NetConnection.Connect.Success", string2 ) == 0 ) if( strcmp( "NetConnection.Connect.Success", string2 ) == 0 )
{ {
rtmp_control_thread->result_connect = 0; p_thread->result_connect = 0;
vlc_mutex_lock( &rtmp_control_thread->lock ); vlc_mutex_lock( &p_thread->lock );
vlc_cond_signal( &rtmp_control_thread->wait ); vlc_cond_signal( &p_thread->wait );
vlc_mutex_unlock( &rtmp_control_thread->lock ); vlc_mutex_unlock( &p_thread->lock );
} }
else if( strcmp( "NetConnection.Connect.InvalidApp", string2 ) == 0 ) else if( strcmp( "NetConnection.Connect.InvalidApp", string2 ) == 0 )
{ {
rtmp_control_thread->b_die = 1; p_thread->b_die = 1;
vlc_mutex_lock( &rtmp_control_thread->lock ); vlc_mutex_lock( &p_thread->lock );
vlc_cond_signal( &rtmp_control_thread->wait ); vlc_cond_signal( &p_thread->wait );
vlc_mutex_unlock( &rtmp_control_thread->lock ); vlc_mutex_unlock( &p_thread->lock );
} }
else if( strcmp( "NetStream.Play.Start", string2 ) == 0 ) else if( strcmp( "NetStream.Play.Start", string2 ) == 0 )
{ {
rtmp_control_thread->result_play = 0; p_thread->result_play = 0;
vlc_mutex_lock( &p_thread->lock );
vlc_cond_signal( &p_thread->wait );
vlc_mutex_unlock( &p_thread->lock );
}
else if( strcmp( "NetStream.Play.Stop", string2 ) == 0 )
{
p_thread->result_stop = 1;
vlc_mutex_lock( &rtmp_control_thread->lock ); block_FifoWake( p_thread->p_fifo_input );
vlc_cond_signal( &rtmp_control_thread->wait );
vlc_mutex_unlock( &rtmp_control_thread->lock );
} }
} }
free( string2 ); free( string2 );
} }
else if( *i == AMF_DATATYPE_NULL) else if( *i == AMF_DATATYPE_NULL )
{
i++;
msg_Dbg( p_thread, "key: %s value: Null", string );
}
else if( *i == AMF_DATATYPE_UNDEFINED )
{ {
i++; i++;
msg_Dbg( rtmp_control_thread, "Key: %s Value: Null", string ); msg_Dbg( p_thread, "key: %s value: undefined (Null)", string );
}
else
{
i++;
msg_Warn( p_thread, "key: %s value: undefined AMF type", string );
} }
free( string ); free( string );
} }
msg_Dbg( rtmp_control_thread, "End of object" ); msg_Dbg( p_thread, "end of object" );
} }
else if( *i == AMF_DATATYPE_NULL) else if( *i == AMF_DATATYPE_NULL)
{ {
i++; i++;
msg_Dbg( rtmp_control_thread, "Null" ); msg_Dbg( p_thread, "null" );
}
else if( *i == AMF_DATATYPE_UNDEFINED )
{
i++;
msg_Dbg( p_thread, "undefined (null)" );
}
else
{
i++;
msg_Warn( p_thread, "undefined AMF type" );
} }
} }
...@@ -1270,6 +1466,7 @@ rtmp_handler_invoke( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t * ...@@ -1270,6 +1466,7 @@ rtmp_handler_invoke( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *
} }
/* length header calculated automatically based on last packet in the same channel */ /* length header calculated automatically based on last packet in the same channel */
/* timestamps passed are always absolute */
static rtmp_packet_t * static rtmp_packet_t *
rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t timestamp, uint8_t content_type, uint32_t src_dst, rtmp_body_t *body ) rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t timestamp, uint8_t content_type, uint32_t src_dst, rtmp_body_t *body )
{ {
...@@ -1277,7 +1474,9 @@ rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t ...@@ -1277,7 +1474,9 @@ rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t
rtmp_packet_t *rtmp_packet; rtmp_packet_t *rtmp_packet;
rtmp_packet = (rtmp_packet_t *) malloc( sizeof( rtmp_packet_t ) ); rtmp_packet = (rtmp_packet_t *) malloc( sizeof( rtmp_packet_t ) );
interchunk_headers = body->length_body / AMF_PACKET_SIZE_VIDEO; interchunk_headers = body->length_body / p_thread->chunk_size_send;
if( body->length_body % p_thread->chunk_size_send == 0 )
interchunk_headers--;
if( src_dst != p_thread->rtmp_headers_send[stream_index].src_dst ) if( src_dst != p_thread->rtmp_headers_send[stream_index].src_dst )
{ {
...@@ -1291,15 +1490,19 @@ rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t ...@@ -1291,15 +1490,19 @@ rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t
else if( content_type != p_thread->rtmp_headers_send[stream_index].content_type else if( content_type != p_thread->rtmp_headers_send[stream_index].content_type
|| body->length_body != p_thread->rtmp_headers_send[stream_index].length_body ) || body->length_body != p_thread->rtmp_headers_send[stream_index].length_body )
{ {
p_thread->rtmp_headers_send[stream_index].timestamp_relative = timestamp; p_thread->rtmp_headers_send[stream_index].timestamp_relative =
timestamp - p_thread->rtmp_headers_send[stream_index].timestamp;
p_thread->rtmp_headers_send[stream_index].timestamp = timestamp;
p_thread->rtmp_headers_send[stream_index].length_body = body->length_body; p_thread->rtmp_headers_send[stream_index].length_body = body->length_body;
p_thread->rtmp_headers_send[stream_index].content_type = content_type; p_thread->rtmp_headers_send[stream_index].content_type = content_type;
rtmp_packet->length_header = 8; rtmp_packet->length_header = 8;
} }
else if( timestamp != p_thread->rtmp_headers_send[stream_index].timestamp_relative ) else if( timestamp != p_thread->rtmp_headers_send[stream_index].timestamp )
{ {
p_thread->rtmp_headers_send[stream_index].timestamp_relative = timestamp; p_thread->rtmp_headers_send[stream_index].timestamp_relative =
timestamp - p_thread->rtmp_headers_send[stream_index].timestamp;
p_thread->rtmp_headers_send[stream_index].timestamp = timestamp;
rtmp_packet->length_header = 4; rtmp_packet->length_header = 4;
} }
...@@ -1307,7 +1510,7 @@ rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t ...@@ -1307,7 +1510,7 @@ rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t
{ {
rtmp_packet->length_header = 1; rtmp_packet->length_header = 1;
} }
/*TODO: puede que no haga falta guardar el timestamp relative */
rtmp_packet->stream_index = stream_index; rtmp_packet->stream_index = stream_index;
if( rtmp_packet->length_header == 12 ) if( rtmp_packet->length_header == 12 )
{ {
...@@ -1316,8 +1519,8 @@ rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t ...@@ -1316,8 +1519,8 @@ rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t
} }
else else
{ {
rtmp_packet->timestamp = p_thread->rtmp_headers_send[stream_index].timestamp + timestamp; rtmp_packet->timestamp = timestamp;
rtmp_packet->timestamp_relative = timestamp; rtmp_packet->timestamp_relative = p_thread->rtmp_headers_send[stream_index].timestamp_relative;
} }
rtmp_packet->length_encoded = rtmp_packet->length_header + body->length_body + interchunk_headers; rtmp_packet->length_encoded = rtmp_packet->length_header + body->length_body + interchunk_headers;
rtmp_packet->length_body = body->length_body; rtmp_packet->length_body = body->length_body;
...@@ -1334,46 +1537,80 @@ rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t ...@@ -1334,46 +1537,80 @@ rtmp_new_packet( rtmp_control_thread_t *p_thread, uint8_t stream_index, uint32_t
return rtmp_packet; return rtmp_packet;
} }
static block_t *
rtmp_new_block( rtmp_control_thread_t *p_thread, uint8_t *buffer, int32_t length_buffer )
{
block_t *p_buffer;
/* DOWN: p_thread->p_empty_blocks->i_depth */
while ( block_FifoCount( p_thread->p_empty_blocks ) > MAX_EMPTY_BLOCKS )
{
p_buffer = block_FifoGet( p_thread->p_empty_blocks );
block_Release( p_buffer );
}
/* DOWN: p_thread->p_empty_blocks->i_depth */
if( block_FifoCount( p_thread->p_empty_blocks ) == 0 )
{
p_buffer = block_New( p_thread, length_buffer );
}
else
{
p_buffer = block_FifoGet( p_thread->p_empty_blocks );
p_buffer = block_Realloc( p_buffer, 0, length_buffer );
}
p_buffer->i_buffer = length_buffer;
memcpy( p_buffer->p_buffer, buffer, p_buffer->i_buffer );
return p_buffer;
}
/* call sequence for each packet rtmp_new_packet -> rtmp_encode_packet -> send */ /* call sequence for each packet rtmp_new_packet -> rtmp_encode_packet -> send */
/* no parallelism allowed because of optimization in header length */ /* no parallelism allowed because of optimization in header length */
static uint8_t * uint8_t *
rtmp_encode_packet( access_t *p_access, rtmp_packet_t *rtmp_packet ) rtmp_encode_packet( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet )
{ {
uint8_t *out; uint8_t *out;
int interchunk_headers; int interchunk_headers;
uint32_t length_body, src_dst; uint32_t timestamp, length_body, src_dst;
int i, j; int i, j;
out = (uint8_t *) malloc( rtmp_packet->length_encoded * sizeof( uint8_t ) ); out = (uint8_t *) malloc( rtmp_packet->length_encoded * sizeof( uint8_t ) );
interchunk_headers = rtmp_packet->body->length_body / AMF_PACKET_SIZE_VIDEO; interchunk_headers = rtmp_packet->body->length_body / p_thread->chunk_size_send;
if( rtmp_packet->body->length_body % p_thread->chunk_size_send == 0 )
interchunk_headers--;
out[0] = rtmp_encode_header_size( p_access, rtmp_packet->length_header ) + rtmp_packet->stream_index; if( rtmp_packet->length_header == 12 )
{
/* Timestamp absolute */
timestamp = hton32( rtmp_packet->timestamp );
memcpy( out, &timestamp, sizeof( uint32_t ) );
if( rtmp_packet->length_header >= 4 ) src_dst = hton32( rtmp_packet->src_dst );
{ /* Timestamp */ memcpy( out + 8, &src_dst, sizeof( uint32_t ) );
out[1] = 0;
out[2] = 0;
out[3] = 0;
} }
if( rtmp_packet->length_header >= 8 ) if( rtmp_packet->length_header >= 8 )
{ {
/* Length without inter chunk headers */ /* Length without inter chunk headers */
length_body = hton32( rtmp_packet->body->length_body ); length_body = hton32( rtmp_packet->body->length_body );
memcpy( out + 3, &length_body, sizeof( uint32_t ) ); memcpy( out + 3, &length_body, sizeof( uint32_t ) );
out[3] = 0;
out[7] = rtmp_packet->content_type; out[7] = rtmp_packet->content_type;
} }
if( rtmp_packet->length_header >= 12 ) if( rtmp_packet->length_header >= 4 && rtmp_packet->length_header != 12 )
{ {
src_dst = hton32( rtmp_packet->src_dst ); /* Timestamp relative */
memcpy( out + 8, &src_dst, sizeof( uint32_t ) ); timestamp = hton32( rtmp_packet->timestamp_relative );
memcpy( out, &timestamp, sizeof( uint32_t ) );
} }
// Insert inter chunk headers out[0] = rtmp_encode_header_size( (vlc_object_t *) p_thread, rtmp_packet->length_header ) + rtmp_packet->stream_index;
/* Insert inter chunk headers */
for(i = 0, j = 0; i < rtmp_packet->body->length_body + interchunk_headers; i++, j++) for(i = 0, j = 0; i < rtmp_packet->body->length_body + interchunk_headers; i++, j++)
{ {
if( j % AMF_PACKET_SIZE_VIDEO == 0 && j != 0 ) if( j % p_thread->chunk_size_send == 0 && j != 0 )
out[rtmp_packet->length_header + i++] = RTMP_HEADER_SIZE_1 + rtmp_packet->stream_index; out[rtmp_packet->length_header + i++] = RTMP_HEADER_SIZE_1 + rtmp_packet->stream_index;
out[rtmp_packet->length_header + i] = rtmp_packet->body->body[j]; out[rtmp_packet->length_header + i] = rtmp_packet->body->body[j];
} }
...@@ -1382,14 +1619,64 @@ rtmp_encode_packet( access_t *p_access, rtmp_packet_t *rtmp_packet ) ...@@ -1382,14 +1619,64 @@ rtmp_encode_packet( access_t *p_access, rtmp_packet_t *rtmp_packet )
} }
static rtmp_packet_t * static rtmp_packet_t *
rtmp_encode_NetConnection_Connect_result( rtmp_control_thread_t *p_thread, double number ) rtmp_encode_onBWDone( rtmp_control_thread_t *p_thread, double number )
{
rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer;
/* Build onBWDone */
rtmp_body = rtmp_body_new( -1 );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "onBWDone" );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_STRING + strlen( "onBWDone" ) );
free( tmp_buffer );
tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER, &number );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer );
tmp_buffer = amf_encode_element( AMF_DATATYPE_NULL, NULL );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NULL );
free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_INVOKE,
0, RTMP_CONTENT_TYPE_INVOKE, RTMP_SRC_DST_CONNECT_OBJECT, rtmp_body );
free( rtmp_body->body );
free( rtmp_body );
return rtmp_packet;
}
static rtmp_packet_t *
rtmp_encode_server_bw( rtmp_control_thread_t *p_thread, uint32_t number )
{
rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body;
/* Build server bw */
rtmp_body = rtmp_body_new( -1 );
rtmp_body_append( rtmp_body, (uint8_t *) &number, sizeof( uint32_t ) );
rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_CONTROL,
0, RTMP_CONTENT_TYPE_SERVER_BW, RTMP_SRC_DST_CONNECT_OBJECT, rtmp_body );
free( rtmp_body->body );
free( rtmp_body );
return rtmp_packet;
}
static rtmp_packet_t *
rtmp_encode_NetConnection_connect_result( rtmp_control_thread_t *p_thread, double number )
{ {
rtmp_packet_t *rtmp_packet; rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body; rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer; uint8_t *tmp_buffer;
/* Build NetConnection.connect result */ /* Build NetConnection.connect result */
rtmp_body = rtmp_body_new(); rtmp_body = rtmp_body_new( -1 );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "_result" ); tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "_result" );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
...@@ -1413,27 +1700,27 @@ rtmp_encode_NetConnection_Connect_result( rtmp_control_thread_t *p_thread, doubl ...@@ -1413,27 +1700,27 @@ rtmp_encode_NetConnection_Connect_result( rtmp_control_thread_t *p_thread, doubl
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "level" ) + AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "level" ) +
AMF_DATATYPE_SIZE_STRING + strlen( "status" ) ); AMF_DATATYPE_SIZE_STRING + strlen( "status" ) );
free ( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "code", tmp_buffer = amf_encode_object_variable( "code",
AMF_DATATYPE_STRING, "NetConnection.Connect.Success" ); AMF_DATATYPE_STRING, "NetConnection.Connect.Success" );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "code" ) + AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "code" ) +
AMF_DATATYPE_SIZE_STRING + strlen( "NetConnection.Connect.Success" ) ); AMF_DATATYPE_SIZE_STRING + strlen( "NetConnection.Connect.Success" ) );
free ( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "description", tmp_buffer = amf_encode_object_variable( "description",
AMF_DATATYPE_STRING, "Connection succeeded." ); AMF_DATATYPE_STRING, "Connection succeeded." );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "description" ) + AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "description" ) +
AMF_DATATYPE_SIZE_STRING + strlen( "Connection succeeded" ) ); AMF_DATATYPE_SIZE_STRING + strlen( "Connection succeeded." ) );
free ( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_element ( AMF_DATATYPE_END_OF_OBJECT, NULL ); tmp_buffer = amf_encode_element ( AMF_DATATYPE_END_OF_OBJECT, NULL );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_END_OF_OBJECT ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_END_OF_OBJECT );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_thread, RTMP_AMF_STREAM_INDEX_DEFAULT, rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_INVOKE,
0, RTMP_CONTENT_TYPE_INVOKE, 0, rtmp_body ); 0, RTMP_CONTENT_TYPE_INVOKE, 0, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
...@@ -1442,20 +1729,102 @@ rtmp_encode_NetConnection_Connect_result( rtmp_control_thread_t *p_thread, doubl ...@@ -1442,20 +1729,102 @@ rtmp_encode_NetConnection_Connect_result( rtmp_control_thread_t *p_thread, doubl
} }
static rtmp_packet_t * static rtmp_packet_t *
rtmp_encode_createStream_result( rtmp_control_thread_t *p_thread, double number ) rtmp_encode_createStream_result( rtmp_control_thread_t *p_thread, double stream_client_id, double stream_server_id )
{ {
rtmp_packet_t *rtmp_packet; rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body; rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer; uint8_t *tmp_buffer;
/* Build createStream result */ /* Build createStream result */
rtmp_body = rtmp_body_new(); rtmp_body = rtmp_body_new( -1 );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "_result" ); tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "_result" );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_STRING + strlen( "_result" ) ); AMF_DATATYPE_SIZE_STRING + strlen( "_result" ) );
free( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER, &stream_client_id );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer );
tmp_buffer = amf_encode_element( AMF_DATATYPE_NULL, NULL );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NULL );
free( tmp_buffer );
tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER, &stream_server_id );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_INVOKE,
0, RTMP_CONTENT_TYPE_INVOKE, 0, rtmp_body );
free( rtmp_body->body );
free( rtmp_body );
return rtmp_packet;
}
static rtmp_packet_t *
rtmp_encode_ping_reset_stream( rtmp_control_thread_t *p_thread )
{
rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer;
/* Build ping reset stream */
rtmp_body = rtmp_body_new( -1 );
tmp_buffer = rtmp_encode_ping( RTMP_PING_RESET_STREAM, RTMP_SRC_DST_CONNECT_OBJECT2, 0, 0 );
rtmp_body_append( rtmp_body, tmp_buffer, RTMP_PING_SIZE_RESET_STREAM );
free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_CONTROL,
0, RTMP_CONTENT_TYPE_PING, 0, rtmp_body );
free( rtmp_body->body );
free( rtmp_body );
return rtmp_packet;
}
static rtmp_packet_t *
rtmp_encode_ping_clear_stream( rtmp_control_thread_t *p_thread, uint32_t src_dst )
{
rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer;
/* Build ping clear stream */
rtmp_body = rtmp_body_new( -1 );
tmp_buffer = rtmp_encode_ping( RTMP_PING_CLEAR_STREAM, src_dst, 0, 0 );
rtmp_body_append( rtmp_body, tmp_buffer, RTMP_PING_SIZE_CLEAR_STREAM );
free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_CONTROL,
0, RTMP_CONTENT_TYPE_PING, 0, rtmp_body );
free( rtmp_body->body );
free( rtmp_body );
return rtmp_packet;
}
static rtmp_packet_t *
rtmp_encode_NetStream_play_reset_onStatus( rtmp_control_thread_t *p_thread, char *psz_media )
{
rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer;
double number;
char *description;
/* Build NetStream.play.reset onStatus */
rtmp_body = rtmp_body_new( -1 );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "onStatus" );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_STRING + strlen( "onStatus" ) );
free( tmp_buffer );
number = 1; /* TODO: review this number*/
tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER, &number ); tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER, &number );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer ); free( tmp_buffer );
...@@ -1464,12 +1833,134 @@ rtmp_encode_createStream_result( rtmp_control_thread_t *p_thread, double number ...@@ -1464,12 +1833,134 @@ rtmp_encode_createStream_result( rtmp_control_thread_t *p_thread, double number
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NULL ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NULL );
free( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_element( AMF_DATATYPE_OBJECT, NULL );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_OBJECT );
free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "level",
AMF_DATATYPE_STRING, "status" );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "level" ) +
AMF_DATATYPE_SIZE_STRING + strlen( "status" ) );
free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "code",
AMF_DATATYPE_STRING, "NetStream.Play.Reset" );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "code" ) +
AMF_DATATYPE_SIZE_STRING + strlen( "NetStream.Play.Reset" ) );
free( tmp_buffer );
description = (char *) malloc( strlen( "Playing and resetting ") + strlen( psz_media ) + strlen( "." ) + 1 );
sprintf( description, "Playing and resetting %s.", psz_media );
tmp_buffer = amf_encode_object_variable( "description",
AMF_DATATYPE_STRING, description );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "description" ) +
AMF_DATATYPE_SIZE_STRING + strlen( description ) );
free( tmp_buffer );
free( description );
tmp_buffer = amf_encode_object_variable( "details",
AMF_DATATYPE_STRING, psz_media );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "details" ) +
AMF_DATATYPE_SIZE_STRING + strlen( psz_media ) );
free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "clientid",
AMF_DATATYPE_NUMBER, &p_thread->stream_client_id );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "clientid" ) +
AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer );
tmp_buffer = amf_encode_element ( AMF_DATATYPE_END_OF_OBJECT, NULL );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_END_OF_OBJECT );
free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_NOTIFY,
0, RTMP_CONTENT_TYPE_INVOKE, RTMP_SRC_DST_DEFAULT, rtmp_body );
free( rtmp_body->body );
free( rtmp_body );
return rtmp_packet;
}
static rtmp_packet_t *
rtmp_encode_NetStream_play_start_onStatus( rtmp_control_thread_t *p_thread, char *psz_media )
{
rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer;
double number;
char *description;
/* Build NetStream.play.start onStatus */
rtmp_body = rtmp_body_new( -1 );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "onStatus" );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_STRING + strlen( "onStatus" ) );
free( tmp_buffer );
number = 1; /* TODO: review this number*/
tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER, &number ); tmp_buffer = amf_encode_element( AMF_DATATYPE_NUMBER, &number );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_thread, RTMP_AMF_STREAM_INDEX_DEFAULT, tmp_buffer = amf_encode_element( AMF_DATATYPE_NULL, NULL );
0, RTMP_CONTENT_TYPE_INVOKE, 0, rtmp_body ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_NULL );
free( tmp_buffer );
tmp_buffer = amf_encode_element( AMF_DATATYPE_OBJECT, NULL );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_OBJECT );
free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "level",
AMF_DATATYPE_STRING, "status" );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "level" ) +
AMF_DATATYPE_SIZE_STRING + strlen( "status" ) );
free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "code",
AMF_DATATYPE_STRING, "NetStream.Play.Start" );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "code" ) +
AMF_DATATYPE_SIZE_STRING + strlen( "NetStream.Play.Start" ) );
free( tmp_buffer );
description = (char *) malloc( strlen( "Started playing ") + strlen( psz_media ) + strlen( "." ) + 1 );
sprintf( description, "Started playing %s.", psz_media );
tmp_buffer = amf_encode_object_variable( "description",
AMF_DATATYPE_STRING, description );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "description" ) +
AMF_DATATYPE_SIZE_STRING + strlen( description ) );
free( tmp_buffer );
free( description );
tmp_buffer = amf_encode_object_variable( "details",
AMF_DATATYPE_STRING, psz_media );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "details" ) +
AMF_DATATYPE_SIZE_STRING + strlen( psz_media ) );
free( tmp_buffer );
tmp_buffer = amf_encode_object_variable( "clientid",
AMF_DATATYPE_NUMBER, &p_thread->stream_client_id );
rtmp_body_append( rtmp_body, tmp_buffer,
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "clientid" ) +
AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer );
tmp_buffer = amf_encode_element ( AMF_DATATYPE_END_OF_OBJECT, NULL );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_END_OF_OBJECT );
free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_thread, RTMP_DEFAULT_STREAM_INDEX_NOTIFY,
0, RTMP_CONTENT_TYPE_INVOKE, RTMP_SRC_DST_DEFAULT, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
...@@ -1477,7 +1968,7 @@ rtmp_encode_createStream_result( rtmp_control_thread_t *p_thread, double number ...@@ -1477,7 +1968,7 @@ rtmp_encode_createStream_result( rtmp_control_thread_t *p_thread, double number
} }
static uint8_t static uint8_t
rtmp_encode_header_size( access_t *p_access, uint8_t header_size ) rtmp_encode_header_size( vlc_object_t *p_this, uint8_t header_size )
{ {
if( header_size == 1 ) if( header_size == 1 )
return RTMP_HEADER_SIZE_1; return RTMP_HEADER_SIZE_1;
...@@ -1489,7 +1980,7 @@ rtmp_encode_header_size( access_t *p_access, uint8_t header_size ) ...@@ -1489,7 +1980,7 @@ rtmp_encode_header_size( access_t *p_access, uint8_t header_size )
return RTMP_HEADER_SIZE_12; return RTMP_HEADER_SIZE_12;
else else
{ {
msg_Err( p_access, "invalid header size for encoding" ); msg_Err( p_this, "invalid header size for encoding" );
return 0; return 0;
} }
} }
...@@ -1512,30 +2003,52 @@ rtmp_decode_header_size( vlc_object_t *p_this, uint8_t header_size ) ...@@ -1512,30 +2003,52 @@ rtmp_decode_header_size( vlc_object_t *p_this, uint8_t header_size )
} }
} }
static uint8_t
rtmp_get_stream_index( uint8_t content_type )
{
if( content_type == RTMP_CONTENT_TYPE_AUDIO_DATA )
return RTMP_DEFAULT_STREAM_INDEX_AUDIO_DATA;
else if( content_type == RTMP_CONTENT_TYPE_VIDEO_DATA )
return RTMP_DEFAULT_STREAM_INDEX_VIDEO_DATA;
else if( content_type == RTMP_CONTENT_TYPE_NOTIFY )
return RTMP_DEFAULT_STREAM_INDEX_NOTIFY;
else
return -1;
}
/***************************************************************************** /*****************************************************************************
* Body handling implementation: * Body handling implementation:
******************************************************************************/ ******************************************************************************/
static rtmp_body_t * rtmp_body_t *
rtmp_body_new( void ) rtmp_body_new( int length_buffer )
{ {
rtmp_body_t *rtmp_body; rtmp_body_t *rtmp_body;
rtmp_body = (rtmp_body_t *) malloc( sizeof( rtmp_body_t ) ); rtmp_body = (rtmp_body_t *) malloc( sizeof( rtmp_body_t ) );
rtmp_body->length_body = 0; rtmp_body->length_body = 0;
rtmp_body->length_buffer = RTMP_BODY_SIZE_ALLOC; if( length_buffer < 0 )
rtmp_body->body = (uint8_t *) malloc( RTMP_BODY_SIZE_ALLOC * sizeof( uint8_t ) ); rtmp_body->length_buffer = RTMP_BODY_SIZE_ALLOC;
else
rtmp_body->length_buffer = length_buffer;
rtmp_body->body = (uint8_t *) malloc( rtmp_body->length_buffer * sizeof( uint8_t ) );
return rtmp_body; return rtmp_body;
} }
void
rtmp_body_reset( rtmp_body_t *rtmp_body )
{
rtmp_body->length_body = 0;
}
static void static void
rtmp_body_append( rtmp_body_t *rtmp_body, uint8_t *buffer, uint8_t length ) rtmp_body_append( rtmp_body_t *rtmp_body, uint8_t *buffer, uint32_t length )
{ {
if( rtmp_body->length_body + length > rtmp_body->length_buffer ) if( rtmp_body->length_body + length > rtmp_body->length_buffer )
{ {
rtmp_body->length_buffer += RTMP_BODY_SIZE_ALLOC; rtmp_body->length_buffer = rtmp_body->length_body + length;
rtmp_body->body = (uint8_t *) realloc( rtmp_body->body, rtmp_body->length_buffer ); rtmp_body->body = (uint8_t *) realloc( rtmp_body->body, rtmp_body->length_buffer * sizeof( uint8_t ) );
} }
memcpy( rtmp_body->body + rtmp_body->length_body, buffer, length ); memcpy( rtmp_body->body + rtmp_body->length_body, buffer, length );
...@@ -1546,7 +2059,7 @@ rtmp_body_append( rtmp_body_t *rtmp_body, uint8_t *buffer, uint8_t length ) ...@@ -1546,7 +2059,7 @@ rtmp_body_append( rtmp_body_t *rtmp_body, uint8_t *buffer, uint8_t length )
* RTMP ping implementation: * RTMP ping implementation:
******************************************************************************/ ******************************************************************************/
static uint8_t * static uint8_t *
ping_encode( uint16_t type, uint32_t src_dst, uint32_t third_arg, uint32_t fourth_arg ) rtmp_encode_ping( uint16_t type, uint32_t src_dst, uint32_t third_arg, uint32_t fourth_arg )
{ {
uint8_t *out; uint8_t *out;
...@@ -1561,10 +2074,11 @@ ping_encode( uint16_t type, uint32_t src_dst, uint32_t third_arg, uint32_t fourt ...@@ -1561,10 +2074,11 @@ ping_encode( uint16_t type, uint32_t src_dst, uint32_t third_arg, uint32_t fourt
third_arg = hton32( third_arg ); third_arg = hton32( third_arg );
memcpy( out + 6, &third_arg, sizeof( uint32_t ) ); memcpy( out + 6, &third_arg, sizeof( uint32_t ) );
} }
/* else if( type == RTMP_PING_RESET_STREAM ) TODO: research this else if( type == RTMP_PING_RESET_STREAM )
{ {
out = (uint8_t *) malloc( RTMP_PING_SIZE_RESET_STREAM * sizeof( uint8_t ) );
} }
else if( type == RTMP_PING_CLIENT_FROM_SERVER ) /* else if( type == RTMP_PING_CLIENT_FROM_SERVER ) TODO: research this
{ {
} }
else if( type == RTMP_PING_PONG_FROM_CLIENT ) else if( type == RTMP_PING_PONG_FROM_CLIENT )
...@@ -1762,36 +2276,8 @@ amf_decode_object( uint8_t **buffer ) ...@@ -1762,36 +2276,8 @@ amf_decode_object( uint8_t **buffer )
/***************************************************************************** /*****************************************************************************
* FLV rebuilding implementation: * FLV rebuilding implementation:
******************************************************************************/ ******************************************************************************/
static block_t *
flv_new_packet( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet )
{
block_t *p_buffer;
/* DOWN: p_thread->p_empty_blocks->i_depth */
while ( block_FifoCount( p_thread->p_empty_blocks ) > MAX_EMPTY_BLOCKS )
{
p_buffer = block_FifoGet( p_thread->p_empty_blocks );
block_Release( p_buffer );
}
/* DOWN: p_thread->p_empty_blocks->i_depth */
if( block_FifoCount( p_thread->p_empty_blocks ) == 0 )
{
p_buffer = block_New( p_thread, rtmp_packet->body->length_body );
}
else
{
p_buffer = block_FifoGet( p_thread->p_empty_blocks );
p_buffer = block_Realloc( p_buffer, 0, rtmp_packet->body->length_body );
}
p_buffer->i_buffer = rtmp_packet->body->length_body;
memcpy( p_buffer->p_buffer, rtmp_packet->body->body, p_buffer->i_buffer );
return p_buffer;
}
static void static void
flv_rebuild( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ) flv_rebuild( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet )
{ {
uint32_t length_tag, timestamp; uint32_t length_tag, timestamp;
...@@ -1801,8 +2287,8 @@ flv_rebuild( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_pac ...@@ -1801,8 +2287,8 @@ flv_rebuild( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_pac
rtmp_packet->body->body, rtmp_packet->body->length_body ); rtmp_packet->body->body, rtmp_packet->body->length_body );
/* Insert tag */ /* Insert tag */
rtmp_control_thread->flv_tag_previous_tag_size = hton32( rtmp_control_thread->flv_tag_previous_tag_size ); p_thread->flv_tag_previous_tag_size = hton32( p_thread->flv_tag_previous_tag_size );
memcpy( rtmp_packet->body->body, &rtmp_control_thread->flv_tag_previous_tag_size, sizeof( uint32_t ) ); memcpy( rtmp_packet->body->body, &p_thread->flv_tag_previous_tag_size, sizeof( uint32_t ) );
/* Fill backwards because of overlapping*/ /* Fill backwards because of overlapping*/
rtmp_packet->body->body[11] = 0x00; rtmp_packet->body->body[11] = 0x00;
...@@ -1819,7 +2305,7 @@ flv_rebuild( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_pac ...@@ -1819,7 +2305,7 @@ flv_rebuild( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_pac
rtmp_packet->body->body[13] = 0x00; rtmp_packet->body->body[13] = 0x00;
rtmp_packet->body->body[14] = 0x00; rtmp_packet->body->body[14] = 0x00;
rtmp_control_thread->flv_tag_previous_tag_size = rtmp_packet->body->length_body + FLV_TAG_SIZE; p_thread->flv_tag_previous_tag_size = rtmp_packet->body->length_body + FLV_TAG_SIZE;
/* Update size */ /* Update size */
rtmp_packet->body->length_body += FLV_TAG_PREVIOUS_TAG_SIZE + FLV_TAG_SIZE; rtmp_packet->body->length_body += FLV_TAG_PREVIOUS_TAG_SIZE + FLV_TAG_SIZE;
...@@ -1827,30 +2313,36 @@ flv_rebuild( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_pac ...@@ -1827,30 +2313,36 @@ flv_rebuild( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_pac
} }
static void static void
flv_get_metadata_audio( rtmp_packet_t *packet_audio, uint8_t *stereo, uint8_t *audiosamplesize, uint8_t *audiosamplerate, uint8_t *audiocodecid ) flv_get_metadata_audio( rtmp_control_thread_t *p_thread, rtmp_packet_t *packet_audio, uint8_t *stereo, uint8_t *audiosamplesize, uint32_t *audiosamplerate, uint8_t *audiocodecid )
{ {
uint8_t data_audio; uint8_t data_audio;
data_audio = *(packet_audio->body->body + packet_audio->length_header); data_audio = *packet_audio->body->body;
if( ( data_audio & FLV_AUDIO_STEREO_MASK ) == FLV_AUDIO_STEREO_MONO ) if( ( data_audio & FLV_AUDIO_STEREO_MASK ) == FLV_AUDIO_STEREO_MONO )
*stereo = FLV_AUDIO_STEREO_MONO; *stereo = FLV_AUDIO_STEREO_MONO;
else if( ( data_audio & FLV_AUDIO_STEREO_MASK ) == FLV_AUDIO_STEREO_STEREO ) else if( ( data_audio & FLV_AUDIO_STEREO_MASK ) == FLV_AUDIO_STEREO_STEREO )
*stereo = FLV_AUDIO_STEREO_STEREO; *stereo = FLV_AUDIO_STEREO_STEREO;
else
msg_Warn( p_thread, "unknown metadata audio stereo" );
if( ( data_audio & FLV_AUDIO_SIZE_MASK ) == FLV_AUDIO_SIZE_8_BIT ) if( ( data_audio & FLV_AUDIO_SIZE_MASK ) == FLV_AUDIO_SIZE_8_BIT )
*audiosamplesize = FLV_AUDIO_SIZE_8_BIT >> 1; *audiosamplesize = FLV_AUDIO_SIZE_8_BIT >> 1;
else if( ( data_audio & FLV_AUDIO_SIZE_MASK ) == FLV_AUDIO_SIZE_16_BIT ) else if( ( data_audio & FLV_AUDIO_SIZE_MASK ) == FLV_AUDIO_SIZE_16_BIT )
*audiosamplesize = FLV_AUDIO_SIZE_16_BIT >> 1; *audiosamplesize = FLV_AUDIO_SIZE_16_BIT >> 1;
else
msg_Warn( p_thread, "unknown metadata audio sample size" );
if( ( data_audio & FLV_AUDIO_RATE_MASK ) == FLV_AUDIO_RATE_5_5_KHZ ) if( ( data_audio & FLV_AUDIO_RATE_MASK ) == FLV_AUDIO_RATE_5_5_KHZ )
*audiosamplerate = FLV_AUDIO_RATE_5_5_KHZ >> 2; *audiosamplerate = 5512;
else if( ( data_audio & FLV_AUDIO_RATE_MASK ) == FLV_AUDIO_RATE_11_KHZ ) else if( ( data_audio & FLV_AUDIO_RATE_MASK ) == FLV_AUDIO_RATE_11_KHZ )
*audiosamplerate = FLV_AUDIO_RATE_11_KHZ >> 2; *audiosamplerate = 11025;
else if( ( data_audio & FLV_AUDIO_RATE_MASK ) == FLV_AUDIO_RATE_22_KHZ ) else if( ( data_audio & FLV_AUDIO_RATE_MASK ) == FLV_AUDIO_RATE_22_KHZ )
*audiosamplerate = FLV_AUDIO_RATE_22_KHZ >> 2; *audiosamplerate = 22050;
else if( ( data_audio & FLV_AUDIO_RATE_MASK ) == FLV_AUDIO_RATE_44_KHZ ) else if( ( data_audio & FLV_AUDIO_RATE_MASK ) == FLV_AUDIO_RATE_44_KHZ )
*audiosamplerate = FLV_AUDIO_RATE_44_KHZ >> 2; *audiosamplerate = 44100;
else
msg_Warn( p_thread, "unknown metadata audio sample rate" );
if( ( data_audio & FLV_AUDIO_CODEC_ID_MASK ) == FLV_AUDIO_CODEC_ID_UNCOMPRESSED ) if( ( data_audio & FLV_AUDIO_CODEC_ID_MASK ) == FLV_AUDIO_CODEC_ID_UNCOMPRESSED )
*audiocodecid = FLV_AUDIO_CODEC_ID_UNCOMPRESSED >> 4; *audiocodecid = FLV_AUDIO_CODEC_ID_UNCOMPRESSED >> 4;
...@@ -1862,14 +2354,16 @@ flv_get_metadata_audio( rtmp_packet_t *packet_audio, uint8_t *stereo, uint8_t *a ...@@ -1862,14 +2354,16 @@ flv_get_metadata_audio( rtmp_packet_t *packet_audio, uint8_t *stereo, uint8_t *a
*audiocodecid = FLV_AUDIO_CODEC_ID_NELLYMOSER_8KHZ_MONO >> 4; *audiocodecid = FLV_AUDIO_CODEC_ID_NELLYMOSER_8KHZ_MONO >> 4;
else if( ( data_audio & FLV_AUDIO_CODEC_ID_MASK ) == FLV_AUDIO_CODEC_ID_NELLYMOSER ) else if( ( data_audio & FLV_AUDIO_CODEC_ID_MASK ) == FLV_AUDIO_CODEC_ID_NELLYMOSER )
*audiocodecid = FLV_AUDIO_CODEC_ID_NELLYMOSER >> 4; *audiocodecid = FLV_AUDIO_CODEC_ID_NELLYMOSER >> 4;
else
msg_Warn( p_thread, "unknown metadata audio codec id" );
} }
static void static void
flv_get_metadata_video( rtmp_packet_t *packet_video, uint8_t *videocodecid, uint8_t *frametype ) flv_get_metadata_video( rtmp_control_thread_t *p_thread, rtmp_packet_t *packet_video, uint8_t *videocodecid, uint8_t *frametype )
{ {
uint8_t data_video; uint8_t data_video;
data_video = *(packet_video->body->body + packet_video->length_header); data_video = *packet_video->body->body;
if( ( data_video & FLV_VIDEO_CODEC_ID_MASK ) == FLV_VIDEO_CODEC_ID_SORENSEN_H263 ) if( ( data_video & FLV_VIDEO_CODEC_ID_MASK ) == FLV_VIDEO_CODEC_ID_SORENSEN_H263 )
*videocodecid = FLV_VIDEO_CODEC_ID_SORENSEN_H263; *videocodecid = FLV_VIDEO_CODEC_ID_SORENSEN_H263;
...@@ -1881,6 +2375,8 @@ flv_get_metadata_video( rtmp_packet_t *packet_video, uint8_t *videocodecid, uint ...@@ -1881,6 +2375,8 @@ flv_get_metadata_video( rtmp_packet_t *packet_video, uint8_t *videocodecid, uint
*videocodecid = FLV_VIDEO_CODEC_ID_ON2_VP6_ALPHA; *videocodecid = FLV_VIDEO_CODEC_ID_ON2_VP6_ALPHA;
else if( ( data_video & FLV_VIDEO_CODEC_ID_MASK ) == FLV_VIDEO_CODEC_ID_SCREEN_VIDEO_2 ) else if( ( data_video & FLV_VIDEO_CODEC_ID_MASK ) == FLV_VIDEO_CODEC_ID_SCREEN_VIDEO_2 )
*videocodecid = FLV_VIDEO_CODEC_ID_SCREEN_VIDEO_2; *videocodecid = FLV_VIDEO_CODEC_ID_SCREEN_VIDEO_2;
else
msg_Warn( p_thread, "unknown metadata video codec id" );
if( ( data_video & FLV_VIDEO_FRAME_TYPE_MASK ) == FLV_VIDEO_FRAME_TYPE_KEYFRAME ) if( ( data_video & FLV_VIDEO_FRAME_TYPE_MASK ) == FLV_VIDEO_FRAME_TYPE_KEYFRAME )
*frametype = FLV_VIDEO_FRAME_TYPE_KEYFRAME >> 4; *frametype = FLV_VIDEO_FRAME_TYPE_KEYFRAME >> 4;
...@@ -1888,17 +2384,19 @@ flv_get_metadata_video( rtmp_packet_t *packet_video, uint8_t *videocodecid, uint ...@@ -1888,17 +2384,19 @@ flv_get_metadata_video( rtmp_packet_t *packet_video, uint8_t *videocodecid, uint
*frametype = FLV_VIDEO_FRAME_TYPE_INTER_FRAME >> 4; *frametype = FLV_VIDEO_FRAME_TYPE_INTER_FRAME >> 4;
else if( ( data_video & FLV_VIDEO_FRAME_TYPE_MASK ) == FLV_VIDEO_FRAME_TYPE_DISPOSABLE_INTER_FRAME ) else if( ( data_video & FLV_VIDEO_FRAME_TYPE_MASK ) == FLV_VIDEO_FRAME_TYPE_DISPOSABLE_INTER_FRAME )
*frametype = FLV_VIDEO_FRAME_TYPE_DISPOSABLE_INTER_FRAME >> 4; *frametype = FLV_VIDEO_FRAME_TYPE_DISPOSABLE_INTER_FRAME >> 4;
else
msg_Warn( p_thread, "unknown metadata video frame type" );
} }
static rtmp_packet_t * static rtmp_packet_t *
flv_build_onMetaData( access_t *p_access, uint64_t duration, uint8_t stereo, uint8_t audiosamplesize, uint8_t audiosamplerate, uint8_t audiocodecid, uint8_t videocodecid ) flv_build_onMetaData( access_t *p_access, uint64_t duration, uint8_t stereo, uint8_t audiosamplesize, uint32_t audiosamplerate, uint8_t audiocodecid, uint8_t videocodecid )
{ {
rtmp_packet_t *rtmp_packet; rtmp_packet_t *rtmp_packet;
rtmp_body_t *rtmp_body; rtmp_body_t *rtmp_body;
uint8_t *tmp_buffer; uint8_t *tmp_buffer;
uint64_t number; double number;
rtmp_body = rtmp_body_new(); rtmp_body = rtmp_body_new( -1 );
tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "onMetaData" ); tmp_buffer = amf_encode_element( AMF_DATATYPE_STRING, "onMetaData" );
rtmp_body_append( rtmp_body, tmp_buffer, rtmp_body_append( rtmp_body, tmp_buffer,
...@@ -1956,13 +2454,12 @@ flv_build_onMetaData( access_t *p_access, uint64_t duration, uint8_t stereo, uin ...@@ -1956,13 +2454,12 @@ flv_build_onMetaData( access_t *p_access, uint64_t duration, uint8_t stereo, uin
AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "videocodecid" ) + AMF_DATATYPE_SIZE_OBJECT_VARIABLE + strlen( "videocodecid" ) +
AMF_DATATYPE_SIZE_NUMBER ); AMF_DATATYPE_SIZE_NUMBER );
free( tmp_buffer ); free( tmp_buffer );
tmp_buffer = amf_encode_element( AMF_DATATYPE_END_OF_OBJECT, NULL ); tmp_buffer = amf_encode_element( AMF_DATATYPE_END_OF_OBJECT, NULL );
rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_END_OF_OBJECT ); rtmp_body_append( rtmp_body, tmp_buffer, AMF_DATATYPE_SIZE_END_OF_OBJECT );
free( tmp_buffer ); free( tmp_buffer );
rtmp_packet = rtmp_new_packet( p_access->p_sys->p_thread, RTMP_AMF_STREAM_INDEX_DEFAULT, rtmp_packet = rtmp_new_packet( p_access->p_sys->p_thread, RTMP_DEFAULT_STREAM_INDEX_INVOKE,
0, RTMP_CONTENT_TYPE_NOTIFY, 0, rtmp_body ); 0, RTMP_CONTENT_TYPE_NOTIFY, 0, rtmp_body );
free( rtmp_body->body ); free( rtmp_body->body );
free( rtmp_body ); free( rtmp_body );
...@@ -1981,7 +2478,7 @@ flv_get_metadata( access_t *p_access ) ...@@ -1981,7 +2478,7 @@ flv_get_metadata( access_t *p_access )
p_sys->p_thread->metadata_samplesize, p_sys->p_thread->metadata_samplerate, p_sys->p_thread->metadata_samplesize, p_sys->p_thread->metadata_samplerate,
p_sys->p_thread->metadata_audiocodecid, p_sys->p_thread->metadata_videocodecid ); p_sys->p_thread->metadata_audiocodecid, p_sys->p_thread->metadata_videocodecid );
flv_rebuild( p_sys->p_thread, flv_metadata_packet ); flv_rebuild( p_sys->p_thread, flv_metadata_packet );
p_buffer = flv_new_packet( p_sys->p_thread, flv_metadata_packet ); p_buffer = rtmp_new_block( p_sys->p_thread, flv_metadata_packet->body->body, flv_metadata_packet->body->length_buffer );
free( flv_metadata_packet->body->body ); free( flv_metadata_packet->body->body );
free( flv_metadata_packet->body ); free( flv_metadata_packet->body );
......
...@@ -26,7 +26,7 @@ ...@@ -26,7 +26,7 @@
typedef struct rtmp_packet_t rtmp_packet_t; typedef struct rtmp_packet_t rtmp_packet_t;
typedef struct rtmp_body_t rtmp_body_t; typedef struct rtmp_body_t rtmp_body_t;
typedef struct rtmp_control_thread_t rtmp_control_thread_t; typedef struct rtmp_control_thread_t rtmp_control_thread_t;
typedef void (*rtmp_handler_t)( rtmp_control_thread_t *rtmp_control_thread, rtmp_packet_t *rtmp_packet ); typedef void (*rtmp_handler_t)( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet );
struct rtmp_packet_t struct rtmp_packet_t
{ {
...@@ -54,51 +54,61 @@ struct rtmp_control_thread_t ...@@ -54,51 +54,61 @@ struct rtmp_control_thread_t
int fd; int fd;
block_fifo_t *p_fifo_media; vlc_url_t url;
char *psz_application;
char *psz_media;
block_fifo_t *p_fifo_input;
block_fifo_t *p_empty_blocks; block_fifo_t *p_empty_blocks;
vlc_mutex_t lock; vlc_mutex_t lock;
vlc_cond_t wait; vlc_cond_t wait;
int result_connect; int result_connect;
int result_publish;
int result_play; int result_play;
int result_publish; int result_stop;
double stream_client; double stream_client_id;
double stream_server; double stream_server_id;
char *psz_publish; char *psz_publish;
/* Rebuild FLV variables */ /* Rebuild FLV variables (access) */
int has_audio; int has_audio;
int has_video; int has_video;
int metadata_received; int metadata_received;
uint8_t metadata_stereo; uint8_t metadata_stereo;
uint8_t metadata_samplesize; uint8_t metadata_samplesize;
uint8_t metadata_samplerate; uint32_t metadata_samplerate;
uint8_t metadata_audiocodecid; uint8_t metadata_audiocodecid;
uint8_t metadata_videocodecid; uint8_t metadata_videocodecid;
uint8_t metadata_frametype; uint8_t metadata_frametype;
int first_media_packet; int first_media_packet;
uint32_t flv_tag_previous_tag_size; uint32_t flv_tag_previous_tag_size;
/* Vars for rebuilding FLV (access_output) */
rtmp_body_t *flv_body;
uint8_t flv_content_type;
uint32_t flv_length_body;
uint32_t flv_timestamp;
/* vars for channel state */ /* vars for channel state */
uint32_t chunk_size_recv;
uint32_t chunk_size_send;
rtmp_packet_t rtmp_headers_recv[64]; /* RTMP_HEADER_STREAM_MAX */ rtmp_packet_t rtmp_headers_recv[64]; /* RTMP_HEADER_STREAM_MAX */
rtmp_packet_t rtmp_headers_send[64]; rtmp_packet_t rtmp_headers_send[64];
rtmp_handler_t rtmp_handler[21]; /* index by RTMP_CONTENT_TYPE */ rtmp_handler_t rtmp_handler[21]; /* index by RTMP_CONTENT_TYPE */
/* Pointer to base module object (later needs to casted) */
void *p_base_object;
}; };
struct access_sys_t struct access_sys_t
{ {
int active; int active;
int fd;
vlc_url_t url;
char *psz_application;
char *psz_media;
/* vars for reading from fifo */ /* vars for reading from fifo */
block_t *flv_packet; block_t *flv_packet;
int read_packet; int read_packet;
...@@ -110,17 +120,27 @@ struct access_sys_t ...@@ -110,17 +120,27 @@ struct access_sys_t
/***************************************************************************** /*****************************************************************************
* RTMP header: * RTMP header:
******************************************************************************/ ******************************************************************************/
int rtmp_handshake_passive( vlc_object_t *p_this ); int rtmp_handshake_active( vlc_object_t *p_this, int fd );
int rtmp_handshake_active( vlc_object_t *p_this ); int rtmp_handshake_passive( vlc_object_t *p_this, int fd );
int rtmp_connect_active( vlc_object_t *p_this ); int rtmp_connect_active( rtmp_control_thread_t *p_thread );
int rtmp_connect_passive( rtmp_control_thread_t *p_thread );
//int rtmp_seek( access_t *p_access, int64_t i_pos ); TODO //int rtmp_seek( access_t *p_access, int64_t i_pos ); TODO
int rtmp_send_bytes_read( access_t *p_access, uint32_t reply ); //
int rtmp_send_publish_start( access_t *p_access ); rtmp_packet_t *rtmp_build_bytes_read( rtmp_control_thread_t *p_thread, uint32_t reply );
rtmp_packet_t *rtmp_build_publish_start( rtmp_control_thread_t *p_thread );
rtmp_packet_t *rtmp_build_flv_over_rtmp( rtmp_control_thread_t *p_thread, block_t *p_buffer );
rtmp_packet_t *rtmp_read_net_packet( rtmp_control_thread_t *p_thread ); rtmp_packet_t *rtmp_read_net_packet( rtmp_control_thread_t *p_thread );
uint8_t *rtmp_encode_packet( rtmp_control_thread_t *p_thread, rtmp_packet_t *rtmp_packet );
void rtmp_init_handler( rtmp_handler_t *rtmp_handler ); void rtmp_init_handler( rtmp_handler_t *rtmp_handler );
/***************************************************************************** /*****************************************************************************
* FLV header: * FLV header:
******************************************************************************/ ******************************************************************************/
block_t *flv_get_metadata( access_t *p_access ); block_t *flv_get_metadata( access_t *p_access );
block_t *flv_insert_header( access_t *p_access, block_t *first_packet ); block_t *flv_insert_header( access_t *p_access, block_t *first_packet );
/*****************************************************************************
* RTMP body header:
******************************************************************************/
rtmp_body_t *rtmp_body_new( int length_buffer );
void rtmp_body_reset( rtmp_body_t * );
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment