1/*
2 * Copyright (c) 1999 Apple Computer, Inc. All rights reserved.
3 *
4 * @APPLE_LICENSE_HEADER_START@
5 *
6 * "Portions Copyright (c) 1999 Apple Computer, Inc.  All Rights
7 * Reserved.  This file contains Original Code and/or Modifications of
8 * Original Code as defined in and that are subject to the Apple Public
9 * Source License Version 1.0 (the 'License').  You may not use this file
10 * except in compliance with the License.  Please obtain a copy of the
11 * License at http://www.apple.com/publicsource and read it before using
12 * this file.
13 *
14 * The Original Code and all software distributed under the License are
15 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
16 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
17 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
19 * License for the specific language governing rights and limitations
20 * under the License."
21 *
22 * @APPLE_LICENSE_HEADER_END@
23 */
24/*
25 * Copyright (c) 1997 Apple Computer, Inc. All Rights Reserved
26 *
27 * HISTORY
28 * 29-Aug-97 Daniel Wade (danielw) at Apple
29 *	Created.
30 *
31 */
32
33
34
35#include <sys/types.h>
36#include <sys/stat.h>
37#include <fcntl.h>
38#include <errno.h>
39#include <stdio.h>
40#include <stdlib.h>
41#include <string.h>
42#include <unistd.h>
43#include <ctype.h>
44#include <err.h>
45
46#define BF_SZ 	512	/* Size of write chunks */
47
48extern void usage(char *, char *);
49extern void create_file(char *, quad_t, int, int);
50extern void err_rm(char *, char *);
51
52int
53main (argc, argv)
54	int argc;
55	char **argv;
56{
57	char *b_num, *prog_name;
58	char *options = "nv";
59	char c;
60	off_t multiplier = 1;
61	off_t file_size;
62	int len;
63	int empty = 0;
64	int verbose = 0;
65	char* endptr = NULL;
66
67	prog_name = argv[0];	/* Get program name */
68    if (1 == argc)
69		usage(prog_name, options);
70
71	/* Get options */
72	opterr=1;
73
74    	while ((c=getopt(argc, argv, options)) != EOF)
75        	switch (c) {
76        	case 'v':   /* Turn on verbose setting */
77			verbose = 1;
78			break;
79		case 'n':   /* Create an empty file */
80			empty = 1;
81			break;
82		default:
83			usage(prog_name, options);
84			break;
85		}
86
87	/* Stop getting options
88	*/
89	argv += optind;
90	if (*argv == NULL)		/* Is there a size given? */
91		usage(prog_name, options);
92
93	b_num = *argv++;		/* Size of file and byte multiplier */
94	len = strlen(b_num) - 1;
95
96	if (!isdigit(b_num[len])) {
97                switch(b_num[len]) {	/* Figure out multiplier */
98			case 'B':
99                        case 'b':
100                                multiplier = 512;
101                                break;
102			case 'K':
103                        case 'k':
104                                multiplier = 1024;
105                        	break;
106			case 'M':
107                        case 'm':
108                                multiplier = 1024 * 1024;
109                                break;
110			case 'G':
111                        case 'g':
112                                multiplier = 1024 * 1024 * 1024;
113                                break;
114                        default:
115                        	usage(prog_name, options);
116                }
117	}
118
119	if (*argv == NULL)		/* Was a file name given? */
120		usage(prog_name, options);
121
122	if ((file_size = strtoll(b_num, &endptr, 10)) == 0 &&
123		(*endptr != 0 && endptr != &b_num[len])) {
124		err(1, "Bad file size!");
125	}
126
127	while ( *argv != NULL ) {	/* Create file for each file_name */
128		create_file(*argv, file_size*multiplier, empty, verbose);
129		argv++;
130	}
131
132	return (0);
133
134}
135
136
137/* Create a file and make it empty (lseek) or zero'd */
138
139void
140create_file(file_name, size, empty, verbose)
141	char *file_name;
142	quad_t size;
143	int empty;
144	int verbose;
145{
146	char buff[BF_SZ];
147	int fd, bytes_written = BF_SZ;
148	quad_t i;
149	mode_t mode = S_IRUSR | S_IWUSR;
150
151	/* If superuser, then set sticky bit */
152	if (!geteuid()) mode |= S_ISVTX;
153
154	if ((fd = open(file_name, O_RDWR | O_CREAT | O_TRUNC, mode)) == -1)
155                err(1, NULL);
156
157
158        if (empty) {			/* Create an empty file */
159                lseek(fd, (off_t)size-1, SEEK_SET);
160                if ( 1 != write(fd, "\0", 1))
161			err_rm(file_name, "Write Error");
162        }
163	else {
164		bzero(buff, BF_SZ);
165
166		/*
167		 * First loop: write BF_SZ chunks until you have
168		 * less then BF_SZ bytes to write.
169		 * Second loop: write the remaining bytes.
170		 * ERRORS in the write process will cause the
171		 * file to be removed before the error is
172		 * reported.
173		 */
174		for (i = size; i > BF_SZ; i -= bytes_written) {
175			bytes_written = write (fd, buff, BF_SZ);
176			if ( bytes_written == -1 )
177                                err_rm (file_name, "Write Error");
178		}
179		for (; i > 0; i -= bytes_written) {
180                   	bytes_written = write (fd, buff, i);
181                        if ( bytes_written == -1 )
182                                err_rm (file_name, "Write Error");
183		}
184	}
185
186	if (fchmod(fd, mode))	/* Change permissions */
187		err_rm(file_name, NULL);
188
189	if ((close(fd)) == -1)
190		err_rm(file_name, NULL);
191
192	if (verbose)
193		(void)fprintf(stderr, "%s %qd bytes\n", file_name, size);
194
195}
196
197/* On error remove the file */
198
199void
200err_rm(filename, msg)
201	char *filename;
202	char *msg;
203{
204	unlink(filename);
205	err(1, "(%s removed) %s", filename, msg);
206}
207
208
209/* Print usage string */
210void
211usage (prog_name, options)
212	char *prog_name;
213	char *options;
214{
215	(void)fprintf(stderr,
216		"usage: %s [-%s] size[b|k|m|g] filename ...\n", prog_name,  options);
217	exit(1);
218
219}
220