mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-18 03:19:31 +02:00
f2c85458c4
Previously, the function would lead to an infinite wait (by returning AVERROR(EAGAIN)) on sockets indicating an error via either of these poll flags. Signed-off-by: Martin Storsjö <martin@martin.st>
205 lines
4.4 KiB
C
205 lines
4.4 KiB
C
/*
|
|
* Copyright (c) 2007 The Libav Project
|
|
*
|
|
* This file is part of Libav.
|
|
*
|
|
* Libav 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.
|
|
*
|
|
* Libav 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 Libav; 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 "config.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() {
|
|
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 */
|