diff options
author | Sergey Poznyakoff <gray@gnu.org.ua> | 2011-08-02 11:24:07 +0000 |
---|---|---|
committer | Sergey Poznyakoff <gray@gnu.org.ua> | 2011-08-02 11:24:07 +0000 |
commit | c17f57c42898b887949c61d9b384fa5a6aa2518a (patch) | |
tree | 8206d215dd848ebecca4f1832cca5ee4999f6750 /tests/gtfetch.c | |
parent | 44a6e9adbbf90b59e8fd32f236dce2fde2fc6a6e (diff) | |
download | gdbm-c17f57c42898b887949c61d9b384fa5a6aa2518a.tar.gz gdbm-c17f57c42898b887949c61d9b384fa5a6aa2518a.tar.bz2 |
Add tests.
Diffstat (limited to 'tests/gtfetch.c')
-rw-r--r-- | tests/gtfetch.c | 141 |
1 files changed, 141 insertions, 0 deletions
diff --git a/tests/gtfetch.c b/tests/gtfetch.c new file mode 100644 index 0000000..a50b83e --- /dev/null +++ b/tests/gtfetch.c @@ -0,0 +1,141 @@ +/* This file is part of GDBM test suite. + Copyright (C) 2011 Free Software Foundation, Inc. + + GDBM is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2, or (at your option) + any later version. + + GDBM is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with GDBM. If not, see <http://www.gnu.org/licenses/>. +*/ +#include "autoconf.h" +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include "gdbm.h" + +void +print_key (FILE *fp, datum key, int delim) +{ + size_t i; + + for (i = 0; i < key.dsize && key.dptr[i]; i++) + { + if (key.dptr[i] == delim || key.dptr[i] == '\\') + fputc ('\\', fp); + fputc (key.dptr[i], fp); + } +} + +const char * +canonical_progname (const char *str) +{ + const char *p; + + p = strrchr (str, '/'); + if (p) + p++; + else + p = str; + if (strncmp (p, "lt-", 3) == 0) + p += 3; + return p; +} + +int +main (int argc, char **argv) +{ + const char *progname = canonical_progname (argv[0]); + const char *dbname; + datum key; + datum data; + int flags = 0; + GDBM_FILE dbf; + int data_z = 0; + int delim = 0; + int rc = 0; + + while (--argc) + { + char *arg = *++argv; + + if (strcmp (arg, "-h") == 0) + { + printf ("usage: %s [-nolock] [-nommap] [-null] [-delim=CHR] DBFILE KEY [KEY...]\n", + progname); + exit (0); + } + else if (strcmp (arg, "-nolock") == 0) + flags |= GDBM_NOLOCK; + else if (strcmp (arg, "-nommap") == 0) + flags |= GDBM_NOMMAP; + else if (strcmp (arg, "-null") == 0) + data_z = 1; + else if (strncmp (arg, "-delim=", 7) == 0) + delim = arg[7]; + else if (strcmp (arg, "--") == 0) + { + --argc; + ++argv; + break; + } + else if (arg[0] == '-') + { + fprintf (stderr, "%s: unknown option %s\n", progname, arg); + exit (1); + } + else + break; + } + + if (argc < 2) + { + fprintf (stderr, "%s: wrong arguments\n", progname); + exit (1); + } + dbname = *argv; + + dbf = gdbm_open (dbname, 0, GDBM_READER|flags, 00664, NULL); + if (!dbf) + { + fprintf (stderr, "gdbm_open failed: %s", gdbm_strerror (gdbm_errno)); + exit (1); + } + + while (--argc) + { + char *arg = *++argv; + + key.dptr = arg; + key.dsize = strlen (arg) + !!data_z; + + data = gdbm_fetch (dbf, key); + if (data.dptr == NULL) + { + rc = 2; + fprintf (stderr, "%s: ", progname); + print_key (stderr, key, delim); + fprintf (stderr, ": not found\n"); + continue; + } + if (delim) + { + print_key (stdout, key, delim); + fputc (delim, stdout); + } + + fwrite (data.dptr, data.dsize - !!data_z, 1, stdout); + free (data.dptr); + + fputc ('\n', stdout); + } + + gdbm_close (dbf); + exit (rc); +} |