git.sesse.net Git - ffmpeg/blob - libavformat/network.h

   1 /*
   2  * Copyright (c) 2007 The FFmpeg Project
   3  *
   4  * This file is part of FFmpeg.
   5  *
   6  * FFmpeg is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * FFmpeg is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with FFmpeg; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 #ifndef AVFORMAT_NETWORK_H
  22 #define AVFORMAT_NETWORK_H
  23
  24 #include <errno.h>
  25 #include <stdint.h>
  26
  27 #include "config.h"
  28 #include "libavutil/error.h"
  29 #include "os_support.h"
  30 #include "avio.h"
  31 #include "url.h"
  32
  33 #if HAVE_UNISTD_H
  34 #include <unistd.h>
  35 #endif
  36
  37 #if HAVE_WINSOCK2_H
  38 #include <winsock2.h>
  39 #include <ws2tcpip.h>
  40
  41 #ifndef EPROTONOSUPPORT
  42 #define EPROTONOSUPPORT WSAEPROTONOSUPPORT
  43 #endif
  44 #ifndef ETIMEDOUT
  45 #define ETIMEDOUT       WSAETIMEDOUT
  46 #endif
  47 #ifndef ECONNREFUSED
  48 #define ECONNREFUSED    WSAECONNREFUSED
  49 #endif
  50 #ifndef EINPROGRESS
  51 #define EINPROGRESS     WSAEINPROGRESS
  52 #endif
  53 #ifndef ENOTCONN
  54 #define ENOTCONN        WSAENOTCONN
  55 #endif
  56
  57 #define getsockopt(a, b, c, d, e) getsockopt(a, b, c, (char*) d, e)
  58 #define setsockopt(a, b, c, d, e) setsockopt(a, b, c, (const char*) d, e)
  59
  60 int ff_neterrno(void);
  61 #else
  62 #include <sys/types.h>
  63 #include <sys/socket.h>
  64 #include <netinet/in.h>
  65 #include <netinet/tcp.h>
  66 #include <netdb.h>
  67
  68 #define ff_neterrno() AVERROR(errno)
  69 #endif /* HAVE_WINSOCK2_H */
  70
  71 #if HAVE_ARPA_INET_H
  72 #include <arpa/inet.h>
  73 #endif
  74
  75 #if HAVE_POLL_H
  76 #include <poll.h>
  77 #endif
  78
  79 int ff_socket_nonblock(int socket, int enable);
  80
  81 int ff_network_init(void);
  82 void ff_network_close(void);
  83
  84 int ff_tls_init(void);
  85 void ff_tls_deinit(void);
  86
  87 int ff_network_wait_fd(int fd, int write);
  88
  89 /**
  90  * This works similarly to ff_network_wait_fd, but waits up to 'timeout' microseconds
  91  * Uses ff_network_wait_fd in a loop
  92  *
  93  * @param fd Socket descriptor
  94  * @param write Set 1 to wait for socket able to be read, 0 to be written
  95  * @param timeout Timeout interval, in microseconds. Actual precision is 100000 mcs, due to ff_network_wait_fd usage
  96  * @param int_cb Interrupt callback, is checked before each ff_network_wait_fd call
  97  * @return 0 if data can be read/written, AVERROR(ETIMEDOUT) if timeout expired, or negative error code
  98  */
  99 int ff_network_wait_fd_timeout(int fd, int write, int64_t timeout, AVIOInterruptCB *int_cb);
 100
 101 /**
 102  * Waits for up to 'timeout' microseconds. If the usert's int_cb is set and
 103  * triggered, return before that.
 104  * @param timeout Timeout in microseconds. Maybe have lower actual precision.
 105  * @param int_cb Interrupt callback, is checked regularly.
 106  * @return AVERROR(ETIMEDOUT) if timeout expirted, AVERROR_EXIT if interrupted by int_cb
 107  */
 108 int ff_network_sleep_interruptible(int64_t timeout, AVIOInterruptCB *int_cb);
 109
 110 #if !HAVE_STRUCT_SOCKADDR_STORAGE
 111 struct sockaddr_storage {
 112 #if HAVE_STRUCT_SOCKADDR_SA_LEN
 113     uint8_t ss_len;
 114     uint8_t ss_family;
 115 #else
 116     uint16_t ss_family;
 117 #endif /* HAVE_STRUCT_SOCKADDR_SA_LEN */
 118     char ss_pad1[6];
 119     int64_t ss_align;
 120     char ss_pad2[112];
 121 };
 122 #endif /* !HAVE_STRUCT_SOCKADDR_STORAGE */
 123
 124 typedef union sockaddr_union {
 125     struct sockaddr_storage storage;
 126     struct sockaddr_in in;
 127 #if HAVE_STRUCT_SOCKADDR_IN6
 128     struct sockaddr_in6 in6;
 129 #endif
 130 } sockaddr_union;
 131
 132 #ifndef MSG_NOSIGNAL
 133 #define MSG_NOSIGNAL 0
 134 #endif
 135
 136 #if !HAVE_STRUCT_ADDRINFO
 137 struct addrinfo {
 138     int ai_flags;
 139     int ai_family;
 140     int ai_socktype;
 141     int ai_protocol;
 142     int ai_addrlen;
 143     struct sockaddr *ai_addr;
 144     char *ai_canonname;
 145     struct addrinfo *ai_next;
 146 };
 147 #endif /* !HAVE_STRUCT_ADDRINFO */
 148
 149 /* getaddrinfo constants */
 150 #ifndef EAI_AGAIN
 151 #define EAI_AGAIN 2
 152 #endif
 153 #ifndef EAI_BADFLAGS
 154 #define EAI_BADFLAGS 3
 155 #endif
 156 #ifndef EAI_FAIL
 157 #define EAI_FAIL 4
 158 #endif
 159 #ifndef EAI_FAMILY
 160 #define EAI_FAMILY 5
 161 #endif
 162 #ifndef EAI_MEMORY
 163 #define EAI_MEMORY 6
 164 #endif
 165 #ifndef EAI_NODATA
 166 #define EAI_NODATA 7
 167 #endif
 168 #ifndef EAI_NONAME
 169 #define EAI_NONAME 8
 170 #endif
 171 #ifndef EAI_SERVICE
 172 #define EAI_SERVICE 9
 173 #endif
 174 #ifndef EAI_SOCKTYPE
 175 #define EAI_SOCKTYPE 10
 176 #endif
 177
 178 #ifndef AI_PASSIVE
 179 #define AI_PASSIVE 1
 180 #endif
 181
 182 #ifndef AI_CANONNAME
 183 #define AI_CANONNAME 2
 184 #endif
 185
 186 #ifndef AI_NUMERICHOST
 187 #define AI_NUMERICHOST 4
 188 #endif
 189
 190 #ifndef NI_NOFQDN
 191 #define NI_NOFQDN 1
 192 #endif
 193
 194 #ifndef NI_NUMERICHOST
 195 #define NI_NUMERICHOST 2
 196 #endif
 197
 198 #ifndef NI_NAMERQD
 199 #define NI_NAMERQD 4
 200 #endif
 201
 202 #ifndef NI_NUMERICSERV
 203 #define NI_NUMERICSERV 8
 204 #endif
 205
 206 #ifndef NI_DGRAM
 207 #define NI_DGRAM 16
 208 #endif
 209
 210 #if !HAVE_GETADDRINFO
 211 int ff_getaddrinfo(const char *node, const char *service,
 212                    const struct addrinfo *hints, struct addrinfo **res);
 213 void ff_freeaddrinfo(struct addrinfo *res);
 214 int ff_getnameinfo(const struct sockaddr *sa, int salen,
 215                    char *host, int hostlen,
 216                    char *serv, int servlen, int flags);
 217 #define getaddrinfo ff_getaddrinfo
 218 #define freeaddrinfo ff_freeaddrinfo
 219 #define getnameinfo ff_getnameinfo
 220 #endif /* !HAVE_GETADDRINFO */
 221
 222 #if !HAVE_GETADDRINFO || HAVE_WINSOCK2_H
 223 const char *ff_gai_strerror(int ecode);
 224 #undef gai_strerror
 225 #define gai_strerror ff_gai_strerror
 226 #endif /* !HAVE_GETADDRINFO || HAVE_WINSOCK2_H */
 227
 228 #ifndef INADDR_LOOPBACK
 229 #define INADDR_LOOPBACK 0x7f000001
 230 #endif
 231
 232 #ifndef INET_ADDRSTRLEN
 233 #define INET_ADDRSTRLEN 16
 234 #endif
 235
 236 #ifndef INET6_ADDRSTRLEN
 237 #define INET6_ADDRSTRLEN INET_ADDRSTRLEN
 238 #endif
 239
 240 #ifndef IN_MULTICAST
 241 #define IN_MULTICAST(a) ((((uint32_t)(a)) & 0xf0000000) == 0xe0000000)
 242 #endif
 243 #ifndef IN6_IS_ADDR_MULTICAST
 244 #define IN6_IS_ADDR_MULTICAST(a) (((uint8_t *) (a))[0] == 0xff)
 245 #endif
 246
 247 int ff_is_multicast_address(struct sockaddr *addr);
 248
 249 #define POLLING_TIME 100 /// Time in milliseconds between interrupt check
 250
 251 /**
 252  * Bind to a file descriptor and poll for a connection.
 253  *
 254  * @param fd      First argument of bind().
 255  * @param addr    Second argument of bind().
 256  * @param addrlen Third argument of bind().
 257  * @param timeout Polling timeout in milliseconds.
 258  * @param h       URLContext providing interrupt check
 259  *                callback and logging context.
 260  * @return        A non-blocking file descriptor on success
 261  *                or an AVERROR on failure.
 262  */
 263 int ff_listen_bind(int fd, const struct sockaddr *addr,
 264                    socklen_t addrlen, int timeout,
 265                    URLContext *h);
 266
 267 /**
 268  * Bind to a file descriptor to an address without accepting connections.
 269  * @param fd      First argument of bind().
 270  * @param addr    Second argument of bind().
 271  * @param addrlen Third argument of bind().
 272  * @return        0 on success or an AVERROR on failure.
 273  */
 274 int ff_listen(int fd, const struct sockaddr *addr, socklen_t addrlen);
 275
 276 /**
 277  * Poll for a single connection on the passed file descriptor.
 278  * @param fd      The listening socket file descriptor.
 279  * @param timeout Polling timeout in milliseconds.
 280  * @param h       URLContext providing interrupt check
 281  *                callback and logging context.
 282  * @return        A non-blocking file descriptor on success
 283  *                or an AVERROR on failure.
 284  */
 285 int ff_accept(int fd, int timeout, URLContext *h);
 286
 287 /**
 288  * Connect to a file descriptor and poll for result.
 289  *
 290  * @param fd       First argument of connect(),
 291  *                 will be set as non-blocking.
 292  * @param addr     Second argument of connect().
 293  * @param addrlen  Third argument of connect().
 294  * @param timeout  Polling timeout in milliseconds.
 295  * @param h        URLContext providing interrupt check
 296  *                 callback and logging context.
 297  * @param will_try_next Whether the caller will try to connect to another
 298  *                 address for the same host name, affecting the form of
 299  *                 logged errors.
 300  * @return         0 on success, AVERROR on failure.
 301  */
 302 int ff_listen_connect(int fd, const struct sockaddr *addr,
 303                       socklen_t addrlen, int timeout,
 304                       URLContext *h, int will_try_next);
 305
 306 int ff_http_match_no_proxy(const char *no_proxy, const char *hostname);
 307
 308 int ff_socket(int domain, int type, int protocol);
 309
 310 void ff_log_net_error(void *ctx, int level, const char* prefix);
 311
 312 /**
 313  * Connect to any of the given addrinfo addresses, with multiple attempts
 314  * running in parallel.
 315  *
 316  * @param addrs    The list of addresses to try to connect to.
 317  *                 This list will be mutated internally, but the list head
 318  *                 will remain as such, so this doesn't affect the caller
 319  *                 freeing the list afterwards.
 320  * @param timeout_ms_per_address The number of milliseconds to wait for each
 321  *                 connection attempt. Since multiple addresses are tried,
 322  *                 some of them in parallel, the total run time will at most
 323  *                 be timeout_ms_per_address*ceil(nb_addrs/parallel) +
 324  *                 (parallel - 1) * NEXT_ATTEMPT_DELAY_MS.
 325  * @param parallel The maximum number of connections to attempt in parallel.
 326  *                 This is limited to an internal maximum capacity.
 327  * @param h        URLContext providing interrupt check
 328  *                 callback and logging context.
 329  * @param fd       If successful, the connected socket is returned here.
 330  * @param customize_fd Function that will be called for each socket created,
 331  *                 to allow the caller to set socket options before calling
 332  *                 connect() on it, may be NULL.
 333  * @param customize_ctx Context parameter passed to customize_fd.
 334  * @return         0 on success, AVERROR on failure.
 335  */
 336 int ff_connect_parallel(struct addrinfo *addrs, int timeout_ms_per_address,
 337                         int parallel, URLContext *h, int *fd,
 338                         void (*customize_fd)(void *, int), void *customize_ctx);
 339
 340 #endif /* AVFORMAT_NETWORK_H */