ffmpeg / libavformat / gopher.c @ bc371aca
History | View | Annotate | Download (3.14 KB)
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 Libav.
|
9 |
*
|
10 |
* Libav 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 |
* Libav 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 Libav; 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 { |
32 |
URLContext *hd; |
33 |
} GopherContext; |
34 |
|
35 |
static int gopher_write(URLContext *h, const uint8_t *buf, int size) |
36 |
{ |
37 |
GopherContext *s = h->priv_data; |
38 |
return url_write(s->hd, buf, size);
|
39 |
} |
40 |
|
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(NULL, 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 |
|
68 |
static int gopher_close(URLContext *h) |
69 |
{ |
70 |
GopherContext *s = h->priv_data; |
71 |
if (s->hd) {
|
72 |
url_close(s->hd); |
73 |
s->hd = NULL;
|
74 |
} |
75 |
av_freep(&h->priv_data); |
76 |
return 0; |
77 |
} |
78 |
|
79 |
static int gopher_open(URLContext *h, const char *uri, int flags) |
80 |
{ |
81 |
GopherContext *s; |
82 |
char hostname[1024], auth[1024], path[1024], buf[1024]; |
83 |
int port, err;
|
84 |
|
85 |
h->is_streamed = 1;
|
86 |
|
87 |
s = av_malloc(sizeof(GopherContext));
|
88 |
if (!s) {
|
89 |
return AVERROR(ENOMEM);
|
90 |
} |
91 |
h->priv_data = s; |
92 |
|
93 |
/* needed in any case to build the host string */
|
94 |
av_url_split(NULL, 0, auth, sizeof(auth), hostname, sizeof(hostname), &port, |
95 |
path, sizeof(path), uri);
|
96 |
|
97 |
if (port < 0) |
98 |
port = 70;
|
99 |
|
100 |
ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port, NULL); |
101 |
|
102 |
s->hd = NULL;
|
103 |
err = ffurl_open(&s->hd, buf, URL_RDWR); |
104 |
if (err < 0) |
105 |
goto fail;
|
106 |
|
107 |
if ((err = gopher_connect(h, path)) < 0) |
108 |
goto fail;
|
109 |
return 0; |
110 |
fail:
|
111 |
gopher_close(h); |
112 |
return err;
|
113 |
} |
114 |
|
115 |
static int gopher_read(URLContext *h, uint8_t *buf, int size) |
116 |
{ |
117 |
GopherContext *s = h->priv_data; |
118 |
int len = ffurl_read(s->hd, buf, size);
|
119 |
return len;
|
120 |
} |
121 |
|
122 |
|
123 |
URLProtocol ff_gopher_protocol = { |
124 |
"gopher",
|
125 |
gopher_open, |
126 |
gopher_read, |
127 |
gopher_write, |
128 |
NULL, /*seek*/ |
129 |
gopher_close, |
130 |
}; |