0
|
1 /*
|
|
2 * TCP protocol
|
|
3 * Copyright (c) 2002 Fabrice Bellard.
|
|
4 *
|
|
5 * This library is free software; you can redistribute it and/or
|
|
6 * modify it under the terms of the GNU Lesser General Public
|
|
7 * License as published by the Free Software Foundation; either
|
|
8 * version 2 of the License, or (at your option) any later version.
|
|
9 *
|
|
10 * This library is distributed in the hope that it will be useful,
|
|
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
13 * Lesser General Public License for more details.
|
|
14 *
|
|
15 * You should have received a copy of the GNU Lesser General Public
|
|
16 * License along with this library; if not, write to the Free Software
|
|
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
18 */
|
|
19 #include "avformat.h"
|
|
20 #include <unistd.h>
|
|
21 #include <ctype.h>
|
|
22 #include <sys/types.h>
|
|
23 #include <sys/socket.h>
|
|
24 #include <netinet/in.h>
|
|
25 #ifndef __BEOS__
|
|
26 # include <arpa/inet.h>
|
|
27 #else
|
|
28 # include "barpainet.h"
|
|
29 #endif
|
|
30 #include <netdb.h>
|
|
31
|
|
32 typedef struct TCPContext {
|
|
33 int fd;
|
|
34 } TCPContext;
|
|
35
|
|
36 /* resolve host with also IP address parsing */
|
|
37 int resolve_host(struct in_addr *sin_addr, const char *hostname)
|
|
38 {
|
|
39 struct hostent *hp;
|
|
40
|
|
41 if ((inet_aton(hostname, sin_addr)) == 0) {
|
|
42 hp = gethostbyname(hostname);
|
|
43 if (!hp)
|
|
44 return -1;
|
|
45 memcpy (sin_addr, hp->h_addr, sizeof(struct in_addr));
|
|
46 }
|
|
47 return 0;
|
|
48 }
|
|
49
|
|
50 /* return non zero if error */
|
|
51 static int tcp_open(URLContext *h, const char *uri, int flags)
|
|
52 {
|
|
53 struct sockaddr_in dest_addr;
|
|
54 char hostname[1024], *q;
|
|
55 int port, fd = -1;
|
|
56 TCPContext *s;
|
|
57 const char *p;
|
|
58
|
|
59 s = av_malloc(sizeof(TCPContext));
|
|
60 if (!s)
|
|
61 return -ENOMEM;
|
|
62 h->priv_data = s;
|
|
63 p = uri;
|
|
64 if (!strstart(p, "tcp://", &p))
|
|
65 goto fail;
|
|
66 q = hostname;
|
|
67 while (*p != ':' && *p != '/' && *p != '\0') {
|
|
68 if ((q - hostname) < sizeof(hostname) - 1)
|
|
69 *q++ = *p;
|
|
70 p++;
|
|
71 }
|
|
72 *q = '\0';
|
|
73 if (*p != ':')
|
|
74 goto fail;
|
|
75 p++;
|
|
76 port = strtoul(p, (char **)&p, 10);
|
|
77 if (port <= 0 || port >= 65536)
|
|
78 goto fail;
|
|
79
|
|
80 dest_addr.sin_family = AF_INET;
|
|
81 dest_addr.sin_port = htons(port);
|
|
82 if (resolve_host(&dest_addr.sin_addr, hostname) < 0)
|
|
83 goto fail;
|
|
84
|
|
85 fd = socket(PF_INET, SOCK_STREAM, 0);
|
|
86 if (fd < 0)
|
|
87 goto fail;
|
|
88
|
|
89 if (connect(fd, (struct sockaddr *)&dest_addr,
|
|
90 sizeof(dest_addr)) < 0)
|
|
91 goto fail;
|
|
92
|
|
93 s->fd = fd;
|
|
94 return 0;
|
|
95
|
|
96 fail:
|
|
97 if (fd >= 0)
|
|
98 close(fd);
|
|
99 av_free(s);
|
|
100 return -EIO;
|
|
101 }
|
|
102
|
|
103 static int tcp_read(URLContext *h, UINT8 *buf, int size)
|
|
104 {
|
|
105 TCPContext *s = h->priv_data;
|
|
106 int size1, len;
|
|
107
|
|
108 size1 = size;
|
|
109 while (size > 0) {
|
|
110 #ifdef CONFIG_BEOS_NETSERVER
|
|
111 len = recv (s->fd, buf, size, 0);
|
|
112 #else
|
|
113 len = read (s->fd, buf, size);
|
|
114 #endif
|
|
115 if (len < 0) {
|
|
116 if (errno != EINTR && errno != EAGAIN)
|
|
117 #ifdef __BEOS__
|
|
118 return errno;
|
|
119 #else
|
|
120 return -errno;
|
|
121 #endif
|
|
122 else
|
|
123 continue;
|
|
124 } else if (len == 0) {
|
|
125 break;
|
|
126 }
|
|
127 size -= len;
|
|
128 buf += len;
|
|
129 }
|
|
130 return size1 - size;
|
|
131 }
|
|
132
|
|
133 static int tcp_write(URLContext *h, UINT8 *buf, int size)
|
|
134 {
|
|
135 TCPContext *s = h->priv_data;
|
|
136 int ret, size1;
|
|
137
|
|
138 size1 = size;
|
|
139 while (size > 0) {
|
|
140 #ifdef CONFIG_BEOS_NETSERVER
|
|
141 ret = send (s->fd, buf, size, 0);
|
|
142 #else
|
|
143 ret = write (s->fd, buf, size);
|
|
144 #endif
|
|
145 if (ret < 0 && errno != EINTR && errno != EAGAIN)
|
|
146 #ifdef __BEOS__
|
|
147 return errno;
|
|
148 #else
|
|
149 return -errno;
|
|
150 #endif
|
|
151 size -= ret;
|
|
152 buf += ret;
|
|
153 }
|
|
154 return size1 - size;
|
|
155 }
|
|
156
|
|
157 static int tcp_close(URLContext *h)
|
|
158 {
|
|
159 TCPContext *s = h->priv_data;
|
|
160 #ifdef CONFIG_BEOS_NETSERVER
|
|
161 closesocket(s->fd);
|
|
162 #else
|
|
163 close(s->fd);
|
|
164 #endif
|
|
165 av_free(s);
|
|
166 return 0;
|
|
167 }
|
|
168
|
|
169 URLProtocol tcp_protocol = {
|
|
170 "tcp",
|
|
171 tcp_open,
|
|
172 tcp_read,
|
|
173 tcp_write,
|
|
174 NULL, /* seek */
|
|
175 tcp_close,
|
|
176 };
|