Annotation of embedaddon/sudo/compat/getcwd.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * Copyright (c) 1989, 1991, 1993
                      3:  *     The Regents of the University of California.  All rights reserved.
                      4:  *
                      5:  * Redistribution and use in source and binary forms, with or without
                      6:  * modification, are permitted provided that the following conditions
                      7:  * are met:
                      8:  * 1. Redistributions of source code must retain the above copyright
                      9:  *    notice, this list of conditions and the following disclaimer.
                     10:  * 2. Redistributions in binary form must reproduce the above copyright
                     11:  *    notice, this list of conditions and the following disclaimer in the
                     12:  *    documentation and/or other materials provided with the distribution.
                     13:  * 3. Neither the name of the University nor the names of its contributors
                     14:  *    may be used to endorse or promote products derived from this software
                     15:  *    without specific prior written permission.
                     16:  *
                     17:  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
                     18:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
                     19:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
                     20:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
                     21:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
                     22:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
                     23:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
                     24:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
                     25:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     26:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     27:  * SUCH DAMAGE.
                     28:  */
                     29: 
                     30: #include <config.h>
                     31: 
                     32: #include <sys/param.h>
                     33: #include <sys/stat.h>
                     34: 
                     35: #include <errno.h>
                     36: #include <stdio.h>
                     37: #ifdef STDC_HEADERS
                     38: # include <stdlib.h>
                     39: # include <stddef.h>
                     40: #else
                     41: # ifdef HAVE_STDLIB_H
                     42: #  include <stdlib.h>
                     43: # endif
                     44: #endif /* STDC_HEADERS */
                     45: #ifdef HAVE_STRING_H
                     46: # include <string.h>
                     47: #endif /* HAVE_STRING_H */
                     48: #ifdef HAVE_STRINGS_H
                     49: # include <strings.h>
                     50: #endif /* HAVE_STRINGS_H */
                     51: #if defined(HAVE_MALLOC_H) && !defined(STDC_HEADERS)
                     52: # include <malloc.h>
                     53: #endif /* HAVE_MALLOC_H && !STDC_HEADERS */
                     54: #ifdef HAVE_UNISTD_H
                     55: # include <unistd.h>
                     56: #endif /* HAVE_UNISTD_H */
                     57: #ifdef HAVE_DIRENT_H
                     58: # include <dirent.h>
                     59: # define NAMLEN(dirent) strlen((dirent)->d_name)
                     60: #else
                     61: # define dirent direct
                     62: # define NAMLEN(dirent) (dirent)->d_namlen
                     63: # ifdef HAVE_SYS_NDIR_H
                     64: #  include <sys/ndir.h>
                     65: # endif
                     66: # ifdef HAVE_SYS_DIR_H
                     67: #  include <sys/dir.h>
                     68: # endif
                     69: # ifdef HAVE_NDIR_H
                     70: #  include <ndir.h>
                     71: # endif
                     72: #endif
                     73: 
                     74: #include "missing.h"
                     75: 
                     76: #define        ISDOT(dp) \
                     77:        (dp->d_name[0] == '.' && (dp->d_name[1] == '\0' || \
                     78:            (dp->d_name[1] == '.' && dp->d_name[2] == '\0')))
                     79: 
                     80: char *
                     81: getcwd(char *pt, size_t size)
                     82: {
                     83:        struct dirent *dp;
                     84:        DIR *dir = NULL;
                     85:        dev_t dev;
                     86:        ino_t ino;
                     87:        int first;
                     88:        char *bpt, *bup;
                     89:        struct stat s;
                     90:        dev_t root_dev;
                     91:        ino_t root_ino;
                     92:        size_t ptsize, upsize;
                     93:        int save_errno;
                     94:        char *ept, *eup, *up;
                     95: 
                     96:        /*
                     97:         * If no buffer specified by the user, allocate one as necessary.
                     98:         * If a buffer is specified, the size has to be non-zero.  The path
                     99:         * is built from the end of the buffer backwards.
                    100:         */
                    101:        if (pt) {
                    102:                ptsize = 0;
                    103:                if (!size) {
                    104:                        errno = EINVAL;
                    105:                        return NULL;
                    106:                }
                    107:                ept = pt + size;
                    108:        } else {
                    109:                if ((pt = malloc(ptsize = 1024 - 4)) == NULL)
                    110:                        return NULL;
                    111:                ept = pt + ptsize;
                    112:        }
                    113:        bpt = ept - 1;
                    114:        *bpt = '\0';
                    115: 
                    116:        /*
                    117:         * Allocate bytes (1024 - malloc space) for the string of "../"'s.
                    118:         * Should always be enough (it's 340 levels).  If it's not, allocate
                    119:         * as necessary.  Special * case the first stat, it's ".", not "..".
                    120:         */
                    121:        if ((up = malloc(upsize = 1024 - 4)) == NULL)
                    122:                goto err;
                    123:        eup = up + PATH_MAX;
                    124:        bup = up;
                    125:        up[0] = '.';
                    126:        up[1] = '\0';
                    127: 
                    128:        /* Save root values, so know when to stop. */
                    129:        if (stat("/", &s))
                    130:                goto err;
                    131:        root_dev = s.st_dev;
                    132:        root_ino = s.st_ino;
                    133: 
                    134:        errno = 0;                      /* XXX readdir has no error return. */
                    135: 
                    136:        for (first = 1;; first = 0) {
                    137:                /* Stat the current level. */
                    138:                if (lstat(up, &s))
                    139:                        goto err;
                    140: 
                    141:                /* Save current node values. */
                    142:                ino = s.st_ino;
                    143:                dev = s.st_dev;
                    144: 
                    145:                /* Check for reaching root. */
                    146:                if (root_dev == dev && root_ino == ino) {
                    147:                        *--bpt = '/';
                    148:                        /*
                    149:                         * It's unclear that it's a requirement to copy the
                    150:                         * path to the beginning of the buffer, but it's always
                    151:                         * been that way and stuff would probably break.
                    152:                         */
                    153:                        bcopy(bpt, pt, ept - bpt);
                    154:                        free(up);
                    155:                        return pt;
                    156:                }
                    157: 
                    158:                /*
                    159:                 * Build pointer to the parent directory, allocating memory
                    160:                 * as necessary.  Max length is 3 for "../", the largest
                    161:                 * possible component name, plus a trailing NULL.
                    162:                 */
                    163:                if (bup + 3  + MAXNAMLEN + 1 >= eup) {
                    164:                        char *nup;
                    165: 
                    166:                        if ((nup = realloc(up, upsize *= 2)) == NULL)
                    167:                                goto err;
                    168:                        up = nup;
                    169:                        bup = up;
                    170:                        eup = up + upsize;
                    171:                }
                    172:                *bup++ = '.';
                    173:                *bup++ = '.';
                    174:                *bup = '\0';
                    175: 
                    176:                /* Open and stat parent directory. */
                    177:                if (!(dir = opendir(up)) || fstat(dirfd(dir), &s))
                    178:                        goto err;
                    179: 
                    180:                /* Add trailing slash for next directory. */
                    181:                *bup++ = '/';
                    182: 
                    183:                /*
                    184:                 * If it's a mount point, have to stat each element because
                    185:                 * the inode number in the directory is for the entry in the
                    186:                 * parent directory, not the inode number of the mounted file.
                    187:                 */
                    188:                save_errno = 0;
                    189:                if (s.st_dev == dev) {
                    190:                        for (;;) {
                    191:                                if (!(dp = readdir(dir)))
                    192:                                        goto notfound;
                    193:                                if (dp->d_fileno == ino)
                    194:                                        break;
                    195:                        }
                    196:                } else
                    197:                        for (;;) {
                    198:                                if (!(dp = readdir(dir)))
                    199:                                        goto notfound;
                    200:                                if (ISDOT(dp))
                    201:                                        continue;
                    202:                                bcopy(dp->d_name, bup, NAMLEN(dp) + 1);
                    203: 
                    204:                                /* Save the first error for later. */
                    205:                                if (lstat(up, &s)) {
                    206:                                        if (!save_errno)
                    207:                                                save_errno = errno;
                    208:                                        errno = 0;
                    209:                                        continue;
                    210:                                }
                    211:                                if (s.st_dev == dev && s.st_ino == ino)
                    212:                                        break;
                    213:                        }
                    214: 
                    215:                /*
                    216:                 * Check for length of the current name, preceding slash,
                    217:                 * leading slash.
                    218:                 */
                    219:                if (bpt - pt <= NAMLEN(dp) + (first ? 1 : 2)) {
                    220:                        size_t len, off;
                    221:                        char *npt;
                    222: 
                    223:                        if (!ptsize) {
                    224:                                errno = ERANGE;
                    225:                                goto err;
                    226:                        }
                    227:                        off = bpt - pt;
                    228:                        len = ept - bpt;
                    229:                        if ((npt = realloc(pt, ptsize *= 2)) == NULL)
                    230:                                goto err;
                    231:                        pt = npt;
                    232:                        bpt = pt + off;
                    233:                        ept = pt + ptsize;
                    234:                        bcopy(bpt, ept - len, len);
                    235:                        bpt = ept - len;
                    236:                }
                    237:                if (!first)
                    238:                        *--bpt = '/';
                    239:                bpt -= NAMLEN(dp);
                    240:                bcopy(dp->d_name, bpt, NAMLEN(dp));
                    241:                (void)closedir(dir);
                    242: 
                    243:                /* Truncate any file name. */
                    244:                *bup = '\0';
                    245:        }
                    246: 
                    247: notfound:
                    248:        /*
                    249:         * If readdir set errno, use it, not any saved error; otherwise,
                    250:         * didn't find the current directory in its parent directory, set
                    251:         * errno to ENOENT.
                    252:         */
                    253:        if (!errno)
                    254:                errno = save_errno ? save_errno : ENOENT;
                    255:        /* FALLTHROUGH */
                    256: err:
                    257:        if (ptsize)
                    258:                free(pt);
                    259:        if (up)
                    260:                free(up);
                    261:        if (dir)
                    262:                (void)closedir(dir);
                    263:        return NULL;
                    264: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>