Home | History | Annotate | Line # | Download | only in db
db.c revision 1.24
      1  1.24    lukem /*	$NetBSD: db.c,v 1.24 2009/01/28 05:48:49 lukem Exp $	*/
      2   1.1    lukem 
      3   1.1    lukem /*-
      4  1.24    lukem  * Copyright (c) 2002-2009 The NetBSD Foundation, Inc.
      5   1.1    lukem  * All rights reserved.
      6   1.1    lukem  *
      7   1.1    lukem  * This code is derived from software contributed to The NetBSD Foundation
      8   1.1    lukem  * by Luke Mewburn of Wasabi Systems.
      9   1.1    lukem  *
     10   1.1    lukem  * Redistribution and use in source and binary forms, with or without
     11   1.1    lukem  * modification, are permitted provided that the following conditions
     12   1.1    lukem  * are met:
     13   1.1    lukem  * 1. Redistributions of source code must retain the above copyright
     14   1.1    lukem  *    notice, this list of conditions and the following disclaimer.
     15   1.1    lukem  * 2. Redistributions in binary form must reproduce the above copyright
     16   1.1    lukem  *    notice, this list of conditions and the following disclaimer in the
     17   1.1    lukem  *    documentation and/or other materials provided with the distribution.
     18   1.1    lukem  *
     19   1.1    lukem  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20   1.1    lukem  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21   1.1    lukem  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22   1.1    lukem  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23   1.1    lukem  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24   1.1    lukem  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25   1.1    lukem  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26   1.1    lukem  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27   1.1    lukem  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28   1.1    lukem  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29   1.1    lukem  * POSSIBILITY OF SUCH DAMAGE.
     30   1.1    lukem  */
     31   1.1    lukem 
     32  1.22    lukem #if HAVE_NBTOOL_CONFIG_H
     33  1.22    lukem #include "nbtool_config.h"
     34  1.22    lukem #endif
     35  1.22    lukem 
     36   1.1    lukem #include <sys/cdefs.h>
     37   1.1    lukem #ifndef lint
     38  1.10  aymeric #ifdef __RCSID
     39  1.24    lukem __RCSID("$NetBSD: db.c,v 1.24 2009/01/28 05:48:49 lukem Exp $");
     40  1.10  aymeric #endif /* __RCSID */
     41   1.1    lukem #endif /* not lint */
     42   1.1    lukem 
     43  1.21    lukem #include <ctype.h>
     44   1.1    lukem #include <db.h>
     45   1.1    lukem #include <err.h>
     46   1.1    lukem #include <fcntl.h>
     47  1.20    lukem #include <limits.h>
     48   1.1    lukem #include <stdio.h>
     49   1.1    lukem #include <stdlib.h>
     50   1.1    lukem #include <string.h>
     51   1.1    lukem #include <unistd.h>
     52   1.8      seb #include <vis.h>
     53   1.1    lukem 
     54   1.3    lukem 
     55   1.1    lukem typedef enum {
     56   1.1    lukem 	F_WRITE		= 1<<0,
     57   1.1    lukem 	F_DELETE	= 1<<1,
     58   1.1    lukem 	F_SHOW_KEY	= 1<<2,
     59   1.1    lukem 	F_SHOW_VALUE	= 1<<3,
     60   1.1    lukem 	F_QUIET		= 1<<10,
     61   1.1    lukem 	F_IGNORECASE	= 1<<11,
     62   1.1    lukem 	F_ENDIAN_BIG	= 1<<12,
     63   1.1    lukem 	F_ENDIAN_LITTLE	= 1<<13,
     64   1.1    lukem 	F_NO_NUL	= 1<<14,
     65   1.1    lukem 	F_CREATENEW	= 1<<20,
     66   1.1    lukem 	F_DUPLICATES	= 1<<21,
     67   1.1    lukem 	F_REPLACE	= 1<<22,
     68   1.8      seb 	F_ENCODE_KEY	= 1<<23,
     69   1.8      seb 	F_ENCODE_VAL	= 1<<24,
     70   1.8      seb 	F_DECODE_KEY	= 1<<25,
     71   1.8      seb 	F_DECODE_VAL	= 1<<26,
     72   1.1    lukem } flags_t;
     73   1.1    lukem 
     74   1.1    lukem int	main(int, char *[]);
     75   1.1    lukem void	db_print(DBT *, DBT *);
     76   1.1    lukem int	db_dump(void);
     77   1.1    lukem int	db_del(char *);
     78   1.1    lukem int	db_get(char *);
     79  1.24    lukem int	db_seq(char *);
     80   1.1    lukem int	db_put(char *, char *);
     81   1.1    lukem int	parseline(FILE *, const char *, char **, char **);
     82   1.8      seb int	encode_data(size_t, char *, char **);
     83   1.8      seb int	decode_data(char *, char **);
     84   1.8      seb void	parse_encode_decode_arg(const char *, int);
     85   1.8      seb int	parse_encode_option(char **);
     86   1.1    lukem void	usage(void);
     87   1.1    lukem 
     88  1.20    lukem flags_t	 	 flags = 0;
     89  1.20    lukem DB		*db;
     90  1.20    lukem const char	*outputsep = "\t";
     91  1.20    lukem int		 visflags = 0;
     92  1.20    lukem const char	*extra_echars = NULL;
     93   1.1    lukem 
     94   1.1    lukem int
     95   1.1    lukem main(int argc, char *argv[])
     96   1.1    lukem {
     97   1.1    lukem 	struct {
     98  1.20    lukem 		char		*file;
     99  1.20    lukem 		char		*type;
    100  1.20    lukem 		DBTYPE	 	dbtype;
    101  1.20    lukem 		void		*info;
    102  1.24    lukem 		int	 	dbflags;
    103  1.20    lukem 		mode_t	 	mode;
    104  1.20    lukem 		unsigned int	pagesize;
    105   1.1    lukem 	} oi;
    106   1.1    lukem 	BTREEINFO	btreeinfo;
    107   1.1    lukem 	HASHINFO	hashinfo;
    108   1.1    lukem 	FILE		*infp;
    109   1.1    lukem 	const char	*infile, *fieldsep;
    110   1.1    lukem 	char		*p, *key, *val;
    111   1.1    lukem 	int		ch, rv;
    112  1.20    lukem 	long		lval;
    113   1.1    lukem 
    114   1.1    lukem 	setprogname(argv[0]);
    115   1.1    lukem 
    116   1.1    lukem 	infile = NULL;
    117   1.1    lukem 	fieldsep = " ";
    118   1.1    lukem 	infp = NULL;
    119   1.1    lukem 	memset(&oi, 0, sizeof(oi));
    120   1.1    lukem 	oi.mode = 0644;
    121  1.20    lukem 	oi.pagesize = 4096;
    122   1.1    lukem 
    123   1.1    lukem 				/* parse arguments */
    124   1.8      seb 	while ( (ch = getopt(argc, argv,
    125  1.20    lukem 			     "CDdE:F:f:iKm:NO:P:qRS:T:U:VwX:")) != -1) {
    126   1.1    lukem 		switch (ch) {
    127   1.1    lukem 
    128   1.1    lukem 		case 'C':
    129   1.1    lukem 			flags |= F_CREATENEW;
    130   1.1    lukem 			break;
    131   1.1    lukem 
    132   1.1    lukem 		case 'D':
    133   1.1    lukem 			flags |= F_DUPLICATES;
    134   1.1    lukem 			break;
    135   1.1    lukem 
    136   1.1    lukem 		case 'd':
    137   1.1    lukem 			flags |= F_DELETE;
    138   1.1    lukem 			break;
    139   1.1    lukem 
    140   1.1    lukem 		case 'E':
    141   1.1    lukem 			if (! optarg[0] || optarg[1])
    142   1.1    lukem 				goto badendian;
    143   1.4    lukem 			switch (toupper((int)optarg[0])) {
    144   1.1    lukem 			case 'B':
    145   1.1    lukem 				flags |= F_ENDIAN_BIG;
    146   1.1    lukem 				break;
    147   1.1    lukem 			case 'L':
    148   1.1    lukem 				flags |= F_ENDIAN_LITTLE;
    149   1.1    lukem 				break;
    150   1.1    lukem 			case 'H':
    151   1.1    lukem 				flags &= ~(F_ENDIAN_BIG | F_ENDIAN_LITTLE);
    152   1.1    lukem 				break;
    153   1.1    lukem 			default:
    154   1.1    lukem  badendian:
    155   1.1    lukem 				errx(1, "Bad endian `%s'", optarg);
    156   1.1    lukem 			}
    157   1.1    lukem 			break;
    158   1.1    lukem 
    159   1.1    lukem 		case 'F':
    160  1.14    lukem 			if (! optarg[0])
    161   1.5      seb 				errx(1, "Invalid field separator `%s'",
    162   1.5      seb 				    optarg);
    163   1.1    lukem 			fieldsep = optarg;
    164   1.1    lukem 			break;
    165   1.1    lukem 
    166   1.1    lukem 		case 'f':
    167   1.1    lukem 			infile = optarg;
    168   1.1    lukem 			break;
    169  1.15    lukem 
    170   1.1    lukem 		case 'i':
    171   1.1    lukem 			flags |= F_IGNORECASE;
    172   1.1    lukem 			break;
    173   1.1    lukem 
    174   1.1    lukem 		case 'K':
    175   1.1    lukem 			flags |= F_SHOW_KEY;
    176   1.1    lukem 			break;
    177   1.1    lukem 
    178   1.1    lukem 		case 'm':
    179  1.20    lukem 			lval = strtol(optarg, &p, 8);
    180   1.1    lukem 			if (p == optarg || *p != '\0')
    181   1.1    lukem 				errx(1, "Invalid octal number `%s'", optarg);
    182  1.20    lukem 			if (lval < 0 || lval > 07777)
    183  1.20    lukem 				errx(1, "Invalid mode `%s'", optarg);
    184  1.20    lukem 			oi.mode = (mode_t)lval;
    185   1.1    lukem 			break;
    186   1.1    lukem 
    187   1.1    lukem 		case 'N':
    188   1.1    lukem 			flags |= F_NO_NUL;
    189   1.1    lukem 			break;
    190   1.1    lukem 
    191   1.6      seb 		case 'O':
    192   1.6      seb 			outputsep = optarg;
    193   1.6      seb 			break;
    194   1.6      seb 
    195  1.20    lukem 		case 'P':
    196  1.20    lukem 			lval = strtol(optarg, &p, 10);
    197  1.20    lukem 			if (p == optarg || *p != '\0')
    198  1.20    lukem 				errx(1, "Invalid pagesize `%s'", optarg);
    199  1.23    lukem 			if (lval < 0 || (unsigned int)lval >= UINT_MAX)
    200  1.20    lukem 				errx(1, "Pagesize `%s' out of range", optarg);
    201  1.20    lukem 			oi.pagesize = (unsigned int)lval;
    202  1.20    lukem 			break;
    203  1.20    lukem 
    204   1.1    lukem 		case 'q':
    205   1.1    lukem 			flags |= F_QUIET;
    206   1.1    lukem 			break;
    207   1.1    lukem 
    208   1.1    lukem 		case 'R':
    209   1.1    lukem 			flags |= F_REPLACE;
    210   1.1    lukem 			break;
    211   1.1    lukem 
    212   1.8      seb 		case 'S':
    213   1.8      seb 			parse_encode_decode_arg(optarg, 1 /* encode */);
    214   1.8      seb 			if (! (flags & (F_ENCODE_KEY | F_ENCODE_VAL)))
    215   1.8      seb 				errx(1, "Invalid encoding argument `%s'",
    216   1.8      seb 				    optarg);
    217   1.8      seb 			break;
    218   1.8      seb 
    219   1.8      seb 		case 'T':
    220  1.19    lukem 			visflags = parse_encode_option(&optarg);
    221  1.19    lukem 			if (! visflags)
    222  1.19    lukem 				errx(1, "Invalid encoding/decoding option `%s'",
    223   1.8      seb 				    optarg);
    224   1.8      seb 			break;
    225   1.8      seb 
    226   1.8      seb 		case 'U':
    227   1.8      seb 			parse_encode_decode_arg(optarg, 0 /* decode */);
    228   1.8      seb 			if (! (flags & (F_DECODE_KEY | F_DECODE_VAL)))
    229   1.8      seb 				errx(1, "Invalid decoding argument `%s'",
    230   1.8      seb 				    optarg);
    231   1.8      seb 			break;
    232   1.8      seb 
    233   1.1    lukem 		case 'V':
    234   1.1    lukem 			flags |= F_SHOW_VALUE;
    235   1.1    lukem 			break;
    236   1.1    lukem 
    237   1.1    lukem 		case 'w':
    238   1.1    lukem 			flags |= F_WRITE;
    239   1.1    lukem 			break;
    240   1.1    lukem 
    241   1.8      seb 		case 'X':
    242   1.8      seb 			extra_echars = optarg;
    243   1.8      seb 			break;
    244   1.8      seb 
    245   1.1    lukem 		default:
    246   1.1    lukem 			usage();
    247   1.1    lukem 
    248   1.1    lukem 		}
    249   1.1    lukem 	}
    250   1.1    lukem 	argc -= optind;
    251   1.1    lukem 	argv += optind;
    252   1.1    lukem 
    253   1.1    lukem 				/* validate arguments */
    254   1.1    lukem 	if (argc < 2)
    255   1.1    lukem 		usage();
    256   1.1    lukem 	oi.type = argv[0];
    257   1.1    lukem 	oi.file = argv[1];
    258   1.1    lukem 	argc -= 2;
    259   1.1    lukem 	argv += 2;
    260   1.1    lukem 
    261   1.1    lukem 	if (flags & F_WRITE) {
    262   1.1    lukem 		if (flags & (F_SHOW_KEY | F_SHOW_VALUE | F_DELETE))
    263   1.1    lukem 			usage();
    264   1.1    lukem 		if ((!infile && argc < 2) || (argc % 2))
    265   1.1    lukem 			usage();
    266  1.19    lukem 		if (0 != (visflags & ~(VIS_HTTPSTYLE)))
    267  1.19    lukem 			errx(1, "Unsupported decoding option provided to -T");
    268  1.24    lukem 		oi.dbflags = O_RDWR | O_CREAT | O_EXLOCK;
    269   1.1    lukem 		if (flags & F_CREATENEW)
    270  1.24    lukem 			oi.dbflags |= O_TRUNC;
    271   1.1    lukem 	} else if (flags & F_DELETE) {
    272   1.1    lukem 		if (flags & (F_SHOW_KEY | F_SHOW_VALUE | F_WRITE))
    273   1.1    lukem 			usage();
    274   1.1    lukem 		if (!infile && argc < 1)
    275   1.1    lukem 			usage();
    276  1.19    lukem 		if (0 != (visflags & ~(VIS_HTTPSTYLE)))
    277  1.19    lukem 			errx(1, "Unsupported decoding option provided to -T");
    278  1.24    lukem 		oi.dbflags = O_RDWR | O_CREAT | O_EXLOCK;
    279   1.1    lukem 	} else {
    280   1.1    lukem 		if (! (flags & (F_SHOW_KEY | F_SHOW_VALUE)))
    281   1.1    lukem 			flags |= (F_SHOW_KEY | F_SHOW_VALUE);
    282  1.24    lukem 		oi.dbflags = O_RDONLY | O_SHLOCK;
    283   1.1    lukem 	}
    284   1.1    lukem 
    285   1.1    lukem 				/* validate oi.type */
    286   1.1    lukem 	if (strcmp(oi.type, "btree") == 0) {
    287   1.1    lukem 		memset(&btreeinfo, 0, sizeof(btreeinfo));
    288   1.1    lukem 		if (flags & F_ENDIAN_BIG)
    289   1.1    lukem 			btreeinfo.lorder = 4321;
    290   1.1    lukem 		else if (flags & F_ENDIAN_LITTLE)
    291   1.1    lukem 			btreeinfo.lorder = 1234;
    292   1.1    lukem 		if (flags & F_DUPLICATES)
    293   1.1    lukem 			btreeinfo.flags = R_DUP;
    294  1.20    lukem 		btreeinfo.psize = oi.pagesize;
    295   1.2    lukem 		btreeinfo.cachesize = 1024 * 1024;
    296   1.1    lukem 		oi.info = &btreeinfo;
    297   1.1    lukem 		oi.dbtype = DB_BTREE;
    298   1.1    lukem 	} else if (strcmp(oi.type, "hash") == 0) {
    299   1.1    lukem 		memset(&hashinfo, 0, sizeof(hashinfo));
    300   1.1    lukem 		if (flags & F_ENDIAN_BIG)
    301   1.1    lukem 			hashinfo.lorder = 4321;
    302   1.1    lukem 		else if (flags & F_ENDIAN_LITTLE)
    303   1.1    lukem 			hashinfo.lorder = 1234;
    304  1.20    lukem 		hashinfo.bsize = oi.pagesize;
    305   1.2    lukem 		hashinfo.cachesize = 1024 * 1024;
    306   1.1    lukem 		oi.info = &hashinfo;
    307   1.1    lukem 		oi.dbtype = DB_HASH;
    308   1.1    lukem 	} else {
    309   1.1    lukem 		warnx("Unknown database type `%s'", oi.type);
    310   1.1    lukem 		usage();
    311   1.1    lukem 	}
    312   1.1    lukem 
    313   1.1    lukem 	if (infile) {
    314   1.1    lukem 		if (strcmp(infile, "-") == 0)
    315   1.1    lukem 			infp = stdin;
    316   1.1    lukem 		else if ((infp = fopen(infile, "r")) == NULL)
    317   1.1    lukem 			err(1, "Opening input file `%s'", infile);
    318   1.1    lukem 	}
    319   1.1    lukem 
    320   1.1    lukem 				/* open database */
    321  1.24    lukem 	db = dbopen(oi.file, oi.dbflags, oi.mode, oi.dbtype, oi.info);
    322   1.1    lukem 	if (db == NULL)
    323   1.1    lukem 		err(1, "Opening database `%s'", oi.file);
    324   1.1    lukem 
    325   1.1    lukem 
    326   1.1    lukem 				/* manipulate database */
    327   1.1    lukem 	rv = 0;
    328   1.1    lukem 	if (flags & F_WRITE) {			/* write entries */
    329   1.1    lukem 		for (ch = 0; ch < argc; ch += 2)
    330   1.1    lukem 			if ((rv = db_put(argv[ch], argv[ch+1])))
    331   1.1    lukem 				goto cleanup;
    332   1.1    lukem 		if (infp) {
    333   1.1    lukem 			while (parseline(infp, fieldsep, &key, &val)) {
    334   1.1    lukem 				if ((rv = db_put(key, val)))
    335   1.1    lukem 					goto cleanup;
    336   1.1    lukem 			}
    337   1.1    lukem 			if (ferror(infp)) {
    338   1.1    lukem 				warnx("Reading `%s'", infile);
    339   1.1    lukem 				goto cleanup;
    340   1.1    lukem 			}
    341   1.1    lukem 		}
    342   1.1    lukem 	} else if (!infp && argc == 0) {	/* read all */
    343   1.1    lukem 		db_dump();
    344   1.1    lukem 	} else {				/* read/delete specific */
    345   1.1    lukem 		int	(*dbop)(char *);
    346   1.1    lukem 
    347   1.1    lukem 		if (flags & F_DELETE)
    348   1.1    lukem 			dbop = db_del;
    349  1.24    lukem 		else if (DB_BTREE == oi.dbtype)
    350  1.24    lukem 			dbop = db_seq;
    351  1.24    lukem 		else if (DB_HASH == oi.dbtype)
    352  1.24    lukem 			dbop = db_get;
    353   1.1    lukem 		else
    354  1.24    lukem 			errx(5, "internal error: unsupported dbtype %d",
    355  1.24    lukem 			    oi.dbtype);
    356   1.1    lukem 		for (ch = 0; ch < argc; ch++) {
    357   1.1    lukem 			if ((rv = dbop(argv[ch])))
    358   1.1    lukem 				goto cleanup;
    359   1.1    lukem 		}
    360   1.1    lukem 		if (infp) {
    361   1.1    lukem 			while (parseline(infp, fieldsep, &key, NULL)) {
    362   1.1    lukem 				if ((rv = dbop(key)))
    363   1.1    lukem 					goto cleanup;
    364   1.1    lukem 			}
    365   1.1    lukem 			if (ferror(infp)) {
    366   1.1    lukem 				warnx("Reading `%s'", infile);
    367   1.1    lukem 				goto cleanup;
    368   1.1    lukem 			}
    369   1.1    lukem 		}
    370   1.1    lukem 	}
    371   1.1    lukem 
    372   1.1    lukem 				/* close database */
    373   1.1    lukem  cleanup:
    374   1.1    lukem 	if (db->close(db) == -1)
    375   1.1    lukem 		err(1, "Closing database `%s'", oi.file);
    376   1.1    lukem 	if (infp)
    377   1.1    lukem 		fclose(infp);
    378   1.1    lukem 	return (rv);
    379   1.1    lukem }
    380   1.1    lukem 
    381   1.1    lukem void
    382   1.1    lukem db_print(DBT *key, DBT *val)
    383   1.1    lukem {
    384  1.13    lukem 	int	len;
    385  1.13    lukem 	char	*data;
    386   1.1    lukem 
    387  1.13    lukem #define	MINUSNUL(x)	((x) > 0  ?  (x) - (flags & F_NO_NUL ? 0 : 1)  :  0)
    388   1.8      seb 
    389   1.8      seb 	if (flags & F_SHOW_KEY) {
    390  1.13    lukem 		if (flags & F_ENCODE_KEY) {
    391  1.13    lukem 			len = encode_data(MINUSNUL(key->size),
    392  1.13    lukem 			    (char *)key->data, &data);
    393  1.13    lukem 		} else {
    394  1.13    lukem 			len = (int)MINUSNUL(key->size);
    395   1.8      seb 			data = (char *)key->data;
    396   1.8      seb 		}
    397   1.8      seb 		printf("%.*s", len, data);
    398   1.8      seb 	}
    399   1.1    lukem 	if ((flags & F_SHOW_KEY) && (flags & F_SHOW_VALUE))
    400   1.5      seb 		printf("%s", outputsep);
    401   1.8      seb 	if (flags & F_SHOW_VALUE) {
    402  1.13    lukem 		if (flags & F_ENCODE_VAL) {
    403  1.13    lukem 			len = encode_data(MINUSNUL(val->size),
    404  1.13    lukem 			    (char *)val->data, &data);
    405  1.13    lukem 		} else {
    406  1.13    lukem 			len = (int)MINUSNUL(val->size);
    407   1.8      seb 			data = (char *)val->data;
    408  1.15    lukem 		}
    409   1.8      seb 		printf("%.*s", len, data);
    410   1.8      seb 	}
    411   1.1    lukem 	printf("\n");
    412   1.1    lukem }
    413   1.1    lukem 
    414   1.1    lukem int
    415   1.1    lukem db_dump(void)
    416   1.1    lukem {
    417   1.1    lukem 	DBT	key, val;
    418   1.1    lukem 	int	rv;
    419   1.1    lukem 
    420   1.1    lukem 	while ((rv = db->seq(db, &key, &val, R_NEXT)) == 0)
    421   1.1    lukem 		db_print(&key, &val);
    422   1.1    lukem 	if (rv == -1)
    423   1.1    lukem 		warn("Error dumping database");
    424   1.1    lukem 	return (rv == 1 ? 0 : 1);
    425   1.1    lukem }
    426   1.1    lukem 
    427   1.1    lukem static void
    428   1.8      seb db_makekey(DBT *key, char *keystr, int downcase, int decode)
    429   1.1    lukem {
    430   1.8      seb 	char	*p, *ks;
    431   1.8      seb 	int	klen;
    432   1.1    lukem 
    433   1.1    lukem 	memset(key, 0, sizeof(*key));
    434   1.8      seb 	if (decode) {
    435   1.8      seb 		if ((klen = decode_data(keystr, &ks)) == -1)
    436  1.15    lukem 			errx(1, "Invalid escape sequence in `%s'", keystr);
    437   1.8      seb 	} else {
    438   1.8      seb 		klen = strlen(keystr);
    439   1.8      seb 		ks = keystr;
    440   1.8      seb 	}
    441   1.8      seb 	key->data = ks;
    442   1.8      seb 	key->size = klen + (flags & F_NO_NUL ? 0 : 1);
    443  1.15    lukem 	if (downcase && (flags & F_IGNORECASE)) {
    444   1.8      seb 		for (p = ks; *p; p++)
    445   1.1    lukem 			if (isupper((int)*p))
    446   1.1    lukem 				*p = tolower((int)*p);
    447   1.1    lukem 	}
    448   1.1    lukem }
    449   1.1    lukem 
    450   1.1    lukem int
    451   1.1    lukem db_del(char *keystr)
    452   1.1    lukem {
    453   1.1    lukem 	DBT	key;
    454  1.21    lukem 	int	r;
    455   1.1    lukem 
    456   1.8      seb 	db_makekey(&key, keystr, 1, (flags & F_DECODE_KEY ? 1 : 0));
    457  1.21    lukem 	r = db->del(db, &key, 0);
    458  1.21    lukem 	switch (r) {
    459   1.1    lukem 	case -1:
    460  1.21    lukem 		if (! (flags & F_QUIET))
    461  1.21    lukem 			warn("Error deleting key `%s'", keystr);
    462   1.8      seb 		r = 1;
    463   1.8      seb 		break;
    464   1.1    lukem 	case 0:
    465   1.1    lukem 		if (! (flags & F_QUIET))
    466   1.1    lukem 			printf("Deleted key `%s'\n", keystr);
    467   1.1    lukem 		break;
    468   1.1    lukem 	case 1:
    469  1.21    lukem 		if (! (flags & F_QUIET))
    470  1.21    lukem 			warnx("Unknown key `%s'", keystr);
    471   1.8      seb 		break;
    472  1.21    lukem 	default:
    473  1.24    lukem 		errx(5, "%s: unexpected result %d from db", __func__, r);
    474   1.1    lukem 	}
    475   1.8      seb 	if (flags & F_DECODE_KEY)
    476   1.8      seb 		free(key.data);
    477   1.8      seb 	return (r);
    478   1.1    lukem }
    479   1.1    lukem 
    480   1.1    lukem int
    481   1.1    lukem db_get(char *keystr)
    482   1.1    lukem {
    483   1.1    lukem 	DBT	key, val;
    484  1.24    lukem 	int	r;
    485  1.24    lukem 
    486  1.24    lukem 	db_makekey(&key, keystr, 1, (flags & F_DECODE_KEY ? 1 : 0));
    487  1.24    lukem 
    488  1.24    lukem 	r = db->get(db, &key, &val, 0);
    489  1.24    lukem 	switch (r) {
    490  1.24    lukem 	case -1:
    491  1.24    lukem 		warn("Error reading key `%s'", keystr);
    492  1.24    lukem 		r = 1;
    493  1.24    lukem 		break;
    494  1.24    lukem 	case 0:
    495  1.24    lukem 		db_print(&key, &val);
    496  1.24    lukem 		break;
    497  1.24    lukem 	case 1:
    498  1.24    lukem 		if (! (flags & F_QUIET)) {
    499  1.24    lukem 			warnx("Unknown key `%s'", keystr);
    500  1.24    lukem 		}
    501  1.24    lukem 		break;
    502  1.24    lukem 	default:
    503  1.24    lukem 		errx(5, "%s: unexpected result %d from db", __func__, r);
    504  1.24    lukem 	}
    505  1.24    lukem 	if (flags & F_DECODE_KEY)
    506  1.24    lukem 		free(key.data);
    507  1.24    lukem 	return (r);
    508  1.24    lukem }
    509  1.24    lukem 
    510  1.24    lukem int
    511  1.24    lukem db_seq(char *keystr)
    512  1.24    lukem {
    513  1.24    lukem 	DBT	key, val, want;
    514  1.15    lukem 	int	r, found;
    515  1.15    lukem 	u_int	seqflags;
    516   1.1    lukem 
    517   1.8      seb 	db_makekey(&key, keystr, 1, (flags & F_DECODE_KEY ? 1 : 0));
    518  1.24    lukem 		/* remember key in want, since db->seq() changes key */
    519  1.24    lukem 	want.data = key.data;
    520  1.24    lukem 	want.size = key.size;
    521  1.15    lukem 
    522  1.15    lukem 	found = 0;
    523  1.15    lukem 	seqflags = R_CURSOR;
    524  1.15    lukem 	while ((r = db->seq(db, &key, &val, seqflags)) == 0) {
    525  1.24    lukem 		if (key.size != want.size ||
    526  1.24    lukem 		    0 != strcmp((char *)key.data, (char *)want.data)) {
    527  1.15    lukem 			r = 1;
    528  1.15    lukem 			break;
    529  1.15    lukem 		}
    530  1.15    lukem 		seqflags = R_NEXT;
    531  1.15    lukem 		found++;
    532  1.15    lukem 		db_print(&key, &val);
    533  1.15    lukem 		if (! (flags & F_DUPLICATES))
    534  1.15    lukem 			break;
    535  1.15    lukem 	}
    536  1.15    lukem 
    537  1.15    lukem 	switch (r) {
    538   1.1    lukem 	case -1:
    539   1.1    lukem 		warn("Error reading key `%s'", keystr);
    540   1.8      seb 		r = 1;
    541   1.8      seb 		break;
    542   1.1    lukem 	case 0:
    543   1.1    lukem 		break;
    544   1.1    lukem 	case 1:
    545  1.15    lukem 		if (found) {
    546  1.15    lukem 			r = 0;
    547  1.15    lukem 			break;
    548  1.15    lukem 		}
    549   1.1    lukem 		if (! (flags & F_QUIET)) {
    550   1.1    lukem 			warnx("Unknown key `%s'", keystr);
    551   1.1    lukem 		}
    552   1.1    lukem 		break;
    553  1.21    lukem 	default:
    554  1.24    lukem 		errx(5, "%s: unexpected result %d from db", __func__, r);
    555   1.1    lukem 	}
    556   1.8      seb 	if (flags & F_DECODE_KEY)
    557  1.24    lukem 		free(want.data);
    558   1.8      seb 	return (r);
    559   1.1    lukem }
    560   1.1    lukem 
    561   1.1    lukem int
    562   1.1    lukem db_put(char *keystr, char *valstr)
    563   1.1    lukem {
    564   1.1    lukem 	DBT	key, val;
    565   1.8      seb 	int	r = 0;
    566   1.1    lukem 
    567   1.8      seb 	db_makekey(&key, keystr, 1, (flags & F_DECODE_KEY ? 1 : 0));
    568   1.8      seb 	db_makekey(&val, valstr, 0, (flags & F_DECODE_VAL ? 1 : 0));
    569  1.21    lukem 	r = db->put(db, &key, &val, (flags & F_REPLACE) ? 0 : R_NOOVERWRITE);
    570  1.21    lukem 	switch (r) {
    571   1.1    lukem 	case -1:
    572   1.1    lukem 		warn("Error writing key `%s'", keystr);
    573   1.8      seb 		r = 1;
    574   1.8      seb 		break;
    575   1.1    lukem 	case 0:
    576   1.1    lukem 		if (! (flags & F_QUIET))
    577   1.1    lukem 			printf("Added key `%s'\n", keystr);
    578   1.1    lukem 		break;
    579   1.1    lukem 	case 1:
    580   1.1    lukem 		if (! (flags & F_QUIET))
    581   1.1    lukem 			warnx("Key `%s' already exists", keystr);
    582   1.8      seb 		break;
    583  1.21    lukem 	default:
    584  1.24    lukem 		errx(5, "Unexpected result %d in %s", r, __func__);
    585   1.1    lukem 	}
    586   1.8      seb 	if (flags & F_DECODE_KEY)
    587   1.8      seb 		free(key.data);
    588   1.8      seb 	if (flags & F_DECODE_VAL)
    589   1.8      seb 		free(val.data);
    590   1.8      seb 	return (r);
    591   1.1    lukem }
    592   1.1    lukem 
    593   1.1    lukem int
    594   1.1    lukem parseline(FILE *fp, const char *sep, char **kp, char **vp)
    595   1.1    lukem {
    596   1.1    lukem 	size_t	len;
    597   1.1    lukem 	char	*key, *val;
    598   1.1    lukem 
    599   1.1    lukem 	key = fgetln(fp, &len);
    600   1.1    lukem 	if (key == NULL)		/* end of file, or error */
    601   1.1    lukem 		return (0);
    602   1.1    lukem 
    603   1.1    lukem 	if (key[len-1] == '\n')		/* check for \n at EOL */
    604   1.1    lukem 		key[--len] = '\0';
    605   1.1    lukem 	else
    606   1.1    lukem 		return (0);
    607   1.1    lukem 
    608   1.1    lukem 	*kp = key;
    609   1.1    lukem 	if (vp == NULL)			/* don't split if don't want value */
    610   1.1    lukem 		return (1);
    611  1.14    lukem 	if ((val = strstr(key, sep)) == NULL)
    612   1.1    lukem 		val = key + len;
    613  1.14    lukem 	else {
    614  1.14    lukem 		*val = '\0';
    615  1.14    lukem 		val += strlen(sep);
    616  1.14    lukem 	}
    617   1.1    lukem 	*vp = val;
    618   1.1    lukem 	return (1);
    619   1.1    lukem }
    620   1.1    lukem 
    621   1.8      seb int
    622   1.8      seb encode_data(size_t len, char *data, char **edata)
    623   1.8      seb {
    624   1.8      seb 	static char	*buf = NULL;
    625  1.11   itojun 	char		*nbuf;
    626   1.8      seb 	static size_t	buflen = 0;
    627   1.8      seb 	size_t		elen;
    628   1.8      seb 
    629   1.8      seb 	elen = 1 + (len * 4);
    630   1.8      seb 	if (elen > buflen) {
    631  1.11   itojun 		if ((nbuf = realloc(buf, elen)) == NULL)
    632   1.8      seb 			err(1, "Cannot allocate encoding buffer");
    633  1.11   itojun 		buf = nbuf;
    634   1.8      seb 		buflen = elen;
    635   1.8      seb 	}
    636   1.8      seb 	*edata = buf;
    637   1.8      seb 	if (extra_echars) {
    638  1.19    lukem 		return (strsvisx(buf, data, len, visflags, extra_echars));
    639   1.8      seb 	} else {
    640  1.19    lukem 		return (strvisx(buf, data, len, visflags));
    641   1.8      seb 	}
    642   1.8      seb }
    643   1.8      seb 
    644   1.8      seb int
    645   1.8      seb decode_data(char *data, char **ddata)
    646   1.8      seb {
    647   1.8      seb 	char	*buf;
    648   1.8      seb 
    649   1.8      seb 	if ((buf = malloc(strlen(data) + 1)) == NULL)
    650   1.8      seb 		err(1, "Cannot allocate decoding buffer");
    651   1.8      seb 	*ddata = buf;
    652  1.19    lukem 	return (strunvisx(buf, data, (visflags & VIS_HTTPSTYLE)));
    653   1.8      seb }
    654   1.8      seb 
    655   1.8      seb void
    656   1.8      seb parse_encode_decode_arg(const char *arg, int encode)
    657   1.8      seb {
    658   1.8      seb 	if (! arg[0] || arg[1])
    659   1.8      seb 		return;
    660   1.8      seb 	if (arg[0] == 'k' || arg[0] == 'b') {
    661   1.8      seb 		if (encode)
    662   1.8      seb 			flags |= F_ENCODE_KEY;
    663   1.8      seb 		else
    664   1.8      seb 			flags |= F_DECODE_KEY;
    665   1.8      seb 	}
    666   1.8      seb 	if (arg[0] == 'v' || arg[0] == 'b') {
    667   1.8      seb 		if (encode)
    668   1.8      seb 			flags |= F_ENCODE_VAL;
    669   1.8      seb 		else
    670   1.8      seb 			flags |= F_DECODE_VAL;
    671   1.8      seb 	}
    672   1.8      seb 	return;
    673   1.8      seb }
    674   1.8      seb 
    675   1.8      seb int
    676   1.8      seb parse_encode_option(char **arg)
    677   1.8      seb {
    678   1.8      seb 	int	r = 0;
    679  1.17    lukem 	int	encmask = ~(VIS_CSTYLE | VIS_HTTPSTYLE | VIS_OCTAL);
    680   1.8      seb 
    681   1.8      seb 	for(; **arg; (*arg)++) {
    682   1.8      seb 		switch (**arg) {
    683   1.8      seb 			case 'b':
    684   1.8      seb 				r |= VIS_NOSLASH;
    685   1.8      seb 				break;
    686   1.8      seb 			case 'c':
    687  1.17    lukem 				r &= encmask;
    688   1.8      seb 				r |= VIS_CSTYLE;
    689   1.8      seb 				break;
    690  1.17    lukem 			case 'h':
    691  1.17    lukem 				r &= encmask;
    692  1.17    lukem 				r |= VIS_HTTPSTYLE;
    693  1.17    lukem 				break;
    694   1.8      seb 			case 'o':
    695  1.17    lukem 				r &= encmask;
    696   1.8      seb 				r |= VIS_OCTAL;
    697   1.8      seb 				break;
    698   1.8      seb 			case 's':
    699   1.8      seb 				r |= VIS_SAFE;
    700   1.8      seb 				break;
    701   1.8      seb 			case 't':
    702   1.8      seb 				r |= VIS_TAB;
    703   1.8      seb 				break;
    704   1.8      seb 			case 'w':
    705   1.8      seb 				r |= VIS_WHITE;
    706   1.8      seb 				break;
    707   1.8      seb 			default:
    708   1.8      seb 				return (0);
    709   1.8      seb 				break;
    710   1.8      seb 		}
    711   1.8      seb 	}
    712   1.8      seb 	return (r);
    713   1.8      seb }
    714   1.8      seb 
    715   1.1    lukem void
    716   1.1    lukem usage(void)
    717   1.1    lukem {
    718   1.1    lukem 	const char *p = getprogname();
    719   1.1    lukem 
    720   1.1    lukem 	fprintf(stderr,
    721  1.24    lukem "usage: %s    [-DKiNqV] [-E endian] [-f infile] [-O outsep] [-S visitem]\n"
    722  1.24    lukem "             [-T visspec] [-U unvisitem] [-X extravis] type dbfile [key [...]]\n"
    723  1.19    lukem "       %s -d [-iNq] [-E endian] [-f infile] [-T visspec] [-U unvisitem]\n"
    724  1.14    lukem "             type dbfile [key [...]]\n"
    725  1.14    lukem "       %s -w [-CDiNqR] [-E endian] [-F isep] [-f infile] [-m mode]\n"
    726  1.20    lukem "             [-P pagesize] [-T visspec] [-U unvisitem]\n"
    727  1.20    lukem "             type dbfile [key value [...]]\n"
    728   1.1    lukem 	    ,p ,p ,p );
    729   1.1    lukem 	fprintf(stderr,
    730  1.14    lukem "Supported modes:\n"
    731  1.14    lukem "                read keys  [default]\n"
    732  1.14    lukem "   -d           delete keys\n"
    733  1.14    lukem "   -w           write (add) keys/values\n"
    734  1.14    lukem "Supported options:\n"
    735  1.14    lukem "   -C           create empty (truncated) database\n"
    736  1.14    lukem "   -D           allow duplicates\n"
    737  1.14    lukem "   -E endian    database endian: `B'ig, `L'ittle, `H'ost  [default: H]\n"
    738  1.20    lukem "   -F isep      input field separator string  [default: a space]\n"
    739  1.14    lukem "   -f infile    file of keys (read|delete) or keys/vals (write)\n"
    740  1.14    lukem "   -i           ignore case of key by converting to lower case\n"
    741  1.14    lukem "   -K           print key\n"
    742  1.14    lukem "   -m mode      mode of created database  [default: 0644]\n"
    743  1.14    lukem "   -N           don't NUL terminate key\n"
    744  1.20    lukem "   -O outsep    output field separator string  [default: a tab]\n"
    745  1.20    lukem "   -P pagesize  database page size  [default: 4096]\n"
    746  1.14    lukem "   -q           quiet operation (missing keys aren't errors)\n"
    747  1.14    lukem "   -R           replace existing keys\n"
    748  1.14    lukem "   -S visitem   items to strvis(3) encode: 'k'ey, 'v'alue, 'b'oth\n"
    749  1.19    lukem "   -T visspec   options to control -S and -U; like vis(1) options\n"
    750  1.14    lukem "   -U unvisitem items to strunvis(3) decode: 'k'ey, 'v'alue, 'b'oth\n"
    751  1.14    lukem "   -V           print value\n"
    752  1.14    lukem "   -X extravis  extra characters to encode with -S\n"
    753   1.1    lukem 	    );
    754   1.1    lukem 	exit(1);
    755   1.1    lukem }
    756