fix(libvideo2x): added missing checks around freeing memory

Signed-off-by: k4yt3x <[email protected]>
k4yt3x · Nov 13, 2024 · 980b44b · 980b44b
1 parent 561671c
commit 980b44b
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/libvideo2x.cpp b/src/libvideo2x.cpp
@@ -62,8 +62,14 @@ static int process_frames(
 
     // Lambda function for cleaning up resources
     auto cleanup = [&]() {
-        av_frame_free(&frame);
-        av_packet_free(&packet);
+        if (frame) {
+            av_frame_free(&frame);
+            frame = nullptr;
+        }
+        if (packet) {
+            av_packet_free(&packet);
+            packet = nullptr;
+        }
         for (AVFrame *&flushed_frame : flushed_frames) {
             if (flushed_frame) {
                 av_frame_free(&flushed_frame);