Mercurial > libavformat.hg
annotate tcp.c @ 3890:76e7b6822fa6 libavformat
cosmetics, remove whitespace
author | bcoudurier |
---|---|
date | Thu, 04 Sep 2008 18:26:31 +0000 |
parents | d55fb12134d1 |
children | c0667b2aa3ab |
rev | line source |
---|---|
0 | 1 /* |
2 * TCP protocol | |
3 * Copyright (c) 2002 Fabrice Bellard. | |
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> | |
1754 | 23 #include "network.h" |
2773 | 24 #include "os_support.h" |
180 | 25 #include <sys/time.h> |
0 | 26 |
27 typedef struct TCPContext { | |
28 int fd; | |
29 } TCPContext; | |
30 | |
31 /* return non zero if error */ | |
32 static int tcp_open(URLContext *h, const char *uri, int flags) | |
33 { | |
34 struct sockaddr_in dest_addr; | |
35 int port, fd = -1; | |
683
095009fc2f35
kill warnings patch by (Mns Rullgrd <mru inprovide com>)
michael
parents:
511
diff
changeset
|
36 TCPContext *s = NULL; |
180 | 37 fd_set wfds; |
38 int fd_max, ret; | |
39 struct timeval tv; | |
40 socklen_t optlen; | |
3765
d55fb12134d1
Remove check for @ in tcp.c which removes the authorization data from the
rbultje
parents:
3756
diff
changeset
|
41 char hostname[1024],proto[1024],path[1024]; |
511
056991ab9f10
HTTP Authentication Patch by (Petr Doubek <doubek at vision dot ee dot ethz dot ch>)
michael
parents:
482
diff
changeset
|
42 |
3753
42ed3629601d
Fix memleak on some OSes in case network initialization fails. See
rbultje
parents:
3752
diff
changeset
|
43 if(!ff_network_init()) |
42ed3629601d
Fix memleak on some OSes in case network initialization fails. See
rbultje
parents:
3752
diff
changeset
|
44 return AVERROR(EIO); |
42ed3629601d
Fix memleak on some OSes in case network initialization fails. See
rbultje
parents:
3752
diff
changeset
|
45 |
511
056991ab9f10
HTTP Authentication Patch by (Petr Doubek <doubek at vision dot ee dot ethz dot ch>)
michael
parents:
482
diff
changeset
|
46 url_split(proto, sizeof(proto), NULL, 0, hostname, sizeof(hostname), |
3756 | 47 &port, path, sizeof(path), uri); |
48 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
|
49 return AVERROR(EINVAL); |
885 | 50 |
0 | 51 dest_addr.sin_family = AF_INET; |
52 dest_addr.sin_port = htons(port); | |
53 if (resolve_host(&dest_addr.sin_addr, hostname) < 0) | |
3755
b9c1bcc6c1ca
On failure, return directly because the fail: case does nothing. This also
rbultje
parents:
3754
diff
changeset
|
54 return AVERROR(EIO); |
0 | 55 |
1810
d85795da84ab
change PF_INET to AF_INET to be consistent in the whole project. PF_INET is deprecated, while AF_INET is referred by the POSIX standards
alex
parents:
1787
diff
changeset
|
56 fd = socket(AF_INET, SOCK_STREAM, 0); |
0 | 57 if (fd < 0) |
3755
b9c1bcc6c1ca
On failure, return directly because the fail: case does nothing. This also
rbultje
parents:
3754
diff
changeset
|
58 return AVERROR(EIO); |
2057
857fbfeb2fa0
implement ff_socket_nonblock and use it in networking code
alex
parents:
2056
diff
changeset
|
59 ff_socket_nonblock(fd, 1); |
885 | 60 |
180 | 61 redo: |
885 | 62 ret = connect(fd, (struct sockaddr *)&dest_addr, |
180 | 63 sizeof(dest_addr)); |
64 if (ret < 0) { | |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
2050
diff
changeset
|
65 if (ff_neterrno() == FF_NETERROR(EINTR)) |
180 | 66 goto redo; |
2321 | 67 if (ff_neterrno() != FF_NETERROR(EINPROGRESS) && |
68 ff_neterrno() != FF_NETERROR(EAGAIN)) | |
180 | 69 goto fail; |
0 | 70 |
180 | 71 /* wait until we are connected or until abort */ |
72 for(;;) { | |
73 if (url_interrupt_cb()) { | |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
74 ret = AVERROR(EINTR); |
180 | 75 goto fail1; |
76 } | |
77 fd_max = fd; | |
78 FD_ZERO(&wfds); | |
79 FD_SET(fd, &wfds); | |
80 tv.tv_sec = 0; | |
81 tv.tv_usec = 100 * 1000; | |
82 ret = select(fd_max + 1, NULL, &wfds, NULL, &tv); | |
83 if (ret > 0 && FD_ISSET(fd, &wfds)) | |
84 break; | |
85 } | |
885 | 86 |
180 | 87 /* test error */ |
88 optlen = sizeof(ret); | |
89 getsockopt (fd, SOL_SOCKET, SO_ERROR, &ret, &optlen); | |
90 if (ret != 0) | |
91 goto fail; | |
92 } | |
3754
8d267b43eaba
Move malloc() down until after all initializations, so that the resource is
rbultje
parents:
3753
diff
changeset
|
93 s = av_malloc(sizeof(TCPContext)); |
8d267b43eaba
Move malloc() down until after all initializations, so that the resource is
rbultje
parents:
3753
diff
changeset
|
94 if (!s) |
8d267b43eaba
Move malloc() down until after all initializations, so that the resource is
rbultje
parents:
3753
diff
changeset
|
95 return AVERROR(ENOMEM); |
8d267b43eaba
Move malloc() down until after all initializations, so that the resource is
rbultje
parents:
3753
diff
changeset
|
96 h->priv_data = s; |
8d267b43eaba
Move malloc() down until after all initializations, so that the resource is
rbultje
parents:
3753
diff
changeset
|
97 h->is_streamed = 1; |
0 | 98 s->fd = fd; |
99 return 0; | |
100 | |
101 fail: | |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2057
diff
changeset
|
102 ret = AVERROR(EIO); |
180 | 103 fail1: |
0 | 104 if (fd >= 0) |
1670 | 105 closesocket(fd); |
180 | 106 return ret; |
0 | 107 } |
108 | |
65 | 109 static int tcp_read(URLContext *h, uint8_t *buf, int size) |
0 | 110 { |
111 TCPContext *s = h->priv_data; | |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
112 int len, fd_max, ret; |
180 | 113 fd_set rfds; |
114 struct timeval tv; | |
0 | 115 |
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
|
116 for (;;) { |
180 | 117 if (url_interrupt_cb()) |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
118 return AVERROR(EINTR); |
180 | 119 fd_max = s->fd; |
120 FD_ZERO(&rfds); | |
121 FD_SET(s->fd, &rfds); | |
122 tv.tv_sec = 0; | |
123 tv.tv_usec = 100 * 1000; | |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
124 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
|
125 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
|
126 len = recv(s->fd, buf, size, 0); |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
127 if (len < 0) { |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
2050
diff
changeset
|
128 if (ff_neterrno() != FF_NETERROR(EINTR) && |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
2050
diff
changeset
|
129 ff_neterrno() != FF_NETERROR(EAGAIN)) |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
130 return AVERROR(errno); |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
131 } else return len; |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
132 } else if (ret < 0) { |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
133 return -1; |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
134 } |
0 | 135 } |
136 } | |
137 | |
65 | 138 static int tcp_write(URLContext *h, uint8_t *buf, int size) |
0 | 139 { |
140 TCPContext *s = h->priv_data; | |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
141 int ret, size1, fd_max, len; |
180 | 142 fd_set wfds; |
143 struct timeval tv; | |
0 | 144 |
145 size1 = size; | |
146 while (size > 0) { | |
180 | 147 if (url_interrupt_cb()) |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
148 return AVERROR(EINTR); |
180 | 149 fd_max = s->fd; |
150 FD_ZERO(&wfds); | |
151 FD_SET(s->fd, &wfds); | |
152 tv.tv_sec = 0; | |
153 tv.tv_usec = 100 * 1000; | |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
154 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
|
155 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
|
156 len = send(s->fd, buf, size, 0); |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
157 if (len < 0) { |
2056
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
2050
diff
changeset
|
158 if (ff_neterrno() != FF_NETERROR(EINTR) && |
eeea52739ff3
use ff_neterrno() and FF_NETERROR() for networking error handling
alex
parents:
2050
diff
changeset
|
159 ff_neterrno() != FF_NETERROR(EAGAIN)) |
1787
eb16c64144ee
This fixes error handling for BeOS, removing the need for some ifdefs.
mmu_man
parents:
1754
diff
changeset
|
160 return AVERROR(errno); |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
161 continue; |
261
5f27f90ed496
Fix a very nasty problem with extra bytes appearing in TCP data streams.
philipjsg
parents:
229
diff
changeset
|
162 } |
388
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
163 size -= len; |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
164 buf += len; |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
165 } else if (ret < 0) { |
9af30d452a0a
tcp select() check and enables pressing 'q' when reading/(writing) from
michael
parents:
385
diff
changeset
|
166 return -1; |
261
5f27f90ed496
Fix a very nasty problem with extra bytes appearing in TCP data streams.
philipjsg
parents:
229
diff
changeset
|
167 } |
0 | 168 } |
169 return size1 - size; | |
170 } | |
171 | |
172 static int tcp_close(URLContext *h) | |
173 { | |
174 TCPContext *s = h->priv_data; | |
175 closesocket(s->fd); | |
2351
b9a881c0967e
Add initialization and cleanup functions for Winsock
ramiro
parents:
2321
diff
changeset
|
176 ff_network_close(); |
0 | 177 av_free(s); |
178 return 0; | |
179 } | |
180 | |
181 URLProtocol tcp_protocol = { | |
182 "tcp", | |
183 tcp_open, | |
184 tcp_read, | |
185 tcp_write, | |
186 NULL, /* seek */ | |
187 tcp_close, | |
188 }; |