1*57718be8SEnji Cooper /* $NetBSD: h_client.c,v 1.8 2012/04/20 05:15:11 jruoho Exp $ */
2*57718be8SEnji Cooper
3*57718be8SEnji Cooper /*
4*57718be8SEnji Cooper * Copyright (c) 2011 The NetBSD Foundation, Inc.
5*57718be8SEnji Cooper * All rights reserved.
6*57718be8SEnji Cooper *
7*57718be8SEnji Cooper * Redistribution and use in source and binary forms, with or without
8*57718be8SEnji Cooper * modification, are permitted provided that the following conditions
9*57718be8SEnji Cooper * are met:
10*57718be8SEnji Cooper * 1. Redistributions of source code must retain the above copyright
11*57718be8SEnji Cooper * notice, this list of conditions and the following disclaimer.
12*57718be8SEnji Cooper * 2. Redistributions in binary form must reproduce the above copyright
13*57718be8SEnji Cooper * notice, this list of conditions and the following disclaimer in the
14*57718be8SEnji Cooper * documentation and/or other materials provided with the distribution.
15*57718be8SEnji Cooper *
16*57718be8SEnji Cooper * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND
17*57718be8SEnji Cooper * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
18*57718be8SEnji Cooper * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19*57718be8SEnji Cooper * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20*57718be8SEnji Cooper * IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY
21*57718be8SEnji Cooper * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22*57718be8SEnji Cooper * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23*57718be8SEnji Cooper * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24*57718be8SEnji Cooper * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25*57718be8SEnji Cooper * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26*57718be8SEnji Cooper * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27*57718be8SEnji Cooper * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28*57718be8SEnji Cooper */
29*57718be8SEnji Cooper
30*57718be8SEnji Cooper #include <sys/types.h>
31*57718be8SEnji Cooper #include <sys/poll.h>
32*57718be8SEnji Cooper #include <sys/select.h>
33*57718be8SEnji Cooper
34*57718be8SEnji Cooper #include <err.h>
35*57718be8SEnji Cooper #include <errno.h>
36*57718be8SEnji Cooper #include <fcntl.h>
37*57718be8SEnji Cooper #include <stdio.h>
38*57718be8SEnji Cooper #include <stdlib.h>
39*57718be8SEnji Cooper #include <string.h>
40*57718be8SEnji Cooper #include <unistd.h>
41*57718be8SEnji Cooper
42*57718be8SEnji Cooper int
main(int argc,char * argv[])43*57718be8SEnji Cooper main(int argc, char *argv[])
44*57718be8SEnji Cooper {
45*57718be8SEnji Cooper
46*57718be8SEnji Cooper if (argc != 2) {
47*57718be8SEnji Cooper errx(1, "need testname as param");
48*57718be8SEnji Cooper }
49*57718be8SEnji Cooper
50*57718be8SEnji Cooper if (strcmp(argv[1], "select_timeout") == 0) {
51*57718be8SEnji Cooper fd_set rfds;
52*57718be8SEnji Cooper struct timeval tv;
53*57718be8SEnji Cooper int pipefd[2];
54*57718be8SEnji Cooper int rv;
55*57718be8SEnji Cooper
56*57718be8SEnji Cooper tv.tv_sec = 0;
57*57718be8SEnji Cooper tv.tv_usec = 1;
58*57718be8SEnji Cooper
59*57718be8SEnji Cooper if (pipe(pipefd) == -1)
60*57718be8SEnji Cooper err(EXIT_FAILURE, "pipe");
61*57718be8SEnji Cooper FD_ZERO(&rfds);
62*57718be8SEnji Cooper FD_SET(pipefd[0], &rfds);
63*57718be8SEnji Cooper
64*57718be8SEnji Cooper rv = select(pipefd[0]+1, &rfds, NULL, NULL, &tv);
65*57718be8SEnji Cooper if (rv == -1)
66*57718be8SEnji Cooper err(EXIT_FAILURE, "select");
67*57718be8SEnji Cooper if (rv != 0)
68*57718be8SEnji Cooper errx(EXIT_FAILURE, "select succesful");
69*57718be8SEnji Cooper
70*57718be8SEnji Cooper if (FD_ISSET(pipefd[0], &rfds))
71*57718be8SEnji Cooper errx(EXIT_FAILURE, "stdin fileno is still set");
72*57718be8SEnji Cooper return EXIT_SUCCESS;
73*57718be8SEnji Cooper } else if (strcmp(argv[1], "select_allunset") == 0) {
74*57718be8SEnji Cooper fd_set fds;
75*57718be8SEnji Cooper struct timeval tv;
76*57718be8SEnji Cooper int rv;
77*57718be8SEnji Cooper
78*57718be8SEnji Cooper tv.tv_sec = 0;
79*57718be8SEnji Cooper tv.tv_usec = 1;
80*57718be8SEnji Cooper
81*57718be8SEnji Cooper FD_ZERO(&fds);
82*57718be8SEnji Cooper
83*57718be8SEnji Cooper rv = select(100, &fds, &fds, &fds, &tv);
84*57718be8SEnji Cooper if (rv == -1)
85*57718be8SEnji Cooper err(EXIT_FAILURE, "select");
86*57718be8SEnji Cooper if (rv != 0)
87*57718be8SEnji Cooper errx(EXIT_FAILURE, "select succesful");
88*57718be8SEnji Cooper
89*57718be8SEnji Cooper rv = select(0, NULL, NULL, NULL, &tv);
90*57718be8SEnji Cooper if (rv == -1)
91*57718be8SEnji Cooper err(EXIT_FAILURE, "select2");
92*57718be8SEnji Cooper if (rv != 0)
93*57718be8SEnji Cooper errx(EXIT_FAILURE, "select2 succesful");
94*57718be8SEnji Cooper
95*57718be8SEnji Cooper return EXIT_SUCCESS;
96*57718be8SEnji Cooper } else if (strcmp(argv[1], "invafd") == 0) {
97*57718be8SEnji Cooper struct pollfd pfd[2];
98*57718be8SEnji Cooper int fd, rv;
99*57718be8SEnji Cooper
100*57718be8SEnji Cooper fd = open("/rump/dev/null", O_RDWR);
101*57718be8SEnji Cooper if (fd == -1)
102*57718be8SEnji Cooper err(EXIT_FAILURE, "open");
103*57718be8SEnji Cooper close(fd);
104*57718be8SEnji Cooper
105*57718be8SEnji Cooper pfd[0].fd = STDIN_FILENO;
106*57718be8SEnji Cooper pfd[0].events = POLLIN;
107*57718be8SEnji Cooper pfd[1].fd = fd;
108*57718be8SEnji Cooper pfd[1].events = POLLIN;
109*57718be8SEnji Cooper
110*57718be8SEnji Cooper if ((rv = poll(pfd, 2, INFTIM)) != 1)
111*57718be8SEnji Cooper errx(EXIT_FAILURE, "poll unexpected rv %d (%d)",
112*57718be8SEnji Cooper rv, errno);
113*57718be8SEnji Cooper if (pfd[1].revents != POLLNVAL || pfd[0].revents != 0)
114*57718be8SEnji Cooper errx(EXIT_FAILURE, "poll unexpected revents");
115*57718be8SEnji Cooper
116*57718be8SEnji Cooper return EXIT_SUCCESS;
117*57718be8SEnji Cooper } else if (strcmp(argv[1], "fdoff8") == 0) {
118*57718be8SEnji Cooper
119*57718be8SEnji Cooper (void)closefrom(0);
120*57718be8SEnji Cooper
121*57718be8SEnji Cooper int fd;
122*57718be8SEnji Cooper
123*57718be8SEnji Cooper do {
124*57718be8SEnji Cooper if ((fd = open("/dev/null", O_RDWR)) == -1)
125*57718be8SEnji Cooper err(EXIT_FAILURE, "open1");
126*57718be8SEnji Cooper } while (fd < 7);
127*57718be8SEnji Cooper fd = open("/dev/null", O_RDWR);
128*57718be8SEnji Cooper if (fd != -1 || errno != ENFILE)
129*57718be8SEnji Cooper errx(EXIT_FAILURE, "unexpected fd8 %d %d", fd, errno);
130*57718be8SEnji Cooper if (fcntl(0, F_MAXFD) != 7)
131*57718be8SEnji Cooper errx(EXIT_FAILURE, "fd leak?");
132*57718be8SEnji Cooper if ((fd = open("/rump/dev/null", O_RDWR)) != 8)
133*57718be8SEnji Cooper errx(EXIT_FAILURE, "rump open %d %d", fd, errno);
134*57718be8SEnji Cooper return EXIT_SUCCESS;
135*57718be8SEnji Cooper } else {
136*57718be8SEnji Cooper return ENOTSUP;
137*57718be8SEnji Cooper }
138*57718be8SEnji Cooper }
139