RepoMirrors/openssh
1
0
Fork 0
mirror of git://anongit.mindrot.org/openssh.git synced 2024-12-27 04:12:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add compat implementation of ppoll using pselect.

Browse Source
This commit is contained in:
Darren Tucker 2021-11-18 23:11:37 +11:00
parent b544ce1ad4
commit 21dd5a9a3f
3 changed files with 69 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
configure.ac
View File

@ -1874,6 +1874,7 @@ AC_CHECK_FUNCS([ \
openlog_r \
pledge \
poll \
ppoll \
prctl \
procctl \
pselect \
@ -3547,6 +3548,26 @@ AC_RUN_IFELSE(
select_works_with_rlimit=yes]
)
AC_CHECK_MEMBERS([struct pollfd.fd], [], [], [[
#include <sys/types.h>
#ifdef HAVE_POLL_H
#include <poll.h>
#endif
#ifdef HAVE_SYS_POLL_H
#include <sys/poll.h>
#endif
]])
AC_CHECK_TYPES([nfds_t], , , [
#include <sys/types.h>
#ifdef HAVE_POLL_H
#include <poll.h>
#endif
#ifdef HAVE_SYS_POLL_H
#include <sys/poll.h>
#endif
])
AC_MSG_CHECKING([if setrlimit(RLIMIT_NOFILE,{0,0}) works])
AC_RUN_IFELSE(
[AC_LANG_PROGRAM([[

40
openbsd-compat/bsd-poll.c
View File

@ -15,7 +15,7 @@
*/
#include "includes.h"
#if !defined(HAVE_POLL)
#if !defined(HAVE_PPOLL) || !defined(HAVE_POLL)
#include <sys/types.h>
#include <sys/time.h>
@ -24,12 +24,14 @@
#endif
#include <errno.h>
#include <signal.h>
#include <stdlib.h>
#include <unistd.h>
#include "bsd-poll.h"
#ifndef HAVE_PPOLL
/*
* A minimal implementation of poll(2), built on top of select(2).
* A minimal implementation of ppoll(2), built on top of pselect(2).
*
* Only supports POLLIN and POLLOUT flags in pfd.events, and POLLIN, POLLOUT
* and POLLERR flags in revents.
@ -38,13 +40,13 @@
*/
int
poll(struct pollfd *fds, nfds_t nfds, int timeout)
ppoll(struct pollfd *fds, nfds_t nfds, const struct timespec *tmoutp,
const sigset_t *sigmask)
{
nfds_t i;
int saved_errno, ret, fd, maxfd = 0;
fd_set *readfds = NULL, *writefds = NULL, *exceptfds = NULL;
size_t nmemb;
struct timeval tv, *tvp = NULL;
for (i = 0; i < nfds; i++) {
fd = fds[i].fd;
@ -79,14 +81,7 @@ poll(struct pollfd *fds, nfds_t nfds, int timeout)
}
}
/* poll timeout is msec, select is timeval (sec + usec) */
if (timeout >= 0) {
tv.tv_sec = timeout / 1000;
tv.tv_usec = (timeout % 1000) * 1000;
tvp = &tv;
}
ret = select(maxfd + 1, readfds, writefds, exceptfds, tvp);
ret = pselect(maxfd + 1, readfds, writefds, exceptfds, tmoutp, sigmask);
saved_errno = errno;
/* scan through select results and set poll() flags */
@ -114,4 +109,23 @@ out:
errno = saved_errno;
return ret;
}
#endif
#endif /* HAVE_PPOLL */
#ifdef HAVE_POLL
int
poll(struct pollfd *fds, nfds_t nfds, int timeout)
{
struct timespec ts, *tsp = NULL;
/* poll timeout is msec, ppoll is timespec (sec + nsec) */
if (timeout >= 0) {
ts.tv_sec = timeout / 1000;
ts.tv_nsec = (timeout % 1000000) * 1000000;
tsp = &ts;
}
return ppoll(fds, nfds, tsp, NULL);
}
#endif /* HAVE_POLL */
#endif /* HAVE_PPOLL || HAVE_POLL */

25
openbsd-compat/bsd-poll.h
View File

@ -27,18 +27,24 @@
/* OPENBSD ORIGINAL: sys/sys/poll.h */
#if !defined(HAVE_POLL) && !defined(HAVE_POLL_H)
#ifndef _COMPAT_POLL_H_
#define _COMPAT_POLL_H_
#include <sys/types.h>
#ifdef HAVE_POLL_H
# include <poll.h>
#endif
#ifdef HAVE_SYS_POLL_H
# include <sys/poll.h>
#endif
#ifndef HAVE_STRUCT_POLLFD_FD
typedef struct pollfd {
int fd;
short events;
short revents;
} pollfd_t;
typedef unsigned int nfds_t;
#define POLLIN 0x0001
#define POLLOUT 0x0004
#define POLLERR 0x0008
@ -55,7 +61,18 @@ typedef unsigned int nfds_t;
#endif
#define INFTIM (-1) /* not standard */
#endif /* !HAVE_STRUCT_POLLFD_FD */
#ifndef HAVE_NFDS_T
typedef unsigned int nfds_t;
#endif
#ifndef HAVE_POLL
int poll(struct pollfd *, nfds_t, int);
#endif
#ifndef HAVE_PPOLL
int ppoll(struct pollfd *, nfds_t, const struct timespec *, const sigset_t *);
#endif
#endif /* !_COMPAT_POLL_H_ */
#endif /* !HAVE_POLL_H */