xref: /illumos-gate/usr/src/test/zfs-tests/cmd/mkfiles/mkfiles.c (revision a61ed2ce7a86a4d6428f2a83eb4739fae945447e)
1 /*
2  * This file and its contents are supplied under the terms of the
3  * Common Development and Distribution License ("CDDL"), version 1.0.
4  * You may only use this file in accordance with the terms of version
5  * 1.0 of the CDDL.
6  *
7  * A full copy of the text of the CDDL should have accompanied this
8  * source.  A copy of the CDDL is also available via the Internet at
9  * http://www.illumos.org/license/CDDL.
10  */
11 
12 /*
13  * Copyright (c) 2016 by Delphix. All rights reserved.
14  */
15 
16 #include <stdio.h>
17 #include <stdlib.h>
18 #include <fcntl.h>
19 #include <string.h>
20 #include <errno.h>
21 #include <unistd.h>
22 #include <sys/param.h>
23 
24 #define	MAX_INT_LENGTH 10
25 
26 static void
27 usage(char *msg, int exit_value)
28 {
29 	(void) fprintf(stderr, "mkfiles basename max_file [min_file]\n");
30 	(void) fprintf(stderr, "%s\n", msg);
31 	exit(exit_value);
32 }
33 
34 int
35 main(int argc, char **argv)
36 {
37 	unsigned int numfiles = 0;
38 	unsigned int first_file = 0;
39 	unsigned int i;
40 	char buf[MAXPATHLEN];
41 
42 	if (argc < 3 || argc > 4)
43 		usage("Invalid number of arguments", -1);
44 
45 	if (sscanf(argv[2], "%u", &numfiles) != 1)
46 		usage("Invalid maximum file", -2);
47 
48 	if (argc == 4 && sscanf(argv[3], "%u", &first_file) != 1)
49 		usage("Invalid first file", -3);
50 
51 	if (numfiles < first_file)
52 		usage("First file larger than last file", -3);
53 
54 	for (i = first_file; i < first_file + numfiles; i++) {
55 		int fd;
56 		(void) snprintf(buf, MAXPATHLEN, "%s%u", argv[1], i);
57 		if ((fd = open(buf, O_CREAT | O_EXCL, O_RDWR)) == -1) {
58 			(void) fprintf(stderr, "Failed to create %s %s\n", buf,
59 			    strerror(errno));
60 			return (-4);
61 		}
62 		(void) close(fd);
63 	}
64 	return (0);
65 }
66