#include <sys/param.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/malloc.h>
#include <sys/socket.h>
#include <net/hostcache.h>
#include <net/route.h>
MALLOC_DEFINE(M_HOSTCACHE, "hostcache", "per-host cache structure");
static struct hctable hctable[AF_MAX];
static int hc_timeout_interval = 120;
static int hc_maxidle = 1800;
static int cmpsa(const struct sockaddr *sa1, const struct sockaddr *sa2);
static void hc_timeout(void *xhct);
static void maybe_bump_hash(struct hctable *hct);
int
hc_init(int af, struct hccallback *hccb, int init_nelem, int primes)
{
struct hctable *hct;
struct hchead *heads;
u_long nelem;
hct = &hctable[af];
nelem = init_nelem;
if (hct->hct_nentries)
return 0;
if (primes) {
heads = phashinit(init_nelem, M_HOSTCACHE, &nelem);
} else {
int i;
MALLOC(heads, struct hchead *, nelem * sizeof *heads,
M_HOSTCACHE, M_WAITOK);
for (i = 0; i < nelem; i++) {
LIST_INIT(&heads[i]);
}
}
hct->hct_heads = heads;
hct->hct_nentries = nelem;
hct->hct_primes = primes;
timeout(hc_timeout, hct, hc_timeout_interval * hz);
return 0;
}
struct hcentry *
hc_get(struct sockaddr *sa)
{
u_long hash;
struct hcentry *hc;
struct hctable *hct;
int s;
hct = &hctable[sa->sa_family];
if (hct->hct_nentries == 0)
return 0;
hash = hct->hct_cb->hccb_hash(sa, hct->hct_nentries);
hc = hct->hct_heads[hash].lh_first;
for (; hc; hc = hc->hc_link.le_next) {
if (cmpsa(hc->hc_host, sa) == 0)
break;
}
if (hc == 0)
return 0;
s = splnet();
if (hc->hc_rt && (hc->hc_rt->rt_flags & RTF_UP) == 0) {
RTFREE(hc->hc_rt);
hc->hc_rt = 0;
}
if (hc->hc_rt == 0) {
hc->hc_rt = rtalloc1(hc->hc_host, 1, 0);
}
hc_ref(hc);
splx(s);
return hc;
}
void
hc_ref(struct hcentry *hc)
{
int s = splnet();
if (hc->hc_refcnt++ == 0) {
hc->hc_hct->hct_idle--;
hc->hc_hct->hct_active++;
}
splx(s);
}
void
hc_rele(struct hcentry *hc)
{
int s = splnet();
#ifdef DIAGNOSTIC
printf("hc_rele: %p: negative refcnt!\n", (void *)hc);
#endif
hc->hc_refcnt--;
if (hc->hc_refcnt == 0) {
hc->hc_hct->hct_idle++;
hc->hc_hct->hct_active--;
hc->hc_idlesince = mono_time;
}
splx(s);
}
int
hc_insert(struct hcentry *hc)
{
struct hcentry *hc2;
struct hctable *hct;
u_long hash;
int s;
hct = &hctable[hc->hc_host->sa_family];
hash = hct->hct_cb->hccb_hash(hc->hc_host, hct->hct_nentries);
hc2 = hct->hct_heads[hash].lh_first;
for (; hc2; hc2 = hc2->hc_link.le_next) {
if (cmpsa(hc2->hc_host, hc->hc_host) == 0)
break;
}
if (hc2 != 0)
return EEXIST;
hc->hc_hct = hct;
s = splnet();
LIST_INSERT_HEAD(&hct->hct_heads[hash], hc, hc_link);
hct->hct_idle++;
if (100 * (hct->hct_idle + hct->hct_active) > 75 * hct->hct_nentries)
maybe_bump_hash(hct);
splx(s);
return 0;
}
int
hc_delete(struct hcentry *hc)
{
struct hctable *hct;
int error, s;
if (hc->hc_refcnt > 0)
return 0;
hct = hc->hc_hct;
error = hct->hct_cb->hccb_delete(hc);
if (error)
return 0;
s = splnet();
LIST_REMOVE(hc, hc_link);
hc->hc_hct->hct_idle--;
splx(s);
FREE(hc, M_HOSTCACHE);
return 0;
}
static void
hc_timeout(void *xhct)
{
struct hcentry *hc;
struct hctable *hct;
int j, s;
time_t start;
hct = xhct;
start = mono_time.tv_sec;
if (hct->hct_idle == 0)
return;
for (j = 0; j < hct->hct_nentries; j++) {
for (hc = hct->hct_heads[j].lh_first; hc;
hc = hc->hc_link.le_next) {
if (hc->hc_refcnt > 0)
continue;
if (hc->hc_idlesince.tv_sec + hc_maxidle <= start) {
if (hct->hct_cb->hccb_delete(hc))
continue;
s = splnet();
LIST_REMOVE(hc, hc_link);
hct->hct_idle--;
splx(s);
}
}
}
timeout(hc_timeout, xhct, hc_timeout_interval * hz);
}
static int
cmpsa(const struct sockaddr *sa1, const struct sockaddr *sa2)
{
if (sa1->sa_len != sa2->sa_len)
return ((int)sa1->sa_len - sa2->sa_len);
return bcmp(sa1, sa2, sa1->sa_len);
}
static void
maybe_bump_hash(struct hctable *hct)
{
;
}