Annotation of embedaddon/sudo/compat/getcwd.c, revision 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>