1 /* 2 * Copyright (c) 2015 Dmitry V. Levin <ldv (at) altlinux.org> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. The name of the author may not be used to endorse or promote products 14 * derived from this software without specific prior written permission. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 */ 27 28 /* 29 * Based on test by Dr. David Alan Gilbert <dave (at) treblig.org> 30 */ 31 #include <stdio.h> 32 #include <signal.h> 33 #include <unistd.h> 34 #include <sys/select.h> 35 #include <sys/syscall.h> 36 #include <sys/time.h> 37 38 #ifdef __NR_pselect6 39 40 #ifndef NSIG 41 # warning NSIG is not defined, using 32 42 # define NSIG 32 43 #endif 44 45 static fd_set set[3][0x1000000 / sizeof(fd_set)]; 46 47 static void 48 handler(int signo) 49 { 50 } 51 52 int main(int ac, char **av) 53 { 54 int fds[2]; 55 struct { 56 struct timespec ts; 57 int pad[2]; 58 } tm_in = { 59 .ts = { .tv_sec = 0xc0de1, .tv_nsec = 0xc0de2 }, 60 .pad = { 0xdeadbeef, 0xbadc0ded } 61 }, tm = tm_in; 62 sigset_t mask; 63 const struct sigaction act = { .sa_handler = handler }; 64 const struct itimerval itv = { .it_value.tv_usec = 111111 }; 65 66 sigemptyset(&mask); 67 sigaddset(&mask, SIGHUP); 68 sigaddset(&mask, SIGCHLD); 69 70 if (pipe(fds)) 71 return 77; 72 73 /* 74 * Start with a nice simple pselect. 75 */ 76 FD_SET(fds[0], set[0]); 77 FD_SET(fds[1], set[0]); 78 FD_SET(fds[0], set[1]); 79 FD_SET(fds[1], set[1]); 80 FD_SET(1, set[2]); 81 FD_SET(2, set[2]); 82 if (pselect(fds[1] + 1, set[0], set[1], set[2], NULL, NULL) != 1) 83 return 77; 84 printf("pselect6(%d, [%d %d], [%d %d], [1 2], NULL, {NULL, %u}) " 85 "= 1 (out [%d])\n", 86 fds[1] + 1, fds[0], fds[1], 87 fds[0], fds[1], 88 NSIG / 8, fds[1]); 89 90 /* 91 * Another simple one, with a timeout. 92 */ 93 FD_SET(1, set[1]); 94 FD_SET(2, set[1]); 95 FD_SET(fds[0], set[1]); 96 FD_SET(fds[1], set[1]); 97 if (syscall(__NR_pselect6, fds[1] + 1, NULL, set[1], NULL, &tm.ts, NULL) != 3) 98 return 77; 99 printf("pselect6(%d, NULL, [1 2 %d %d], NULL, {%lld, %lld}, NULL)" 100 " = 3 (out [1 2 %d], left {%lld, %lld})\n", 101 fds[1] + 1, fds[0], fds[1], 102 (long long) tm_in.ts.tv_sec, (long long) tm_in.ts.tv_nsec, 103 fds[1], 104 (long long) tm.ts.tv_sec, (long long) tm.ts.tv_nsec); 105 106 /* 107 * Now the crash case that trinity found, negative nfds 108 * but with a pointer to a large chunk of valid memory. 109 */ 110 FD_ZERO(set[0]); 111 FD_SET(fds[1],set[0]); 112 if (pselect(-1, NULL, set[0], NULL, NULL, &mask) != -1) 113 return 77; 114 printf("pselect6(-1, NULL, %p, NULL, NULL, {[HUP CHLD], %u}) " 115 "= -1 EINVAL (Invalid argument)\n", set[0], NSIG / 8); 116 117 /* 118 * Another variant, with nfds exceeding FD_SETSIZE limit. 119 */ 120 FD_ZERO(set[0]); 121 FD_SET(fds[0],set[0]); 122 FD_ZERO(set[1]); 123 tm.ts.tv_sec = 0; 124 tm.ts.tv_nsec = 123; 125 if (pselect(FD_SETSIZE + 1, set[0], set[1], NULL, &tm.ts, &mask) != 0) 126 return 77; 127 printf("pselect6(%d, [%d], [], NULL, {0, 123}, {[HUP CHLD], %u}) " 128 "= 0 (Timeout)\n", FD_SETSIZE + 1, fds[0], NSIG / 8); 129 130 /* 131 * See how timeouts are decoded. 132 */ 133 if (sigaction(SIGALRM, &act, NULL) != 0) 134 return 77; 135 if (setitimer(ITIMER_REAL, &itv, NULL) != 0) 136 return 77; 137 138 tm.ts.tv_nsec = 222222222; 139 if (pselect(0, NULL, NULL, NULL, &tm.ts, &mask) != -1) 140 return 77; 141 puts("pselect6(0, NULL, NULL, NULL, {0, 222222222}, {[HUP CHLD], 8})" 142 " = ? ERESTARTNOHAND (To be restarted if no handler)"); 143 puts("--- SIGALRM {si_signo=SIGALRM, si_code=SI_KERNEL} ---"); 144 145 puts("+++ exited with 0 +++"); 146 return 0; 147 } 148 149 #else 150 151 int 152 main(void) 153 { 154 return 77; 155 } 156 157 #endif 158
