ceph/src/dpdk/lib/librte_eal/bsdapp/nic_uio/nic_uio.c

   1 /* -
   2  *   BSD LICENSE
   3  *
   4  *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
   5  *   All rights reserved.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following conditions
   9  *   are met:
  10  *
  11  *     * Redistributions of source code must retain the above copyright
  12  *       notice, this list of conditions and the following disclaimer.
  13  *     * Redistributions in binary form must reproduce the above copyright
  14  *       notice, this list of conditions and the following disclaimer in
  15  *       the documentation and/or other materials provided with the
  16  *       distribution.
  17  *     * Neither the name of Intel Corporation nor the names of its
  18  *       contributors may be used to endorse or promote products derived
  19  *       from this software without specific prior written permission.
  20  *
  21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33 #include <sys/cdefs.h>
  34 __FBSDID("$FreeBSD$");
  35
  36 #include <sys/param.h> /* defines used in kernel.h */
  37 #include <sys/module.h>
  38 #include <sys/kernel.h> /* types used in module initialization */
  39 #include <sys/conf.h> /* cdevsw struct */
  40 #include <sys/bus.h> /* structs, prototypes for pci bus stuff and DEVMETHOD */
  41 #include <sys/rman.h>
  42 #include <sys/systm.h>
  43 #include <sys/rwlock.h>
  44 #include <sys/proc.h>
  45
  46 #include <machine/bus.h>
  47 #include <dev/pci/pcivar.h> /* For pci_get macros! */
  48 #include <dev/pci/pcireg.h> /* The softc holds our per-instance data. */
  49 #include <vm/vm.h>
  50 #include <vm/uma.h>
  51 #include <vm/vm_object.h>
  52 #include <vm/vm_page.h>
  53 #include <vm/vm_pager.h>
  54
  55
  56 #define MAX_BARS (PCIR_MAX_BAR_0 + 1)
  57
  58 #define MAX_DETACHED_DEVICES    128
  59 static device_t detached_devices[MAX_DETACHED_DEVICES] = {};
  60 static int num_detached = 0;
  61
  62 struct nic_uio_softc {
  63         device_t        dev_t;
  64         struct cdev     *my_cdev;
  65         int              bar_id[MAX_BARS];
  66         struct resource *bar_res[MAX_BARS];
  67         u_long           bar_start[MAX_BARS];
  68         u_long           bar_size[MAX_BARS];
  69 };
  70
  71 /* Function prototypes */
  72 static d_open_t         nic_uio_open;
  73 static d_close_t        nic_uio_close;
  74 static d_mmap_t         nic_uio_mmap;
  75 static d_mmap_single_t  nic_uio_mmap_single;
  76 static int              nic_uio_probe(device_t dev);
  77 static int              nic_uio_attach(device_t dev);
  78 static int              nic_uio_detach(device_t dev);
  79 static int              nic_uio_shutdown(void);
  80 static int              nic_uio_modevent(module_t mod, int type, void *arg);
  81
  82 static struct cdevsw uio_cdevsw = {
  83                 .d_name        = "nic_uio",
  84                 .d_version     = D_VERSION,
  85                 .d_open        = nic_uio_open,
  86                 .d_close       = nic_uio_close,
  87                 .d_mmap        = nic_uio_mmap,
  88                 .d_mmap_single = nic_uio_mmap_single,
  89 };
  90
  91 static device_method_t nic_uio_methods[] = {
  92         DEVMETHOD(device_probe,    nic_uio_probe),
  93         DEVMETHOD(device_attach,   nic_uio_attach),
  94         DEVMETHOD(device_detach,   nic_uio_detach),
  95         DEVMETHOD_END
  96 };
  97
  98 struct device {
  99     int vend;
 100     int dev;
 101 };
 102
 103 struct pci_bdf {
 104         uint32_t bus;
 105         uint32_t devid;
 106         uint32_t function;
 107 };
 108
 109 static devclass_t nic_uio_devclass;
 110
 111 DEFINE_CLASS_0(nic_uio, nic_uio_driver, nic_uio_methods, sizeof(struct nic_uio_softc));
 112 DRIVER_MODULE(nic_uio, pci, nic_uio_driver, nic_uio_devclass, nic_uio_modevent, 0);
 113
 114 static int
 115 nic_uio_mmap(struct cdev *cdev, vm_ooffset_t offset, vm_paddr_t *paddr,
 116                 int prot, vm_memattr_t *memattr)
 117 {
 118         *paddr = offset;
 119         return 0;
 120 }
 121
 122 static int
 123 nic_uio_mmap_single(struct cdev *cdev, vm_ooffset_t *offset, vm_size_t size,
 124                 struct vm_object **obj, int nprot)
 125 {
 126         /*
 127          * The BAR index is encoded in the offset.  Divide the offset by
 128          *  PAGE_SIZE to get the index of the bar requested by the user
 129          *  app.
 130          */
 131         unsigned bar = *offset/PAGE_SIZE;
 132         struct nic_uio_softc *sc = cdev->si_drv1;
 133
 134         if (bar >= MAX_BARS)
 135                 return EINVAL;
 136
 137         if (sc->bar_res[bar] == NULL) {
 138                 sc->bar_id[bar] = PCIR_BAR(bar);
 139
 140                 if (PCI_BAR_IO(pci_read_config(sc->dev_t, sc->bar_id[bar], 4)))
 141                         sc->bar_res[bar] = bus_alloc_resource_any(sc->dev_t, SYS_RES_IOPORT,
 142                                         &sc->bar_id[bar], RF_ACTIVE);
 143                 else
 144                         sc->bar_res[bar] = bus_alloc_resource_any(sc->dev_t, SYS_RES_MEMORY,
 145                                         &sc->bar_id[bar], RF_ACTIVE);
 146         }
 147         if (sc->bar_res[bar] == NULL)
 148                 return ENXIO;
 149
 150         sc->bar_start[bar] = rman_get_start(sc->bar_res[bar]);
 151         sc->bar_size[bar] = rman_get_size(sc->bar_res[bar]);
 152
 153         device_printf(sc->dev_t, "Bar %u @ %lx, size %lx\n", bar,
 154                         sc->bar_start[bar], sc->bar_size[bar]);
 155
 156         *offset = sc->bar_start[bar];
 157         *obj = vm_pager_allocate(OBJT_DEVICE, cdev, size, nprot, *offset,
 158                                 curthread->td_ucred);
 159         return 0;
 160 }
 161
 162
 163 int
 164 nic_uio_open(struct cdev *dev, int oflags, int devtype, struct thread *td)
 165 {
 166         return 0;
 167 }
 168
 169 int
 170 nic_uio_close(struct cdev *dev, int fflag, int devtype, struct thread *td)
 171 {
 172         return 0;
 173 }
 174
 175 static int
 176 nic_uio_probe (device_t dev)
 177 {
 178         int i;
 179         unsigned int bus = pci_get_bus(dev);
 180         unsigned int device = pci_get_slot(dev);
 181         unsigned int function = pci_get_function(dev);
 182
 183         for (i = 0; i < num_detached; i++)
 184                 if (bus == pci_get_bus(detached_devices[i]) &&
 185                     device == pci_get_slot(detached_devices[i]) &&
 186                     function == pci_get_function(detached_devices[i])) {
 187                         device_set_desc(dev, "DPDK PCI Device");
 188                         return BUS_PROBE_SPECIFIC;
 189                 }
 190
 191         return ENXIO;
 192 }
 193
 194 static int
 195 nic_uio_attach(device_t dev)
 196 {
 197         int i;
 198         struct nic_uio_softc *sc;
 199
 200         sc = device_get_softc(dev);
 201         sc->dev_t = dev;
 202         sc->my_cdev = make_dev(&uio_cdevsw, device_get_unit(dev),
 203                         UID_ROOT, GID_WHEEL, 0600, "uio@pci:%u:%u:%u",
 204                         pci_get_bus(dev), pci_get_slot(dev), pci_get_function(dev));
 205         if (sc->my_cdev == NULL)
 206                 return ENXIO;
 207         sc->my_cdev->si_drv1 = sc;
 208
 209         for (i = 0; i < MAX_BARS; i++)
 210                 sc->bar_res[i] = NULL;
 211
 212         pci_enable_busmaster(dev);
 213
 214         return 0;
 215 }
 216
 217 static int
 218 nic_uio_detach(device_t dev)
 219 {
 220         int i;
 221         struct nic_uio_softc *sc;
 222         sc = device_get_softc(dev);
 223
 224         for (i = 0; i < MAX_BARS; i++)
 225                 if (sc->bar_res[i] != NULL) {
 226
 227                         if (PCI_BAR_IO(pci_read_config(dev, sc->bar_id[i], 4)))
 228                                 bus_release_resource(dev, SYS_RES_IOPORT, sc->bar_id[i],
 229                                                 sc->bar_res[i]);
 230                         else
 231                                 bus_release_resource(dev, SYS_RES_MEMORY, sc->bar_id[i],
 232                                                 sc->bar_res[i]);
 233                 }
 234
 235         if (sc->my_cdev != NULL)
 236                 destroy_dev(sc->my_cdev);
 237         return 0;
 238 }
 239
 240 static void
 241 nic_uio_load(void)
 242 {
 243         uint32_t bus, device, function;
 244         device_t dev;
 245         char bdf_str[256];
 246         char *token, *remaining;
 247
 248         memset(bdf_str, 0, sizeof(bdf_str));
 249         TUNABLE_STR_FETCH("hw.nic_uio.bdfs", bdf_str, sizeof(bdf_str));
 250         remaining = bdf_str;
 251         /*
 252          * Users should specify PCI BDFs in the format "b:d:f,b:d:f,b:d:f".
 253          *  But the code below does not try differentiate between : and ,
 254          *  and just blindly uses 3 tokens at a time to construct a
 255          *  bus/device/function tuple.
 256          *
 257          * There is no checking on strtol() return values, but this should
 258          *  be OK.  Worst case is it cannot convert and returns 0.  This
 259          *  could give us a different BDF than intended, but as long as the
 260          *  PCI device/vendor ID does not match it will not matter.
 261          */
 262         while (1) {
 263                 if (remaining == NULL || remaining[0] == '\0')
 264                         break;
 265                 token = strsep(&remaining, ",:");
 266                 if (token == NULL)
 267                         break;
 268                 bus = strtol(token, NULL, 10);
 269                 token = strsep(&remaining, ",:");
 270                 if (token == NULL)
 271                         break;
 272                 device = strtol(token, NULL, 10);
 273                 token = strsep(&remaining, ",:");
 274                 if (token == NULL)
 275                         break;
 276                 function = strtol(token, NULL, 10);
 277
 278                 dev = pci_find_bsf(bus, device, function);
 279                 if (dev == NULL)
 280                         continue;
 281
 282                 if (num_detached < MAX_DETACHED_DEVICES) {
 283                         printf("nic_uio_load: detaching and storing dev=%p\n",
 284                                dev);
 285                         detached_devices[num_detached++] = dev;
 286                 } else {
 287                         printf("nic_uio_load: reached MAX_DETACHED_DEVICES=%d. dev=%p won't be reattached\n",
 288                                MAX_DETACHED_DEVICES, dev);
 289                 }
 290                 device_detach(dev);
 291         }
 292 }
 293
 294 static void
 295 nic_uio_unload(void)
 296 {
 297         int i;
 298         printf("nic_uio_unload: entered...\n");
 299
 300         for (i = 0; i < num_detached; i++) {
 301                 printf("nic_uio_unload: calling to device_probe_and_attach for dev=%p...\n",
 302                         detached_devices[i]);
 303                 device_probe_and_attach(detached_devices[i]);
 304                 printf("nic_uio_unload: done.\n");
 305         }
 306
 307         printf("nic_uio_unload: leaving...\n");
 308 }
 309
 310 static int
 311 nic_uio_shutdown(void)
 312 {
 313         return 0;
 314 }
 315
 316 static int
 317 nic_uio_modevent(module_t mod, int type, void *arg)
 318 {
 319
 320         switch (type) {
 321         case MOD_LOAD:
 322                 nic_uio_load();
 323                 break;
 324         case MOD_UNLOAD:
 325                 nic_uio_unload();
 326                 break;
 327         case MOD_SHUTDOWN:
 328                 nic_uio_shutdown();
 329                 break;
 330         default:
 331                 break;
 332         }
 333
 334         return 0;
 335 }