path-sandbox/trunk/getcwd.c

/* These functions find the absolute path to the current working directory.  */

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <dirent.h>
#include <unistd.h>
#include <string.h>

#include "config.h"
#include <sys/param.h>
#include "localdecls.h"

/* Modified: 08 June 2005; Martin Schlemmer <azarah@gentoo.org>
 * 
 *  Cleaned up unneeded stuff.  Add a wrapper to try and detect when
 *  we have a kernel whose getcwd system call do not handle directory
 *  names longer than PATH_MAX, and if so, use our generic version.
 */

#ifndef __set_errno
# define __set_errno(val) errno = (val)
#endif

/* If the syscall is not present, we have to walk up the
 * directory tree till we hit the root.  Now we _could_
 * use /proc/self/cwd if /proc is mounted... That approach
 * is left an an exercise for the reader... */


/* Seems a few broken filesystems (like coda) don't like this */
/*#undef FAST_DIR_SEARCH_POSSIBLE on Linux */


/* Routine to find the step back down */
SB_STATIC char *search_dir(dev_t this_dev, ino_t this_ino, char *path_buf, size_t path_size)
{
        DIR *dp;
        struct dirent *d;
        char *ptr;
        size_t slen;
        struct stat st;

#ifdef FAST_DIR_SEARCH_POSSIBLE
        /* The test is for ELKS lib 0.0.9, this should be fixed in the real kernel */
        size_t slow_search = (sizeof(ino_t) != sizeof(d->d_ino));
#endif

//      if (stat(path_buf, &st) < 0) {
//              goto oops;
//      }
#ifdef FAST_DIR_SEARCH_POSSIBLE
        if (this_dev != st.st_dev)
                slow_search = 1;
#endif

        slen = strlen(path_buf);
        ptr = path_buf + slen - 1;
        if (*ptr != '/') {
                if (slen + 2 > path_size) {
                        goto oops;
                }
                snprintf(++ptr, 2, "/");
                slen++;
        }
        slen++;

#ifdef OUTSIDE_LIBSANDBOX
        dp = opendir(path_buf);
#else
        check_dlsym(opendir);
        dp = true_opendir(path_buf);
#endif
        if (dp == 0) {
            goto oops;
        }

        while ((d = readdir(dp)) != 0) {
#ifdef FAST_DIR_SEARCH_POSSIBLE
                if (slow_search || this_ino == d->d_ino) {
#endif
                        if (slen + strlen(d->d_name) > path_size) {
                            goto oops;
                        }
                        snprintf(ptr + 1, sizeof(d->d_name) + 1, "%s", d->d_name);
                        if (lstat(path_buf, &st) < 0)
                                continue;
                        if (st.st_ino == this_ino && st.st_dev == this_dev) {
                                closedir(dp);
                                return path_buf;
                        }
#ifdef FAST_DIR_SEARCH_POSSIBLE
                }
#endif
        }

        closedir(dp);
        return 0;

oops:
        __set_errno(ERANGE);
        return 0;
}

/* Routine to go up tree */
SB_STATIC char *recurser(char *path_buf, size_t path_size, dev_t root_dev, ino_t root_ino)
{
        struct stat st;
        dev_t this_dev;
        ino_t this_ino;

        if (lstat(path_buf, &st) < 0) {
            if (errno != EFAULT)
                goto oops;
            return 0;
        }
        this_dev = st.st_dev;
        this_ino = st.st_ino;
        if (this_dev == root_dev && this_ino == root_ino) {
                if (path_size < 2) {
                    goto oops;
                }
                snprintf(path_buf, 2, "/");
                return path_buf;
        }
        if (strlen(path_buf) + 4 > path_size) {
            goto oops;
        }
        snprintf(path_buf, 4, "/..");
        if (recurser(path_buf, path_size, root_dev, root_ino) == 0)
                return 0;

        return search_dir(this_dev, this_ino, path_buf, path_size);
oops:
        __set_errno(ERANGE);
        return 0;
}

SB_STATIC inline
size_t __syscall_egetcwd(char * buf, unsigned long size)
{
    size_t len;
    char *cwd;
    struct stat st;
    size_t olderrno;

    olderrno = errno;
    len = -1;
    cwd = recurser(buf, size, st.st_dev, st.st_ino);
    if (cwd) {
        len = strlen(buf);
        __set_errno(olderrno);
    }
    return len;
}

SB_STATIC char *__egetcwd(char *buf, size_t size)
{
    size_t ret;
    char *path;
    size_t alloc_size = size;

    if (size == 0) {
        if (buf != NULL) {
            __set_errno(EINVAL);
            return NULL;
        }
        alloc_size = SB_PATH_MAX;
    }
    path=buf;
    if (buf == NULL) {
        path = malloc(alloc_size);
        if (path == NULL)
            return NULL;
    }
    ret = __syscall_egetcwd(path, alloc_size);
    if (ret >= 0)
    {
        if (buf == NULL && size == 0)
            buf = realloc(path, ret);
        if (buf == NULL)
            buf = path;
        return buf;
    }
    if (buf == NULL)
        free (path);
    return NULL;
}

SB_STATIC char *egetcwd(char *buf, size_t size)
{
        struct stat st;
        char *tmpbuf;

        __set_errno(0);
        tmpbuf = getcwd(buf, size);

        if (tmpbuf) {
                lstat(buf, &st);
        } else {
                return tmpbuf;
        }

        if (errno) {
                /* If lstat() failed with eerror = ENOENT, then its
                 * possible that we are running on an older kernel,
                 * so use our generic version which *should* not fail.
                 */
                if (errno == ENOENT) {
                        return __egetcwd(buf, size);
                } else {
                        return tmpbuf;
                }
        }

        return tmpbuf;
}
1	/* These functions find the absolute path to the current working directory. */
2
3	#include <stdio.h>
4	#include <stdlib.h>
5	#include <errno.h>
6	#include <sys/types.h>
7	#include <sys/stat.h>
8	#include <dirent.h>
9	#include <unistd.h>
10	#include <string.h>
11
12	#include "config.h"
13	#include <sys/param.h>
14	#include "localdecls.h"
15
16	/* Modified: 08 June 2005; Martin Schlemmer <azarah@gentoo.org>
17	*
18	* Cleaned up unneeded stuff. Add a wrapper to try and detect when
19	* we have a kernel whose getcwd system call do not handle directory
20	* names longer than PATH_MAX, and if so, use our generic version.
21	*/
22
23	#ifndef __set_errno
24	# define __set_errno(val) errno = (val)
25	#endif
26
27	/* If the syscall is not present, we have to walk up the
28	* directory tree till we hit the root. Now we _could_
29	* use /proc/self/cwd if /proc is mounted... That approach
30	* is left an an exercise for the reader... */
31
32
33	/* Seems a few broken filesystems (like coda) don't like this */
34	/#undef FAST_DIR_SEARCH_POSSIBLE on Linux /
35
36
37	/* Routine to find the step back down */
38	SB_STATIC char search_dir(dev_t this_dev, ino_t this_ino, char path_buf, size_t path_size)
39	{
40	DIR *dp;
41	struct dirent *d;
42	char *ptr;
43	size_t slen;
44	struct stat st;
45
46	#ifdef FAST_DIR_SEARCH_POSSIBLE
47	/* The test is for ELKS lib 0.0.9, this should be fixed in the real kernel */
48	size_t slow_search = (sizeof(ino_t) != sizeof(d->d_ino));
49	#endif
50
51	// if (stat(path_buf, &st) < 0) {
52	// goto oops;
53	// }
54	#ifdef FAST_DIR_SEARCH_POSSIBLE
55	if (this_dev != st.st_dev)
56	slow_search = 1;
57	#endif
58
59	slen = strlen(path_buf);
60	ptr = path_buf + slen - 1;
61	if (*ptr != '/') {
62	if (slen + 2 > path_size) {
63	goto oops;
64	}
65	snprintf(++ptr, 2, "/");
66	slen++;
67	}
68	slen++;
69
70	#ifdef OUTSIDE_LIBSANDBOX
71	dp = opendir(path_buf);
72	#else
73	check_dlsym(opendir);
74	dp = true_opendir(path_buf);
75	#endif
76	if (dp == 0) {
77	goto oops;
78	}
79
80	while ((d = readdir(dp)) != 0) {
81	#ifdef FAST_DIR_SEARCH_POSSIBLE
82	if (slow_search \|\| this_ino == d->d_ino) {
83	#endif
84	if (slen + strlen(d->d_name) > path_size) {
85	goto oops;
86	}
87	snprintf(ptr + 1, sizeof(d->d_name) + 1, "%s", d->d_name);
88	if (lstat(path_buf, &st) < 0)
89	continue;
90	if (st.st_ino == this_ino && st.st_dev == this_dev) {
91	closedir(dp);
92	return path_buf;
93	}
94	#ifdef FAST_DIR_SEARCH_POSSIBLE
95	}
96	#endif
97	}
98
99	closedir(dp);
100	return 0;
101
102	oops:
103	__set_errno(ERANGE);
104	return 0;
105	}
106
107	/* Routine to go up tree */
108	SB_STATIC char recurser(char path_buf, size_t path_size, dev_t root_dev, ino_t root_ino)
109	{
110	struct stat st;
111	dev_t this_dev;
112	ino_t this_ino;
113
114	if (lstat(path_buf, &st) < 0) {
115	if (errno != EFAULT)
116	goto oops;
117	return 0;
118	}
119	this_dev = st.st_dev;
120	this_ino = st.st_ino;
121	if (this_dev == root_dev && this_ino == root_ino) {
122	if (path_size < 2) {
123	goto oops;
124	}
125	snprintf(path_buf, 2, "/");
126	return path_buf;
127	}
128	if (strlen(path_buf) + 4 > path_size) {
129	goto oops;
130	}
131	snprintf(path_buf, 4, "/..");
132	if (recurser(path_buf, path_size, root_dev, root_ino) == 0)
133	return 0;
134
135	return search_dir(this_dev, this_ino, path_buf, path_size);
136	oops:
137	__set_errno(ERANGE);
138	return 0;
139	}
140
141	SB_STATIC inline
142	size_t __syscall_egetcwd(char * buf, unsigned long size)
143	{
144	size_t len;
145	char *cwd;
146	struct stat st;
147	size_t olderrno;
148
149	olderrno = errno;
150	len = -1;
151	cwd = recurser(buf, size, st.st_dev, st.st_ino);
152	if (cwd) {
153	len = strlen(buf);
154	__set_errno(olderrno);
155	}
156	return len;
157	}
158
159	SB_STATIC char __egetcwd(char buf, size_t size)
160	{
161	size_t ret;
162	char *path;
163	size_t alloc_size = size;
164
165	if (size == 0) {
166	if (buf != NULL) {
167	__set_errno(EINVAL);
168	return NULL;
169	}
170	alloc_size = SB_PATH_MAX;
171	}
172	path=buf;
173	if (buf == NULL) {
174	path = malloc(alloc_size);
175	if (path == NULL)
176	return NULL;
177	}
178	ret = __syscall_egetcwd(path, alloc_size);
179	if (ret >= 0)
180	{
181	if (buf == NULL && size == 0)
182	buf = realloc(path, ret);
183	if (buf == NULL)
184	buf = path;
185	return buf;
186	}
187	if (buf == NULL)
188	free (path);
189	return NULL;
190	}
191
192	SB_STATIC char egetcwd(char buf, size_t size)
193	{
194	struct stat st;
195	char *tmpbuf;
196
197	__set_errno(0);
198	tmpbuf = getcwd(buf, size);
199
200	if (tmpbuf) {
201	lstat(buf, &st);
202	} else {
203	return tmpbuf;
204	}
205
206	if (errno) {
207	/* If lstat() failed with eerror = ENOENT, then its
208	* possible that we are running on an older kernel,
209	* so use our generic version which should not fail.
210	*/
211	if (errno == ENOENT) {
212	return __egetcwd(buf, size);
213	} else {
214	return tmpbuf;
215	}
216	}
217
218	return tmpbuf;
219	}
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision