Fix return code and don't output spurious newlines.
[oweals/busybox.git] / ln.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini ln implementation for busybox
4  *
5  * Copyright (C) 1999,2000,2001 by Lineo, inc.
6  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21  *
22  */
23
24 #include <stdio.h>
25 #include <dirent.h>
26 #include <string.h>
27 #include <stdlib.h>
28 #include <errno.h>
29 #include <unistd.h>
30 #include "busybox.h"
31 #define BB_DECLARE_EXTERN
32 #define bb_need_not_a_directory
33 #include "messages.c"
34
35
36 static const int LN_SYMLINK = 1;
37 static const int LN_FORCE = 2;
38 static const int LN_NODEREFERENCE = 4;
39
40 /*
41  * linkDestName is where the link points to,
42  * linkSrcName is the name of the link to be created.
43  */
44 static int fs_link(const char *link_destname, const char *link_srcname, 
45                 const int flag)
46 {
47         int status;
48         int src_is_dir;
49         char *src_name;
50
51         if (link_destname==NULL)
52                 return(FALSE);
53
54         src_name = (char *) xmalloc(strlen(link_srcname)+strlen(link_destname)+1);
55
56         if (link_srcname==NULL)
57                 strcpy(src_name, link_destname);
58         else
59                 strcpy(src_name, link_srcname);
60
61         if (flag&LN_NODEREFERENCE)
62                 src_is_dir = is_directory(src_name, TRUE, NULL);
63         else
64                 src_is_dir = is_directory(src_name, FALSE, NULL);       
65         
66         if ((src_is_dir==TRUE)&&((flag&LN_NODEREFERENCE)==0)) {
67                 char* srcdir_name;
68
69                 srcdir_name = xstrdup(link_destname);
70                 strcat(src_name, "/");
71                 strcat(src_name, get_last_path_component(srcdir_name));
72                 free(srcdir_name);
73         }
74         
75         if (flag&LN_FORCE)
76                 unlink(src_name);
77                 
78         if (flag&LN_SYMLINK)
79                 status = symlink(link_destname, src_name);
80         else
81                 status = link(link_destname, src_name);
82
83         if (status != 0) {
84                 perror_msg(src_name);
85                 return(FALSE);
86         }
87         return(TRUE);
88 }
89
90 extern int ln_main(int argc, char **argv)
91 {
92         int status = EXIT_SUCCESS;
93         int flag = 0;
94         int opt;
95         
96         /* Parse any options */
97         while ((opt=getopt(argc, argv, "sfn")) != -1) {
98                 switch(opt) {
99                         case 's':
100                                 flag |= LN_SYMLINK;
101                                 break;
102                         case 'f':
103                                 flag |= LN_FORCE;
104                                 break;
105                         case 'n':
106                                 flag |= LN_NODEREFERENCE;
107                                 break;
108                         default:
109                                 show_usage();
110                 }
111         }
112         if (optind > (argc-1)) {
113                 show_usage();
114         } 
115         if (optind == (argc-1)) {
116                 if (fs_link(argv[optind], 
117                                         get_last_path_component(argv[optind]), flag)==FALSE)
118                         status = EXIT_FAILURE;
119         }
120         while(optind<(argc-1)) {
121                 if (fs_link(argv[optind], argv[argc-1], flag)==FALSE)
122                         status = EXIT_FAILURE;
123                 optind++;
124         }
125         exit(status);
126 }
127
128 /*
129 Local Variables:
130 c-file-style: "linux"
131 c-basic-offset: 4
132 tab-width: 4
133 End:
134 */