Gopher protocol, patch by Toshimitsu Kimura, lovesyao gmail com

Originally committed as revision 17159 to svn://svn.ffmpeg.org/ffmpeg/trunk

5 files changed, 132 insertions, 1 deletions

diff --git a/Changelog b/Changelog
index 977c983494..7ad2140db7 100644
--- a/Changelog
+++ b/Changelog
@@ -149,6 +149,7 @@ version <next>
 - Electronic Arts TQI decoder
 - OpenJPEG based JPEG 2000 decoder
 - NC (NC4600) cameras file demuxer
+- Gopher client support
 
 version 0.4.9-pre1:
 
diff --git a/libavformat/Makefile b/libavformat/Makefile
index bc2d9dac67..e79860c937 100644
--- a/libavformat/Makefile
+++ b/libavformat/Makefile
@@ -208,6 +208,7 @@ OBJS-$(CONFIG_VHOOK)                     += framehook.o
 OBJS+= avio.o aviobuf.o
 
 OBJS-$(CONFIG_FILE_PROTOCOL)             += file.o
+OBJS-$(CONFIG_GOPHER_PROTOCOL)           += gopher.o
 OBJS-$(CONFIG_HTTP_PROTOCOL)             += http.o
 OBJS-$(CONFIG_PIPE_PROTOCOL)             += file.o
 OBJS-$(CONFIG_RTP_PROTOCOL)              += rtpproto.o
diff --git a/libavformat/allformats.c b/libavformat/allformats.c
index 8dcbd55477..181415a156 100644
--- a/libavformat/allformats.c
+++ b/libavformat/allformats.c
@@ -199,6 +199,7 @@ void av_register_all(void)
 
     /* protocols */
     REGISTER_PROTOCOL (FILE, file);
+    REGISTER_PROTOCOL (GOPHER, gopher);
     REGISTER_PROTOCOL (HTTP, http);
     REGISTER_PROTOCOL (PIPE, pipe);
     REGISTER_PROTOCOL (RTP, rtp);
diff --git a/libavformat/avformat.h b/libavformat/avformat.h
index 06bba43570..a1a9b601ef 100644
--- a/libavformat/avformat.h
+++ b/libavformat/avformat.h
@@ -22,7 +22,7 @@
 #define AVFORMAT_AVFORMAT_H
 
 #define LIBAVFORMAT_VERSION_MAJOR 52
-#define LIBAVFORMAT_VERSION_MINOR 26
+#define LIBAVFORMAT_VERSION_MINOR 27
 #define LIBAVFORMAT_VERSION_MICRO  0
 
 #define LIBAVFORMAT_VERSION_INT AV_VERSION_INT(LIBAVFORMAT_VERSION_MAJOR, \
diff --git a/libavformat/gopher.c b/libavformat/gopher.c
new file mode 100644
index 0000000000..abd1cc394a
--- /dev/null
+++ b/libavformat/gopher.c
@@ -0,0 +1,128 @@
+/*
+ * Gopher protocol
+ *
+ * Copyright (c) 2009 Toshimitsu Kimura
+ *
+ * based on libavformat/http.c, Copyright (c) 2000, 2001 Fabrice Bellard
+ *
+ * This file is part of FFmpeg.
+ *
+ * FFmpeg is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * FFmpeg is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with FFmpeg; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+
+#include "libavutil/avstring.h"
+#include "avformat.h"
+#include "network.h"
+
+typedef struct {
+    URLContext *hd;
+} GopherContext;
+
+static int gopher_write(URLContext *h, uint8_t *buf, int size)
+{
+    GopherContext *s = h->priv_data;
+    return url_write(s->hd, buf, size);
+}
+
+static int gopher_connect(URLContext *h, const char *path)
+{
+    char buffer[1024];
+
+    if (!*path) return AVERROR(EINVAL);
+    switch (*++path) {
+        case '5':
+        case '9':
+            path = strchr(path, '/');
+            if (!path) return AVERROR(EINVAL);
+            break;
+        default:
+            av_log(NULL, AV_LOG_WARNING,
+                   "Gopher protocol type '%c' not supported yet!\n",
+                   *path);
+            return AVERROR(EINVAL);
+    }
+
+    /* send gopher sector */
+    snprintf(buffer, sizeof(buffer), "%s\r\n", path);
+
+    if (gopher_write(h, buffer, strlen(buffer)) < 0)
+        return AVERROR(EIO);
+
+    return 0;
+}
+
+static int gopher_close(URLContext *h)
+{
+    GopherContext *s = h->priv_data;
+    if (s->hd) {
+        url_close(s->hd);
+        s->hd = NULL;
+    }
+    av_freep(&h->priv_data);
+    return 0;
+}
+
+static int gopher_open(URLContext *h, const char *uri, int flags)
+{
+    GopherContext *s;
+    char hostname[1024], auth[1024], path[1024], buf[1024];
+    int port, err;
+
+    h->is_streamed = 1;
+
+    s = av_malloc(sizeof(GopherContext));
+    if (!s) {
+        return AVERROR(ENOMEM);
+    }
+    h->priv_data = s;
+
+    /* needed in any case to build the host string */
+    url_split(NULL, 0, auth, sizeof(auth), hostname, sizeof(hostname), &port,
+              path, sizeof(path), uri);
+
+    if (port < 0)
+        port = 70;
+
+    snprintf(buf, sizeof(buf), "tcp://%s:%d", hostname, port);
+
+    s->hd = NULL;
+    err = url_open(&s->hd, buf, URL_RDWR);
+    if (err < 0)
+        goto fail;
+
+    if ((err = gopher_connect(h, path)) < 0)
+        goto fail;
+    return 0;
+ fail:
+    gopher_close(h);
+    return err;
+}
+
+static int gopher_read(URLContext *h, uint8_t *buf, int size)
+{
+    GopherContext *s = h->priv_data;
+    int len = url_read(s->hd, buf, size);
+    return len;
+}
+
+
+URLProtocol gopher_protocol = {
+    "gopher",
+    gopher_open,
+    gopher_read,
+    gopher_write,
+    NULL, /*seek*/
+    gopher_close,
+};