123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571 |
- /* vi: set sw=4 ts=4: */
- /*
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License version 2 as
- * published by the Free Software Foundation.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU Library General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
- *
- */
- #define _GNU_SOURCE
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <fcntl.h>
- #include <getopt.h>
- #include <time.h>
- #include <pwd.h>
- #include <grp.h>
- #include <unistd.h>
- #include <ctype.h>
- #include <errno.h>
- #include <libgen.h>
- #include <stdarg.h>
- #include <sys/stat.h>
- #include <sys/types.h>
- #if !defined(__APPLE__) && !defined(__FreeBSD__) && !defined(__OpenBSD__) && !defined(__NetBSD__)
- #include <sys/sysmacros.h> /* major() and minor() */
- #endif
- #include <ftw.h>
- const char *bb_applet_name;
- uid_t recursive_uid;
- gid_t recursive_gid;
- unsigned int recursive_mode;
- void bb_verror_msg(const char *s, va_list p)
- {
- fflush(stdout);
- fprintf(stderr, "%s: ", bb_applet_name);
- vfprintf(stderr, s, p);
- }
- void bb_error_msg(const char *s, ...)
- {
- va_list p;
- va_start(p, s);
- bb_verror_msg(s, p);
- va_end(p);
- putc('\n', stderr);
- }
- void bb_error_msg_and_die(const char *s, ...)
- {
- va_list p;
- va_start(p, s);
- bb_verror_msg(s, p);
- va_end(p);
- putc('\n', stderr);
- exit(1);
- }
- void bb_vperror_msg(const char *s, va_list p)
- {
- int err=errno;
- if(s == 0) s = "";
- bb_verror_msg(s, p);
- if (*s) s = ": ";
- fprintf(stderr, "%s%s\n", s, strerror(err));
- }
- void bb_perror_msg(const char *s, ...)
- {
- va_list p;
- va_start(p, s);
- bb_vperror_msg(s, p);
- va_end(p);
- }
- void bb_perror_msg_and_die(const char *s, ...)
- {
- va_list p;
- va_start(p, s);
- bb_vperror_msg(s, p);
- va_end(p);
- exit(1);
- }
- FILE *bb_xfopen(const char *path, const char *mode)
- {
- FILE *fp;
- if ((fp = fopen(path, mode)) == NULL)
- bb_perror_msg_and_die("%s", path);
- return fp;
- }
- enum {
- FILEUTILS_PRESERVE_STATUS = 1,
- FILEUTILS_DEREFERENCE = 2,
- FILEUTILS_RECUR = 4,
- FILEUTILS_FORCE = 8,
- FILEUTILS_INTERACTIVE = 16
- };
- int bb_make_directory (char *path, long mode, int flags)
- {
- mode_t mask;
- const char *fail_msg;
- char *s = path;
- char c;
- struct stat st;
- mask = umask(0);
- if (mode == -1) {
- umask(mask);
- mode = (S_IXUSR | S_IXGRP | S_IXOTH |
- S_IWUSR | S_IWGRP | S_IWOTH |
- S_IRUSR | S_IRGRP | S_IROTH) & ~mask;
- } else {
- umask(mask & ~0300);
- }
- do {
- c = 0;
- if (flags & FILEUTILS_RECUR) { /* Get the parent. */
- /* Bypass leading non-'/'s and then subsequent '/'s. */
- while (*s) {
- if (*s == '/') {
- do {
- ++s;
- } while (*s == '/');
- c = *s; /* Save the current char */
- *s = 0; /* and replace it with nul. */
- break;
- }
- ++s;
- }
- }
- if (mkdir(path, 0777) < 0) {
- /* If we failed for any other reason than the directory
- * already exists, output a diagnostic and return -1.*/
- if ((errno != EEXIST && errno != EISDIR)
- || !(flags & FILEUTILS_RECUR)
- || (stat(path, &st) < 0 || !S_ISDIR(st.st_mode))) {
- fail_msg = "create";
- umask(mask);
- break;
- }
- /* Since the directory exists, don't attempt to change
- * permissions if it was the full target. Note that
- * this is not an error conditon. */
- if (!c) {
- umask(mask);
- return 0;
- }
- }
- if (!c) {
- /* Done. If necessary, updated perms on the newly
- * created directory. Failure to update here _is_
- * an error.*/
- umask(mask);
- if ((mode != -1) && (chmod(path, mode) < 0)){
- fail_msg = "set permissions of";
- break;
- }
- return 0;
- }
- /* Remove any inserted nul from the path (recursive mode). */
- *s = c;
- } while (1);
- bb_perror_msg ("Cannot %s directory `%s'", fail_msg, path);
- return -1;
- }
- const char * const bb_msg_memory_exhausted = "memory exhausted";
- void *xmalloc(size_t size)
- {
- void *ptr = malloc(size);
- if (ptr == NULL && size != 0)
- bb_error_msg_and_die(bb_msg_memory_exhausted);
- return ptr;
- }
- void *xcalloc(size_t nmemb, size_t size)
- {
- void *ptr = calloc(nmemb, size);
- if (ptr == NULL && nmemb != 0 && size != 0)
- bb_error_msg_and_die(bb_msg_memory_exhausted);
- return ptr;
- }
- void *xrealloc(void *ptr, size_t size)
- {
- ptr = realloc(ptr, size);
- if (ptr == NULL && size != 0)
- bb_error_msg_and_die(bb_msg_memory_exhausted);
- return ptr;
- }
- char *private_get_line_from_file(FILE *file, int c)
- {
- #define GROWBY (80) /* how large we will grow strings by */
- int ch;
- int idx = 0;
- char *linebuf = NULL;
- int linebufsz = 0;
- while ((ch = getc(file)) != EOF) {
- /* grow the line buffer as necessary */
- if (idx > linebufsz - 2) {
- linebuf = xrealloc(linebuf, linebufsz += GROWBY);
- }
- linebuf[idx++] = (char)ch;
- if (!ch) return linebuf;
- if (c<2 && ch == '\n') {
- if (c) {
- --idx;
- }
- break;
- }
- }
- if (linebuf) {
- if (ferror(file)) {
- free(linebuf);
- return NULL;
- }
- linebuf[idx] = 0;
- }
- return linebuf;
- }
- char *bb_get_chomped_line_from_file(FILE *file)
- {
- return private_get_line_from_file(file, 1);
- }
- long my_getpwnam(const char *name)
- {
- struct passwd *myuser;
- myuser = getpwnam(name);
- if (myuser==NULL)
- bb_error_msg_and_die("unknown user name: %s", name);
- return myuser->pw_uid;
- }
- long my_getgrnam(const char *name)
- {
- struct group *mygroup;
- mygroup = getgrnam(name);
- if (mygroup==NULL)
- bb_error_msg_and_die("unknown group name: %s", name);
- return (mygroup->gr_gid);
- }
- unsigned long get_ug_id(const char *s, long (*my_getxxnam)(const char *))
- {
- unsigned long r;
- char *p;
- r = strtoul(s, &p, 10);
- if (*p || (s == p)) {
- r = my_getxxnam(s);
- }
- return r;
- }
- char * last_char_is(const char *s, int c)
- {
- char *sret = (char *)s;
- if (sret) {
- sret = strrchr(sret, c);
- if(sret != NULL && *(sret+1) != 0)
- sret = NULL;
- }
- return sret;
- }
- void bb_xasprintf(char **string_ptr, const char *format, ...)
- {
- va_list p;
- int r;
- va_start(p, format);
- r = vasprintf(string_ptr, format, p);
- va_end(p);
- if (r < 0) {
- bb_perror_msg_and_die("bb_xasprintf");
- }
- }
- char *concat_path_file(const char *path, const char *filename)
- {
- char *outbuf;
- char *lc;
- if (!path)
- path = "";
- lc = last_char_is(path, '/');
- while (*filename == '/')
- filename++;
- bb_xasprintf(&outbuf, "%s%s%s", path, (lc==NULL ? "/" : ""), filename);
- return outbuf;
- }
- void bb_show_usage(void)
- {
- fprintf(stderr, "%s: [-d device_table] rootdir\n\n", bb_applet_name);
- fprintf(stderr, "Creates a batch of special files as specified in a device table.\n");
- fprintf(stderr, "Device table entries take the form of:\n");
- fprintf(stderr, "name type mode user group major minor start increment count\n\n");
- fprintf(stderr, "Where name is the file name, type can be one of:\n");
- fprintf(stderr, " f A regular file\n");
- fprintf(stderr, " d Directory\n");
- fprintf(stderr, " r Directory recursively\n");
- fprintf(stderr, " c Character special device file\n");
- fprintf(stderr, " b Block special device file\n");
- fprintf(stderr, " p Fifo (named pipe)\n");
- fprintf(stderr, "uid is the user id for the target file, gid is the group id for the\n");
- fprintf(stderr, "target file. The rest of the entries (major, minor, etc) apply to\n");
- fprintf(stderr, "to device special files. A '-' may be used for blank entries.\n\n");
- fprintf(stderr, "For example:\n");
- fprintf(stderr, "<name> <type> <mode> <uid> <gid> <major> <minor> <start> <inc> <count>\n");
- fprintf(stderr, "/dev d 755 0 0 - - - - -\n");
- fprintf(stderr, "/dev/console c 666 0 0 5 1 - - -\n");
- fprintf(stderr, "/dev/null c 666 0 0 1 3 0 0 -\n");
- fprintf(stderr, "/dev/zero c 666 0 0 1 5 0 0 -\n");
- fprintf(stderr, "/dev/hda b 640 0 0 3 0 0 0 -\n");
- fprintf(stderr, "/dev/hda b 640 0 0 3 1 1 1 15\n");
- fprintf(stderr, "/dev/rtp b 640 0 0 250 0 0 1 5\n");
- fprintf(stderr, "/dev/gps b 640 0 0 251 0 1 1 5\n");
- fprintf(stderr, "/dev/uio b 640 0 0 252 0 1 2 5\n");
- fprintf(stderr, "/dev/uio b 640 0 0 252 1 6 2 5\n\n");
- fprintf(stderr, "Will Produce:\n");
- fprintf(stderr, "/dev\n");
- fprintf(stderr, "/dev/console\n");
- fprintf(stderr, "/dev/null\n");
- fprintf(stderr, "/dev/zero\n");
- fprintf(stderr, "/dev/hda\n");
- fprintf(stderr, "/dev/hda[1-15] with minor numbers [1-15]\n");
- fprintf(stderr, "/dev/rtp[0-4] with minor numbers [0-4]\n");
- fprintf(stderr, "/dev/gps[1-5] with minor numbers [0-4]\n");
- fprintf(stderr, "/dev/uio[1-5] with minor numbers 0,2,4,6,8\n");
- fprintf(stderr, "/dev/uio[6-10] with minor numbers 1,3,5,7,9\n");
- exit(1);
- }
- int bb_recursive(const char *fpath, const struct stat *sb,
- int tflag, struct FTW *ftwbuf){
- if (chown(fpath, recursive_uid, recursive_gid) == -1) {
- bb_perror_msg("chown failed for %s", fpath);
- return -1;
- }
- if (recursive_mode != -1) {
- if (chmod(fpath, recursive_mode) < 0) {
- bb_perror_msg("chmod failed for %s", fpath);
- return -1;
- }
- }
- return 0;
- }
- int main(int argc, char **argv)
- {
- int opt;
- FILE *table = stdin;
- char *rootdir = NULL;
- char *line = NULL;
- int linenum = 0;
- int ret = EXIT_SUCCESS;
- bb_applet_name = basename(argv[0]);
- while ((opt = getopt(argc, argv, "d:")) != -1) {
- switch(opt) {
- case 'd':
- table = bb_xfopen((line=optarg), "r");
- break;
- default:
- bb_show_usage();
- }
- }
- if (optind >= argc || (rootdir=argv[optind])==NULL) {
- bb_error_msg_and_die("root directory not speficied");
- }
- if (chdir(rootdir) != 0) {
- bb_perror_msg_and_die("Couldnt chdir to %s", rootdir);
- }
- umask(0);
- printf("rootdir=%s\n", rootdir);
- if (line) {
- printf("table='%s'\n", line);
- } else {
- printf("table=<stdin>\n");
- }
- while ((line = bb_get_chomped_line_from_file(table))) {
- char type;
- unsigned int mode = 0755;
- unsigned int major = 0;
- unsigned int minor = 0;
- unsigned int count = 0;
- unsigned int increment = 0;
- unsigned int start = 0;
- char name[4096];
- char user[41];
- char group[41];
- char *full_name;
- uid_t uid;
- gid_t gid;
- linenum++;
- if ((2 > sscanf(line, "%4095s %c %o %40s %40s %u %u %u %u %u", name,
- &type, &mode, user, group, &major,
- &minor, &start, &increment, &count)) ||
- ((major | minor | start | count | increment) > 0xfffff))
- {
- if (*line=='\0' || *line=='#' || isspace(*line))
- continue;
- bb_error_msg("line %d invalid: '%s'\n", linenum, line);
- ret = EXIT_FAILURE;
- continue;
- }
- if (name[0] == '#') {
- continue;
- }
- if (*group) {
- gid = get_ug_id(group, my_getgrnam);
- } else {
- gid = getgid();
- }
- if (*user) {
- uid = get_ug_id(user, my_getpwnam);
- } else {
- uid = getuid();
- }
- full_name = concat_path_file(rootdir, name);
- if (type == 'd') {
- bb_make_directory(full_name, mode | S_IFDIR, FILEUTILS_RECUR);
- if (chown(full_name, uid, gid) == -1) {
- bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
- ret = EXIT_FAILURE;
- goto loop;
- }
- if ((mode != -1) && (chmod(full_name, mode) < 0)){
- bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
- ret = EXIT_FAILURE;
- goto loop;
- }
- } else if (type == 'f') {
- struct stat st;
- if ((stat(full_name, &st) < 0 || !S_ISREG(st.st_mode))) {
- bb_perror_msg("line %d: regular file '%s' does not exist", linenum, full_name);
- ret = EXIT_FAILURE;
- goto loop;
- }
- if (chown(full_name, uid, gid) == -1) {
- bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
- ret = EXIT_FAILURE;
- goto loop;
- }
- if ((mode != -1) && (chmod(full_name, mode) < 0)){
- bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
- ret = EXIT_FAILURE;
- goto loop;
- }
- } else if (type == 'r') {
- recursive_uid = uid;
- recursive_gid = gid;
- recursive_mode = mode;
- if (nftw(full_name, bb_recursive, 20, FTW_MOUNT | FTW_PHYS) < 0) {
- bb_perror_msg("line %d: recursive failed for %s", linenum, full_name);
- ret = EXIT_FAILURE;
- goto loop;
- }
- } else
- {
- dev_t rdev;
- if (type == 'p') {
- mode |= S_IFIFO;
- }
- else if (type == 'c') {
- mode |= S_IFCHR;
- }
- else if (type == 'b') {
- mode |= S_IFBLK;
- } else {
- bb_error_msg("line %d: Unsupported file type %c", linenum, type);
- ret = EXIT_FAILURE;
- goto loop;
- }
- if (count > 0) {
- int i;
- char *full_name_inc;
- full_name_inc = xmalloc(strlen(full_name) + 8);
- for (i = 0; i < count; i++) {
- sprintf(full_name_inc, "%s%d", full_name, start + i);
- rdev = makedev(major, minor + i * increment);
- if (mknod(full_name_inc, mode, rdev) == -1) {
- bb_perror_msg("line %d: Couldnt create node %s", linenum, full_name_inc);
- ret = EXIT_FAILURE;
- }
- else if (chown(full_name_inc, uid, gid) == -1) {
- bb_perror_msg("line %d: chown failed for %s", linenum, full_name_inc);
- ret = EXIT_FAILURE;
- }
- if ((mode != -1) && (chmod(full_name_inc, mode) < 0)){
- bb_perror_msg("line %d: chmod failed for %s", linenum, full_name_inc);
- ret = EXIT_FAILURE;
- }
- }
- free(full_name_inc);
- } else {
- rdev = makedev(major, minor);
- if (mknod(full_name, mode, rdev) == -1) {
- bb_perror_msg("line %d: Couldnt create node %s", linenum, full_name);
- ret = EXIT_FAILURE;
- }
- else if (chown(full_name, uid, gid) == -1) {
- bb_perror_msg("line %d: chown failed for %s", linenum, full_name);
- ret = EXIT_FAILURE;
- }
- if ((mode != -1) && (chmod(full_name, mode) < 0)){
- bb_perror_msg("line %d: chmod failed for %s", linenum, full_name);
- ret = EXIT_FAILURE;
- }
- }
- }
- loop:
- free(line);
- free(full_name);
- }
- fclose(table);
- return ret;
- }
|