1 /*        $NetBSD: ip_htable.c,v 1.7 2016/06/09 04:43:46 pgoyette Exp $         */
2 
3 /*
4  * Copyright (C) 2012 by Darren Reed.
5  *
6  * See the IPFILTER.LICENCE file for details on licencing.
7  */
8 #if defined(KERNEL) || defined(_KERNEL)
9 # undef KERNEL
10 # undef _KERNEL
11 # define        KERNEL        1
12 # define        _KERNEL       1
13 #endif
14 #include <sys/param.h>
15 #if defined(__NetBSD__)
16 # if (NetBSD >= 199905) && !defined(IPFILTER_LKM) && defined(_KERNEL)
17 #  if (__NetBSD_Version__ >= 799003000)
18 #   if defined(_KERNEL_OPT)
19 #    include "opt_ipfilter.h"
20 #   endif
21 #  else
22 #   include "opt_ipfilter.h"
23 #  endif
24 # endif
25 #endif
26 #include <sys/types.h>
27 #include <sys/errno.h>
28 #include <sys/time.h>
29 #include <sys/file.h>
30 #if !defined(_KERNEL)
31 # include <stdlib.h>
32 # include <string.h>
33 # define _KERNEL
34 # ifdef __OpenBSD__
35 struct file;
36 # endif
37 # include <sys/uio.h>
38 # undef _KERNEL
39 #endif
40 #include <sys/socket.h>
41 #if defined(__FreeBSD_version) && (__FreeBSD_version >= 300000)
42 # include <sys/malloc.h>
43 #endif
44 #if defined(__FreeBSD__)
45 #  include <sys/cdefs.h>
46 #  include <sys/proc.h>
47 #endif
48 #if !defined(__svr4__) && !defined(__SVR4) && !defined(__hpux) && \
49     !defined(linux)
50 # include <sys/mbuf.h>
51 #endif
52 #if defined(_KERNEL)
53 # include <sys/systm.h>
54 #else
55 # include "ipf.h"
56 #endif
57 #include <netinet/in.h>
58 #include <net/if.h>
59 
60 #include "netinet/ip_compat.h"
61 #include "netinet/ip_fil.h"
62 #include "netinet/ip_lookup.h"
63 #include "netinet/ip_htable.h"
64 /* END OF INCLUDES */
65 
66 #if !defined(lint)
67 #if defined(__NetBSD__)
68 #include <sys/cdefs.h>
69 __KERNEL_RCSID(0, "$NetBSD: ip_htable.c,v 1.7 2016/06/09 04:43:46 pgoyette Exp $");
70 #else
71 static const char rcsid[] = "@(#)Id: ip_htable.c,v 1.1.1.2 2012/07/22 13:45:19 darrenr Exp";
72 #endif
73 #endif
74 
75 # ifdef USE_INET6
76 static iphtent_t *ipf_iphmfind6(iphtable_t *, i6addr_t *);
77 # endif
78 static iphtent_t *ipf_iphmfind(iphtable_t *, struct in_addr *);
79 static int ipf_iphmfindip(ipf_main_softc_t *, void *, int, void *, u_int);
80 static int ipf_htable_clear(ipf_main_softc_t *, void *, iphtable_t *);
81 static int ipf_htable_create(ipf_main_softc_t *, void *, iplookupop_t *);
82 static int ipf_htable_deref(ipf_main_softc_t *, void *, void *);
83 static int ipf_htable_destroy(ipf_main_softc_t *, void *, int, char *);
84 static void *ipf_htable_exists(void *, int, char *);
85 static size_t ipf_htable_flush(ipf_main_softc_t *, void *,
86                                             iplookupflush_t *);
87 static void ipf_htable_free(void *, iphtable_t *);
88 static int ipf_htable_iter_deref(ipf_main_softc_t *, void *, int,
89                                               int, void *);
90 static int ipf_htable_iter_next(ipf_main_softc_t *, void *, ipftoken_t *,
91                                              ipflookupiter_t *);
92 static int ipf_htable_node_add(ipf_main_softc_t *, void *,
93                                             iplookupop_t *, int);
94 static int ipf_htable_node_del(ipf_main_softc_t *, void *,
95                                             iplookupop_t *, int);
96 static int ipf_htable_remove(ipf_main_softc_t *, void *, iphtable_t *);
97 static void *ipf_htable_soft_create(ipf_main_softc_t *);
98 static void ipf_htable_soft_destroy(ipf_main_softc_t *, void *);
99 static int ipf_htable_soft_init(ipf_main_softc_t *, void *);
100 static void ipf_htable_soft_fini(ipf_main_softc_t *, void *);
101 static int ipf_htable_stats_get(ipf_main_softc_t *, void *,
102                                              iplookupop_t *);
103 static int ipf_htable_table_add(ipf_main_softc_t *, void *,
104                                              iplookupop_t *);
105 static int ipf_htable_table_del(ipf_main_softc_t *, void *,
106                                              iplookupop_t *);
107 static int ipf_htent_deref(void *, iphtent_t *);
108 static iphtent_t *ipf_htent_find(iphtable_t *, iphtent_t *);
109 static int ipf_htent_insert(ipf_main_softc_t *, void *, iphtable_t *,
110                                          iphtent_t *);
111 static int ipf_htent_remove(ipf_main_softc_t *, void *, iphtable_t *,
112                                          iphtent_t *);
113 static void *ipf_htable_select_add_ref(void *, int, char *);
114 static void ipf_htable_expire(ipf_main_softc_t *, void *);
115 
116 
117 typedef struct ipf_htable_softc_s {
118           u_long              ipht_nomem[LOOKUP_POOL_SZ];
119           u_long              ipf_nhtables[LOOKUP_POOL_SZ];
120           u_long              ipf_nhtnodes[LOOKUP_POOL_SZ];
121           iphtable_t          *ipf_htables[LOOKUP_POOL_SZ];
122           iphtent_t *ipf_node_explist;
123 } ipf_htable_softc_t;
124 
125 ipf_lookup_t ipf_htable_backend = {
126           IPLT_HASH,
127           ipf_htable_soft_create,
128           ipf_htable_soft_destroy,
129           ipf_htable_soft_init,
130           ipf_htable_soft_fini,
131           ipf_iphmfindip,
132           ipf_htable_flush,
133           ipf_htable_iter_deref,
134           ipf_htable_iter_next,
135           ipf_htable_node_add,
136           ipf_htable_node_del,
137           ipf_htable_stats_get,
138           ipf_htable_table_add,
139           ipf_htable_table_del,
140           ipf_htable_deref,
141           ipf_htable_exists,
142           ipf_htable_select_add_ref,
143           NULL,
144           ipf_htable_expire,
145           NULL
146 };
147 
148 
149 /* ------------------------------------------------------------------------ */
150 /* Function:    ipf_htable_soft_create                                      */
151 /* Returns:     void *   - NULL = failure, else pointer to local context    */
152 /* Parameters:  softc(I) - pointer to soft context main structure           */
153 /*                                                                          */
154 /* Initialise the routing table data structures where required.             */
155 /* ------------------------------------------------------------------------ */
156 static void *
ipf_htable_soft_create(ipf_main_softc_t * softc)157 ipf_htable_soft_create(ipf_main_softc_t *softc)
158 {
159           ipf_htable_softc_t *softh;
160 
161           KMALLOC(softh, ipf_htable_softc_t *);
162           if (softh == NULL) {
163                     IPFERROR(30026);
164                     return NULL;
165           }
166 
167           bzero((char *)softh, sizeof(*softh));
168 
169           return softh;
170 }
171 
172 
173 /* ------------------------------------------------------------------------ */
174 /* Function:    ipf_htable_soft_destroy                                     */
175 /* Returns:     Nil                                                         */
176 /* Parameters:  softc(I) - pointer to soft context main structure           */
177 /*              arg(I)   - pointer to local context to use                  */
178 /*                                                                          */
179 /* Clean up the pool by free'ing the radix tree associated with it and free */
180 /* up the pool context too.                                                 */
181 /* ------------------------------------------------------------------------ */
182 static void
ipf_htable_soft_destroy(ipf_main_softc_t * softc,void * arg)183 ipf_htable_soft_destroy(ipf_main_softc_t *softc, void *arg)
184 {
185           ipf_htable_softc_t *softh = arg;
186 
187           KFREE(softh);
188 }
189 
190 
191 /* ------------------------------------------------------------------------ */
192 /* Function:    ipf_htable_soft_init                                        */
193 /* Returns:     int     - 0 = success, else error                           */
194 /* Parameters:  softc(I) - pointer to soft context main structure           */
195 /*              arg(I)   - pointer to local context to use                  */
196 /*                                                                          */
197 /* Initialise the hash table ready for use.                                 */
198 /* ------------------------------------------------------------------------ */
199 static int
ipf_htable_soft_init(ipf_main_softc_t * softc,void * arg)200 ipf_htable_soft_init(ipf_main_softc_t *softc, void *arg)
201 {
202           ipf_htable_softc_t *softh = arg;
203 
204           bzero((char *)softh, sizeof(*softh));
205 
206           return 0;
207 }
208 
209 
210 /* ------------------------------------------------------------------------ */
211 /* Function:    ipf_htable_soft_fini                                        */
212 /* Returns:     Nil                                                         */
213 /* Parameters:  softc(I) - pointer to soft context main structure           */
214 /*              arg(I)   - pointer to local context to use                  */
215 /* Locks:       WRITE(ipf_global)                                           */
216 /*                                                                          */
217 /* Clean up all the pool data structures allocated and call the cleanup     */
218 /* function for the radix tree that supports the pools. ipf_pool_destroy is */
219 /* used to delete the pools one by one to ensure they're properly freed up. */
220 /* ------------------------------------------------------------------------ */
221 static void
ipf_htable_soft_fini(ipf_main_softc_t * softc,void * arg)222 ipf_htable_soft_fini(ipf_main_softc_t *softc, void *arg)
223 {
224           iplookupflush_t fop;
225 
226           fop.iplf_type = IPLT_HASH;
227           fop.iplf_unit = IPL_LOGALL;
228           fop.iplf_arg = 0;
229           fop.iplf_count = 0;
230           *fop.iplf_name = '\0';
231           ipf_htable_flush(softc, arg, &fop);
232 }
233 
234 
235 /* ------------------------------------------------------------------------ */
236 /* Function:    ipf_htable_stats_get                                        */
237 /* Returns:     int - 0 = success, else error                               */
238 /* Parameters:  softc(I) - pointer to soft context main structure           */
239 /*              arg(I)   - pointer to local context to use                  */
240 /*              op(I)    - pointer to lookup operation data                 */
241 /*                                                                          */
242 /* Copy the relevant statistics out of internal structures and into the     */
243 /* structure used to export statistics.                                     */
244 /* ------------------------------------------------------------------------ */
245 static int
ipf_htable_stats_get(ipf_main_softc_t * softc,void * arg,iplookupop_t * op)246 ipf_htable_stats_get(ipf_main_softc_t *softc, void *arg, iplookupop_t *op)
247 {
248           ipf_htable_softc_t *softh = arg;
249           iphtstat_t stats;
250           int err;
251 
252           if (op->iplo_size != sizeof(stats)) {
253                     IPFERROR(30001);
254                     return EINVAL;
255           }
256 
257           stats.iphs_tables = softh->ipf_htables[op->iplo_unit + 1];
258           stats.iphs_numtables = softh->ipf_nhtables[op->iplo_unit + 1];
259           stats.iphs_numnodes = softh->ipf_nhtnodes[op->iplo_unit + 1];
260           stats.iphs_nomem = softh->ipht_nomem[op->iplo_unit + 1];
261 
262           err = COPYOUT(&stats, op->iplo_struct, sizeof(stats));
263           if (err != 0) {
264                     IPFERROR(30013);
265                     return EFAULT;
266           }
267           return 0;
268 
269 }
270 
271 
272 /* ------------------------------------------------------------------------ */
273 /* Function:    ipf_htable_create                                           */
274 /* Returns:     int - 0 = success, else error                               */
275 /* Parameters:  softc(I) - pointer to soft context main structure           */
276 /*              arg(I)   - pointer to local context to use                  */
277 /*              op(I)    - pointer to lookup operation data                 */
278 /*                                                                          */
279 /* Create a new hash table using the template passed.                       */
280 /* ------------------------------------------------------------------------ */
281 static int
ipf_htable_create(ipf_main_softc_t * softc,void * arg,iplookupop_t * op)282 ipf_htable_create(ipf_main_softc_t *softc, void *arg, iplookupop_t *op)
283 {
284           ipf_htable_softc_t *softh = arg;
285           iphtable_t htab, *iph, *oiph;
286           char name[FR_GROUPLEN];
287           int err, i, unit;
288 
289           if (op->iplo_size != sizeof(htab)) {
290                     IPFERROR(30024);
291                     return EINVAL;
292           }
293           err = COPYIN(op->iplo_struct, &htab, sizeof(htab));
294           if (err != 0) {
295                     IPFERROR(30003);
296                     return EFAULT;
297           }
298 
299           unit = op->iplo_unit;
300           if (htab.iph_unit != unit) {
301                     IPFERROR(30005);
302                     return EINVAL;
303           }
304           if (htab.iph_size < 1) {
305                     IPFERROR(30025);
306                     return EINVAL;
307           }
308 
309 
310           if ((op->iplo_arg & IPHASH_ANON) == 0) {
311                     iph = ipf_htable_exists(softh, unit, op->iplo_name);
312                     if (iph != NULL) {
313                               if ((iph->iph_flags & IPHASH_DELETE) == 0) {
314                                         IPFERROR(30004);
315                                         return EEXIST;
316                               }
317                               iph->iph_flags &= ~IPHASH_DELETE;
318                               iph->iph_ref++;
319                               return 0;
320                     }
321           }
322 
323           KMALLOC(iph, iphtable_t *);
324           if (iph == NULL) {
325                     softh->ipht_nomem[op->iplo_unit + 1]++;
326                     IPFERROR(30002);
327                     return ENOMEM;
328           }
329           *iph = htab;
330 
331           if ((op->iplo_arg & IPHASH_ANON) != 0) {
332                     i = IPHASH_ANON;
333                     do {
334                               i++;
335                               snprintf(name, sizeof(name), "%u", i);
336                               for (oiph = softh->ipf_htables[unit + 1]; oiph != NULL;
337                                    oiph = oiph->iph_next)
338                                         if (strncmp(oiph->iph_name, name,
339                                                       sizeof(oiph->iph_name)) == 0)
340                                                   break;
341                     } while (oiph != NULL);
342 
343                     (void)strncpy(iph->iph_name, name, sizeof(iph->iph_name));
344                     (void)strncpy(op->iplo_name, name, sizeof(op->iplo_name));
345                     iph->iph_type |= IPHASH_ANON;
346           } else {
347                     (void)strncpy(iph->iph_name, op->iplo_name,
348                                     sizeof(iph->iph_name));
349                     iph->iph_name[sizeof(iph->iph_name) - 1] = '\0';
350           }
351 
352           KMALLOCS(iph->iph_table, iphtent_t **,
353                      iph->iph_size * sizeof(*iph->iph_table));
354           if (iph->iph_table == NULL) {
355                     KFREE(iph);
356                     softh->ipht_nomem[unit + 1]++;
357                     IPFERROR(30006);
358                     return ENOMEM;
359           }
360 
361           bzero((char *)iph->iph_table, iph->iph_size * sizeof(*iph->iph_table));
362           iph->iph_maskset[0] = 0;
363           iph->iph_maskset[1] = 0;
364           iph->iph_maskset[2] = 0;
365           iph->iph_maskset[3] = 0;
366 
367           iph->iph_ref = 1;
368           iph->iph_list = NULL;
369           iph->iph_tail = &iph->iph_list;
370           iph->iph_next = softh->ipf_htables[unit + 1];
371           iph->iph_pnext = &softh->ipf_htables[unit + 1];
372           if (softh->ipf_htables[unit + 1] != NULL)
373                     softh->ipf_htables[unit + 1]->iph_pnext = &iph->iph_next;
374           softh->ipf_htables[unit + 1] = iph;
375 
376           softh->ipf_nhtables[unit + 1]++;
377 
378           return 0;
379 }
380 
381 
382 /* ------------------------------------------------------------------------ */
383 /* Function:    ipf_htable_table_del                                        */
384 /* Returns:     int      - 0 = success, else error                          */
385 /* Parameters:  softc(I) - pointer to soft context main structure           */
386 /*              arg(I)   - pointer to local context to use                  */
387 /*              op(I)    - pointer to lookup operation data                 */
388 /*                                                                          */
389 /* ------------------------------------------------------------------------ */
390 static int
ipf_htable_table_del(ipf_main_softc_t * softc,void * arg,iplookupop_t * op)391 ipf_htable_table_del(ipf_main_softc_t *softc, void *arg, iplookupop_t *op)
392 {
393           return ipf_htable_destroy(softc, arg, op->iplo_unit, op->iplo_name);
394 }
395 
396 
397 /* ------------------------------------------------------------------------ */
398 /* Function:    ipf_htable_destroy                                          */
399 /* Returns:     int      - 0 = success, else error                          */
400 /* Parameters:  softc(I) - pointer to soft context main structure           */
401 /*              arg(I)   - pointer to local context to use                  */
402 /*              op(I)    - pointer to lookup operation data                 */
403 /*                                                                          */
404 /* Find the hash table that belongs to the relevant part of ipfilter with a */
405 /* matching name and attempt to destroy it.  If it is in use, empty it out  */
406 /* and mark it for deletion so that when all the references disappear, it   */
407 /* can be removed.                                                          */
408 /* ------------------------------------------------------------------------ */
409 static int
ipf_htable_destroy(ipf_main_softc_t * softc,void * arg,int unit,char * name)410 ipf_htable_destroy(ipf_main_softc_t *softc, void *arg, int unit, char *name)
411 {
412           iphtable_t *iph;
413 
414           iph = ipf_htable_find(arg, unit, name);
415           if (iph == NULL) {
416                     IPFERROR(30007);
417                     return ESRCH;
418           }
419 
420           if (iph->iph_unit != unit) {
421                     IPFERROR(30008);
422                     return EINVAL;
423           }
424 
425           if (iph->iph_ref != 0) {
426                     ipf_htable_clear(softc, arg, iph);
427                     iph->iph_flags |= IPHASH_DELETE;
428                     return 0;
429           }
430 
431           ipf_htable_remove(softc, arg, iph);
432 
433           return 0;
434 }
435 
436 
437 /* ------------------------------------------------------------------------ */
438 /* Function:    ipf_htable_clear                                            */
439 /* Returns:     int      - 0 = success, else error                          */
440 /* Parameters:  softc(I) - pointer to soft context main structure           */
441 /*              arg(I)   - pointer to local context to use                  */
442 /*              iph(I)   - pointer to hash table to destroy                 */
443 /*                                                                          */
444 /* Clean out the hash table by walking the list of entries and removing     */
445 /* each one, one by one.                                                    */
446 /* ------------------------------------------------------------------------ */
447 static int
ipf_htable_clear(ipf_main_softc_t * softc,void * arg,iphtable_t * iph)448 ipf_htable_clear(ipf_main_softc_t *softc, void *arg, iphtable_t *iph)
449 {
450           iphtent_t *ipe;
451 
452           while ((ipe = iph->iph_list) != NULL)
453                     if (ipf_htent_remove(softc, arg, iph, ipe) != 0)
454                               return 1;
455           return 0;
456 }
457 
458 
459 /* ------------------------------------------------------------------------ */
460 /* Function:    ipf_htable_free                                             */
461 /* Returns:     Nil                                                         */
462 /* Parameters:  arg(I) - pointer to local context to use                    */
463 /*              iph(I) - pointer to hash table to destroy                   */
464 /*                                                                          */
465 /* ------------------------------------------------------------------------ */
466 static void
ipf_htable_free(void * arg,iphtable_t * iph)467 ipf_htable_free(void *arg, iphtable_t *iph)
468 {
469           ipf_htable_softc_t *softh = arg;
470 
471           if (iph->iph_next != NULL)
472                     iph->iph_next->iph_pnext = iph->iph_pnext;
473           if (iph->iph_pnext != NULL)
474                     *iph->iph_pnext = iph->iph_next;
475           iph->iph_pnext = NULL;
476           iph->iph_next = NULL;
477 
478           softh->ipf_nhtables[iph->iph_unit + 1]--;
479 
480           KFREES(iph->iph_table, iph->iph_size * sizeof(*iph->iph_table));
481           KFREE(iph);
482 }
483 
484 
485 /* ------------------------------------------------------------------------ */
486 /* Function:    ipf_htable_remove                                           */
487 /* Returns:     int      - 0 = success, else error                          */
488 /* Parameters:  softc(I) - pointer to soft context main structure           */
489 /*              arg(I)   - pointer to local context to use                  */
490 /*              iph(I)   - pointer to hash table to destroy                 */
491 /*                                                                          */
492 /* It is necessary to unlink here as well as free (called by deref) so that */
493 /* the while loop in ipf_htable_flush() functions properly.                 */
494 /* ------------------------------------------------------------------------ */
495 static int
ipf_htable_remove(ipf_main_softc_t * softc,void * arg,iphtable_t * iph)496 ipf_htable_remove(ipf_main_softc_t *softc, void *arg, iphtable_t *iph)
497 {
498 
499           if (ipf_htable_clear(softc, arg, iph) != 0)
500                     return 1;
501 
502           if (iph->iph_pnext != NULL)
503                     *iph->iph_pnext = iph->iph_next;
504           if (iph->iph_next != NULL)
505                     iph->iph_next->iph_pnext = iph->iph_pnext;
506           iph->iph_pnext = NULL;
507           iph->iph_next = NULL;
508 
509           return ipf_htable_deref(softc, arg, iph);
510 }
511 
512 
513 /* ------------------------------------------------------------------------ */
514 /* Function:    ipf_htable_node_del                                         */
515 /* Returns:     int      - 0 = success, else error                          */
516 /* Parameters:  softc(I) - pointer to soft context main structure           */
517 /*              arg(I)   - pointer to local context to use                  */
518 /*              op(I)    - pointer to lookup operation data                 */
519 /*              uid(I)   - real uid of process doing operation              */
520 /*                                                                          */
521 /* ------------------------------------------------------------------------ */
522 static int
ipf_htable_node_del(ipf_main_softc_t * softc,void * arg,iplookupop_t * op,int uid)523 ipf_htable_node_del(ipf_main_softc_t *softc, void *arg, iplookupop_t *op,
524     int uid)
525 {
526         iphtable_t *iph;
527         iphtent_t hte, *ent;
528           int err;
529 
530           if (op->iplo_size != sizeof(hte)) {
531                     IPFERROR(30014);
532                     return EINVAL;
533           }
534 
535           err = COPYIN(op->iplo_struct, &hte, sizeof(hte));
536           if (err != 0) {
537                     IPFERROR(30015);
538                     return EFAULT;
539           }
540 
541           iph = ipf_htable_find(arg, op->iplo_unit, op->iplo_name);
542           if (iph == NULL) {
543                     IPFERROR(30016);
544                     return ESRCH;
545           }
546 
547           ent = ipf_htent_find(iph, &hte);
548           if (ent == NULL) {
549                     IPFERROR(30022);
550                     return ESRCH;
551           }
552 
553           if ((uid != 0) && (ent->ipe_uid != uid)) {
554                     IPFERROR(30023);
555                     return EACCES;
556           }
557 
558           err = ipf_htent_remove(softc, arg, iph, ent);
559 
560           return err;
561 }
562 
563 
564 /* ------------------------------------------------------------------------ */
565 /* Function:    ipf_htable_node_del                                         */
566 /* Returns:     int      - 0 = success, else error                          */
567 /* Parameters:  softc(I) - pointer to soft context main structure           */
568 /*              arg(I)   - pointer to local context to use                  */
569 /*              op(I)    - pointer to lookup operation data                 */
570 /*                                                                          */
571 /* ------------------------------------------------------------------------ */
572 static int
ipf_htable_table_add(ipf_main_softc_t * softc,void * arg,iplookupop_t * op)573 ipf_htable_table_add(ipf_main_softc_t *softc, void *arg, iplookupop_t *op)
574 {
575           int err;
576 
577           if (ipf_htable_find(arg, op->iplo_unit, op->iplo_name) != NULL) {
578                     IPFERROR(30017);
579                     err = EEXIST;
580           } else {
581                     err = ipf_htable_create(softc, arg, op);
582           }
583 
584           return err;
585 }
586 
587 
588 /* ------------------------------------------------------------------------ */
589 /* Function:    ipf_htent_remove                                            */
590 /* Returns:     int      - 0 = success, else error                          */
591 /* Parameters:  softc(I) - pointer to soft context main structure           */
592 /*              arg(I)   - pointer to local context to use                  */
593 /*              iph(I)   - pointer to hash table                            */
594 /*              ipe(I)   - pointer to hash table entry to remove            */
595 /*                                                                          */
596 /* Delete an entry from a hash table.                                       */
597 /* ------------------------------------------------------------------------ */
598 static int
ipf_htent_remove(ipf_main_softc_t * softc,void * arg,iphtable_t * iph,iphtent_t * ipe)599 ipf_htent_remove(ipf_main_softc_t *softc, void *arg, iphtable_t *iph,
600     iphtent_t *ipe)
601 {
602 
603           if (iph->iph_tail == &ipe->ipe_next)
604                     iph->iph_tail = ipe->ipe_pnext;
605 
606           if (ipe->ipe_hnext != NULL)
607                     ipe->ipe_hnext->ipe_phnext = ipe->ipe_phnext;
608           if (ipe->ipe_phnext != NULL)
609                     *ipe->ipe_phnext = ipe->ipe_hnext;
610           ipe->ipe_phnext = NULL;
611           ipe->ipe_hnext = NULL;
612 
613           if (ipe->ipe_dnext != NULL)
614                     ipe->ipe_dnext->ipe_pdnext = ipe->ipe_pdnext;
615           if (ipe->ipe_pdnext != NULL)
616                     *ipe->ipe_pdnext = ipe->ipe_dnext;
617           ipe->ipe_pdnext = NULL;
618           ipe->ipe_dnext = NULL;
619 
620           if (ipe->ipe_next != NULL)
621                     ipe->ipe_next->ipe_pnext = ipe->ipe_pnext;
622           if (ipe->ipe_pnext != NULL)
623                     *ipe->ipe_pnext = ipe->ipe_next;
624           ipe->ipe_pnext = NULL;
625           ipe->ipe_next = NULL;
626 
627           switch (iph->iph_type & ~IPHASH_ANON)
628           {
629           case IPHASH_GROUPMAP :
630                     ipf_group_del(softc, ipe->ipe_ptr, NULL);
631                     break;
632 
633           default :
634                     ipe->ipe_ptr = NULL;
635                     ipe->ipe_value = 0;
636                     break;
637           }
638 
639           return ipf_htent_deref(arg, ipe);
640 }
641 
642 
643 /* ------------------------------------------------------------------------ */
644 /* Function:    ipf_htable_deref                                            */
645 /* Returns:     int       - 0 = success, else error                         */
646 /* Parameters:  softc(I)  - pointer to soft context main structure          */
647 /*              arg(I)    - pointer to local context to use                 */
648 /*              object(I) - pointer to hash table                           */
649 /*                                                                          */
650 /* ------------------------------------------------------------------------ */
651 static int
ipf_htable_deref(ipf_main_softc_t * softc,void * arg,void * object)652 ipf_htable_deref(ipf_main_softc_t *softc, void *arg, void *object)
653 {
654           ipf_htable_softc_t *softh = arg;
655           iphtable_t *iph = object;
656           int refs;
657 
658           iph->iph_ref--;
659           refs = iph->iph_ref;
660 
661           if (iph->iph_ref == 0) {
662                     ipf_htable_free(softh, iph);
663           }
664 
665           return refs;
666 }
667 
668 
669 /* ------------------------------------------------------------------------ */
670 /* Function:    ipf_htent_deref                                             */
671 /* Parameters:  arg(I) - pointer to local context to use                    */
672 /*              ipe(I) -                                                    */
673 /*                                                                          */
674 /* ------------------------------------------------------------------------ */
675 static int
ipf_htent_deref(void * arg,iphtent_t * ipe)676 ipf_htent_deref(void *arg, iphtent_t *ipe)
677 {
678           ipf_htable_softc_t *softh = arg;
679 
680           ipe->ipe_ref--;
681           if (ipe->ipe_ref == 0) {
682                     softh->ipf_nhtnodes[ipe->ipe_unit + 1]--;
683                     KFREE(ipe);
684 
685                     return 0;
686           }
687 
688           return ipe->ipe_ref;
689 }
690 
691 
692 /* ------------------------------------------------------------------------ */
693 /* Function:    ipf_htable_exists                                           */
694 /* Parameters:  arg(I) - pointer to local context to use                    */
695 /*                                                                          */
696 /* ------------------------------------------------------------------------ */
697 static void *
ipf_htable_exists(void * arg,int unit,char * name)698 ipf_htable_exists(void *arg, int unit, char *name)
699 {
700           ipf_htable_softc_t *softh = arg;
701           iphtable_t *iph;
702 
703           if (unit == IPL_LOGALL) {
704                     int i;
705 
706                     for (i = 0; i <= LOOKUP_POOL_MAX; i++) {
707                               for (iph = softh->ipf_htables[i]; iph != NULL;
708                                    iph = iph->iph_next) {
709                                         if (strncmp(iph->iph_name, name,
710                                                       sizeof(iph->iph_name)) == 0)
711                                                   break;
712                               }
713                               if (iph != NULL)
714                                         break;
715                     }
716           } else {
717                     for (iph = softh->ipf_htables[unit + 1]; iph != NULL;
718                          iph = iph->iph_next) {
719                               if (strncmp(iph->iph_name, name,
720                                             sizeof(iph->iph_name)) == 0)
721                                         break;
722                     }
723           }
724           return iph;
725 }
726 
727 
728 /* ------------------------------------------------------------------------ */
729 /* Function:    ipf_htable_select_add_ref                                   */
730 /* Returns:     void *  - NULL = failure, else pointer to the hash table    */
731 /* Parameters:  arg(I)  - pointer to local context to use                   */
732 /*              unit(I) - ipfilter device to which we are working on        */
733 /*              name(I) - name of the hash table                            */
734 /*                                                                          */
735 /* ------------------------------------------------------------------------ */
736 static void *
ipf_htable_select_add_ref(void * arg,int unit,char * name)737 ipf_htable_select_add_ref(void *arg, int unit, char *name)
738 {
739           iphtable_t *iph;
740 
741           iph = ipf_htable_exists(arg, unit, name);
742           if (iph != NULL) {
743                     ATOMIC_INC32(iph->iph_ref);
744           }
745           return iph;
746 }
747 
748 
749 /* ------------------------------------------------------------------------ */
750 /* Function:    ipf_htable_find                                             */
751 /* Returns:     void *  - NULL = failure, else pointer to the hash table    */
752 /* Parameters:  arg(I)  - pointer to local context to use                   */
753 /*              unit(I) - ipfilter device to which we are working on        */
754 /*              name(I) - name of the hash table                            */
755 /*                                                                          */
756 /* This function is exposed becaues it is used in the group-map feature.    */
757 /* ------------------------------------------------------------------------ */
758 iphtable_t *
ipf_htable_find(void * arg,int unit,char * name)759 ipf_htable_find(void *arg, int unit, char *name)
760 {
761           iphtable_t *iph;
762 
763           iph = ipf_htable_exists(arg, unit, name);
764           if ((iph != NULL) && (iph->iph_flags & IPHASH_DELETE) == 0)
765                     return iph;
766 
767           return NULL;
768 }
769 
770 
771 /* ------------------------------------------------------------------------ */
772 /* Function:    ipf_htable_flush                                            */
773 /* Returns:     size_t   - number of entries flushed                        */
774 /* Parameters:  softc(I) - pointer to soft context main structure           */
775 /*              arg(I)   - pointer to local context to use                  */
776 /*              op(I)    - pointer to lookup operation data                 */
777 /*                                                                          */
778 /* ------------------------------------------------------------------------ */
779 static size_t
ipf_htable_flush(ipf_main_softc_t * softc,void * arg,iplookupflush_t * op)780 ipf_htable_flush(ipf_main_softc_t *softc, void *arg, iplookupflush_t *op)
781 {
782           ipf_htable_softc_t *softh = arg;
783           iphtable_t *iph;
784           size_t freed;
785           int i;
786 
787           freed = 0;
788 
789           for (i = -1; i <= IPL_LOGMAX; i++) {
790                     if (op->iplf_unit == i || op->iplf_unit == IPL_LOGALL) {
791                               while ((iph = softh->ipf_htables[i + 1]) != NULL) {
792                                         if (ipf_htable_remove(softc, arg, iph) == 0) {
793                                                   freed++;
794                                         } else {
795                                                   iph->iph_flags |= IPHASH_DELETE;
796                                         }
797                               }
798                     }
799           }
800 
801           return freed;
802 }
803 
804 
805 /* ------------------------------------------------------------------------ */
806 /* Function:    ipf_htable_node_add                                         */
807 /* Returns:     int      - 0 = success, else error                          */
808 /* Parameters:  softc(I) - pointer to soft context main structure           */
809 /*              arg(I)   - pointer to local context to use                  */
810 /*              op(I)    - pointer to lookup operation data                 */
811 /*              uid(I)   - real uid of process doing operation              */
812 /*                                                                          */
813 /* ------------------------------------------------------------------------ */
814 static int
ipf_htable_node_add(ipf_main_softc_t * softc,void * arg,iplookupop_t * op,int uid)815 ipf_htable_node_add(ipf_main_softc_t *softc, void *arg, iplookupop_t *op,
816     int uid)
817 {
818           iphtable_t *iph;
819           iphtent_t hte;
820           int err;
821 
822           if (op->iplo_size != sizeof(hte)) {
823                     IPFERROR(30018);
824                     return EINVAL;
825           }
826 
827           err = COPYIN(op->iplo_struct, &hte, sizeof(hte));
828           if (err != 0) {
829                     IPFERROR(30019);
830                     return EFAULT;
831           }
832           hte.ipe_uid = uid;
833 
834           iph = ipf_htable_find(arg, op->iplo_unit, op->iplo_name);
835           if (iph == NULL) {
836                     IPFERROR(30020);
837                     return ESRCH;
838           }
839 
840           if (ipf_htent_find(iph, &hte) != NULL) {
841                     IPFERROR(30021);
842                     return EEXIST;
843           }
844 
845           err = ipf_htent_insert(softc, arg, iph, &hte);
846 
847           return err;
848 }
849 
850 
851 /* ------------------------------------------------------------------------ */
852 /* Function:    ipf_htent_insert                                            */
853 /* Returns:     int      - 0 = success, -1 =  error                         */
854 /* Parameters:  softc(I) - pointer to soft context main structure           */
855 /*              arg(I)   - pointer to local context to use                  */
856 /*              op(I)    - pointer to lookup operation data                 */
857 /*              ipeo(I)  -                                                  */
858 /*                                                                          */
859 /* Add an entry to a hash table.                                            */
860 /* ------------------------------------------------------------------------ */
861 static int
ipf_htent_insert(ipf_main_softc_t * softc,void * arg,iphtable_t * iph,iphtent_t * ipeo)862 ipf_htent_insert(ipf_main_softc_t *softc, void *arg, iphtable_t *iph,
863     iphtent_t *ipeo)
864 {
865           ipf_htable_softc_t *softh = arg;
866           iphtent_t *ipe;
867           u_int hv;
868           int bits;
869 
870           KMALLOC(ipe, iphtent_t *);
871           if (ipe == NULL)
872                     return -1;
873 
874           bcopy((char *)ipeo, (char *)ipe, sizeof(*ipe));
875           ipe->ipe_addr.i6[0] &= ipe->ipe_mask.i6[0];
876           if (ipe->ipe_family == AF_INET) {
877                     bits = count4bits(ipe->ipe_mask.in4_addr);
878                     ipe->ipe_addr.i6[1] = 0;
879                     ipe->ipe_addr.i6[2] = 0;
880                     ipe->ipe_addr.i6[3] = 0;
881                     ipe->ipe_mask.i6[1] = 0;
882                     ipe->ipe_mask.i6[2] = 0;
883                     ipe->ipe_mask.i6[3] = 0;
884                     hv = IPE_V4_HASH_FN(ipe->ipe_addr.in4_addr,
885                                             ipe->ipe_mask.in4_addr, iph->iph_size);
886           } else
887 #ifdef USE_INET6
888           if (ipe->ipe_family == AF_INET6) {
889                     ipe->ipe_addr.i6[1] &= ipe->ipe_mask.i6[1];
890                     ipe->ipe_addr.i6[2] &= ipe->ipe_mask.i6[2];
891                     ipe->ipe_addr.i6[3] &= ipe->ipe_mask.i6[3];
892 
893                     bits = count6bits(ipe->ipe_mask.i6);
894                     hv = IPE_V6_HASH_FN(ipe->ipe_addr.i6,
895                                             ipe->ipe_mask.i6, iph->iph_size);
896           } else
897 #endif
898           {
899                     KFREE(ipe);
900                     return -1;
901           }
902 
903           ipe->ipe_owner = iph;
904           ipe->ipe_ref = 1;
905           ipe->ipe_hnext = iph->iph_table[hv];
906           ipe->ipe_phnext = iph->iph_table + hv;
907 
908           if (iph->iph_table[hv] != NULL)
909                     iph->iph_table[hv]->ipe_phnext = &ipe->ipe_hnext;
910           iph->iph_table[hv] = ipe;
911 
912           ipe->ipe_pnext = iph->iph_tail;
913           *iph->iph_tail = ipe;
914           iph->iph_tail = &ipe->ipe_next;
915           ipe->ipe_next = NULL;
916 
917           if (ipe->ipe_die != 0) {
918                     /*
919                      * If the new node has a given expiration time, insert it
920                      * into the list of expiring nodes with the ones to be
921                      * removed first added to the front of the list. The
922                      * insertion is O(n) but it is kept sorted for quick scans
923                      * at expiration interval checks.
924                      */
925                     iphtent_t *n;
926 
927                     ipe->ipe_die = softc->ipf_ticks + IPF_TTLVAL(ipe->ipe_die);
928                     for (n = softh->ipf_node_explist; n != NULL; n = n->ipe_dnext) {
929                               if (ipe->ipe_die < n->ipe_die)
930                                         break;
931                               if (n->ipe_dnext == NULL) {
932                                         /*
933                                          * We've got to the last node and everything
934                                          * wanted to be expired before this new node,
935                                          * so we have to tack it on the end...
936                                          */
937                                         n->ipe_dnext = ipe;
938                                         ipe->ipe_pdnext = &n->ipe_dnext;
939                                         n = NULL;
940                                         break;
941                               }
942                     }
943 
944                     if (softh->ipf_node_explist == NULL) {
945                               softh->ipf_node_explist = ipe;
946                               ipe->ipe_pdnext = &softh->ipf_node_explist;
947                     } else if (n != NULL) {
948                               ipe->ipe_dnext = n;
949                               ipe->ipe_pdnext = n->ipe_pdnext;
950                               n->ipe_pdnext = &ipe->ipe_dnext;
951                     }
952           }
953 
954           if (ipe->ipe_family == AF_INET) {
955                     ipf_inet_mask_add(bits, &iph->iph_v4_masks);
956           }
957 #ifdef USE_INET6
958           else if (ipe->ipe_family == AF_INET6) {
959                     ipf_inet6_mask_add(bits, &ipe->ipe_mask, &iph->iph_v6_masks);
960           }
961 #endif
962 
963           switch (iph->iph_type & ~IPHASH_ANON)
964           {
965           case IPHASH_GROUPMAP :
966                     ipe->ipe_ptr = ipf_group_add(softc, ipe->ipe_group, NULL,
967                                                      iph->iph_flags, IPL_LOGIPF,
968                                                      softc->ipf_active);
969                     break;
970 
971           default :
972                     ipe->ipe_ptr = NULL;
973                     ipe->ipe_value = 0;
974                     break;
975           }
976 
977           ipe->ipe_unit = iph->iph_unit;
978           softh->ipf_nhtnodes[ipe->ipe_unit + 1]++;
979 
980           return 0;
981 }
982 
983 
984 /* ------------------------------------------------------------------------ */
985 /* Function:    ipf_htent_find                                              */
986 /* Returns:     int     - 0 = success, else error                           */
987 /* Parameters:  iph(I)  - pointer to table to search                        */
988 /*              ipeo(I) - pointer to entry to find                          */
989 /*                                                                          */
990 /* While it isn't absolutely necessary to for the address and mask to be    */
991 /* passed in through an iphtent_t structure, one is always present when it  */
992 /* is time to call this function, so it is just more convenient.            */
993 /* ------------------------------------------------------------------------ */
994 static iphtent_t *
ipf_htent_find(iphtable_t * iph,iphtent_t * ipeo)995 ipf_htent_find(iphtable_t *iph, iphtent_t *ipeo)
996 {
997           iphtent_t ipe, *ent;
998           u_int hv;
999 
1000           bcopy((char *)ipeo, (char *)&ipe, sizeof(ipe));
1001           ipe.ipe_addr.i6[0] &= ipe.ipe_mask.i6[0];
1002           ipe.ipe_addr.i6[1] &= ipe.ipe_mask.i6[1];
1003           ipe.ipe_addr.i6[2] &= ipe.ipe_mask.i6[2];
1004           ipe.ipe_addr.i6[3] &= ipe.ipe_mask.i6[3];
1005           if (ipe.ipe_family == AF_INET) {
1006                     ipe.ipe_addr.i6[1] = 0;
1007                     ipe.ipe_addr.i6[2] = 0;
1008                     ipe.ipe_addr.i6[3] = 0;
1009                     ipe.ipe_mask.i6[1] = 0;
1010                     ipe.ipe_mask.i6[2] = 0;
1011                     ipe.ipe_mask.i6[3] = 0;
1012                     hv = IPE_V4_HASH_FN(ipe.ipe_addr.in4_addr,
1013                                             ipe.ipe_mask.in4_addr, iph->iph_size);
1014           } else
1015 #ifdef USE_INET6
1016           if (ipe.ipe_family == AF_INET6) {
1017                     hv = IPE_V6_HASH_FN(ipe.ipe_addr.i6,
1018                                             ipe.ipe_mask.i6, iph->iph_size);
1019           } else
1020 #endif
1021                     return NULL;
1022 
1023           for (ent = iph->iph_table[hv]; ent != NULL; ent = ent->ipe_hnext) {
1024                     if (ent->ipe_family != ipe.ipe_family)
1025                               continue;
1026                     if (IP6_NEQ(&ipe.ipe_addr, &ent->ipe_addr))
1027                               continue;
1028                     if (IP6_NEQ(&ipe.ipe_mask, &ent->ipe_mask))
1029                               continue;
1030                     break;
1031           }
1032 
1033           return ent;
1034 }
1035 
1036 
1037 /* ------------------------------------------------------------------------ */
1038 /* Function:    ipf_iphmfindgroup                                           */
1039 /* Returns:     int      - 0 = success, else error                          */
1040 /* Parameters:  softc(I) - pointer to soft context main structure           */
1041 /*              tptr(I)  -                                                  */
1042 /*              aptr(I)  -                                                  */
1043 /*                                                                          */
1044 /* Search a hash table for a matching entry and return the pointer stored   */
1045 /* in it for use as the next group of rules to search.                      */
1046 /*                                                                          */
1047 /* This function is exposed becaues it is used in the group-map feature.    */
1048 /* ------------------------------------------------------------------------ */
1049 void *
ipf_iphmfindgroup(ipf_main_softc_t * softc,void * tptr,void * aptr)1050 ipf_iphmfindgroup(ipf_main_softc_t *softc, void *tptr, void *aptr)
1051 {
1052           struct in_addr *addr;
1053           iphtable_t *iph;
1054           iphtent_t *ipe;
1055           void *rval;
1056 
1057           READ_ENTER(&softc->ipf_poolrw);
1058           iph = tptr;
1059           addr = aptr;
1060 
1061           ipe = ipf_iphmfind(iph, addr);
1062           if (ipe != NULL)
1063                     rval = ipe->ipe_ptr;
1064           else
1065                     rval = NULL;
1066           RWLOCK_EXIT(&softc->ipf_poolrw);
1067           return rval;
1068 }
1069 
1070 
1071 /* ------------------------------------------------------------------------ */
1072 /* Function:    ipf_iphmfindip                                              */
1073 /* Returns:     int     - 0 == +ve match, -1 == error, 1 == -ve/no match    */
1074 /* Parameters:  softc(I)     - pointer to soft context main structure       */
1075 /*              tptr(I)      - pointer to the pool to search                */
1076 /*              ipversion(I) - IP protocol version (4 or 6)                 */
1077 /*              aptr(I)      - pointer to address information               */
1078 /*              bytes(I)     - packet length                                */
1079 /*                                                                          */
1080 /* Search the hash table for a given address and return a search result.    */
1081 /* ------------------------------------------------------------------------ */
1082 static int
ipf_iphmfindip(ipf_main_softc_t * softc,void * tptr,int ipversion,void * aptr,u_int bytes)1083 ipf_iphmfindip(ipf_main_softc_t *softc, void *tptr, int ipversion, void *aptr,
1084     u_int bytes)
1085 {
1086           struct in_addr *addr;
1087           iphtable_t *iph;
1088           iphtent_t *ipe;
1089           int rval;
1090 
1091           if (tptr == NULL || aptr == NULL)
1092                     return -1;
1093 
1094           iph = tptr;
1095           addr = aptr;
1096 
1097           READ_ENTER(&softc->ipf_poolrw);
1098           if (ipversion == 4) {
1099                     ipe = ipf_iphmfind(iph, addr);
1100 #ifdef USE_INET6
1101           } else if (ipversion == 6) {
1102                     ipe = ipf_iphmfind6(iph, (i6addr_t *)addr);
1103 #endif
1104           } else {
1105                     ipe = NULL;
1106           }
1107 
1108           if (ipe != NULL) {
1109                     rval = 0;
1110                     ipe->ipe_hits++;
1111                     ipe->ipe_bytes += bytes;
1112           } else {
1113                     rval = 1;
1114           }
1115           RWLOCK_EXIT(&softc->ipf_poolrw);
1116           return rval;
1117 }
1118 
1119 
1120 /* ------------------------------------------------------------------------ */
1121 /* Function:    ipf_iphmfindip                                              */
1122 /* Parameters:  iph(I)  - pointer to hash table                             */
1123 /*              addr(I) - pointer to IPv4 address                           */
1124 /* Locks:  ipf_poolrw                                                       */
1125 /*                                                                          */
1126 /* ------------------------------------------------------------------------ */
1127 static iphtent_t *
ipf_iphmfind(iphtable_t * iph,struct in_addr * addr)1128 ipf_iphmfind(iphtable_t *iph, struct in_addr *addr)
1129 {
1130           u_32_t msk, ips;
1131           iphtent_t *ipe;
1132           u_int hv;
1133           int i;
1134 
1135           i = 0;
1136 maskloop:
1137           msk = iph->iph_v4_masks.imt4_active[i];
1138           ips = addr->s_addr & msk;
1139           hv = IPE_V4_HASH_FN(ips, msk, iph->iph_size);
1140           for (ipe = iph->iph_table[hv]; (ipe != NULL); ipe = ipe->ipe_hnext) {
1141                     if ((ipe->ipe_family != AF_INET) ||
1142                         (ipe->ipe_mask.in4_addr != msk) ||
1143                         (ipe->ipe_addr.in4_addr != ips)) {
1144                               continue;
1145                     }
1146                     break;
1147           }
1148 
1149           if (ipe == NULL) {
1150                     i++;
1151                     if (i < iph->iph_v4_masks.imt4_max)
1152                               goto maskloop;
1153           }
1154           return ipe;
1155 }
1156 
1157 
1158 /* ------------------------------------------------------------------------ */
1159 /* Function:    ipf_htable_iter_next                                        */
1160 /* Returns:     int      - 0 = success, else error                          */
1161 /* Parameters:  softc(I) - pointer to soft context main structure           */
1162 /*              arg(I)   - pointer to local context to use                  */
1163 /*              token(I) -                                                  */
1164 /*              ilp(I)   -                                                  */
1165 /*                                                                          */
1166 /* ------------------------------------------------------------------------ */
1167 static int
ipf_htable_iter_next(ipf_main_softc_t * softc,void * arg,ipftoken_t * token,ipflookupiter_t * ilp)1168 ipf_htable_iter_next(ipf_main_softc_t *softc, void *arg, ipftoken_t *token,
1169     ipflookupiter_t *ilp)
1170 {
1171           ipf_htable_softc_t *softh = arg;
1172           iphtent_t *node, zn, *nextnode;
1173           iphtable_t *iph, zp, *nextiph;
1174           void *hnext;
1175           int err;
1176 
1177           err = 0;
1178           iph = NULL;
1179           node = NULL;
1180           nextiph = NULL;
1181           nextnode = NULL;
1182 
1183           READ_ENTER(&softc->ipf_poolrw);
1184 
1185           switch (ilp->ili_otype)
1186           {
1187           case IPFLOOKUPITER_LIST :
1188                     iph = token->ipt_data;
1189                     if (iph == NULL) {
1190                               nextiph = softh->ipf_htables[(int)ilp->ili_unit + 1];
1191                     } else {
1192                               nextiph = iph->iph_next;
1193                     }
1194 
1195                     if (nextiph != NULL) {
1196                               ATOMIC_INC(nextiph->iph_ref);
1197                               token->ipt_data = nextiph;
1198                     } else {
1199                               bzero((char *)&zp, sizeof(zp));
1200                               nextiph = &zp;
1201                               token->ipt_data = NULL;
1202                     }
1203                     hnext = nextiph->iph_next;
1204                     break;
1205 
1206           case IPFLOOKUPITER_NODE :
1207                     node = token->ipt_data;
1208                     if (node == NULL) {
1209                               iph = ipf_htable_find(arg, ilp->ili_unit,
1210                                                         ilp->ili_name);
1211                               if (iph == NULL) {
1212                                         IPFERROR(30009);
1213                                         err = ESRCH;
1214                               } else {
1215                                         nextnode = iph->iph_list;
1216                               }
1217                     } else {
1218                               nextnode = node->ipe_next;
1219                     }
1220 
1221                     if (nextnode != NULL) {
1222                               ATOMIC_INC(nextnode->ipe_ref);
1223                               token->ipt_data = nextnode;
1224                     } else {
1225                               bzero((char *)&zn, sizeof(zn));
1226                               nextnode = &zn;
1227                               token->ipt_data = NULL;
1228                     }
1229                     hnext = nextnode->ipe_next;
1230                     break;
1231 
1232           default :
1233                     IPFERROR(30010);
1234                     err = EINVAL;
1235                     hnext = NULL;
1236                     break;
1237           }
1238 
1239           RWLOCK_EXIT(&softc->ipf_poolrw);
1240           if (err != 0)
1241                     return err;
1242 
1243           switch (ilp->ili_otype)
1244           {
1245           case IPFLOOKUPITER_LIST :
1246                     err = COPYOUT(nextiph, ilp->ili_data, sizeof(*nextiph));
1247                     if (err != 0) {
1248                               IPFERROR(30011);
1249                               err = EFAULT;
1250                     }
1251                     if (iph != NULL) {
1252                               WRITE_ENTER(&softc->ipf_poolrw);
1253                               ipf_htable_deref(softc, softh, iph);
1254                               RWLOCK_EXIT(&softc->ipf_poolrw);
1255                     }
1256                     break;
1257 
1258           case IPFLOOKUPITER_NODE :
1259                     err = COPYOUT(nextnode, ilp->ili_data, sizeof(*nextnode));
1260                     if (err != 0) {
1261                               IPFERROR(30012);
1262                               err = EFAULT;
1263                     }
1264                     if (node != NULL) {
1265                               WRITE_ENTER(&softc->ipf_poolrw);
1266                               ipf_htent_deref(softc, node);
1267                               RWLOCK_EXIT(&softc->ipf_poolrw);
1268                     }
1269                     break;
1270           }
1271 
1272           if (hnext == NULL)
1273                     ipf_token_mark_complete(token);
1274 
1275           return err;
1276 }
1277 
1278 
1279 /* ------------------------------------------------------------------------ */
1280 /* Function:    ipf_htable_iter_deref                                       */
1281 /* Returns:     int      - 0 = success, else  error                         */
1282 /* Parameters:  softc(I) - pointer to soft context main structure           */
1283 /*              arg(I)   - pointer to local context to use                  */
1284 /*              otype(I) - which data structure type is being walked        */
1285 /*              unit(I)  - ipfilter device to which we are working on       */
1286 /*              data(I)  - pointer to old data structure                    */
1287 /*                                                                          */
1288 /* ------------------------------------------------------------------------ */
1289 static int
ipf_htable_iter_deref(ipf_main_softc_t * softc,void * arg,int otype,int unit,void * data)1290 ipf_htable_iter_deref(ipf_main_softc_t *softc, void *arg, int otype, int unit,
1291     void *data)
1292 {
1293 
1294           if (data == NULL)
1295                     return EFAULT;
1296 
1297           if (unit < -1 || unit > IPL_LOGMAX)
1298                     return EINVAL;
1299 
1300           switch (otype)
1301           {
1302           case IPFLOOKUPITER_LIST :
1303                     ipf_htable_deref(softc, arg, (iphtable_t *)data);
1304                     break;
1305 
1306           case IPFLOOKUPITER_NODE :
1307                     ipf_htent_deref(arg, (iphtent_t *)data);
1308                     break;
1309           default :
1310                     break;
1311           }
1312 
1313           return 0;
1314 }
1315 
1316 
1317 #ifdef USE_INET6
1318 /* ------------------------------------------------------------------------ */
1319 /* Function:    ipf_iphmfind6                                               */
1320 /* Parameters:  iph(I)  - pointer to hash table                             */
1321 /*              addr(I) - pointer to IPv6 address                           */
1322 /* Locks:  ipf_poolrw                                                       */
1323 /*                                                                          */
1324 /* ------------------------------------------------------------------------ */
1325 static iphtent_t *
ipf_iphmfind6(iphtable_t * iph,i6addr_t * addr)1326 ipf_iphmfind6(iphtable_t *iph, i6addr_t *addr)
1327 {
1328           i6addr_t *msk, ips;
1329           iphtent_t *ipe;
1330           u_int hv;
1331           int i;
1332 
1333           i = 0;
1334 maskloop:
1335           msk = iph->iph_v6_masks.imt6_active + i;
1336           ips.i6[0] = addr->i6[0] & msk->i6[0];
1337           ips.i6[1] = addr->i6[1] & msk->i6[1];
1338           ips.i6[2] = addr->i6[2] & msk->i6[2];
1339           ips.i6[3] = addr->i6[3] & msk->i6[3];
1340           hv = IPE_V6_HASH_FN(ips.i6, msk->i6, iph->iph_size);
1341           for (ipe = iph->iph_table[hv]; (ipe != NULL); ipe = ipe->ipe_next) {
1342                     if ((ipe->ipe_family != AF_INET6) ||
1343                         IP6_NEQ(&ipe->ipe_mask, msk) ||
1344                         IP6_NEQ(&ipe->ipe_addr, &ips)) {
1345                               continue;
1346                     }
1347                     break;
1348           }
1349 
1350           if (ipe == NULL) {
1351                     i++;
1352                     if (i < iph->iph_v6_masks.imt6_max)
1353                               goto maskloop;
1354           }
1355           return ipe;
1356 }
1357 #endif
1358 
1359 
1360 static void
ipf_htable_expire(ipf_main_softc_t * softc,void * arg)1361 ipf_htable_expire(ipf_main_softc_t *softc, void *arg)
1362 {
1363           ipf_htable_softc_t *softh = arg;
1364           iphtent_t *n;
1365 
1366           while ((n = softh->ipf_node_explist) != NULL) {
1367                     if (n->ipe_die > softc->ipf_ticks)
1368                               break;
1369 
1370                     ipf_htent_remove(softc, softh, n->ipe_owner, n);
1371           }
1372 }
1373 
1374 
1375 #ifndef _KERNEL
1376 
1377 /* ------------------------------------------------------------------------ */
1378 /*                                                                          */
1379 /* ------------------------------------------------------------------------ */
1380 void
ipf_htable_dump(softc,arg)1381 ipf_htable_dump(softc, arg)
1382           ipf_main_softc_t *softc;
1383           void *arg;
1384 {
1385           ipf_htable_softc_t *softh = arg;
1386           iphtable_t *iph;
1387           int i;
1388 
1389           printf("List of configured hash tables\n");
1390           for (i = 0; i < IPL_LOGSIZE; i++)
1391                     for (iph = softh->ipf_htables[i]; iph != NULL;
1392                          iph = iph->iph_next)
1393                               printhash(iph, bcopywrap, NULL, opts, NULL);
1394 
1395 }
1396 #endif
1397