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
22 #include "libavutil/parseutils.h"
23 #include "libavutil/opt.h"
24 #include "libavutil/time.h"
27 #include "os_support.h"
33 typedef struct TCPContext
{
42 #define OFFSET(x) offsetof(TCPContext, x)
43 #define D AV_OPT_FLAG_DECODING_PARAM
44 #define E AV_OPT_FLAG_ENCODING_PARAM
45 static const AVOption options
[] = {
46 {"listen", "listen on port instead of connecting", OFFSET(listen
), AV_OPT_TYPE_INT
, {.i64
= 0}, 0, 1, D
|E
},
47 {"timeout", "set timeout of socket I/O operations", OFFSET(rw_timeout
), AV_OPT_TYPE_INT
, {.i64
= -1}, -1, INT_MAX
, D
|E
},
48 {"listen_timeout", "set connection awaiting timeout", OFFSET(listen_timeout
), AV_OPT_TYPE_INT
, {.i64
= -1}, -1, INT_MAX
, D
|E
},
52 static const AVClass tcp_context_class
= {
54 .item_name
= av_default_item_name
,
56 .version
= LIBAVUTIL_VERSION_INT
,
59 /* return non zero if error */
60 static int tcp_open(URLContext
*h
, const char *uri
, int flags
)
62 struct addrinfo hints
= { 0 }, *ai
, *cur_ai
;
64 TCPContext
*s
= h
->priv_data
;
68 char hostname
[1024],proto
[1024],path
[1024];
70 s
->open_timeout
= 5000000;
72 av_url_split(proto
, sizeof(proto
), NULL
, 0, hostname
, sizeof(hostname
),
73 &port
, path
, sizeof(path
), uri
);
74 if (strcmp(proto
, "tcp"))
75 return AVERROR(EINVAL
);
76 if (port
<= 0 || port
>= 65536) {
77 av_log(h
, AV_LOG_ERROR
, "Port missing in uri\n");
78 return AVERROR(EINVAL
);
82 if (av_find_info_tag(buf
, sizeof(buf
), "listen", p
)) {
84 s
->listen
= strtol(buf
, &endptr
, 10);
85 /* assume if no digits were found it is a request to enable it */
89 if (av_find_info_tag(buf
, sizeof(buf
), "timeout", p
)) {
90 s
->rw_timeout
= strtol(buf
, NULL
, 10);
92 if (av_find_info_tag(buf
, sizeof(buf
), "listen_timeout", p
)) {
93 s
->listen_timeout
= strtol(buf
, NULL
, 10);
96 if (s
->rw_timeout
>= 0) {
98 h
->rw_timeout
= s
->rw_timeout
;
100 hints
.ai_family
= AF_UNSPEC
;
101 hints
.ai_socktype
= SOCK_STREAM
;
102 snprintf(portstr
, sizeof(portstr
), "%d", port
);
104 hints
.ai_flags
|= AI_PASSIVE
;
106 ret
= getaddrinfo(NULL
, portstr
, &hints
, &ai
);
108 ret
= getaddrinfo(hostname
, portstr
, &hints
, &ai
);
110 av_log(h
, AV_LOG_ERROR
,
111 "Failed to resolve hostname %s: %s\n",
112 hostname
, gai_strerror(ret
));
119 fd
= ff_socket(cur_ai
->ai_family
,
121 cur_ai
->ai_protocol
);
128 if ((fd
= ff_listen_bind(fd
, cur_ai
->ai_addr
, cur_ai
->ai_addrlen
,
129 s
->listen_timeout
, h
)) < 0) {
134 if ((ret
= ff_listen_connect(fd
, cur_ai
->ai_addr
, cur_ai
->ai_addrlen
,
135 s
->open_timeout
/ 1000, h
, !!cur_ai
->ai_next
)) < 0) {
137 if (ret
== AVERROR_EXIT
)
150 if (cur_ai
->ai_next
) {
151 /* Retry with the next sockaddr */
152 cur_ai
= cur_ai
->ai_next
;
165 static int tcp_read(URLContext
*h
, uint8_t *buf
, int size
)
167 TCPContext
*s
= h
->priv_data
;
170 if (!(h
->flags
& AVIO_FLAG_NONBLOCK
)) {
171 ret
= ff_network_wait_fd_timeout(s
->fd
, 0, h
->rw_timeout
, &h
->interrupt_callback
);
175 ret
= recv(s
->fd
, buf
, size
, 0);
176 return ret
< 0 ? ff_neterrno() : ret
;
179 static int tcp_write(URLContext
*h
, const uint8_t *buf
, int size
)
181 TCPContext
*s
= h
->priv_data
;
184 if (!(h
->flags
& AVIO_FLAG_NONBLOCK
)) {
185 ret
= ff_network_wait_fd_timeout(s
->fd
, 1, h
->rw_timeout
, &h
->interrupt_callback
);
189 ret
= send(s
->fd
, buf
, size
, MSG_NOSIGNAL
);
190 return ret
< 0 ? ff_neterrno() : ret
;
193 static int tcp_shutdown(URLContext
*h
, int flags
)
195 TCPContext
*s
= h
->priv_data
;
198 if (flags
& AVIO_FLAG_WRITE
&& flags
& AVIO_FLAG_READ
) {
200 } else if (flags
& AVIO_FLAG_WRITE
) {
206 return shutdown(s
->fd
, how
);
209 static int tcp_close(URLContext
*h
)
211 TCPContext
*s
= h
->priv_data
;
216 static int tcp_get_file_handle(URLContext
*h
)
218 TCPContext
*s
= h
->priv_data
;
222 URLProtocol ff_tcp_protocol
= {
224 .url_open
= tcp_open
,
225 .url_read
= tcp_read
,
226 .url_write
= tcp_write
,
227 .url_close
= tcp_close
,
228 .url_get_file_handle
= tcp_get_file_handle
,
229 .url_shutdown
= tcp_shutdown
,
230 .priv_data_size
= sizeof(TCPContext
),
231 .priv_data_class
= &tcp_context_class
,
232 .flags
= URL_PROTOCOL_FLAG_NETWORK
,