8396df67f47c0e4a72f7642899ace3bac675f7df
[dragonfly.git] / bin / sh / cd.c
1 /*-
2  * Copyright (c) 1991, 1993
3  *      The Regents of the University of California.  All rights reserved.
4  *
5  * This code is derived from software contributed to Berkeley by
6  * Kenneth Almquist.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. All advertising materials mentioning features or use of this software
17  *    must display the following acknowledgement:
18  *      This product includes software developed by the University of
19  *      California, Berkeley and its contributors.
20  * 4. Neither the name of the University nor the names of its contributors
21  *    may be used to endorse or promote products derived from this software
22  *    without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34  * SUCH DAMAGE.
35  *
36  * @(#)cd.c     8.2 (Berkeley) 5/4/95
37  * $FreeBSD: src/bin/sh/cd.c,v 1.48 2011/05/27 20:01:46 jilles Exp $
38  */
39
40 #include <sys/types.h>
41 #include <sys/stat.h>
42 #include <stdlib.h>
43 #include <string.h>
44 #include <unistd.h>
45 #include <errno.h>
46 #include <limits.h>
47
48 /*
49  * The cd and pwd commands.
50  */
51
52 #include "shell.h"
53 #include "var.h"
54 #include "nodes.h"      /* for jobs.h */
55 #include "jobs.h"
56 #include "options.h"
57 #include "output.h"
58 #include "memalloc.h"
59 #include "error.h"
60 #include "exec.h"
61 #include "redir.h"
62 #include "mystring.h"
63 #include "show.h"
64 #include "cd.h"
65
66 static int cdlogical(char *);
67 static int cdphysical(char *);
68 static int docd(char *, int, int);
69 static char *getcomponent(void);
70 static char *findcwd(char *);
71 static void updatepwd(char *);
72 static char *getpwd(void);
73 static char *getpwd2(void);
74
75 static char *curdir = NULL;     /* current working directory */
76 static char *prevdir;           /* previous working directory */
77 static char *cdcomppath;
78
79 int
80 cdcmd(int argc, char **argv)
81 {
82         const char *dest;
83         const char *path;
84         char *p;
85         struct stat statb;
86         int ch, phys, print = 0, getcwderr = 0;
87         int rc;
88         int errno1 = ENOENT;
89
90         optreset = 1; optind = 1; opterr = 0; /* initialize getopt */
91         phys = Pflag;
92         while ((ch = getopt(argc, argv, "eLP")) != -1) {
93                 switch (ch) {
94                 case 'e':
95                         getcwderr = 1;
96                         break;
97                 case 'L':
98                         phys = 0;
99                         break;
100                 case 'P':
101                         phys = 1;
102                         break;
103                 default:
104                         error("unknown option: -%c", optopt);
105                         break;
106                 }
107         }
108         argc -= optind;
109         argv += optind;
110
111         if (argc > 1)
112                 error("too many arguments");
113
114         if ((dest = *argv) == NULL && (dest = bltinlookup("HOME", 1)) == NULL)
115                 error("HOME not set");
116         if (*dest == '\0')
117                 dest = ".";
118         if (dest[0] == '-' && dest[1] == '\0') {
119                 dest = prevdir ? prevdir : curdir;
120                 if (dest)
121                         print = 1;
122                 else
123                         dest = ".";
124         }
125         if (dest[0] == '/' ||
126             (dest[0] == '.' && (dest[1] == '/' || dest[1] == '\0')) ||
127             (dest[0] == '.' && dest[1] == '.' && (dest[2] == '/' || dest[2] == '\0')) ||
128             (path = bltinlookup("CDPATH", 1)) == NULL)
129                 path = nullstr;
130         while ((p = padvance(&path, dest)) != NULL) {
131                 if (stat(p, &statb) >= 0 && S_ISDIR(statb.st_mode)) {
132                         if (!print) {
133                                 /*
134                                  * XXX - rethink
135                                  */
136                                 if (p[0] == '.' && p[1] == '/' && p[2] != '\0')
137                                         print = strcmp(p + 2, dest);
138                                 else
139                                         print = strcmp(p, dest);
140                         }
141                         rc = docd(p, print, phys);
142                         if (rc >= 0)
143                                 return getcwderr ? rc : 0;
144                         if (errno != ENOENT)
145                                 errno1 = errno;
146                 }
147         }
148         error("%s: %s", dest, strerror(errno1));
149         /*NOTREACHED*/
150         return 0;
151 }
152
153
154 /*
155  * Actually change the directory.  In an interactive shell, print the
156  * directory name if "print" is nonzero.
157  */
158 static int
159 docd(char *dest, int print, int phys)
160 {
161         int rc;
162
163         TRACE(("docd(\"%s\", %d, %d) called\n", dest, print, phys));
164
165         /* If logical cd fails, fall back to physical. */
166         if ((phys || (rc = cdlogical(dest)) < 0) && (rc = cdphysical(dest)) < 0)
167                 return (-1);
168
169         if (print && iflag && curdir)
170                 out1fmt("%s\n", curdir);
171
172         return (rc);
173 }
174
175 static int
176 cdlogical(char *dest)
177 {
178         char *p;
179         char *q;
180         char *component;
181         struct stat statb;
182         int first;
183         int badstat;
184
185         /*
186          *  Check each component of the path. If we find a symlink or
187          *  something we can't stat, clear curdir to force a getcwd()
188          *  next time we get the value of the current directory.
189          */
190         badstat = 0;
191         cdcomppath = stalloc(strlen(dest) + 1);
192         scopy(dest, cdcomppath);
193         STARTSTACKSTR(p);
194         if (*dest == '/') {
195                 STPUTC('/', p);
196                 cdcomppath++;
197         }
198         first = 1;
199         while ((q = getcomponent()) != NULL) {
200                 if (q[0] == '\0' || (q[0] == '.' && q[1] == '\0'))
201                         continue;
202                 if (! first)
203                         STPUTC('/', p);
204                 first = 0;
205                 component = q;
206                 STPUTS(q, p);
207                 if (equal(component, ".."))
208                         continue;
209                 STACKSTRNUL(p);
210                 if (lstat(stackblock(), &statb) < 0) {
211                         badstat = 1;
212                         break;
213                 }
214         }
215
216         INTOFF;
217         if ((p = findcwd(badstat ? NULL : dest)) == NULL || chdir(p) < 0) {
218                 INTON;
219                 return (-1);
220         }
221         updatepwd(p);
222         INTON;
223         return (0);
224 }
225
226 static int
227 cdphysical(char *dest)
228 {
229         char *p;
230         int rc = 0;
231
232         INTOFF;
233         if (chdir(dest) < 0) {
234                 INTON;
235                 return (-1);
236         }
237         p = findcwd(NULL);
238         if (p == NULL) {
239                 warning("warning: failed to get name of current directory");
240                 rc = 1;
241         }
242         updatepwd(p);
243         INTON;
244         return (rc);
245 }
246
247 /*
248  * Get the next component of the path name pointed to by cdcomppath.
249  * This routine overwrites the string pointed to by cdcomppath.
250  */
251 static char *
252 getcomponent(void)
253 {
254         char *p;
255         char *start;
256
257         if ((p = cdcomppath) == NULL)
258                 return NULL;
259         start = cdcomppath;
260         while (*p != '/' && *p != '\0')
261                 p++;
262         if (*p == '\0') {
263                 cdcomppath = NULL;
264         } else {
265                 *p++ = '\0';
266                 cdcomppath = p;
267         }
268         return start;
269 }
270
271
272 static char *
273 findcwd(char *dir)
274 {
275         char *new;
276         char *p;
277
278         /*
279          * If our argument is NULL, we don't know the current directory
280          * any more because we traversed a symbolic link or something
281          * we couldn't stat().
282          */
283         if (dir == NULL || curdir == NULL)
284                 return getpwd2();
285         cdcomppath = stalloc(strlen(dir) + 1);
286         scopy(dir, cdcomppath);
287         STARTSTACKSTR(new);
288         if (*dir != '/') {
289                 STPUTS(curdir, new);
290                 if (STTOPC(new) == '/')
291                         STUNPUTC(new);
292         }
293         while ((p = getcomponent()) != NULL) {
294                 if (equal(p, "..")) {
295                         while (new > stackblock() && (STUNPUTC(new), *new) != '/');
296                 } else if (*p != '\0' && ! equal(p, ".")) {
297                         STPUTC('/', new);
298                         STPUTS(p, new);
299                 }
300         }
301         if (new == stackblock())
302                 STPUTC('/', new);
303         STACKSTRNUL(new);
304         return stackblock();
305 }
306
307 /*
308  * Update curdir (the name of the current directory) in response to a
309  * cd command.  We also call hashcd to let the routines in exec.c know
310  * that the current directory has changed.
311  */
312 static void
313 updatepwd(char *dir)
314 {
315         hashcd();                               /* update command hash table */
316
317         if (prevdir)
318                 ckfree(prevdir);
319         prevdir = curdir;
320         curdir = dir ? savestr(dir) : NULL;
321         setvar("PWD", curdir, VEXPORT);
322         setvar("OLDPWD", prevdir, VEXPORT);
323 }
324
325 int
326 pwdcmd(int argc, char **argv)
327 {
328         char *p;
329         int ch, phys;
330
331         optreset = 1; optind = 1; opterr = 0; /* initialize getopt */
332         phys = Pflag;
333         while ((ch = getopt(argc, argv, "LP")) != -1) {
334                 switch (ch) {
335                 case 'L':
336                         phys = 0;
337                         break;
338                 case 'P':
339                         phys = 1;
340                         break;
341                 default:
342                         error("unknown option: -%c", optopt);
343                         break;
344                 }
345         }
346         argc -= optind;
347         argv += optind;
348
349         if (argc != 0)
350                 error("too many arguments");
351
352         if (!phys && getpwd()) {
353                 out1str(curdir);
354                 out1c('\n');
355         } else {
356                 if ((p = getpwd2()) == NULL)
357                         error(".: %s", strerror(errno));
358                 out1str(p);
359                 out1c('\n');
360         }
361
362         return 0;
363 }
364
365 /*
366  * Get the current directory and cache the result in curdir.
367  */
368 static char *
369 getpwd(void)
370 {
371         char *p;
372
373         if (curdir)
374                 return curdir;
375
376         p = getpwd2();
377         if (p != NULL)
378                 curdir = savestr(p);
379
380         return curdir;
381 }
382
383 #define MAXPWD 256
384
385 /*
386  * Return the current directory.
387  */
388 static char *
389 getpwd2(void)
390 {
391         char *pwd;
392         int i;
393
394         for (i = MAXPWD;; i *= 2) {
395                 pwd = stalloc(i);
396                 if (getcwd(pwd, i) != NULL)
397                         return pwd;
398                 stunalloc(pwd);
399                 if (errno != ERANGE)
400                         break;
401         }
402
403         return NULL;
404 }
405
406 /*
407  * Initialize PWD in a new shell.
408  * If the shell is interactive, we need to warn if this fails.
409  */
410 void
411 pwd_init(int warn)
412 {
413         char *pwd;
414         struct stat stdot, stpwd;
415
416         pwd = lookupvar("PWD");
417         if (pwd && *pwd == '/' && stat(".", &stdot) != -1 &&
418             stat(pwd, &stpwd) != -1 &&
419             stdot.st_dev == stpwd.st_dev &&
420             stdot.st_ino == stpwd.st_ino) {
421                 if (curdir)
422                         ckfree(curdir);
423                 curdir = savestr(pwd);
424         }
425         if (getpwd() == NULL && warn)
426                 out2fmt_flush("sh: cannot determine working directory\n");
427         setvar("PWD", curdir, VEXPORT);
428 }