From 62f63b24bfec22134bc5b9312bc6afe824e63199 Mon Sep 17 00:00:00 2001
From: "Jeyapal, Karthick" <kjeyapal@akamai.com>
Date: Wed, 29 Nov 2017 14:31:34 +0800
Subject: [PATCH] libavformat/avio: Utility function to return URLContext

---
 libavformat/avio_internal.h |  8 ++++++++
 libavformat/aviobuf.c       | 13 +++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/libavformat/avio_internal.h b/libavformat/avio_internal.h
index c01835df96..04c1ad5157 100644
--- a/libavformat/avio_internal.h
+++ b/libavformat/avio_internal.h
@@ -132,6 +132,14 @@ int ffio_open_dyn_packet_buf(AVIOContext **s, int max_packet_size);
  */
 int ffio_fdopen(AVIOContext **s, URLContext *h);
 
+/**
+ * Return the URLContext associated with the AVIOContext
+ *
+ * @param s IO context
+ * @return pointer to URLContext or NULL.
+ */
+URLContext *ffio_geturlcontext(AVIOContext *s);
+
 /**
  * Open a write-only fake memory stream. The written data is not stored
  * anywhere - this is only used for measuring the amount of data
diff --git a/libavformat/aviobuf.c b/libavformat/aviobuf.c
index 3b4c8439d6..86eb6579f4 100644
--- a/libavformat/aviobuf.c
+++ b/libavformat/aviobuf.c
@@ -980,6 +980,19 @@ fail:
     return AVERROR(ENOMEM);
 }
 
+URLContext* ffio_geturlcontext(AVIOContext *s)
+{
+    AVIOInternal *internal;
+    if (!s)
+        return NULL;
+
+    internal = s->opaque;
+    if (internal && s->read_packet == io_read_packet)
+        return internal->h;
+    else
+        return NULL;
+}
+
 int ffio_ensure_seekback(AVIOContext *s, int64_t buf_size)
 {
     uint8_t *buffer;