Quit wf-recorder when the recording target fails

src/frame-writer.cpp

@@ -601,7 +601,9 @@ void FrameWriter::finish_frame(AVCodecContext *enc_ctx, AVPacket& pkt)
         pending_mutex.unlock();
     }
 
-    av_interleaved_write_frame(fmtCtx, &pkt);
+    if (av_interleaved_write_frame(fmtCtx, &pkt) != 0) {
+        params.write_aborted_flag = true;
+    }
     av_packet_unref(&pkt);
 
     if (params.enable_audio)

src/frame-writer.hpp

@@ -8,6 +8,7 @@
 #include <string>
 #include <vector>
 #include <map>
+#include <atomic>
 #include "config.h"
 
 #define AUDIO_RATE 44100
@@ -64,6 +65,9 @@ struct FrameWriterParams
     bool force_yuv;
     int opencl_device;
     int bframes;
+
+    std::atomic<bool>& write_aborted_flag;
+    FrameWriterParams(std::atomic<bool>& flag): write_aborted_flag(flag) {}
 };
 
 class FrameWriter

src/main.cpp

@@ -584,7 +584,7 @@ With no FILE, start recording the current screen.
 
 int main(int argc, char *argv[])
 {
-    FrameWriterParams params;
+    FrameWriterParams params = FrameWriterParams(exit_main_loop);
     params.file = "recording.mp4";
     params.codec = DEFAULT_CODEC;
     params.enable_ffmpeg_debug_output = false;