coreutils/install.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  *  Copyright (C) 2003 by Glenn McGrath <bug1@iinet.net.au>
   4  *
   5  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
   6  *
   7  * TODO: -d option, need a way of recursively making directories and changing
   8  *           owner/group, will probably modify bb_make_directory(...)
   9  */
  10
  11 #include <sys/stat.h>
  12 #include <sys/types.h>
  13 #include <errno.h>
  14 #include <stdlib.h>
  15 #include <string.h>
  16 #include <unistd.h>
  17 #include <getopt.h> /* struct option */
  18
  19 #include "busybox.h"
  20 #include "libcoreutils/coreutils.h"
  21
  22 #define INSTALL_OPT_CMD 1
  23 #define INSTALL_OPT_DIRECTORY   2
  24 #define INSTALL_OPT_PRESERVE_TIME       4
  25 #define INSTALL_OPT_STRIP       8
  26 #define INSTALL_OPT_GROUP  16
  27 #define INSTALL_OPT_MODE  32
  28 #define INSTALL_OPT_OWNER  64
  29
  30 #if ENABLE_FEATURE_INSTALL_LONG_OPTIONS
  31 static const struct option install_long_options[] = {
  32         { "directory",  0,      NULL,   'd' },
  33         { "preserve-timestamps",        0,      NULL,   'p' },
  34         { "strip",      0,      NULL,   's' },
  35         { "group",      0,      NULL,   'g' },
  36         { "mode",       0,      NULL,   'm' },
  37         { "owner",      0,      NULL,   'o' },
  38         { 0,    0,      0,      0 }
  39 };
  40 #endif
  41
  42 int install_main(int argc, char **argv)
  43 {
  44         mode_t mode;
  45         uid_t uid;
  46         gid_t gid;
  47         char *gid_str = "-1";
  48         char *uid_str = "-1";
  49         char *mode_str = "0755";
  50         int copy_flags = FILEUTILS_DEREFERENCE | FILEUTILS_FORCE;
  51         int ret = EXIT_SUCCESS, flags, i, isdir;
  52
  53 #if ENABLE_FEATURE_INSTALL_LONG_OPTIONS
  54         bb_applet_long_options = install_long_options;
  55 #endif
  56         bb_opt_complementally = "?:s--d:d--s";
  57         /* -c exists for backwards compatibility, its needed */
  58         flags = bb_getopt_ulflags(argc, argv, "cdpsg:m:o:", &gid_str, &mode_str, &uid_str);     /* 'a' must be 2nd */
  59
  60         /* preserve access and modification time, this is GNU behaviour, BSD only preserves modification time */
  61         if (flags & INSTALL_OPT_PRESERVE_TIME) {
  62                 copy_flags |= FILEUTILS_PRESERVE_STATUS;
  63         }
  64         bb_parse_mode(mode_str, &mode);
  65         gid = get_ug_id(gid_str, bb_xgetgrnam);
  66         uid = get_ug_id(uid_str, bb_xgetpwnam);
  67         umask(0);
  68
  69         /* Create directories
  70          * dont use bb_make_directory() as it cant change uid or gid
  71          * perhaps bb_make_directory() should be improved.
  72          */
  73         if (flags & INSTALL_OPT_DIRECTORY) {
  74                 for (argv += optind; *argv; argv++) {
  75                         char *old_argv_ptr = *argv + 1;
  76                         char *argv_ptr;
  77                         do {
  78                                 argv_ptr = strchr(old_argv_ptr, '/');
  79                                 old_argv_ptr = argv_ptr;
  80                                 if (argv_ptr) {
  81                                         *argv_ptr = '\0';
  82                                         old_argv_ptr++;
  83                                 }
  84                                 if (mkdir(*argv, mode) == -1) {
  85                                         if (errno != EEXIST) {
  86                                                 bb_perror_msg("coulnt create %s", *argv);
  87                                                 ret = EXIT_FAILURE;
  88                                                 break;
  89                                         }
  90                                 }
  91                                 else if (lchown(*argv, uid, gid) == -1) {
  92                                         bb_perror_msg("cannot change ownership of %s", *argv);
  93                                         ret = EXIT_FAILURE;
  94                                         break;
  95                                 }
  96                                 if (argv_ptr) {
  97                                         *argv_ptr = '/';
  98                                 }
  99                         } while (old_argv_ptr);
 100                 }
 101                 return(ret);
 102         }
 103
 104         {
 105                 struct stat statbuf;
 106                 isdir = lstat(argv[argc - 1], &statbuf)<0
 107                                         ? 0 : S_ISDIR(statbuf.st_mode);
 108         }
 109         for (i = optind; i < argc - 1; i++) {
 110                 char *dest;
 111
 112                 dest = argv[argc - 1];
 113                 if (isdir) dest = concat_path_file(argv[argc - 1], basename(argv[i]));
 114                 ret |= copy_file(argv[i], dest, copy_flags);
 115
 116                 /* Set the file mode */
 117                 if (chmod(dest, mode) == -1) {
 118                         bb_perror_msg("cannot change permissions of %s", dest);
 119                         ret = EXIT_FAILURE;
 120                 }
 121
 122                 /* Set the user and group id */
 123                 if (lchown(dest, uid, gid) == -1) {
 124                         bb_perror_msg("cannot change ownership of %s", dest);
 125                         ret = EXIT_FAILURE;
 126                 }
 127                 if (flags & INSTALL_OPT_STRIP) {
 128                         if (execlp("strip", "strip", dest, NULL) == -1) {
 129                                 bb_error_msg("strip failed");
 130                                 ret = EXIT_FAILURE;
 131                         }
 132                 }
 133                 if(ENABLE_FEATURE_CLEAN_UP && isdir) free(dest);
 134         }
 135
 136         return(ret);
 137 }