ex_access.c

/*-
 * See the file LICENSE for redistribution information.
 *
 * Copyright (c) 1997-2005
 *      Sleepycat Software.  All rights reserved.
 *
 * $Id: ex_access.c,v 12.1 2005/06/16 20:22:00 bostic Exp $
 */

#include <sys/types.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#ifdef _WIN32
extern int getopt(int, char * const *, const char *);
#else
#include <unistd.h>
#endif

#include <db.h>

#define DATABASE        "access.db"
int main __P((int, char *[]));
int usage __P((void));

int
main(argc, argv)
        int argc;
        char *argv[];
{
        extern int optind;
        DB *dbp;
        DBC *dbcp;
        DBT key, data;
        size_t len;
        int ch, ret, rflag;
        char *database, *p, *t, buf[1024], rbuf[1024];
        const char *progname = "ex_access";             /* Program name. */

        rflag = 0;
        while ((ch = getopt(argc, argv, "r")) != EOF)
                switch (ch) {
                case 'r':
                        rflag = 1;
                        break;
                case '?':
                default:
                        return (usage());
                }
        argc -= optind;
        argv += optind;

        /* Accept optional database name. */
        database = *argv == NULL ? DATABASE : argv[0];

        /* Optionally discard the database. */
        if (rflag)
                (void)remove(database);

        /* Create and initialize database object, open the database. */
        if ((ret = db_create(&dbp, NULL, 0)) != 0) {
                fprintf(stderr,
                    "%s: db_create: %s\n", progname, db_strerror(ret));
                return (EXIT_FAILURE);
        }
        dbp->set_errfile(dbp, stderr);
        dbp->set_errpfx(dbp, progname);
        if ((ret = dbp->set_pagesize(dbp, 1024)) != 0) {
                dbp->err(dbp, ret, "set_pagesize");
                goto err1;
        }
        if ((ret = dbp->set_cachesize(dbp, 0, 32 * 1024, 0)) != 0) {
                dbp->err(dbp, ret, "set_cachesize");
                goto err1;
        }
        if ((ret = dbp->open(dbp,
            NULL, database, NULL, DB_BTREE, DB_CREATE, 0664)) != 0) {
                dbp->err(dbp, ret, "%s: open", database);
                goto err1;
        }

        /*
         * Insert records into the database, where the key is the user
         * input and the data is the user input in reverse order.
         */
        memset(&key, 0, sizeof(DBT));
        memset(&data, 0, sizeof(DBT));
        for (;;) {
                printf("input> ");
                fflush(stdout);
                if (fgets(buf, sizeof(buf), stdin) == NULL)
                        break;
                if (strcmp(buf, "exit\n") == 0 || strcmp(buf, "quit\n") == 0)
                        break;
                if ((len = strlen(buf)) <= 1)
                        continue;
                for (t = rbuf, p = buf + (len - 2); p >= buf;)
                        *t++ = *p--;
                *t++ = '\0';

                key.data = buf;
                data.data = rbuf;
                data.size = key.size = (u_int32_t)len - 1;

                switch (ret =
                    dbp->put(dbp, NULL, &key, &data, DB_NOOVERWRITE)) {
                case 0:
                        break;
                default:
                        dbp->err(dbp, ret, "DB->put");
                        if (ret != DB_KEYEXIST)
                                goto err1;
                        break;
                }
        }
        printf("\n");

        /* Acquire a cursor for the database. */
        if ((ret = dbp->cursor(dbp, NULL, &dbcp, 0)) != 0) {
                dbp->err(dbp, ret, "DB->cursor");
                goto err1;
        }

        /* Initialize the key/data pair so the flags aren't set. */
        memset(&key, 0, sizeof(key));
        memset(&data, 0, sizeof(data));

        /* Walk through the database and print out the key/data pairs. */
        while ((ret = dbcp->c_get(dbcp, &key, &data, DB_NEXT)) == 0)
                printf("%.*s : %.*s\n",
                    (int)key.size, (char *)key.data,
                    (int)data.size, (char *)data.data);
        if (ret != DB_NOTFOUND) {
                dbp->err(dbp, ret, "DBcursor->get");
                goto err2;
        }

        /* Close everything down. */
        if ((ret = dbcp->c_close(dbcp)) != 0) {
                dbp->err(dbp, ret, "DBcursor->close");
                goto err1;
        }
        if ((ret = dbp->close(dbp, 0)) != 0) {
                fprintf(stderr,
                    "%s: DB->close: %s\n", progname, db_strerror(ret));
                return (EXIT_FAILURE);
        }
        return (EXIT_SUCCESS);

err2:   (void)dbcp->c_close(dbcp);
err1:   (void)dbp->close(dbp, 0);
        return (EXIT_FAILURE);
}

int
usage()
{
        (void)fprintf(stderr, "usage: ex_access [-r] [database]\n");
        return (EXIT_FAILURE);
}

---