/* * 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 */