From ba99cfc2130025cd63a0ad767a1ee9293c2936f1 Mon Sep 17 00:00:00 2001
From: "Ronald S. Bultje" <rsbultje@gmail.com>
Date: Tue, 19 Aug 2008 23:44:23 +0000
Subject: [PATCH] Implement url_open_protocol(), which is basiclly the former
 url_open() but which opens a URLProtocol instead of a filename. url_open() is
 reimplemented to call url_open_protocol(). See discussion on
 "url_open_protocol" on ffmpeg-devel.

Originally committed as revision 14857 to svn://svn.ffmpeg.org/ffmpeg/trunk
---
 libavformat/avio.c | 69 +++++++++++++++++++++++++---------------------
 libavformat/avio.h |  2 ++
 2 files changed, 39 insertions(+), 32 deletions(-)

diff --git a/libavformat/avio.c b/libavformat/avio.c
index 743cc88e98..1278827802 100644
--- a/libavformat/avio.c
+++ b/libavformat/avio.c
@@ -59,41 +59,12 @@ int register_protocol(URLProtocol *protocol)
     return 0;
 }
 
-int url_open(URLContext **puc, const char *filename, int flags)
+int url_open_protocol (URLContext **puc, struct URLProtocol *up,
+                       const char *filename, int flags)
 {
     URLContext *uc;
-    URLProtocol *up;
-    const char *p;
-    char proto_str[128], *q;
     int err;
 
-    p = filename;
-    q = proto_str;
-    while (*p != '\0' && *p != ':') {
-        /* protocols can only contain alphabetic chars */
-        if (!isalpha(*p))
-            goto file_proto;
-        if ((q - proto_str) < sizeof(proto_str) - 1)
-            *q++ = *p;
-        p++;
-    }
-    /* if the protocol has length 1, we consider it is a dos drive */
-    if (*p == '\0' || (q - proto_str) <= 1) {
-    file_proto:
-        strcpy(proto_str, "file");
-    } else {
-        *q = '\0';
-    }
-
-    up = first_protocol;
-    while (up != NULL) {
-        if (!strcmp(proto_str, up->name))
-            goto found;
-        up = up->next;
-    }
-    err = AVERROR(ENOENT);
-    goto fail;
- found:
     uc = av_malloc(sizeof(URLContext) + strlen(filename) + 1);
     if (!uc) {
         err = AVERROR(ENOMEM);
@@ -117,7 +88,7 @@ int url_open(URLContext **puc, const char *filename, int flags)
 
     //We must be carefull here as url_seek() could be slow, for example for http
     if(   (flags & (URL_WRONLY | URL_RDWR))
-       || !strcmp(proto_str, "file"))
+       || !strcmp(up->name, "file"))
         if(!uc->is_streamed && url_seek(uc, 0, SEEK_SET) < 0)
             uc->is_streamed= 1;
     *puc = uc;
@@ -127,6 +98,40 @@ int url_open(URLContext **puc, const char *filename, int flags)
     return err;
 }
 
+int url_open(URLContext **puc, const char *filename, int flags)
+{
+    URLProtocol *up;
+    const char *p;
+    char proto_str[128], *q;
+
+    p = filename;
+    q = proto_str;
+    while (*p != '\0' && *p != ':') {
+        /* protocols can only contain alphabetic chars */
+        if (!isalpha(*p))
+            goto file_proto;
+        if ((q - proto_str) < sizeof(proto_str) - 1)
+            *q++ = *p;
+        p++;
+    }
+    /* if the protocol has length 1, we consider it is a dos drive */
+    if (*p == '\0' || (q - proto_str) <= 1) {
+    file_proto:
+        strcpy(proto_str, "file");
+    } else {
+        *q = '\0';
+    }
+
+    up = first_protocol;
+    while (up != NULL) {
+        if (!strcmp(proto_str, up->name))
+            return url_open_protocol (puc, up, filename, flags);
+        up = up->next;
+    }
+    *puc = NULL;
+    return AVERROR(ENOENT);
+}
+
 int url_read(URLContext *h, unsigned char *buf, int size)
 {
     int ret;
diff --git a/libavformat/avio.h b/libavformat/avio.h
index 2d2e26999d..8137cb54dc 100644
--- a/libavformat/avio.h
+++ b/libavformat/avio.h
@@ -62,6 +62,8 @@ typedef struct URLPollEntry {
 
 typedef int URLInterruptCB(void);
 
+int url_open_protocol (URLContext **puc, struct URLProtocol *up,
+                       const char *filename, int flags);
 int url_open(URLContext **h, const char *filename, int flags);
 int url_read(URLContext *h, unsigned char *buf, int size);
 int url_write(URLContext *h, unsigned char *buf, int size);