Home | History | Annotate | Download | only in libmicro
      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 }
    155