2 * Buffered I/O for ffmpeg system
3 * Copyright (c) 2000,2001 Fabrice Bellard
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 #include "libavutil/crc.h"
23 #include "libavutil/intreadwrite.h"
28 #define IO_BUFFER_SIZE 32768
30 static void fill_buffer(ByteIOContext
*s
);
32 int init_put_byte(ByteIOContext
*s
,
33 unsigned char *buffer
,
37 int (*read_packet
)(void *opaque
, uint8_t *buf
, int buf_size
),
38 int (*write_packet
)(void *opaque
, uint8_t *buf
, int buf_size
),
39 int64_t (*seek
)(void *opaque
, int64_t offset
, int whence
))
42 s
->buffer_size
= buffer_size
;
45 url_resetbuf(s
, write_flag
? URL_WRONLY
: URL_RDONLY
);
46 s
->write_packet
= write_packet
;
47 s
->read_packet
= read_packet
;
54 s
->max_packet_size
= 0;
55 s
->update_checksum
= NULL
;
56 if(!read_packet
&& !write_flag
){
58 s
->buf_end
= s
->buffer
+ buffer_size
;
65 ByteIOContext
*av_alloc_put_byte(
66 unsigned char *buffer
,
70 int (*read_packet
)(void *opaque
, uint8_t *buf
, int buf_size
),
71 int (*write_packet
)(void *opaque
, uint8_t *buf
, int buf_size
),
72 int64_t (*seek
)(void *opaque
, int64_t offset
, int whence
))
74 ByteIOContext
*s
= av_mallocz(sizeof(ByteIOContext
));
75 init_put_byte(s
, buffer
, buffer_size
, write_flag
, opaque
,
76 read_packet
, write_packet
, seek
);
80 static void flush_buffer(ByteIOContext
*s
)
82 if (s
->buf_ptr
> s
->buffer
) {
83 if (s
->write_packet
&& !s
->error
){
84 int ret
= s
->write_packet(s
->opaque
, s
->buffer
, s
->buf_ptr
- s
->buffer
);
89 if(s
->update_checksum
){
90 s
->checksum
= s
->update_checksum(s
->checksum
, s
->checksum_ptr
, s
->buf_ptr
- s
->checksum_ptr
);
91 s
->checksum_ptr
= s
->buffer
;
93 s
->pos
+= s
->buf_ptr
- s
->buffer
;
95 s
->buf_ptr
= s
->buffer
;
98 void put_byte(ByteIOContext
*s
, int b
)
101 if (s
->buf_ptr
>= s
->buf_end
)
105 void put_buffer(ByteIOContext
*s
, const unsigned char *buf
, int size
)
108 int len
= FFMIN(s
->buf_end
- s
->buf_ptr
, size
);
109 memcpy(s
->buf_ptr
, buf
, len
);
112 if (s
->buf_ptr
>= s
->buf_end
)
120 void put_flush_packet(ByteIOContext
*s
)
126 int64_t url_fseek(ByteIOContext
*s
, int64_t offset
, int whence
)
132 return AVERROR(EINVAL
);
134 pos
= s
->pos
- (s
->write_flag
? 0 : (s
->buf_end
- s
->buffer
));
136 if (whence
!= SEEK_CUR
&& whence
!= SEEK_SET
)
137 return AVERROR(EINVAL
);
139 if (whence
== SEEK_CUR
) {
140 offset1
= pos
+ (s
->buf_ptr
- s
->buffer
);
145 offset1
= offset
- pos
;
146 if (!s
->must_flush
&&
147 offset1
>= 0 && offset1
<= (s
->buf_end
- s
->buffer
)) {
148 /* can do the seek inside the buffer */
149 s
->buf_ptr
= s
->buffer
+ offset1
;
150 } else if(s
->is_streamed
&& !s
->write_flag
&&
151 offset1
>= 0 && offset1
< (s
->buf_end
- s
->buffer
) + (1<<16)){
152 while(s
->pos
< offset
&& !s
->eof_reached
)
155 return AVERROR(EPIPE
);
156 s
->buf_ptr
= s
->buf_end
+ offset
- s
->pos
;
158 int64_t res
= AVERROR(EPIPE
);
160 #if CONFIG_MUXERS || CONFIG_NETWORK
165 #endif /* CONFIG_MUXERS || CONFIG_NETWORK */
166 if (!s
->seek
|| (res
= s
->seek(s
->opaque
, offset
, SEEK_SET
)) < 0)
169 s
->buf_end
= s
->buffer
;
170 s
->buf_ptr
= s
->buffer
;
177 void url_fskip(ByteIOContext
*s
, int64_t offset
)
179 url_fseek(s
, offset
, SEEK_CUR
);
182 int64_t url_ftell(ByteIOContext
*s
)
184 return url_fseek(s
, 0, SEEK_CUR
);
187 int64_t url_fsize(ByteIOContext
*s
)
192 return AVERROR(EINVAL
);
195 return AVERROR(EPIPE
);
196 size
= s
->seek(s
->opaque
, 0, AVSEEK_SIZE
);
198 if ((size
= s
->seek(s
->opaque
, -1, SEEK_END
)) < 0)
201 s
->seek(s
->opaque
, s
->pos
, SEEK_SET
);
206 int url_feof(ByteIOContext
*s
)
210 return s
->eof_reached
;
213 int url_ferror(ByteIOContext
*s
)
220 void put_le32(ByteIOContext
*s
, unsigned int val
)
223 put_byte(s
, val
>> 8);
224 put_byte(s
, val
>> 16);
225 put_byte(s
, val
>> 24);
228 void put_be32(ByteIOContext
*s
, unsigned int val
)
230 put_byte(s
, val
>> 24);
231 put_byte(s
, val
>> 16);
232 put_byte(s
, val
>> 8);
236 void put_strz(ByteIOContext
*s
, const char *str
)
239 put_buffer(s
, (const unsigned char *) str
, strlen(str
) + 1);
244 void put_le64(ByteIOContext
*s
, uint64_t val
)
246 put_le32(s
, (uint32_t)(val
& 0xffffffff));
247 put_le32(s
, (uint32_t)(val
>> 32));
250 void put_be64(ByteIOContext
*s
, uint64_t val
)
252 put_be32(s
, (uint32_t)(val
>> 32));
253 put_be32(s
, (uint32_t)(val
& 0xffffffff));
256 void put_le16(ByteIOContext
*s
, unsigned int val
)
259 put_byte(s
, val
>> 8);
262 void put_be16(ByteIOContext
*s
, unsigned int val
)
264 put_byte(s
, val
>> 8);
268 void put_le24(ByteIOContext
*s
, unsigned int val
)
270 put_le16(s
, val
& 0xffff);
271 put_byte(s
, val
>> 16);
274 void put_be24(ByteIOContext
*s
, unsigned int val
)
276 put_be16(s
, val
>> 8);
280 void put_tag(ByteIOContext
*s
, const char *tag
)
289 static void fill_buffer(ByteIOContext
*s
)
291 uint8_t *dst
= !s
->max_packet_size
&& s
->buf_end
- s
->buffer
< s
->buffer_size
? s
->buf_ptr
: s
->buffer
;
292 int len
= s
->buffer_size
- (dst
- s
->buffer
);
294 assert(s
->buf_ptr
== s
->buf_end
);
296 /* no need to do anything if EOF already reached */
300 if(s
->update_checksum
&& dst
== s
->buffer
){
301 if(s
->buf_end
> s
->checksum_ptr
)
302 s
->checksum
= s
->update_checksum(s
->checksum
, s
->checksum_ptr
, s
->buf_end
- s
->checksum_ptr
);
303 s
->checksum_ptr
= s
->buffer
;
307 len
= s
->read_packet(s
->opaque
, dst
, len
);
311 /* do not modify buffer if EOF reached so that a seek back can
312 be done without rereading data */
319 s
->buf_end
= dst
+ len
;
323 unsigned long ff_crc04C11DB7_update(unsigned long checksum
, const uint8_t *buf
,
326 return av_crc(av_crc_get_table(AV_CRC_32_IEEE
), checksum
, buf
, len
);
329 unsigned long get_checksum(ByteIOContext
*s
)
331 s
->checksum
= s
->update_checksum(s
->checksum
, s
->checksum_ptr
, s
->buf_ptr
- s
->checksum_ptr
);
332 s
->update_checksum
= NULL
;
336 void init_checksum(ByteIOContext
*s
,
337 unsigned long (*update_checksum
)(unsigned long c
, const uint8_t *p
, unsigned int len
),
338 unsigned long checksum
)
340 s
->update_checksum
= update_checksum
;
341 if(s
->update_checksum
){
342 s
->checksum
= checksum
;
343 s
->checksum_ptr
= s
->buf_ptr
;
347 /* XXX: put an inline version */
348 int get_byte(ByteIOContext
*s
)
350 if (s
->buf_ptr
< s
->buf_end
) {
351 return *s
->buf_ptr
++;
354 if (s
->buf_ptr
< s
->buf_end
)
355 return *s
->buf_ptr
++;
361 int url_fgetc(ByteIOContext
*s
)
363 if (s
->buf_ptr
< s
->buf_end
) {
364 return *s
->buf_ptr
++;
367 if (s
->buf_ptr
< s
->buf_end
)
368 return *s
->buf_ptr
++;
374 int get_buffer(ByteIOContext
*s
, unsigned char *buf
, int size
)
380 len
= s
->buf_end
- s
->buf_ptr
;
384 if(size
> s
->buffer_size
&& !s
->update_checksum
){
386 len
= s
->read_packet(s
->opaque
, buf
, size
);
388 /* do not modify buffer if EOF reached so that a seek back can
389 be done without rereading data */
398 s
->buf_ptr
= s
->buffer
;
399 s
->buf_end
= s
->buffer
/* + len*/;
403 len
= s
->buf_end
- s
->buf_ptr
;
408 memcpy(buf
, s
->buf_ptr
, len
);
415 if (url_ferror(s
)) return url_ferror(s
);
416 if (url_feof(s
)) return AVERROR_EOF
;
421 int get_partial_buffer(ByteIOContext
*s
, unsigned char *buf
, int size
)
428 len
= s
->buf_end
- s
->buf_ptr
;
431 len
= s
->buf_end
- s
->buf_ptr
;
435 memcpy(buf
, s
->buf_ptr
, len
);
438 if (url_ferror(s
)) return url_ferror(s
);
439 if (url_feof(s
)) return AVERROR_EOF
;
444 unsigned int get_le16(ByteIOContext
*s
)
448 val
|= get_byte(s
) << 8;
452 unsigned int get_le24(ByteIOContext
*s
)
456 val
|= get_byte(s
) << 16;
460 unsigned int get_le32(ByteIOContext
*s
)
464 val
|= get_le16(s
) << 16;
468 uint64_t get_le64(ByteIOContext
*s
)
471 val
= (uint64_t)get_le32(s
);
472 val
|= (uint64_t)get_le32(s
) << 32;
476 unsigned int get_be16(ByteIOContext
*s
)
479 val
= get_byte(s
) << 8;
484 unsigned int get_be24(ByteIOContext
*s
)
487 val
= get_be16(s
) << 8;
491 unsigned int get_be32(ByteIOContext
*s
)
494 val
= get_be16(s
) << 16;
499 char *get_strz(ByteIOContext
*s
, char *buf
, int maxlen
)
504 while ((c
= get_byte(s
))) {
509 buf
[i
] = 0; /* Ensure null terminated, but may be truncated */
514 uint64_t get_be64(ByteIOContext
*s
)
517 val
= (uint64_t)get_be32(s
) << 32;
518 val
|= (uint64_t)get_be32(s
);
522 uint64_t ff_get_v(ByteIOContext
*bc
){
528 val
= (val
<<7) + (tmp
&127);
533 int url_fdopen(ByteIOContext
**s
, URLContext
*h
)
536 int buffer_size
, max_packet_size
;
539 max_packet_size
= url_get_max_packet_size(h
);
540 if (max_packet_size
) {
541 buffer_size
= max_packet_size
; /* no need to bufferize more than one packet */
543 buffer_size
= IO_BUFFER_SIZE
;
545 buffer
= av_malloc(buffer_size
);
547 return AVERROR(ENOMEM
);
549 *s
= av_mallocz(sizeof(ByteIOContext
));
552 return AVERROR(ENOMEM
);
555 if (init_put_byte(*s
, buffer
, buffer_size
,
556 (h
->flags
& URL_WRONLY
|| h
->flags
& URL_RDWR
), h
,
557 url_read
, url_write
, url_seek
) < 0) {
562 (*s
)->is_streamed
= h
->is_streamed
;
563 (*s
)->max_packet_size
= max_packet_size
;
565 (*s
)->read_pause
= (int (*)(void *, int))h
->prot
->url_read_pause
;
566 (*s
)->read_seek
= (int64_t (*)(void *, int, int64_t, int))h
->prot
->url_read_seek
;
571 int url_setbufsize(ByteIOContext
*s
, int buf_size
)
574 buffer
= av_malloc(buf_size
);
576 return AVERROR(ENOMEM
);
580 s
->buffer_size
= buf_size
;
582 url_resetbuf(s
, s
->write_flag
? URL_WRONLY
: URL_RDONLY
);
586 int url_resetbuf(ByteIOContext
*s
, int flags
)
588 URLContext
*h
= s
->opaque
;
589 if ((flags
& URL_RDWR
) || (h
&& h
->flags
!= flags
&& !h
->flags
& URL_RDWR
))
590 return AVERROR(EINVAL
);
592 if (flags
& URL_WRONLY
) {
593 s
->buf_end
= s
->buffer
+ s
->buffer_size
;
596 s
->buf_end
= s
->buffer
;
602 int url_fopen(ByteIOContext
**s
, const char *filename
, int flags
)
607 err
= url_open(&h
, filename
, flags
);
610 err
= url_fdopen(s
, h
);
618 int url_fclose(ByteIOContext
*s
)
620 URLContext
*h
= s
->opaque
;
627 URLContext
*url_fileno(ByteIOContext
*s
)
633 int url_fprintf(ByteIOContext
*s
, const char *fmt
, ...)
640 ret
= vsnprintf(buf
, sizeof(buf
), fmt
, ap
);
642 put_buffer(s
, buf
, strlen(buf
));
645 #endif //CONFIG_MUXERS
647 char *url_fgets(ByteIOContext
*s
, char *buf
, int buf_size
)
657 if (c
== EOF
|| c
== '\n')
659 if ((q
- buf
) < buf_size
- 1)
668 int url_fget_max_packet_size(ByteIOContext
*s
)
670 return s
->max_packet_size
;
673 int av_url_read_fpause(ByteIOContext
*s
, int pause
)
676 return AVERROR(ENOSYS
);
677 return s
->read_pause(s
->opaque
, pause
);
680 int64_t av_url_read_fseek(ByteIOContext
*s
, int stream_index
,
681 int64_t timestamp
, int flags
)
683 URLContext
*h
= s
->opaque
;
686 return AVERROR(ENOSYS
);
687 ret
= s
->read_seek(h
, stream_index
, timestamp
, flags
);
689 s
->buf_ptr
= s
->buf_end
; // Flush buffer
690 s
->pos
= s
->seek(h
, 0, SEEK_CUR
);
695 /* url_open_dyn_buf and url_close_dyn_buf are used in rtp.c to send a response
696 * back to the server even if CONFIG_MUXERS is false. */
697 #if CONFIG_MUXERS || CONFIG_NETWORK
698 /* buffer handling */
699 int url_open_buf(ByteIOContext
**s
, uint8_t *buf
, int buf_size
, int flags
)
702 *s
= av_mallocz(sizeof(ByteIOContext
));
704 return AVERROR(ENOMEM
);
705 ret
= init_put_byte(*s
, buf
, buf_size
,
706 (flags
& URL_WRONLY
|| flags
& URL_RDWR
),
707 NULL
, NULL
, NULL
, NULL
);
713 int url_close_buf(ByteIOContext
*s
)
716 return s
->buf_ptr
- s
->buffer
;
719 /* output in a dynamic buffer */
721 typedef struct DynBuffer
{
722 int pos
, size
, allocated_size
;
725 uint8_t io_buffer
[1];
728 static int dyn_buf_write(void *opaque
, uint8_t *buf
, int buf_size
)
730 DynBuffer
*d
= opaque
;
731 unsigned new_size
, new_allocated_size
;
733 /* reallocate buffer if needed */
734 new_size
= d
->pos
+ buf_size
;
735 new_allocated_size
= d
->allocated_size
;
736 if(new_size
< d
->pos
|| new_size
> INT_MAX
/2)
738 while (new_size
> new_allocated_size
) {
739 if (!new_allocated_size
)
740 new_allocated_size
= new_size
;
742 new_allocated_size
+= new_allocated_size
/ 2 + 1;
745 if (new_allocated_size
> d
->allocated_size
) {
746 d
->buffer
= av_realloc(d
->buffer
, new_allocated_size
);
747 if(d
->buffer
== NULL
)
748 return AVERROR(ENOMEM
);
749 d
->allocated_size
= new_allocated_size
;
751 memcpy(d
->buffer
+ d
->pos
, buf
, buf_size
);
753 if (d
->pos
> d
->size
)
758 static int dyn_packet_buf_write(void *opaque
, uint8_t *buf
, int buf_size
)
760 unsigned char buf1
[4];
763 /* packetized write: output the header */
764 AV_WB32(buf1
, buf_size
);
765 ret
= dyn_buf_write(opaque
, buf1
, 4);
770 return dyn_buf_write(opaque
, buf
, buf_size
);
773 static int64_t dyn_buf_seek(void *opaque
, int64_t offset
, int whence
)
775 DynBuffer
*d
= opaque
;
777 if (whence
== SEEK_CUR
)
779 else if (whence
== SEEK_END
)
781 if (offset
< 0 || offset
> 0x7fffffffLL
)
787 static int url_open_dyn_buf_internal(ByteIOContext
**s
, int max_packet_size
)
791 unsigned io_buffer_size
= max_packet_size
? max_packet_size
: 1024;
793 if(sizeof(DynBuffer
) + io_buffer_size
< io_buffer_size
)
795 d
= av_mallocz(sizeof(DynBuffer
) + io_buffer_size
);
797 return AVERROR(ENOMEM
);
798 *s
= av_mallocz(sizeof(ByteIOContext
));
801 return AVERROR(ENOMEM
);
803 d
->io_buffer_size
= io_buffer_size
;
804 ret
= init_put_byte(*s
, d
->io_buffer
, io_buffer_size
,
806 max_packet_size
? dyn_packet_buf_write
: dyn_buf_write
,
807 max_packet_size
? NULL
: dyn_buf_seek
);
809 (*s
)->max_packet_size
= max_packet_size
;
817 int url_open_dyn_buf(ByteIOContext
**s
)
819 return url_open_dyn_buf_internal(s
, 0);
822 int url_open_dyn_packet_buf(ByteIOContext
**s
, int max_packet_size
)
824 if (max_packet_size
<= 0)
826 return url_open_dyn_buf_internal(s
, max_packet_size
);
829 int url_close_dyn_buf(ByteIOContext
*s
, uint8_t **pbuffer
)
831 DynBuffer
*d
= s
->opaque
;
836 *pbuffer
= d
->buffer
;
842 #endif /* CONFIG_MUXERS || CONFIG_NETWORK */