diff options
author | Nicolas George <nicolas.george@normalesup.org> | 2011-02-04 19:12:37 +0100 |
---|---|---|
committer | Ronald S. Bultje <rsbultje@gmail.com> | 2011-02-05 20:29:59 -0500 |
commit | 90441276e4f661c6aec5e4d2c5718cde1ff1946d (patch) | |
tree | d79d57cb33c9bdcc3a78d5484f72e41b12fe08e0 | |
parent | fe174fc8fc4bbdb050014a945de7eb9b28ba358e (diff) | |
download | ffmpeg-streaming-90441276e4f661c6aec5e4d2c5718cde1ff1946d.zip ffmpeg-streaming-90441276e4f661c6aec5e4d2c5718cde1ff1946d.tar.gz |
Non-blocking protocol: core wrapper functions
Signed-off-by: Nicolas George <nicolas.george@normalesup.org>
Signed-off-by: Ronald S. Bultje <rsbultje@gmail.com>
-rw-r--r-- | libavformat/avio.c | 32 | ||||
-rw-r--r-- | libavformat/avio.h | 1 |
2 files changed, 19 insertions, 14 deletions
diff --git a/libavformat/avio.c b/libavformat/avio.c index 9a4d735..a19ec37 100644 --- a/libavformat/avio.c +++ b/libavformat/avio.c @@ -206,24 +206,21 @@ int url_open(URLContext **puc, const char *filename, int flags) return ret; } -int url_read(URLContext *h, unsigned char *buf, int size) -{ - int ret; - if (h->flags & URL_WRONLY) - return AVERROR(EIO); - ret = h->prot->url_read(h, buf, size); - return ret; -} - -static inline int retry_transfer_wrapper(URLContext *h, unsigned char *buf, int size, +static inline int retry_transfer_wrapper(URLContext *h, unsigned char *buf, int size, int size_min, int (*transfer_func)(URLContext *h, unsigned char *buf, int size)) { int ret, len; int fast_retries = 5; len = 0; - while (len < size) { + while (len < size_min) { + if (url_interrupt_cb()) + return AVERROR(EINTR); ret = transfer_func(h, buf+len, size-len); + if (ret == AVERROR(EINTR)) + continue; + if (h->flags & URL_FLAG_NONBLOCK) + return ret; if (ret == AVERROR(EAGAIN)) { ret = 0; if (fast_retries) @@ -239,9 +236,18 @@ static inline int retry_transfer_wrapper(URLContext *h, unsigned char *buf, int return len; } +int url_read(URLContext *h, unsigned char *buf, int size) +{ + if (h->flags & URL_WRONLY) + return AVERROR(EIO); + return retry_transfer_wrapper(h, buf, size, 1, h->prot->url_read); +} + int url_read_complete(URLContext *h, unsigned char *buf, int size) { - return retry_transfer_wrapper(h, buf, size, url_read); + if (h->flags & URL_WRONLY) + return AVERROR(EIO); + return retry_transfer_wrapper(h, buf, size, size, h->prot->url_read); } int url_write(URLContext *h, const unsigned char *buf, int size) @@ -252,7 +258,7 @@ int url_write(URLContext *h, const unsigned char *buf, int size) if (h->max_packet_size && size > h->max_packet_size) return AVERROR(EIO); - return retry_transfer_wrapper(h, buf, size, h->prot->url_write); + return retry_transfer_wrapper(h, buf, size, size, h->prot->url_write); } int64_t url_seek(URLContext *h, int64_t pos, int whence) diff --git a/libavformat/avio.h b/libavformat/avio.h index d05cab1..c899c0d 100644 --- a/libavformat/avio.h +++ b/libavformat/avio.h @@ -151,7 +151,6 @@ int url_read(URLContext *h, unsigned char *buf, int size); /** * Read as many bytes as possible (up to size), calling the * read function multiple times if necessary. - * Will also retry if the read function returns AVERROR(EAGAIN). * This makes special short-read handling in applications * unnecessary, if the return value is < size then it is * certain there was either an error or the end of file was reached. |