1
0
mirror of https://gitlab.com/libvirt/libvirt.git synced 2024-12-25 01:34:11 +03:00
libvirt/tests/virportallocatormock.c
Daniel P. Berrangé fc920f704c src: convert all code to use virsocket.h
There are a large number of different header files that
are related to the sockets APIs. The virsocket.h header
includes all of the relevant headers for Windows and UNIX
in one convenient place. If virsocketaddr.h is already
included, then there's no need for virsocket.h

Reviewed-by: Pavel Hrdina <phrdina@redhat.com>
Signed-off-by: Daniel P. Berrangé <berrange@redhat.com>
2020-01-29 14:51:40 +00:00

103 lines
2.5 KiB
C

/*
* Copyright (C) 2013-2014 Red Hat, Inc.
*
* This library 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.
*
* This library 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 this library. If not, see
* <http://www.gnu.org/licenses/>.
*/
#include <config.h>
#if HAVE_DLFCN_H
# include <dlfcn.h>
#endif
#if defined(__linux__) && defined(RTLD_NEXT)
# include "virsocket.h"
# include <unistd.h>
static bool host_has_ipv6;
static int (*realsocket)(int domain, int type, int protocol);
static void init_syms(void)
{
int fd;
if (realsocket)
return;
realsocket = dlsym(RTLD_NEXT, "socket");
if (!realsocket) {
fprintf(stderr, "Unable to find 'socket' symbol\n");
abort();
}
fd = realsocket(AF_INET6, SOCK_STREAM, 0);
if (fd < 0)
return;
host_has_ipv6 = true;
close(fd);
}
int socket(int domain,
int type,
int protocol)
{
init_syms();
if (getenv("LIBVIRT_TEST_IPV4ONLY") && domain == AF_INET6) {
errno = EAFNOSUPPORT;
return -1;
}
return realsocket(domain, type, protocol);
}
int bind(int sockfd G_GNUC_UNUSED,
const struct sockaddr *addr,
socklen_t addrlen G_GNUC_UNUSED)
{
struct sockaddr_in saddr;
memcpy(&saddr, addr, sizeof(saddr));
if (host_has_ipv6 && !getenv("LIBVIRT_TEST_IPV4ONLY")) {
if (saddr.sin_port == htons(5900) ||
(saddr.sin_family == AF_INET &&
saddr.sin_port == htons(5904)) ||
(saddr.sin_family == AF_INET6 &&
(saddr.sin_port == htons(5905) ||
saddr.sin_port == htons(5906)))) {
errno = EADDRINUSE;
return -1;
}
return 0;
}
if (saddr.sin_port == htons(5900) ||
saddr.sin_port == htons(5904) ||
saddr.sin_port == htons(5905) ||
saddr.sin_port == htons(5906)) {
errno = EADDRINUSE;
return -1;
}
return 0;
}
#else /* defined(__linux__) && defined(RTLD_NEXT) */
/* Nothing to override on other platforms. */
#endif