| /* |
| * 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 "config.h" |
| #include "libavutil/error.h" |
| #include "os_support.h" |
| |
| #if HAVE_WINSOCK2_H |
| #include <winsock2.h> |
| #include <ws2tcpip.h> |
| |
| #define EPROTONOSUPPORT WSAEPROTONOSUPPORT |
| #define ETIMEDOUT WSAETIMEDOUT |
| #define ECONNREFUSED WSAECONNREFUSED |
| #define EINPROGRESS WSAEINPROGRESS |
| |
| static inline int ff_neterrno(void) |
| { |
| int err = WSAGetLastError(); |
| switch (err) { |
| case WSAEWOULDBLOCK: |
| return AVERROR(EAGAIN); |
| case WSAEINTR: |
| return AVERROR(EINTR); |
| } |
| return -err; |
| } |
| #else |
| #include <sys/types.h> |
| #include <sys/socket.h> |
| #include <netinet/in.h> |
| #include <netdb.h> |
| |
| #define ff_neterrno() AVERROR(errno) |
| #endif |
| |
| #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); |
| |
| static inline int ff_network_init(void) |
| { |
| #if HAVE_WINSOCK2_H |
| WSADATA wsaData; |
| if (WSAStartup(MAKEWORD(1,1), &wsaData)) |
| return 0; |
| #endif |
| return 1; |
| } |
| |
| static inline int ff_network_wait_fd(int fd, int write) |
| { |
| int ev = write ? POLLOUT : POLLIN; |
| struct pollfd p = { .fd = fd, .events = ev, .revents = 0 }; |
| int ret; |
| ret = poll(&p, 1, 100); |
| return ret < 0 ? ff_neterrno() : p.revents & (ev | POLLERR | POLLHUP) ? 0 : AVERROR(EAGAIN); |
| } |
| |
| static inline void ff_network_close(void) |
| { |
| #if HAVE_WINSOCK2_H |
| WSACleanup(); |
| #endif |
| } |
| |
| int ff_inet_aton (const char * str, struct in_addr * add); |
| |
| #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 |
| char ss_pad1[6]; |
| int64_t ss_align; |
| char ss_pad2[112]; |
| }; |
| #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 |
| |
| /* getaddrinfo constants */ |
| #ifndef EAI_FAIL |
| #define EAI_FAIL 4 |
| #endif |
| |
| #ifndef EAI_FAMILY |
| #define EAI_FAMILY 5 |
| #endif |
| |
| #ifndef EAI_NONAME |
| #define EAI_NONAME 8 |
| #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); |
| const char *ff_gai_strerror(int ecode); |
| #define getaddrinfo ff_getaddrinfo |
| #define freeaddrinfo ff_freeaddrinfo |
| #define getnameinfo ff_getnameinfo |
| #define gai_strerror ff_gai_strerror |
| #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 |
| |
| static inline int ff_is_multicast_address(struct sockaddr *addr) |
| { |
| if (addr->sa_family == AF_INET) { |
| return IN_MULTICAST(ntohl(((struct sockaddr_in *)addr)->sin_addr.s_addr)); |
| } |
| #if HAVE_STRUCT_SOCKADDR_IN6 |
| if (addr->sa_family == AF_INET6) { |
| return IN6_IS_ADDR_MULTICAST(&((struct sockaddr_in6 *)addr)->sin6_addr); |
| } |
| #endif |
| |
| return 0; |
| } |
| |
| #endif /* AVFORMAT_NETWORK_H */ |