1/*
2 * Copyright (c) 1999 Apple Computer, Inc. All rights reserved.
3 *
4 * @APPLE_LICENSE_HEADER_START@
5 *
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. Please obtain a copy of the License at
10 * http://www.opensource.apple.com/apsl/ and read it before using this
11 * file.
12 *
13 * The Original Code and all software distributed under the License are
14 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
15 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
16 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
18 * Please see the License for the specific language governing rights and
19 * limitations under the License.
20 *
21 * @APPLE_LICENSE_HEADER_END@
22 */
23/*-
24 * Copyright (c) 1990, 1993
25 *	The Regents of the University of California.  All rights reserved.
26 *
27 * Redistribution and use in source and binary forms, with or without
28 * modification, are permitted provided that the following conditions
29 * are met:
30 * 1. Redistributions of source code must retain the above copyright
31 *    notice, this list of conditions and the following disclaimer.
32 * 2. Redistributions in binary form must reproduce the above copyright
33 *    notice, this list of conditions and the following disclaimer in the
34 *    documentation and/or other materials provided with the distribution.
35 * 3. All advertising materials mentioning features or use of this software
36 *    must display the following acknowledgement:
37 *	This product includes software developed by the University of
38 *	California, Berkeley and its contributors.
39 * 4. Neither the name of the University nor the names of its contributors
40 *    may be used to endorse or promote products derived from this software
41 *    without specific prior written permission.
42 *
43 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
44 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
45 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
46 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
47 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
48 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
49 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
50 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
51 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
52 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
53 * SUCH DAMAGE.
54 */
55
56/******
57static char copyright[] =
58"@(#) Copyright (c) 1990, 1993\n\
59	The Regents of the University of California.  All rights reserved.\n";
60******/
61
62/******
63static char sccsid[] = "@(#)dev_mkdb.c	8.1 (Berkeley) 6/6/93";
64******/
65
66#include <sys/param.h>
67#include <sys/stat.h>
68
69#include <fcntl.h>
70#undef DIRBLKSIZ
71#include <dirent.h>
72#include <nlist.h>
73#include <db.h>
74#include <errno.h>
75#include <unistd.h>
76#include <stdio.h>
77#include <paths.h>
78#include <stdlib.h>
79#include <string.h>
80
81void	err __P((const char *, ...));
82void	usage __P((void));
83
84int
85main(argc, argv)
86	int argc;
87	char *argv[];
88{
89	register DIR *dirp;
90	register struct dirent *dp;
91	struct stat sb;
92	struct {
93		mode_t type;
94		dev_t dev;
95	} bkey;
96	DB *db;
97	DBT data, key;
98	int ch;
99	u_char buf[MAXNAMLEN + 1];
100	char dbtmp[MAXPATHLEN + 1], dbname[MAXPATHLEN + 1];
101
102	while ((ch = getopt(argc, argv, "")) != EOF)
103		switch((char)ch) {
104		case '?':
105		default:
106			usage();
107		}
108	argc -= optind;
109	argv += optind;
110
111	if (argc > 0)
112		usage();
113
114	if (chdir(_PATH_DEV))
115		err("%s: %s", _PATH_DEV, strerror(errno));
116
117	dirp = opendir(".");
118
119	(void)snprintf(dbtmp, sizeof(dbtmp), "%sdev.tmp", _PATH_VARRUN);
120	(void)snprintf(dbname, sizeof(dbtmp), "%sdev.db", _PATH_VARRUN);
121	db = dbopen(dbtmp, O_CREAT|O_EXLOCK|O_RDWR|O_TRUNC,
122	    S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH, DB_HASH, NULL);
123	if (db == NULL)
124		err("%s: %s", dbtmp, strerror(errno));
125
126	/*
127	 * Keys are a mode_t followed by a dev_t.  The former is the type of
128	 * the file (mode & S_IFMT), the latter is the st_rdev field.  Note
129	 * that the structure may contain padding, so we have to clear it
130	 * out here.
131	 */
132	bzero(&bkey, sizeof(bkey));
133	key.data = &bkey;
134	key.size = sizeof(bkey);
135	data.data = buf;
136	while ((dp = readdir(dirp))) {
137		if (lstat(dp->d_name, &sb)) {
138			(void)fprintf(stderr,
139			    "dev_mkdb: %s: %s\n", dp->d_name, strerror(errno));
140			continue;
141		}
142
143		/* Create the key. */
144		if (S_ISCHR(sb.st_mode))
145			bkey.type = S_IFCHR;
146		else if (S_ISBLK(sb.st_mode))
147			bkey.type = S_IFBLK;
148		else
149			continue;
150		bkey.dev = sb.st_rdev;
151
152		/*
153		 * Create the data; nul terminate the name so caller doesn't
154		 * have to.
155		 */
156		bcopy(dp->d_name, buf, dp->d_namlen);
157		buf[dp->d_namlen] = '\0';
158		data.size = dp->d_namlen + 1;
159		if ((db->put)(db, &key, &data, 0))
160			err("dbput %s: %s\n", dbtmp, strerror(errno));
161	}
162	(void)(db->close)(db);
163	if (rename(dbtmp, dbname))
164		err("rename %s to %s: %s", dbtmp, dbname, strerror(errno));
165	exit(0);
166}
167
168void
169usage()
170{
171	(void)fprintf(stderr, "usage: dev_mkdb\n");
172	exit(1);
173}
174
175#if __STDC__
176#include <stdarg.h>
177#else
178#include <varargs.h>
179#endif
180
181void
182#if __STDC__
183err(const char *fmt, ...)
184#else
185err(fmt, va_alist)
186	char *fmt;
187        va_dcl
188#endif
189{
190	va_list ap;
191#if __STDC__
192	va_start(ap, fmt);
193#else
194	va_start(ap);
195#endif
196	(void)fprintf(stderr, "dev_mkdb: ");
197	(void)vfprintf(stderr, fmt, ap);
198	va_end(ap);
199	(void)fprintf(stderr, "\n");
200	exit(1);
201	/* NOTREACHED */
202}
203