1/* $NetBSD: ln.c,v 1.40 2018/08/26 23:01:06 sevan Exp $ */
2
3/*-
4 * Copyright (c) 1987, 1993, 1994
5 * The Regents of the University of California. All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the University nor the names of its contributors
16 * may be used to endorse or promote products derived from this software
17 * without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32#if 0
33#ifndef lint
34static char const copyright[] =
35"@(#) Copyright (c) 1987, 1993, 1994\n\
36 The Regents of the University of California. All rights reserved.\n";
37#endif /* not lint */
38
39#ifndef lint
40static char sccsid[] = "@(#)ln.c 8.2 (Berkeley) 3/31/94";
41#endif /* not lint */
42#endif
43#include <sys/cdefs.h>
44#ifdef __FBSDID
45__FBSDID("$FreeBSD: head/bin/ln/ln.c 251261 2013-06-02 17:55:00Z eadler $");
46#endif
47__RCSID("$NetBSD: ln.c,v 1.40 2018/08/26 23:01:06 sevan Exp $");
48
49#include <sys/param.h>
50#include <sys/stat.h>
51
52#include <err.h>
53#include <errno.h>
54#include <fcntl.h>
55#include <libgen.h>
56#include <limits.h>
57#include <stdio.h>
58#include <stdlib.h>
59#include <string.h>
60#include <unistd.h>
61
62static int fflag; /* Unlink existing files. */
63static int Fflag; /* Remove empty directories also. */
64static int hflag; /* Check new name for symlink first. */
65static int iflag; /* Interactive mode. */
66static int Pflag; /* Create hard links to symlinks. */
67static int sflag; /* Symbolic, not hard, link. */
68static int vflag; /* Verbose output. */
69static int wflag; /* Warn if symlink target does not
70 * exist, and -f is not enabled. */
71static char linkch;
72
73static int linkit(const char *, const char *, int);
74static __dead void usage(void);
75
76int
77main(int argc, char *argv[])
78{
79 struct stat sb;
80 char *p, *targetdir;
81 int ch, exitval;
82
83 /*
84 * Test for the special case where the utility is called as
85 * "link", for which the functionality provided is greatly
86 * simplified.
87 */
88 if ((p = strrchr(argv[0], '/')) == NULL)
89 p = argv[0];
90 else
91 ++p;
92 if (strcmp(p, "link") == 0) {
93 while (getopt(argc, argv, "") != -1)
94 usage();
95 argc -= optind;
96 argv += optind;
97 if (argc != 2)
98 usage();
99 if (link(argv[0], argv[1]) == -1)
100 err(EXIT_FAILURE, NULL);
101 exit(EXIT_SUCCESS);
102 }
103
104 while ((ch = getopt(argc, argv, "FLPfhinsvw")) != -1)
105 switch (ch) {
106 case 'F':
107 Fflag = 1;
108 break;
109 case 'L':
110 Pflag = 0;
111 break;
112 case 'P':
113 Pflag = 1;
114 break;
115 case 'f':
116 fflag = 1;
117 iflag = 0;
118 wflag = 0;
119 break;
120 case 'h':
121 case 'n':
122 hflag = 1;
123 break;
124 case 'i':
125 iflag = 1;
126 fflag = 0;
127 break;
128 case 's':
129 sflag = 1;
130 break;
131 case 'v':
132 vflag = 1;
133 break;
134 case 'w':
135 wflag = 1;
136 break;
137 case '?':
138 default:
139 usage();
140 }
141
142 argv += optind;
143 argc -= optind;
144
145 linkch = sflag ? '-' : '=';
146 if (sflag == 0)
147 Fflag = 0;
148 if (Fflag == 1 && iflag == 0) {
149 fflag = 1;
150 wflag = 0; /* Implied when fflag != 0 */
151 }
152
153 switch(argc) {
154 case 0:
155 usage();
156 /* NOTREACHED */
157 case 1: /* ln source */
158 exit(linkit(argv[0], ".", 1));
159 case 2: /* ln source target */
160 exit(linkit(argv[0], argv[1], 0));
161 default:
162 ;
163 }
164 /* ln source1 source2 directory */
165 targetdir = argv[argc - 1];
166 if (hflag && lstat(targetdir, &sb) == 0 && S_ISLNK(sb.st_mode)) {
167 /*
168 * We were asked not to follow symlinks, but found one at
169 * the target--simulate "not a directory" error
170 */
171 errno = ENOTDIR;
172 err(1, "%s", targetdir);
173 }
174 if (stat(targetdir, &sb))
175 err(1, "%s", targetdir);
176 if (!S_ISDIR(sb.st_mode))
177 usage();
178 for (exitval = 0; *argv != targetdir; ++argv)
179 exitval |= linkit(*argv, targetdir, 1);
180 exit(exitval);
181}
182
183/*
184 * Two pathnames refer to the same directory entry if the directories match
185 * and the final components' names match.
186 */
187static int
188samedirent(const char *path1, const char *path2)
189{
190 const char *file1, *file2;
191 char pathbuf[PATH_MAX];
192 struct stat sb1, sb2;
193
194 if (strcmp(path1, path2) == 0)
195 return 1;
196 file1 = strrchr(path1, '/');
197 if (file1 != NULL)
198 file1++;
199 else
200 file1 = path1;
201 file2 = strrchr(path2, '/');
202 if (file2 != NULL)
203 file2++;
204 else
205 file2 = path2;
206 if (strcmp(file1, file2) != 0)
207 return 0;
208 if (file1 - path1 >= PATH_MAX || file2 - path2 >= PATH_MAX)
209 return 0;
210 if (file1 == path1)
211 memcpy(pathbuf, ".", 2);
212 else {
213 memcpy(pathbuf, path1, file1 - path1);
214 pathbuf[file1 - path1] = '\0';
215 }
216 if (stat(pathbuf, &sb1) != 0)
217 return 0;
218 if (file2 == path2)
219 memcpy(pathbuf, ".", 2);
220 else {
221 memcpy(pathbuf, path2, file2 - path2);
222 pathbuf[file2 - path2] = '\0';
223 }
224 if (stat(pathbuf, &sb2) != 0)
225 return 0;
226 return sb1.st_dev == sb2.st_dev && sb1.st_ino == sb2.st_ino;
227}
228
229static int
230linkit(const char *source, const char *target, int isdir)
231{
232 struct stat sb;
233 const char *p;
234 int ch, exists, first;
235 char path[PATH_MAX];
236 char wbuf[PATH_MAX];
237 char bbuf[PATH_MAX];
238
239 if (!sflag) {
240 /* If source doesn't exist, quit now. */
241 if ((Pflag ? lstat : stat)(source, &sb)) {
242 warn("%s", source);
243 return (1);
244 }
245 /* Only symbolic links to directories. */
246 if (S_ISDIR(sb.st_mode)) {
247 errno = EISDIR;
248 warn("%s", source);
249 return (1);
250 }
251 }
252
253 /*
254 * If the target is a directory (and not a symlink if hflag),
255 * append the source's name.
256 */
257 if (isdir ||
258 (lstat(target, &sb) == 0 && S_ISDIR(sb.st_mode)) ||
259 (!hflag && stat(target, &sb) == 0 && S_ISDIR(sb.st_mode))) {
260 if (strlcpy(bbuf, source, sizeof(bbuf)) >= sizeof(bbuf) ||
261 (p = basename(bbuf)) == NULL ||
262 snprintf(path, sizeof(path), "%s/%s", target, p) >=
263 (ssize_t)sizeof(path)) {
264 errno = ENAMETOOLONG;
265 warn("%s", source);
266 return (1);
267 }
268 target = path;
269 }
270
271 /*
272 * If the link source doesn't exist, and a symbolic link was
273 * requested, and -w was specified, give a warning.
274 */
275 if (sflag && wflag) {
276 if (*source == '/') {
277 /* Absolute link source. */
278 if (stat(source, &sb) != 0)
279 warn("warning: %s inaccessible", source);
280 } else {
281 /*
282 * Relative symlink source. Try to construct the
283 * absolute path of the source, by appending `source'
284 * to the parent directory of the target.
285 */
286 strlcpy(bbuf, target, sizeof(bbuf));
287 p = dirname(bbuf);
288 if (p != NULL) {
289 (void)snprintf(wbuf, sizeof(wbuf), "%s/%s",
290 p, source);
291 if (stat(wbuf, &sb) != 0)
292 warn("warning: %s", source);
293 }
294 }
295 }
296
297 /*
298 * If the file exists, first check it is not the same directory entry.
299 */
300 exists = !lstat(target, &sb);
301 if (exists) {
302 if (!sflag && samedirent(source, target)) {
303 warnx("%s and %s are the same directory entry",
304 source, target);
305 return (1);
306 }
307 }
308 /*
309 * Then unlink it forcibly if -f was specified
310 * and interactively if -i was specified.
311 */
312 if (fflag && exists) {
313 if (Fflag && S_ISDIR(sb.st_mode)) {
314 if (rmdir(target)) {
315 warn("%s", target);
316 return (1);
317 }
318 } else if (unlink(target)) {
319 warn("%s", target);
320 return (1);
321 }
322 } else if (iflag && exists) {
323 fflush(stdout);
324 fprintf(stderr, "replace %s? ", target);
325
326 first = ch = getchar();
327 while(ch != '\n' && ch != EOF)
328 ch = getchar();
329 if (first != 'y' && first != 'Y') {
330 fprintf(stderr, "not replaced\n");
331 return (1);
332 }
333
334 if (Fflag && S_ISDIR(sb.st_mode)) {
335 if (rmdir(target)) {
336 warn("%s", target);
337 return (1);
338 }
339 } else if (unlink(target)) {
340 warn("%s", target);
341 return (1);
342 }
343 }
344
345 /* Attempt the link. */
346 if (sflag ? symlink(source, target) :
347 linkat(AT_FDCWD, source, AT_FDCWD, target,
348 Pflag ? 0 : AT_SYMLINK_FOLLOW)) {
349 warn("%s", target);
350 return (1);
351 }
352 if (vflag)
353 (void)printf("%s %c> %s\n", target, linkch, source);
354 return (0);
355}
356
357static __dead void
358usage(void)
359{
360 (void)fprintf(stderr,
361 "usage: %s [-L | -P | -s [-F]] [-f | -iw] [-hnv] source_file [target_file]\n"
362 " %s [-L | -P | -s [-F]] [-f | -iw] [-hnv] source_file ... target_dir\n"
363 " link source_file target_file\n", getprogname(), getprogname());
364 exit(1);
365}
366