80d094a0deaa36f1e637186491a75e488c64c880
[ffmpeg.git] / libavformat / network.h
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  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * Libav is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * 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  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #ifndef AVFORMAT_NETWORK_H
22 #define AVFORMAT_NETWORK_H
23
24 #include <errno.h>
25
26 #include "config.h"
27 #include "libavutil/error.h"
28 #include "os_support.h"
29
30 #if HAVE_WINSOCK2_H
31 #include <winsock2.h>
32 #include <ws2tcpip.h>
33
34 #define EPROTONOSUPPORT WSAEPROTONOSUPPORT
35 #define ETIMEDOUT       WSAETIMEDOUT
36 #define ECONNREFUSED    WSAECONNREFUSED
37 #define EINPROGRESS     WSAEINPROGRESS
38
39 static inline int ff_neterrno(void)
40 {
41     int err = WSAGetLastError();
42     switch (err) {
43     case WSAEWOULDBLOCK:
44         return AVERROR(EAGAIN);
45     case WSAEINTR:
46         return AVERROR(EINTR);
47     }
48     return -err;
49 }
50 #else
51 #include <sys/types.h>
52 #include <sys/socket.h>
53 #include <netinet/in.h>
54 #include <netdb.h>
55
56 #define ff_neterrno() AVERROR(errno)
57 #endif
58
59 #if HAVE_ARPA_INET_H
60 #include <arpa/inet.h>
61 #endif
62
63 #if HAVE_POLL_H
64 #include <poll.h>
65 #endif
66
67 int ff_socket_nonblock(int socket, int enable);
68
69 static inline int ff_network_init(void)
70 {
71 #if HAVE_WINSOCK2_H
72     WSADATA wsaData;
73     if (WSAStartup(MAKEWORD(1,1), &wsaData))
74         return 0;
75 #endif
76     return 1;
77 }
78
79 static inline int ff_network_wait_fd(int fd, int write)
80 {
81     int ev = write ? POLLOUT : POLLIN;
82     struct pollfd p = { .fd = fd, .events = ev, .revents = 0 };
83     int ret;
84     ret = poll(&p, 1, 100);
85     return ret < 0 ? ff_neterrno() : p.revents & (ev | POLLERR | POLLHUP) ? 0 : AVERROR(EAGAIN);
86 }
87
88 static inline void ff_network_close(void)
89 {
90 #if HAVE_WINSOCK2_H
91     WSACleanup();
92 #endif
93 }
94
95 int ff_inet_aton (const char * str, struct in_addr * add);
96
97 #if !HAVE_STRUCT_SOCKADDR_STORAGE
98 struct sockaddr_storage {
99 #if HAVE_STRUCT_SOCKADDR_SA_LEN
100     uint8_t ss_len;
101     uint8_t ss_family;
102 #else
103     uint16_t ss_family;
104 #endif
105     char ss_pad1[6];
106     int64_t ss_align;
107     char ss_pad2[112];
108 };
109 #endif
110
111 #if !HAVE_STRUCT_ADDRINFO
112 struct addrinfo {
113     int ai_flags;
114     int ai_family;
115     int ai_socktype;
116     int ai_protocol;
117     int ai_addrlen;
118     struct sockaddr *ai_addr;
119     char *ai_canonname;
120     struct addrinfo *ai_next;
121 };
122 #endif
123
124 /* getaddrinfo constants */
125 #ifndef EAI_FAIL
126 #define EAI_FAIL 4
127 #endif
128
129 #ifndef EAI_FAMILY
130 #define EAI_FAMILY 5
131 #endif
132
133 #ifndef EAI_NONAME
134 #define EAI_NONAME 8
135 #endif
136
137 #ifndef AI_PASSIVE
138 #define AI_PASSIVE 1
139 #endif
140
141 #ifndef AI_CANONNAME
142 #define AI_CANONNAME 2
143 #endif
144
145 #ifndef AI_NUMERICHOST
146 #define AI_NUMERICHOST 4
147 #endif
148
149 #ifndef NI_NOFQDN
150 #define NI_NOFQDN 1
151 #endif
152
153 #ifndef NI_NUMERICHOST
154 #define NI_NUMERICHOST 2
155 #endif
156
157 #ifndef NI_NAMERQD
158 #define NI_NAMERQD 4
159 #endif
160
161 #ifndef NI_NUMERICSERV
162 #define NI_NUMERICSERV 8
163 #endif
164
165 #ifndef NI_DGRAM
166 #define NI_DGRAM 16
167 #endif
168
169 #if !HAVE_GETADDRINFO
170 int ff_getaddrinfo(const char *node, const char *service,
171                    const struct addrinfo *hints, struct addrinfo **res);
172 void ff_freeaddrinfo(struct addrinfo *res);
173 int ff_getnameinfo(const struct sockaddr *sa, int salen,
174                    char *host, int hostlen,
175                    char *serv, int servlen, int flags);
176 const char *ff_gai_strerror(int ecode);
177 #define getaddrinfo ff_getaddrinfo
178 #define freeaddrinfo ff_freeaddrinfo
179 #define getnameinfo ff_getnameinfo
180 #define gai_strerror ff_gai_strerror
181 #endif
182
183 #ifndef INET6_ADDRSTRLEN
184 #define INET6_ADDRSTRLEN INET_ADDRSTRLEN
185 #endif
186
187 #ifndef IN_MULTICAST
188 #define IN_MULTICAST(a) ((((uint32_t)(a)) & 0xf0000000) == 0xe0000000)
189 #endif
190 #ifndef IN6_IS_ADDR_MULTICAST
191 #define IN6_IS_ADDR_MULTICAST(a) (((uint8_t *) (a))[0] == 0xff)
192 #endif
193
194 static inline int ff_is_multicast_address(struct sockaddr *addr)
195 {
196     if (addr->sa_family == AF_INET) {
197         return IN_MULTICAST(ntohl(((struct sockaddr_in *)addr)->sin_addr.s_addr));
198     }
199 #if HAVE_STRUCT_SOCKADDR_IN6
200     if (addr->sa_family == AF_INET6) {
201         return IN6_IS_ADDR_MULTICAST(&((struct sockaddr_in6 *)addr)->sin6_addr);
202     }
203 #endif
204
205     return 0;
206 }
207
208 #endif /* AVFORMAT_NETWORK_H */