Use busybox error handling functions wherever possible.
[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 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 "busybox.h"
25 #define BB_DECLARE_EXTERN
26 #define bb_need_not_a_directory
27 #include "messages.c"
28
29 #include <stdio.h>
30 #include <dirent.h>
31 #include <errno.h>
32
33 #define LN_SYMLINK              1
34 #define LN_FORCE                        2
35 #define LN_NODEREFERENCE        4
36
37 /*
38  * linkDestName is where the link points to,
39  * linkSrcName is the name of the link to be created.
40  */
41 static int fs_link(const char *link_DestName, const char *link_SrcName, const int flag)
42 {
43         int status;
44         int srcIsDir;
45         char *srcName;
46
47         if (link_DestName==NULL)
48                 return(FALSE);
49
50         srcName = (char *) malloc(strlen(link_SrcName)+strlen(link_DestName)+1);
51
52         if (link_SrcName==NULL)
53                 strcpy(srcName, link_DestName);
54         else
55                 strcpy(srcName, link_SrcName);
56
57         if (flag&LN_NODEREFERENCE)
58                 srcIsDir = is_directory(srcName, TRUE, NULL);
59         else
60                 srcIsDir = is_directory(srcName, FALSE, NULL);  
61         
62         if ((srcIsDir==TRUE)&&((flag&LN_NODEREFERENCE)==0)) {
63                 strcat(srcName, "/");
64                 strcat(srcName, link_DestName);
65         }
66         
67         if (flag&LN_FORCE)
68                 unlink(srcName);
69                 
70         if (flag&LN_SYMLINK)
71                 status = symlink(link_DestName, srcName);
72         else
73                 status = link(link_DestName, srcName);
74
75         if (status != 0) {
76                 perror(srcName);
77                 return(FALSE);
78         }
79         return(TRUE);
80 }
81
82 extern int ln_main(int argc, char **argv)
83 {
84         int status = EXIT_SUCCESS;
85         int flag = 0;
86         int opt;
87         
88         /* Parse any options */
89         while ((opt=getopt(argc, argv, "sfn")) != -1) {
90                 switch(opt) {
91                         case 's':
92                                 flag |= LN_SYMLINK;
93                                 break;
94                         case 'f':
95                                 flag |= LN_FORCE;
96                                 break;
97                         case 'n':
98                                 flag |= LN_NODEREFERENCE;
99                                 break;
100                         default:
101                                 usage(ln_usage);
102                 }
103         }
104         while(optind<(argc-1)) {
105                 if (fs_link(argv[optind], argv[argc-1], flag)==FALSE)
106                         status = EXIT_FAILURE;
107                 optind++;
108         }
109         return(status);
110 }
111
112 /*
113 Local Variables:
114 c-file-style: "linux"
115 c-basic-offset: 4
116 tab-width: 4
117 End:
118 */