Changeset View
Changeset View
Standalone View
Standalone View
contrib/blacklist/bin/state.c
- This file was added.
Property | Old Value | New Value |
---|---|---|
svn:eol-style | null | native \ No newline at end of property |
svn:keywords | null | FreeBSD=%H \ No newline at end of property |
svn:mime-type | null | text/plain \ No newline at end of property |
/* $NetBSD: state.c,v 1.18 2016/04/04 15:52:56 christos 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. | |||||
*/ | |||||
#ifdef HAVE_CONFIG_H | |||||
#include "config.h" | |||||
#endif | |||||
#include <sys/cdefs.h> | |||||
__RCSID("$NetBSD: state.c,v 1.18 2016/04/04 15:52:56 christos Exp $"); | |||||
#include <sys/types.h> | |||||
#include <sys/socket.h> | |||||
#include <stdio.h> | |||||
#include <string.h> | |||||
#include <errno.h> | |||||
#include <fcntl.h> | |||||
#include <syslog.h> | |||||
#include <netinet/in.h> | |||||
#include "bl.h" | |||||
#include "internal.h" | |||||
#include "conf.h" | |||||
#include "support.h" | |||||
#include "state.h" | |||||
static HASHINFO openinfo = { | |||||
4096, /* bsize */ | |||||
32, /* ffactor */ | |||||
256, /* nelem */ | |||||
8 * 1024 * 1024,/* cachesize */ | |||||
NULL, /* hash() */ | |||||
0 /* lorder */ | |||||
}; | |||||
int | |||||
state_close(DB *db) | |||||
{ | |||||
if (db == NULL) | |||||
return -1; | |||||
if ((*db->close)(db) == -1) { | |||||
(*lfun)(LOG_ERR, "%s: can't close db (%m)", __func__); | |||||
return -1; | |||||
} | |||||
return 0; | |||||
} | |||||
DB * | |||||
state_open(const char *dbname, int flags, mode_t perm) | |||||
{ | |||||
DB *db; | |||||
#ifdef __APPLE__ | |||||
flags &= O_CREAT|O_EXCL|O_EXLOCK|O_NONBLOCK|O_RDONLY| | |||||
O_RDWR|O_SHLOCK|O_TRUNC; | |||||
#endif | |||||
db = dbopen(dbname, flags, perm, DB_HASH, &openinfo); | |||||
if (db == NULL) { | |||||
if (errno == ENOENT && (flags & O_CREAT) == 0) | |||||
return NULL; | |||||
(*lfun)(LOG_ERR, "%s: can't open `%s' (%m)", __func__, dbname); | |||||
} | |||||
return db; | |||||
} | |||||
static int | |||||
state_sizecheck(const DBT *t) | |||||
{ | |||||
if (sizeof(struct conf) == t->size) | |||||
return 0; | |||||
(*lfun)(LOG_ERR, "Key size mismatch %zu != %zu", sizeof(struct conf), | |||||
t->size); | |||||
return -1; | |||||
} | |||||
static void | |||||
dumpkey(const struct conf *k) | |||||
{ | |||||
char buf[10240]; | |||||
blhexdump(buf, sizeof(buf), __func__, k, sizeof(*k)); | |||||
(*lfun)(LOG_DEBUG, "%s", buf); | |||||
(*lfun)(LOG_DEBUG, "%s: %s", __func__, | |||||
conf_print(buf, sizeof(buf), "", "", k)); | |||||
} | |||||
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, 0)) { | |||||
case 0: | |||||
case 1: | |||||
if (debug > 1) { | |||||
(*lfun)(LOG_DEBUG, "%s: returns %d", __func__, rv); | |||||
(*db->sync)(db, 0); | |||||
} | |||||
return 0; | |||||
default: | |||||
(*lfun)(LOG_ERR, "%s: failed (%m)", __func__); | |||||
return -1; | |||||
} | |||||
} | |||||
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) | |||||
(*lfun)(LOG_DEBUG, "%s: returns %d", __func__, rv); | |||||
return 0; | |||||
default: | |||||
(*lfun)(LOG_ERR, "%s: failed (%m)", __func__); | |||||
return -1; | |||||
} | |||||
} | |||||
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) { | |||||
(*lfun)(LOG_DEBUG, "%s: returns %d", __func__, rv); | |||||
(*db->sync)(db, 0); | |||||
} | |||||
return 0; | |||||
case 1: | |||||
errno = EEXIST; | |||||
/*FALLTHROUGH*/ | |||||
default: | |||||
(*lfun)(LOG_ERR, "%s: failed (%m)", __func__); | |||||
return -1; | |||||
} | |||||
} | |||||
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)); | |||||
if (debug > 2) | |||||
dumpkey(c); | |||||
memcpy(dbi, v.data, sizeof(*dbi)); | |||||
if (debug > 1) | |||||
(*lfun)(LOG_DEBUG, "%s: returns %d", __func__, rv); | |||||
return 1; | |||||
case 1: | |||||
if (debug > 1) | |||||
(*lfun)(LOG_DEBUG, "%s: returns %d", __func__, rv); | |||||
return 0; | |||||
default: | |||||
(*lfun)(LOG_ERR, "%s: failed (%m)", __func__); | |||||
return -1; | |||||
} | |||||
} | |||||
int | |||||
state_sync(DB *db) | |||||
{ | |||||
return (*db->sync)(db, 0); | |||||
} |