2 * This file is part of MPlayer.
4 * MPlayer is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * MPlayer is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License along
15 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
23 #include <sys/types.h>
26 #include <sys/ioctl.h>
41 #include <libavutil/common.h>
44 #include "osdep/shmem.h"
45 #include "osdep/timer.h"
48 #include "libmpdemux/demuxer.h"
49 #include "ffmpeg_files/intreadwrite.h"
57 static int (*stream_check_interrupt_cb
)(struct input_ctx
*ctx
, int time
);
58 static struct input_ctx
*stream_check_interrupt_ctx
;
60 extern const stream_info_t stream_info_vcd
;
61 extern const stream_info_t stream_info_cdda
;
62 extern const stream_info_t stream_info_netstream
;
63 extern const stream_info_t stream_info_pnm
;
64 extern const stream_info_t stream_info_asf
;
65 extern const stream_info_t stream_info_rtsp
;
66 extern const stream_info_t stream_info_rtp
;
67 extern const stream_info_t stream_info_udp
;
68 extern const stream_info_t stream_info_http1
;
69 extern const stream_info_t stream_info_http2
;
70 extern const stream_info_t stream_info_dvb
;
71 extern const stream_info_t stream_info_tv
;
72 extern const stream_info_t stream_info_radio
;
73 extern const stream_info_t stream_info_pvr
;
74 extern const stream_info_t stream_info_ftp
;
75 extern const stream_info_t stream_info_vstream
;
76 extern const stream_info_t stream_info_dvdnav
;
77 extern const stream_info_t stream_info_smb
;
78 extern const stream_info_t stream_info_sdp
;
79 extern const stream_info_t stream_info_rtsp_sip
;
81 extern const stream_info_t stream_info_cue
;
82 extern const stream_info_t stream_info_null
;
83 extern const stream_info_t stream_info_mf
;
84 extern const stream_info_t stream_info_ffmpeg
;
85 extern const stream_info_t stream_info_file
;
86 extern const stream_info_t stream_info_ifo
;
87 extern const stream_info_t stream_info_dvd
;
88 extern const stream_info_t stream_info_bluray
;
90 static const stream_info_t
* const auto_open_streams
[] = {
97 #ifdef CONFIG_NETWORKING
98 &stream_info_netstream
,
103 #ifdef CONFIG_LIVE555
105 &stream_info_rtsp_sip
,
126 #ifdef CONFIG_VSTREAM
127 &stream_info_vstream
,
129 #ifdef CONFIG_LIBSMBCLIENT
133 #ifdef CONFIG_DVDREAD
140 #ifdef CONFIG_LIBBLURAY
153 static stream_t
*open_stream_plugin(const stream_info_t
*sinfo
,
154 const char *filename
,
155 int mode
, struct MPOpts
*options
,
156 int *file_format
, int *ret
,
157 char **redirected_url
)
161 m_struct_t
* desc
= (m_struct_t
*)sinfo
->opts
;
165 arg
= m_struct_alloc(desc
);
166 if(sinfo
->opts_url
) {
168 { "stream url", arg
, CONF_TYPE_CUSTOM_URL
, 0, 0 ,0, (void *)sinfo
->opts
};
169 if(m_option_parse(&url_opt
,"stream url",filename
,arg
,M_CONFIG_FILE
) < 0) {
170 mp_msg(MSGT_OPEN
,MSGL_ERR
, "URL parsing failed on url %s\n",filename
);
171 m_struct_free(desc
,arg
);
176 s
= new_stream(-2,-2);
178 s
->url
=strdup(filename
);
180 *ret
= sinfo
->open(s
,mode
,arg
,file_format
);
181 if((*ret
) != STREAM_OK
) {
182 #ifdef CONFIG_NETWORKING
183 if (*ret
== STREAM_REDIRECTED
&& redirected_url
) {
184 if (s
->streaming_ctrl
&& s
->streaming_ctrl
->url
185 && s
->streaming_ctrl
->url
->url
)
186 *redirected_url
= strdup(s
->streaming_ctrl
->url
->url
);
188 *redirected_url
= NULL
;
190 streaming_ctrl_free(s
->streaming_ctrl
);
197 mp_msg(MSGT_OPEN
,MSGL_WARN
, "Warning streams need a type !!!!\n");
198 if(s
->flags
& MP_STREAM_SEEK
&& !s
->seek
)
199 s
->flags
&= ~MP_STREAM_SEEK
;
200 if(s
->seek
&& !(s
->flags
& MP_STREAM_SEEK
))
201 s
->flags
|= MP_STREAM_SEEK
;
205 mp_msg(MSGT_OPEN
,MSGL_V
, "STREAM: [%s] %s\n",sinfo
->name
,filename
);
206 mp_msg(MSGT_OPEN
,MSGL_V
, "STREAM: Description: %s\n",sinfo
->info
);
207 mp_msg(MSGT_OPEN
,MSGL_V
, "STREAM: Author: %s\n", sinfo
->author
);
208 mp_msg(MSGT_OPEN
,MSGL_V
, "STREAM: Comment: %s\n", sinfo
->comment
);
214 stream_t
*open_stream_full(const char *filename
, int mode
,
215 struct MPOpts
*options
, int *file_format
)
218 const stream_info_t
* sinfo
;
220 char *redirected_url
= NULL
;
222 for(i
= 0 ; auto_open_streams
[i
] ; i
++) {
223 sinfo
= auto_open_streams
[i
];
224 if(!sinfo
->protocols
) {
225 mp_msg(MSGT_OPEN
,MSGL_WARN
, "Stream type %s has protocols == NULL, it's a bug\n", sinfo
->name
);
228 for(j
= 0 ; sinfo
->protocols
[j
] ; j
++) {
229 l
= strlen(sinfo
->protocols
[j
]);
230 // l == 0 => Don't do protocol matching (ie network and filenames)
231 if((l
== 0 && !strstr(filename
, "://")) ||
232 ((strncasecmp(sinfo
->protocols
[j
],filename
,l
) == 0) &&
233 (strncmp("://",filename
+l
,3) == 0))) {
234 *file_format
= DEMUXER_TYPE_UNKNOWN
;
235 s
= open_stream_plugin(sinfo
,filename
,mode
,options
,file_format
,&r
,
238 if(r
== STREAM_REDIRECTED
&& redirected_url
) {
239 mp_msg(MSGT_OPEN
,MSGL_V
, "[%s] open %s redirected to %s\n",
240 sinfo
->info
, filename
, redirected_url
);
241 s
= open_stream_full(redirected_url
, mode
, options
, file_format
);
242 free(redirected_url
);
245 else if(r
!= STREAM_UNSUPPORTED
) {
246 mp_tmsg(MSGT_OPEN
,MSGL_ERR
, "Failed to open %s.\n",filename
);
254 mp_msg(MSGT_OPEN
,MSGL_ERR
, "No stream found to handle url %s\n",filename
);
258 stream_t
*open_output_stream(const char *filename
, struct MPOpts
*options
)
260 int file_format
; //unused
262 mp_msg(MSGT_OPEN
,MSGL_ERR
,"open_output_stream(), NULL filename, report this bug\n");
266 return open_stream_full(filename
,STREAM_WRITE
,options
,&file_format
);
269 //=================== STREAMER =========================
271 void stream_capture_do(stream_t
*s
)
273 if (fwrite(s
->buffer
, s
->buf_len
, 1, s
->capture_file
) < 1) {
274 mp_tmsg(MSGT_GLOBAL
, MSGL_ERR
, "Error writing capture file: %s\n",
276 fclose(s
->capture_file
);
277 s
->capture_file
= NULL
;
281 int stream_read_internal(stream_t
*s
, void *buf
, int len
)
284 // we will retry even if we already reached EOF previously.
286 case STREAMTYPE_STREAM
:
287 #ifdef CONFIG_NETWORKING
288 if( s
->streaming_ctrl
!=NULL
&& s
->streaming_ctrl
->streaming_read
) {
289 len
=s
->streaming_ctrl
->streaming_read(s
->fd
, buf
, len
, s
->streaming_ctrl
);
293 len
= s
->fill_buffer(s
, buf
, len
);
295 len
= read(s
->fd
, buf
, len
);
298 len
= demux_read_data((demux_stream_t
*)s
->priv
, buf
, len
);
303 len
= s
->fill_buffer
? s
->fill_buffer(s
, buf
, len
) : 0;
306 // dvdnav has some horrible hacks to "suspend" reads,
307 // we need to skip this code or seeks will hang.
308 if (!s
->eof
&& s
->type
!= STREAMTYPE_DVDNAV
) {
309 // just in case this is an error e.g. due to network
310 // timeout reset and retry
311 // Seeking is used as a hack to make network streams
312 // reopen the connection, ideally they would implement
313 // e.g. a STREAM_CTRL_RECONNECT to do this
317 stream_seek_internal(s
, pos
);
318 // make sure EOF is set to ensure no endless loops
320 return stream_read_internal(s
, buf
, orig_len
);
325 // When reading succeeded we are obviously not at eof.
326 // This e.g. avoids issues with eof getting stuck when lavf seeks in MPEG-TS
332 int stream_fill_buffer(stream_t
*s
){
333 int len
= stream_read_internal(s
, s
->buffer
, STREAM_BUFFER_SIZE
);
338 // printf("[%d]",len);fflush(stdout);
340 stream_capture_do(s
);
344 int stream_write_buffer(stream_t
*s
, unsigned char *buf
, int len
) {
348 rd
= s
->write_buffer(s
, buf
, len
);
352 assert(rd
== len
&& "stream_write_buffer(): unexpected short write");
356 int stream_seek_internal(stream_t
*s
, off_t newpos
)
358 if(newpos
==0 || newpos
!=s
->pos
){
360 case STREAMTYPE_STREAM
:
361 //s->pos=newpos; // real seek
362 // Some streaming protocol allow to seek backward and forward
363 // A function call that return -1 can tell that the protocol
364 // doesn't support seeking.
365 #ifdef CONFIG_NETWORKING
366 if(s
->seek
) { // new stream seek is much cleaner than streaming_ctrl one
367 if(!s
->seek(s
,newpos
)) {
368 mp_msg(MSGT_STREAM
,MSGL_ERR
, "Seek failed\n");
374 if( s
->streaming_ctrl
!=NULL
&& s
->streaming_ctrl
->streaming_seek
) {
375 if( s
->streaming_ctrl
->streaming_seek( s
->fd
, newpos
, s
->streaming_ctrl
)<0 ) {
376 mp_msg(MSGT_STREAM
,MSGL_INFO
,"Stream not seekable!\n");
383 mp_msg(MSGT_STREAM
,MSGL_INFO
,"Cannot seek backward in linear streams!\n");
388 // This should at the beginning as soon as all streams are converted
392 if(!s
->seek(s
,newpos
)) {
393 mp_msg(MSGT_STREAM
,MSGL_ERR
, "Seek failed\n");
397 // putchar('.');fflush(stdout);
399 // putchar('%');fflush(stdout);
404 int stream_seek_long(stream_t
*s
,off_t pos
){
408 // if( mp_msg_test(MSGT_STREAM,MSGL_DBG3) ) printf("seek_long to 0x%X\n",(unsigned int)pos);
410 s
->buf_pos
=s
->buf_len
=0;
412 if(s
->mode
== STREAM_WRITE
) {
413 if(!s
->seek
|| !s
->seek(s
,pos
))
419 newpos
= (pos
/s
->sector_size
)*s
->sector_size
;
421 newpos
= pos
&(~((off_t
)STREAM_BUFFER_SIZE
-1));
423 if( mp_msg_test(MSGT_STREAM
,MSGL_DBG3
) ){
424 mp_msg(MSGT_STREAM
,MSGL_DBG3
, "s->pos=%"PRIX64
" newpos=%"PRIX64
" new_bufpos=%"PRIX64
" buflen=%X \n",
425 (int64_t)s
->pos
,(int64_t)newpos
,(int64_t)pos
,s
->buf_len
);
429 res
= stream_seek_internal(s
, newpos
);
433 while(s
->pos
<newpos
){
434 if(stream_fill_buffer(s
)<=0) break; // EOF
437 s
->eof
= 0; // EOF reset when seek succeeds.
438 while (stream_fill_buffer(s
) > 0) {
440 s
->buf_pos
=pos
; // byte position in sector
445 // Fill failed, but seek still is a success.
450 mp_msg(MSGT_STREAM
,MSGL_V
,
451 "stream_seek: Seek to/past EOF: no buffer preloaded.\n");
456 void stream_reset(stream_t
*s
){
459 s
->buf_pos
=s
->buf_len
=0;
462 if(s
->control
) s
->control(s
,STREAM_CTRL_RESET
,NULL
);
466 int stream_control(stream_t
*s
, int cmd
, void *arg
){
467 if(!s
->control
) return STREAM_UNSUPPORTED
;
468 #ifdef CONFIG_STREAM_CACHE
470 return cache_do_control(s
, cmd
, arg
);
472 return s
->control(s
, cmd
, arg
);
475 stream_t
* new_memory_stream(unsigned char* data
,int len
){
480 s
=calloc(1, sizeof(stream_t
)+len
);
482 s
->type
=STREAMTYPE_MEMORY
;
483 s
->buf_pos
=0; s
->buf_len
=len
;
484 s
->start_pos
=0; s
->end_pos
=len
;
487 memcpy(s
->buffer
,data
,len
);
491 stream_t
* new_stream(int fd
,int type
){
492 stream_t
*s
=calloc(1, sizeof(stream_t
));
493 if(s
==NULL
) return NULL
;
498 int temp
= WSAStartup(0x0202, &wsdata
); // there might be a better place for this (-> later)
499 mp_msg(MSGT_STREAM
,MSGL_V
,"WINSOCK2 init: %i\n", temp
);
505 s
->buf_pos
=s
->buf_len
=0;
506 s
->start_pos
=s
->end_pos
=0;
514 void free_stream(stream_t
*s
){
515 // printf("\n*** free_stream() called ***\n");
516 #ifdef CONFIG_STREAM_CACHE
519 if (s
->capture_file
) {
520 fclose(s
->capture_file
);
521 s
->capture_file
= NULL
;
524 if(s
->close
) s
->close(s
);
526 /* on unix we define closesocket to close
527 on windows however we have to distinguish between
528 network socket and file */
529 if(s
->url
&& strstr(s
->url
,"://"))
534 mp_msg(MSGT_STREAM
,MSGL_V
,"WINSOCK2 uninit\n");
535 WSACleanup(); // there might be a better place for this (-> later)
537 // Disabled atm, i don't like that. s->priv can be anything after all
538 // streams should destroy their priv on close
544 stream_t
* new_ds_stream(demux_stream_t
*ds
) {
545 stream_t
* s
= new_stream(-1,STREAMTYPE_DS
);
550 void stream_set_interrupt_callback(int (*cb
)(struct input_ctx
*, int),
551 struct input_ctx
*ctx
)
553 stream_check_interrupt_cb
= cb
;
554 stream_check_interrupt_ctx
= ctx
;
557 int stream_check_interrupt(int time
) {
558 if(!stream_check_interrupt_cb
) {
559 usec_sleep(time
* 1000);
562 return stream_check_interrupt_cb(stream_check_interrupt_ctx
, time
);
566 * Helper function to read 16 bits little-endian and advance pointer
568 static uint16_t get_le16_inc(const uint8_t **buf
)
570 uint16_t v
= AV_RL16(*buf
);
576 * Helper function to read 16 bits big-endian and advance pointer
578 static uint16_t get_be16_inc(const uint8_t **buf
)
580 uint16_t v
= AV_RB16(*buf
);
586 * Find a newline character in buffer
587 * \param buf buffer to search
588 * \param len amount of bytes to search in buffer, may not overread
589 * \param utf16 chose between UTF-8/ASCII/other and LE and BE UTF-16
590 * 0 = UTF-8/ASCII/other, 1 = UTF-16-LE, 2 = UTF-16-BE
592 static const uint8_t *find_newline(const uint8_t *buf
, int len
, int utf16
)
595 const uint8_t *end
= buf
+ len
;
598 return (uint8_t *)memchr(buf
, '\n', len
);
600 while (buf
< end
- 1) {
601 GET_UTF16(c
, buf
< end
- 1 ? get_le16_inc(&buf
) : 0, return NULL
;)
602 if (buf
<= end
&& c
== '\n')
607 while (buf
< end
- 1) {
608 GET_UTF16(c
, buf
< end
- 1 ? get_be16_inc(&buf
) : 0, return NULL
;)
609 if (buf
<= end
&& c
== '\n')
618 * Copy a number of bytes, converting to UTF-8 if input is UTF-16
619 * \param dst buffer to copy to
620 * \param dstsize size of dst buffer
621 * \param src buffer to copy from
622 * \param len amount of bytes to copy from src
623 * \param utf16 chose between UTF-8/ASCII/other and LE and BE UTF-16
624 * 0 = UTF-8/ASCII/other, 1 = UTF-16-LE, 2 = UTF-16-BE
626 static int copy_characters(uint8_t *dst
, int dstsize
,
627 const uint8_t *src
, int *len
, int utf16
)
630 uint8_t *dst_end
= dst
+ dstsize
;
631 const uint8_t *end
= src
+ *len
;
636 memcpy(dst
, src
, *len
);
639 while (src
< end
- 1 && dst_end
- dst
> 8) {
641 GET_UTF16(c
, src
< end
- 1 ? get_le16_inc(&src
) : 0, ;)
642 PUT_UTF8(c
, tmp
, *dst
++ = tmp
;)
645 return dstsize
- (dst_end
- dst
);
647 while (src
< end
- 1 && dst_end
- dst
> 8) {
649 GET_UTF16(c
, src
< end
- 1 ? get_be16_inc(&src
) : 0, ;)
650 PUT_UTF8(c
, tmp
, *dst
++ = tmp
;)
653 return dstsize
- (dst_end
- dst
);
658 unsigned char* stream_read_line(stream_t
*s
,unsigned char* mem
, int max
, int utf16
) {
660 const unsigned char *end
;
661 unsigned char *ptr
= mem
;
662 if (max
< 1) return NULL
;
663 max
--; // reserve one for 0-termination
665 len
= s
->buf_len
-s
->buf_pos
;
666 // try to fill the buffer
668 (!cache_stream_fill_buffer(s
) ||
669 (len
= s
->buf_len
-s
->buf_pos
) <= 0)) break;
670 end
= find_newline(s
->buffer
+s
->buf_pos
, len
, utf16
);
671 if(end
) len
= end
- (s
->buffer
+s
->buf_pos
) + 1;
672 if(len
> 0 && max
> 0) {
673 int l
= copy_characters(ptr
, max
, s
->buffer
+s
->buf_pos
, &len
, utf16
);
682 if(s
->eof
&& ptr
== mem
) return NULL
;
686 struct bstr
stream_read_complete(struct stream
*s
, void *talloc_ctx
,
687 int max_size
, int padding_bytes
)
689 if (max_size
> 1000000000)
694 int padding
= FFMAX(padding_bytes
, 1);
696 if (s
->end_pos
> max_size
)
697 return (struct bstr
){NULL
, 0};
699 bufsize
= s
->end_pos
+ padding
;
703 buf
= talloc_realloc_size(talloc_ctx
, buf
, bufsize
);
704 int readsize
= stream_read(s
, buf
+ total_read
, bufsize
- total_read
);
705 total_read
+= readsize
;
706 if (total_read
< bufsize
)
708 if (bufsize
> max_size
) {
710 return (struct bstr
){NULL
, 0};
712 bufsize
= FFMIN(bufsize
+ (bufsize
>> 1), max_size
+ padding
);
714 buf
= talloc_realloc_size(talloc_ctx
, buf
, total_read
+ padding
);
715 return (struct bstr
){buf
, total_read
};