Re-organized some sections and added a whole new section on avoiding the

[oweals/busybox.git] / ln.c

diff --git a/ln.c b/ln.c
index 8ef8d05decaf69fe543dd62d03bbbafa125c92a0..e81dbafaec49667f99cb70187ad8d7202928a466 100644 (file)
--- a/ln.c
+++ b/ln.c
@@ -30,96 +30,83 @@
 #include <dirent.h>
 #include <errno.h>
 
-static int symlinkFlag = FALSE;
-static int removeoldFlag = FALSE;
-static int followLinks = TRUE;
+#define LN_SYMLINK             1
+#define LN_FORCE                       2
+#define LN_NODEREFERENCE       4
 
-extern int ln_main(int argc, char **argv)
+/*
+ * 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)
 {
-       char *linkName, *dirName=NULL;
-       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;
-       }
-
-       if (argc < 2) {
-               usage(ln_usage);
+       int status;
+       int srcIsDir;
+       char *srcName;
+
+       if (link_DestName==NULL)
+               return(FALSE);
+
+       srcName = (char *) malloc(strlen(link_SrcName)+strlen(link_DestName)+1);
+
+       if (link_SrcName==NULL)
+               strcpy(srcName, link_DestName);
+       else
+               strcpy(srcName, link_SrcName);
+
+       if (flag&LN_NODEREFERENCE)
+               srcIsDir = isDirectory(srcName, TRUE, NULL);
+       else
+               srcIsDir = isDirectory(srcName, FALSE, NULL);   
+       
+       if ((srcIsDir==TRUE)&&((flag&LN_NODEREFERENCE)==0)) {
+               strcat(srcName, "/");
+               strcat(srcName, link_DestName);
        }
-
-       linkName = argv[argc - 1];
-
-       linkIntoDirFlag = isDirectory(linkName, followLinks, NULL);
-       if ((argc >= 3) && linkIntoDirFlag == FALSE) {
-               errorMsg(not_a_directory, linkName);
-               exit FALSE;
+       
+       if (flag&LN_FORCE)
+               unlink(srcName);
+               
+       if (flag&LN_SYMLINK)
+               status = symlink(link_DestName, srcName);
+       else
+               status = link(link_DestName, srcName);
+
+       if (status != 0) {
+               perror(srcName);
+               return(FALSE);
        }
+       return(TRUE);
+}
 
-       if (linkIntoDirFlag == TRUE)
-               dirName = linkName;
-
-       while (argc-- >= 2) {
-               int status;
-
-               if (linkIntoDirFlag == TRUE) {
-                       char *baseName = get_last_path_component(*argv);
-                       linkName = (char *)xmalloc(strlen(dirName)+strlen(baseName)+2);
-                       strcpy(linkName, dirName);
-                       if(dirName[strlen(dirName)-1] != '/')
-                               strcat(linkName, "/");
-                       strcat(linkName,baseName);
-               }
-
-               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;
+extern int ln_main(int argc, char **argv)
+{
+       int status = EXIT_SUCCESS;
+       int flag = 0;
+       int opt;
+       
+       /* Parse any options */
+       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:
+                               usage(ln_usage);
                }
-
-               if (linkIntoDirFlag == TRUE)
-                       free(linkName);
-
-               argv++;
        }
-       return( TRUE);
+       while(optind<(argc-1)) {
+               if (fs_link(argv[optind], argv[argc-1], flag)==FALSE)
+                       status = EXIT_FAILURE;
+               optind++;
+       }
+       return(status);
 }
 
 /*