BSD 4_3_Net_2 release
[unix-history] / usr / src / bin / cp / cp.c
index 2b8ac42..8124e8d 100644 (file)
@@ -5,17 +5,33 @@
  * This code is derived from software contributed to Berkeley by
  * David Hitz of Auspex Systems Inc.
  *
  * This code is derived from software contributed to Berkeley by
  * David Hitz of Auspex Systems Inc.
  *
- * Redistribution and use in source and binary forms are permitted
- * provided that the above copyright notice and this paragraph are
- * duplicated in all such forms and that any documentation,
- * advertising materials, and other materials related to such
- * distribution and use acknowledge that the software was developed
- * by the University of California, Berkeley.  The name of the
- * University may not be used to endorse or promote products derived
- * from this software without specific prior written permission.
- * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR
- * IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
- * WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE.
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ * 3. All advertising materials mentioning features or use of this software
+ *    must display the following acknowledgement:
+ *     This product includes software developed by the University of
+ *     California, Berkeley and its contributors.
+ * 4. Neither the name of the University nor the names of its contributors
+ *    may be used to endorse or promote products derived from this software
+ *    without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
  */
 
 #ifndef lint
  */
 
 #ifndef lint
@@ -25,13 +41,13 @@ char copyright[] =
 #endif /* not lint */
 
 #ifndef lint
 #endif /* not lint */
 
 #ifndef lint
-static char sccsid[] = "@(#)cp.c       5.2 (Berkeley) %G%";
+static char sccsid[] = "@(#)cp.c       5.24 (Berkeley) 5/6/91";
 #endif /* not lint */
 
 /*
  * cp copies source files to target files.
  * 
 #endif /* not lint */
 
 /*
  * cp copies source files to target files.
  * 
- * The global path_t structures "to" and "from" always contain paths to the
+ * The global PATH_T structures "to" and "from" always contain paths to the
  * current source and target files, respectively.  Since cp does not change
  * directories, these paths can be either absolute or dot-realative.
  * 
  * current source and target files, respectively.  Since cp does not change
  * directories, these paths can be either absolute or dot-realative.
  * 
@@ -40,65 +56,65 @@ static char sccsid[] = "@(#)cp.c    5.2 (Berkeley) %G%";
  * copy copies the data.  If "from" is a directory, copy creates the
  * corresponding "to" directory, and calls itself recursively on all of
  * the entries in the "from" directory.
  * copy copies the data.  If "from" is a directory, copy creates the
  * corresponding "to" directory, and calls itself recursively on all of
  * the entries in the "from" directory.
- * 
- * Instead of handling directory entries in the order they appear on disk,
- * copy() does non-directory files before directory files.
- * 
- * There are two reasons to do directories last.  The first is efficiency.
- * Files tend to be in the same cylinder group as their parent, whereas
- * directories tend not to be. Copying files all at once reduces seeking.
- * 
- * Second, deeply nested tree's could use up all the file descriptors if we
- * didn't close one directory before recursivly starting on the next.
  */
 
 #include <sys/param.h>
 #include <sys/stat.h>
  */
 
 #include <sys/param.h>
 #include <sys/stat.h>
-#include <sys/file.h>
-#include <sys/dir.h>
 #include <sys/time.h>
 #include <sys/time.h>
-
-#include <stdio.h>
+#include <dirent.h>
+#include <fcntl.h>
 #include <errno.h>
 #include <errno.h>
-#include <strings.h>
-
-typedef struct {
-       char    *p_path;        /* Pointer to the start of a path. */
-       char    *p_end;         /* Pointer to NULL at end of path. */
-} path_t;
+#include <unistd.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include "cp.h"
 
 
-char *path_append(), *path_basename();
-void path_restore();
+PATH_T from = { from.p_path, "" };
+PATH_T to = { to.p_path, "" };
 
 
-int exit_val, my_umask;
-int interactive_flag, preserve_flag, recursive_flag;
-char *buf;                             /* I/O; malloc for best alignment. */
-char from_buf[MAXPATHLEN + 1],         /* Source path buffer. */
-     to_buf[MAXPATHLEN + 1];           /* Target path buffer. */
-path_t from = {from_buf, from_buf};
-path_t to = {to_buf, to_buf};
+uid_t myuid;
+int exit_val, myumask;
+int iflag, pflag, orflag, rflag;
+int (*statfcn)();
+char *buf, *progname;
 
 main(argc, argv)
        int argc;
        char **argv;
 {
 
 main(argc, argv)
        int argc;
        char **argv;
 {
-       extern int optind, errno;
+       extern int optind;
        struct stat to_stat;
        register int c, r;
        struct stat to_stat;
        register int c, r;
-       char *old_to, *malloc();
+       int symfollow, lstat(), stat();
+       char *old_to, *p;
+
+       /*
+        * The utility cp(1) is used by mv(1) -- except for usage statements,
+        * print the "called as" program name.
+        */
+       progname = (p = rindex(*argv,'/')) ? ++p : *argv;
 
 
-       while ((c = getopt(argc, argv, "Ripr")) != EOF) {
-       switch ((char) c) {
+       symfollow = 0;
+       while ((c = getopt(argc, argv, "Rfhipr")) != EOF) {
+       switch ((char)c) {
+               case 'f':
+                       iflag = 0;
+                       break;
+               case 'h':
+                       symfollow = 1;
+                       break;
                case 'i':
                case 'i':
-                       interactive_flag = isatty(fileno(stdin));
+                       iflag = isatty(fileno(stdin));
                        break;
                case 'p':
                        break;
                case 'p':
-                       preserve_flag = 1;
-                       (void)umask(0);
+                       pflag = 1;
                        break;
                        break;
-               case 'r':
                case 'R':
                case 'R':
-                       recursive_flag = 1;
+                       rflag = 1;
+                       break;
+               case 'r':
+                       orflag = 1;
                        break;
                case '?':
                default:
                        break;
                case '?':
                default:
@@ -106,31 +122,41 @@ main(argc, argv)
                        break;
                }
        }
                        break;
                }
        }
-       argc -= optind;         /* argc is count of remaining arguments. */
-       argv += optind;         /* argv[0] is first remaining argument. */
+       argc -= optind;
+       argv += optind;
 
        if (argc < 2)
                usage();
 
 
        if (argc < 2)
                usage();
 
-       my_umask = umask(0);
-       (void)umask(my_umask);
+       if (rflag && orflag) {
+               (void)fprintf(stderr,
+                   "cp: the -R and -r options are mutually exclusive.\n");
+               exit(1);
+       }
 
        buf = (char *)malloc(MAXBSIZE);
        if (!buf) {
 
        buf = (char *)malloc(MAXBSIZE);
        if (!buf) {
-               fprintf(stderr, "cp: out of space.\n");
+               (void)fprintf(stderr, "%s: out of space.\n", progname);
                exit(1);
        }
 
                exit(1);
        }
 
-       /* Consume last argument first. */
+       myuid = getuid();
+
+       /* copy the umask for explicit mode setting */
+       myumask = umask(0);
+       (void)umask(myumask);
+
+       /* consume last argument first. */
        if (!path_set(&to, argv[--argc]))
        if (!path_set(&to, argv[--argc]))
-               exit(exit_val);
+               exit(1);
+
+       statfcn = symfollow || !rflag ? stat : lstat;
 
        /*
         * Cp has two distinct cases:
         *
 
        /*
         * Cp has two distinct cases:
         *
-        * Case (1)       $ cp [-rip] source target
-        *
-        * Case (2)       $ cp [-rip] source1 ... directory
+        * % cp [-rip] source target
+        * % cp [-rip] source1 ... directory
         *
         * In both cases, source can be either a file or a directory.
         *
         *
         * In both cases, source can be either a file or a directory.
         *
@@ -146,7 +172,7 @@ main(argc, argv)
                error(to.p_path);
                exit(1);
        }
                error(to.p_path);
                exit(1);
        }
-       if (r == -1 || (to_stat.st_mode & S_IFMT) != S_IFDIR) {
+       if (r == -1 || !S_ISDIR(to_stat.st_mode)) {
                /*
                 * Case (1).  Target is not a directory.
                 */
                /*
                 * Case (1).  Target is not a directory.
                 */
@@ -155,345 +181,370 @@ main(argc, argv)
                        exit(1);
                }
                if (!path_set(&from, *argv))
                        exit(1);
                }
                if (!path_set(&from, *argv))
-                       exit(exit_val);
+                       exit(1);
                copy();
        }
        else {
                /*
                 * Case (2).  Target is a directory.
                 */
                copy();
        }
        else {
                /*
                 * Case (2).  Target is a directory.
                 */
-               for (; argc; --argc, ++argv) {
-                       if (!path_set(&from, *argv))
+               for (;; ++argv) {
+                       if (!path_set(&from, *argv)) {
+                               exit_val = 1;
                                continue;
                                continue;
+                       }
                        old_to = path_append(&to, path_basename(&from), -1);
                        old_to = path_append(&to, path_basename(&from), -1);
-                       if (!old_to)
+                       if (!old_to) {
+                               exit_val = 1;
                                continue;
                                continue;
+                       }
                        copy();
                        copy();
+                       if (!--argc)
+                               break;
                        path_restore(&to, old_to);
                }
        }
        exit(exit_val);
 }
 
                        path_restore(&to, old_to);
                }
        }
        exit(exit_val);
 }
 
-/*
- * Copy file or directory at "from" to "to".
- */
+/* copy file or directory at "from" to "to". */
 copy()
 {
        struct stat from_stat, to_stat;
 copy()
 {
        struct stat from_stat, to_stat;
-       int new_target_dir = 0;
+       int dne, statval;
 
 
-       if (stat(from.p_path, &from_stat) == -1) {
+       statval = statfcn(from.p_path, &from_stat);
+       if (statval == -1) {
                error(from.p_path);
                return;
        }
                error(from.p_path);
                return;
        }
-       /*
-        * This is not an error, but we need to remember that it happened.
-        */
-       if (stat(to.p_path, &to_stat) == -1)
-               to_stat.st_ino = -1;
-       else if (to_stat.st_dev == from_stat.st_dev &&
-           to_stat.st_ino == from_stat.st_ino) {
-               fprintf(stderr,
-                   "cp: \"%s\" and \"%s\" are identical (not copied).\n",
-                   to.p_path, from.p_path);
-               exit_val = 1;
-               return;
-       }
 
 
-       if ((from_stat.st_mode & S_IFMT) != S_IFDIR) {
-               if (!copy_file(from_stat.st_mode))
+       /* not an error, but need to remember it happened */
+       if (stat(to.p_path, &to_stat) == -1)
+               dne = 1;
+       else {
+               if (to_stat.st_dev == from_stat.st_dev &&
+                   to_stat.st_ino == from_stat.st_ino) {
+                       (void)fprintf(stderr,
+                           "%s: %s and %s are identical (not copied).\n",
+                           progname, to.p_path, from.p_path);
+                       exit_val = 1;
                        return;
                        return;
+               }
+               dne = 0;
        }
        }
-       else {
-               if (!recursive_flag) {
+
+       switch(from_stat.st_mode & S_IFMT) {
+       case S_IFLNK:
+               copy_link(!dne);
+               return;
+       case S_IFDIR:
+               if (!rflag && !orflag) {
                        (void)fprintf(stderr,
                        (void)fprintf(stderr,
-                          "cp: \"%s\" is a directory (not copied).\n",
-                          from.p_path);
+                           "%s: %s is a directory (not copied).\n",
+                           progname, from.p_path);
                        exit_val = 1;
                        return;
                }
                        exit_val = 1;
                        return;
                }
-               if (to_stat.st_ino == -1) {
-                       if (mkdir(to.p_path, 0777) < 0) {
+               if (dne) {
+                       /*
+                        * If the directory doesn't exist, create the new
+                        * one with the from file mode plus owner RWX bits,
+                        * modified by the umask.  Trade-off between being
+                        * able to write the directory (if from directory is
+                        * 555) and not causing a permissions race.  If the
+                        * umask blocks owner writes cp fails.
+                        */
+                       if (mkdir(to.p_path, from_stat.st_mode|S_IRWXU) < 0) {
                                error(to.p_path);
                                return;
                        }
                                error(to.p_path);
                                return;
                        }
-                       new_target_dir = 1;
                }
                }
-               else if ((to_stat.st_mode & S_IFMT) != S_IFDIR) {
-                       (void)fprintf(stderr,
-                          "cp: \"%s\": not a directory.\n",
-                          to.p_path);
+               else if (!S_ISDIR(to_stat.st_mode) != S_IFDIR) {
+                       (void)fprintf(stderr, "%s: %s: not a directory.\n",
+                           progname, to.p_path);
                        return;
                }
                copy_dir();
                        return;
                }
                copy_dir();
+               /*
+                * If not -p and directory didn't exist, set it to be the
+                * same as the from directory, umodified by the umask;
+                * arguably wrong, but it's been that way forever.
+                */
+               if (pflag)
+                       setfile(&from_stat, 0);
+               else if (dne)
+                       (void)chmod(to.p_path, from_stat.st_mode);
+               return;
+       case S_IFCHR:
+       case S_IFBLK:
+               if (rflag) {
+                       copy_special(&from_stat, !dne);
+                       return;
+               }
+               break;
+       case S_IFIFO:
+               if (rflag) {
+                       copy_fifo(&from_stat, !dne);
+                       return;
+               }
+               break;
        }
        }
-       /* Preserve old times/modes if necessary. */
-       if (preserve_flag)
-               (void)chmod(to.p_path, (int) from_stat.st_mode);
-       else if (new_target_dir)
-               (void)chmod(to.p_path, (int) from_stat.st_mode & ~my_umask);
-       if (preserve_flag || new_target_dir) {
-               static struct timeval tv[2];
-
-               tv[0].tv_sec = from_stat.st_atime;
-               tv[1].tv_sec = from_stat.st_mtime;
-               if (utimes(to.p_path, tv))
-                       error(to.p_path);
-       }
+       copy_file(&from_stat, dne);
 }
 
 }
 
-copy_file(mode)
-       u_short mode;                   /* Permissions for new file. */
+copy_file(fs, dne)
+       struct stat *fs;
+       int dne;
 {
 {
-       int from_fd, to_fd, rcount, wcount, r;
-       char c;
+       register int from_fd, to_fd, rcount, wcount;
+       struct stat to_stat;
 
 
-       from_fd = open(from.p_path, O_RDONLY, 0);
-       if (from_fd == -1) {
+       if ((from_fd = open(from.p_path, O_RDONLY, 0)) == -1) {
                error(from.p_path);
                error(from.p_path);
-               (void)close(from_fd);
-               return(0);
+               return;
        }
 
        /*
        }
 
        /*
-        * In the interactive case, use O_EXCL to notice existing files. If
-        * the file exists, verify with the user.
+        * If the file exists and we're interactive, verify with the user.
+        * If the file DNE, set the mode to be the from file, minus setuid
+        * bits, modified by the umask; arguably wrong, but it makes copying
+        * executables work right and it's been that way forever.  (The
+        * other choice is 666 or'ed with the execute bits on the from file
+        * modified by the umask.)
         */
         */
-       to_fd = open(to.p_path,
-            (interactive_flag ? O_EXCL : 0) | O_WRONLY | O_CREAT | O_TRUNC,
-                mode);
-
-       if (to_fd == -1 && errno == EEXIST && interactive_flag) {
-               (void)fprintf(stderr, "overwrite \"%s\"? ", to.p_path);
-               r = scanf("%1s", &c);
-               if (r != 1 || c != 'y')
-                       return(0);
-               /* Try again. */
-               to_fd = open(to.p_path, O_WRONLY | O_CREAT | O_TRUNC, mode);
-       }
+       if (!dne) {
+               if (iflag) {
+                       int checkch, ch;
+
+                       (void)fprintf(stderr, "overwrite %s? ", to.p_path);
+                       checkch = ch = getchar();
+                       while (ch != '\n' && ch != EOF)
+                               ch = getchar();
+                       if (checkch != 'y') {
+                               (void)close(from_fd);
+                               return;
+                       }
+               }
+               to_fd = open(to.p_path, O_WRONLY|O_TRUNC, 0);
+       } else
+               to_fd = open(to.p_path, O_WRONLY|O_CREAT|O_TRUNC,
+                   fs->st_mode & ~(S_ISUID|S_ISGID));
 
        if (to_fd == -1) {
                error(to.p_path);
                (void)close(from_fd);
 
        if (to_fd == -1) {
                error(to.p_path);
                (void)close(from_fd);
-               return(0);
+               return;
        }
 
        while ((rcount = read(from_fd, buf, MAXBSIZE)) > 0) {
                wcount = write(to_fd, buf, rcount);
                if (rcount != wcount || wcount == -1) {
        }
 
        while ((rcount = read(from_fd, buf, MAXBSIZE)) > 0) {
                wcount = write(to_fd, buf, rcount);
                if (rcount != wcount || wcount == -1) {
-                       error(from.p_path);
+                       error(to.p_path);
                        break;
                }
        }
                        break;
                }
        }
+       if (rcount < 0)
+               error(from.p_path);
+       if (pflag)
+               setfile(fs, to_fd);
+       /*
+        * If the source was setuid or setgid, lose the bits unless the
+        * copy is owned by the same user and group.
+        */
+       else if (fs->st_mode & (S_ISUID|S_ISGID) && fs->st_uid == myuid)
+               if (fstat(to_fd, &to_stat))
+                       error(to.p_path);
+#define        RETAINBITS      (S_ISUID|S_ISGID|S_ISVTX|S_IRWXU|S_IRWXG|S_IRWXO)
+               else if (fs->st_gid == to_stat.st_gid && fchmod(to_fd,
+                   fs->st_mode & RETAINBITS & ~myumask))
+                       error(to.p_path);
        (void)close(from_fd);
        (void)close(from_fd);
-       (void)close(to_fd);
-       return(1);
+       if (close(to_fd))
+               error(to.p_path);
 }
 
 copy_dir()
 {
        struct stat from_stat;
 }
 
 copy_dir()
 {
        struct stat from_stat;
+       struct dirent *dp, **dir_list;
+       register int dir_cnt, i;
        char *old_from, *old_to;
        char *old_from, *old_to;
-       struct direct *dp, **dir_list;
-       int dir_cnt, i;
 
        dir_cnt = scandir(from.p_path, &dir_list, NULL, NULL);
        if (dir_cnt == -1) {
 
        dir_cnt = scandir(from.p_path, &dir_list, NULL, NULL);
        if (dir_cnt == -1) {
-               (void)fprintf(stderr, "cp: Can't read directory \"%s\".\n",
-                   from.p_path);
+               (void)fprintf(stderr, "%s: can't read directory %s.\n",
+                   progname, from.p_path);
                exit_val = 1;
        }
 
                exit_val = 1;
        }
 
-       /* Copy files first. */
+       /*
+        * Instead of handling directory entries in the order they appear
+        * on disk, do non-directory files before directory files.
+        * There are two reasons to do directories last.  The first is
+        * efficiency.  Files tend to be in the same cylinder group as
+        * their parent, whereas directories tend not to be.  Copying files
+        * all at once reduces seeking.  Second, deeply nested tree's
+        * could use up all the file descriptors if we didn't close one
+        * directory before recursivly starting on the next.
+        */
+       /* copy files */
        for (i = 0; i < dir_cnt; ++i) {
                dp = dir_list[i];
                if (dp->d_namlen <= 2 && dp->d_name[0] == '.'
        for (i = 0; i < dir_cnt; ++i) {
                dp = dir_list[i];
                if (dp->d_namlen <= 2 && dp->d_name[0] == '.'
-                   && (dp->d_name[1] == NULL || dp->d_name[1] == '.')) {
-                       (void)free((char *)dp);
-                       dir_list[i] = NULL;
-                       continue;
-               }
+                   && (dp->d_name[1] == NULL || dp->d_name[1] == '.'))
+                       goto done;
                old_from = path_append(&from, dp->d_name, (int)dp->d_namlen);
                if (!old_from) {
                old_from = path_append(&from, dp->d_name, (int)dp->d_namlen);
                if (!old_from) {
-                       dir_list[i] = NULL;
-                       (void)free((char *)dp);
-                       continue;
+                       exit_val = 1;
+                       goto done;
                }
 
                }
 
-               if (stat(from.p_path, &from_stat) < 0) {
+               if (statfcn(from.p_path, &from_stat) < 0) {
                        error(dp->d_name);
                        error(dp->d_name);
+                       path_restore(&from, old_from);
+                       goto done;
+               }
+               if (S_ISDIR(from_stat.st_mode)) {
                        path_restore(&from, old_from);
                        continue;
                }
                        path_restore(&from, old_from);
                        continue;
                }
-
-               if ((from_stat.st_mode & S_IFMT) != S_IFDIR) {
-                       old_to = path_append(&to, dp->d_name,
-                           (int)dp->d_namlen);
-                       if (!old_to) {
-                               dir_list[i] = NULL;
-                               (void)free((char *)dp);
-                               continue;
-                       }
+               old_to = path_append(&to, dp->d_name, (int)dp->d_namlen);
+               if (old_to) {
                        copy();
                        path_restore(&to, old_to);
                        copy();
                        path_restore(&to, old_to);
-                       dir_list[i] = NULL;
-                       (void)free((char *)dp);
-               }
+               } else
+                       exit_val = 1;
                path_restore(&from, old_from);
                path_restore(&from, old_from);
+done:          dir_list[i] = NULL;
+               (void)free((void *)dp);
        }
 
        }
 
-       /* Then copy directories. */
+       /* copy directories */
        for (i = 0; i < dir_cnt; ++i) {
                dp = dir_list[i];
                if (!dp)
                        continue;
                old_from = path_append(&from, dp->d_name, (int) dp->d_namlen);
                if (!old_from) {
        for (i = 0; i < dir_cnt; ++i) {
                dp = dir_list[i];
                if (!dp)
                        continue;
                old_from = path_append(&from, dp->d_name, (int) dp->d_namlen);
                if (!old_from) {
-                       (void)free((char *)dp);
+                       exit_val = 1;
+                       (void)free((void *)dp);
                        continue;
                }
                old_to = path_append(&to, dp->d_name, (int) dp->d_namlen);
                if (!old_to) {
                        continue;
                }
                old_to = path_append(&to, dp->d_name, (int) dp->d_namlen);
                if (!old_to) {
-                       (void)free((char *)dp);
+                       exit_val = 1;
+                       (void)free((void *)dp);
                        path_restore(&from, old_from);
                        continue;
                }
                copy();
                        path_restore(&from, old_from);
                        continue;
                }
                copy();
-               free((char *)dp);
+               free((void *)dp);
                path_restore(&from, old_from);
                path_restore(&to, old_to);
        }
                path_restore(&from, old_from);
                path_restore(&to, old_to);
        }
-       free((char *)dir_list);
+       free((void *)dir_list);
 }
 
 }
 
-error(s)
-       char *s;
+copy_link(exists)
+       int exists;
 {
 {
-       extern int errno;
+       int len;
+       char link[MAXPATHLEN];
 
 
-       exit_val = 1;
-       (void)fprintf(stderr, "cp: %s: %s\n", s, strerror(errno));
+       if ((len = readlink(from.p_path, link, sizeof(link))) == -1) {
+               error(from.p_path);
+               return;
+       }
+       link[len] = '\0';
+       if (exists && unlink(to.p_path)) {
+               error(to.p_path);
+               return;
+       }
+       if (symlink(link, to.p_path)) {
+               error(link);
+               return;
+       }
 }
 
 }
 
-/********************************************************************
- * Path Manipulation Routines.
- ********************************************************************/
-
-/*
- * These functions manipulate paths in "path_t" structures.
- * 
- * They eliminate multiple slashes in paths when they notice them, and keep
- * the path non-slash terminated.
- *
- * Both path_set() and path_append() return 0 if the requested name
- * would be too long.
- */
-
-#define        STRIP_TRAILING_SLASH(p) { \
-       while ((p)->p_end > (p)->p_path && (p)->p_end[-1] == '/') \
-               *--(p)->p_end = 0; \
+copy_fifo(from_stat, exists)
+       struct stat *from_stat;
+       int exists;
+{
+       if (exists && unlink(to.p_path)) {
+               error(to.p_path);
+               return;
+       }
+       if (mkfifo(to.p_path, from_stat->st_mode)) {
+               error(to.p_path);
+               return;
        }
        }
+       if (pflag)
+               setfile(from_stat, 0);
+}
 
 
-/*
- * Move specified string into path.  Convert "" to "." to handle BSD
- * semantics for a null path.  Strip trailing slashes.
- */
-path_set(p, string)
-       register path_t *p;
-       char *string;
+copy_special(from_stat, exists)
+       struct stat *from_stat;
+       int exists;
 {
 {
-       if (strlen(string) > MAXPATHLEN) {
-               fprintf(stderr, "cp: \"%s\": Name too long.\n", string);
-               exit_val = 1;
-               return(0);
+       if (exists && unlink(to.p_path)) {
+               error(to.p_path);
+               return;
        }
        }
-
-       (void)strcpy(p->p_path, string);
-       p->p_end = p->p_path + strlen(p->p_path);
-
-       if (p->p_path == p->p_end) {
-               *p->p_end++ = '.';
-               *p->p_end = 0;
+       if (mknod(to.p_path, from_stat->st_mode,  from_stat->st_rdev)) {
+               error(to.p_path);
+               return;
        }
        }
-
-       STRIP_TRAILING_SLASH(p);
-       return(1);
+       if (pflag)
+               setfile(from_stat, 0);
 }
 
 }
 
-/*
- * Append specified string to path, inserting '/' if necessary.  Return a
- * pointer to the old end of path for restoration.
- */
-char *
-path_append(p, name, len)
-       register path_t *p;
-       char *name;
-       int len;
+setfile(fs, fd)
+       register struct stat *fs;
+       int fd;
 {
 {
-       char *old;
+       static struct timeval tv[2];
+       char path[100];
 
 
-       old = p->p_end;
-       if (len == -1)
-               len = strlen(name);
+       fs->st_mode &= S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
 
 
-       /*
-        * The final "+ 1" accounts for the '/' between old path and name.
-        */
-       if ((len + p->p_end - p->p_path + 1) > MAXPATHLEN) {
-               fprintf(stderr,
-                   "cp: \"%s/%s\": Name too long.\n", p->p_path, name);
-               exit_val = 1;
-               return(0);
+       tv[0].tv_sec = fs->st_atime;
+       tv[1].tv_sec = fs->st_mtime;
+       if (utimes(to.p_path, tv)) {
+               (void)snprintf(path, sizeof(path), "utimes: %s", to.p_path);
+               error(path);
        }
        }
-
        /*
        /*
-        * This code should always be executed, since paths shouldn't
-        * end in '/'.
+        * Changing the ownership probably won't succeed, unless we're root
+        * or POSIX_CHOWN_RESTRICTED is not set.  Set uid/gid before setting
+        * the mode; current BSD behavior is to remove all setuid bits on
+        * chown.  If chown fails, lose setuid/setgid bits.
         */
         */
-       if (p->p_end[-1] != '/') {
-               *p->p_end++ = '/';
-               *p->p_end = 0;
+       if (fd ? fchown(fd, fs->st_uid, fs->st_gid) :
+           chown(to.p_path, fs->st_uid, fs->st_gid)) {
+               if (errno != EPERM) {
+                       (void)snprintf(path, sizeof(path),
+                           "chown: %s", to.p_path);
+                       error(path);
+               }
+               fs->st_mode &= ~(S_ISUID|S_ISGID);
+       }
+       if (fd ? fchmod(fd, fs->st_mode) : chmod(to.p_path, fs->st_mode)) {
+               (void)snprintf(path, sizeof(path), "chown: %s", to.p_path);
+               error(path);
        }
        }
-
-       (void)strncat(p->p_end, name, len);
-       p->p_end += len;
-       *p->p_end = 0;
-
-       STRIP_TRAILING_SLASH(p);
-       return(old);
-}
-
-/*
- * Restore path to previous value.  (As returned by path_append.)
- */
-void
-path_restore(p, old)
-       path_t *p;
-       char *old;
-{
-       p->p_end = old;
-       *p->p_end = 0;
 }
 
 }
 
-/*
- * Return basename of path.  (Like basename(1).)
- */
-char *
-path_basename(p)
-       path_t *p;
+error(s)
+       char *s;
 {
 {
-       char *basename;
-
-       basename = rindex(p->p_path, '/');
-       if (!basename)
-               basename = p->p_path;
-       return(basename);
+       exit_val = 1;
+       (void)fprintf(stderr, "%s: %s: %s\n", progname, s, strerror(errno));
 }
 
 usage()
 {
        (void)fprintf(stderr,
 }
 
 usage()
 {
        (void)fprintf(stderr,
-          "usage: cp [-ip] f1 f2; or: cp [-irp] f1 ... fn directory\n");
+"usage: cp [-Rfhip] src target;\n   or: cp [-Rfhip] src1 ... srcN directory\n");
        exit(1);
 }
        exit(1);
 }