85 lines
2.3 KiB
C
85 lines
2.3 KiB
C
/*
|
|
* test_mkfds - make various file descriptors
|
|
*
|
|
* Written by Masatake YAMATO <yamato@redhat.com>
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it would be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#ifndef TEST_MKFDS_H
|
|
#define TEST_MKFDS_H
|
|
|
|
#include <asm/unistd.h>
|
|
#include <stdbool.h>
|
|
#include <stddef.h>
|
|
|
|
enum multiplexing_mode {
|
|
MX_READ = 1 << 0,
|
|
MX_WRITE = 1 << 1,
|
|
MX_EXCEPT = 1 << 2,
|
|
};
|
|
|
|
struct fdesc {
|
|
int fd;
|
|
void (*close)(int, void *);
|
|
unsigned int mx_modes;
|
|
void *data;
|
|
};
|
|
|
|
#define DEFUN_WAIT_EVENT_POLL(NAME,SYSCALL,XDECLS,SETUP_SIG_HANDLER,SYSCALL_INVOCATION) \
|
|
void wait_event_##NAME(bool add_stdin, struct fdesc *fdescs, size_t n_fdescs) \
|
|
{ \
|
|
int n = add_stdin? 1: 0; \
|
|
int n0 = 0; \
|
|
struct pollfd *pfds = NULL; \
|
|
\
|
|
XDECLS \
|
|
\
|
|
for (size_t i = 0; i < n_fdescs; i++) \
|
|
if (fdescs[i].mx_modes) \
|
|
n++; \
|
|
\
|
|
pfds = xcalloc(n, sizeof(pfds[0])); \
|
|
\
|
|
for (size_t i = 0; i < n_fdescs; i++) { \
|
|
if (!fdescs[i].mx_modes) \
|
|
continue; \
|
|
pfds[n0].fd = fdescs[i].fd; \
|
|
if (fdescs[i].mx_modes & MX_READ) \
|
|
pfds[n0].events |= POLLIN; \
|
|
if (fdescs[i].mx_modes & MX_WRITE) \
|
|
pfds[n0].events |= POLLOUT; \
|
|
if (fdescs[i].mx_modes & MX_EXCEPT) \
|
|
pfds[n0].events |= POLLHUP; \
|
|
n0++; \
|
|
} \
|
|
\
|
|
if (add_stdin) { \
|
|
pfds[n0].fd = 0; \
|
|
pfds[n0].events |= POLLIN; \
|
|
} \
|
|
\
|
|
SETUP_SIG_HANDLER \
|
|
\
|
|
if (SYSCALL_INVOCATION < 0 \
|
|
&& errno != EINTR) \
|
|
err(EXIT_FAILURE, "failed in " SYSCALL); \
|
|
free(pfds); \
|
|
}
|
|
|
|
#ifdef __NR_ppoll
|
|
void wait_event_ppoll(bool add_stdin, struct fdesc *fdescs, size_t n_fdescs);
|
|
#endif
|
|
|
|
#endif /* TEST_MKFDS_H */
|