tools/tests/libMicro/apple/lmbench_write.c

   1 /*
   2  * Copyright (c) 2006 Apple Inc.  All Rights Reserved.
   3  *
   4  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. The rights granted to you under the License
  10  * may not be used to create, or enable the creation or redistribution of,
  11  * unlawful or unlicensed copies of an Apple operating system, or to
  12  * circumvent, violate, or enable the circumvention or violation of, any
  13  * terms of an Apple operating system software license agreement.
  14  *
  15  * Please obtain a copy of the License at
  16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
  17  *
  18  * The Original Code and all software distributed under the License are
  19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  23  * Please see the License for the specific language governing rights and
  24  * limitations under the License.
  25  *
  26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  27  */
  28
  29
  30 #ifdef  __sun
  31 #pragma ident   "@(#)write.c    1.3     05/08/04 Apple Inc."
  32 #endif
  33
  34 #ifdef linux
  35 #define _XOPEN_SOURCE 500
  36 #endif
  37
  38 #include <unistd.h>
  39 #include <stdlib.h>
  40 #include <stdio.h>
  41 #include <fcntl.h>
  42
  43 #include "../libmicro.h"
  44
  45 typedef struct {
  46         char                    *ts_buf;
  47         int                     ts_fd;
  48 } tsd_t;
  49
  50 #define DEFF                    "/dev/null"
  51 #define DEFS                    1024
  52
  53 static int                      optc = 0;
  54 static char                     *optf = DEFF;
  55 static long long                opts = DEFS;
  56 static int                      optd;
  57
  58 int
  59 benchmark_init()
  60 {
  61         lm_tsdsize = sizeof (tsd_t);
  62
  63         (void) sprintf(lm_optstr, "cdf:s:");
  64
  65         (void) sprintf(lm_usage,
  66             "       [-f file-to-write (default %s)]\n"
  67             "       [-s buffer-size (default %d)]\n"
  68             "       [-c ] (make sure buffer is in cache)\n"
  69 #ifdef __sun
  70             "       [-d ] use directio"
  71 #endif
  72             "notes: measures lmbench_write()\n",
  73             DEFF, DEFS);
  74
  75         (void) sprintf(lm_header, "%8s", "size");
  76
  77         lm_defB = 1;
  78
  79         return (0);
  80 }
  81
  82 int
  83 benchmark_optswitch(int opt, char *optarg)
  84 {
  85         switch (opt) {
  86
  87         case 'd':
  88                 optd++;
  89                 break;
  90         case 'c':
  91                 optc++;
  92                 break;
  93         case 'f':
  94                 optf = optarg;
  95                 break;
  96         case 's':
  97                 opts = sizetoll(optarg);
  98                 break;
  99         default:
 100                 return (-1);
 101         }
 102         return (0);
 103 }
 104
 105 int
 106 benchmark_initbatch(void *tsd)
 107 {
 108         tsd_t                   *ts = (tsd_t *)tsd;
 109         int                     i;
 110
 111         if (ts->ts_buf == NULL) {
 112                 ts->ts_buf = malloc(opts);
 113                 ts->ts_fd = open(optf, O_WRONLY);
 114
 115 #ifdef __sun
 116                 if (optd)
 117                         (void) directio(ts->ts_fd, DIRECTIO_ON);
 118 #endif
 119                 /*
 120                  * bring buf into cache if specified.
 121                  */
 122
 123                 if (optc)
 124                         for (i = 0; i < opts; i++)
 125                                 ts->ts_buf[i] = 0;
 126         }
 127
 128         (void) lseek(ts->ts_fd, 0, SEEK_SET);
 129
 130         return (0);
 131 }
 132
 133 int
 134 benchmark(void *tsd, result_t *res)
 135 {
 136         tsd_t                   *ts = (tsd_t *)tsd;
 137         int                     i;
 138
 139 /*
 140  * The libmicro test uses a for loop as below:
 141  *   for (i = 0; i < lm_optB; i++) {
 142  *
 143  * we can probably get away with using lm_optB
 144  * in the while loop below
 145  *
 146  */
 147         i = 0;
 148
 149         while (i++ < lm_optB) {
 150                 if (write(ts->ts_fd, ts->ts_buf, opts) != opts) {
 151                         res->re_errors++;
 152                 }
 153         }
 154         res->re_count = i;
 155
 156         return (0);
 157 }
 158
 159 char *
 160 benchmark_result()
 161 {
 162         static char             result[256];
 163
 164         (void) sprintf(result, "%8lld", opts);
 165
 166         return (result);
 167 }