load_hash.c revision 153881 
1/*	$FreeBSD: head/contrib/ipfilter/lib/load_hash.c 153881 2005-12-30 11:52:26Z guido $	*/
2
3/*
4 * Copyright (C) 2002 by Darren Reed.
5 *
6 * See the IPFILTER.LICENCE file for details on licencing.
7 *
8 * $Id: load_hash.c,v 1.11.2.3 2005/11/13 15:41:12 darrenr Exp $
9 */
10
11#include <fcntl.h>
12#include <sys/ioctl.h>
13#include "ipf.h"
14#include "netinet/ip_lookup.h"
15#include "netinet/ip_htable.h"
16
17static int hashfd = -1;
18
19
20int load_hash(iphp, list, iocfunc)
21iphtable_t *iphp;
22iphtent_t *list;
23ioctlfunc_t iocfunc;
24{
25	iplookupop_t op;
26	iphtable_t iph;
27	iphtent_t *a;
28	size_t size;
29	int n;
30
31	if ((hashfd == -1) && ((opts & OPT_DONOTHING) == 0))
32		hashfd = open(IPLOOKUP_NAME, O_RDWR);
33	if ((hashfd == -1) && ((opts & OPT_DONOTHING) == 0))
34		return -1;
35
36	for (n = 0, a = list; a != NULL; a = a->ipe_next)
37		n++;
38
39	op.iplo_arg = 0;
40	op.iplo_type = IPLT_HASH;
41	op.iplo_unit = iphp->iph_unit;
42	strncpy(op.iplo_name, iphp->iph_name, sizeof(op.iplo_name));
43	if (*op.iplo_name == '\0')
44		op.iplo_arg = IPHASH_ANON;
45	op.iplo_size = sizeof(iph);
46	op.iplo_struct = &iph;
47	iph.iph_unit = iphp->iph_unit;
48	iph.iph_type = iphp->iph_type;
49	strncpy(iph.iph_name, iphp->iph_name, sizeof(iph.iph_name));
50	iph.iph_flags = iphp->iph_flags;
51	if (n <= 0)
52		n = 1;
53	if (iphp->iph_size == 0)
54		size = n * 2 - 1;
55	else
56		size = iphp->iph_size;
57	if ((list == NULL) && (size == 1)) {
58		fprintf(stderr,
59			"WARNING: empty hash table %s, recommend setting %s\n",
60			iphp->iph_name, "size to match expected use");
61	}
62	iph.iph_size = size;
63	iph.iph_seed = iphp->iph_seed;
64	iph.iph_table = NULL;
65	iph.iph_ref = 0;
66
67	if ((opts & OPT_REMOVE) == 0) {
68		if ((*iocfunc)(hashfd, SIOCLOOKUPADDTABLE, &op))
69			if ((opts & OPT_DONOTHING) == 0) {
70				perror("load_hash:SIOCLOOKUPADDTABLE");
71				return -1;
72			}
73	}
74
75	strncpy(iph.iph_name, op.iplo_name, sizeof(op.iplo_name));
76	strncpy(iphp->iph_name, op.iplo_name, sizeof(op.iplo_name));
77
78	if (opts & OPT_VERBOSE) {
79		for (a = list; a != NULL; a = a->ipe_next) {
80			a->ipe_addr.in4_addr = ntohl(a->ipe_addr.in4_addr);
81			a->ipe_mask.in4_addr = ntohl(a->ipe_mask.in4_addr);
82		}
83		iph.iph_table = calloc(size, sizeof(*iph.iph_table));
84		if (iph.iph_table == NULL) {
85			perror("calloc(size, sizeof(*iph.iph_table))");
86			return -1;
87		}
88		iph.iph_table[0] = list;
89		printhash(&iph, bcopywrap, iph.iph_name, opts);
90		free(iph.iph_table);
91
92		for (a = list; a != NULL; a = a->ipe_next) {
93			a->ipe_addr.in4_addr = htonl(a->ipe_addr.in4_addr);
94			a->ipe_mask.in4_addr = htonl(a->ipe_mask.in4_addr);
95		}
96	}
97
98	if (opts & OPT_DEBUG)
99		printf("Hash %s:\n", iph.iph_name);
100
101	for (a = list; a != NULL; a = a->ipe_next)
102		load_hashnode(iphp->iph_unit, iph.iph_name, a, iocfunc);
103
104	if ((opts & OPT_REMOVE) != 0) {
105		if ((*iocfunc)(hashfd, SIOCLOOKUPDELTABLE, &op))
106			if ((opts & OPT_DONOTHING) == 0) {
107				perror("load_hash:SIOCLOOKUPDELTABLE");
108				return -1;
109			}
110	}
111	return 0;
112}
113