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 <time.h>
   8 #include <ctype.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 /* Licensed under the GPL v2 or later, see the file LICENSE in this tarball. */
  79
  80 extern int makedevs_main(int argc, char **argv)
  81 {
  82         FILE *table = stdin;
  83         char *rootdir = NULL;
  84         char *line = NULL;
  85         int linenum = 0;
  86         int ret = EXIT_SUCCESS;
  87
  88         unsigned long flags;
  89         flags = bb_getopt_ulflags(argc, argv, "d:", &line);
  90         if (line)
  91                 table = bb_xfopen(line, "r");
  92
  93         if (optind >= argc || (rootdir=argv[optind])==NULL) {
  94                 bb_error_msg_and_die("root directory not specified");
  95         }
  96
  97         if (chdir(rootdir) != 0) {
  98                 bb_perror_msg_and_die("could not chdir to %s", rootdir);
  99         }
 100
 101         umask(0);
 102
 103         printf("rootdir=%s\n", rootdir);
 104         if (line) {
 105                 printf("table='%s'\n", line);
 106         } else {
 107                 printf("table=<stdin>\n");
 108         }
 109
 110         while ((line = bb_get_chomped_line_from_file(table))) {
 111                 char type;
 112                 unsigned int mode = 0755;
 113                 unsigned int major = 0;
 114                 unsigned int minor = 0;
 115                 unsigned int count = 0;
 116                 unsigned int increment = 0;
 117                 unsigned int start = 0;
 118                 char name[41];
 119                 char user[41];
 120                 char group[41];
 121                 char *full_name;
 122                 uid_t uid;
 123                 gid_t gid;
 124
 125                 linenum++;
 126
 127                 if ((2 > sscanf(line, "%40s %c %o %40s %40s %u %u %u %u %u", name,
 128                                                 &type, &mode, user, group, &major,
 129                                                 &minor, &start, &increment, &count)) ||
 130                                 ((major | minor | start | count | increment) > 255))
 131                 {
 132                         if (*line=='\0' || *line=='#' || isspace(*line))
 133                                 continue;
 134                         bb_error_msg("line %d invalid: '%s'\n", linenum, line);
 135                         ret = EXIT_FAILURE;
 136                         continue;
 137                 }
 138                 if (name[0] == '#') {
 139                         continue;
 140                 }
 141
 142                 gid = (*group) ? get_ug_id(group, bb_xgetgrnam) : getgid();
 143                 uid = (*user) ? get_ug_id(user, bb_xgetpwnam) : getuid();
 144                 full_name = concat_path_file(rootdir, name);
 145
 146                 if (type == 'd') {
 147                         bb_make_directory(full_name, mode | S_IFDIR, FILEUTILS_RECUR);
 148                         if (chown(full_name, uid, gid) == -1) {
 149                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
 150                                 ret = EXIT_FAILURE;
 151                                 goto loop;
 152                         }
 153                         if ((mode != -1) && (chmod(full_name, mode) < 0)){
 154                                 bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
 155                                 ret = EXIT_FAILURE;
 156                                 goto loop;
 157                         }
 158                 } else if (type == 'f') {
 159                         struct stat st;
 160                         if ((stat(full_name, &st) < 0 || !S_ISREG(st.st_mode))) {
 161                                 bb_perror_msg("line %d: regular file '%s' does not exist", linenum, full_name);
 162                                 ret = EXIT_FAILURE;
 163                                 goto loop;
 164                         }
 165                         if (chown(full_name, uid, gid) == -1) {
 166                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
 167                                 ret = EXIT_FAILURE;
 168                                 goto loop;
 169                         }
 170                         if ((mode != -1) && (chmod(full_name, mode) < 0)){
 171                                 bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
 172                                 ret = EXIT_FAILURE;
 173                                 goto loop;
 174                         }
 175                 } else
 176                 {
 177                         dev_t rdev;
 178
 179                         if (type == 'p') {
 180                                 mode |= S_IFIFO;
 181                         }
 182                         else if (type == 'c') {
 183                                 mode |= S_IFCHR;
 184                         }
 185                         else if (type == 'b') {
 186                                 mode |= S_IFBLK;
 187                         } else {
 188                                 bb_error_msg("line %d: unsupported file type %c", linenum, type);
 189                                 ret = EXIT_FAILURE;
 190                                 goto loop;
 191                         }
 192
 193                         if (count > 0) {
 194                                 int i;
 195                                 char *full_name_inc;
 196
 197                                 full_name_inc = xmalloc(strlen(full_name) + 4);
 198                                 for (i = start; i < count; i++) {
 199                                         sprintf(full_name_inc, "%s%d", full_name, i);
 200                                         rdev = (major << 8) + minor + (i * increment - start);
 201                                         if (mknod(full_name_inc, mode, rdev) == -1) {
 202                                                 bb_perror_msg("line %d: could not create node %s", linenum, full_name_inc);
 203                                                 ret = EXIT_FAILURE;
 204                                         }
 205                                         else if (chown(full_name_inc, uid, gid) == -1) {
 206                                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name_inc);
 207                                                 ret = EXIT_FAILURE;
 208                                         }
 209                                         if ((mode != -1) && (chmod(full_name_inc, mode) < 0)){
 210                                                 bb_perror_msg("line %d: chmod 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: could not 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                                 if ((mode != -1) && (chmod(full_name, mode) < 0)){
 226                                         bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
 227                                         ret = EXIT_FAILURE;
 228                                 }
 229                         }
 230                 }
 231 loop:
 232                 free(line);
 233                 free(full_name);
 234         }
 235         fclose(table);
 236
 237         return 0;
 238 }
 239
 240 #else
 241 # error makdedevs configuration error, either leaf or table must be selected
 242 #endif