1 /*
2 * Gopher protocol
3 *
4 * Copyright (c) 2009 Toshimitsu Kimura
5 *
6 * based on libavformat/http.c, Copyright (c) 2000, 2001 Fabrice Bellard
7 *
8 * This file is part of FFmpeg.
9 *
10 * FFmpeg is free software; you can redistribute it and/or
11 * modify it under the terms of the GNU Lesser General Public
12 * License as published by the Free Software Foundation; either
13 * version 2.1 of the License, or (at your option) any later version.
14 *
15 * FFmpeg is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * Lesser General Public License for more details.
19 *
20 * You should have received a copy of the GNU Lesser General Public
21 * License along with FFmpeg; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 */
24
25 #include "libavutil/avstring.h"
26 #include "avformat.h"
27 #include "internal.h"
28 #include "network.h"
29 #include "url.h"
30
31 typedef struct GopherContext {
32 URLContext *hd;
33 } GopherContext;
34
gopher_write(URLContext * h,const uint8_t * buf,int size)35 static int gopher_write(URLContext *h, const uint8_t *buf, int size)
36 {
37 GopherContext *s = h->priv_data;
38 return ffurl_write(s->hd, buf, size);
39 }
40
gopher_connect(URLContext * h,const char * path)41 static int gopher_connect(URLContext *h, const char *path)
42 {
43 char buffer[1024];
44
45 if (!*path) return AVERROR(EINVAL);
46 switch (*++path) {
47 case '5':
48 case '9':
49 path = strchr(path, '/');
50 if (!path) return AVERROR(EINVAL);
51 break;
52 default:
53 av_log(h, AV_LOG_WARNING,
54 "Gopher protocol type '%c' not supported yet!\n",
55 *path);
56 return AVERROR(EINVAL);
57 }
58
59 /* send gopher sector */
60 snprintf(buffer, sizeof(buffer), "%s\r\n", path);
61
62 if (gopher_write(h, buffer, strlen(buffer)) < 0)
63 return AVERROR(EIO);
64
65 return 0;
66 }
67
gopher_close(URLContext * h)68 static int gopher_close(URLContext *h)
69 {
70 GopherContext *s = h->priv_data;
71 ffurl_closep(&s->hd);
72 return 0;
73 }
74
gopher_open(URLContext * h,const char * uri,int flags)75 static int gopher_open(URLContext *h, const char *uri, int flags)
76 {
77 GopherContext *s = h->priv_data;
78 char hostname[1024], auth[1024], path[1024], buf[1024];
79 int port, err;
80
81 h->is_streamed = 1;
82
83 /* needed in any case to build the host string */
84 av_url_split(NULL, 0, auth, sizeof(auth), hostname, sizeof(hostname), &port,
85 path, sizeof(path), uri);
86
87 if (port < 0)
88 port = 70;
89
90 ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port, NULL);
91
92 s->hd = NULL;
93 err = ffurl_open_whitelist(&s->hd, buf, AVIO_FLAG_READ_WRITE,
94 &h->interrupt_callback, NULL, h->protocol_whitelist, h->protocol_blacklist, h);
95 if (err < 0)
96 goto fail;
97
98 if ((err = gopher_connect(h, path)) < 0)
99 goto fail;
100 return 0;
101 fail:
102 gopher_close(h);
103 return err;
104 }
105
gopher_read(URLContext * h,uint8_t * buf,int size)106 static int gopher_read(URLContext *h, uint8_t *buf, int size)
107 {
108 GopherContext *s = h->priv_data;
109 int len = ffurl_read(s->hd, buf, size);
110 return len;
111 }
112
113
114 const URLProtocol ff_gopher_protocol = {
115 .name = "gopher",
116 .url_open = gopher_open,
117 .url_read = gopher_read,
118 .url_write = gopher_write,
119 .url_close = gopher_close,
120 .priv_data_size = sizeof(GopherContext),
121 .flags = URL_PROTOCOL_FLAG_NETWORK,
122 };
123