/*

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