granicus.if.org Git - strace/blob - tests/poll.c

   1 /*
   2  * This file is part of poll strace test.
   3  *
   4  * Copyright (c) 2016 Dmitry V. Levin <ldv@altlinux.org>
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. The name of the author may not be used to endorse or promote products
  16  *    derived from this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  19  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  20  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  21  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  22  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  23  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29
  30 #include "tests.h"
  31 #include <asm/unistd.h>
  32
  33 #ifdef __NR_poll
  34
  35 # include <assert.h>
  36 # include <errno.h>
  37 # include <poll.h>
  38 # include <stdio.h>
  39 # include <stdlib.h>
  40 # include <unistd.h>
  41
  42 #define PRINT_EVENT(flag, member) \
  43         if (member & flag) { \
  44                 if (member != pfd->member) \
  45                         tprintf("|"); \
  46                 tprintf(#flag); \
  47                 member &= ~flag; \
  48         }
  49
  50 static void
  51 print_pollfd_entering(const struct pollfd *const pfd)
  52 {
  53         tprintf("{fd=%d", pfd->fd);
  54         if (pfd->fd >= 0) {
  55                 tprintf(", events=");
  56                 short events = pfd->events;
  57
  58                 if (pfd->events) {
  59                         PRINT_EVENT(POLLIN, events)
  60                         PRINT_EVENT(POLLPRI, events)
  61                         PRINT_EVENT(POLLOUT, events)
  62 #ifdef POLLRDNORM
  63                         PRINT_EVENT(POLLRDNORM, events)
  64 #endif
  65 #ifdef POLLWRNORM
  66                         PRINT_EVENT(POLLWRNORM, events)
  67 #endif
  68 #ifdef POLLRDBAND
  69                         PRINT_EVENT(POLLRDBAND, events)
  70 #endif
  71 #ifdef POLLWRBAND
  72                         PRINT_EVENT(POLLWRBAND, events)
  73 #endif
  74                         PRINT_EVENT(POLLERR, events)
  75                         PRINT_EVENT(POLLHUP, events)
  76                         PRINT_EVENT(POLLNVAL, events)
  77                 } else
  78                         tprintf("0");
  79         }
  80         tprintf("}");
  81 }
  82
  83 static void
  84 print_pollfd_array_entering(const struct pollfd *const pfd,
  85                             const unsigned int size,
  86                             const unsigned int valid,
  87                             const unsigned int abbrev)
  88 {
  89         tprintf("[");
  90         unsigned int i;
  91         for (i = 0; i < size; ++i) {
  92                 if (i)
  93                         tprintf(", ");
  94                 if (i >= valid) {
  95                         tprintf("%p", &pfd[i]);
  96                         break;
  97                 }
  98                 if (i >= abbrev) {
  99                         tprintf("...");
 100                         break;
 101                 }
 102                 print_pollfd_entering(&pfd[i]);
 103         }
 104         tprintf("]");
 105 }
 106
 107 static void
 108 print_pollfd_exiting(const struct pollfd *const pfd,
 109                      unsigned int *const seen,
 110                      const unsigned int abbrev)
 111 {
 112         if (!pfd->revents || pfd->fd < 0 || *seen > abbrev)
 113                 return;
 114
 115         if (*seen)
 116                 tprintf(", ");
 117         ++(*seen);
 118
 119         if (*seen > abbrev) {
 120                 tprintf("...");
 121                 return;
 122         }
 123         tprintf("{fd=%d, revents=", pfd->fd);
 124         short revents = pfd->revents;
 125
 126         PRINT_EVENT(POLLIN, revents)
 127         PRINT_EVENT(POLLPRI, revents)
 128         PRINT_EVENT(POLLOUT, revents)
 129 #ifdef POLLRDNORM
 130         PRINT_EVENT(POLLRDNORM, revents)
 131 #endif
 132 #ifdef POLLWRNORM
 133         PRINT_EVENT(POLLWRNORM, revents)
 134 #endif
 135 #ifdef POLLRDBAND
 136         PRINT_EVENT(POLLRDBAND, revents)
 137 #endif
 138 #ifdef POLLWRBAND
 139         PRINT_EVENT(POLLWRBAND, revents)
 140 #endif
 141         PRINT_EVENT(POLLERR, revents)
 142         PRINT_EVENT(POLLHUP, revents)
 143         PRINT_EVENT(POLLNVAL, revents)
 144         tprintf("}");
 145 }
 146
 147 static void
 148 print_pollfd_array_exiting(const struct pollfd *const pfd,
 149                            const unsigned int size,
 150                            const unsigned int abbrev)
 151 {
 152         tprintf("[");
 153         unsigned int seen = 0;
 154         unsigned int i;
 155         for (i = 0; i < size; ++i)
 156                 print_pollfd_exiting(&pfd[i], &seen, abbrev);
 157         tprintf("]");
 158 }
 159
 160 int
 161 main(int ac, char **av)
 162 {
 163         tprintf("%s", "");
 164
 165         assert(syscall(__NR_poll, NULL, 42, 0) == -1);
 166         if (ENOSYS == errno)
 167                 perror_msg_and_skip("poll");
 168         tprintf("poll(NULL, 42, 0) = -1 EFAULT (%m)\n");
 169
 170         int fds[2];
 171         if (pipe(fds) || pipe(fds))
 172                 perror_msg_and_fail("pipe");
 173
 174         const unsigned int abbrev = (ac > 1) ? atoi(av[1]) : -1;
 175         const struct pollfd pfds0[] = {
 176                 { .fd = 0, .events = POLLIN | POLLPRI | POLLRDNORM | POLLRDBAND },
 177                 { .fd = 1, .events = POLLOUT | POLLWRNORM | POLLWRBAND },
 178                 { .fd = fds[0], .events = POLLIN | POLLPRI },
 179                 { .fd = fds[1], .events = POLLOUT },
 180                 { .fd = 2, .events = POLLOUT | POLLWRBAND }
 181         };
 182         struct pollfd *const tail_fds0 = tail_memdup(pfds0, sizeof(pfds0));
 183         const int timeout = 42;
 184         int rc = syscall(__NR_poll, tail_fds0, 0, timeout);
 185         assert(rc == 0);
 186
 187         tprintf("poll([], 0, %d) = %d (Timeout)\n", timeout, rc);
 188
 189         rc = syscall(__NR_poll, tail_fds0, ARRAY_SIZE(pfds0), timeout);
 190         assert(rc == 3);
 191
 192         tprintf("poll(");
 193         print_pollfd_array_entering(tail_fds0, ARRAY_SIZE(pfds0),
 194                                     ARRAY_SIZE(pfds0), abbrev);
 195         tprintf(", %u, %d) = %d (", ARRAY_SIZE(pfds0), timeout, rc);
 196         print_pollfd_array_exiting(tail_fds0, ARRAY_SIZE(pfds0), abbrev);
 197         tprintf(")\n");
 198
 199         tail_fds0[0].fd = -1;
 200         tail_fds0[2].fd = -3;
 201         tail_fds0[4].events = 0;
 202         rc = syscall(__NR_poll, tail_fds0, ARRAY_SIZE(pfds0), timeout);
 203         assert(rc == 2);
 204
 205         tprintf("poll(");
 206         print_pollfd_array_entering(tail_fds0, ARRAY_SIZE(pfds0),
 207                                     ARRAY_SIZE(pfds0), abbrev);
 208         tprintf(", %u, %d) = %d (", ARRAY_SIZE(pfds0), timeout, rc);
 209         print_pollfd_array_exiting(tail_fds0, ARRAY_SIZE(pfds0), abbrev);
 210         tprintf(")\n");
 211
 212         tail_fds0[1].fd = -2;
 213         tail_fds0[4].fd = -5;
 214         rc = syscall(__NR_poll, tail_fds0, ARRAY_SIZE(pfds0), timeout);
 215         assert(rc == 1);
 216
 217         tprintf("poll(");
 218         print_pollfd_array_entering(tail_fds0, ARRAY_SIZE(pfds0),
 219                                     ARRAY_SIZE(pfds0), abbrev);
 220         tprintf(", %u, %d) = %d (", ARRAY_SIZE(pfds0), timeout, rc);
 221         print_pollfd_array_exiting(tail_fds0, ARRAY_SIZE(pfds0), abbrev);
 222         tprintf(")\n");
 223
 224         struct pollfd pfds1[] = {
 225                 { .fd = 1, .events = POLLIN | POLLPRI | POLLRDNORM | POLLRDBAND },
 226                 { .fd = 0, .events = POLLOUT | POLLWRNORM | POLLWRBAND }
 227         };
 228         struct pollfd *const tail_fds1 = tail_memdup(pfds1, sizeof(pfds1));
 229         rc = syscall(__NR_poll, tail_fds1, ARRAY_SIZE(pfds1), timeout);
 230         assert(rc == 0);
 231
 232         tprintf("poll(");
 233         print_pollfd_array_entering(tail_fds1, ARRAY_SIZE(pfds1),
 234                                     ARRAY_SIZE(pfds1), abbrev);
 235         tprintf(", %u, %d) = %d (Timeout)\n", ARRAY_SIZE(pfds1), timeout, rc);
 236
 237         const void *const efault = tail_fds0 + ARRAY_SIZE(pfds0);
 238         rc = syscall(__NR_poll, efault, 1, 0);
 239         assert(rc == -1);
 240         tprintf("poll(%p, 1, 0) = -1 EFAULT (%m)\n", efault);
 241
 242         const unsigned int valid = 1;
 243         const void *const epfds = tail_fds0 + ARRAY_SIZE(pfds0) - valid;
 244         rc = syscall(__NR_poll, epfds, valid + 1, 0);
 245         assert(rc == -1);
 246         tprintf("poll(");
 247         print_pollfd_array_entering(epfds, valid + 1, valid, abbrev);
 248         errno = EFAULT;
 249         tprintf(", %u, 0) = -1 EFAULT (%m)\n", valid + 1);
 250
 251         tprintf("+++ exited with 0 +++\n");
 252         return 0;
 253 }
 254
 255 #else
 256
 257 SKIP_MAIN_UNDEFINED("__NR_poll")
 258
 259 #endif