341 lines
9.8 KiB
C
341 lines
9.8 KiB
C
|
/*
|
||
|
* Copyright (c) 2007 The FFmpeg Project
|
||
|
*
|
||
|
* This file is part of FFmpeg.
|
||
|
*
|
||
|
* FFmpeg is free software; you can redistribute it and/or
|
||
|
* modify it under the terms of the GNU Lesser General Public
|
||
|
* License as published by the Free Software Foundation; either
|
||
|
* version 2.1 of the License, or (at your option) any later version.
|
||
|
*
|
||
|
* FFmpeg is distributed in the hope that it will be useful,
|
||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
* Lesser General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU Lesser General Public
|
||
|
* License along with FFmpeg; if not, write to the Free Software
|
||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||
|
*/
|
||
|
|
||
|
#ifndef AVFORMAT_NETWORK_H
|
||
|
#define AVFORMAT_NETWORK_H
|
||
|
|
||
|
#include <errno.h>
|
||
|
#include <stdint.h>
|
||
|
|
||
|
#include "config.h"
|
||
|
#include "libavutil/error.h"
|
||
|
#include "os_support.h"
|
||
|
#include "avio.h"
|
||
|
#include "url.h"
|
||
|
|
||
|
#if HAVE_UNISTD_H
|
||
|
#include <unistd.h>
|
||
|
#endif
|
||
|
|
||
|
#if HAVE_WINSOCK2_H
|
||
|
#include <winsock2.h>
|
||
|
#include <ws2tcpip.h>
|
||
|
|
||
|
#ifndef EPROTONOSUPPORT
|
||
|
#define EPROTONOSUPPORT WSAEPROTONOSUPPORT
|
||
|
#endif
|
||
|
#ifndef ETIMEDOUT
|
||
|
#define ETIMEDOUT WSAETIMEDOUT
|
||
|
#endif
|
||
|
#ifndef ECONNREFUSED
|
||
|
#define ECONNREFUSED WSAECONNREFUSED
|
||
|
#endif
|
||
|
#ifndef EINPROGRESS
|
||
|
#define EINPROGRESS WSAEINPROGRESS
|
||
|
#endif
|
||
|
#ifndef ENOTCONN
|
||
|
#define ENOTCONN WSAENOTCONN
|
||
|
#endif
|
||
|
|
||
|
#define getsockopt(a, b, c, d, e) getsockopt(a, b, c, (char*) d, e)
|
||
|
#define setsockopt(a, b, c, d, e) setsockopt(a, b, c, (const char*) d, e)
|
||
|
|
||
|
int ff_neterrno(void);
|
||
|
#else
|
||
|
#include <sys/types.h>
|
||
|
#include <sys/socket.h>
|
||
|
#include <netinet/in.h>
|
||
|
#include <netinet/tcp.h>
|
||
|
#include <netdb.h>
|
||
|
|
||
|
#define ff_neterrno() AVERROR(errno)
|
||
|
#endif /* HAVE_WINSOCK2_H */
|
||
|
|
||
|
#if HAVE_ARPA_INET_H
|
||
|
#include <arpa/inet.h>
|
||
|
#endif
|
||
|
|
||
|
#if HAVE_POLL_H
|
||
|
#include <poll.h>
|
||
|
#endif
|
||
|
|
||
|
int ff_socket_nonblock(int socket, int enable);
|
||
|
|
||
|
int ff_network_init(void);
|
||
|
void ff_network_close(void);
|
||
|
|
||
|
int ff_tls_init(void);
|
||
|
void ff_tls_deinit(void);
|
||
|
|
||
|
int ff_network_wait_fd(int fd, int write);
|
||
|
|
||
|
/**
|
||
|
* This works similarly to ff_network_wait_fd, but waits up to 'timeout' microseconds
|
||
|
* Uses ff_network_wait_fd in a loop
|
||
|
*
|
||
|
* @param fd Socket descriptor
|
||
|
* @param write Set 1 to wait for socket able to be read, 0 to be written
|
||
|
* @param timeout Timeout interval, in microseconds. Actual precision is 100000 mcs, due to ff_network_wait_fd usage
|
||
|
* @param int_cb Interrupt callback, is checked before each ff_network_wait_fd call
|
||
|
* @return 0 if data can be read/written, AVERROR(ETIMEDOUT) if timeout expired, or negative error code
|
||
|
*/
|
||
|
int ff_network_wait_fd_timeout(int fd, int write, int64_t timeout, AVIOInterruptCB *int_cb);
|
||
|
|
||
|
/**
|
||
|
* Waits for up to 'timeout' microseconds. If the usert's int_cb is set and
|
||
|
* triggered, return before that.
|
||
|
* @param timeout Timeout in microseconds. Maybe have lower actual precision.
|
||
|
* @param int_cb Interrupt callback, is checked regularly.
|
||
|
* @return AVERROR(ETIMEDOUT) if timeout expirted, AVERROR_EXIT if interrupted by int_cb
|
||
|
*/
|
||
|
int ff_network_sleep_interruptible(int64_t timeout, AVIOInterruptCB *int_cb);
|
||
|
|
||
|
#if !HAVE_STRUCT_SOCKADDR_STORAGE
|
||
|
struct sockaddr_storage {
|
||
|
#if HAVE_STRUCT_SOCKADDR_SA_LEN
|
||
|
uint8_t ss_len;
|
||
|
uint8_t ss_family;
|
||
|
#else
|
||
|
uint16_t ss_family;
|
||
|
#endif /* HAVE_STRUCT_SOCKADDR_SA_LEN */
|
||
|
char ss_pad1[6];
|
||
|
int64_t ss_align;
|
||
|
char ss_pad2[112];
|
||
|
};
|
||
|
#endif /* !HAVE_STRUCT_SOCKADDR_STORAGE */
|
||
|
|
||
|
typedef union sockaddr_union {
|
||
|
struct sockaddr_storage storage;
|
||
|
struct sockaddr_in in;
|
||
|
#if HAVE_STRUCT_SOCKADDR_IN6
|
||
|
struct sockaddr_in6 in6;
|
||
|
#endif
|
||
|
} sockaddr_union;
|
||
|
|
||
|
#ifndef MSG_NOSIGNAL
|
||
|
#define MSG_NOSIGNAL 0
|
||
|
#endif
|
||
|
|
||
|
#if !HAVE_STRUCT_ADDRINFO
|
||
|
struct addrinfo {
|
||
|
int ai_flags;
|
||
|
int ai_family;
|
||
|
int ai_socktype;
|
||
|
int ai_protocol;
|
||
|
int ai_addrlen;
|
||
|
struct sockaddr *ai_addr;
|
||
|
char *ai_canonname;
|
||
|
struct addrinfo *ai_next;
|
||
|
};
|
||
|
#endif /* !HAVE_STRUCT_ADDRINFO */
|
||
|
|
||
|
/* getaddrinfo constants */
|
||
|
#ifndef EAI_AGAIN
|
||
|
#define EAI_AGAIN 2
|
||
|
#endif
|
||
|
#ifndef EAI_BADFLAGS
|
||
|
#define EAI_BADFLAGS 3
|
||
|
#endif
|
||
|
#ifndef EAI_FAIL
|
||
|
#define EAI_FAIL 4
|
||
|
#endif
|
||
|
#ifndef EAI_FAMILY
|
||
|
#define EAI_FAMILY 5
|
||
|
#endif
|
||
|
#ifndef EAI_MEMORY
|
||
|
#define EAI_MEMORY 6
|
||
|
#endif
|
||
|
#ifndef EAI_NODATA
|
||
|
#define EAI_NODATA 7
|
||
|
#endif
|
||
|
#ifndef EAI_NONAME
|
||
|
#define EAI_NONAME 8
|
||
|
#endif
|
||
|
#ifndef EAI_SERVICE
|
||
|
#define EAI_SERVICE 9
|
||
|
#endif
|
||
|
#ifndef EAI_SOCKTYPE
|
||
|
#define EAI_SOCKTYPE 10
|
||
|
#endif
|
||
|
|
||
|
#ifndef AI_PASSIVE
|
||
|
#define AI_PASSIVE 1
|
||
|
#endif
|
||
|
|
||
|
#ifndef AI_CANONNAME
|
||
|
#define AI_CANONNAME 2
|
||
|
#endif
|
||
|
|
||
|
#ifndef AI_NUMERICHOST
|
||
|
#define AI_NUMERICHOST 4
|
||
|
#endif
|
||
|
|
||
|
#ifndef NI_NOFQDN
|
||
|
#define NI_NOFQDN 1
|
||
|
#endif
|
||
|
|
||
|
#ifndef NI_NUMERICHOST
|
||
|
#define NI_NUMERICHOST 2
|
||
|
#endif
|
||
|
|
||
|
#ifndef NI_NAMERQD
|
||
|
#define NI_NAMERQD 4
|
||
|
#endif
|
||
|
|
||
|
#ifndef NI_NUMERICSERV
|
||
|
#define NI_NUMERICSERV 8
|
||
|
#endif
|
||
|
|
||
|
#ifndef NI_DGRAM
|
||
|
#define NI_DGRAM 16
|
||
|
#endif
|
||
|
|
||
|
#if !HAVE_GETADDRINFO
|
||
|
int ff_getaddrinfo(const char *node, const char *service,
|
||
|
const struct addrinfo *hints, struct addrinfo **res);
|
||
|
void ff_freeaddrinfo(struct addrinfo *res);
|
||
|
int ff_getnameinfo(const struct sockaddr *sa, int salen,
|
||
|
char *host, int hostlen,
|
||
|
char *serv, int servlen, int flags);
|
||
|
#define getaddrinfo ff_getaddrinfo
|
||
|
#define freeaddrinfo ff_freeaddrinfo
|
||
|
#define getnameinfo ff_getnameinfo
|
||
|
#endif /* !HAVE_GETADDRINFO */
|
||
|
|
||
|
#if !HAVE_GETADDRINFO || HAVE_WINSOCK2_H
|
||
|
const char *ff_gai_strerror(int ecode);
|
||
|
#undef gai_strerror
|
||
|
#define gai_strerror ff_gai_strerror
|
||
|
#endif /* !HAVE_GETADDRINFO || HAVE_WINSOCK2_H */
|
||
|
|
||
|
#ifndef INADDR_LOOPBACK
|
||
|
#define INADDR_LOOPBACK 0x7f000001
|
||
|
#endif
|
||
|
|
||
|
#ifndef INET_ADDRSTRLEN
|
||
|
#define INET_ADDRSTRLEN 16
|
||
|
#endif
|
||
|
|
||
|
#ifndef INET6_ADDRSTRLEN
|
||
|
#define INET6_ADDRSTRLEN INET_ADDRSTRLEN
|
||
|
#endif
|
||
|
|
||
|
#ifndef IN_MULTICAST
|
||
|
#define IN_MULTICAST(a) ((((uint32_t)(a)) & 0xf0000000) == 0xe0000000)
|
||
|
#endif
|
||
|
#ifndef IN6_IS_ADDR_MULTICAST
|
||
|
#define IN6_IS_ADDR_MULTICAST(a) (((uint8_t *) (a))[0] == 0xff)
|
||
|
#endif
|
||
|
|
||
|
int ff_is_multicast_address(struct sockaddr *addr);
|
||
|
|
||
|
#define POLLING_TIME 100 /// Time in milliseconds between interrupt check
|
||
|
|
||
|
/**
|
||
|
* Bind to a file descriptor and poll for a connection.
|
||
|
*
|
||
|
* @param fd First argument of bind().
|
||
|
* @param addr Second argument of bind().
|
||
|
* @param addrlen Third argument of bind().
|
||
|
* @param timeout Polling timeout in milliseconds.
|
||
|
* @param h URLContext providing interrupt check
|
||
|
* callback and logging context.
|
||
|
* @return A non-blocking file descriptor on success
|
||
|
* or an AVERROR on failure.
|
||
|
*/
|
||
|
int ff_listen_bind(int fd, const struct sockaddr *addr,
|
||
|
socklen_t addrlen, int timeout,
|
||
|
URLContext *h);
|
||
|
|
||
|
/**
|
||
|
* Bind to a file descriptor to an address without accepting connections.
|
||
|
* @param fd First argument of bind().
|
||
|
* @param addr Second argument of bind().
|
||
|
* @param addrlen Third argument of bind().
|
||
|
* @return 0 on success or an AVERROR on failure.
|
||
|
*/
|
||
|
int ff_listen(int fd, const struct sockaddr *addr, socklen_t addrlen);
|
||
|
|
||
|
/**
|
||
|
* Poll for a single connection on the passed file descriptor.
|
||
|
* @param fd The listening socket file descriptor.
|
||
|
* @param timeout Polling timeout in milliseconds.
|
||
|
* @param h URLContext providing interrupt check
|
||
|
* callback and logging context.
|
||
|
* @return A non-blocking file descriptor on success
|
||
|
* or an AVERROR on failure.
|
||
|
*/
|
||
|
int ff_accept(int fd, int timeout, URLContext *h);
|
||
|
|
||
|
/**
|
||
|
* Connect to a file descriptor and poll for result.
|
||
|
*
|
||
|
* @param fd First argument of connect(),
|
||
|
* will be set as non-blocking.
|
||
|
* @param addr Second argument of connect().
|
||
|
* @param addrlen Third argument of connect().
|
||
|
* @param timeout Polling timeout in milliseconds.
|
||
|
* @param h URLContext providing interrupt check
|
||
|
* callback and logging context.
|
||
|
* @param will_try_next Whether the caller will try to connect to another
|
||
|
* address for the same host name, affecting the form of
|
||
|
* logged errors.
|
||
|
* @return 0 on success, AVERROR on failure.
|
||
|
*/
|
||
|
int ff_listen_connect(int fd, const struct sockaddr *addr,
|
||
|
socklen_t addrlen, int timeout,
|
||
|
URLContext *h, int will_try_next);
|
||
|
|
||
|
int ff_http_match_no_proxy(const char *no_proxy, const char *hostname);
|
||
|
|
||
|
int ff_socket(int domain, int type, int protocol);
|
||
|
|
||
|
void ff_log_net_error(void *ctx, int level, const char* prefix);
|
||
|
|
||
|
/**
|
||
|
* Connect to any of the given addrinfo addresses, with multiple attempts
|
||
|
* running in parallel.
|
||
|
*
|
||
|
* @param addrs The list of addresses to try to connect to.
|
||
|
* This list will be mutated internally, but the list head
|
||
|
* will remain as such, so this doesn't affect the caller
|
||
|
* freeing the list afterwards.
|
||
|
* @param timeout_ms_per_address The number of milliseconds to wait for each
|
||
|
* connection attempt. Since multiple addresses are tried,
|
||
|
* some of them in parallel, the total run time will at most
|
||
|
* be timeout_ms_per_address*ceil(nb_addrs/parallel) +
|
||
|
* (parallel - 1) * NEXT_ATTEMPT_DELAY_MS.
|
||
|
* @param parallel The maximum number of connections to attempt in parallel.
|
||
|
* This is limited to an internal maximum capacity.
|
||
|
* @param h URLContext providing interrupt check
|
||
|
* callback and logging context.
|
||
|
* @param fd If successful, the connected socket is returned here.
|
||
|
* @param customize_fd Function that will be called for each socket created,
|
||
|
* to allow the caller to set socket options before calling
|
||
|
* connect() on it, may be NULL.
|
||
|
* @param customize_ctx Context parameter passed to customize_fd.
|
||
|
* @return 0 on success, AVERROR on failure.
|
||
|
*/
|
||
|
int ff_connect_parallel(struct addrinfo *addrs, int timeout_ms_per_address,
|
||
|
int parallel, URLContext *h, int *fd,
|
||
|
void (*customize_fd)(void *, int), void *customize_ctx);
|
||
|
|
||
|
#endif /* AVFORMAT_NETWORK_H */
|