1 /*****************************************************************************
2 * access.c: RTMP input.
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>
32 #include <vlc_access.h>
34 #include <vlc_network.h> /* DOWN: #include <network.h> */
36 #include <vlc_block.h>
38 #include "rtmp_amf_flv.h"
40 /*****************************************************************************
42 *****************************************************************************/
43 #define CACHING_TEXT N_("Caching value in ms")
44 #define CACHING_LONGTEXT N_( \
45 "Caching value for RTMP streams. This " \
46 "value should be set in milliseconds." )
48 static int Open ( vlc_object_t
* );
49 static void Close( vlc_object_t
* );
52 set_description( N_("RTMP input") );
53 set_shortname( N_("RTMP") );
54 set_category( CAT_INPUT
);
55 set_subcategory( SUBCAT_INPUT_ACCESS
);
57 add_integer( "rtmp-caching", DEFAULT_PTS_DELAY
/ 1000, NULL
, CACHING_TEXT
,
58 CACHING_LONGTEXT
, true );
60 set_capability( "access", 10 );
61 set_callbacks( Open
, Close
);
62 add_shortcut( "rtmp" );
66 /*****************************************************************************
68 *****************************************************************************/
69 static int Read( access_t
*, uint8_t *, size_t ); /*DOWN: last parameter int */
70 static int Seek( access_t
*, int64_t );
71 static int Control( access_t
*, int, va_list );
73 static void ThreadControl( vlc_object_t
* );
75 /*****************************************************************************
76 * Open: open the rtmp connection
77 *****************************************************************************/
78 static int Open( vlc_object_t
*p_this
)
80 access_t
*p_access
= (access_t
*) p_this
;
83 int length_path
, length_media_name
;
86 STANDARD_READ_ACCESS_INIT
89 vlc_object_create( p_access
, sizeof( rtmp_control_thread_t
) );
90 if( !p_sys
->p_thread
)
92 vlc_object_attach( p_sys
->p_thread
, p_access
);
94 /* Parse URI - remove spaces */
95 p
= psz
= strdup( p_access
->psz_path
);
96 while( (p
= strchr( p
, ' ' )) != NULL
)
98 vlc_UrlParse( &p_sys
->p_thread
->url
, psz
, 0 );
101 if( p_sys
->p_thread
->url
.psz_host
== NULL
102 || *p_sys
->p_thread
->url
.psz_host
== '\0' )
104 msg_Warn( p_access
, "invalid host" );
108 if( p_sys
->p_thread
->url
.i_port
<= 0 )
109 p_sys
->p_thread
->url
.i_port
= 1935;
111 if( p_sys
->p_thread
->url
.psz_path
== NULL
)
113 msg_Warn( p_access
, "invalid path" );
117 length_path
= strlen( p_sys
->p_thread
->url
.psz_path
);
118 length_media_name
= strlen( strrchr( p_sys
->p_thread
->url
.psz_path
, '/' ) ) - 1;
120 p_sys
->p_thread
->psz_application
= strndup( p_sys
->p_thread
->url
.psz_path
+ 1, length_path
- length_media_name
- 2 );
121 p_sys
->p_thread
->psz_media
= strdup( p_sys
->p_thread
->url
.psz_path
+ ( length_path
- length_media_name
) );
123 msg_Dbg( p_access
, "rtmp: host='%s' port=%d path='%s'",
124 p_sys
->p_thread
->url
.psz_host
, p_sys
->p_thread
->url
.i_port
, p_sys
->p_thread
->url
.psz_path
);
126 if( p_sys
->p_thread
->url
.psz_username
&& *p_sys
->p_thread
->url
.psz_username
)
128 msg_Dbg( p_access
, " user='%s', pwd='%s'",
129 p_sys
->p_thread
->url
.psz_username
, p_sys
->p_thread
->url
.psz_password
);
132 /* Initialize thread variables */
133 p_sys
->p_thread
->b_die
= 0;
134 p_sys
->p_thread
->b_error
= 0;
135 p_sys
->p_thread
->p_fifo_input
= block_FifoNew();
136 p_sys
->p_thread
->p_empty_blocks
= block_FifoNew();
137 p_sys
->p_thread
->has_audio
= 0;
138 p_sys
->p_thread
->has_video
= 0;
139 p_sys
->p_thread
->metadata_received
= 0;
140 p_sys
->p_thread
->first_media_packet
= 1;
141 p_sys
->p_thread
->flv_tag_previous_tag_size
= 0x00000000; /* FLV_TAG_FIRST_PREVIOUS_TAG_SIZE */
142 p_sys
->p_thread
->chunk_size_recv
= 128; /* RTMP_DEFAULT_CHUNK_SIZE */
143 p_sys
->p_thread
->chunk_size_send
= 128; /* RTMP_DEFAULT_CHUNK_SIZE */
144 for(i
= 0; i
< 64; i
++)
146 memset( &p_sys
->p_thread
->rtmp_headers_recv
[i
], 0, sizeof( rtmp_packet_t
) );
147 p_sys
->p_thread
->rtmp_headers_send
[i
].length_header
= -1;
148 p_sys
->p_thread
->rtmp_headers_send
[i
].stream_index
= -1;
149 p_sys
->p_thread
->rtmp_headers_send
[i
].timestamp
= -1;
150 p_sys
->p_thread
->rtmp_headers_send
[i
].timestamp_relative
= -1;
151 p_sys
->p_thread
->rtmp_headers_send
[i
].length_encoded
= -1;
152 p_sys
->p_thread
->rtmp_headers_send
[i
].length_body
= -1;
153 p_sys
->p_thread
->rtmp_headers_send
[i
].content_type
= -1;
154 p_sys
->p_thread
->rtmp_headers_send
[i
].src_dst
= -1;
155 p_sys
->p_thread
->rtmp_headers_send
[i
].body
= NULL
;
158 p_sys
->p_thread
->p_base_object
= p_this
;
160 vlc_cond_init( p_sys
->p_thread
, &p_sys
->p_thread
->wait
);
162 vlc_mutex_init( &p_sys
->p_thread
->lock
);
164 p_sys
->p_thread
->result_connect
= 1;
165 p_sys
->p_thread
->result_play
= 1;
166 p_sys
->p_thread
->result_stop
= 0;
168 /* Open connection */
169 p_sys
->p_thread
->fd
= net_ConnectTCP( p_access
, p_sys
->p_thread
->url
.psz_host
, p_sys
->p_thread
->url
.i_port
);
170 if( p_sys
->p_thread
->fd
== -1 )
174 msg_Warn( p_access
, "cannot connect to %s:%d", p_sys
->p_thread
->url
.psz_host
, p_sys
->p_thread
->url
.i_port
);
175 msg_Dbg( p_access
, "switching to passive mode" );
179 p_fd_listen
= net_ListenTCP( p_access
, p_sys
->p_thread
->url
.psz_host
, p_sys
->p_thread
->url
.i_port
);
180 if( p_fd_listen
== NULL
)
182 msg_Err( p_access
, "cannot listen to %s port %i", p_sys
->p_thread
->url
.psz_host
, p_sys
->p_thread
->url
.i_port
);
186 p_sys
->p_thread
->fd
= net_Accept( p_access
, p_fd_listen
, -1 );
188 net_ListenClose( p_fd_listen
);
190 if( rtmp_handshake_passive( p_this
, p_sys
->p_thread
->fd
) < 0 )
192 msg_Err( p_access
, "handshake passive failed");
196 p_sys
->p_thread
->result_publish
= 1;
202 if( rtmp_handshake_active( p_this
, p_sys
->p_thread
->fd
) < 0 )
204 msg_Err( p_access
, "handshake active failed");
208 p_sys
->p_thread
->result_publish
= 0;
211 if( vlc_thread_create( p_sys
->p_thread
, "rtmp control thread", ThreadControl
,
212 VLC_THREAD_PRIORITY_INPUT
, false ) )
214 msg_Err( p_access
, "cannot spawn rtmp control thread" );
220 if( rtmp_connect_active( p_sys
->p_thread
) < 0 )
222 msg_Err( p_access
, "connect active failed");
227 /* Set vars for reading from fifo */
228 p_access
->p_sys
->flv_packet
= NULL
;
229 p_access
->p_sys
->read_packet
= 1;
231 /* Update default_pts to a suitable value for rtmp access */
232 var_Create( p_access
, "rtmp-caching", VLC_VAR_INTEGER
| VLC_VAR_DOINHERIT
);
237 vlc_cond_destroy( &p_sys
->p_thread
->wait
);
238 vlc_mutex_destroy( &p_sys
->p_thread
->lock
);
240 free( p_sys
->p_thread
->psz_application
);
241 free( p_sys
->p_thread
->psz_media
);
243 net_Close( p_sys
->p_thread
->fd
);
245 vlc_UrlClean( &p_sys
->p_thread
->url
);
247 vlc_object_detach( p_sys
->p_thread
);
248 vlc_object_release( p_sys
->p_thread
);
255 /*****************************************************************************
256 * Close: close the rtmp connection
257 *****************************************************************************/
258 static void Close( vlc_object_t
* p_this
)
260 access_t
*p_access
= (access_t
*) p_this
;
261 access_sys_t
*p_sys
= p_access
->p_sys
;
264 /* p_sys->p_thread->b_die = true;*/
265 vlc_object_kill( p_sys
->p_thread
);
266 block_FifoWake( p_sys
->p_thread
->p_fifo_input
);
267 block_FifoWake( p_sys
->p_thread
->p_empty_blocks
);
269 vlc_thread_join( p_sys
->p_thread
);
271 vlc_cond_destroy( &p_sys
->p_thread
->wait
);
272 vlc_mutex_destroy( &p_sys
->p_thread
->lock
);
274 block_FifoRelease( p_sys
->p_thread
->p_fifo_input
);
275 block_FifoRelease( p_sys
->p_thread
->p_empty_blocks
);
277 for( i
= 0; i
< 64; i
++ ) /* RTMP_HEADER_STREAM_INDEX_MASK */
279 if( p_sys
->p_thread
->rtmp_headers_recv
[i
].body
!= NULL
)
281 free( p_sys
->p_thread
->rtmp_headers_recv
[i
].body
->body
);
282 free( p_sys
->p_thread
->rtmp_headers_recv
[i
].body
);
286 net_Close( p_sys
->p_thread
->fd
);
288 var_Destroy( p_access
, "rtmp-caching" );
290 vlc_UrlClean( &p_sys
->p_thread
->url
);
291 free( p_sys
->p_thread
->psz_application
);
292 free( p_sys
->p_thread
->psz_media
);
294 vlc_object_detach( p_sys
->p_thread
);
295 vlc_object_release( p_sys
->p_thread
);
299 /*****************************************************************************
300 * Read: standard read on a file descriptor.
301 *****************************************************************************/
302 static int Read( access_t
*p_access
, uint8_t *p_buffer
, size_t i_len
)
304 access_sys_t
*p_sys
= p_access
->p_sys
;
305 rtmp_packet_t
*rtmp_packet
;
312 while( i_len_tmp
< i_len
)
314 if( p_sys
->p_thread
->result_stop
|| p_access
->info
.b_eof
|| p_access
->b_die
)
316 p_access
->info
.b_eof
= true;
320 if( p_sys
->read_packet
)
322 if( !p_sys
->p_thread
->metadata_received
)
324 /* Wait until enough data is received for extracting metadata */
325 if( block_FifoCount( p_sys
->p_thread
->p_fifo_input
) < 10 )
331 p_sys
->flv_packet
= flv_get_metadata( p_access
);
333 p_sys
->p_thread
->metadata_received
= 1;
337 p_sys
->flv_packet
= block_FifoGet( p_sys
->p_thread
->p_fifo_input
);
338 if( p_sys
->flv_packet
== NULL
)
339 continue; /* Forced wake-up */
342 if( p_sys
->p_thread
->first_media_packet
)
344 p_sys
->flv_packet
= flv_insert_header( p_access
, p_sys
->flv_packet
);
346 p_sys
->p_thread
->first_media_packet
= 0;
349 if( i_len
- i_len_tmp
>= p_sys
->flv_packet
->i_buffer
)
351 p_sys
->read_packet
= 1;
353 memcpy( p_buffer
+ i_len_tmp
, p_sys
->flv_packet
->p_buffer
, p_sys
->flv_packet
->i_buffer
);
354 block_FifoPut( p_sys
->p_thread
->p_empty_blocks
, p_sys
->flv_packet
);
356 i_len_tmp
+= p_sys
->flv_packet
->i_buffer
;
360 p_sys
->read_packet
= 0;
362 memcpy( p_buffer
+ i_len_tmp
, p_sys
->flv_packet
->p_buffer
, i_len
- i_len_tmp
);
363 p_sys
->flv_packet
->i_buffer
-= i_len
- i_len_tmp
;
364 memmove( p_sys
->flv_packet
->p_buffer
, p_sys
->flv_packet
->p_buffer
+ i_len
- i_len_tmp
, p_sys
->flv_packet
->i_buffer
);
366 i_len_tmp
+= i_len
- i_len_tmp
;
371 if( p_sys
->p_thread
->result_publish
)
373 /* Send publish onStatus event only once */
374 p_sys
->p_thread
->result_publish
= 0;
376 rtmp_packet
= rtmp_build_publish_start( p_sys
->p_thread
);
378 tmp_buffer
= rtmp_encode_packet( p_sys
->p_thread
, rtmp_packet
);
380 i_ret
= net_Write( p_sys
->p_thread
, p_sys
->p_thread
->fd
, NULL
, tmp_buffer
, rtmp_packet
->length_encoded
);
381 if( i_ret
!= rtmp_packet
->length_encoded
)
383 free( rtmp_packet
->body
->body
);
384 free( rtmp_packet
->body
);
387 msg_Err( p_access
, "failed send publish start" );
390 free( rtmp_packet
->body
->body
);
391 free( rtmp_packet
->body
);
396 p_access
->info
.i_pos
+= i_len_tmp
;
398 rtmp_packet
= rtmp_build_bytes_read( p_sys
->p_thread
, p_access
->info
.i_pos
);
400 tmp_buffer
= rtmp_encode_packet( p_sys
->p_thread
, rtmp_packet
);
402 i_ret
= net_Write( p_sys
->p_thread
, p_sys
->p_thread
->fd
, NULL
, tmp_buffer
, rtmp_packet
->length_encoded
);
403 if( i_ret
!= rtmp_packet
->length_encoded
)
405 free( rtmp_packet
->body
->body
);
406 free( rtmp_packet
->body
);
409 msg_Err( p_access
, "failed send bytes read" );
412 free( rtmp_packet
->body
->body
);
413 free( rtmp_packet
->body
);
421 /*****************************************************************************
422 * Seek: seek to a specific location in a file
423 *****************************************************************************/
424 static int Seek( access_t
*p_access
, int64_t i_pos
)
426 VLC_UNUSED( p_access
);
428 /*msg_Warn ( p_access, "Seek to %lld", i_pos);
429 switch( rtmp_seek( p_access, i_pos ) )
436 msg_Err( p_access, "You should not be here" );
443 /*****************************************************************************
445 *****************************************************************************/
446 static int Control( access_t
*p_access
, int i_query
, va_list args
)
455 case ACCESS_CAN_SEEK
:
456 case ACCESS_CAN_FASTSEEK
:
457 pb_bool
= (bool*) va_arg( args
, bool* );
458 *pb_bool
= false; /* TODO */
461 case ACCESS_CAN_PAUSE
:
462 pb_bool
= (bool*) va_arg( args
, bool* );
463 *pb_bool
= false; /* TODO */
466 case ACCESS_CAN_CONTROL_PACE
:
467 pb_bool
= (bool*) va_arg( args
, bool* );
473 pi_int
= (int*) va_arg( args
, int * );
477 case ACCESS_GET_PTS_DELAY
:
478 pi_64
= (int64_t*) va_arg( args
, int64_t * );
479 *pi_64
= var_GetInteger( p_access
, "rtmp-caching" ) * INT64_C(1000);
483 case ACCESS_SET_PAUSE_STATE
:
487 case ACCESS_GET_TITLE_INFO
:
488 case ACCESS_SET_TITLE
:
489 case ACCESS_SET_SEEKPOINT
:
490 case ACCESS_SET_PRIVATE_ID_STATE
:
491 case ACCESS_GET_META
:
492 case ACCESS_GET_CONTENT_TYPE
: /* DOWN: comment this line */
496 msg_Warn( p_access
, "unimplemented query in control" );
503 /*****************************************************************************
504 * ThreadControl: manage control messages and pipe media to Read
505 *****************************************************************************/
506 static void ThreadControl( vlc_object_t
*p_this
)
508 rtmp_control_thread_t
*p_thread
= (rtmp_control_thread_t
*) p_this
;
509 rtmp_packet_t
*rtmp_packet
;
511 rtmp_init_handler( p_thread
->rtmp_handler
);
513 while( !p_thread
->b_die
)
515 rtmp_packet
= rtmp_read_net_packet( p_thread
);
516 if( rtmp_packet
!= NULL
)
518 if( rtmp_packet
->content_type
< 0x01 /* RTMP_CONTENT_TYPE_CHUNK_SIZE */
519 || rtmp_packet
->content_type
> 0x14 ) /* RTMP_CONTENT_TYPE_INVOKE */
521 free( rtmp_packet
->body
->body
);
522 free( rtmp_packet
->body
);
525 msg_Warn( p_thread
, "unknown content type received" );
528 p_thread
->rtmp_handler
[rtmp_packet
->content_type
]( p_thread
, rtmp_packet
);
532 /* Sometimes server close connection too soon */
533 if( p_thread
->result_connect
)
535 vlc_mutex_lock( &p_thread
->lock
);
536 vlc_cond_signal( &p_thread
->wait
);
537 vlc_mutex_unlock( &p_thread
->lock
);
541 ((access_t
*) p_thread
->p_base_object
)->info
.b_eof
= true;
543 block_FifoWake( p_thread
->p_fifo_input
);