miscutils/makedevs.c

   1 /* vi: set sw=4 ts=4: */
   2
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <fcntl.h>
   7 #include <getopt.h>
   8 #include <time.h>
   9 #include <unistd.h>
  10 #include <sys/types.h>
  11 #include <sys/sysmacros.h>     /* major() and minor() */
  12 #include "busybox.h"
  13
  14 #ifdef CONFIG_FEATURE_MAKEDEVS_LEAF
  15 /*
  16  * public domain -- Dave 'Kill a Cop' Cinege <dcinege@psychosis.com>
  17  *
  18  * makedevs
  19  * Make ranges of device files quickly.
  20  * known bugs: can't deal with alpha ranges
  21  */
  22 int makedevs_main(int argc, char **argv)
  23 {
  24         mode_t mode;
  25         char *basedev, *type, *nodname, buf[255];
  26         int Smajor, Sminor, S, E;
  27
  28         if (argc < 7 || *argv[1]=='-')
  29                 bb_show_usage();
  30
  31         basedev = argv[1];
  32         type = argv[2];
  33         Smajor = atoi(argv[3]);
  34         Sminor = atoi(argv[4]);
  35         S = atoi(argv[5]);
  36         E = atoi(argv[6]);
  37         nodname = argc == 8 ? basedev : buf;
  38
  39         mode = 0660;
  40
  41         switch (type[0]) {
  42         case 'c':
  43                 mode |= S_IFCHR;
  44                 break;
  45         case 'b':
  46                 mode |= S_IFBLK;
  47                 break;
  48         case 'f':
  49                 mode |= S_IFIFO;
  50                 break;
  51         default:
  52                 bb_show_usage();
  53         }
  54
  55         while (S <= E) {
  56                 int sz;
  57
  58                 sz = snprintf(buf, sizeof(buf), "%s%d", basedev, S);
  59                 if(sz<0 || sz>=sizeof(buf))  /* libc different */
  60                         bb_error_msg_and_die("%s too large", basedev);
  61
  62         /* if mode != S_IFCHR and != S_IFBLK third param in mknod() ignored */
  63
  64                 if (mknod(nodname, mode, makedev(Smajor, Sminor)))
  65                         bb_error_msg("Failed to create: %s", nodname);
  66
  67                 if (nodname == basedev) /* ex. /dev/hda - to /dev/hda1 ... */
  68                         nodname = buf;
  69                 S++;
  70                 Sminor++;
  71         }
  72
  73         return 0;
  74 }
  75
  76 #elif defined CONFIG_FEATURE_MAKEDEVS_TABLE
  77
  78 /*
  79  *  This program is free software; you can redistribute it and/or modify
  80  *  it under the terms of the GNU General Public License version 2 as
  81  *  published by the Free Software Foundation.
  82  *
  83  *  This program is distributed in the hope that it will be useful,
  84  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  85  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  86  *  GNU Library General Public License for more details.
  87  *
  88  *  You should have received a copy of the GNU General Public License
  89  *  along with this program; if not, write to the Free Software
  90  *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  91  *
  92  */
  93
  94 extern int makedevs_main(int argc, char **argv)
  95 {
  96         int opt;
  97         FILE *table = stdin;
  98         char *rootdir = NULL;
  99         char *line = NULL;
 100         int linenum = 0;
 101         int ret = EXIT_SUCCESS;
 102
 103         while ((opt = getopt(argc, argv, "d:")) != -1) {
 104                 switch(opt) {
 105                         case 'd':
 106                                 table = bb_xfopen((line=optarg), "r");
 107                                 break;
 108                         default:
 109                                 bb_show_usage();
 110                 }
 111         }
 112
 113         if (optind >= argc || (rootdir=argv[optind])==NULL) {
 114                 bb_error_msg_and_die("root directory not speficied");
 115         }
 116
 117         if (chdir(rootdir) != 0) {
 118                 bb_perror_msg_and_die("Couldnt chdir to %s", rootdir);
 119         }
 120
 121         umask(0);
 122
 123         printf("rootdir=%s\n", rootdir);
 124         if (line) {
 125                 printf("table='%s'\n", line);
 126         } else {
 127                 printf("table=<stdin>\n");
 128         }
 129
 130         while ((line = bb_get_chomped_line_from_file(table))) {
 131                 char type;
 132                 unsigned int mode = 0755;
 133                 unsigned int major = 0;
 134                 unsigned int minor = 0;
 135                 unsigned int count = 0;
 136                 unsigned int increment = 0;
 137                 unsigned int start = 0;
 138                 char name[41];
 139                 char user[41];
 140                 char group[41];
 141                 char *full_name;
 142                 uid_t uid;
 143                 gid_t gid;
 144
 145                 linenum++;
 146
 147                 if ((2 > sscanf(line, "%40s %c %o %40s %40s %u %u %u %u %u", name,
 148                                                 &type, &mode, user, group, &major,
 149                                                 &minor, &start, &increment, &count)) ||
 150                                 ((major | minor | start | count | increment) > 255))
 151                 {
 152                         if (*line=='\0' || *line=='#' || isspace(*line))
 153                                 continue;
 154                         bb_error_msg("line %d invalid: '%s'\n", linenum, line);
 155                         ret = EXIT_FAILURE;
 156                         continue;
 157                 }
 158                 if (name[0] == '#') {
 159                         continue;
 160                 }
 161                 if (group) {
 162                         gid = get_ug_id(group, my_getgrnam);
 163                 } else {
 164                         gid = getgid();
 165                 }
 166                 if (user) {
 167                         uid = get_ug_id(user, my_getpwnam);
 168                 } else {
 169                         uid = getuid();
 170                 }
 171                 full_name = concat_path_file(rootdir, name);
 172
 173                 if (type == 'd') {
 174                         bb_make_directory(full_name, mode | S_IFDIR, FILEUTILS_RECUR);
 175                         if (chown(full_name, uid, gid) == -1) {
 176                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
 177                                 ret = EXIT_FAILURE;
 178                                 goto loop;
 179                         }
 180                 } else {
 181                         dev_t rdev;
 182
 183                         if (type == 'p') {
 184                                 mode |= S_IFIFO;
 185                         }
 186                         else if (type == 'c') {
 187                                 mode |= S_IFCHR;
 188                         }
 189                         else if (type == 'b') {
 190                                 mode |= S_IFBLK;
 191                         } else {
 192                                 bb_error_msg("line %d: Unsupported file type %c", linenum, type);
 193                                 ret = EXIT_FAILURE;
 194                                 goto loop;
 195                         }
 196
 197                         if (count > 0) {
 198                                 int i;
 199                                 char *full_name_inc;
 200
 201                                 full_name_inc = xmalloc(strlen(full_name) + 4);
 202                                 for (i = start; i < count; i++) {
 203                                         sprintf(full_name_inc, "%s%d", full_name, i);
 204                                         rdev = (major << 8) + minor + (i * increment - start);
 205                                         if (mknod(full_name_inc, mode, rdev) == -1) {
 206                                                 bb_perror_msg("line %d: Couldnt create node %s", linenum, full_name_inc);
 207                                                 ret = EXIT_FAILURE;
 208                                         }
 209                                         else if (chown(full_name_inc, uid, gid) == -1) {
 210                                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name_inc);
 211                                                 ret = EXIT_FAILURE;
 212                                         }
 213                                 }
 214                                 free(full_name_inc);
 215                         } else {
 216                                 rdev = (major << 8) + minor;
 217                                 if (mknod(full_name, mode, rdev) == -1) {
 218                                         bb_perror_msg("line %d: Couldnt create node %s", linenum, full_name);
 219                                         ret = EXIT_FAILURE;
 220                                 }
 221                                 else if (chown(full_name, uid, gid) == -1) {
 222                                         bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
 223                                         ret = EXIT_FAILURE;
 224                                 }
 225                         }
 226                 }
 227 loop:
 228                 free(line);
 229                 free(full_name);
 230         }
 231         fclose(table);
 232
 233         return 0;
 234 }
 235
 236 #else
 237 # error makdedevs configuration error, either leaf or table must be selected
 238 #endif