These messages can be .rodata, so make them even more const.

[oweals/busybox.git] / ln.c

diff --git a/ln.c b/ln.c
index 37148225121795e15d26b92e644d42da64ffb9a9..7412a86fd124dd7a581a0227efa03e1d0c231f86 100644 (file)
--- a/ln.c
+++ b/ln.c
@@ -2,7 +2,7 @@
 /*
  * Mini ln implementation for busybox
  *
- * Copyright (C) 1999,2000 by Lineo, inc.
+ * Copyright (C) 1999,2000,2001 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
@@ -21,122 +21,105 @@
  *
  */
 
-#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 <string.h>
+#include <stdlib.h>
 #include <errno.h>
+#include <unistd.h>
+#include "busybox.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 const int LN_SYMLINK = 1;
+static const int LN_FORCE = 2;
+static const int LN_NODEREFERENCE = 4;
 
-static int symlinkFlag = FALSE;
-static int removeoldFlag = FALSE;
-static int followLinks = TRUE;
+/*
+ * linkDestName is where the link points to,
+ * linkSrcName is the name of the link to be created.
+ */
+static int fs_link(const char *link_destname, const char *link_srcname, 
+               const int flag)
+{
+       int status;
+       int src_is_dir;
+       char *src_name;
+
+       if (link_destname==NULL)
+               return(FALSE);
+
+       src_name = (char *) xmalloc(strlen(link_srcname)+strlen(link_destname)+1);
+
+       if (link_srcname==NULL)
+               strcpy(src_name, link_destname);
+       else
+               strcpy(src_name, link_srcname);
+
+       if (flag&LN_NODEREFERENCE)
+               src_is_dir = is_directory(src_name, TRUE, NULL);
+       else
+               src_is_dir = is_directory(src_name, FALSE, NULL);       
+       
+       if ((src_is_dir==TRUE)&&((flag&LN_NODEREFERENCE)==0)) {
+               char* srcdir_name;
+
+               srcdir_name = xstrdup(link_destname);
+               strcat(src_name, "/");
+               strcat(src_name, get_last_path_component(srcdir_name));
+               free(srcdir_name);
+       }
+       
+       if (flag&LN_FORCE)
+               unlink(src_name);
+               
+       if (flag&LN_SYMLINK)
+               status = symlink(link_destname, src_name);
+       else
+               status = link(link_destname, src_name);
+
+       if (status != 0) {
+               perror_msg(src_name);
+               return(FALSE);
+       }
+       return(TRUE);
+}
 
 extern int ln_main(int argc, char **argv)
 {
-       char *linkName;
-       int linkIntoDirFlag;
-       int stopIt = FALSE;
-
-       argc--;
-       argv++;
-
+       int status = EXIT_SUCCESS;
+       int flag = 0;
+       int opt;
+       
        /* 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++;
+       while ((opt=getopt(argc, argv, "sfn")) != -1) {
+               switch(opt) {
+                       case 's':
+                               flag |= LN_SYMLINK;
+                               break;
+                       case 'f':
+                               flag |= LN_FORCE;
+                               break;
+                       case 'n':
+                               flag |= LN_NODEREFERENCE;
+                               break;
+                       default:
+                               show_usage();
                }
-               else
-                       break;
-       }
-
-       if (argc < 2) {
-               usage(ln_usage);
        }
-
-       linkName = argv[argc - 1];
-
-       if (strlen(linkName) > BUFSIZ) {
-               fprintf(stderr, name_too_long, "ln");
-               exit FALSE;
+       if (optind > (argc-1)) {
+               show_usage();
+       } 
+       if (optind == (argc-1)) {
+               if (fs_link(argv[optind], 
+                                       get_last_path_component(argv[optind]), flag)==FALSE)
+                       status = EXIT_FAILURE;
        }
-
-       linkIntoDirFlag = isDirectory(linkName, TRUE, NULL);
-
-       if ((argc >= 3) && linkIntoDirFlag == FALSE) {
-               fprintf(stderr, not_a_directory, "ln", linkName);
-               exit FALSE;
-       }
-
-       while (argc-- >= 2) {
-               char srcName[BUFSIZ + 1];
-               int nChars, status;
-
-               if (strlen(*argv) > BUFSIZ) {
-                       fprintf(stderr, name_too_long, "ln");
-                       exit FALSE;
-               }
-
-               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';
-               }
-
-               if (removeoldFlag == TRUE) {
-                       status = (unlink(linkName) && errno != ENOENT);
-                       if (status != 0) {
-                               perror(linkName);
-                               exit FALSE;
-                       }
-               }
-
-               if (symlinkFlag == TRUE)
-                       status = symlink(*argv, linkName);
-               else
-                       status = link(*argv, linkName);
-               if (status != 0) {
-                       perror(linkName);
-                       exit FALSE;
-               }
+       while(optind<(argc-1)) {
+               if (fs_link(argv[optind], argv[argc-1], flag)==FALSE)
+                       status = EXIT_FAILURE;
+               optind++;
        }
-       exit TRUE;
+       exit(status);
 }
 
 /*