1d583b39bSJohn Wren Kennedy /*
2d583b39bSJohn Wren Kennedy * CDDL HEADER START
3d583b39bSJohn Wren Kennedy *
4d583b39bSJohn Wren Kennedy * The contents of this file are subject to the terms of the
5d583b39bSJohn Wren Kennedy * Common Development and Distribution License (the "License").
6d583b39bSJohn Wren Kennedy * You may not use this file except in compliance with the License.
7d583b39bSJohn Wren Kennedy *
8d583b39bSJohn Wren Kennedy * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9d583b39bSJohn Wren Kennedy * or http://www.opensolaris.org/os/licensing.
10d583b39bSJohn Wren Kennedy * See the License for the specific language governing permissions
11d583b39bSJohn Wren Kennedy * and limitations under the License.
12d583b39bSJohn Wren Kennedy *
13d583b39bSJohn Wren Kennedy * When distributing Covered Code, include this CDDL HEADER in each
14d583b39bSJohn Wren Kennedy * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15d583b39bSJohn Wren Kennedy * If applicable, add the following below this CDDL HEADER, with the
16d583b39bSJohn Wren Kennedy * fields enclosed by brackets "[]" replaced with your own identifying
17d583b39bSJohn Wren Kennedy * information: Portions Copyright [yyyy] [name of copyright owner]
18d583b39bSJohn Wren Kennedy *
19d583b39bSJohn Wren Kennedy * CDDL HEADER END
20d583b39bSJohn Wren Kennedy */
21d583b39bSJohn Wren Kennedy
22d583b39bSJohn Wren Kennedy /*
23d583b39bSJohn Wren Kennedy * Copyright 2007 Sun Microsystems, Inc. All rights reserved.
24d583b39bSJohn Wren Kennedy * Use is subject to license terms.
25d583b39bSJohn Wren Kennedy */
26d583b39bSJohn Wren Kennedy
27*b533ad07SDan McDonald /*
28*b533ad07SDan McDonald * Copyright 2022 MNX Cloud, Inc.
29*b533ad07SDan McDonald */
30*b533ad07SDan McDonald
31d583b39bSJohn Wren Kennedy #include "../file_common.h"
32d583b39bSJohn Wren Kennedy #include <libgen.h>
33d583b39bSJohn Wren Kennedy
34d583b39bSJohn Wren Kennedy static unsigned char bigbuffer[BIGBUFFERSIZE];
35d583b39bSJohn Wren Kennedy
36d583b39bSJohn Wren Kennedy /*
37d583b39bSJohn Wren Kennedy * Writes (or appends) a given value to a file repeatedly.
38d583b39bSJohn Wren Kennedy * See header file for defaults.
39d583b39bSJohn Wren Kennedy */
40d583b39bSJohn Wren Kennedy
41d583b39bSJohn Wren Kennedy static void usage(void);
42d583b39bSJohn Wren Kennedy
43d583b39bSJohn Wren Kennedy int
main(int argc,char ** argv)44d583b39bSJohn Wren Kennedy main(int argc, char **argv)
45d583b39bSJohn Wren Kennedy {
46d583b39bSJohn Wren Kennedy int bigfd;
47d583b39bSJohn Wren Kennedy int c;
48d583b39bSJohn Wren Kennedy int oflag = 0;
49d583b39bSJohn Wren Kennedy int err = 0;
50d583b39bSJohn Wren Kennedy int k;
51d583b39bSJohn Wren Kennedy long i;
52d583b39bSJohn Wren Kennedy int64_t good_writes = 0;
53d583b39bSJohn Wren Kennedy uchar_t nxtfillchar;
54d583b39bSJohn Wren Kennedy /*
55d583b39bSJohn Wren Kennedy * Default Parameters
56d583b39bSJohn Wren Kennedy */
57d583b39bSJohn Wren Kennedy int write_count = BIGFILESIZE;
58d583b39bSJohn Wren Kennedy uchar_t fillchar = DATA;
59d583b39bSJohn Wren Kennedy int block_size = BLOCKSZ;
60d583b39bSJohn Wren Kennedy char *filename = NULL;
61d583b39bSJohn Wren Kennedy char *operation = NULL;
62d583b39bSJohn Wren Kennedy offset_t noffset, offset = 0;
63d583b39bSJohn Wren Kennedy int verbose = 0;
64d583b39bSJohn Wren Kennedy int rsync = 0;
65d583b39bSJohn Wren Kennedy int wsync = 0;
66*b533ad07SDan McDonald int exitcode;
67d583b39bSJohn Wren Kennedy
68d583b39bSJohn Wren Kennedy /*
69d583b39bSJohn Wren Kennedy * Process Arguments
70d583b39bSJohn Wren Kennedy */
71d583b39bSJohn Wren Kennedy while ((c = getopt(argc, argv, "b:c:d:s:f:o:vwr")) != -1) {
72d583b39bSJohn Wren Kennedy switch (c) {
73d583b39bSJohn Wren Kennedy case 'b':
74d583b39bSJohn Wren Kennedy block_size = atoi(optarg);
75d583b39bSJohn Wren Kennedy break;
76d583b39bSJohn Wren Kennedy case 'c':
77d583b39bSJohn Wren Kennedy write_count = atoi(optarg);
78d583b39bSJohn Wren Kennedy break;
79d583b39bSJohn Wren Kennedy case 'd':
80d583b39bSJohn Wren Kennedy fillchar = atoi(optarg);
81d583b39bSJohn Wren Kennedy break;
82d583b39bSJohn Wren Kennedy case 's':
83d583b39bSJohn Wren Kennedy offset = atoll(optarg);
84d583b39bSJohn Wren Kennedy break;
85d583b39bSJohn Wren Kennedy case 'f':
86d583b39bSJohn Wren Kennedy filename = optarg;
87d583b39bSJohn Wren Kennedy break;
88d583b39bSJohn Wren Kennedy case 'o':
89d583b39bSJohn Wren Kennedy operation = optarg;
90d583b39bSJohn Wren Kennedy break;
91d583b39bSJohn Wren Kennedy case 'v':
92d583b39bSJohn Wren Kennedy verbose = 1;
93d583b39bSJohn Wren Kennedy break;
94d583b39bSJohn Wren Kennedy case 'w':
95d583b39bSJohn Wren Kennedy wsync = 1;
96d583b39bSJohn Wren Kennedy break;
97d583b39bSJohn Wren Kennedy case 'r':
98d583b39bSJohn Wren Kennedy rsync = 1;
99d583b39bSJohn Wren Kennedy break;
100d583b39bSJohn Wren Kennedy case '?':
101d583b39bSJohn Wren Kennedy (void) printf("unknown arg %c\n", optopt);
102d583b39bSJohn Wren Kennedy usage();
103d583b39bSJohn Wren Kennedy break;
104d583b39bSJohn Wren Kennedy }
105d583b39bSJohn Wren Kennedy }
106d583b39bSJohn Wren Kennedy
107d583b39bSJohn Wren Kennedy /*
108d583b39bSJohn Wren Kennedy * Validate Parameters
109d583b39bSJohn Wren Kennedy */
110d583b39bSJohn Wren Kennedy if (!filename) {
111d583b39bSJohn Wren Kennedy (void) printf("Filename not specified (-f <file>)\n");
112d583b39bSJohn Wren Kennedy err++;
113d583b39bSJohn Wren Kennedy }
114d583b39bSJohn Wren Kennedy
115d583b39bSJohn Wren Kennedy if (!operation) {
116d583b39bSJohn Wren Kennedy (void) printf("Operation not specified (-o <operation>).\n");
117d583b39bSJohn Wren Kennedy err++;
118d583b39bSJohn Wren Kennedy }
119d583b39bSJohn Wren Kennedy
120d583b39bSJohn Wren Kennedy if (block_size > BIGBUFFERSIZE) {
121d583b39bSJohn Wren Kennedy (void) printf("block_size is too large max==%d.\n",
122d583b39bSJohn Wren Kennedy BIGBUFFERSIZE);
123d583b39bSJohn Wren Kennedy err++;
124d583b39bSJohn Wren Kennedy }
125d583b39bSJohn Wren Kennedy
126d583b39bSJohn Wren Kennedy if (err) usage();
127d583b39bSJohn Wren Kennedy
128d583b39bSJohn Wren Kennedy /*
129d583b39bSJohn Wren Kennedy * Prepare the buffer and determine the requested operation
130d583b39bSJohn Wren Kennedy */
131d583b39bSJohn Wren Kennedy nxtfillchar = fillchar;
132d583b39bSJohn Wren Kennedy k = 0;
133d583b39bSJohn Wren Kennedy
134d583b39bSJohn Wren Kennedy for (i = 0; i < block_size; i++) {
135d583b39bSJohn Wren Kennedy bigbuffer[i] = nxtfillchar;
136d583b39bSJohn Wren Kennedy
137d583b39bSJohn Wren Kennedy if (fillchar == 0) {
138d583b39bSJohn Wren Kennedy if ((k % DATA_RANGE) == 0) {
139d583b39bSJohn Wren Kennedy k = 0;
140d583b39bSJohn Wren Kennedy }
141d583b39bSJohn Wren Kennedy nxtfillchar = k++;
142d583b39bSJohn Wren Kennedy }
143d583b39bSJohn Wren Kennedy }
144d583b39bSJohn Wren Kennedy
145d583b39bSJohn Wren Kennedy /*
146d583b39bSJohn Wren Kennedy * using the strncmp of operation will make the operation match the
147d583b39bSJohn Wren Kennedy * first shortest match - as the operations are unique from the first
148d583b39bSJohn Wren Kennedy * character this means that we match single character operations
149d583b39bSJohn Wren Kennedy */
150d583b39bSJohn Wren Kennedy if ((strncmp(operation, "create", strlen(operation) + 1)) == 0 ||
151d583b39bSJohn Wren Kennedy (strncmp(operation, "overwrite", strlen(operation) + 1)) == 0) {
152d583b39bSJohn Wren Kennedy oflag = (O_RDWR|O_CREAT);
153d583b39bSJohn Wren Kennedy } else if ((strncmp(operation, "append", strlen(operation) + 1)) == 0) {
154d583b39bSJohn Wren Kennedy oflag = (O_RDWR|O_APPEND);
155d583b39bSJohn Wren Kennedy } else {
156d583b39bSJohn Wren Kennedy (void) printf("valid operations are <create|append> not '%s'\n",
157d583b39bSJohn Wren Kennedy operation);
158d583b39bSJohn Wren Kennedy usage();
159d583b39bSJohn Wren Kennedy }
160d583b39bSJohn Wren Kennedy
161d583b39bSJohn Wren Kennedy if (rsync) {
162d583b39bSJohn Wren Kennedy oflag = oflag | O_RSYNC;
163d583b39bSJohn Wren Kennedy }
164d583b39bSJohn Wren Kennedy
165d583b39bSJohn Wren Kennedy if (wsync) {
166d583b39bSJohn Wren Kennedy oflag = oflag | O_SYNC;
167d583b39bSJohn Wren Kennedy }
168d583b39bSJohn Wren Kennedy
169d583b39bSJohn Wren Kennedy /*
170d583b39bSJohn Wren Kennedy * Given an operation (create/overwrite/append), open the file
171d583b39bSJohn Wren Kennedy * accordingly and perform a write of the appropriate type.
172d583b39bSJohn Wren Kennedy */
173d583b39bSJohn Wren Kennedy if ((bigfd = open(filename, oflag, 0666)) == -1) {
174*b533ad07SDan McDonald exitcode = errno;
175d583b39bSJohn Wren Kennedy (void) printf("open %s: failed [%s]%d. Aborting!\n", filename,
176d583b39bSJohn Wren Kennedy strerror(errno), errno);
177*b533ad07SDan McDonald exit(exitcode);
178d583b39bSJohn Wren Kennedy }
179d583b39bSJohn Wren Kennedy noffset = llseek(bigfd, offset, SEEK_SET);
180d583b39bSJohn Wren Kennedy if (noffset != offset) {
181*b533ad07SDan McDonald exitcode = errno;
182d583b39bSJohn Wren Kennedy (void) printf("llseek %s (%lld/%lld) failed [%s]%d.Aborting!\n",
183d583b39bSJohn Wren Kennedy filename, offset, noffset, strerror(errno), errno);
184*b533ad07SDan McDonald exit(exitcode);
185d583b39bSJohn Wren Kennedy }
186d583b39bSJohn Wren Kennedy
187d583b39bSJohn Wren Kennedy if (verbose) {
188d583b39bSJohn Wren Kennedy (void) printf("%s: block_size = %d, write_count = %d, "
189d583b39bSJohn Wren Kennedy "offset = %lld, data = %s%d\n", filename, block_size,
190d583b39bSJohn Wren Kennedy write_count, offset,
191d583b39bSJohn Wren Kennedy (fillchar == 0) ? "0->" : "",
192d583b39bSJohn Wren Kennedy (fillchar == 0) ? DATA_RANGE : fillchar);
193d583b39bSJohn Wren Kennedy }
194d583b39bSJohn Wren Kennedy
195d583b39bSJohn Wren Kennedy for (i = 0; i < write_count; i++) {
196d583b39bSJohn Wren Kennedy ssize_t n;
197d583b39bSJohn Wren Kennedy
198d583b39bSJohn Wren Kennedy if ((n = write(bigfd, &bigbuffer, block_size)) == -1) {
199*b533ad07SDan McDonald exitcode = errno;
200d583b39bSJohn Wren Kennedy (void) printf("write failed (%ld), good_writes = %lld, "
201d583b39bSJohn Wren Kennedy "error: %s[%d]\n", (long)n, good_writes,
202d583b39bSJohn Wren Kennedy strerror(errno),
203d583b39bSJohn Wren Kennedy errno);
204*b533ad07SDan McDonald exit(exitcode);
205d583b39bSJohn Wren Kennedy }
206d583b39bSJohn Wren Kennedy good_writes++;
207d583b39bSJohn Wren Kennedy }
208d583b39bSJohn Wren Kennedy
209d583b39bSJohn Wren Kennedy if (verbose) {
210d583b39bSJohn Wren Kennedy (void) printf("Success: good_writes = %lld (%lld)\n",
211d583b39bSJohn Wren Kennedy good_writes, (good_writes * block_size));
212d583b39bSJohn Wren Kennedy }
213d583b39bSJohn Wren Kennedy
214d583b39bSJohn Wren Kennedy return (0);
215d583b39bSJohn Wren Kennedy }
216d583b39bSJohn Wren Kennedy
217d583b39bSJohn Wren Kennedy static void
usage(void)218d583b39bSJohn Wren Kennedy usage(void)
219d583b39bSJohn Wren Kennedy {
220d583b39bSJohn Wren Kennedy char *base = (char *)"file_write";
221d583b39bSJohn Wren Kennedy char *exec = (char *)getexecname();
222d583b39bSJohn Wren Kennedy
223d583b39bSJohn Wren Kennedy if (exec != NULL)
224d583b39bSJohn Wren Kennedy exec = strdup(exec);
225d583b39bSJohn Wren Kennedy if (exec != NULL)
226d583b39bSJohn Wren Kennedy base = basename(exec);
227d583b39bSJohn Wren Kennedy
228d583b39bSJohn Wren Kennedy (void) printf("Usage: %s [-v] -o {create,overwrite,append} -f file_name"
229d583b39bSJohn Wren Kennedy " [-b block_size]\n"
230d583b39bSJohn Wren Kennedy "\t[-s offset] [-c write_count] [-d data]\n"
231d583b39bSJohn Wren Kennedy "\twhere [data] equal to zero causes chars "
232d583b39bSJohn Wren Kennedy "0->%d to be repeated throughout\n", base, DATA_RANGE);
233d583b39bSJohn Wren Kennedy
234d583b39bSJohn Wren Kennedy if (exec) {
235d583b39bSJohn Wren Kennedy free(exec);
236d583b39bSJohn Wren Kennedy }
237d583b39bSJohn Wren Kennedy
238d583b39bSJohn Wren Kennedy exit(1);
239d583b39bSJohn Wren Kennedy }
240