avutil/fifo: Fix the case where func() returns less bytes than requested in av_fifo_generic_write()

Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
10 years ago · fcbea93cf8
parent 6e80fe1ecd
commit fcbea93cf8
1 changed files with 2 additions and 1 deletions
--- a/libavutil/fifo.c
+++ b/libavutil/fifo.c
@ -129,7 +129,8 @@ int av_fifo_generic_write(AVFifoBuffer *f, void *src, int size,
    do {
        int len = FFMIN(f->end - wptr, size);
        if (func) {
-            if (func(src, wptr, len) <= 0)
+            len = func(src, wptr, len);
+            if (len <= 0)
                break;
        } else {
            memcpy(wptr, src, len);