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) 2012, 2014 by Delphix. All rights reserved.
14 * Copyright 2017, loli10K <ezomori.nozomu@gmail.com>. All rights reserved.
15 */
16
17 #include <fcntl.h>
18 #include <sys/stat.h>
19 #include <sys/types.h>
20 #include <unistd.h>
21 #include <errno.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24
25 #define FSIZE 256*1024*1024
26
27 static long fsize = FSIZE;
28 static int errflag = 0;
29 static char *filename = NULL;
30 static int ftruncflag = 0;
31
32 static void parse_options(int argc, char *argv[]);
33
34 static void
usage(char * execname)35 usage(char *execname)
36 {
37 (void) fprintf(stderr,
38 "usage: %s [-s filesize] [-f] /path/to/file\n", execname);
39 (void) exit(1);
40 }
41
42 int
main(int argc,char * argv[])43 main(int argc, char *argv[])
44 {
45 int fd;
46
47 parse_options(argc, argv);
48
49 if (ftruncflag) {
50 fd = open(filename, O_RDWR|O_CREAT, 0666);
51 if (fd < 0) {
52 perror("open");
53 return (1);
54 }
55 if (ftruncate(fd, fsize) < 0) {
56 perror("ftruncate");
57 return (1);
58 }
59 if (close(fd)) {
60 perror("close");
61 return (1);
62 }
63 } else {
64 if (truncate(filename, fsize) < 0) {
65 perror("truncate");
66 return (1);
67 }
68 }
69 return (0);
70 }
71
72 static void
parse_options(int argc,char * argv[])73 parse_options(int argc, char *argv[])
74 {
75 int c;
76 extern char *optarg;
77 extern int optind, optopt;
78
79 while ((c = getopt(argc, argv, "s:f")) != -1) {
80 switch (c) {
81 case 's':
82 fsize = atoi(optarg);
83 break;
84 case 'f':
85 ftruncflag++;
86 break;
87 case ':':
88 (void) fprintf(stderr,
89 "Option -%c requires an operand\n", optopt);
90 errflag++;
91 break;
92 }
93 if (errflag) {
94 (void) usage(argv[0]);
95 }
96 }
97
98 if (argc <= optind) {
99 (void) fprintf(stderr, "No filename specified\n");
100 usage(argv[0]);
101 }
102 filename = argv[optind];
103 }
104