3 * Copyright (c) 2002 Fabrice Bellard.
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
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.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
24 #include "os_support.h"
27 typedef struct TCPContext
{
31 /* return non zero if error */
32 static int tcp_open(URLContext
*h
, const char *uri
, int flags
)
34 struct sockaddr_in dest_addr
;
35 char hostname
[1024], *q
;
42 char proto
[1024],path
[1024],tmp
[1024];
44 if(!ff_network_init())
47 url_split(proto
, sizeof(proto
), NULL
, 0, hostname
, sizeof(hostname
),
48 &port
, path
, sizeof(path
), uri
);
49 if (strcmp(proto
,"tcp")) goto fail
;
50 if ((q
= strchr(hostname
,'@'))) { strcpy(tmp
,q
+1); strcpy(hostname
,tmp
); }
52 if (port
<= 0 || port
>= 65536)
55 dest_addr
.sin_family
= AF_INET
;
56 dest_addr
.sin_port
= htons(port
);
57 if (resolve_host(&dest_addr
.sin_addr
, hostname
) < 0)
60 fd
= socket(AF_INET
, SOCK_STREAM
, 0);
63 ff_socket_nonblock(fd
, 1);
66 ret
= connect(fd
, (struct sockaddr
*)&dest_addr
,
69 if (ff_neterrno() == FF_NETERROR(EINTR
))
71 if (ff_neterrno() != FF_NETERROR(EINPROGRESS
) &&
72 ff_neterrno() != FF_NETERROR(EAGAIN
))
75 /* wait until we are connected or until abort */
77 if (url_interrupt_cb()) {
85 tv
.tv_usec
= 100 * 1000;
86 ret
= select(fd_max
+ 1, NULL
, &wfds
, NULL
, &tv
);
87 if (ret
> 0 && FD_ISSET(fd
, &wfds
))
93 getsockopt (fd
, SOL_SOCKET
, SO_ERROR
, &ret
, &optlen
);
97 s
= av_malloc(sizeof(TCPContext
));
99 return AVERROR(ENOMEM
);
113 static int tcp_read(URLContext
*h
, uint8_t *buf
, int size
)
115 TCPContext
*s
= h
->priv_data
;
116 int len
, fd_max
, ret
;
121 if (url_interrupt_cb())
122 return AVERROR(EINTR
);
125 FD_SET(s
->fd
, &rfds
);
127 tv
.tv_usec
= 100 * 1000;
128 ret
= select(fd_max
+ 1, &rfds
, NULL
, NULL
, &tv
);
129 if (ret
> 0 && FD_ISSET(s
->fd
, &rfds
)) {
130 len
= recv(s
->fd
, buf
, size
, 0);
132 if (ff_neterrno() != FF_NETERROR(EINTR
) &&
133 ff_neterrno() != FF_NETERROR(EAGAIN
))
134 return AVERROR(errno
);
136 } else if (ret
< 0) {
142 static int tcp_write(URLContext
*h
, uint8_t *buf
, int size
)
144 TCPContext
*s
= h
->priv_data
;
145 int ret
, size1
, fd_max
, len
;
151 if (url_interrupt_cb())
152 return AVERROR(EINTR
);
155 FD_SET(s
->fd
, &wfds
);
157 tv
.tv_usec
= 100 * 1000;
158 ret
= select(fd_max
+ 1, NULL
, &wfds
, NULL
, &tv
);
159 if (ret
> 0 && FD_ISSET(s
->fd
, &wfds
)) {
160 len
= send(s
->fd
, buf
, size
, 0);
162 if (ff_neterrno() != FF_NETERROR(EINTR
) &&
163 ff_neterrno() != FF_NETERROR(EAGAIN
))
164 return AVERROR(errno
);
169 } else if (ret
< 0) {
176 static int tcp_close(URLContext
*h
)
178 TCPContext
*s
= h
->priv_data
;
185 URLProtocol tcp_protocol
= {