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