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