OpenBSD-4.6/usr.sbin/ypserv/makedbm/makedbm.c
/* $OpenBSD: makedbm.c,v 1.28 2007/02/19 21:42:41 jmc Exp $ */
/*
* Copyright (c) 1994-97 Mats O Jansson <moj@stacken.kth.se>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
* OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#ifndef lint
static const char rcsid[] = "$OpenBSD: makedbm.c,v 1.28 2007/02/19 21:42:41 jmc Exp $";
#endif
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <ctype.h>
#include <sys/stat.h>
#include <sys/param.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include "ypdb.h"
#include "ypdef.h"
#include "db.h"
extern char *__progname; /* from crt0.o */
/*
* Read one line
*/
static int
read_line(FILE *fp, char *buf, int size)
{
int done = 0;
do {
while (fgets(buf, size, fp)) {
int len = strlen(buf);
done += len;
if (len > 1 && buf[len-2] == '\\' &&
buf[len-1] == '\n') {
int ch;
buf += len - 2;
size -= len - 2;
*buf = '\n'; buf[1] = '\0';
/* Skip leading white space on next line */
while ((ch = getc(fp)) != EOF &&
isascii(ch) && isspace(ch))
;
(void) ungetc(ch, fp);
} else {
return done;
}
}
} while (size > 0 && !feof(fp));
return done;
}
static void
add_record(DBM *db, char *str1, char *str2, int check)
{
datum key, val;
int status;
key.dptr = str1;
key.dsize = strlen(str1);
if (check) {
val = ypdb_fetch(db, key);
if (val.dptr != NULL)
return; /* already there */
}
val.dptr = str2;
val.dsize = strlen(str2);
status = ypdb_store(db, key, val, YPDB_INSERT);
if (status != 0) {
printf("%s: problem storing %s %s\n",
__progname, str1, str2);
exit(1);
}
}
static char *
file_date(char *filename)
{
struct stat finfo;
static char datestr[11];
int status;
if (strcmp(filename,"-") == 0) {
snprintf(datestr, sizeof datestr, "%010u", time(0));
} else {
status = stat(filename, &finfo);
if (status < 0) {
fprintf(stderr, "%s: can't stat %s\n", __progname, filename);
exit(1);
}
snprintf(datestr, sizeof datestr, "%010u", finfo.st_mtime);
}
return datestr;
}
static void
list_database(char *database, int Uflag)
{
DBM *db;
datum key, val;
db = ypdb_open(database, O_RDONLY, 0444);
if (db == NULL) {
if (Uflag != 0)
if (db_hash_list_database(database))
return;
fprintf(stderr, "%s: can't open database %s: %s\n", __progname,
database, strerror(errno));
exit(1);
}
key = ypdb_firstkey(db);
while (key.dptr != NULL) {
val = ypdb_fetch(db, key);
printf("%*.*s %*.*s\n",
key.dsize, key.dsize, key.dptr,
val.dsize, val.dsize, val.dptr);
key = ypdb_nextkey(db);
}
ypdb_close(db);
}
static void
create_database(char *infile, char *database, char *yp_input_file,
char *yp_output_file, char *yp_master_name, char *yp_domain_name,
int bflag, int lflag, int sflag)
{
FILE *data_file;
char data_line[4096]; /* XXX: DB bsize = 4096 in ypdb.c */
char myname[MAXHOSTNAMELEN];
int line_no = 0, len;
char *p, *k, *v, *slash;
DBM *new_db;
static char mapname[] = "ypdbXXXXXXXXXX";
char db_mapname[MAXPATHLEN], db_outfile[MAXPATHLEN];
char db_tempname[MAXPATHLEN], empty_str[] = "";
if (strcmp(infile,"-") == 0) {
data_file = stdin;
} else {
data_file = fopen(infile, "r");
if (errno != 0) {
(void)fprintf(stderr,"%s: ", __progname);
perror(infile);
exit(1);
}
}
if (strlen(database) + strlen(YPDB_SUFFIX) > MAXPATHLEN) {
fprintf(stderr,"%s: %s: file name too long\n",
__progname, database);
exit(1);
}
snprintf(db_outfile, sizeof(db_outfile), "%s%s", database, YPDB_SUFFIX);
slash = strrchr(database, '/');
if (slash != NULL)
slash[1] = 0; /* truncate to dir */
else
*database = 0; /* eliminate */
/* note: database is now directory where map goes ! */
if (strlen(database) + strlen(mapname) +
strlen(YPDB_SUFFIX) > MAXPATHLEN) {
fprintf(stderr,"%s: %s: directory name too long\n",
__progname, database);
exit(1);
}
snprintf(db_mapname, sizeof(db_mapname), "%s%s%s",
database, mapname, YPDB_SUFFIX);
mkstemps(db_mapname, sizeof(YPDB_SUFFIX));
strlcpy(db_tempname, db_mapname, sizeof db_tempname);
db_tempname[strlen(db_tempname) - sizeof(YPDB_SUFFIX) + 1] = '\0';
new_db = ypdb_open(db_tempname, O_RDWR|O_CREAT, 0444);
if (new_db == NULL) {
fprintf(stderr, "%s: Unable to open output database %s\n",
__progname, db_outfile);
exit(1);
}
while (read_line(data_file, data_line, sizeof(data_line))) {
line_no++;
len = strlen(data_line);
/* Check if we have the whole line */
if (data_line[len-1] != '\n') {
fprintf(stderr, "line %d in \"%s\" is too long",
line_no, infile);
} else {
data_line[len-1] = '\0';
}
p = (char *) &data_line;
k = p; /* save start of key */
while (!isspace(*p)) { /* find first "space" */
if (lflag && isupper(*p)) /* if force lower case */
*p = (char)tolower(*p); /* fix it */
p++;
}
while (isspace(*p)) { /* replace space with <NUL> */
*p = '\0';
p++;
}
v = p; /* save start of value */
while (*p != '\0') /* find end of string */
p++;
add_record(new_db, k, v, TRUE); /* save record */
}
if (strcmp(infile,"-") != 0)
(void) fclose(data_file);
add_record(new_db, YP_LAST_KEY, file_date(infile), FALSE);
if (yp_input_file)
add_record(new_db, YP_INPUT_KEY, yp_input_file, FALSE);
if (yp_output_file)
add_record(new_db, YP_OUTPUT_KEY, yp_output_file, FALSE);
if (yp_master_name)
add_record(new_db, YP_MASTER_KEY, yp_master_name, FALSE);
else {
gethostname(myname, sizeof(myname));
add_record(new_db, YP_MASTER_KEY, myname, FALSE);
}
if (yp_domain_name)
add_record(new_db, YP_DOMAIN_KEY, yp_domain_name, FALSE);
if (bflag)
add_record(new_db, YP_INTERDOMAIN_KEY, empty_str, FALSE);
if (sflag)
add_record(new_db, YP_SECURE_KEY, empty_str, FALSE);
ypdb_close(new_db);
if (rename(db_mapname, db_outfile) < 0) {
perror("rename");
fprintf(stderr,"rename %s -> %s failed!\n", db_mapname,
db_outfile);
exit(1);
}
}
static void
usage(void)
{
fprintf(stderr,"usage: makedbm [-blsUu] [-d yp_domain_name] "
"[-i yp_input_file]\n"
"\t[-m yp_master_name] [-o yp_output_file] infile outfile\n");
exit(1);
}
int
main(int argc, char *argv[])
{
int aflag, uflag, bflag, lflag, sflag, Uflag;
char *yp_input_file, *yp_output_file;
char *yp_master_name, *yp_domain_name;
char *infile, *outfile;
int ch;
extern int optind;
extern char *optarg;
yp_input_file = yp_output_file = NULL;
yp_master_name = yp_domain_name = NULL;
aflag = uflag = bflag = lflag = sflag = Uflag = 0;
infile = outfile = NULL;
while ((ch = getopt(argc, argv, "Ublsui:o:m:d:")) != -1)
switch (ch) {
case 'U':
uflag++;
Uflag++;
break;
case 'b':
bflag++;
aflag++;
break;
case 'l':
lflag++;
aflag++;
break;
case 's':
sflag++;
aflag++;
break;
case 'i':
yp_input_file = optarg;
aflag++;
break;
case 'o':
yp_output_file = optarg;
aflag++;
break;
case 'm':
yp_master_name = optarg;
aflag++;
break;
case 'd':
yp_domain_name = optarg;
aflag++;
break;
case 'u':
uflag++;
break;
default:
usage();
break;
}
if ((uflag != 0) && (aflag != 0))
usage();
if (uflag != 0) {
if (argc == (optind + 1))
infile = argv[optind];
else
usage();
} else {
if (argc == (optind + 2)) {
infile = argv[optind];
outfile = argv[optind+1];
} else
usage();
}
if (uflag != 0) {
list_database(infile,Uflag);
} else {
create_database(infile, outfile, yp_input_file,
yp_output_file, yp_master_name, yp_domain_name,
bflag, lflag, sflag);
}
return(0);
}