ANSI

[unix-history] / usr / src / usr.sbin / dev_mkdb / dev_mkdb.c

/*-
* Copyright (c) 1990 The Regents of the University of California.
* All rights reserved.
*
* %sccs.include.redist.c%
*/

#ifndef lint
char copyright[] =
"@(#) Copyright (c) 1990 The Regents of the University of California.\n\
All rights reserved.\n";
#endif /* not lint */

#ifndef lint
static char sccsid[] = "@(#)dev_mkdb.c  5.3 (Berkeley) %G%";
#endif /* not lint */

#include <sys/param.h>
#include <sys/stat.h>
#include <sys/user.h>
#include <fcntl.h>
#include <ndbm.h>
#undef DIRBLKSIZ
#include <dirent.h>
#include <kvm.h>
#include <errno.h>
#include <stdio.h>
#include <paths.h>
#include <string.h>

main(argc, argv)
       int argc;
       char **argv;
{
       extern int optind;
       register DIR *dirp;
       register struct dirent *dp;
       struct stat sb;
       DBM *db;
       datum key, data;
       int ch;
       char buf[MAXNAMLEN + 1], dbtmp[MAXPATHLEN + 1], dbname[MAXPATHLEN + 1];

       while ((ch = getopt(argc, argv, "")) != EOF)
               switch((char)ch) {
               case '?':
               default:
                       usage();
               }
       argc -= optind;
       argv += optind;

       if (chdir(_PATH_DEV))
               error(_PATH_DEV);

       dirp = opendir(".");

       (void)sprintf(dbtmp, "%s/dev.tmp", _PATH_VARRUN);
       (void)sprintf(dbname, "%s/dev.db", _PATH_VARRUN);
       if ((db = dbm_open(dbtmp, O_CREAT|O_WRONLY|O_EXCL,
           S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH)) == NULL)
               error(dbtmp);

       key.dptr = (char *)&sb.st_rdev;
       key.dsize = sizeof(sb.st_rdev);
       data.dptr = buf;
       while (dp = readdir(dirp)) {
               if (stat(dp->d_name, &sb))
                       error(dp->d_name);
               if (!S_ISCHR(sb.st_mode))
                       continue;

               /* Nul terminate the name so ps doesn't have to. */
               bcopy(dp->d_name, buf, dp->d_namlen);
               buf[dp->d_namlen] = '\0';
               data.dsize = dp->d_namlen + 1;
               if (dbm_store(db, key, data, DBM_INSERT) < 0)
                       error("dbm_store");
       }
       (void)dbm_close(db);
       (void)strcat(dbtmp, DBM_SUFFIX);
       if (rename(dbtmp, dbname)) {
               (void)fprintf(stderr, "dev_mkdb: %s to %s: %s.\n",
                   dbtmp, dbname, strerror(errno));
               exit(1);
       }
       exit(0);
}

error(n)
       char *n;
{
       (void)fprintf(stderr, "kvm_mkdb: %s: %s\n", n, strerror(errno));
       exit(1);
}

usage()
{
       (void)fprintf(stderr, "usage: dev_mkdb\n");
       exit(1);
}