git.maemo.org Git - busybox4maemo/blob - 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 "libbb.h"
  11
  12 #if ENABLE_FEATURE_MAKEDEVS_LEAF
  13 int makedevs_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  14 int makedevs_main(int argc, char **argv)
  15 {
  16         mode_t mode;
  17         char *basedev, *type, *nodname, buf[255];
  18         int Smajor, Sminor, S, E;
  19
  20         if (argc < 7 || *argv[1]=='-')
  21                 bb_show_usage();
  22
  23         basedev = argv[1];
  24         type = argv[2];
  25         Smajor = xatoi_u(argv[3]);
  26         Sminor = xatoi_u(argv[4]);
  27         S = xatoi_u(argv[5]);
  28         E = xatoi_u(argv[6]);
  29         nodname = argc == 8 ? basedev : buf;
  30
  31         mode = 0660;
  32
  33         switch (type[0]) {
  34         case 'c':
  35                 mode |= S_IFCHR;
  36                 break;
  37         case 'b':
  38                 mode |= S_IFBLK;
  39                 break;
  40         case 'f':
  41                 mode |= S_IFIFO;
  42                 break;
  43         default:
  44                 bb_show_usage();
  45         }
  46
  47         while (S <= E) {
  48                 int sz;
  49
  50                 sz = snprintf(buf, sizeof(buf), "%s%d", basedev, S);
  51                 if (sz < 0 || sz >= sizeof(buf))  /* libc different */
  52                         bb_error_msg_and_die("%s too large", basedev);
  53
  54         /* if mode != S_IFCHR and != S_IFBLK third param in mknod() ignored */
  55
  56                 if (mknod(nodname, mode, makedev(Smajor, Sminor)))
  57                         bb_error_msg("failed to create: %s", nodname);
  58
  59                 if (nodname == basedev) /* ex. /dev/hda - to /dev/hda1 ... */
  60                         nodname = buf;
  61                 S++;
  62                 Sminor++;
  63         }
  64
  65         return 0;
  66 }
  67
  68 #elif ENABLE_FEATURE_MAKEDEVS_TABLE
  69
  70 /* Licensed under the GPL v2 or later, see the file LICENSE in this tarball. */
  71
  72 int makedevs_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  73 int makedevs_main(int argc, char **argv)
  74 {
  75         FILE *table = stdin;
  76         char *rootdir = NULL;
  77         char *line = NULL;
  78         int linenum = 0;
  79         int ret = EXIT_SUCCESS;
  80
  81         getopt32(argv, "d:", &line);
  82         if (line)
  83                 table = xfopen(line, "r");
  84
  85         if (optind >= argc || (rootdir=argv[optind])==NULL) {
  86                 bb_error_msg_and_die("root directory not specified");
  87         }
  88
  89         xchdir(rootdir);
  90
  91         umask(0);
  92
  93         printf("rootdir=%s\n", rootdir);
  94         if (line) {
  95                 printf("table='%s'\n", line);
  96         } else {
  97                 printf("table=<stdin>\n");
  98         }
  99
 100         while ((line = xmalloc_getline(table))) {
 101                 char type;
 102                 unsigned int mode = 0755;
 103                 unsigned int major = 0;
 104                 unsigned int minor = 0;
 105                 unsigned int count = 0;
 106                 unsigned int increment = 0;
 107                 unsigned int start = 0;
 108                 char name[41];
 109                 char user[41];
 110                 char group[41];
 111                 char *full_name;
 112                 uid_t uid;
 113                 gid_t gid;
 114
 115                 linenum++;
 116
 117                 if ((2 > sscanf(line, "%40s %c %o %40s %40s %u %u %u %u %u", name,
 118                                                 &type, &mode, user, group, &major,
 119                                                 &minor, &start, &increment, &count)) ||
 120                                 ((major | minor | start | count | increment) > 255))
 121                 {
 122                         if (*line=='\0' || *line=='#' || isspace(*line))
 123                                 continue;
 124                         bb_error_msg("line %d invalid: '%s'", linenum, line);
 125                         ret = EXIT_FAILURE;
 126                         continue;
 127                 }
 128                 if (name[0] == '#') {
 129                         continue;
 130                 }
 131
 132                 gid = (*group) ? get_ug_id(group, xgroup2gid) : getgid();
 133                 uid = (*user) ? get_ug_id(user, xuname2uid) : getuid();
 134                 full_name = concat_path_file(rootdir, name);
 135
 136                 if (type == 'd') {
 137                         bb_make_directory(full_name, mode | S_IFDIR, FILEUTILS_RECUR);
 138                         if (chown(full_name, uid, gid) == -1) {
 139                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
 140                                 ret = EXIT_FAILURE;
 141                                 goto loop;
 142                         }
 143                         if ((mode != -1) && (chmod(full_name, mode) < 0)){
 144                                 bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
 145                                 ret = EXIT_FAILURE;
 146                                 goto loop;
 147                         }
 148                 } else if (type == 'f') {
 149                         struct stat st;
 150                         if ((stat(full_name, &st) < 0 || !S_ISREG(st.st_mode))) {
 151                                 bb_perror_msg("line %d: regular file '%s' does not exist", linenum, full_name);
 152                                 ret = EXIT_FAILURE;
 153                                 goto loop;
 154                         }
 155                         if (chown(full_name, uid, gid) == -1) {
 156                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
 157                                 ret = EXIT_FAILURE;
 158                                 goto loop;
 159                         }
 160                         if ((mode != -1) && (chmod(full_name, mode) < 0)){
 161                                 bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
 162                                 ret = EXIT_FAILURE;
 163                                 goto loop;
 164                         }
 165                 } else {
 166                         dev_t rdev;
 167
 168                         if (type == 'p') {
 169                                 mode |= S_IFIFO;
 170                         }
 171                         else if (type == 'c') {
 172                                 mode |= S_IFCHR;
 173                         }
 174                         else if (type == 'b') {
 175                                 mode |= S_IFBLK;
 176                         } else {
 177                                 bb_error_msg("line %d: unsupported file type %c", linenum, type);
 178                                 ret = EXIT_FAILURE;
 179                                 goto loop;
 180                         }
 181
 182                         if (count > 0) {
 183                                 int i;
 184                                 char *full_name_inc;
 185
 186                                 full_name_inc = xmalloc(strlen(full_name) + 4);
 187                                 for (i = start; i < count; i++) {
 188                                         sprintf(full_name_inc, "%s%d", full_name, i);
 189                                         rdev = makedev(major, minor + (i * increment - start));
 190                                         if (mknod(full_name_inc, mode, rdev) == -1) {
 191                                                 bb_perror_msg("line %d: cannot create node %s", linenum, full_name_inc);
 192                                                 ret = EXIT_FAILURE;
 193                                         }
 194                                         else if (chown(full_name_inc, uid, gid) == -1) {
 195                                                 bb_perror_msg("line %d: chown failed for %s", linenum, full_name_inc);
 196                                                 ret = EXIT_FAILURE;
 197                                         }
 198                                         if ((mode != -1) && (chmod(full_name_inc, mode) < 0)){
 199                                                 bb_perror_msg("line %d: chmod failed for %s", linenum, full_name_inc);
 200                                                 ret = EXIT_FAILURE;
 201                                         }
 202                                 }
 203                                 free(full_name_inc);
 204                         } else {
 205                                 rdev = makedev(major, minor);
 206                                 if (mknod(full_name, mode, rdev) == -1) {
 207                                         bb_perror_msg("line %d: cannot create node %s", linenum, full_name);
 208                                         ret = EXIT_FAILURE;
 209                                 }
 210                                 else if (chown(full_name, uid, gid) == -1) {
 211                                         bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
 212                                         ret = EXIT_FAILURE;
 213                                 }
 214                                 if ((mode != -1) && (chmod(full_name, mode) < 0)){
 215                                         bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
 216                                         ret = EXIT_FAILURE;
 217                                 }
 218                         }
 219                 }
 220 loop:
 221                 free(line);
 222                 free(full_name);
 223         }
 224         fclose(table);
 225
 226         return ret;
 227 }
 228
 229 #else
 230 # error makedevs configuration error, either leaf or table must be selected
 231 #endif