load_hash.c revision 153881
1145519Sdarrenr/* $FreeBSD: head/contrib/ipfilter/lib/load_hash.c 153881 2005-12-30 11:52:26Z guido $ */ 2145510Sdarrenr 3145510Sdarrenr/* 4145510Sdarrenr * Copyright (C) 2002 by Darren Reed. 5145510Sdarrenr * 6145510Sdarrenr * See the IPFILTER.LICENCE file for details on licencing. 7145510Sdarrenr * 8153881Sguido * $Id: load_hash.c,v 1.11.2.3 2005/11/13 15:41:12 darrenr Exp $ 9145510Sdarrenr */ 10145510Sdarrenr 11145510Sdarrenr#include <fcntl.h> 12145510Sdarrenr#include <sys/ioctl.h> 13145510Sdarrenr#include "ipf.h" 14145510Sdarrenr#include "netinet/ip_lookup.h" 15145510Sdarrenr#include "netinet/ip_htable.h" 16145510Sdarrenr 17145510Sdarrenrstatic int hashfd = -1; 18145510Sdarrenr 19145510Sdarrenr 20145510Sdarrenrint load_hash(iphp, list, iocfunc) 21145510Sdarrenriphtable_t *iphp; 22145510Sdarrenriphtent_t *list; 23145510Sdarrenrioctlfunc_t iocfunc; 24145510Sdarrenr{ 25145510Sdarrenr iplookupop_t op; 26145510Sdarrenr iphtable_t iph; 27145510Sdarrenr iphtent_t *a; 28145510Sdarrenr size_t size; 29145510Sdarrenr int n; 30145510Sdarrenr 31145510Sdarrenr if ((hashfd == -1) && ((opts & OPT_DONOTHING) == 0)) 32145510Sdarrenr hashfd = open(IPLOOKUP_NAME, O_RDWR); 33145510Sdarrenr if ((hashfd == -1) && ((opts & OPT_DONOTHING) == 0)) 34145510Sdarrenr return -1; 35145510Sdarrenr 36145510Sdarrenr for (n = 0, a = list; a != NULL; a = a->ipe_next) 37145510Sdarrenr n++; 38145510Sdarrenr 39145510Sdarrenr op.iplo_arg = 0; 40145510Sdarrenr op.iplo_type = IPLT_HASH; 41145510Sdarrenr op.iplo_unit = iphp->iph_unit; 42145510Sdarrenr strncpy(op.iplo_name, iphp->iph_name, sizeof(op.iplo_name)); 43145510Sdarrenr if (*op.iplo_name == '\0') 44145510Sdarrenr op.iplo_arg = IPHASH_ANON; 45145510Sdarrenr op.iplo_size = sizeof(iph); 46145510Sdarrenr op.iplo_struct = &iph; 47145510Sdarrenr iph.iph_unit = iphp->iph_unit; 48145510Sdarrenr iph.iph_type = iphp->iph_type; 49145510Sdarrenr strncpy(iph.iph_name, iphp->iph_name, sizeof(iph.iph_name)); 50145510Sdarrenr iph.iph_flags = iphp->iph_flags; 51145510Sdarrenr if (n <= 0) 52145510Sdarrenr n = 1; 53145510Sdarrenr if (iphp->iph_size == 0) 54145510Sdarrenr size = n * 2 - 1; 55145510Sdarrenr else 56145510Sdarrenr size = iphp->iph_size; 57145510Sdarrenr if ((list == NULL) && (size == 1)) { 58145510Sdarrenr fprintf(stderr, 59145510Sdarrenr "WARNING: empty hash table %s, recommend setting %s\n", 60145510Sdarrenr iphp->iph_name, "size to match expected use"); 61145510Sdarrenr } 62145510Sdarrenr iph.iph_size = size; 63145510Sdarrenr iph.iph_seed = iphp->iph_seed; 64145510Sdarrenr iph.iph_table = NULL; 65145510Sdarrenr iph.iph_ref = 0; 66145510Sdarrenr 67145510Sdarrenr if ((opts & OPT_REMOVE) == 0) { 68145510Sdarrenr if ((*iocfunc)(hashfd, SIOCLOOKUPADDTABLE, &op)) 69145510Sdarrenr if ((opts & OPT_DONOTHING) == 0) { 70145510Sdarrenr perror("load_hash:SIOCLOOKUPADDTABLE"); 71145510Sdarrenr return -1; 72145510Sdarrenr } 73145510Sdarrenr } 74145510Sdarrenr 75153881Sguido strncpy(iph.iph_name, op.iplo_name, sizeof(op.iplo_name)); 76153881Sguido strncpy(iphp->iph_name, op.iplo_name, sizeof(op.iplo_name)); 77145510Sdarrenr 78145510Sdarrenr if (opts & OPT_VERBOSE) { 79145510Sdarrenr for (a = list; a != NULL; a = a->ipe_next) { 80145510Sdarrenr a->ipe_addr.in4_addr = ntohl(a->ipe_addr.in4_addr); 81145510Sdarrenr a->ipe_mask.in4_addr = ntohl(a->ipe_mask.in4_addr); 82145510Sdarrenr } 83145510Sdarrenr iph.iph_table = calloc(size, sizeof(*iph.iph_table)); 84145510Sdarrenr if (iph.iph_table == NULL) { 85145510Sdarrenr perror("calloc(size, sizeof(*iph.iph_table))"); 86145510Sdarrenr return -1; 87145510Sdarrenr } 88145510Sdarrenr iph.iph_table[0] = list; 89145510Sdarrenr printhash(&iph, bcopywrap, iph.iph_name, opts); 90145510Sdarrenr free(iph.iph_table); 91145510Sdarrenr 92145510Sdarrenr for (a = list; a != NULL; a = a->ipe_next) { 93145510Sdarrenr a->ipe_addr.in4_addr = htonl(a->ipe_addr.in4_addr); 94145510Sdarrenr a->ipe_mask.in4_addr = htonl(a->ipe_mask.in4_addr); 95145510Sdarrenr } 96145510Sdarrenr } 97145510Sdarrenr 98145510Sdarrenr if (opts & OPT_DEBUG) 99145510Sdarrenr printf("Hash %s:\n", iph.iph_name); 100145510Sdarrenr 101145510Sdarrenr for (a = list; a != NULL; a = a->ipe_next) 102145510Sdarrenr load_hashnode(iphp->iph_unit, iph.iph_name, a, iocfunc); 103145510Sdarrenr 104145510Sdarrenr if ((opts & OPT_REMOVE) != 0) { 105145510Sdarrenr if ((*iocfunc)(hashfd, SIOCLOOKUPDELTABLE, &op)) 106145510Sdarrenr if ((opts & OPT_DONOTHING) == 0) { 107145510Sdarrenr perror("load_hash:SIOCLOOKUPDELTABLE"); 108145510Sdarrenr return -1; 109145510Sdarrenr } 110145510Sdarrenr } 111145510Sdarrenr return 0; 112145510Sdarrenr} 113