load_hash.c revision 170268
1/* $FreeBSD: head/contrib/ipfilter/lib/load_hash.c 170268 2007-06-04 02:54:36Z darrenr $ */ 2 3/* 4 * Copyright (C) 2002-2005 by Darren Reed. 5 * 6 * See the IPFILTER.LICENCE file for details on licencing. 7 * 8 * $Id: load_hash.c,v 1.11.2.5 2006/07/14 06:12:25 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_list = NULL; 66 iph.iph_ref = 0; 67 68 if ((opts & OPT_REMOVE) == 0) { 69 if ((*iocfunc)(hashfd, SIOCLOOKUPADDTABLE, &op)) 70 if ((opts & OPT_DONOTHING) == 0) { 71 perror("load_hash:SIOCLOOKUPADDTABLE"); 72 return -1; 73 } 74 } 75 76 strncpy(iph.iph_name, op.iplo_name, sizeof(op.iplo_name)); 77 strncpy(iphp->iph_name, op.iplo_name, sizeof(op.iplo_name)); 78 79 if (opts & OPT_VERBOSE) { 80 for (a = list; a != NULL; a = a->ipe_next) { 81 a->ipe_addr.in4_addr = ntohl(a->ipe_addr.in4_addr); 82 a->ipe_mask.in4_addr = ntohl(a->ipe_mask.in4_addr); 83 } 84 iph.iph_table = calloc(size, sizeof(*iph.iph_table)); 85 if (iph.iph_table == NULL) { 86 perror("calloc(size, sizeof(*iph.iph_table))"); 87 return -1; 88 } 89 iph.iph_list = list; 90 printhash(&iph, bcopywrap, iph.iph_name, opts); 91 free(iph.iph_table); 92 iph.iph_list = NULL; 93 94 for (a = list; a != NULL; a = a->ipe_next) { 95 a->ipe_addr.in4_addr = htonl(a->ipe_addr.in4_addr); 96 a->ipe_mask.in4_addr = htonl(a->ipe_mask.in4_addr); 97 } 98 } 99 100 if (opts & OPT_DEBUG) 101 printf("Hash %s:\n", iph.iph_name); 102 103 for (a = list; a != NULL; a = a->ipe_next) 104 load_hashnode(iphp->iph_unit, iph.iph_name, a, iocfunc); 105 106 if ((opts & OPT_REMOVE) != 0) { 107 if ((*iocfunc)(hashfd, SIOCLOOKUPDELTABLE, &op)) 108 if ((opts & OPT_DONOTHING) == 0) { 109 perror("load_hash:SIOCLOOKUPDELTABLE"); 110 return -1; 111 } 112 } 113 return 0; 114} 115