Mercurial > libavformat.hg
annotate tcp.c @ 6488:96d44d29ac98 libavformat
mpegtsenc: Indent
author | thardin |
---|---|
date | Mon, 27 Sep 2010 12:27:26 +0000 |
parents | 3e7384f85f1d |
children |
rev | line source |
---|---|
0 | 1 /* |
2 * TCP protocol | |
4251
77e0c7511d41
cosmetics: Remove pointless period after copyright statement non-sentences.
diego
parents:
4206
diff
changeset
|
3 * Copyright (c) 2002 Fabrice Bellard |
0 | 4 * |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
923
diff
changeset
|
5 * This file is part of FFmpeg. |
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
923
diff
changeset
|
6 * |
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
923
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
0 | 8 * modify it under the terms of the GNU Lesser General Public |
9 * License as published by the Free Software Foundation; either | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
923
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
0 | 11 * |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
923
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
0 | 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
923
diff
changeset
|
18 * License along with FFmpeg; if not, write to the Free Software |
896
edbe5c3717f9
Update licensing information: The FSF changed postal address.
diego
parents:
885
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
0 | 20 */ |
21 #include "avformat.h" | |
22 #include <unistd.h> | |
5837
d605f589f0be
move ff_url_split() and ff_url_join() declarations to internal.h
aurel
parents:
5775
diff
changeset
|
23 #include "internal.h" |
1754 | 24 #include "network.h" |
2773 | 25 #include "os_support.h" |
4206 | 26 #if HAVE_SYS_SELECT_H |
3937
c0667b2aa3ab
include sys/select.h to get select, according to posix 2001,
bcoudurier
parents:
3765
diff
changeset
|
27 #include <sys/select.h> |
3941
53c5b89b8dff
only include sys/select.h if present, fix mingw compilation
bcoudurier
parents:
3937
diff
changeset
|
28 #endif |
180 | 29 #include <sys/time.h> |
0 | 30 |
31 typedef struct TCPContext { | |
32 int fd; | |
33 } TCPContext; | |
34 | |
35 /* return non zero if error */ | |
36 static int tcp_open(URLContext *h, const char *uri, int flags) | |
37 { | |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
38 struct addrinfo hints, *ai, *cur_ai; |
0 | 39 int port, fd = -1; |
683
095009fc2f35
kill warnings patch by (Mns Rullgrd <mru inprovide com>)
michael
parents:
511
diff
changeset
|
40 TCPContext *s = NULL; |
6479
3e7384f85f1d
tcp: Check both wfds and efds when waiting for the result from connect
mstorsjo
parents:
6378
diff
changeset
|
41 fd_set wfds, efds; |
180 | 42 int fd_max, ret; |
43 struct timeval tv; | |
44 socklen_t optlen; | |
3765
d55fb12134d1
Remove check for @ in tcp.c which removes the authorization data from the
rbultje
parents:
3756
diff
changeset
|
45 char hostname[1024],proto[1024],path[1024]; |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
46 char portstr[10]; |
511
056991ab9f10
HTTP Authentication Patch by (Petr Doubek <doubek at vision dot ee dot ethz dot ch>)
michael
parents:
482
diff
changeset
|
47 |
6182 | 48 av_url_split(proto, sizeof(proto), NULL, 0, hostname, sizeof(hostname), |
3756 | 49 &port, path, sizeof(path), uri); |
50 if (strcmp(proto,"tcp") || port <= 0 || port >= 65536) | |
3755
b9c1bcc6c1ca
On failure, return directly because the fail: case does nothing. This also
rbultje
parents:
3754
diff
changeset
|
51 return AVERROR(EINVAL); |
885 | 52 |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
53 memset(&hints, 0, sizeof(hints)); |
5595
0980c09b05c1
Fix BSD compile (PF_UNSPEC is not a standard define, AF_UNSPEC is).
rbultje
parents:
5518
diff
changeset
|
54 hints.ai_family = AF_UNSPEC; |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
55 hints.ai_socktype = SOCK_STREAM; |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
56 snprintf(portstr, sizeof(portstr), "%d", port); |
6378
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
57 ret = getaddrinfo(hostname, portstr, &hints, &ai); |
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
58 if (ret) { |
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
59 av_log(NULL, AV_LOG_ERROR, |
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
60 "Failed to resolve hostname %s: %s\n", |
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
61 hostname, gai_strerror(ret)); |
3755
b9c1bcc6c1ca
On failure, return directly because the fail: case does nothing. This also
rbultje
parents:
3754
diff
changeset
|
62 return AVERROR(EIO); |
6378
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
63 } |
0 | 64 |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
65 cur_ai = ai; |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
66 |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
67 restart: |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
68 fd = socket(cur_ai->ai_family, cur_ai->ai_socktype, cur_ai->ai_protocol); |
0 | 69 if (fd < 0) |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
70 goto fail; |
2057
857fbfeb2fa0
implement ff_socket_nonblock and use it in networking code
alex
parents:
2056
diff
changeset
|
71 ff_socket_nonblock(fd, 1); |
885 | 72 |
180 | 73 redo: |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
74 ret = connect(fd, cur_ai->ai_addr, cur_ai->ai_addrlen); |
180 | 75 if (ret < 0) { |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
2050
diff
changeset
|
76 if (ff_neterrno() == FF_NETERROR(EINTR)) |
180 | 77 goto redo; |
2321 | 78 if (ff_neterrno() != FF_NETERROR(EINPROGRESS) && |
79 ff_neterrno() != FF_NETERROR(EAGAIN)) | |
180 | 80 goto fail; |
0 | 81 |
180 | 82 /* wait until we are connected or until abort */ |
83 for(;;) { | |
84 if (url_interrupt_cb()) { | |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
85 ret = AVERROR(EINTR); |
180 | 86 goto fail1; |
87 } | |
88 fd_max = fd; | |
89 FD_ZERO(&wfds); | |
6479
3e7384f85f1d
tcp: Check both wfds and efds when waiting for the result from connect
mstorsjo
parents:
6378
diff
changeset
|
90 FD_ZERO(&efds); |
180 | 91 FD_SET(fd, &wfds); |
6479
3e7384f85f1d
tcp: Check both wfds and efds when waiting for the result from connect
mstorsjo
parents:
6378
diff
changeset
|
92 FD_SET(fd, &efds); |
180 | 93 tv.tv_sec = 0; |
94 tv.tv_usec = 100 * 1000; | |
6479
3e7384f85f1d
tcp: Check both wfds and efds when waiting for the result from connect
mstorsjo
parents:
6378
diff
changeset
|
95 ret = select(fd_max + 1, NULL, &wfds, &efds, &tv); |
3e7384f85f1d
tcp: Check both wfds and efds when waiting for the result from connect
mstorsjo
parents:
6378
diff
changeset
|
96 if (ret > 0 && (FD_ISSET(fd, &wfds) || FD_ISSET(fd, &efds))) |
180 | 97 break; |
98 } | |
885 | 99 |
180 | 100 /* test error */ |
101 optlen = sizeof(ret); | |
102 getsockopt (fd, SOL_SOCKET, SO_ERROR, &ret, &optlen); | |
6378
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
103 if (ret != 0) { |
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
104 av_log(NULL, AV_LOG_ERROR, |
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
105 "TCP connection to %s:%d failed: %s\n", |
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
106 hostname, port, strerror(ret)); |
180 | 107 goto fail; |
6378
a5b3f5d68db6
Print error messages in case of connection failure or name resolution failure
rbultje
parents:
6182
diff
changeset
|
108 } |
180 | 109 } |
3754
8d267b43eaba
Move malloc() down until after all initializations, so that the resource is
rbultje
parents:
3753
diff
changeset
|
110 s = av_malloc(sizeof(TCPContext)); |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
111 if (!s) { |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
112 freeaddrinfo(ai); |
3754
8d267b43eaba
Move malloc() down until after all initializations, so that the resource is
rbultje
parents:
3753
diff
changeset
|
113 return AVERROR(ENOMEM); |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
114 } |
3754
8d267b43eaba
Move malloc() down until after all initializations, so that the resource is
rbultje
parents:
3753
diff
changeset
|
115 h->priv_data = s; |
8d267b43eaba
Move malloc() down until after all initializations, so that the resource is
rbultje
parents:
3753
diff
changeset
|
116 h->is_streamed = 1; |
0 | 117 s->fd = fd; |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
118 freeaddrinfo(ai); |
0 | 119 return 0; |
120 | |
121 fail: | |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
122 if (cur_ai->ai_next) { |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
123 /* Retry with the next sockaddr */ |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
124 cur_ai = cur_ai->ai_next; |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
125 if (fd >= 0) |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
126 closesocket(fd); |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
127 goto restart; |
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
128 } |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2057
diff
changeset
|
129 ret = AVERROR(EIO); |
180 | 130 fail1: |
0 | 131 if (fd >= 0) |
1670 | 132 closesocket(fd); |
5518
0dc71c7f6653
Use getaddrinfo() instead of resolve_host(). Patch by Martin Storsj
rbultje
parents:
5102
diff
changeset
|
133 freeaddrinfo(ai); |
180 | 134 return ret; |
0 | 135 } |
136 | |
65 | 137 static int tcp_read(URLContext *h, uint8_t *buf, int size) |
0 | 138 { |
139 TCPContext *s = h->priv_data; | |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
140 int len, fd_max, ret; |
180 | 141 fd_set rfds; |
142 struct timeval tv; | |
0 | 143 |
385
2f56d366a787
no read loop tcp/http and http CRLF fix by (Leon van Stuivenberg <l dot vanstuivenberg at chello dot nl>)
michael
parents:
261
diff
changeset
|
144 for (;;) { |
180 | 145 if (url_interrupt_cb()) |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
146 return AVERROR(EINTR); |
180 | 147 fd_max = s->fd; |
148 FD_ZERO(&rfds); | |
149 FD_SET(s->fd, &rfds); | |
150 tv.tv_sec = 0; | |
151 tv.tv_usec = 100 * 1000; | |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
152 ret = select(fd_max + 1, &rfds, NULL, NULL, &tv); |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
153 if (ret > 0 && FD_ISSET(s->fd, &rfds)) { |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
154 len = recv(s->fd, buf, size, 0); |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
155 if (len < 0) { |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
2050
diff
changeset
|
156 if (ff_neterrno() != FF_NETERROR(EINTR) && |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
2050
diff
changeset
|
157 ff_neterrno() != FF_NETERROR(EAGAIN)) |
5102 | 158 return AVERROR(ff_neterrno()); |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
159 } else return len; |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
160 } else if (ret < 0) { |
5896
395592984ef0
Don't report EINTR from select as an error, retry select instead
mstorsjo
parents:
5837
diff
changeset
|
161 if (ff_neterrno() == FF_NETERROR(EINTR)) |
395592984ef0
Don't report EINTR from select as an error, retry select instead
mstorsjo
parents:
5837
diff
changeset
|
162 continue; |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
163 return -1; |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
164 } |
0 | 165 } |
166 } | |
167 | |
6068 | 168 static int tcp_write(URLContext *h, const uint8_t *buf, int size) |
0 | 169 { |
170 TCPContext *s = h->priv_data; | |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
171 int ret, size1, fd_max, len; |
180 | 172 fd_set wfds; |
173 struct timeval tv; | |
0 | 174 |
175 size1 = size; | |
176 while (size > 0) { | |
180 | 177 if (url_interrupt_cb()) |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
178 return AVERROR(EINTR); |
180 | 179 fd_max = s->fd; |
180 FD_ZERO(&wfds); | |
181 FD_SET(s->fd, &wfds); | |
182 tv.tv_sec = 0; | |
183 tv.tv_usec = 100 * 1000; | |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
184 ret = select(fd_max + 1, NULL, &wfds, NULL, &tv); |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
185 if (ret > 0 && FD_ISSET(s->fd, &wfds)) { |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
186 len = send(s->fd, buf, size, 0); |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
187 if (len < 0) { |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
2050
diff
changeset
|
188 if (ff_neterrno() != FF_NETERROR(EINTR) && |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
2050
diff
changeset
|
189 ff_neterrno() != FF_NETERROR(EAGAIN)) |
5102 | 190 return AVERROR(ff_neterrno()); |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
191 continue; |
261
5f27f90ed496
Fix a very nasty problem with extra bytes appearing in TCP data streams.
philipjsg
parents:
229
diff
changeset
|
192 } |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
193 size -= len; |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
194 buf += len; |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
195 } else if (ret < 0) { |
5896
395592984ef0
Don't report EINTR from select as an error, retry select instead
mstorsjo
parents:
5837
diff
changeset
|
196 if (ff_neterrno() == FF_NETERROR(EINTR)) |
395592984ef0
Don't report EINTR from select as an error, retry select instead
mstorsjo
parents:
5837
diff
changeset
|
197 continue; |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
198 return -1; |
261
5f27f90ed496
Fix a very nasty problem with extra bytes appearing in TCP data streams.
philipjsg
parents:
229
diff
changeset
|
199 } |
0 | 200 } |
201 return size1 - size; | |
202 } | |
203 | |
204 static int tcp_close(URLContext *h) | |
205 { | |
206 TCPContext *s = h->priv_data; | |
207 closesocket(s->fd); | |
208 av_free(s); | |
209 return 0; | |
210 } | |
211 | |
4640
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4251
diff
changeset
|
212 static int tcp_get_file_handle(URLContext *h) |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4251
diff
changeset
|
213 { |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4251
diff
changeset
|
214 TCPContext *s = h->priv_data; |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4251
diff
changeset
|
215 return s->fd; |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4251
diff
changeset
|
216 } |
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4251
diff
changeset
|
217 |
0 | 218 URLProtocol tcp_protocol = { |
219 "tcp", | |
220 tcp_open, | |
221 tcp_read, | |
222 tcp_write, | |
223 NULL, /* seek */ | |
224 tcp_close, | |
4640
b34d9614b887
Add url_get_file_handle(), which is used to get the file descriptor
rbultje
parents:
4251
diff
changeset
|
225 .url_get_file_handle = tcp_get_file_handle, |
0 | 226 }; |