xref: /freebsd-src/sys/contrib/openzfs/tests/zfs-tests/cmd/file/randfree_file.c (revision 271171e0d97b88ba2a7c3bf750c9672b484c1c13)
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
9*271171e0SMartin 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 by Delphix. All rights reserved.
29716fd348SMartin Matuska  */
30716fd348SMartin Matuska 
31716fd348SMartin Matuska #include "file_common.h"
32716fd348SMartin Matuska #include <sys/types.h>
33716fd348SMartin Matuska #include <unistd.h>
34716fd348SMartin Matuska #include <fcntl.h>
35716fd348SMartin Matuska #include <string.h>
36716fd348SMartin Matuska #include <linux/falloc.h>
37716fd348SMartin Matuska 
38716fd348SMartin Matuska /*
39716fd348SMartin Matuska  * Create a file with assigned size and then free the specified
40716fd348SMartin Matuska  * section of the file
41716fd348SMartin Matuska  */
42716fd348SMartin Matuska 
43716fd348SMartin Matuska static void usage(char *progname);
44716fd348SMartin Matuska 
45716fd348SMartin Matuska static void
usage(char * progname)46716fd348SMartin Matuska usage(char *progname)
47716fd348SMartin Matuska {
48716fd348SMartin Matuska 	(void) fprintf(stderr,
49716fd348SMartin Matuska 	    "usage: %s [-l filesize] [-s start-offset]"
50716fd348SMartin Matuska 	    "[-n section-len] filename\n", progname);
51716fd348SMartin Matuska 	exit(1);
52716fd348SMartin Matuska }
53716fd348SMartin Matuska 
54716fd348SMartin Matuska int
main(int argc,char * argv[])55716fd348SMartin Matuska main(int argc, char *argv[])
56716fd348SMartin Matuska {
57716fd348SMartin Matuska 	char *filename = NULL;
58716fd348SMartin Matuska 	char *buf = NULL;
59716fd348SMartin Matuska 	size_t filesize = 0;
60716fd348SMartin Matuska 	off_t start_off = 0;
61716fd348SMartin Matuska 	off_t off_len = 0;
62716fd348SMartin Matuska 	int  fd, ch;
63716fd348SMartin Matuska 	mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
64716fd348SMartin Matuska 
65716fd348SMartin Matuska 	while ((ch = getopt(argc, argv, "l:s:n:")) != EOF) {
66716fd348SMartin Matuska 		switch (ch) {
67716fd348SMartin Matuska 		case 'l':
68716fd348SMartin Matuska 			filesize = atoll(optarg);
69716fd348SMartin Matuska 			break;
70716fd348SMartin Matuska 		case 's':
71716fd348SMartin Matuska 			start_off = atoll(optarg);
72716fd348SMartin Matuska 			break;
73716fd348SMartin Matuska 		case 'n':
74716fd348SMartin Matuska 			off_len = atoll(optarg);
75716fd348SMartin Matuska 			break;
76716fd348SMartin Matuska 		default:
77716fd348SMartin Matuska 			usage(argv[0]);
78716fd348SMartin Matuska 			break;
79716fd348SMartin Matuska 		}
80716fd348SMartin Matuska 	}
81716fd348SMartin Matuska 
82716fd348SMartin Matuska 	if (optind == argc - 1)
83716fd348SMartin Matuska 		filename = argv[optind];
84716fd348SMartin Matuska 	else
85716fd348SMartin Matuska 		usage(argv[0]);
86716fd348SMartin Matuska 
87716fd348SMartin Matuska 	if ((fd = open(filename, O_RDWR | O_CREAT | O_TRUNC, mode)) < 0) {
88716fd348SMartin Matuska 		perror("open");
89716fd348SMartin Matuska 		return (1);
90716fd348SMartin Matuska 	}
91716fd348SMartin Matuska 
92716fd348SMartin Matuska 	buf = (char *)calloc(1, filesize);
93716fd348SMartin Matuska 	if (buf == NULL) {
94716fd348SMartin Matuska 		perror("write");
95716fd348SMartin Matuska 		close(fd);
96716fd348SMartin Matuska 		return (1);
97716fd348SMartin Matuska 	}
98716fd348SMartin Matuska 	memset(buf, 'c', filesize);
99716fd348SMartin Matuska 
100716fd348SMartin Matuska 	if (write(fd, buf, filesize) < filesize) {
101716fd348SMartin Matuska 		free(buf);
102716fd348SMartin Matuska 		perror("write");
103716fd348SMartin Matuska 		close(fd);
104716fd348SMartin Matuska 		return (1);
105716fd348SMartin Matuska 	}
106716fd348SMartin Matuska 
107716fd348SMartin Matuska 	free(buf);
108716fd348SMartin Matuska 
109716fd348SMartin Matuska #if defined(FALLOC_FL_PUNCH_HOLE) && defined(FALLOC_FL_KEEP_SIZE)
110716fd348SMartin Matuska 	if (fallocate(fd, FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE,
111716fd348SMartin Matuska 	    start_off, off_len) < 0) {
112716fd348SMartin Matuska 		perror("fallocate");
113716fd348SMartin Matuska 		close(fd);
114716fd348SMartin Matuska 		return (1);
115716fd348SMartin Matuska 	}
116716fd348SMartin Matuska #else /* !(defined(FALLOC_FL_PUNCH_HOLE) && defined(FALLOC_FL_KEEP_SIZE)) */
117716fd348SMartin Matuska 	{
118716fd348SMartin Matuska 		perror("FALLOC_FL_PUNCH_HOLE unsupported");
119716fd348SMartin Matuska 		close(fd);
120716fd348SMartin Matuska 		return (1);
121716fd348SMartin Matuska 	}
122716fd348SMartin Matuska #endif /* defined(FALLOC_FL_PUNCH_HOLE) && defined(FALLOC_FL_KEEP_SIZE) */
123716fd348SMartin Matuska 	close(fd);
124716fd348SMartin Matuska 	return (0);
125716fd348SMartin Matuska }
126