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