*: add most of the required setup_common_bufsiz() calls
[oweals/busybox.git] / archival / cpio.c
index c2a5b8ab9edd17b5aad0cd42b51000cff691a820..3b15507200e770176bf2212951ee092189ebb528 100644 (file)
@@ -9,13 +9,45 @@
  * Limitations:
  * Doesn't check CRC's
  * Only supports new ASCII and CRC formats
- *
  */
 #include "libbb.h"
+#include "common_bufsiz.h"
 #include "bb_archive.h"
 
+//config:config CPIO
+//config:      bool "cpio"
+//config:      default y
+//config:      help
+//config:        cpio is an archival utility program used to create, modify, and
+//config:        extract contents from archives.
+//config:        cpio has 110 bytes of overheads for every stored file.
+//config:
+//config:        This implementation of cpio can extract cpio archives created in the
+//config:        "newc" or "crc" format, it cannot create or modify them.
+//config:
+//config:        Unless you have a specific application which requires cpio, you
+//config:        should probably say N here.
+//config:
+//config:config FEATURE_CPIO_O
+//config:      bool "Support for archive creation"
+//config:      default y
+//config:      depends on CPIO
+//config:      help
+//config:        This implementation of cpio can create cpio archives in the "newc"
+//config:        format only.
+//config:
+//config:config FEATURE_CPIO_P
+//config:      bool "Support for passthrough mode"
+//config:      default y
+//config:      depends on FEATURE_CPIO_O
+//config:      help
+//config:        Passthrough mode. Rarely used.
+
+//applet:IF_CPIO(APPLET(cpio, BB_DIR_BIN, BB_SUID_DROP))
+//kbuild:lib-$(CONFIG_CPIO) += cpio.o
+
 //usage:#define cpio_trivial_usage
-//usage:       "[-dmvu] [-F FILE]" IF_FEATURE_CPIO_O(" [-H newc]")
+//usage:       "[-dmvu] [-F FILE] [-R USER[:GRP]]" IF_FEATURE_CPIO_O(" [-H newc]")
 //usage:       " [-ti"IF_FEATURE_CPIO_O("o")"]" IF_FEATURE_CPIO_P(" [-p DIR]")
 //usage:       " [EXTR_FILE]..."
 //usage:#define cpio_full_usage "\n\n"
 //usage:       IF_FEATURE_CPIO_P(
 //usage:     "\n       -p DIR  Copy files to DIR"
 //usage:       )
+//usage:     "\nOptions:"
 //usage:     "\n       -d      Make leading directories"
 //usage:     "\n       -m      Preserve mtime"
 //usage:     "\n       -v      Verbose"
 //usage:     "\n       -u      Overwrite"
 //usage:     "\n       -F FILE Input (-t,-i,-p) or output (-o) file"
+//usage:     "\n       -R USER[:GRP]   Set owner of created files"
 //usage:       IF_FEATURE_CPIO_O(
 //usage:     "\n       -H newc Archive format"
 //usage:       )
   -I FILE                    File to use instead of standard input
   -L, --dereference          Dereference symbolic links (copy the files
                              that they point to instead of copying the links)
-  -R, --owner=[USER][:.][GROUP] Set owner of created files
+  -R, --owner=[USER][:.][GRP] Set owner of created files
 
  Options valid in --extract and --pass-through modes:
   -d, --make-directories     Create leading directories where needed
@@ -118,7 +152,8 @@ enum {
        OPT_PRESERVE_MTIME     = (1 << 6),
        OPT_DEREF              = (1 << 7),
        OPT_FILE               = (1 << 8),
-       OPTBIT_FILE = 8,
+       OPT_OWNER              = (1 << 9),
+       OPTBIT_OWNER = 9,
        IF_FEATURE_CPIO_O(OPTBIT_CREATE     ,)
        IF_FEATURE_CPIO_O(OPTBIT_FORMAT     ,)
        IF_FEATURE_CPIO_P(OPTBIT_PASSTHROUGH,)
@@ -131,7 +166,18 @@ enum {
        OPT_2STDOUT            = IF_LONG_OPTS(     (1 << OPTBIT_2STDOUT    )) + 0,
 };
 
-#define OPTION_STR "it0uvdmLF:"
+#define OPTION_STR "it0uvdmLF:R:"
+
+struct globals {
+       struct bb_uidgid_t owner_ugid;
+} FIX_ALIASING;
+#define G (*(struct globals*)bb_common_bufsiz1)
+void BUG_cpio_globals_too_big(void);
+#define INIT_G() do { \
+       setup_common_bufsiz(); \
+       G.owner_ugid.uid = -1L; \
+       G.owner_ugid.gid = -1L; \
+} while (0)
 
 #if ENABLE_FEATURE_CPIO_O
 static off_t cpio_pad4(off_t size)
@@ -149,7 +195,6 @@ static off_t cpio_pad4(off_t size)
  * It's ok to exit instead of return. */
 static NOINLINE int cpio_o(void)
 {
-       static const char trailer[] ALIGN1 = "TRAILER!!!";
        struct name_s {
                struct name_s *next;
                char name[1];
@@ -191,6 +236,11 @@ static NOINLINE int cpio_o(void)
                                bb_simple_perror_msg_and_die(name);
                        }
 
+                       if (G.owner_ugid.uid != (uid_t)-1L)
+                               st.st_uid = G.owner_ugid.uid;
+                       if (G.owner_ugid.gid != (gid_t)-1L)
+                               st.st_gid = G.owner_ugid.gid;
+
                        if (!(S_ISLNK(st.st_mode) || S_ISREG(st.st_mode)))
                                st.st_size = 0; /* paranoia */
 
@@ -225,7 +275,6 @@ static NOINLINE int cpio_o(void)
                                free(line);
                                continue;
                        }
-
                } else { /* line == NULL: EOF */
  next_link:
                        if (links) {
@@ -244,7 +293,7 @@ static NOINLINE int cpio_o(void)
                        } else {
                                /* If no (more) hardlinks to output,
                                 * output "trailer" entry */
-                               name = trailer;
+                               name = cpio_TRAILER;
                                /* st.st_size == 0 is a must, but for uniformity
                                 * in the output, we zero out everything */
                                memset(&st, 0, sizeof(st));
@@ -253,24 +302,24 @@ static NOINLINE int cpio_o(void)
                }
 
                bytes += printf("070701"
-                               "%08X%08X%08X%08X%08X%08X%08X"
-                               "%08X%08X%08X%08X" /* GNU cpio uses uppercase hex */
+                               "%08X%08X%08X%08X%08X%08X%08X"
+                               "%08X%08X%08X%08X" /* GNU cpio uses uppercase hex */
                                /* strlen+1: */ "%08X"
                                /* chksum: */   "00000000" /* (only for "070702" files) */
                                /* name,NUL: */ "%s%c",
-                               (unsigned)(uint32_t) st.st_ino,
-                               (unsigned)(uint32_t) st.st_mode,
-                               (unsigned)(uint32_t) st.st_uid,
-                               (unsigned)(uint32_t) st.st_gid,
-                               (unsigned)(uint32_t) st.st_nlink,
-                               (unsigned)(uint32_t) st.st_mtime,
-                               (unsigned)(uint32_t) st.st_size,
-                               (unsigned)(uint32_t) major(st.st_dev),
-                               (unsigned)(uint32_t) minor(st.st_dev),
-                               (unsigned)(uint32_t) major(st.st_rdev),
-                               (unsigned)(uint32_t) minor(st.st_rdev),
-                               (unsigned)(strlen(name) + 1),
-                               name, '\0');
+                               (unsigned)(uint32_t) st.st_ino,
+                               (unsigned)(uint32_t) st.st_mode,
+                               (unsigned)(uint32_t) st.st_uid,
+                               (unsigned)(uint32_t) st.st_gid,
+                               (unsigned)(uint32_t) st.st_nlink,
+                               (unsigned)(uint32_t) st.st_mtime,
+                               (unsigned)(uint32_t) st.st_size,
+                               (unsigned)(uint32_t) major(st.st_dev),
+                               (unsigned)(uint32_t) minor(st.st_dev),
+                               (unsigned)(uint32_t) major(st.st_rdev),
+                               (unsigned)(uint32_t) minor(st.st_rdev),
+                               (unsigned)(strlen(name) + 1),
+                               name, '\0');
                bytes = cpio_pad4(bytes);
 
                if (st.st_size) {
@@ -292,7 +341,7 @@ static NOINLINE int cpio_o(void)
                }
 
                if (!line) {
-                       if (name != trailer)
+                       if (name != cpio_TRAILER)
                                goto next_link;
                        /* TODO: GNU cpio pads trailer to 512 bytes, do we want that? */
                        return EXIT_SUCCESS;
@@ -308,6 +357,7 @@ int cpio_main(int argc UNUSED_PARAM, char **argv)
 {
        archive_handle_t *archive_handle;
        char *cpio_filename;
+       char *cpio_owner;
        IF_FEATURE_CPIO_O(const char *cpio_fmt = "";)
        unsigned opt;
 
@@ -322,12 +372,14 @@ int cpio_main(int argc UNUSED_PARAM, char **argv)
                "pass-through\0" No_argument       "p"
 #endif
 #endif
+               "owner\0"        Required_argument "R"
                "verbose\0"      No_argument       "v"
                "quiet\0"        No_argument       "\xff"
                "to-stdout\0"    No_argument       "\xfe"
                ;
 #endif
 
+       INIT_G();
        archive_handle = init_handle();
        /* archive_handle->src_fd = STDIN_FILENO; - done by init_handle */
        archive_handle->ah_flags = ARCHIVE_EXTRACT_NEWER;
@@ -338,14 +390,21 @@ int cpio_main(int argc UNUSED_PARAM, char **argv)
        /* -L makes sense only with -o or -p */
 
 #if !ENABLE_FEATURE_CPIO_O
-       opt = getopt32(argv, OPTION_STR, &cpio_filename);
+       opt = getopt32(argv, OPTION_STR, &cpio_filename, &cpio_owner);
+#else
+       opt = getopt32(argv, OPTION_STR "oH:" IF_FEATURE_CPIO_P("p"),
+                      &cpio_filename, &cpio_owner, &cpio_fmt);
+#endif
        argv += optind;
+       if (opt & OPT_OWNER) { /* -R */
+               parse_chown_usergroup_or_die(&G.owner_ugid, cpio_owner);
+               archive_handle->cpio__owner = G.owner_ugid;
+       }
+#if !ENABLE_FEATURE_CPIO_O
        if (opt & OPT_FILE) { /* -F */
                xmove_fd(xopen(cpio_filename, O_RDONLY), STDIN_FILENO);
        }
 #else
-       opt = getopt32(argv, OPTION_STR "oH:" IF_FEATURE_CPIO_P("p"), &cpio_filename, &cpio_fmt);
-       argv += optind;
        if ((opt & (OPT_FILE|OPT_CREATE)) == OPT_FILE) { /* -F without -o */
                xmove_fd(xopen(cpio_filename, O_RDONLY), STDIN_FILENO);
        }
@@ -384,6 +443,7 @@ int cpio_main(int argc UNUSED_PARAM, char **argv)
                        goto dump;
                }
                /* parent */
+               USE_FOR_NOMMU(argv[-optind][0] &= 0x7f); /* undo fork_or_rexec() damage */
                xchdir(*argv++);
                close(pp.wr);
                xmove_fd(pp.rd, STDIN_FILENO);