1 /* 2 * Copyright (c) 2015-2016 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 32 #include "tests.h" 33 #include "nsig.h" 34 #include <assert.h> 35 #include <stdio.h> 36 #include <unistd.h> 37 #include <sys/select.h> 38 #include <asm/unistd.h> 39 #include <sys/time.h> 40 41 #ifdef __NR_pselect6 42 43 static fd_set set[3][0x1000000 / sizeof(fd_set)]; 44 45 static void 46 handler(int signo) 47 { 48 } 49 50 int main(int ac, char **av) 51 { 52 int fds[2]; 53 struct { 54 struct timespec ts; 55 int pad[2]; 56 } tm_in = { 57 .ts = { .tv_sec = 0xc0de1, .tv_nsec = 0xc0de2 }, 58 .pad = { 0xdeadbeef, 0xbadc0ded } 59 }, tm = tm_in; 60 sigset_t mask; 61 const struct sigaction act = { .sa_handler = handler }; 62 const struct itimerval itv = { .it_value.tv_usec = 111111 }; 63 64 sigemptyset(&mask); 65 sigaddset(&mask, SIGHUP); 66 sigaddset(&mask, SIGCHLD); 67 68 if (pipe(fds)) 69 perror_msg_and_fail("pipe"); 70 71 /* 72 * Start with a nice simple pselect. 73 */ 74 FD_SET(fds[0], set[0]); 75 FD_SET(fds[1], set[0]); 76 FD_SET(fds[0], set[1]); 77 FD_SET(fds[1], set[1]); 78 FD_SET(1, set[2]); 79 FD_SET(2, set[2]); 80 int rc = pselect(fds[1] + 1, set[0], set[1], set[2], NULL, NULL); 81 if (rc < 0) 82 perror_msg_and_skip("pselect"); 83 assert(rc == 1); 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_BYTES, 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 assert(syscall(__NR_pselect6, fds[1] + 1, NULL, set[1], NULL, &tm.ts, NULL) == 3); 98 printf("pselect6(%d, NULL, [1 2 %d %d], NULL, " 99 "{tv_sec=%lld, tv_nsec=%lld}, NULL) = 3 (out [1 2 %d], left " 100 "{tv_sec=%lld, tv_nsec=%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 assert(pselect(-1, NULL, set[0], NULL, NULL, &mask) == -1); 113 printf("pselect6(-1, NULL, %p, NULL, NULL, {[HUP CHLD], %u}) " 114 "= -1 EINVAL (%m)\n", set[0], NSIG_BYTES); 115 116 /* 117 * Another variant, with nfds exceeding FD_SETSIZE limit. 118 */ 119 FD_ZERO(set[0]); 120 FD_SET(fds[0],set[0]); 121 FD_ZERO(set[1]); 122 tm.ts.tv_sec = 0; 123 tm.ts.tv_nsec = 123; 124 assert(pselect(FD_SETSIZE + 1, set[0], set[1], NULL, &tm.ts, &mask) == 0); 125 printf("pselect6(%d, [%d], [], NULL, {tv_sec=0, tv_nsec=123}, " 126 "{[HUP CHLD], %u}) = 0 (Timeout)\n", 127 FD_SETSIZE + 1, fds[0], NSIG_BYTES); 128 129 /* 130 * See how timeouts are decoded. 131 */ 132 assert(sigaction(SIGALRM, &act, NULL) == 0); 133 assert(setitimer(ITIMER_REAL, &itv, NULL) == 0); 134 135 tm.ts.tv_nsec = 222222222; 136 assert(pselect(0, NULL, NULL, NULL, &tm.ts, &mask) == -1); 137 printf("pselect6(0, NULL, NULL, NULL, {tv_sec=0, tv_nsec=222222222}, " 138 "{[HUP CHLD], %u}) = " 139 "? ERESTARTNOHAND (To be restarted if no handler)\n", 140 NSIG_BYTES); 141 puts("--- SIGALRM {si_signo=SIGALRM, si_code=SI_KERNEL} ---"); 142 143 puts("+++ exited with 0 +++"); 144 return 0; 145 } 146 147 #else 148 149 SKIP_MAIN_UNDEFINED("__NR_pselect6") 150 151 #endif 152
