1716fd348SMartin Matuska /*
2716fd348SMartin Matuska * CDDL HEADER START
3716fd348SMartin Matuska *
4716fd348SMartin Matuska * The contents of this file are subject to the terms of the
5716fd348SMartin Matuska * Common Development and Distribution License (the "License").
6716fd348SMartin Matuska * You may not use this file except in compliance with the License.
7716fd348SMartin Matuska *
8716fd348SMartin Matuska * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9271171e0SMartin Matuska * or https://opensource.org/licenses/CDDL-1.0.
10716fd348SMartin Matuska * See the License for the specific language governing permissions
11716fd348SMartin Matuska * and limitations under the License.
12716fd348SMartin Matuska *
13716fd348SMartin Matuska * When distributing Covered Code, include this CDDL HEADER in each
14716fd348SMartin Matuska * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15716fd348SMartin Matuska * If applicable, add the following below this CDDL HEADER, with the
16716fd348SMartin Matuska * fields enclosed by brackets "[]" replaced with your own identifying
17716fd348SMartin Matuska * information: Portions Copyright [yyyy] [name of copyright owner]
18716fd348SMartin Matuska *
19716fd348SMartin Matuska * CDDL HEADER END
20716fd348SMartin Matuska */
21716fd348SMartin Matuska
22716fd348SMartin Matuska /*
23716fd348SMartin Matuska * Copyright 2007 Sun Microsystems, Inc. All rights reserved.
24716fd348SMartin Matuska * Use is subject to license terms.
25716fd348SMartin Matuska */
26716fd348SMartin Matuska
27716fd348SMartin Matuska /*
28716fd348SMartin Matuska * Copyright (c) 2012, 2014 by Delphix. All rights reserved.
29716fd348SMartin Matuska */
30716fd348SMartin Matuska
31716fd348SMartin Matuska #include <stdio.h>
32716fd348SMartin Matuska #include <stdlib.h>
33716fd348SMartin Matuska #include <unistd.h>
34716fd348SMartin Matuska #include <limits.h>
35716fd348SMartin Matuska #include <errno.h>
36716fd348SMartin Matuska #include <fcntl.h>
37716fd348SMartin Matuska #include <sys/types.h>
38716fd348SMartin Matuska #include <sys/stat.h>
39716fd348SMartin Matuska #include <sys/statvfs.h>
40716fd348SMartin Matuska #include <sys/time.h>
41716fd348SMartin Matuska #include <sys/ioctl.h>
42716fd348SMartin Matuska #include <sys/wait.h>
43716fd348SMartin Matuska #include <sys/param.h>
44716fd348SMartin Matuska #include <string.h>
45716fd348SMartin Matuska #include <time.h>
46716fd348SMartin Matuska #include <inttypes.h>
47716fd348SMartin Matuska
48716fd348SMartin Matuska #define FSIZE 256*1024*1024
49716fd348SMartin Matuska #define BSIZE 512
50716fd348SMartin Matuska
51716fd348SMartin Matuska /* Initialize Globals */
52716fd348SMartin Matuska static long fsize = FSIZE;
53716fd348SMartin Matuska static size_t bsize = BSIZE;
54716fd348SMartin Matuska static int count = 0;
55716fd348SMartin Matuska static int rflag = 0;
56*be181ee2SMartin Matuska static uint_t seed = 0;
57716fd348SMartin Matuska static int vflag = 0;
58716fd348SMartin Matuska static int errflag = 0;
59716fd348SMartin Matuska static off_t offset = 0;
60716fd348SMartin Matuska static char *filename = NULL;
61716fd348SMartin Matuska
62716fd348SMartin Matuska static void usage(char *execname);
63716fd348SMartin Matuska static void parse_options(int argc, char *argv[]);
64716fd348SMartin Matuska static void do_write(int fd);
65716fd348SMartin Matuska static void do_trunc(int fd);
66716fd348SMartin Matuska
67716fd348SMartin Matuska static void
usage(char * execname)68716fd348SMartin Matuska usage(char *execname)
69716fd348SMartin Matuska {
70716fd348SMartin Matuska (void) fprintf(stderr,
71716fd348SMartin Matuska "usage: %s [-b blocksize] [-c count] [-f filesize]"
72716fd348SMartin Matuska " [-o offset] [-s seed] [-r] [-v] filename\n", execname);
73716fd348SMartin Matuska (void) exit(1);
74716fd348SMartin Matuska }
75716fd348SMartin Matuska
76716fd348SMartin Matuska int
main(int argc,char * argv[])77716fd348SMartin Matuska main(int argc, char *argv[])
78716fd348SMartin Matuska {
79716fd348SMartin Matuska int i = 0;
80716fd348SMartin Matuska int fd = -1;
81716fd348SMartin Matuska
82716fd348SMartin Matuska parse_options(argc, argv);
83716fd348SMartin Matuska
84716fd348SMartin Matuska fd = open(filename, O_RDWR|O_CREAT|O_TRUNC, 0666);
85716fd348SMartin Matuska if (fd < 0) {
86716fd348SMartin Matuska perror("open");
87716fd348SMartin Matuska exit(3);
88716fd348SMartin Matuska }
89716fd348SMartin Matuska
90716fd348SMartin Matuska for (i = 0; count == 0 || i < count; i++) {
91716fd348SMartin Matuska (void) do_write(fd);
92716fd348SMartin Matuska (void) do_trunc(fd);
93716fd348SMartin Matuska }
94716fd348SMartin Matuska
95716fd348SMartin Matuska (void) close(fd);
96716fd348SMartin Matuska return (0);
97716fd348SMartin Matuska }
98716fd348SMartin Matuska
99716fd348SMartin Matuska static void
parse_options(int argc,char * argv[])100716fd348SMartin Matuska parse_options(int argc, char *argv[])
101716fd348SMartin Matuska {
102716fd348SMartin Matuska int c;
103716fd348SMartin Matuska
104716fd348SMartin Matuska extern char *optarg;
105716fd348SMartin Matuska extern int optind, optopt;
106716fd348SMartin Matuska
107716fd348SMartin Matuska count = fsize / bsize;
108*be181ee2SMartin Matuska seed = (uint_t)time(NULL);
109716fd348SMartin Matuska while ((c = getopt(argc, argv, "b:c:f:o:rs:v")) != -1) {
110716fd348SMartin Matuska switch (c) {
111716fd348SMartin Matuska case 'b':
112716fd348SMartin Matuska bsize = atoi(optarg);
113716fd348SMartin Matuska break;
114716fd348SMartin Matuska
115716fd348SMartin Matuska case 'c':
116716fd348SMartin Matuska count = atoi(optarg);
117716fd348SMartin Matuska break;
118716fd348SMartin Matuska
119716fd348SMartin Matuska case 'f':
120716fd348SMartin Matuska fsize = atoi(optarg);
121716fd348SMartin Matuska break;
122716fd348SMartin Matuska
123716fd348SMartin Matuska case 'o':
124716fd348SMartin Matuska offset = atoi(optarg);
125716fd348SMartin Matuska break;
126716fd348SMartin Matuska
127716fd348SMartin Matuska case 'r':
128716fd348SMartin Matuska rflag++;
129716fd348SMartin Matuska break;
130716fd348SMartin Matuska
131716fd348SMartin Matuska case 's':
132716fd348SMartin Matuska seed = atoi(optarg);
133716fd348SMartin Matuska break;
134716fd348SMartin Matuska
135716fd348SMartin Matuska case 'v':
136716fd348SMartin Matuska vflag++;
137716fd348SMartin Matuska break;
138716fd348SMartin Matuska
139716fd348SMartin Matuska case ':':
140716fd348SMartin Matuska (void) fprintf(stderr,
141716fd348SMartin Matuska "Option -%c requires an operand\n", optopt);
142716fd348SMartin Matuska errflag++;
143716fd348SMartin Matuska break;
144716fd348SMartin Matuska
145716fd348SMartin Matuska case '?':
146716fd348SMartin Matuska (void) fprintf(stderr,
147716fd348SMartin Matuska "Unrecognized option: -%c\n", optopt);
148716fd348SMartin Matuska errflag++;
149716fd348SMartin Matuska break;
150716fd348SMartin Matuska }
151716fd348SMartin Matuska
152716fd348SMartin Matuska if (errflag) {
153716fd348SMartin Matuska (void) usage(argv[0]);
154716fd348SMartin Matuska }
155716fd348SMartin Matuska }
156716fd348SMartin Matuska if (argc <= optind) {
157716fd348SMartin Matuska (void) fprintf(stderr,
158716fd348SMartin Matuska "No filename specified\n");
159716fd348SMartin Matuska usage(argv[0]);
160716fd348SMartin Matuska }
161716fd348SMartin Matuska filename = argv[optind];
162716fd348SMartin Matuska
163716fd348SMartin Matuska if (vflag) {
164716fd348SMartin Matuska (void) fprintf(stderr, "Seed = %d\n", seed);
165716fd348SMartin Matuska }
166716fd348SMartin Matuska srandom(seed);
167716fd348SMartin Matuska }
168716fd348SMartin Matuska
169716fd348SMartin Matuska static void
do_write(int fd)170716fd348SMartin Matuska do_write(int fd)
171716fd348SMartin Matuska {
172716fd348SMartin Matuska off_t roffset = 0;
173716fd348SMartin Matuska char *buf = NULL;
174716fd348SMartin Matuska char *rbuf = NULL;
175716fd348SMartin Matuska
176716fd348SMartin Matuska buf = (char *)calloc(1, bsize);
177716fd348SMartin Matuska rbuf = (char *)calloc(1, bsize);
178716fd348SMartin Matuska if (buf == NULL || rbuf == NULL) {
179716fd348SMartin Matuska perror("malloc");
180716fd348SMartin Matuska exit(4);
181716fd348SMartin Matuska }
182716fd348SMartin Matuska
183716fd348SMartin Matuska roffset = random() % fsize;
184716fd348SMartin Matuska if (lseek64(fd, (offset + roffset), SEEK_SET) < 0) {
185716fd348SMartin Matuska perror("lseek");
186716fd348SMartin Matuska exit(5);
187716fd348SMartin Matuska }
188716fd348SMartin Matuska
189716fd348SMartin Matuska (void) strcpy(buf, "ZFS Test Suite Truncation Test");
190716fd348SMartin Matuska if (write(fd, buf, bsize) < bsize) {
191716fd348SMartin Matuska perror("write");
192716fd348SMartin Matuska exit(6);
193716fd348SMartin Matuska }
194716fd348SMartin Matuska
195716fd348SMartin Matuska if (rflag) {
196716fd348SMartin Matuska if (lseek64(fd, (offset + roffset), SEEK_SET) < 0) {
197716fd348SMartin Matuska perror("lseek");
198716fd348SMartin Matuska exit(7);
199716fd348SMartin Matuska }
200716fd348SMartin Matuska
201716fd348SMartin Matuska if (read(fd, rbuf, bsize) < bsize) {
202716fd348SMartin Matuska perror("read");
203716fd348SMartin Matuska exit(8);
204716fd348SMartin Matuska }
205716fd348SMartin Matuska
206716fd348SMartin Matuska if (memcmp(buf, rbuf, bsize) != 0) {
207716fd348SMartin Matuska perror("memcmp");
208716fd348SMartin Matuska exit(9);
209716fd348SMartin Matuska }
210716fd348SMartin Matuska }
211716fd348SMartin Matuska if (vflag) {
212716fd348SMartin Matuska (void) fprintf(stderr,
213716fd348SMartin Matuska "Wrote to offset %" PRId64 "\n", (offset + roffset));
214716fd348SMartin Matuska if (rflag) {
215716fd348SMartin Matuska (void) fprintf(stderr,
216716fd348SMartin Matuska "Read back from offset %" PRId64 "\n",
217716fd348SMartin Matuska (offset + roffset));
218716fd348SMartin Matuska }
219716fd348SMartin Matuska }
220716fd348SMartin Matuska
221716fd348SMartin Matuska (void) free(buf);
222716fd348SMartin Matuska (void) free(rbuf);
223716fd348SMartin Matuska }
224716fd348SMartin Matuska
225716fd348SMartin Matuska static void
do_trunc(int fd)226716fd348SMartin Matuska do_trunc(int fd)
227716fd348SMartin Matuska {
228716fd348SMartin Matuska off_t roffset = 0;
229716fd348SMartin Matuska
230716fd348SMartin Matuska roffset = random() % fsize;
231716fd348SMartin Matuska if (ftruncate64(fd, (offset + roffset)) < 0) {
232716fd348SMartin Matuska perror("truncate");
233716fd348SMartin Matuska exit(7);
234716fd348SMartin Matuska }
235716fd348SMartin Matuska
236716fd348SMartin Matuska if (vflag) {
237716fd348SMartin Matuska (void) fprintf(stderr, "Truncated at offset %" PRId64 "\n",
238716fd348SMartin Matuska (offset + roffset));
239716fd348SMartin Matuska }
240716fd348SMartin Matuska }
241