mkfile.tproj/mkfile.c

   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
  48 extern void usage(char *, char *);
  49 extern void create_file(char *, quad_t, int, int);
  50 extern void err_rm(char *, char *);
  51
  52 int
  53 main (argc, argv)
  54         int argc;
  55         char **argv;
  56 {
  57         char *b_num, *prog_name;
  58         char *options = "nv";
  59         char c;
  60         quad_t multiplier = 1;
  61         quad_t file_size;
  62         int len;
  63         int empty = 0;
  64         int verbose = 0;
  65
  66         prog_name = argv[0];    /* Get program name */
  67     if (1 == argc)
  68                 usage(prog_name, options);
  69
  70         /* Get options */
  71         opterr=1;
  72
  73         while ((c=getopt(argc, argv, options)) != EOF)
  74                 switch (c) {
  75                 case 'v':   /* Turn on verbose setting */
  76                         verbose = 1;
  77                         break;
  78                 case 'n':   /* Create an empty file */
  79                         empty = 1;
  80                         break;
  81                 default:
  82                         usage(prog_name, options);
  83                         break;
  84                 }
  85
  86         /* Stop getting options
  87         */
  88         argv += optind;
  89         if (*argv == NULL)              /* Is there a size given? */
  90                 usage(prog_name, options);
  91
  92         b_num = *argv++;                /* Size of file and byte multiplier */
  93         len = strlen(b_num) - 1;
  94
  95         if (!isdigit(b_num[len])) {
  96                 switch(b_num[len]) {    /* Figure out multiplier */
  97                         case 'B':
  98                         case 'b':
  99                                 multiplier = 512;
 100                                 break;
 101                         case 'K':
 102                         case 'k':
 103                                 multiplier = 1024;
 104                                 break;
 105                         case 'M':
 106                         case 'm':
 107                                 multiplier = 1024 * 1024;
 108                                 break;
 109                         case 'G':
 110                         case 'g':
 111                                 multiplier = 1024 * 1024 * 1024;
 112                                 break;
 113                         default:
 114                                 usage(prog_name, options);
 115                 }
 116         }
 117
 118         if (*argv == NULL)              /* Was a file name given? */
 119                 usage(prog_name, options);
 120
 121         if ((file_size = strtoq(b_num, NULL, 10)) == 0 )
 122                 err(1, "Bad file size!");
 123
 124         while ( *argv != NULL ) {       /* Create file for each file_name */
 125                 create_file(*argv, file_size*multiplier, empty, verbose);
 126                 argv++;
 127         }
 128
 129         return (0);
 130
 131 }
 132
 133
 134 /* Create a file and make it empty (lseek) or zero'd */
 135
 136 void
 137 create_file(file_name, size, empty, verbose)
 138         char *file_name;
 139         quad_t size;
 140         int empty;
 141         int verbose;
 142 {
 143         char buff[BF_SZ];
 144         int fd, bytes_written = BF_SZ;
 145         quad_t i;
 146         mode_t mode = S_IRUSR | S_IWUSR;
 147
 148         /* If superuser, then set sticky bit */
 149         if (!geteuid()) mode |= S_ISVTX;
 150
 151         if ((fd = open(file_name, O_RDWR | O_CREAT | O_TRUNC, mode)) == -1)
 152                 err(1, NULL);
 153
 154
 155         if (empty) {                    /* Create an empty file */
 156                 lseek(fd, (off_t)size-1, SEEK_SET);
 157                 if ( 1 != write(fd, "\0", 1))
 158                         err_rm(file_name, "Write Error");
 159         }
 160         else {
 161                 bzero(buff, BF_SZ);
 162
 163                 /*
 164                  * First loop: write BF_SZ chunks until you have
 165                  * less then BF_SZ bytes to write.
 166                  * Second loop: write the remaining bytes.
 167                  * ERRORS in the write process will cause the
 168                  * file to be removed before the error is
 169                  * reported.
 170                  */
 171                 for (i = size; i > BF_SZ; i -= bytes_written) {
 172                         bytes_written = write (fd, buff, BF_SZ);
 173                         if ( bytes_written == -1 )
 174                                 err_rm (file_name, "Write Error");
 175                 }
 176                 for (; i > 0; i -= bytes_written) {
 177                         bytes_written = write (fd, buff, i);
 178                         if ( bytes_written == -1 )
 179                                 err_rm (file_name, "Write Error");
 180                 }
 181         }
 182
 183         if (fchmod(fd, mode))   /* Change permissions */
 184                 err_rm(file_name, NULL);
 185
 186         if ((close(fd)) == -1)
 187                 err_rm(file_name, NULL);
 188
 189         if (verbose)
 190                 (void)fprintf(stderr, "%s %qd bytes\n", file_name, size);
 191
 192 }
 193
 194 /* On error remove the file */
 195
 196 void
 197 err_rm(filename, msg)
 198         char *filename;
 199         char *msg;
 200 {
 201         unlink(filename);
 202         err(1, "(%s removed) %s", filename, msg);
 203 }
 204
 205
 206 /* Print usage string */
 207 void
 208 usage (prog_name, options)
 209         char *prog_name;
 210         char *options;
 211 {
 212         (void)fprintf(stderr,
 213                 "usage: %s [-%s] size[b|k|m|g] filename ...\n", prog_name,  options);
 214         exit(1);
 215
 216 }