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