#ifndef LINT
static char rcsid[] = "$OpenBSD: makedbm.c,v 1.9 1997/08/18 03:11:34 millert Exp $";
#endif
#include <stdio.h>
#include <fcntl.h>
#include <ctype.h>
#include <sys/stat.h>
#include <sys/param.h>
#include <unistd.h>
#include <string.h>
#include <sys/errno.h>
#include "ypdb.h"
#include "ypdef.h"
#include "db.h"
extern char *__progname;
static int read_line(fp, buf, size)
FILE *fp;
char *buf;
int size;
{
int done;
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';
while ((ch = getc(fp)) != EOF &&
isascii(ch) && isspace(ch))
;
(void) ungetc(ch, fp);
} else {
return done;
}
}
} while (size > 0 && !feof(fp));
return done;
}
void
add_record(db, str1, str2, check)
DBM *db;
char *str1, *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;
}
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(filename)
char *filename;
{
struct stat finfo;
static char datestr[11];
int status;
if (strcmp(filename,"-") == 0) {
sprintf(datestr, "%010d", time(0));
} else {
status = stat(filename, &finfo);
if (status < 0) {
fprintf(stderr, "%s: can't stat %s\n", __progname, filename);
exit(1);
}
sprintf(datestr, "%010d", finfo.st_mtime);
}
return datestr;
}
void
list_database(database,Uflag)
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\n", __progname, database);
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);
}
void
create_database(infile,database,
yp_input_file,yp_output_file,
yp_master_name,yp_domain_name,
bflag, lflag, sflag)
char *infile, *database;
char *yp_input_file, *yp_output_file;
char *yp_master_name, *yp_domain_name;
int bflag, lflag, sflag;
{
FILE *data_file;
char data_line[4096];
char myname[MAXHOSTNAMELEN];
int line_no = 0;
int len;
char *p,*k,*v;
char *slash;
DBM *new_db;
static char mapname[] = "ypdbXXXXXXXXXX";
char db_mapname[MAXPATHLEN],db_outfile[MAXPATHLEN],
db_tempname[MAXPATHLEN];
char 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;
else
*database = 0;
if (strlen(database) + strlen(mapname)
+ strlen(YPDB_SUFFIX) > MAXPATHLEN) {
fprintf(stderr,"%s: %s: directory name too long\n",
__progname, database);
exit(1);
}
snprintf(db_tempname, sizeof(db_tempname), "%s%s", database,
mapname);
mktemp(db_tempname);
snprintf(db_mapname, sizeof(db_mapname), "%s%s", db_tempname,
YPDB_SUFFIX);
new_db = ypdb_open(db_tempname, O_RDWR|O_CREAT, 0444);
while (read_line(data_file,data_line,sizeof(data_line))) {
line_no++;
len = strlen(data_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;
while (!isspace(*p)) {
if (lflag && isupper(*p))
*p = tolower(*p);
p++;
};
while (isspace(*p)) {
*p = '\0';
p++;
};
v = p;
while(*p != '\0') { p++; };
add_record(new_db, k, v, TRUE);
}
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) - 1);
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);
}
}
int
main (argc,argv)
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 usage = 0;
int ch;
extern int optind;
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 = argv[optind];
aflag++;
break;
case 'o':
yp_output_file = argv[optind];
aflag++;
break;
case 'm':
yp_master_name = argv[optind];
aflag++;
break;
case 'd':
yp_domain_name = argv[optind];
aflag++;
break;
case 'u':
uflag++;
break;
default:
usage++;
break;
}
if ((uflag != 0) && (aflag != 0)) {
usage++;
} else {
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 (usage) {
fprintf(stderr,"%s%s%s",
"usage:\tmakedbm [-u|-U] file\n\tmakedbm [-bls]",
" [-i YP_INPUT_FILE] [-o YP_OUTPUT_FILE]\n\t\t",
"[-d YP_DOMAIN_NAME] [-m YP_MASTER_NAME] infile outfile\n");
exit(1);
}
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);
}