xref: /dragonfly/lib/libc/gen/opendir.c (revision 0720b42f)
1 /*
2  * Copyright (c) 1983, 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. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed by the University of
16  *	California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * $FreeBSD: src/lib/libc/gen/opendir.c,v 1.10.2.1 2001/06/04 20:59:48 joerg Exp $
34  *
35  * @(#)opendir.c	8.8 (Berkeley) 5/1/95
36  */
37 
38 #include "namespace.h"
39 #include <sys/param.h>
40 #include <sys/stat.h>
41 
42 #include <dirent.h>
43 #include <errno.h>
44 #include <fcntl.h>
45 #include <stdlib.h>
46 #include <string.h>
47 #include <unistd.h>
48 #include "un-namespace.h"
49 
50 #include "gen_private.h"
51 
52 #define DEFAULT_FLAGS (DTF_HIDEW | DTF_NODUP)
53 
54 /*
55  * Open a directory given its path.
56  */
57 DIR *
58 opendir(const char *name)
59 {
60 	return (__opendir2(name, DEFAULT_FLAGS));
61 }
62 
63 /*
64  * Open a directory given a descriptor representing it.
65  */
66 DIR *
67 fdopendir(int fd)
68 {
69 	return (__fdopendir2(fd, DEFAULT_FLAGS));
70 }
71 
72 DIR *
73 __opendir2(const char *name, int flags)
74 {
75 	int fd;
76 	struct stat statb;
77 	DIR *dirp;
78 	int saved_errno;
79 
80 	/*
81 	 * stat() before _open() because opening of special files may be
82 	 * harmful.
83 	 */
84 	if (stat(name, &statb) != 0)
85 		return (NULL);
86 
87 	fd = _open(name, O_RDONLY | O_NONBLOCK | O_DIRECTORY | O_CLOEXEC);
88 	if (fd == -1)
89 		return (NULL);
90 	dirp = __fdopendir2(fd, flags);
91 	if (dirp == NULL) {
92 		saved_errno = errno;
93 		_close(fd);
94 		errno = saved_errno;
95 	}
96 
97 	return (dirp);
98 }
99 
100 DIR *
101 __fdopendir2(int fd, int flags)
102 {
103 	DIR *dirp;
104 	int incr;
105 	int saved_errno;
106 	struct stat statb;
107 
108 	dirp = NULL;
109 
110 	if (_fstat(fd, &statb) != 0)
111 		goto fail;
112 	if (!S_ISDIR(statb.st_mode)) {
113 		errno = ENOTDIR;
114 		goto fail;
115 	}
116 	if (_fcntl(fd, F_SETFD, FD_CLOEXEC) == -1 ||
117 	    (dirp = malloc(sizeof(DIR))) == NULL)
118 		goto fail;
119 
120 	/*
121 	 * Use the system page size if that is a multiple of DIRBLKSIZ.
122 	 * Hopefully this can be a big win someday by allowing page
123 	 * trades to user space to be done by _getdirentries().
124 	 */
125 	incr = getpagesize();
126 	if ((incr % DIRBLKSIZ) != 0)
127 		incr = DIRBLKSIZ;
128 
129 	dirp->dd_len = incr;
130 	dirp->dd_buf = malloc(dirp->dd_len);
131 	if (dirp->dd_buf == NULL)
132 		goto fail;
133 	flags &= ~DTF_REWIND;
134 
135 	dirp->dd_loc = 0;
136 	dirp->dd_fd = fd;
137 	dirp->dd_flags = flags;
138 	dirp->dd_lock = NULL;
139 
140 	/*
141 	 * Set up seek point for rewinddir.
142 	 */
143 	dirp->dd_seek = 0;
144 	dirp->dd_rewind = telldir(dirp);
145 
146 	/*
147 	 * The file offset of the fd passed to fdopendir() determines the
148 	 * initial entry returned by readdir().  Save this offset so that
149 	 * telldir() right after fdopendir() returns a value pointing to this
150 	 * initial entry.
151 	 * We don't have to worry about misaligned file offsets.  The kernel
152 	 * deals with these.
153 	 */
154 	if ((dirp->dd_seek = lseek(fd, 0, SEEK_CUR)) < 0)
155 		goto fail;
156 
157 	return (dirp);
158 
159 fail:
160 	saved_errno = errno;
161 	if (dirp != NULL) {
162 		_reclaim_telldir(dirp);
163 		free(dirp->dd_buf);
164 	}
165 	free(dirp);
166 	errno = saved_errno;
167 	return (NULL);
168 }
169