Ticket #8989: ffmpeg_defect_ticket_8989_wip_fix_vignesh_ravichandran.patch

libavformat/hlsenc.c

@@ -25 +25 @@
 #include <stdint.h>
 #if HAVE_UNISTD_H
 #include <unistd.h>
+#include <stdbool.h>
+
 #endif
 
 #if CONFIG_GCRYPT
@@ -88 +90 @@
     char iv_string[KEYSIZE*2 + 1];
 
     struct HLSSegment *next;
+    double discont_program_date_time;
 } HLSSegment;
 
 typedef enum HLSFlags {
@@ -1124 +1127 @@
     en->keyframe_size     = vs->video_keyframe_size;
     en->next     = NULL;
     en->discont  = 0;
+    en->discont_program_date_time = 0;
 
     if (vs->discontinuity) {
         en->discont = 1;
@@ -1182 +1186 @@
     char line[MAX_URL_SIZE];
     const char *ptr;
     const char *end;
+    bool is_discont_detected = false;
+    double discont_program_date_time = 0;
 
     if ((ret = ffio_open_whitelist(&in, url, AVIO_FLAG_READ,
                                    &s->interrupt_callback, NULL,
@@ -1211 +1217 @@
         } else if (av_strstart(line, "#EXT-X-DISCONTINUITY", &ptr)) {
             is_segment = 1;
             vs->discontinuity = 1;
+            is_discont_detected = true;
         } else if (av_strstart(line, "#EXTINF:", &ptr)) {
             is_segment = 1;
             vs->duration = atof(ptr);
@@ -1236 +1243 @@
                     av_strlcpy(vs->iv_string, ptr, sizeof(vs->iv_string));
                 }
             }
-
+        } else if (av_strstart(line, "#EXT-X-PROGRAM-DATE-TIME:", &ptr) && is_discont_detected) {
+            av_log(hls, AV_LOG_VERBOSE, "program_date_time: %s\n", ptr);
+            struct tm program_date_time;
+            int y,M,d,h,m,s;
+            double ms;
+            sscanf(ptr, "%d-%d-%dT%d:%d:%d.%lf", &y, &M, &d, &h, &m, &s, &ms);
+
+            program_date_time.tm_year = y - 1900;
+            program_date_time.tm_mon = M - 1;
+            program_date_time.tm_mday = d;
+            program_date_time.tm_hour = h;
+            program_date_time.tm_min = m;
+            program_date_time.tm_sec = s;
+            program_date_time.tm_isdst = -1;
+
+            discont_program_date_time = mktime(&program_date_time);
+            discont_program_date_time += (double)(ms / 1000);
+            is_discont_detected = false;
         } else if (av_strstart(line, "#", NULL)) {
             continue;
         } else if (line[0]) {
@@ -1254 +1278 @@
                 ret = hls_append_segment(s, hls, vs, vs->duration, vs->start_pos, vs->size);
                 if (ret < 0)
                     goto fail;
+                vs->last_segment->discont_program_date_time = discont_program_date_time;
+                discont_program_date_time += vs->duration;
                 vs->start_pos = new_start_pos;
             }
         }
@@ -1572 +1598 @@
         ret = ff_hls_write_file_entry(byterange_mode ? hls->m3u8_out : vs->out, en->discont, byterange_mode,
                                       en->duration, hls->flags & HLS_ROUND_DURATIONS,
                                       en->size, en->pos, hls->baseurl,
-                                      en->filename, prog_date_time_p, en->keyframe_size, en->keyframe_pos, hls->flags & HLS_I_FRAMES_ONLY);
+                                      en->filename,
+                                      en->discont_program_date_time ? &en->discont_program_date_time : prog_date_time_p,
+                                      en->keyframe_size, en->keyframe_pos, hls->flags & HLS_I_FRAMES_ONLY);
         if (ret < 0) {
             av_log(s, AV_LOG_WARNING, "ff_hls_write_file_entry get error\n");
         }
+        if (en->discont_program_date_time)
+            en->discont_program_date_time -= en->duration;
     }
 
     if (last && (hls->flags & HLS_OMIT_ENDLIST)==0)