contrib/ipfilter/lib/load_hash.c

   1 /*      $FreeBSD$       */
   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
  17 static int hashfd = -1;
  18
  19
  20 int load_hash(iphp, list, iocfunc)
  21 iphtable_t *iphp;
  22 iphtent_t *list;
  23 ioctlfunc_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 }