bin/ln/ln.c

   1 /*-
   2  * Copyright (c) 1987, 1993, 1994
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 4. Neither the name of the University nor the names of its contributors
  14  *    may be used to endorse or promote products derived from this software
  15  *    without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  */
  29
  30 #if 0
  31 #ifndef lint
  32 static char const copyright[] =
  33 "@(#) Copyright (c) 1987, 1993, 1994\n\
  34         The Regents of the University of California.  All rights reserved.\n";
  35 #endif /* not lint */
  36
  37 #ifndef lint
  38 static char sccsid[] = "@(#)ln.c        8.2 (Berkeley) 3/31/94";
  39 #endif /* not lint */
  40 #endif
  41 #include <sys/cdefs.h>
  42 __FBSDID("$FreeBSD$");
  43
  44 #include <sys/param.h>
  45 #include <sys/stat.h>
  46
  47 #include <err.h>
  48 #include <errno.h>
  49 #include <limits.h>
  50 #include <stdio.h>
  51 #include <stdlib.h>
  52 #include <string.h>
  53 #include <unistd.h>
  54
  55 int     fflag;                          /* Unlink existing files. */
  56 int     Fflag;                          /* Remove empty directories also. */
  57 int     hflag;                          /* Check new name for symlink first. */
  58 int     iflag;                          /* Interactive mode. */
  59 int     sflag;                          /* Symbolic, not hard, link. */
  60 int     vflag;                          /* Verbose output. */
  61                                         /* System link call. */
  62 int (*linkf)(const char *, const char *);
  63 char    linkch;
  64
  65 int     linkit(const char *, const char *, int);
  66 void    usage(void);
  67
  68 int
  69 main(int argc, char *argv[])
  70 {
  71         struct stat sb;
  72         char *p, *sourcedir;
  73         int ch, exitval;
  74
  75         /*
  76          * Test for the special case where the utility is called as
  77          * "link", for which the functionality provided is greatly
  78          * simplified.
  79          */
  80         if ((p = rindex(argv[0], '/')) == NULL)
  81                 p = argv[0];
  82         else
  83                 ++p;
  84         if (strcmp(p, "link") == 0) {
  85                 while (getopt(argc, argv, "") != -1)
  86                         usage();
  87                 argc -= optind;
  88                 argv += optind;
  89                 if (argc != 2)
  90                         usage();
  91                 linkf = link;
  92                 exit(linkit(argv[0], argv[1], 0));
  93         }
  94
  95         while ((ch = getopt(argc, argv, "Ffhinsv")) != -1)
  96                 switch (ch) {
  97                 case 'F':
  98                         Fflag = 1;
  99                         break;
 100                 case 'f':
 101                         fflag = 1;
 102                         iflag = 0;
 103                         break;
 104                 case 'h':
 105                 case 'n':
 106                         hflag = 1;
 107                         break;
 108                 case 'i':
 109                         iflag = 1;
 110                         fflag = 0;
 111                         break;
 112                 case 's':
 113                         sflag = 1;
 114                         break;
 115                 case 'v':
 116                         vflag = 1;
 117                         break;
 118                 case '?':
 119                 default:
 120                         usage();
 121                 }
 122
 123         argv += optind;
 124         argc -= optind;
 125
 126         linkf = sflag ? symlink : link;
 127         linkch = sflag ? '-' : '=';
 128         if (sflag == 0)
 129                 Fflag = 0;
 130         if (Fflag == 1 && iflag == 0)
 131                 fflag = 1;
 132
 133         switch(argc) {
 134         case 0:
 135                 usage();
 136                 /* NOTREACHED */
 137         case 1:                         /* ln target */
 138                 exit(linkit(argv[0], ".", 1));
 139         case 2:                         /* ln target source */
 140                 exit(linkit(argv[0], argv[1], 0));
 141         default:
 142                 ;
 143         }
 144                                         /* ln target1 target2 directory */
 145         sourcedir = argv[argc - 1];
 146         if (hflag && lstat(sourcedir, &sb) == 0 && S_ISLNK(sb.st_mode)) {
 147                 /*
 148                  * We were asked not to follow symlinks, but found one at
 149                  * the target--simulate "not a directory" error
 150                  */
 151                 errno = ENOTDIR;
 152                 err(1, "%s", sourcedir);
 153         }
 154         if (stat(sourcedir, &sb))
 155                 err(1, "%s", sourcedir);
 156         if (!S_ISDIR(sb.st_mode))
 157                 usage();
 158         for (exitval = 0; *argv != sourcedir; ++argv)
 159                 exitval |= linkit(*argv, sourcedir, 1);
 160         exit(exitval);
 161 }
 162
 163 int
 164 linkit(const char *target, const char *source, int isdir)
 165 {
 166         struct stat sb;
 167         const char *p;
 168         int ch, exists, first;
 169         char path[PATH_MAX];
 170
 171         if (!sflag) {
 172                 /* If target doesn't exist, quit now. */
 173                 if (stat(target, &sb)) {
 174                         warn("%s", target);
 175                         return (1);
 176                 }
 177                 /* Only symbolic links to directories. */
 178                 if (S_ISDIR(sb.st_mode)) {
 179                         errno = EISDIR;
 180                         warn("%s", target);
 181                         return (1);
 182                 }
 183         }
 184
 185         /*
 186          * If the source is a directory (and not a symlink if hflag),
 187          * append the target's name.
 188          */
 189         if (isdir ||
 190             (lstat(source, &sb) == 0 && S_ISDIR(sb.st_mode)) ||
 191             (!hflag && stat(source, &sb) == 0 && S_ISDIR(sb.st_mode))) {
 192                 if ((p = strrchr(target, '/')) == NULL)
 193                         p = target;
 194                 else
 195                         ++p;
 196                 if (snprintf(path, sizeof(path), "%s/%s", source, p) >=
 197                     (ssize_t)sizeof(path)) {
 198                         errno = ENAMETOOLONG;
 199                         warn("%s", target);
 200                         return (1);
 201                 }
 202                 source = path;
 203         }
 204
 205         exists = !lstat(source, &sb);
 206         /*
 207          * If the file exists, then unlink it forcibly if -f was specified
 208          * and interactively if -i was specified.
 209          */
 210         if (fflag && exists) {
 211                 if (Fflag && S_ISDIR(sb.st_mode)) {
 212                         if (rmdir(source)) {
 213                                 warn("%s", source);
 214                                 return (1);
 215                         }
 216                 } else if (unlink(source)) {
 217                         warn("%s", source);
 218                         return (1);
 219                 }
 220         } else if (iflag && exists) {
 221                 fflush(stdout);
 222                 fprintf(stderr, "replace %s? ", source);
 223
 224                 first = ch = getchar();
 225                 while(ch != '\n' && ch != EOF)
 226                         ch = getchar();
 227                 if (first != 'y' && first != 'Y') {
 228                         fprintf(stderr, "not replaced\n");
 229                         return (1);
 230                 }
 231
 232                 if (Fflag && S_ISDIR(sb.st_mode)) {
 233                         if (rmdir(source)) {
 234                                 warn("%s", source);
 235                                 return (1);
 236                         }
 237                 } else if (unlink(source)) {
 238                         warn("%s", source);
 239                         return (1);
 240                 }
 241         }
 242
 243         /* Attempt the link. */
 244         if ((*linkf)(target, source)) {
 245                 warn("%s", source);
 246                 return (1);
 247         }
 248         if (vflag)
 249                 (void)printf("%s %c> %s\n", source, linkch, target);
 250         return (0);
 251 }
 252
 253 void
 254 usage(void)
 255 {
 256         (void)fprintf(stderr, "%s\n%s\n%s\n",
 257             "usage: ln [-Ffhinsv] source_file [target_file]",
 258             "       ln [-Ffhinsv] source_file ... target_dir",
 259             "       link source_file target_file");
 260         exit(1);
 261 }