hlsenc: write playlist into a temp file and replace the original atomically

Signed-off-by: Martin Storsjö <martin@martin.st>

1 files changed, 5 insertions, 1 deletions

diff --git a/libavformat/hlsenc.c b/libavformat/hlsenc.c
index d8fd2546ef..21c952630d 100644
--- a/libavformat/hlsenc.c
+++ b/libavformat/hlsenc.c
@@ -134,9 +134,11 @@ static int hls_window(AVFormatContext *s, int last)
     int target_duration = 0;
     int ret = 0;
     AVIOContext *out = NULL;
+    char temp_filename[1024];
     int64_t sequence = FFMAX(hls->start_sequence, hls->sequence - hls->size);
 
-    if ((ret = avio_open2(&out, s->filename, AVIO_FLAG_WRITE,
+    snprintf(temp_filename, sizeof(temp_filename), "%s.tmp", s->filename);
+    if ((ret = avio_open2(&out, temp_filename, AVIO_FLAG_WRITE,
                           &s->interrupt_callback, NULL)) < 0)
         goto fail;
 
@@ -168,6 +170,8 @@ static int hls_window(AVFormatContext *s, int last)
 
 fail:
     avio_closep(&out);
+    if (ret >= 0)
+        ff_rename(temp_filename, s->filename);
     return ret;
 }