sys/dev/wtap/if_wtap_module.c

   1 /*-
   2  * Copyright (c) 2010-2011 Monthadar Al Jaberi, TerraNet AB
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer,
  10  *    without modification.
  11  * 2. Redistributions in binary form must reproduce at minimum a disclaimer
  12  *    similar to the "NO WARRANTY" disclaimer below ("Disclaimer") and any
  13  *    redistribution must be conditioned upon including a substantially
  14  *    similar Disclaimer requirement for further binary redistribution.
  15  *
  16  * NO WARRANTY
  17  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  18  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  19  * LIMITED TO, THE IMPLIED WARRANTIES OF NONINFRINGEMENT, MERCHANTIBILITY
  20  * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
  21  * THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR SPECIAL, EXEMPLARY,
  22  * OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  23  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  24  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
  25  * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  26  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  27  * THE POSSIBILITY OF SUCH DAMAGES.
  28  *
  29  * $FreeBSD$
  30  */
  31 #include <sys/param.h>
  32 #include <sys/module.h>
  33 #include <sys/kernel.h>
  34 #include <sys/systm.h>
  35 #include <sys/sysctl.h>
  36 #include <sys/mbuf.h>
  37 #include <sys/malloc.h>
  38 #include <sys/lock.h>
  39 #include <sys/mutex.h>
  40 #include <sys/proc.h>
  41 #include <sys/ucred.h>
  42 #include <sys/jail.h>
  43
  44 #include <sys/sockio.h>
  45 #include <sys/socket.h>
  46 #include <sys/socketvar.h>
  47 #include <sys/errno.h>
  48 #include <sys/callout.h>
  49 #include <sys/endian.h>
  50 #include <sys/kthread.h>
  51 #include <sys/taskqueue.h>
  52 #include <sys/priv.h>
  53 #include <sys/sysctl.h>
  54
  55 #include <machine/bus.h>
  56
  57 #include <net/if.h>
  58 #include <net/if_dl.h>
  59 #include <net/if_media.h>
  60 #include <net/if_types.h>
  61 #include <net/if_arp.h>
  62 #include <net/ethernet.h>
  63 #include <net/if_llc.h>
  64 #include <net/vnet.h>
  65
  66 #include <net80211/ieee80211_var.h>
  67 #include <net80211/ieee80211_regdomain.h>
  68
  69 #include <net/bpf.h>
  70
  71
  72 #include <sys/errno.h>
  73 #include <sys/conf.h>   /* cdevsw struct */
  74 #include <sys/uio.h>    /* uio struct */
  75
  76
  77 #include <netinet/in.h>
  78 #include <netinet/if_ether.h>
  79
  80 #include "if_wtapvar.h"
  81 #include "if_wtapioctl.h"
  82 #include "if_medium.h"
  83 #include "wtap_hal/hal.h"
  84
  85 /* WTAP PLUGINS */
  86 #include "plugins/visibility.h"
  87
  88 MALLOC_DEFINE(M_WTAP, "wtap", "wtap wireless simulator");
  89 MALLOC_DEFINE(M_WTAP_PACKET, "wtap packet", "wtap wireless simulator packet");
  90 MALLOC_DEFINE(M_WTAP_RXBUF, "wtap rxbuf",
  91     "wtap wireless simulator receive buffer");
  92 MALLOC_DEFINE(M_WTAP_PLUGIN, "wtap plugin", "wtap wireless simulator plugin");
  93
  94 static struct wtap_hal          *hal;
  95
  96 /* Function prototypes */
  97 static d_ioctl_t        wtap_ioctl;
  98
  99 static struct cdev *sdev;
 100 static struct cdevsw wtap_cdevsw = {
 101         .d_version =    D_VERSION,
 102         .d_flags =      0,
 103         .d_ioctl =      wtap_ioctl,
 104         .d_name =       "wtapctl",
 105 };
 106
 107 int
 108 wtap_ioctl(struct cdev *dev, u_long cmd, caddr_t data,
 109     int fflag, struct thread *td)
 110 {
 111         int error = 0;
 112
 113         CURVNET_SET(CRED_TO_VNET(curthread->td_ucred));
 114
 115         switch(cmd) {
 116         case WTAPIOCTLCRT:
 117                 if(new_wtap(hal, *(int *)data))
 118                         error = EINVAL;
 119                 break;
 120         case WTAPIOCTLDEL:
 121                 if(free_wtap(hal, *(int *)data))
 122                         error = EINVAL;
 123                 break;
 124         default:
 125                 DWTAP_PRINTF("Unknown WTAP IOCTL\n");
 126                 error = EINVAL;
 127         }
 128
 129         CURVNET_RESTORE();
 130         return error;
 131 }
 132
 133
 134 /* The function called at load/unload. */
 135 static int
 136 event_handler(module_t module, int event, void *arg)
 137 {
 138         struct visibility_plugin *plugin;
 139         int e = 0; /* Error, 0 for normal return status */
 140
 141         switch (event) {
 142         case MOD_LOAD:
 143                 sdev = make_dev(&wtap_cdevsw,0,UID_ROOT,
 144                     GID_WHEEL,0600,(const char *)"wtapctl");
 145                 hal = (struct wtap_hal *)malloc(sizeof(struct wtap_hal),
 146                     M_WTAP, M_NOWAIT | M_ZERO);
 147                 bzero(hal, sizeof(struct wtap_hal));
 148
 149                 init_hal(hal);
 150
 151                 /* Setting up a simple plugin */
 152                 plugin = (struct visibility_plugin *)malloc
 153                     (sizeof(struct visibility_plugin), M_WTAP_PLUGIN,
 154                     M_NOWAIT | M_ZERO);
 155                 plugin->base.wp_hal  = hal;
 156                 plugin->base.init = visibility_init;
 157                 plugin->base.deinit = visibility_deinit;
 158                 plugin->base.work = visibility_work;
 159                 register_plugin(hal, (struct wtap_plugin *)plugin);
 160
 161                 printf("Loaded wtap wireless simulator\n");
 162                 break;
 163         case MOD_UNLOAD:
 164                 destroy_dev(sdev);
 165                 deregister_plugin(hal);
 166                 deinit_hal(hal);
 167                 free(hal, M_WTAP);
 168                 printf("Unloading wtap wireless simulator\n");
 169                 break;
 170         default:
 171                 e = EOPNOTSUPP; /* Error, Operation Not Supported */
 172                 break;
 173         }
 174
 175         return(e);
 176 }
 177
 178 /* The second argument of DECLARE_MODULE. */
 179 static moduledata_t wtap_conf = {
 180         "wtap",         /* module name */
 181         event_handler,  /* event handler */
 182         NULL            /* extra data */
 183 };
 184
 185 DECLARE_MODULE(wtap, wtap_conf, SI_SUB_DRIVERS, SI_ORDER_MIDDLE);
 186 MODULE_DEPEND(wtap, wlan, 1, 1, 1);     /* 802.11 media layer */