aboutsummaryrefslogtreecommitdiffstats
path: root/tests/zfs-tests/cmd/mkfiles/mkfiles.c
blob: 32abfd0c3d67c7be4b8973bff71a1a676bc508fd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright (c) 2016 by Delphix. All rights reserved.
 */

#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <string.h>
#include <errno.h>
#include <unistd.h>
#include <sys/param.h>

#define	MAX_INT_LENGTH 10

static void
usage(char *msg, int exit_value)
{
	(void) fprintf(stderr, "mkfiles basename max_file [min_file]\n");
	(void) fprintf(stderr, "%s\n", msg);
	exit(exit_value);
}

int
main(int argc, char **argv)
{
	unsigned int numfiles = 0;
	unsigned int first_file = 0;
	unsigned int i;
	char buf[MAXPATHLEN];

	if (argc < 3 || argc > 4)
		usage("Invalid number of arguments", -1);

	if (sscanf(argv[2], "%u", &numfiles) != 1)
		usage("Invalid maximum file", -2);

	if (argc == 4 && sscanf(argv[3], "%u", &first_file) != 1)
		usage("Invalid first file", -3);

	for (i = first_file; i < first_file + numfiles; i++) {
		int fd;
		(void) snprintf(buf, MAXPATHLEN, "%s%u", argv[1], i);
		if ((fd = open(buf, O_CREAT | O_EXCL, O_RDWR)) == -1) {
			(void) fprintf(stderr, "Failed to create %s %s\n", buf,
			    strerror(errno));
			return (-4);
		} else if (fchown(fd, getuid(), getgid()) < 0) {
			(void) fprintf(stderr, "Failed to chown %s %s\n", buf,
			    strerror(errno));
			return (-5);
		}
		(void) close(fd);
	}
	return (0);
}