contrib/ipfilter/mlf_rule.c

   1 /*      $FreeBSD$       */
   2
   3 /*
   4  * Copyright (C) 2012 by Darren Reed.
   5  *
   6  * See the IPFILTER.LICENCE file for details on licencing.
   7  */
   8 /*
   9  * 29/12/94 Added code from Marc Huber <huber@fzi.de> to allow it to allocate
  10  * its own major char number! Way cool patch!
  11  */
  12
  13
  14 #include <sys/param.h>
  15
  16 #if defined(__FreeBSD__) && (__FreeBSD__ > 1)
  17 # ifdef IPFILTER_LKM
  18 #  include <osreldate.h>
  19 #  define       ACTUALLY_LKM_NOT_KERNEL
  20 # else
  21 #  include <sys/osreldate.h>
  22 # endif
  23 #endif
  24 #include <sys/systm.h>
  25 #if defined(__FreeBSD_version) && (__FreeBSD_version >= 220000)
  26 # include <sys/conf.h>
  27 # include <sys/kernel.h>
  28 # ifdef DEVFS
  29 #  include <sys/devfsext.h>
  30 # endif /*DEVFS*/
  31 #endif
  32 #include <sys/conf.h>
  33 #include <sys/file.h>
  34 #if defined(__FreeBSD_version) && (__FreeBSD_version >= 300000)
  35 # include <sys/lock.h>
  36 #endif
  37 #include <sys/stat.h>
  38 #include <sys/proc.h>
  39 #include <sys/kernel.h>
  40 #include <sys/vnode.h>
  41 #include <sys/namei.h>
  42 #include <sys/malloc.h>
  43 #include <sys/mount.h>
  44 #include <sys/exec.h>
  45 #include <sys/mbuf.h>
  46 #if     BSD >= 199506
  47 # include <sys/sysctl.h>
  48 #endif
  49 #if (__FreeBSD_version >= 300000)
  50 # include <sys/socket.h>
  51 #endif
  52 #if (__FreeBSD_version >= 199511)
  53 #include <net/if.h>
  54 #include <netinet/in_systm.h>
  55 #include <netinet/in.h>
  56 #include <netinet/ip.h>
  57 #include <net/route.h>
  58 #include <netinet/ip_var.h>
  59 #include <netinet/tcp.h>
  60 #include <netinet/tcpip.h>
  61 #endif
  62 #if (__FreeBSD__ > 1)
  63 # include <sys/sysent.h>
  64 #endif
  65 #include <sys/lkm.h>
  66 #include "netinet/ip_compat.h"
  67 #include "netinet/ip_fil.h"
  68 #include "netinet/ip_rules.h"
  69
  70
  71 int     xxxinit __P((struct lkm_table *, int, int));
  72
  73 #if !defined(__FreeBSD_version) || (__FreeBSD_version < 220000)
  74 MOD_DEV(IPL_VERSION, LM_DT_CHAR, -1, &ipldevsw);
  75 #endif
  76
  77 static int ipfrule_ioctl __P((struct lkm_table *, int));
  78
  79 #if defined(__FreeBSD_version) && (__FreeBSD_version < 220000)
  80
  81 int xxxinit(lkmtp, cmd, ver)
  82         struct lkm_table *lkmtp;
  83         int cmd, ver;
  84 {
  85         DISPATCH(lkmtp, cmd, ver, ipfrule_ioctl, ipfrule_ioctl, ipfrule_ioctl);
  86 }
  87 #else   /* __FREEBSD_version >= 220000 */
  88 # ifdef IPFILTER_LKM
  89 #  include <sys/exec.h>
  90
  91 #  if (__FreeBSD_version >= 300000)
  92 MOD_MISC(ipfrule);
  93 #  else
  94 MOD_DECL(ipfrule);
  95
  96
  97 static struct lkm_misc _module = {
  98         LM_MISC,
  99         LKM_VERSION,
 100         "IP Filter rules",
 101         0,
 102 };
 103 #  endif
 104
 105
 106 int ipfrule __P((struct lkm_table *, int, int));
 107
 108
 109 int ipfrule(lkmtp, cmd, ver)
 110         struct lkm_table *lkmtp;
 111         int cmd, ver;
 112 {
 113 #  if (__FreeBSD_version >= 300000)
 114         MOD_DISPATCH(ipfrule, lkmtp, cmd, ver, ipfrule_ioctl, ipfrule_ioctl,
 115                      ipfrule_ioctl);
 116 #  else
 117         DISPATCH(lkmtp, cmd, ver, ipfrule_ioctl, ipfrule_ioctl, ipfrule_ioctl);
 118 #  endif
 119 }
 120 # endif /* IPFILTER_LKM */
 121
 122
 123 int ipfrule_load(lkmtp, cmd)
 124         struct lkm_table *lkmtp;
 125         int cmd;
 126 {
 127         return ipfrule_add();
 128 }
 129
 130
 131 int ipfrule_unload(lkmtp, cmd)
 132         struct lkm_table *lkmtp;
 133         int cmd;
 134 {
 135         return ipfrule_remove();
 136 }
 137
 138
 139 static int ipfrule_ioctl(lkmtp, cmd)
 140         struct lkm_table *lkmtp;
 141         int cmd;
 142 {
 143         int err = 0;
 144
 145         switch (cmd)
 146         {
 147         case LKM_E_LOAD :
 148                 if (lkmexists(lkmtp))
 149                         return EEXIST;
 150
 151                 err = ipfrule_load(lkmtp, cmd);
 152                 if (!err)
 153                         ipf_refcnt++;
 154                 break;
 155         case LKM_E_UNLOAD :
 156                 err = ipfrule_unload(lkmtp, cmd);
 157                 if (!err)
 158                         ipf_refcnt--;
 159                 break;
 160         case LKM_E_STAT :
 161                 break;
 162         default:
 163                 err = EIO;
 164                 break;
 165         }
 166         return err;
 167 }
 168 #endif /* _FreeBSD_version */