1 /*****************************************************************************
3 *****************************************************************************
4 * Copyright (C) URJC - LADyR - Luis Lopez Fernandez
6 * Author: Miguel Angel Cabrera Moya
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
21 *****************************************************************************/
23 /*****************************************************************************
25 *****************************************************************************/
30 #include <vlc_common.h>
31 #include <vlc_plugin.h>
34 #include <vlc_network.h> /* DOWN: #include <network.h> */
36 #include <vlc_block.h>
38 #include "../access/rtmp/rtmp_amf_flv.h"
40 /*****************************************************************************
42 *****************************************************************************/
44 #define RTMP_CONNECT_TEXT N_( "Active TCP connection" )
45 #define RTMP_CONNECT_LONGTEXT N_( \
46 "If enabled, VLC will connect to a remote destination instead of " \
47 "waiting for an incoming connection." )
49 static int Open ( vlc_object_t
* );
50 static void Close( vlc_object_t
* );
52 #define SOUT_CFG_PREFIX "sout-rtmp-"
55 set_description( N_("RTMP stream output") )
56 set_shortname( N_("RTMP" ) )
57 set_capability( "sout access", 50 )
58 set_category( CAT_SOUT
)
59 set_subcategory( SUBCAT_SOUT_STREAM
)
60 add_shortcut( "rtmp" )
61 set_callbacks( Open
, Close
)
62 add_bool( "rtmp-connect", false, NULL
, RTMP_CONNECT_TEXT
,
63 RTMP_CONNECT_LONGTEXT
, false );
66 /*****************************************************************************
68 *****************************************************************************/
69 static ssize_t
Write( sout_access_out_t
*, block_t
* );
70 static int Seek ( sout_access_out_t
*, off_t
);
71 static void* ThreadControl( vlc_object_t
* );
73 struct sout_access_out_sys_t
77 /* thread for filtering and handling control messages */
78 rtmp_control_thread_t
*p_thread
;
81 /*****************************************************************************
82 * Open: open the rtmp connection
83 *****************************************************************************/
84 static int Open( vlc_object_t
*p_this
)
86 sout_access_out_t
*p_access
= (sout_access_out_t
*) p_this
;
87 sout_access_out_sys_t
*p_sys
;
89 int length_path
, length_media_name
;
92 if( !( p_sys
= calloc ( 1, sizeof( sout_access_out_sys_t
) ) ) )
94 p_access
->p_sys
= p_sys
;
97 vlc_object_create( p_access
, sizeof( rtmp_control_thread_t
) );
98 if( !p_sys
->p_thread
)
100 vlc_object_attach( p_sys
->p_thread
, p_access
);
102 /* Parse URI - remove spaces */
103 p
= psz
= strdup( p_access
->psz_path
);
104 while( ( p
= strchr( p
, ' ' ) ) != NULL
)
106 vlc_UrlParse( &p_sys
->p_thread
->url
, psz
, 0 );
109 if( p_sys
->p_thread
->url
.psz_host
== NULL
110 || *p_sys
->p_thread
->url
.psz_host
== '\0' )
112 msg_Warn( p_access
, "invalid host" );
116 if( p_sys
->p_thread
->url
.i_port
<= 0 )
117 p_sys
->p_thread
->url
.i_port
= 1935;
119 if ( p_sys
->p_thread
->url
.psz_path
== NULL
)
121 msg_Warn( p_access
, "invalid path" );
125 length_path
= strlen( p_sys
->p_thread
->url
.psz_path
);
126 char* psz_tmp
= strrchr( p_sys
->p_thread
->url
.psz_path
, '/' );
129 length_media_name
= strlen( psz_tmp
) - 1;
131 p_sys
->p_thread
->psz_application
= strndup( p_sys
->p_thread
->url
.psz_path
+ 1, length_path
- length_media_name
- 2 );
132 p_sys
->p_thread
->psz_media
= strdup( p_sys
->p_thread
->url
.psz_path
+ ( length_path
- length_media_name
) );
134 msg_Dbg( p_access
, "rtmp: host='%s' port=%d path='%s'",
135 p_sys
->p_thread
->url
.psz_host
, p_sys
->p_thread
->url
.i_port
, p_sys
->p_thread
->url
.psz_path
);
137 if( p_sys
->p_thread
->url
.psz_username
&& *p_sys
->p_thread
->url
.psz_username
)
139 msg_Dbg( p_access
, " user='%s', pwd='%s'",
140 p_sys
->p_thread
->url
.psz_username
, p_sys
->p_thread
->url
.psz_password
);
143 /* Initialize thread variables */
144 p_sys
->p_thread
->b_die
= 0;
145 p_sys
->p_thread
->b_error
= 0;
146 p_sys
->p_thread
->p_fifo_input
= block_FifoNew();
147 p_sys
->p_thread
->p_empty_blocks
= block_FifoNew();
148 p_sys
->p_thread
->has_audio
= 0;
149 p_sys
->p_thread
->has_video
= 0;
150 p_sys
->p_thread
->metadata_received
= 0;
151 p_sys
->p_thread
->first_media_packet
= 1;
152 p_sys
->p_thread
->flv_tag_previous_tag_size
= 0x00000000; /* FLV_TAG_FIRST_PREVIOUS_TAG_SIZE */
154 p_sys
->p_thread
->flv_body
= rtmp_body_new( -1 );
155 p_sys
->p_thread
->flv_length_body
= 0;
157 p_sys
->p_thread
->chunk_size_recv
= 128; /* RTMP_DEFAULT_CHUNK_SIZE */
158 p_sys
->p_thread
->chunk_size_send
= 128; /* RTMP_DEFAULT_CHUNK_SIZE */
159 for(i
= 0; i
< 64; i
++)
161 memset( &p_sys
->p_thread
->rtmp_headers_recv
[i
], 0, sizeof( rtmp_packet_t
) );
162 p_sys
->p_thread
->rtmp_headers_send
[i
].length_header
= -1;
163 p_sys
->p_thread
->rtmp_headers_send
[i
].stream_index
= -1;
164 p_sys
->p_thread
->rtmp_headers_send
[i
].timestamp
= -1;
165 p_sys
->p_thread
->rtmp_headers_send
[i
].timestamp_relative
= -1;
166 p_sys
->p_thread
->rtmp_headers_send
[i
].length_encoded
= -1;
167 p_sys
->p_thread
->rtmp_headers_send
[i
].length_body
= -1;
168 p_sys
->p_thread
->rtmp_headers_send
[i
].content_type
= -1;
169 p_sys
->p_thread
->rtmp_headers_send
[i
].src_dst
= -1;
170 p_sys
->p_thread
->rtmp_headers_send
[i
].body
= NULL
;
173 vlc_cond_init( &p_sys
->p_thread
->wait
);
174 vlc_mutex_init( &p_sys
->p_thread
->lock
);
176 p_sys
->p_thread
->result_connect
= 1;
177 /* p_sys->p_thread->result_publish = only used on access */
178 p_sys
->p_thread
->result_play
= 1;
179 p_sys
->p_thread
->result_stop
= 0;
180 p_sys
->p_thread
->fd
= -1;
182 /* Open connection */
183 if( var_CreateGetBool( p_access
, "rtmp-connect" ) > 0 )
186 p_sys
->p_thread
->fd
= net_ConnectTCP( p_access
,
187 p_sys
->p_thread
->url
.psz_host
,
188 p_sys
->p_thread
->url
.i_port
);
190 msg_Err( p_access
, "to be implemented" );
198 p_fd_listen
= net_ListenTCP( p_access
, p_sys
->p_thread
->url
.psz_host
,
199 p_sys
->p_thread
->url
.i_port
);
200 if( p_fd_listen
== NULL
)
202 msg_Warn( p_access
, "cannot listen to %s port %i",
203 p_sys
->p_thread
->url
.psz_host
,
204 p_sys
->p_thread
->url
.i_port
);
209 p_sys
->p_thread
->fd
= net_Accept( p_access
, p_fd_listen
, -1 );
210 while( p_sys
->p_thread
->fd
== -1 );
211 net_ListenClose( p_fd_listen
);
213 if( rtmp_handshake_passive( p_this
, p_sys
->p_thread
->fd
) < 0 )
215 msg_Err( p_access
, "handshake passive failed");
220 if( vlc_thread_create( p_sys
->p_thread
, "rtmp control thread", ThreadControl
,
221 VLC_THREAD_PRIORITY_INPUT
, false ) )
223 msg_Err( p_access
, "cannot spawn rtmp control thread" );
229 if( rtmp_connect_passive( p_sys
->p_thread
) < 0 )
231 msg_Err( p_access
, "connect passive failed");
236 p_access
->pf_write
= Write
;
237 p_access
->pf_seek
= Seek
;
242 vlc_cond_destroy( &p_sys
->p_thread
->wait
);
243 vlc_mutex_destroy( &p_sys
->p_thread
->lock
);
245 free( p_sys
->p_thread
->psz_application
);
246 free( p_sys
->p_thread
->psz_media
);
248 if( p_sys
->p_thread
->fd
!= -1 )
249 net_Close( p_sys
->p_thread
->fd
);
251 vlc_object_detach( p_sys
->p_thread
);
252 vlc_object_release( p_sys
->p_thread
);
254 vlc_UrlClean( &p_sys
->p_thread
->url
);
260 /*****************************************************************************
261 * Close: close the target
262 *****************************************************************************/
263 static void Close( vlc_object_t
* p_this
)
265 sout_access_out_t
*p_access
= (sout_access_out_t
*) p_this
;
266 sout_access_out_sys_t
*p_sys
= p_access
->p_sys
;
269 // p_sys->p_thread->b_die = true;
270 vlc_object_kill( p_sys
->p_thread
);
271 block_FifoWake( p_sys
->p_thread
->p_fifo_input
);
273 vlc_thread_join( p_sys
->p_thread
);
275 vlc_cond_destroy( &p_sys
->p_thread
->wait
);
276 vlc_mutex_destroy( &p_sys
->p_thread
->lock
);
278 block_FifoRelease( p_sys
->p_thread
->p_fifo_input
);
279 block_FifoRelease( p_sys
->p_thread
->p_empty_blocks
);
281 for( i
= 0; i
< 64; i
++ ) /* RTMP_HEADER_STREAM_INDEX_MASK */
283 if( p_sys
->p_thread
->rtmp_headers_recv
[i
].body
!= NULL
)
285 free( p_sys
->p_thread
->rtmp_headers_recv
[i
].body
->body
);
286 free( p_sys
->p_thread
->rtmp_headers_recv
[i
].body
);
290 net_Close( p_sys
->p_thread
->fd
);
292 vlc_object_detach( p_sys
->p_thread
);
293 vlc_object_release( p_sys
->p_thread
);
295 vlc_UrlClean( &p_sys
->p_thread
->url
);
296 free( p_sys
->p_thread
->psz_application
);
297 free( p_sys
->p_thread
->psz_media
);
301 /*****************************************************************************
302 * Write: standard write on a file descriptor.
303 *****************************************************************************/
304 static ssize_t
Write( sout_access_out_t
*p_access
, block_t
*p_buffer
)
306 rtmp_packet_t
*rtmp_packet
;
311 if( p_access
->p_sys
->p_thread
->first_media_packet
)
313 /* 13 == FLV_HEADER_SIZE + PreviousTagSize*/
314 memmove( p_buffer
->p_buffer
, p_buffer
->p_buffer
+ 13, p_buffer
->i_buffer
- 13 );
315 p_buffer
= block_Realloc( p_buffer
, 0, p_buffer
->i_buffer
- 13 );
317 p_access
->p_sys
->p_thread
->first_media_packet
= 0;
322 block_t
*p_next
= p_buffer
->p_next
;
323 //////////////////////////////
324 /*msg_Warn(p_access, "XXXXXXXXXXXXXXXXX");
326 for(i = 0; i < p_buffer->i_buffer; i += 16)
328 msg_Warn(p_access,"%.2x%.2x %.2x%.2x %.2x%.2x %.2x%.2x %.2x%.2x %.2x%.2x %.2x%.2x %.2x%.2x",
329 p_buffer->p_buffer[i], p_buffer->p_buffer[i+1], p_buffer->p_buffer[i+2], p_buffer->p_buffer[i+3], p_buffer->p_buffer[i+4], p_buffer->p_buffer[i+5], p_buffer->p_buffer[i+6], p_buffer->p_buffer[i+7],
330 p_buffer->p_buffer[i+8], p_buffer->p_buffer[i+9], p_buffer->p_buffer[i+10], p_buffer->p_buffer[i+11], p_buffer->p_buffer[i+12], p_buffer->p_buffer[i+13], p_buffer->p_buffer[i+14], p_buffer->p_buffer[i+15]);
332 ////////////////////////
333 msg_Warn(p_access
, "rtmp.c:360 i_dts %"PRIu64
" i_pts %"PRIu64
,
334 p_buffer
->i_dts
, p_buffer
->i_pts
);
335 rtmp_packet
= rtmp_build_flv_over_rtmp( p_access
->p_sys
->p_thread
, p_buffer
);
339 tmp_buffer
= rtmp_encode_packet( p_access
->p_sys
->p_thread
, rtmp_packet
);
341 i_ret
= net_Write( p_access
->p_sys
->p_thread
, p_access
->p_sys
->p_thread
->fd
, NULL
, tmp_buffer
, rtmp_packet
->length_encoded
);
342 if( i_ret
!= rtmp_packet
->length_encoded
)
344 free( rtmp_packet
->body
->body
);
345 free( rtmp_packet
->body
);
348 msg_Err( p_access
->p_sys
->p_thread
, "failed send flv packet" );
351 free( rtmp_packet
->body
->body
);
352 free( rtmp_packet
->body
);
357 i_write
+= p_buffer
->i_buffer
;
365 /********************a*********************************************************
366 * Seek: seek to a specific location in a file
367 *****************************************************************************/
368 static int Seek( sout_access_out_t
*p_access
, off_t i_pos
)
371 msg_Err( p_access
, "RTMP sout access cannot seek" );
375 /*****************************************************************************
376 * ThreadControl: manage control messages and pipe media to Read
377 *****************************************************************************/
378 static void* ThreadControl( vlc_object_t
*p_this
)
380 rtmp_control_thread_t
*p_thread
= (rtmp_control_thread_t
*) p_this
;
381 rtmp_packet_t
*rtmp_packet
;
382 int canc
= vlc_savecancel ();
384 rtmp_init_handler( p_thread
->rtmp_handler
);
386 while( vlc_object_alive (p_thread
) )
388 rtmp_packet
= rtmp_read_net_packet( p_thread
);
389 if( rtmp_packet
!= NULL
)
391 if( rtmp_packet
->content_type
< 0x01 /* RTMP_CONTENT_TYPE_CHUNK_SIZE */
392 || rtmp_packet
->content_type
> 0x14 ) /* RTMP_CONTENT_TYPE_INVOKE */
394 free( rtmp_packet
->body
->body
);
395 free( rtmp_packet
->body
);
398 msg_Warn( p_thread
, "unknown content type received" );
401 p_thread
->rtmp_handler
[rtmp_packet
->content_type
]( p_thread
, rtmp_packet
);
405 /* Sometimes server close connection too soon */
406 if( p_thread
->result_connect
)
408 vlc_mutex_lock( &p_thread
->lock
);
409 vlc_cond_signal( &p_thread
->wait
);
410 vlc_mutex_unlock( &p_thread
->lock
);
416 vlc_restorecancel (canc
);