comm: almost total rewrite; add testsuite
[oweals/busybox.git] / coreutils / ln.c
index 2233a1d68d032d318d13cd6e43ae745e34526d2b..eb7171959cbe62b479753124fb817eda6df90309 100644 (file)
 /*
  * Mini ln implementation for busybox
  *
- * Copyright (C) 1999,2000 by Lineo, inc.
- * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
  *
+ * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  */
 
-#include "internal.h"
-#define BB_DECLARE_EXTERN
-#define bb_need_name_too_long
-#define bb_need_not_a_directory
-#include "messages.c"
-
-#include <stdio.h>
-#include <dirent.h>
-#include <errno.h>
-
-static const char ln_usage[] =
-       "ln [OPTION] TARGET... LINK_NAME|DIRECTORY\n"
-#ifndef BB_FEATURE_TRIVIAL_HELP
-       "\nCreate a link named LINK_NAME or DIRECTORY to the specified TARGET\n\n"
-       "Options:\n"
-       "\t-s\tmake symbolic links instead of hard links\n"
-
-       "\t-f\tremove existing destination files\n"
-       "\t-n\tno dereference symlinks - treat like normal file\n"
-#endif
-       ;
-
-static int symlinkFlag = FALSE;
-static int removeoldFlag = FALSE;
-static int followLinks = TRUE;
-
-extern int ln_main(int argc, char **argv)
-{
-       char *linkName;
-       int linkIntoDirFlag;
-       int stopIt = FALSE;
-
-       argc--;
-       argv++;
-
-       /* Parse any options */
-       while (argc > 0 && stopIt == FALSE) {
-               if (**argv == '-') {
-                       while (*++(*argv))
-                               switch (**argv) {
-                                       case 's':
-                                               symlinkFlag = TRUE;
-                                               break;
-                                       case 'f':
-                                               removeoldFlag = TRUE;
-                                               break;
-                                       case 'n':
-                                               followLinks = FALSE;
-                                               break;
-                                       case '-':
-                                               stopIt = TRUE;
-                                               break;
-                                       default:
-                                               usage(ln_usage);
-                               }
-                       argc--;
-                       argv++;
-               }
-               else
-                       break;
-       }
+/* BB_AUDIT SUSv3 compliant */
+/* BB_AUDIT GNU options missing: -d, -F, -i, and -v. */
+/* http://www.opengroup.org/onlinepubs/007904975/utilities/ln.html */
 
-       if (argc < 2) {
-               usage(ln_usage);
-       }
+#include "libbb.h"
 
-       linkName = argv[argc - 1];
+/* This is a NOEXEC applet. Be very careful! */
 
-       if (strlen(linkName) > BUFSIZ) {
-               fprintf(stderr, name_too_long, "ln");
-               exit FALSE;
-       }
 
-       linkIntoDirFlag = isDirectory(linkName, TRUE, NULL);
+#define LN_SYMLINK          1
+#define LN_FORCE            2
+#define LN_NODEREFERENCE    4
+#define LN_BACKUP           8
+#define LN_SUFFIX           16
 
-       if ((argc >= 3) && linkIntoDirFlag == FALSE) {
-               fprintf(stderr, not_a_directory, "ln", linkName);
-               exit FALSE;
+int ln_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int ln_main(int argc, char **argv)
+{
+       int status = EXIT_SUCCESS;
+       int flag;
+       char *last;
+       char *src_name;
+       char *src;
+       char *suffix = (char*)"~";
+       struct stat statbuf;
+       int (*link_func)(const char *, const char *);
+
+       flag = getopt32(argv, "sfnbS:", &suffix);
+
+       if (argc == optind) {
+               bb_show_usage();
        }
 
-       while (argc-- >= 2) {
-               char srcName[BUFSIZ + 1];
-               int nChars, status;
+       last = argv[argc - 1];
+       argv += optind;
 
-               if (strlen(*argv) > BUFSIZ) {
-                       fprintf(stderr, name_too_long, "ln");
-                       exit FALSE;
-               }
+       if (argc == optind + 1) {
+               *--argv = last;
+               last = bb_get_last_path_component_strip(xstrdup(last));
+       }
 
-               if (followLinks == FALSE) {
-                       strcpy(srcName, *argv);
-               } else {
-                       /* Warning!  This can silently truncate if > BUFSIZ, but
-                          I don't think that there can be one > BUFSIZ anyway. */
-                       nChars = readlink(*argv, srcName, BUFSIZ);
-                       srcName[nChars] = '\0';
+       do {
+               src_name = NULL;
+               src = last;
+
+               if (is_directory(src,
+                               (flag & LN_NODEREFERENCE) ^ LN_NODEREFERENCE,
+                               NULL)
+               ) {
+                       src_name = xstrdup(*argv);
+                       src = concat_path_file(src, bb_get_last_path_component_strip(src_name));
+                       free(src_name);
+                       src_name = src;
+               }
+               if (!(flag & LN_SYMLINK) && stat(*argv, &statbuf)) {
+                       // coreutils: "ln dangling_symlink new_hardlink" works
+                       if (lstat(*argv, &statbuf) || !S_ISLNK(statbuf.st_mode)) {
+                               bb_simple_perror_msg(*argv);
+                               status = EXIT_FAILURE;
+                               free(src_name);
+                               continue;
+                       }
                }
 
-               if (removeoldFlag == TRUE) {
-                       status = (unlink(linkName) && errno != ENOENT);
-                       if (status != 0) {
-                               perror(linkName);
-                               exit FALSE;
+               if (flag & LN_BACKUP) {
+                       char *backup;
+                       backup = xasprintf("%s%s", src, suffix);
+                       if (rename(src, backup) < 0 && errno != ENOENT) {
+                               bb_simple_perror_msg(src);
+                               status = EXIT_FAILURE;
+                               free(backup);
+                               continue;
                        }
+                       free(backup);
+                       /*
+                        * When the source and dest are both hard links to the same
+                        * inode, a rename may succeed even though nothing happened.
+                        * Therefore, always unlink().
+                        */
+                       unlink(src);
+               } else if (flag & LN_FORCE) {
+                       unlink(src);
                }
 
-               if (symlinkFlag == TRUE)
-                       status = symlink(*argv, linkName);
-               else
-                       status = link(*argv, linkName);
-               if (status != 0) {
-                       perror(linkName);
-                       exit FALSE;
+               link_func = link;
+               if (flag & LN_SYMLINK) {
+                       link_func = symlink;
                }
-       }
-       return( TRUE);
-}
 
-/*
-Local Variables:
-c-file-style: "linux"
-c-basic-offset: 4
-tab-width: 4
-End:
-*/
+               if (link_func(*argv, src) != 0) {
+                       bb_simple_perror_msg(src);
+                       status = EXIT_FAILURE;
+               }
+
+               free(src_name);
+
+       } while ((++argv)[1]);
+
+       return status;
+}