00daac1553
* src/posix-io.c (get_max_fds): use getdents64 instead of getdents. -- getdents64 was introduced in linux 2.4, so it should be widely available. some Linux architectures which post-date 2.4 (e.g. arm64) appear to not have getdents at all, so it's probably better to use the more modern interface. Signed-off-by: Daniel Kahn Gillmor <dkg@fifthhorseman.net>
876 lines
20 KiB
C
876 lines
20 KiB
C
/* posix-io.c - Posix I/O functions
|
||
Copyright (C) 2000 Werner Koch (dd9jn)
|
||
Copyright (C) 2001, 2002, 2004, 2005, 2007, 2010 g10 Code GmbH
|
||
|
||
This file is part of GPGME.
|
||
|
||
GPGME 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.
|
||
|
||
GPGME 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/>.
|
||
*/
|
||
|
||
#ifdef HAVE_CONFIG_H
|
||
#include <config.h>
|
||
#endif
|
||
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#ifdef HAVE_STDINT_H
|
||
# include <stdint.h>
|
||
#endif
|
||
#include <string.h>
|
||
#include <assert.h>
|
||
#include <errno.h>
|
||
#include <signal.h>
|
||
#include <fcntl.h>
|
||
#ifdef HAVE_UNISTD_H
|
||
# include <unistd.h>
|
||
#endif
|
||
#ifdef HAVE_SYS_TIME_H
|
||
# include <sys/time.h>
|
||
#endif
|
||
#ifdef HAVE_SYS_TYPES_H
|
||
# include <sys/types.h>
|
||
#endif
|
||
#include <sys/wait.h>
|
||
#ifdef HAVE_SYS_UIO_H
|
||
# include <sys/uio.h>
|
||
#endif
|
||
#include <ctype.h>
|
||
#include <sys/resource.h>
|
||
|
||
#ifdef USE_LINUX_GETDENTS
|
||
# include <sys/syscall.h>
|
||
# include <sys/types.h>
|
||
# include <dirent.h>
|
||
#endif /*USE_LINUX_GETDENTS*/
|
||
|
||
|
||
#include "util.h"
|
||
#include "priv-io.h"
|
||
#include "sema.h"
|
||
#include "ath.h"
|
||
#include "debug.h"
|
||
|
||
|
||
|
||
void
|
||
_gpgme_io_subsystem_init (void)
|
||
{
|
||
struct sigaction act;
|
||
|
||
sigaction (SIGPIPE, NULL, &act);
|
||
if (act.sa_handler == SIG_DFL)
|
||
{
|
||
act.sa_handler = SIG_IGN;
|
||
sigemptyset (&act.sa_mask);
|
||
act.sa_flags = 0;
|
||
sigaction (SIGPIPE, &act, NULL);
|
||
}
|
||
}
|
||
|
||
|
||
/* Write the printable version of FD to the buffer BUF of length
|
||
BUFLEN. The printable version is the representation on the command
|
||
line that the child process expects. */
|
||
int
|
||
_gpgme_io_fd2str (char *buf, int buflen, int fd)
|
||
{
|
||
return snprintf (buf, buflen, "%d", fd);
|
||
}
|
||
|
||
|
||
/* The table to hold notification handlers. We use a linear search
|
||
and extend the table as needed. */
|
||
struct notify_table_item_s
|
||
{
|
||
int fd; /* -1 indicates an unused entry. */
|
||
_gpgme_close_notify_handler_t handler;
|
||
void *value;
|
||
};
|
||
typedef struct notify_table_item_s *notify_table_item_t;
|
||
|
||
static notify_table_item_t notify_table;
|
||
static size_t notify_table_size;
|
||
DEFINE_STATIC_LOCK (notify_table_lock);
|
||
|
||
|
||
|
||
int
|
||
_gpgme_io_read (int fd, void *buffer, size_t count)
|
||
{
|
||
int nread;
|
||
TRACE_BEG2 (DEBUG_SYSIO, "_gpgme_io_read", fd,
|
||
"buffer=%p, count=%u", buffer, count);
|
||
|
||
do
|
||
{
|
||
nread = _gpgme_ath_read (fd, buffer, count);
|
||
}
|
||
while (nread == -1 && errno == EINTR);
|
||
|
||
TRACE_LOGBUF (buffer, nread);
|
||
return TRACE_SYSRES (nread);
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_write (int fd, const void *buffer, size_t count)
|
||
{
|
||
int nwritten;
|
||
TRACE_BEG2 (DEBUG_SYSIO, "_gpgme_io_write", fd,
|
||
"buffer=%p, count=%u", buffer, count);
|
||
TRACE_LOGBUF (buffer, count);
|
||
|
||
do
|
||
{
|
||
nwritten = _gpgme_ath_write (fd, buffer, count);
|
||
}
|
||
while (nwritten == -1 && errno == EINTR);
|
||
|
||
return TRACE_SYSRES (nwritten);
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_pipe (int filedes[2], int inherit_idx)
|
||
{
|
||
int saved_errno;
|
||
int err;
|
||
TRACE_BEG2 (DEBUG_SYSIO, "_gpgme_io_pipe", filedes,
|
||
"inherit_idx=%i (GPGME uses it for %s)",
|
||
inherit_idx, inherit_idx ? "reading" : "writing");
|
||
|
||
err = pipe (filedes);
|
||
if (err < 0)
|
||
return TRACE_SYSRES (err);
|
||
|
||
/* FIXME: Should get the old flags first. */
|
||
err = fcntl (filedes[1 - inherit_idx], F_SETFD, FD_CLOEXEC);
|
||
saved_errno = errno;
|
||
if (err < 0)
|
||
{
|
||
close (filedes[0]);
|
||
close (filedes[1]);
|
||
}
|
||
errno = saved_errno;
|
||
if (err)
|
||
return TRACE_SYSRES (err);
|
||
|
||
return TRACE_SUC2 ("read=0x%x, write=0x%x", filedes[0], filedes[1]);
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_close (int fd)
|
||
{
|
||
int res;
|
||
_gpgme_close_notify_handler_t handler = NULL;
|
||
void *handler_value;
|
||
int idx;
|
||
|
||
TRACE_BEG (DEBUG_SYSIO, "_gpgme_io_close", fd);
|
||
|
||
if (fd == -1)
|
||
{
|
||
errno = EINVAL;
|
||
return TRACE_SYSRES (-1);
|
||
}
|
||
|
||
/* First call the notify handler. */
|
||
LOCK (notify_table_lock);
|
||
for (idx=0; idx < notify_table_size; idx++)
|
||
{
|
||
if (notify_table[idx].fd == fd)
|
||
{
|
||
handler = notify_table[idx].handler;
|
||
handler_value = notify_table[idx].value;
|
||
notify_table[idx].handler = NULL;
|
||
notify_table[idx].value = NULL;
|
||
notify_table[idx].fd = -1; /* Mark slot as free. */
|
||
break;
|
||
}
|
||
}
|
||
UNLOCK (notify_table_lock);
|
||
if (handler)
|
||
{
|
||
TRACE_LOG2 ("invoking close handler %p/%p", handler, handler_value);
|
||
handler (fd, handler_value);
|
||
}
|
||
|
||
/* Then do the close. */
|
||
res = close (fd);
|
||
return TRACE_SYSRES (res);
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_set_close_notify (int fd, _gpgme_close_notify_handler_t handler,
|
||
void *value)
|
||
{
|
||
int res = 0;
|
||
int idx;
|
||
|
||
TRACE_BEG2 (DEBUG_SYSIO, "_gpgme_io_set_close_notify", fd,
|
||
"close_handler=%p/%p", handler, value);
|
||
|
||
assert (fd != -1);
|
||
|
||
LOCK (notify_table_lock);
|
||
for (idx=0; idx < notify_table_size; idx++)
|
||
if (notify_table[idx].fd == -1)
|
||
break;
|
||
if (idx == notify_table_size)
|
||
{
|
||
/* We need to increase the size of the table. The approach we
|
||
take is straightforward to minimize the risk of bugs. */
|
||
notify_table_item_t newtbl;
|
||
size_t newsize = notify_table_size + 64;
|
||
|
||
newtbl = calloc (newsize, sizeof *newtbl);
|
||
if (!newtbl)
|
||
{
|
||
res = -1;
|
||
goto leave;
|
||
}
|
||
for (idx=0; idx < notify_table_size; idx++)
|
||
newtbl[idx] = notify_table[idx];
|
||
for (; idx < newsize; idx++)
|
||
{
|
||
newtbl[idx].fd = -1;
|
||
newtbl[idx].handler = NULL;
|
||
newtbl[idx].value = NULL;
|
||
}
|
||
free (notify_table);
|
||
notify_table = newtbl;
|
||
idx = notify_table_size;
|
||
notify_table_size = newsize;
|
||
}
|
||
notify_table[idx].fd = fd;
|
||
notify_table[idx].handler = handler;
|
||
notify_table[idx].value = value;
|
||
|
||
leave:
|
||
UNLOCK (notify_table_lock);
|
||
|
||
return TRACE_SYSRES (res);
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_set_nonblocking (int fd)
|
||
{
|
||
int flags;
|
||
int res;
|
||
TRACE_BEG (DEBUG_SYSIO, "_gpgme_io_set_nonblocking", fd);
|
||
|
||
flags = fcntl (fd, F_GETFL, 0);
|
||
if (flags == -1)
|
||
return TRACE_SYSRES (-1);
|
||
flags |= O_NONBLOCK;
|
||
res = fcntl (fd, F_SETFL, flags);
|
||
return TRACE_SYSRES (res);
|
||
}
|
||
|
||
|
||
#ifdef USE_LINUX_GETDENTS
|
||
/* This is not declared in public headers; getdents64(2) says that we must
|
||
* define it ourselves. */
|
||
struct linux_dirent64
|
||
{
|
||
ino64_t d_ino;
|
||
off64_t d_off;
|
||
unsigned short d_reclen;
|
||
unsigned char d_type;
|
||
char d_name[];
|
||
};
|
||
|
||
# define DIR_BUF_SIZE 1024
|
||
#endif /*USE_LINUX_GETDENTS*/
|
||
|
||
|
||
static long int
|
||
get_max_fds (void)
|
||
{
|
||
const char *source = NULL;
|
||
long int fds = -1;
|
||
int rc;
|
||
|
||
/* Under Linux we can figure out the highest used file descriptor by
|
||
* reading /proc/self/fd. This is in the common cases much faster
|
||
* than for example doing 4096 close calls where almost all of them
|
||
* will fail.
|
||
*
|
||
* We can't use the normal opendir/readdir/closedir interface between
|
||
* fork and exec in a multi-threaded process because opendir uses
|
||
* malloc and thus a mutex which may deadlock with a malloc in another
|
||
* thread. However, the underlying getdents system call is safe. */
|
||
#ifdef USE_LINUX_GETDENTS
|
||
{
|
||
int dir_fd;
|
||
char dir_buf[DIR_BUF_SIZE];
|
||
struct linux_dirent64 *dir_entry;
|
||
int r, pos;
|
||
const char *s;
|
||
int x;
|
||
|
||
dir_fd = open ("/proc/self/fd", O_RDONLY | O_DIRECTORY);
|
||
if (dir_fd != -1)
|
||
{
|
||
for (;;)
|
||
{
|
||
r = syscall(SYS_getdents64, dir_fd, dir_buf, DIR_BUF_SIZE);
|
||
if (r == -1)
|
||
{
|
||
/* Fall back to other methods. */
|
||
fds = -1;
|
||
break;
|
||
}
|
||
if (r == 0)
|
||
break;
|
||
|
||
for (pos = 0; pos < r; pos += dir_entry->d_reclen)
|
||
{
|
||
dir_entry = (struct linux_dirent64 *) (dir_buf + pos);
|
||
s = dir_entry->d_name;
|
||
if (*s < '0' || *s > '9')
|
||
continue;
|
||
/* atoi is not guaranteed to be async-signal-safe. */
|
||
for (x = 0; *s >= '0' && *s <= '9'; s++)
|
||
x = x * 10 + (*s - '0');
|
||
if (!*s && x > fds && x != dir_fd)
|
||
fds = x;
|
||
}
|
||
}
|
||
|
||
close (dir_fd);
|
||
}
|
||
if (fds != -1)
|
||
{
|
||
fds++;
|
||
source = "/proc";
|
||
}
|
||
}
|
||
#endif /*USE_LINUX_GETDENTS*/
|
||
|
||
#ifdef RLIMIT_NOFILE
|
||
if (fds == -1)
|
||
{
|
||
struct rlimit rl;
|
||
rc = getrlimit (RLIMIT_NOFILE, &rl);
|
||
if (rc == 0)
|
||
{
|
||
source = "RLIMIT_NOFILE";
|
||
fds = rl.rlim_max;
|
||
}
|
||
}
|
||
#endif
|
||
#ifdef RLIMIT_OFILE
|
||
if (fds == -1)
|
||
{
|
||
struct rlimit rl;
|
||
rc = getrlimit (RLIMIT_OFILE, &rl);
|
||
if (rc == 0)
|
||
{
|
||
source = "RLIMIT_OFILE";
|
||
fds = rl.rlim_max;
|
||
}
|
||
}
|
||
#endif
|
||
#ifdef _SC_OPEN_MAX
|
||
if (fds == -1)
|
||
{
|
||
long int scres;
|
||
scres = sysconf (_SC_OPEN_MAX);
|
||
if (scres >= 0)
|
||
{
|
||
source = "_SC_OPEN_MAX";
|
||
return scres;
|
||
}
|
||
}
|
||
#endif
|
||
#ifdef OPEN_MAX
|
||
if (fds == -1)
|
||
{
|
||
source = "OPEN_MAX";
|
||
fds = OPEN_MAX;
|
||
}
|
||
#endif
|
||
|
||
#if !defined(RLIMIT_NOFILE) && !defined(RLIMIT_OFILE) \
|
||
&& !defined(_SC_OPEN_MAX) && !defined(OPEN_MAX)
|
||
#warning "No known way to get the maximum number of file descriptors."
|
||
#endif
|
||
if (fds == -1)
|
||
{
|
||
source = "arbitrary";
|
||
/* Arbitrary limit. */
|
||
fds = 1024;
|
||
}
|
||
|
||
/* AIX returns INT32_MAX instead of a proper value. We assume that
|
||
* this is always an error and use a more reasonable limit. */
|
||
#ifdef INT32_MAX
|
||
if (fds == INT32_MAX)
|
||
{
|
||
source = "aix-fix";
|
||
fds = 1024;
|
||
}
|
||
#endif
|
||
|
||
TRACE2 (DEBUG_SYSIO, "gpgme:max_fds", 0, "max fds=%i (%s)", fds, source);
|
||
return fds;
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_waitpid (int pid, int hang, int *r_status, int *r_signal)
|
||
{
|
||
int status;
|
||
pid_t ret;
|
||
|
||
*r_status = 0;
|
||
*r_signal = 0;
|
||
do
|
||
ret = _gpgme_ath_waitpid (pid, &status, hang? 0 : WNOHANG);
|
||
while (ret == (pid_t)(-1) && errno == EINTR);
|
||
|
||
if (ret == pid)
|
||
{
|
||
if (WIFSIGNALED (status))
|
||
{
|
||
*r_status = 4; /* Need some value here. */
|
||
*r_signal = WTERMSIG (status);
|
||
}
|
||
else if (WIFEXITED (status))
|
||
*r_status = WEXITSTATUS (status);
|
||
else
|
||
*r_status = 4; /* Oops. */
|
||
return 1;
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
|
||
/* Returns 0 on success, -1 on error. */
|
||
int
|
||
_gpgme_io_spawn (const char *path, char *const argv[], unsigned int flags,
|
||
struct spawn_fd_item_s *fd_list,
|
||
void (*atfork) (void *opaque, int reserved),
|
||
void *atforkvalue, pid_t *r_pid)
|
||
{
|
||
pid_t pid;
|
||
int i;
|
||
int status;
|
||
int signo;
|
||
|
||
TRACE_BEG1 (DEBUG_SYSIO, "_gpgme_io_spawn", path,
|
||
"path=%s", path);
|
||
i = 0;
|
||
while (argv[i])
|
||
{
|
||
TRACE_LOG2 ("argv[%2i] = %s", i, argv[i]);
|
||
i++;
|
||
}
|
||
for (i = 0; fd_list[i].fd != -1; i++)
|
||
if (fd_list[i].dup_to == -1)
|
||
TRACE_LOG2 ("fd[%i] = 0x%x", i, fd_list[i].fd);
|
||
else
|
||
TRACE_LOG3 ("fd[%i] = 0x%x -> 0x%x", i, fd_list[i].fd, fd_list[i].dup_to);
|
||
|
||
pid = fork ();
|
||
if (pid == -1)
|
||
return TRACE_SYSRES (-1);
|
||
|
||
if (!pid)
|
||
{
|
||
/* Intermediate child to prevent zombie processes. */
|
||
if ((pid = fork ()) == 0)
|
||
{
|
||
/* Child. */
|
||
int max_fds = -1;
|
||
int fd;
|
||
int seen_stdin = 0;
|
||
int seen_stdout = 0;
|
||
int seen_stderr = 0;
|
||
|
||
if (atfork)
|
||
atfork (atforkvalue, 0);
|
||
|
||
/* First close all fds which will not be inherited. If we
|
||
* have closefrom(2) we first figure out the highest fd we
|
||
* do not want to close, then call closefrom, and on success
|
||
* use the regular code to close all fds up to the start
|
||
* point of closefrom. Note that Solaris' and FreeBSD's closefrom do
|
||
* not return errors. */
|
||
#ifdef HAVE_CLOSEFROM
|
||
{
|
||
fd = -1;
|
||
for (i = 0; fd_list[i].fd != -1; i++)
|
||
if (fd_list[i].fd > fd)
|
||
fd = fd_list[i].fd;
|
||
fd++;
|
||
#if defined(__sun) || defined(__FreeBSD__)
|
||
closefrom (fd);
|
||
max_fds = fd;
|
||
#else /*!__sun */
|
||
while ((i = closefrom (fd)) && errno == EINTR)
|
||
;
|
||
if (!i || errno == EBADF)
|
||
max_fds = fd;
|
||
#endif /*!__sun*/
|
||
}
|
||
#endif /*HAVE_CLOSEFROM*/
|
||
if (max_fds == -1)
|
||
max_fds = get_max_fds ();
|
||
for (fd = 0; fd < max_fds; fd++)
|
||
{
|
||
for (i = 0; fd_list[i].fd != -1; i++)
|
||
if (fd_list[i].fd == fd)
|
||
break;
|
||
if (fd_list[i].fd == -1)
|
||
close (fd);
|
||
}
|
||
|
||
/* And now dup and close those to be duplicated. */
|
||
for (i = 0; fd_list[i].fd != -1; i++)
|
||
{
|
||
int child_fd;
|
||
int res;
|
||
|
||
if (fd_list[i].dup_to != -1)
|
||
child_fd = fd_list[i].dup_to;
|
||
else
|
||
child_fd = fd_list[i].fd;
|
||
|
||
if (child_fd == 0)
|
||
seen_stdin = 1;
|
||
else if (child_fd == 1)
|
||
seen_stdout = 1;
|
||
else if (child_fd == 2)
|
||
seen_stderr = 1;
|
||
|
||
if (fd_list[i].dup_to == -1)
|
||
continue;
|
||
|
||
res = dup2 (fd_list[i].fd, fd_list[i].dup_to);
|
||
if (res < 0)
|
||
{
|
||
#if 0
|
||
/* FIXME: The debug file descriptor is not
|
||
dup'ed anyway, so we can't see this. */
|
||
TRACE_LOG1 ("dup2 failed in child: %s\n",
|
||
strerror (errno));
|
||
#endif
|
||
_exit (8);
|
||
}
|
||
|
||
close (fd_list[i].fd);
|
||
}
|
||
|
||
if (! seen_stdin || ! seen_stdout || !seen_stderr)
|
||
{
|
||
fd = open ("/dev/null", O_RDWR);
|
||
if (fd == -1)
|
||
{
|
||
/* The debug file descriptor is not dup'ed, so we
|
||
can't do a trace output. */
|
||
_exit (8);
|
||
}
|
||
/* Make sure that the process has connected stdin. */
|
||
if (! seen_stdin && fd != 0)
|
||
{
|
||
if (dup2 (fd, 0) == -1)
|
||
_exit (8);
|
||
}
|
||
if (! seen_stdout && fd != 1)
|
||
{
|
||
if (dup2 (fd, 1) == -1)
|
||
_exit (8);
|
||
}
|
||
if (! seen_stderr && fd != 2)
|
||
{
|
||
if (dup2 (fd, 2) == -1)
|
||
_exit (8);
|
||
}
|
||
if (fd != 0 && fd != 1 && fd != 2)
|
||
close (fd);
|
||
}
|
||
|
||
execv (path, (char *const *) argv);
|
||
/* Hmm: in that case we could write a special status code to the
|
||
status-pipe. */
|
||
_exit (8);
|
||
/* End child. */
|
||
}
|
||
if (pid == -1)
|
||
_exit (1);
|
||
else
|
||
_exit (0);
|
||
}
|
||
|
||
TRACE_LOG1 ("waiting for child process pid=%i", pid);
|
||
_gpgme_io_waitpid (pid, 1, &status, &signo);
|
||
if (status)
|
||
return TRACE_SYSRES (-1);
|
||
|
||
for (i = 0; fd_list[i].fd != -1; i++)
|
||
{
|
||
if (! (flags & IOSPAWN_FLAG_NOCLOSE))
|
||
_gpgme_io_close (fd_list[i].fd);
|
||
/* No handle translation. */
|
||
fd_list[i].peer_name = fd_list[i].fd;
|
||
}
|
||
|
||
if (r_pid)
|
||
*r_pid = pid;
|
||
|
||
return TRACE_SYSRES (0);
|
||
}
|
||
|
||
|
||
/* Select on the list of fds. Returns: -1 = error, 0 = timeout or
|
||
nothing to select, > 0 = number of signaled fds. */
|
||
int
|
||
_gpgme_io_select (struct io_select_fd_s *fds, size_t nfds, int nonblock)
|
||
{
|
||
fd_set readfds;
|
||
fd_set writefds;
|
||
unsigned int i;
|
||
int any;
|
||
int max_fd;
|
||
int n;
|
||
int count;
|
||
/* Use a 1s timeout. */
|
||
struct timeval timeout = { 1, 0 };
|
||
void *dbg_help = NULL;
|
||
TRACE_BEG2 (DEBUG_SYSIO, "_gpgme_io_select", fds,
|
||
"nfds=%u, nonblock=%u", nfds, nonblock);
|
||
|
||
FD_ZERO (&readfds);
|
||
FD_ZERO (&writefds);
|
||
max_fd = 0;
|
||
if (nonblock)
|
||
timeout.tv_sec = 0;
|
||
|
||
TRACE_SEQ (dbg_help, "select on [ ");
|
||
|
||
any = 0;
|
||
for (i = 0; i < nfds; i++)
|
||
{
|
||
if (fds[i].fd == -1)
|
||
continue;
|
||
if (fds[i].for_read)
|
||
{
|
||
if (fds[i].fd >= FD_SETSIZE)
|
||
{
|
||
TRACE_END (dbg_help, " -BAD- ]");
|
||
gpg_err_set_errno (EMFILE);
|
||
return TRACE_SYSRES (-1);
|
||
}
|
||
assert (!FD_ISSET (fds[i].fd, &readfds));
|
||
FD_SET (fds[i].fd, &readfds);
|
||
if (fds[i].fd > max_fd)
|
||
max_fd = fds[i].fd;
|
||
TRACE_ADD1 (dbg_help, "r0x%x ", fds[i].fd);
|
||
any = 1;
|
||
}
|
||
else if (fds[i].for_write)
|
||
{
|
||
if (fds[i].fd >= FD_SETSIZE)
|
||
{
|
||
TRACE_END (dbg_help, " -BAD- ]");
|
||
gpg_err_set_errno (EMFILE);
|
||
return TRACE_SYSRES (-1);
|
||
}
|
||
assert (!FD_ISSET (fds[i].fd, &writefds));
|
||
FD_SET (fds[i].fd, &writefds);
|
||
if (fds[i].fd > max_fd)
|
||
max_fd = fds[i].fd;
|
||
TRACE_ADD1 (dbg_help, "w0x%x ", fds[i].fd);
|
||
any = 1;
|
||
}
|
||
fds[i].signaled = 0;
|
||
}
|
||
TRACE_END (dbg_help, "]");
|
||
if (!any)
|
||
return TRACE_SYSRES (0);
|
||
|
||
do
|
||
{
|
||
count = _gpgme_ath_select (max_fd + 1, &readfds, &writefds, NULL,
|
||
&timeout);
|
||
}
|
||
while (count < 0 && errno == EINTR);
|
||
if (count < 0)
|
||
return TRACE_SYSRES (-1);
|
||
|
||
TRACE_SEQ (dbg_help, "select OK [ ");
|
||
if (TRACE_ENABLED (dbg_help))
|
||
{
|
||
for (i = 0; i <= max_fd; i++)
|
||
{
|
||
if (FD_ISSET (i, &readfds))
|
||
TRACE_ADD1 (dbg_help, "r0x%x ", i);
|
||
if (FD_ISSET (i, &writefds))
|
||
TRACE_ADD1 (dbg_help, "w0x%x ", i);
|
||
}
|
||
TRACE_END (dbg_help, "]");
|
||
}
|
||
|
||
/* The variable N is used to optimize it a little bit. */
|
||
for (n = count, i = 0; i < nfds && n; i++)
|
||
{
|
||
if (fds[i].fd == -1)
|
||
;
|
||
else if (fds[i].for_read)
|
||
{
|
||
if (FD_ISSET (fds[i].fd, &readfds))
|
||
{
|
||
fds[i].signaled = 1;
|
||
n--;
|
||
}
|
||
}
|
||
else if (fds[i].for_write)
|
||
{
|
||
if (FD_ISSET (fds[i].fd, &writefds))
|
||
{
|
||
fds[i].signaled = 1;
|
||
n--;
|
||
}
|
||
}
|
||
}
|
||
return TRACE_SYSRES (count);
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_recvmsg (int fd, struct msghdr *msg, int flags)
|
||
{
|
||
int nread;
|
||
int saved_errno;
|
||
struct iovec *iov;
|
||
TRACE_BEG2 (DEBUG_SYSIO, "_gpgme_io_recvmsg", fd,
|
||
"msg=%p, flags=%i", msg, flags);
|
||
|
||
nread = 0;
|
||
iov = msg->msg_iov;
|
||
while (iov < msg->msg_iov + msg->msg_iovlen)
|
||
{
|
||
nread += iov->iov_len;
|
||
iov++;
|
||
}
|
||
|
||
TRACE_LOG1 ("about to receive %d bytes", nread);
|
||
|
||
do
|
||
{
|
||
nread = _gpgme_ath_recvmsg (fd, msg, flags);
|
||
}
|
||
while (nread == -1 && errno == EINTR);
|
||
saved_errno = errno;
|
||
if (nread > 0)
|
||
{
|
||
int nr = nread;
|
||
|
||
iov = msg->msg_iov;
|
||
while (nr > 0)
|
||
{
|
||
int len = nr > iov->iov_len ? iov->iov_len : nr;
|
||
TRACE_LOGBUF (msg->msg_iov->iov_base, len);
|
||
iov++;
|
||
nr -= len;
|
||
}
|
||
}
|
||
errno = saved_errno;
|
||
return TRACE_SYSRES (nread);
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_sendmsg (int fd, const struct msghdr *msg, int flags)
|
||
{
|
||
int nwritten;
|
||
struct iovec *iov;
|
||
TRACE_BEG2 (DEBUG_SYSIO, "_gpgme_io_sendmsg", fd,
|
||
"msg=%p, flags=%i", msg, flags);
|
||
|
||
nwritten = 0;
|
||
iov = msg->msg_iov;
|
||
while (iov < msg->msg_iov + msg->msg_iovlen)
|
||
{
|
||
nwritten += iov->iov_len;
|
||
iov++;
|
||
}
|
||
|
||
TRACE_LOG1 ("about to receive %d bytes", nwritten);
|
||
iov = msg->msg_iov;
|
||
while (nwritten > 0)
|
||
{
|
||
int len = nwritten > iov->iov_len ? iov->iov_len : nwritten;
|
||
TRACE_LOGBUF (msg->msg_iov->iov_base, len);
|
||
iov++;
|
||
nwritten -= len;
|
||
}
|
||
|
||
do
|
||
{
|
||
nwritten = _gpgme_ath_sendmsg (fd, msg, flags);
|
||
}
|
||
while (nwritten == -1 && errno == EINTR);
|
||
return TRACE_SYSRES (nwritten);
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_dup (int fd)
|
||
{
|
||
int new_fd;
|
||
|
||
do
|
||
new_fd = dup (fd);
|
||
while (new_fd == -1 && errno == EINTR);
|
||
|
||
TRACE1 (DEBUG_SYSIO, "_gpgme_io_dup", fd, "new fd==%i", new_fd);
|
||
|
||
return new_fd;
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_socket (int domain, int type, int proto)
|
||
{
|
||
int res;
|
||
|
||
TRACE_BEG2 (DEBUG_SYSIO, "_gpgme_io_socket", domain,
|
||
"type=%i, proto=%i", type, proto);
|
||
|
||
res = socket (domain, type, proto);
|
||
|
||
return TRACE_SYSRES (res);
|
||
}
|
||
|
||
|
||
int
|
||
_gpgme_io_connect (int fd, struct sockaddr *addr, int addrlen)
|
||
{
|
||
int res;
|
||
|
||
TRACE_BEG2 (DEBUG_SYSIO, "_gpgme_io_connect", fd,
|
||
"addr=%p, addrlen=%i", addr, addrlen);
|
||
|
||
do
|
||
res = ath_connect (fd, addr, addrlen);
|
||
while (res == -1 && errno == EINTR);
|
||
|
||
return TRACE_SYSRES (res);
|
||
}
|