tools/tests/libMicro/pwrite.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 #ifdef linux
  32 #define _XOPEN_SOURCE 500
  33 #endif
  34
  35 #include <unistd.h>
  36 #include <stdlib.h>
  37 #include <stdio.h>
  38 #include <fcntl.h>
  39
  40 #include "libmicro.h"
  41
  42 typedef struct {
  43         char                    *ts_buf;
  44 } tsd_t;
  45
  46 #define DEFF                    "/dev/null"
  47 #define DEFS                    1024
  48
  49 static int                      optc = 0;
  50 static char                     *optf = DEFF;
  51 static long long                opts = DEFS;
  52 static int                      fd = -1;
  53
  54 int
  55 benchmark_init()
  56 {
  57         lm_tsdsize = sizeof (tsd_t);
  58
  59         (void) sprintf(lm_optstr, "cf:s:");
  60
  61         (void) sprintf(lm_usage,
  62             "       [-f file-to-write (default %s)]\n"
  63             "       [-s buffer-size (default %d)]\n"
  64             "       [-c ] (make sure buffer is in cache)\n"
  65             "notes: measures pwrite()\n",
  66             DEFF, DEFS);
  67
  68         (void) sprintf(lm_header, "%8s", "size");
  69
  70         return (0);
  71 }
  72
  73 int
  74 benchmark_optswitch(int opt, char *optarg)
  75 {
  76         switch (opt) {
  77         case 'c':
  78                 optc++;
  79                 break;
  80         case 'f':
  81                 optf = optarg;
  82                 break;
  83         case 's':
  84                 opts = sizetoll(optarg);
  85                 break;
  86         default:
  87                 return (-1);
  88         }
  89         return (0);
  90 }
  91
  92 int
  93 benchmark_initrun()
  94 {
  95         fd = open(optf, O_WRONLY);
  96         if (fd == -1) {
  97                 return (-1);
  98         }
  99
 100         return (0);
 101 }
 102
 103 int
 104 benchmark_finirun()
 105 {
 106         return (0);
 107 }
 108
 109 int
 110 benchmark_initbatch(void *tsd)
 111 {
 112         tsd_t                   *ts = (tsd_t *)tsd;
 113         int                     i;
 114
 115         if (ts->ts_buf == NULL) {
 116                 ts->ts_buf = malloc(opts);
 117
 118                 /*
 119                  * bring buf into cache if specified.
 120                  */
 121
 122                 if (optc)
 123                         for (i = 0; i < opts; i++)
 124                                 ts->ts_buf[i] = 0;
 125         }
 126
 127         return (0);
 128 }
 129
 130 int
 131 benchmark(void *tsd, result_t *res)
 132 {
 133         tsd_t                   *ts = (tsd_t *)tsd;
 134         int                     i;
 135
 136         for (i = 0; i < lm_optB; i++) {
 137                 if (pwrite(fd, ts->ts_buf, opts, 0) != opts) {
 138                         res->re_errors++;
 139                 }
 140         }
 141         res->re_count = i;
 142
 143         return (0);
 144 }
 145
 146 char *
 147 benchmark_result()
 148 {
 149         static char             result[256];
 150
 151         (void) sprintf(result, "%8lld", opts);
 152
 153         return (result);
 154 }