From 33c1d38a21cb14c0b151f5bc7e52217cf4296819 Mon Sep 17 00:00:00 2001 From: Adrian Shum Date: Tue, 18 Oct 2022 12:38:18 +0800 Subject: [PATCH] refactor(ffmpeg): handles EOF with buf exists --- ffmpeg/callback.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/ffmpeg/callback.go b/ffmpeg/callback.go index 2477801..24e7fdf 100644 --- a/ffmpeg/callback.go +++ b/ffmpeg/callback.go @@ -23,8 +23,10 @@ func goPacketRead(opaque unsafe.Pointer, buffer *C.uint8_t, bufSize C.int) C.int } buf := *(*[]byte)(unsafe.Pointer(sh)) n, err := ctx.reader.Read(buf) - if err == io.EOF && n == 0 { - return C.int(ErrEOF) + if err == io.EOF { + if n == 0 { + return C.int(ErrEOF) + } } else if err != nil { return C.int(ErrUnknown) }