tools/tests/libMicro/poll.c

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms
   5  * of the Common Development and Distribution License
   6  * (the "License").  You may not use this file except
   7  * in compliance with the License.
   8  *
   9  * You can obtain a copy of the license at
  10  * src/OPENSOLARIS.LICENSE
  11  * or http://www.opensolaris.org/os/licensing.
  12  * See the License for the specific language governing
  13  * permissions and limitations under the License.
  14  *
  15  * When distributing Covered Code, include this CDDL
  16  * HEADER in each file and include the License file at
  17  * usr/src/OPENSOLARIS.LICENSE.  If applicable,
  18  * add the following below this CDDL HEADER, with the
  19  * fields enclosed by brackets "[]" replaced with your
  20  * own identifying information: Portions Copyright [yyyy]
  21  * [name of copyright owner]
  22  *
  23  * CDDL HEADER END
  24  */
  25
  26 /*
  27  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
  28  * Use is subject to license terms.
  29  */
  30
  31 #define MAX(x, y)               ((x) > (y) ? (x) : (y))
  32 #define MIN(x, y)               ((x) > (y) ? (y) : (x))
  33
  34 #include <unistd.h>
  35 #include <stdlib.h>
  36 #include <stdio.h>
  37 #include <string.h>
  38 #include <poll.h>
  39 #include <sys/socket.h>
  40
  41 #include "libmicro.h"
  42
  43 #define DEFN                    256
  44
  45 static int                      optn = DEFN;
  46 static int                      optr = 0;
  47 static int                      optw = 0;
  48 static int                      optx = 0;
  49 static int                      *fds;
  50 static int                      target = 0;
  51
  52 typedef struct pollfd           pfd_t;
  53
  54 typedef struct {
  55         int                     ts_once;
  56         pfd_t           *ts_pfds;
  57 } tsd_t;
  58
  59 int
  60 benchmark_init()
  61 {
  62         lm_tsdsize = sizeof (tsd_t);
  63
  64         (void) sprintf(lm_optstr, "n:r:w:x");
  65
  66         (void) sprintf(lm_usage,
  67             "       [-n fds-per-thread (default %d)]\n"
  68             "       [-r readable-fds (default 0)]\n"
  69             "       [-w writeable-fds (default 0)]\n"
  70             "       [-x] (start -r option with highest fd first; "
  71             "default is lowest first)\n"
  72             "notes: measures poll()\n",
  73             DEFN);
  74
  75         (void) sprintf(lm_header, "%8s %5s", "nfds", "flags");
  76
  77         return (0);
  78 }
  79
  80 int
  81 benchmark_optswitch(int opt, char *optarg)
  82 {
  83         switch (opt) {
  84         case 'n':
  85                 optn = atoi(optarg);
  86                 break;
  87         case 'r':
  88                 optr = atoi(optarg);
  89                 break;
  90         case 'w':
  91                 optw = atoi(optarg);
  92                 break;
  93         case 'x':
  94                 optx = 1;
  95                 break;
  96         default:
  97                 return (-1);
  98         }
  99         return (0);
 100 }
 101
 102 int
 103 benchmark_initrun()
 104 {
 105         int                     i;
 106         int                     j;
 107         int                     pair[2];
 108
 109         if (optn % 2 != 0) {
 110                 (void) printf("ERROR: -n value must be even\n");
 111                 optn = optr = optw = 0;
 112                 return (-1);
 113         }
 114
 115         if (optn < 0 || optr < 0 || optw < 0) {
 116                 (void) printf("ERROR: -n, -r and -w values must be > 0\n");
 117                 optn = optr = optw = 0;
 118                 return (-1);
 119         }
 120
 121         if (optr > optn || optw > optn) {
 122                 (void) printf("ERROR: -r and -w values must be <= maxfd\n");
 123                 optn = optr = optw = 0;
 124                 return (-1);
 125         }
 126
 127         fds = (int *)malloc(optn * sizeof (int));
 128         if (fds == NULL) {
 129                 (void) printf("ERROR: malloc() failed\n");
 130                 optn = optr = optw = 0;
 131                 return (-1);
 132         }
 133
 134         (void) setfdlimit(optn + 10);
 135
 136
 137         for (i = 0; i < optn; i += 2) {
 138                 if (socketpair(PF_UNIX, SOCK_STREAM, 0, pair) == -1) {
 139                         (void) printf("ERROR: socketpair() failed\n");
 140                         return (-1);
 141                 }
 142
 143                 fds[i] = MIN(pair[0], pair[1]);
 144                 fds[i+1] = MAX(pair[0], pair[1]);
 145         }
 146
 147         if (optx) {
 148                 target = MIN(optr + optw, optn);
 149                 for (i = 0, j = optn - 1; i < optr; i++, j--) {
 150                         (void) write(fds[j+1 - (2*(j%2))], "", 1);
 151                 }
 152         } else {
 153                 target = MAX(optr, optw);
 154                 for (i = 0; i < optr; i++) {
 155                         (void) write(fds[i+1 - (2*(i%2))], "", 1);
 156                 }
 157         }
 158
 159         return (0);
 160 }
 161
 162 int
 163 benchmark_initbatch(void *tsd)
 164 {
 165         tsd_t                   *ts = (tsd_t *)tsd;
 166         int                     i;
 167         int                     errors = 0;
 168
 169         if (ts->ts_once++ == 0) {
 170                 ts->ts_pfds = (pfd_t *)malloc(optn * sizeof (pfd_t));
 171                 if (ts->ts_pfds == NULL) {
 172                         errors++;
 173                 }
 174
 175                 for (i = 0; i < optn; i++) {
 176                         ts->ts_pfds[i].fd = fds[i];
 177                         ts->ts_pfds[i].events = POLLIN;
 178                 }
 179
 180                 for (i = 0; i < optw; i++) {
 181                         ts->ts_pfds[i].events |= POLLOUT;
 182                 }
 183         }
 184
 185         return (errors);
 186 }
 187
 188 int
 189 benchmark(void *tsd, result_t *res)
 190 {
 191         tsd_t                   *ts = (tsd_t *)tsd;
 192         int                     i;
 193
 194         for (i = 0; i < lm_optB; i++) {
 195                 if (poll(ts->ts_pfds, optn, 0) != target) {
 196                         res->re_errors++;
 197                 }
 198         }
 199         res->re_count = i;
 200
 201         return (0);
 202 }
 203
 204 char *
 205 benchmark_result()
 206 {
 207         static char             result[256];
 208         char                    flags[4];
 209
 210         flags[0] = optr ? 'r' : '-';
 211         flags[1] = optw ? 'w' : '-';
 212         flags[2] = optx ? 'x' : '-';
 213         flags[3] = 0;
 214
 215         (void) sprintf(result, "%8d %5s", optn, flags);
 216
 217         return (result);
 218 }