network.h 4.44 KB
Newer Older
1
/*
2
 * Copyright (c) 2007 The Libav Project
3
 *
4
 * This file is part of Libav.
5
 *
6
 * Libav is free software; you can redistribute it and/or
7 8 9 10
 * 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.
 *
11
 * Libav is distributed in the hope that it will be useful,
12 13 14 15 16
 * 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
17
 * License along with Libav; if not, write to the Free Software
18 19 20
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

21 22
#ifndef AVFORMAT_NETWORK_H
#define AVFORMAT_NETWORK_H
23

24
#include "config.h"
25
#include "os_support.h"
26

27
#if HAVE_WINSOCK2_H
28 29 30
#include <winsock2.h>
#include <ws2tcpip.h>

31 32 33 34 35
#define EPROTONOSUPPORT WSAEPROTONOSUPPORT
#define ETIMEDOUT       WSAETIMEDOUT
#define ECONNREFUSED    WSAECONNREFUSED
#define EINPROGRESS     WSAEINPROGRESS

36 37
static inline int ff_neterrno(void)
{
38 39 40 41 42 43 44 45 46
    int err = WSAGetLastError();
    switch (err) {
    case WSAEWOULDBLOCK:
        return AVERROR(EAGAIN);
    case WSAEINTR:
        return AVERROR(EINTR);
    }
    return -err;
}
47
#else
48 49 50 51 52
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>

53
#define ff_neterrno() AVERROR(errno)
54 55
#endif

56
#if HAVE_ARPA_INET_H
57 58
#include <arpa/inet.h>
#endif
59

60 61 62 63
#if HAVE_POLL_H
#include <poll.h>
#endif

64 65
int ff_socket_nonblock(int socket, int enable);

66 67
static inline int ff_network_init(void)
{
68
#if HAVE_WINSOCK2_H
69 70 71 72 73 74 75
    WSADATA wsaData;
    if (WSAStartup(MAKEWORD(1,1), &wsaData))
        return 0;
#endif
    return 1;
}

76 77 78 79 80 81
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);
82
    return ret < 0 ? ff_neterrno() : p.revents & (ev | POLLERR | POLLHUP) ? 0 : AVERROR(EAGAIN);
83 84
}

85 86
static inline void ff_network_close(void)
{
87
#if HAVE_WINSOCK2_H
88 89 90 91
    WSACleanup();
#endif
}

92
int ff_inet_aton (const char * str, struct in_addr * add);
93

94 95
#if !HAVE_STRUCT_SOCKADDR_STORAGE
struct sockaddr_storage {
96 97 98 99 100 101 102 103 104
#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];
105 106 107
};
#endif

108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125
#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

126 127 128 129 130 131 132 133
#ifndef EAI_FAMILY
#define EAI_FAMILY 5
#endif

#ifndef EAI_NONAME
#define EAI_NONAME 8
#endif

134 135 136 137 138 139 140 141 142 143 144 145
#ifndef AI_PASSIVE
#define AI_PASSIVE 1
#endif

#ifndef AI_CANONNAME
#define AI_CANONNAME 2
#endif

#ifndef AI_NUMERICHOST
#define AI_NUMERICHOST 4
#endif

146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165
#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

166 167 168 169
#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);
170 171 172
int ff_getnameinfo(const struct sockaddr *sa, int salen,
                   char *host, int hostlen,
                   char *serv, int servlen, int flags);
173
const char *ff_gai_strerror(int ecode);
174 175
#define getaddrinfo ff_getaddrinfo
#define freeaddrinfo ff_freeaddrinfo
176
#define getnameinfo ff_getnameinfo
177
#define gai_strerror ff_gai_strerror
178 179
#endif

180 181 182 183
#ifndef INET6_ADDRSTRLEN
#define INET6_ADDRSTRLEN INET_ADDRSTRLEN
#endif

184 185 186 187 188 189 190
#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

191
static inline int ff_is_multicast_address(struct sockaddr *addr)
192
{
193
    if (addr->sa_family == AF_INET) {
194 195 196
        return IN_MULTICAST(ntohl(((struct sockaddr_in *)addr)->sin_addr.s_addr));
    }
#if HAVE_STRUCT_SOCKADDR_IN6
197
    if (addr->sa_family == AF_INET6) {
198 199 200 201 202 203 204
        return IN6_IS_ADDR_MULTICAST(&((struct sockaddr_in6 *)addr)->sin6_addr);
    }
#endif

    return 0;
}

205
#endif /* AVFORMAT_NETWORK_H */