miscutils/makedevs.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * public domain -- Dave 'Kill a Cop' Cinege <dcinege@psychosis.com>
   4  *
   5  * makedevs
   6  * Make ranges of device files quickly.
   7  * known bugs: can't deal with alpha ranges
   8  */
   9
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13 #include <fcntl.h>
  14 #include <time.h>
  15 #include <ctype.h>
  16 #include <unistd.h>
  17 #include <sys/types.h>
  18 #include <sys/sysmacros.h>     /* major() and minor() */
  19 #include "busybox.h"
  20
  21 #ifdef CONFIG_FEATURE_MAKEDEVS_LEAF
  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 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         bb_xchdir(rootdir);
  98
  99         umask(0);
 100
 101         printf("rootdir=%s\n", rootdir);
 102         if (line) {
 103                 printf("table='%s'\n", line);
 104         } else {
 105                 printf("table=<stdin>\n");
 106         }
 107
 108         while ((line = bb_get_chomped_line_from_file(table))) {
 109                 char type;
 110                 unsigned int mode = 0755;
 111                 unsigned int major = 0;
 112                 unsigned int minor = 0;
 113                 unsigned int count = 0;
 114                 unsigned int increment = 0;
 115                 unsigned int start = 0;
 116                 char name[41];
 117                 char user[41];
 118                 char group[41];
 119                 char *full_name;
 120                 uid_t uid;
 121                 gid_t gid;
 122
 123                 linenum++;
 124
 125                 if ((2 > sscanf(line, "%40s %c %o %40s %40s %u %u %u %u %u", name,
 126                                                 &type, &mode, user, group, &major,
 127                                                 &minor, &start, &increment, &count)) ||
 128                                 ((major | minor | start | count | increment) > 255))
 129                 {
 130                         if (*line=='\0' || *line=='#' || isspace(*line))
 131                                 continue;
 132                         bb_error_msg("line %d invalid: '%s'\n", linenum, line);
 133                         ret = EXIT_FAILURE;
 134                         continue;
 135                 }
 136                 if (name[0] == '#') {
 137                         continue;
 138                 }
 139
 140                 gid = (*group) ? get_ug_id(group, bb_xgetgrnam) : getgid();
 141                 uid = (*user) ? get_ug_id(user, bb_xgetpwnam) : getuid();
 142                 full_name = concat_path_file(rootdir, name);
 143
 144                 if (type == 'd') {
 145                         bb_make_directory(full_name, mode | S_IFDIR, FILEUTILS_RECUR);
 146                         if (chown(full_name, uid, gid) == -1) {
 147                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
 148                                 ret = EXIT_FAILURE;
 149                                 goto loop;
 150                         }
 151                         if ((mode != -1) && (chmod(full_name, mode) < 0)){
 152                                 bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
 153                                 ret = EXIT_FAILURE;
 154                                 goto loop;
 155                         }
 156                 } else if (type == 'f') {
 157                         struct stat st;
 158                         if ((stat(full_name, &st) < 0 || !S_ISREG(st.st_mode))) {
 159                                 bb_perror_msg("line %d: regular file '%s' does not exist", linenum, full_name);
 160                                 ret = EXIT_FAILURE;
 161                                 goto loop;
 162                         }
 163                         if (chown(full_name, uid, gid) == -1) {
 164                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
 165                                 ret = EXIT_FAILURE;
 166                                 goto loop;
 167                         }
 168                         if ((mode != -1) && (chmod(full_name, mode) < 0)){
 169                                 bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
 170                                 ret = EXIT_FAILURE;
 171                                 goto loop;
 172                         }
 173                 } else
 174                 {
 175                         dev_t rdev;
 176
 177                         if (type == 'p') {
 178                                 mode |= S_IFIFO;
 179                         }
 180                         else if (type == 'c') {
 181                                 mode |= S_IFCHR;
 182                         }
 183                         else if (type == 'b') {
 184                                 mode |= S_IFBLK;
 185                         } else {
 186                                 bb_error_msg("line %d: unsupported file type %c", linenum, type);
 187                                 ret = EXIT_FAILURE;
 188                                 goto loop;
 189                         }
 190
 191                         if (count > 0) {
 192                                 int i;
 193                                 char *full_name_inc;
 194
 195                                 full_name_inc = xmalloc(strlen(full_name) + 4);
 196                                 for (i = start; i < count; i++) {
 197                                         sprintf(full_name_inc, "%s%d", full_name, i);
 198                                         rdev = (major << 8) + minor + (i * increment - start);
 199                                         if (mknod(full_name_inc, mode, rdev) == -1) {
 200                                                 bb_perror_msg("line %d: could not create node %s", linenum, full_name_inc);
 201                                                 ret = EXIT_FAILURE;
 202                                         }
 203                                         else if (chown(full_name_inc, uid, gid) == -1) {
 204                                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name_inc);
 205                                                 ret = EXIT_FAILURE;
 206                                         }
 207                                         if ((mode != -1) && (chmod(full_name_inc, mode) < 0)){
 208                                                 bb_perror_msg("line %d: chmod failed for %s", linenum, full_name_inc);
 209                                                 ret = EXIT_FAILURE;
 210                                         }
 211                                 }
 212                                 free(full_name_inc);
 213                         } else {
 214                                 rdev = (major << 8) + minor;
 215                                 if (mknod(full_name, mode, rdev) == -1) {
 216                                         bb_perror_msg("line %d: could not create node %s", linenum, full_name);
 217                                         ret = EXIT_FAILURE;
 218                                 }
 219                                 else if (chown(full_name, uid, gid) == -1) {
 220                                         bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
 221                                         ret = EXIT_FAILURE;
 222                                 }
 223                                 if ((mode != -1) && (chmod(full_name, mode) < 0)){
 224                                         bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
 225                                         ret = EXIT_FAILURE;
 226                                 }
 227                         }
 228                 }
 229 loop:
 230                 free(line);
 231                 free(full_name);
 232         }
 233         fclose(table);
 234
 235         return ret;
 236 }
 237
 238 #else
 239 # error makedevs configuration error, either leaf or table must be selected
 240 #endif