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
42typedef struct {
43	char			*ts_buf;
44} tsd_t;
45
46#define	DEFF			"/dev/null"
47#define	DEFS			1024
48
49static int			optc = 0;
50static char			*optf = DEFF;
51static long long		opts = DEFS;
52static int			fd = -1;
53
54int
55benchmark_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
73int
74benchmark_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
92int
93benchmark_initrun()
94{
95	fd = open(optf, O_WRONLY);
96	if (fd == -1) {
97		return (-1);
98	}
99
100	return (0);
101}
102
103int
104benchmark_finirun()
105{
106	return (0);
107}
108
109int
110benchmark_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
130int
131benchmark(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
146char *
147benchmark_result()
148{
149	static char		result[256];
150
151	(void) sprintf(result, "%8lld", opts);
152
153	return (result);
154}
155