Fixup device table based makedevs so it actually works
[people/mcb30/busybox.git] / 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