mirror of
https://github.com/libguestfs/libguestfs.git
synced 2026-03-21 22:53:37 +00:00
e2c7bddf10
Run this command across the source:
perl -pi.bak -e 's/(20[012][0-9])-20[12][012]/$1-2023/g' `git ls-files`
and remove changes to po{,-docs}/*.po{,t} (these will be regenerated
later when we run 'make dist').
140 lines
3.5 KiB
C
140 lines
3.5 KiB
C
/* Accept a connection on a socket, with specific opening flags.
|
|
Copyright (C) 2009-2023 Free Software Foundation, Inc.
|
|
|
|
This file 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 3 of the
|
|
License, or (at your option) any later version.
|
|
|
|
This file 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 program. If not, see <https://www.gnu.org/licenses/>. */
|
|
|
|
#include <config.h>
|
|
|
|
#ifndef HAVE_ACCEPT4
|
|
|
|
/* Specification. */
|
|
#include <sys/socket.h>
|
|
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#if GNULIB_MSVC_NOTHROW
|
|
# include "msvc-nothrow.h"
|
|
#endif
|
|
|
|
#ifndef SOCK_CLOEXEC
|
|
# define SOCK_CLOEXEC 0
|
|
#endif
|
|
#ifndef SOCK_NONBLOCK
|
|
# define SOCK_NONBLOCK 0
|
|
#endif
|
|
|
|
#ifndef O_TEXT
|
|
#define O_TEXT 0
|
|
#endif
|
|
#ifndef O_BINARY
|
|
#define O_BINARY 0
|
|
#endif
|
|
|
|
int
|
|
accept4 (int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags)
|
|
{
|
|
int fd;
|
|
|
|
/* Check the supported flags. */
|
|
if ((flags & ~(SOCK_CLOEXEC | SOCK_NONBLOCK | O_TEXT | O_BINARY)) != 0)
|
|
{
|
|
errno = EINVAL;
|
|
return -1;
|
|
}
|
|
|
|
fd = accept (sockfd, addr, addrlen);
|
|
if (fd < 0)
|
|
return -1;
|
|
|
|
#if SOCK_CLOEXEC
|
|
# if defined _WIN32 && ! defined __CYGWIN__
|
|
/* Native Windows API. */
|
|
if (flags & SOCK_CLOEXEC)
|
|
{
|
|
HANDLE curr_process = GetCurrentProcess ();
|
|
HANDLE old_handle = (HANDLE) _get_osfhandle (fd);
|
|
HANDLE new_handle;
|
|
int nfd;
|
|
|
|
if (!DuplicateHandle (curr_process, /* SourceProcessHandle */
|
|
old_handle, /* SourceHandle */
|
|
curr_process, /* TargetProcessHandle */
|
|
(PHANDLE) &new_handle, /* TargetHandle */
|
|
(DWORD) 0, /* DesiredAccess */
|
|
FALSE, /* InheritHandle */
|
|
DUPLICATE_SAME_ACCESS)) /* Options */
|
|
{
|
|
close (fd);
|
|
errno = EBADF; /* arbitrary */
|
|
return -1;
|
|
}
|
|
|
|
/* Closing fd before allocating the new fd ensures that the new fd will
|
|
have the minimum possible value. */
|
|
close (fd);
|
|
nfd = _open_osfhandle ((intptr_t) new_handle,
|
|
O_NOINHERIT | (flags & (O_TEXT | O_BINARY)));
|
|
if (nfd < 0)
|
|
{
|
|
CloseHandle (new_handle);
|
|
return -1;
|
|
}
|
|
return nfd;
|
|
}
|
|
# else
|
|
/* Unix API. */
|
|
if (flags & SOCK_CLOEXEC)
|
|
{
|
|
int fcntl_flags;
|
|
|
|
if ((fcntl_flags = fcntl (fd, F_GETFD, 0)) < 0
|
|
|| fcntl (fd, F_SETFD, fcntl_flags | FD_CLOEXEC) == -1)
|
|
{
|
|
int saved_errno = errno;
|
|
close (fd);
|
|
errno = saved_errno;
|
|
return -1;
|
|
}
|
|
}
|
|
# endif
|
|
#endif
|
|
|
|
#if SOCK_NONBLOCK
|
|
if (flags & SOCK_NONBLOCK)
|
|
{
|
|
int fcntl_flags;
|
|
|
|
if ((fcntl_flags = fcntl (fd, F_GETFL, 0)) < 0
|
|
|| fcntl (fd, F_SETFL, fcntl_flags | O_NONBLOCK) == -1)
|
|
{
|
|
int saved_errno = errno;
|
|
close (fd);
|
|
errno = saved_errno;
|
|
return -1;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
#if O_BINARY
|
|
if (flags & O_BINARY)
|
|
set_binary_mode (fd, O_BINARY);
|
|
else if (flags & O_TEXT)
|
|
set_binary_mode (fd, O_TEXT);
|
|
#endif
|
|
|
|
return fd;
|
|
}
|
|
|
|
#endif /* !HAVE_ACCEPT4 */
|