coreutils/cp.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini cp implementation for busybox
   4  *
   5  * Copyright (C) 2000 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15  * General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  20  *
  21  */
  22
  23 /* BB_AUDIT SUSv3 defects - unsupported options -H, -L, and -P. */
  24 /* BB_AUDIT GNU defects - only extension options supported are -a and -d.  */
  25 /* http://www.opengroup.org/onlinepubs/007904975/utilities/cp.html */
  26
  27 /* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
  28  *
  29  * Size reduction.
  30  */
  31
  32 #include <sys/types.h>
  33 #include <sys/stat.h>
  34 #include <unistd.h>
  35 #include <fcntl.h>
  36 #include <utime.h>
  37 #include <errno.h>
  38 #include <dirent.h>
  39 #include <stdlib.h>
  40 #include <assert.h>
  41 #include "busybox.h"
  42 #include "libcoreutils/coreutils.h"
  43
  44 static const char cp_opts[] = "pdRfia"; /* WARNING!! ORDER IS IMPORTANT!! */
  45
  46 extern int cp_main(int argc, char **argv)
  47 {
  48         struct stat source_stat;
  49         struct stat dest_stat;
  50         const char *last;
  51         const char *dest;
  52         int s_flags;
  53         int d_flags;
  54         int flags;
  55         int status = 0;
  56
  57         /* Since these are enums, #if tests will not work.  So use assert()s. */
  58         assert(FILEUTILS_PRESERVE_STATUS == 1);
  59         assert(FILEUTILS_DEREFERENCE == 2);
  60         assert(FILEUTILS_RECUR == 4);
  61         assert(FILEUTILS_FORCE == 8);
  62         assert(FILEUTILS_INTERACTIVE == 16);
  63
  64         flags = bb_getopt_ulflags(argc, argv, cp_opts);
  65
  66         if (flags & 32) {
  67                 flags |= (FILEUTILS_PRESERVE_STATUS | FILEUTILS_RECUR | FILEUTILS_DEREFERENCE);
  68         }
  69
  70         flags ^= FILEUTILS_DEREFERENCE;         /* The sense of this flag was reversed. */
  71
  72         if (optind + 2 > argc) {
  73                 bb_show_usage();
  74         }
  75
  76         last = argv[argc - 1];
  77         argv += optind;
  78
  79         /* If there are only two arguments and...  */
  80         if (optind + 2 == argc) {
  81                 s_flags = cp_mv_stat2(*argv, &source_stat,
  82                                                                  (flags & FILEUTILS_DEREFERENCE) ? stat : lstat);
  83                 if ((s_flags < 0) || ((d_flags = cp_mv_stat(last, &dest_stat)) < 0)) {
  84                         exit(EXIT_FAILURE);
  85                 }
  86                 /* ...if neither is a directory or...  */
  87                 if ( !((s_flags | d_flags) & 2) ||
  88                         /* ...recursing, the 1st is a directory, and the 2nd doesn't exist... */
  89                         /* ((flags & FILEUTILS_RECUR) && (s_flags & 2) && !d_flags) */
  90                         /* Simplify the above since FILEUTILS_RECUR >> 1 == 2. */
  91                         ((((flags & FILEUTILS_RECUR) >> 1) & s_flags) && !d_flags)
  92                 ) {
  93                         /* ...do a simple copy.  */
  94                                 dest = last;
  95                                 goto DO_COPY; /* Note: optind+2==argc implies argv[1]==last below. */
  96                 }
  97         }
  98
  99         do {
 100                 dest = concat_path_file(last, bb_get_last_path_component(*argv));
 101         DO_COPY:
 102                 if (copy_file(*argv, dest, flags) < 0) {
 103                         status = 1;
 104                 }
 105                 if (*++argv == last) {
 106                         break;
 107                 }
 108                 free((void *) dest);
 109         } while (1);
 110
 111         exit(status);
 112 }