From 722f9b63c20918cb8f503b65e6bcafcc303e355c Mon Sep 17 00:00:00 2001
From: wm4 <wm4@nowhere>
Date: Sat, 30 Dec 2017 17:39:08 +0100
Subject: [PATCH] stream_lavf: minor fixes to HTTP reconnection support

Don't drop the stream buffers, because the read call (that must have
been failing) might try to extend an existing read buffer in the first
place. Just move the messy seek logic to stream_lavf.c. (In theory,
stream_lavf should probably make libavformat connect at the correct
offset instead of using a seek to reconnect it again. This patch doesn't
fix it, but at least it's a good argument to have the messing with the
position not in the generic code.)

Also update the comment about avio not supporting reconnecting. It has
that feature now. Maybe we should use it, but only after it gets fixed.
---
 stream/stream.c      |  3 +--
 stream/stream_lavf.c | 17 +++++++++++++----
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/stream/stream.c b/stream/stream.c
index 44a773cd1f..dba30d5df2 100644
--- a/stream/stream.c
+++ b/stream/stream.c
@@ -341,7 +341,6 @@ static bool stream_reconnect(stream_t *s)
     if (!s->streaming || s->caching || !s->seekable || !s->cancel)
         return false;
 
-    int64_t pos = s->pos;
     double sleep_secs = 0;
     for (int retry = 0; retry < 6; retry++) {
         if (mp_cancel_wait(s->cancel, sleep_secs))
@@ -350,7 +349,7 @@ static bool stream_reconnect(stream_t *s)
         int r = stream_control(s, STREAM_CTRL_RECONNECT, NULL);
         if (r == STREAM_UNSUPPORTED)
             break;
-        if (r == STREAM_OK && stream_seek_unbuffered(s, pos) && s->pos == pos) {
+        if (r == STREAM_OK) {
             MP_WARN(s, "Reconnected successfully.\n");
             return true;
         }
diff --git a/stream/stream_lavf.c b/stream/stream_lavf.c
index 80a27bd3e6..2230247f28 100644
--- a/stream/stream_lavf.c
+++ b/stream/stream_lavf.c
@@ -178,12 +178,21 @@ static int control(stream_t *s, int cmd, void *arg)
     case STREAM_CTRL_RECONNECT: {
         if (avio && avio->write_flag)
             break; // don't bother with this
-        // avio doesn't seem to support this - emulate it by reopening
+        // avio supports reconneting for http (as private avio option), but it
+        // seems somewhat broken and drops part of the stream if the first
+        // reconnect does not work. emulate it.
         close_f(s);
         s->priv = NULL;
-        stream_drop_buffers(s);
-        s->pos = 0;
-        return open_f(s);
+        int res = open_f(s);
+        if (res == STREAM_OK) {
+            if (!seek(s, s->pos)) {
+                MP_WARN(s, "Reconnecting failed.\n");
+                close_f(s);
+                s->priv = NULL;
+                return STREAM_UNSUPPORTED;
+            }
+        }
+        return res;
     }
     }
     return STREAM_UNSUPPORTED;