diskdev_cmds/dev_mkdb.tproj/dev_mkdb.c

   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 /******
  57 static char copyright[] =
  58 "@(#) Copyright (c) 1990, 1993\n\
  59         The Regents of the University of California.  All rights reserved.\n";
  60 ******/
  61
  62 /******
  63 static 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
  81 void    err __P((const char *, ...));
  82 void    usage __P((void));
  83
  84 int
  85 main(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
 168 void
 169 usage()
 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
 181 void
 182 #if __STDC__
 183 err(const char *fmt, ...)
 184 #else
 185 err(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 }