Merge commit 'bff0349d9da527084e1433167466d8afc9e25c7f'

* commit 'bff0349d9da527084e1433167466d8afc9e25c7f':
  avio: Add avio_read wrapper to simplify error checking

Merged-by: Michael Niedermayer <michaelni@gmx.at>
pull/136/head
Michael Niedermayer 10 years ago
commit ebab902a1c
  1. 7
      libavformat/avio_internal.h
  2. 8
      libavformat/aviobuf.c

@ -85,6 +85,13 @@ int ffio_rewind_with_probe_data(AVIOContext *s, unsigned char **buf, int buf_siz
uint64_t ffio_read_varlen(AVIOContext *bc); uint64_t ffio_read_varlen(AVIOContext *bc);
/**
* Read size bytes from AVIOContext into buf.
* Check that exactly size bytes have been read.
* @return number of bytes read or AVERROR
*/
int ffio_read_size(AVIOContext *s, unsigned char *buf, int size);
/** @warning must be called before any I/O */ /** @warning must be called before any I/O */
int ffio_set_buf_size(AVIOContext *s, int buf_size); int ffio_set_buf_size(AVIOContext *s, int buf_size);

@ -580,6 +580,14 @@ int avio_read(AVIOContext *s, unsigned char *buf, int size)
return size1 - size; return size1 - size;
} }
int ffio_read_size(AVIOContext *s, unsigned char *buf, int size)
{
int ret = avio_read(s, buf, size);
if (ret != size)
return AVERROR_INVALIDDATA;
return ret;
}
int ffio_read_indirect(AVIOContext *s, unsigned char *buf, int size, const unsigned char **data) int ffio_read_indirect(AVIOContext *s, unsigned char *buf, int size, const unsigned char **data)
{ {
if (s->buf_end - s->buf_ptr >= size && !s->write_flag) { if (s->buf_end - s->buf_ptr >= size && !s->write_flag) {

Loading…
Cancel
Save