/*      $NetBSD: t_db_hash_seq.c,v 1.4 2020/09/07 00:28:44 mrg Exp $    */

/*-
 * Copyright (c) 2015 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Christos Zoulas.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */
#include <sys/cdefs.h>
__RCSID("$NetBSD: t_db_hash_seq.c,v 1.4 2020/09/07 00:28:44 mrg Exp $");

#include <sys/types.h>
#include <sys/socket.h>
#include <db.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <stdlib.h>
#include <fcntl.h>
#include <syslog.h>
#include <netinet/in.h>

#define ATF

struct conf {
        struct sockaddr_storage c_ss;
        int                     c_lmask;
        int                     c_port;
        int                     c_proto;
        int                     c_family;
        int                     c_uid;
        int                     c_nfail;
        char                    c_name[128];
        int                     c_rmask;
        int                     c_duration;
};

struct dbinfo {
        int count;
        time_t last;
        char id[64];
};

#ifdef ATF
#include <atf-c.h>

#define DO_ERR(msg, ...)        ATF_REQUIRE_MSG(0, msg, __VA_ARGS__)
#define DO_WARNX(msg, ...)      ATF_REQUIRE_MSG(0, msg, __VA_ARGS__)
#else
#include <err.h>

#define DO_ERR(fmt, ...)        err(EXIT_FAILURE, fmt, __VA_ARGS__)
#define DO_WARNX(fmt, ...)      warnx(fmt, __VA_ARGS__)
#endif

#define DO_DEBUG(fmt, ...)      fprintf(stderr, fmt, __VA_ARGS__)

static HASHINFO openinfo = {
        4096,           /* bsize */
        32,             /* ffactor */
        256,            /* nelem */
        8 * 1024 * 1024,/* cachesize */
        NULL,           /* hash() */
        0               /* lorder */
};

static int debug = 0;

static int
state_close(DB *db)
{
        if (db == NULL)
                return -1;
        if ((*db->close)(db) == -1)
                DO_ERR("%s: can't close db", __func__);
        return 0;
}

static DB *
state_open(const char *dbname, int flags, mode_t perm)
{
        DB *db;

        db = dbopen(dbname, flags, perm, DB_HASH, &openinfo);
        if (db == NULL) {
                if (errno == ENOENT && (flags & O_CREAT) == 0)
                        return NULL;
                DO_ERR("%s: can't open `%s'", __func__,
                    dbname ? dbname : "<memory>");
        }
        return db;
}

static int
state_sizecheck(const DBT *t)
{
        if (sizeof(struct conf) == t->size)
                return 0;
        DO_WARNX("Key size mismatch %zu != %zu", sizeof(struct conf), t->size);
        return 0;
}

static int
state_del(DB *db, const struct conf *c)
{
        int rv;
        DBT k;

        if (db == NULL)
                return -1;

        k.data = __UNCONST(c);
        k.size = sizeof(*c);

        switch (rv = (*db->del)(db, &k, 1)) {
        case 0:
        case 1:
                if (debug > 1) {
                        DO_DEBUG("%s: returns %d", __func__, rv);
                        (*db->sync)(db, 0);
                }
                return 0;
        default:
                DO_ERR("%s: failed", __func__);
                return -1;
        }
}

#if 0
static int
state_get(DB *db, const struct conf *c, struct dbinfo *dbi)
{
        int rv;
        DBT k, v;

        if (db == NULL)
                return -1;

        k.data = __UNCONST(c);
        k.size = sizeof(*c);

        switch (rv = (*db->get)(db, &k, &v, 0)) {
        case 0:
        case 1:
                if (rv)
                        memset(dbi, 0, sizeof(*dbi));
                else
                        memcpy(dbi, v.data, sizeof(*dbi));
                if (debug > 1)
                        DO_DEBUG("%s: returns %d", __func__, rv);
                return 0;
        default:
                DO_ERR("%s: failed", __func__);
                return -1;
        }
}
#endif

static int
state_put(DB *db, const struct conf *c, const struct dbinfo *dbi)
{
        int rv;
        DBT k, v;

        if (db == NULL)
                return -1;

        k.data = __UNCONST(c);
        k.size = sizeof(*c);
        v.data = __UNCONST(dbi);
        v.size = sizeof(*dbi);

        switch (rv = (*db->put)(db, &k, &v, 0)) {
        case 0:
                if (debug > 1) {
                        DO_DEBUG("%s: returns %d", __func__, rv);
                        (*db->sync)(db, 0);
                }
                return 0;
        case 1:
                errno = EEXIST;
                /*FALLTHROUGH*/
        default:
                DO_ERR("%s: failed", __func__);
        }
}

static int
state_iterate(DB *db, struct conf *c, struct dbinfo *dbi, unsigned int first)
{
        int rv;
        DBT k, v;

        if (db == NULL)
                return -1;

        first = first ? R_FIRST : R_NEXT;

        switch (rv = (*db->seq)(db, &k, &v, first)) {
        case 0:
                if (state_sizecheck(&k) == -1)
                        return -1;
                memcpy(c, k.data, sizeof(*c));
                memcpy(dbi, v.data, sizeof(*dbi));
                if (debug > 1)
                        DO_DEBUG("%s: returns %d", __func__, rv);
                return 1;
        case 1:
                if (debug > 1)
                        DO_DEBUG("%s: returns %d", __func__, rv);
                return 0;
        default:
                DO_ERR("%s: failed", __func__);
                return -1;
        }
}

#define MAXB 100

static int
testdb(int skip)
{
        size_t i;
        int f;
        char flag[MAXB];
        DB *db;
        struct conf c;
        struct dbinfo d;

        db = state_open(NULL, O_RDWR|O_CREAT|O_TRUNC, 0600);
        if (db == NULL)
                DO_ERR("%s: cannot open `%s'", __func__, "foo");

        memset(&c, 0, sizeof(c));
        memset(&d, 0, sizeof(d));
        memset(flag, 0, sizeof(flag));

        for (i = 0; i < __arraycount(flag); i++) {
                c.c_port = i;
                state_put(db, &c, &d);
        }

        for (f = 1, i = 0; state_iterate(db, &c, &d, f) == 1; f = 0, i++) {
                if (debug > 1)
                        DO_DEBUG("%zu %d\n", i, c.c_port);
                if (flag[c.c_port])
                        DO_WARNX("Already visited %d", c.c_port);
                flag[c.c_port] = 1;
                if (skip == 0 || c.c_port % skip != 0)
                        continue;
                state_del(db, &c);
        }
        state_close(db);
        for (i = 0; i < __arraycount(flag); i++) {
                if (flag[i] == 0)
                        DO_WARNX("Not visited %zu", i);
        }
        return 0;
}

#ifndef ATF
int
main(int argc, char *argv[])
{
        return testdb(6);
}
#else

ATF_TC(test_hash_del_none);
ATF_TC_HEAD(test_hash_del_none, tc)
{
        atf_tc_set_md_var(tc, "descr", "Check sequential scan of hash tables deleting none");
}

ATF_TC_BODY(test_hash_del_none, tc)
{
        testdb(0);
}

ATF_TC(test_hash_del_all);
ATF_TC_HEAD(test_hash_del_all, tc)
{
        atf_tc_set_md_var(tc, "descr", "Check sequential scan of hash tables deleting all");
}

ATF_TC_BODY(test_hash_del_all, tc)
{
        testdb(1);
}

ATF_TC(test_hash_del_alt);
ATF_TC_HEAD(test_hash_del_alt, tc)
{
        atf_tc_set_md_var(tc, "descr", "Check sequential scan of hash tables alternating deletets");
}

ATF_TC_BODY(test_hash_del_alt, tc)
{
        testdb(2);
}

ATF_TC(test_hash_del_every_7);
ATF_TC_HEAD(test_hash_del_every_7, tc)
{
        atf_tc_set_md_var(tc, "descr", "Check sequential scan of hash tables deleting every 7 elements");
}

ATF_TC_BODY(test_hash_del_every_7, tc)
{
        testdb(7);
}

ATF_TP_ADD_TCS(tp)
{
        ATF_TP_ADD_TC(tp, test_hash_del_none);
        ATF_TP_ADD_TC(tp, test_hash_del_all);
        ATF_TP_ADD_TC(tp, test_hash_del_alt);
        ATF_TP_ADD_TC(tp, test_hash_del_every_7);

        return atf_no_error();
}
#endif