sys/compat/linuxkpi/common/src/linux_pci.c

   1 /*-
   2  * Copyright (c) 2015-2016 Mellanox Technologies, Ltd.
   3  * All rights reserved.
   4  * Copyright (c) 2020-2022 The FreeBSD Foundation
   5  *
   6  * Portions of this software were developed by Björn Zeeb
   7  * under sponsorship from the FreeBSD Foundation.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice unmodified, this list of conditions, and the following
  14  *    disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  20  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  21  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  22  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  23  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  24  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  28  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  */
  30
  31 #include <sys/cdefs.h>
  32 __FBSDID("$FreeBSD$");
  33
  34 #include <sys/param.h>
  35 #include <sys/systm.h>
  36 #include <sys/bus.h>
  37 #include <sys/malloc.h>
  38 #include <sys/kernel.h>
  39 #include <sys/sysctl.h>
  40 #include <sys/lock.h>
  41 #include <sys/mutex.h>
  42 #include <sys/fcntl.h>
  43 #include <sys/file.h>
  44 #include <sys/filio.h>
  45 #include <sys/pciio.h>
  46 #include <sys/pctrie.h>
  47 #include <sys/rwlock.h>
  48
  49 #include <vm/vm.h>
  50 #include <vm/pmap.h>
  51
  52 #include <machine/stdarg.h>
  53
  54 #include <dev/pci/pcivar.h>
  55 #include <dev/pci/pci_private.h>
  56 #include <dev/pci/pci_iov.h>
  57 #include <dev/backlight/backlight.h>
  58
  59 #include <linux/kobject.h>
  60 #include <linux/device.h>
  61 #include <linux/slab.h>
  62 #include <linux/module.h>
  63 #include <linux/cdev.h>
  64 #include <linux/file.h>
  65 #include <linux/sysfs.h>
  66 #include <linux/mm.h>
  67 #include <linux/io.h>
  68 #include <linux/vmalloc.h>
  69 #include <linux/pci.h>
  70 #include <linux/compat.h>
  71
  72 #include <linux/backlight.h>
  73
  74 #include "backlight_if.h"
  75 #include "pcib_if.h"
  76
  77 /* Undef the linux function macro defined in linux/pci.h */
  78 #undef pci_get_class
  79
  80 static device_probe_t linux_pci_probe;
  81 static device_attach_t linux_pci_attach;
  82 static device_detach_t linux_pci_detach;
  83 static device_suspend_t linux_pci_suspend;
  84 static device_resume_t linux_pci_resume;
  85 static device_shutdown_t linux_pci_shutdown;
  86 static pci_iov_init_t linux_pci_iov_init;
  87 static pci_iov_uninit_t linux_pci_iov_uninit;
  88 static pci_iov_add_vf_t linux_pci_iov_add_vf;
  89 static int linux_backlight_get_status(device_t dev, struct backlight_props *props);
  90 static int linux_backlight_update_status(device_t dev, struct backlight_props *props);
  91 static int linux_backlight_get_info(device_t dev, struct backlight_info *info);
  92
  93 static device_method_t pci_methods[] = {
  94         DEVMETHOD(device_probe, linux_pci_probe),
  95         DEVMETHOD(device_attach, linux_pci_attach),
  96         DEVMETHOD(device_detach, linux_pci_detach),
  97         DEVMETHOD(device_suspend, linux_pci_suspend),
  98         DEVMETHOD(device_resume, linux_pci_resume),
  99         DEVMETHOD(device_shutdown, linux_pci_shutdown),
 100         DEVMETHOD(pci_iov_init, linux_pci_iov_init),
 101         DEVMETHOD(pci_iov_uninit, linux_pci_iov_uninit),
 102         DEVMETHOD(pci_iov_add_vf, linux_pci_iov_add_vf),
 103
 104         /* backlight interface */
 105         DEVMETHOD(backlight_update_status, linux_backlight_update_status),
 106         DEVMETHOD(backlight_get_status, linux_backlight_get_status),
 107         DEVMETHOD(backlight_get_info, linux_backlight_get_info),
 108         DEVMETHOD_END
 109 };
 110
 111 struct linux_dma_priv {
 112         uint64_t        dma_mask;
 113         bus_dma_tag_t   dmat;
 114         uint64_t        dma_coherent_mask;
 115         bus_dma_tag_t   dmat_coherent;
 116         struct mtx      lock;
 117         struct pctrie   ptree;
 118 };
 119 #define DMA_PRIV_LOCK(priv) mtx_lock(&(priv)->lock)
 120 #define DMA_PRIV_UNLOCK(priv) mtx_unlock(&(priv)->lock)
 121
 122 static bool
 123 linux_is_drm(struct pci_driver *pdrv)
 124 {
 125         return (pdrv->name != NULL && strcmp(pdrv->name, "drmn") == 0);
 126 }
 127
 128 static int
 129 linux_pdev_dma_uninit(struct pci_dev *pdev)
 130 {
 131         struct linux_dma_priv *priv;
 132
 133         priv = pdev->dev.dma_priv;
 134         if (priv->dmat)
 135                 bus_dma_tag_destroy(priv->dmat);
 136         if (priv->dmat_coherent)
 137                 bus_dma_tag_destroy(priv->dmat_coherent);
 138         mtx_destroy(&priv->lock);
 139         pdev->dev.dma_priv = NULL;
 140         free(priv, M_DEVBUF);
 141         return (0);
 142 }
 143
 144 static int
 145 linux_pdev_dma_init(struct pci_dev *pdev)
 146 {
 147         struct linux_dma_priv *priv;
 148         int error;
 149
 150         priv = malloc(sizeof(*priv), M_DEVBUF, M_WAITOK | M_ZERO);
 151
 152         mtx_init(&priv->lock, "lkpi-priv-dma", NULL, MTX_DEF);
 153         pctrie_init(&priv->ptree);
 154
 155         pdev->dev.dma_priv = priv;
 156
 157         /* Create a default DMA tags. */
 158         error = linux_dma_tag_init(&pdev->dev, DMA_BIT_MASK(64));
 159         if (error != 0)
 160                 goto err;
 161         /* Coherent is lower 32bit only by default in Linux. */
 162         error = linux_dma_tag_init_coherent(&pdev->dev, DMA_BIT_MASK(32));
 163         if (error != 0)
 164                 goto err;
 165
 166         return (error);
 167
 168 err:
 169         linux_pdev_dma_uninit(pdev);
 170         return (error);
 171 }
 172
 173 int
 174 linux_dma_tag_init(struct device *dev, u64 dma_mask)
 175 {
 176         struct linux_dma_priv *priv;
 177         int error;
 178
 179         priv = dev->dma_priv;
 180
 181         if (priv->dmat) {
 182                 if (priv->dma_mask == dma_mask)
 183                         return (0);
 184
 185                 bus_dma_tag_destroy(priv->dmat);
 186         }
 187
 188         priv->dma_mask = dma_mask;
 189
 190         error = bus_dma_tag_create(bus_get_dma_tag(dev->bsddev),
 191             1, 0,                       /* alignment, boundary */
 192             dma_mask,                   /* lowaddr */
 193             BUS_SPACE_MAXADDR,          /* highaddr */
 194             NULL, NULL,                 /* filtfunc, filtfuncarg */
 195             BUS_SPACE_MAXSIZE,          /* maxsize */
 196             1,                          /* nsegments */
 197             BUS_SPACE_MAXSIZE,          /* maxsegsz */
 198             0,                          /* flags */
 199             NULL, NULL,                 /* lockfunc, lockfuncarg */
 200             &priv->dmat);
 201         return (-error);
 202 }
 203
 204 int
 205 linux_dma_tag_init_coherent(struct device *dev, u64 dma_mask)
 206 {
 207         struct linux_dma_priv *priv;
 208         int error;
 209
 210         priv = dev->dma_priv;
 211
 212         if (priv->dmat_coherent) {
 213                 if (priv->dma_coherent_mask == dma_mask)
 214                         return (0);
 215
 216                 bus_dma_tag_destroy(priv->dmat_coherent);
 217         }
 218
 219         priv->dma_coherent_mask = dma_mask;
 220
 221         error = bus_dma_tag_create(bus_get_dma_tag(dev->bsddev),
 222             1, 0,                       /* alignment, boundary */
 223             dma_mask,                   /* lowaddr */
 224             BUS_SPACE_MAXADDR,          /* highaddr */
 225             NULL, NULL,                 /* filtfunc, filtfuncarg */
 226             BUS_SPACE_MAXSIZE,          /* maxsize */
 227             1,                          /* nsegments */
 228             BUS_SPACE_MAXSIZE,          /* maxsegsz */
 229             0,                          /* flags */
 230             NULL, NULL,                 /* lockfunc, lockfuncarg */
 231             &priv->dmat_coherent);
 232         return (-error);
 233 }
 234
 235 static struct pci_driver *
 236 linux_pci_find(device_t dev, const struct pci_device_id **idp)
 237 {
 238         const struct pci_device_id *id;
 239         struct pci_driver *pdrv;
 240         uint16_t vendor;
 241         uint16_t device;
 242         uint16_t subvendor;
 243         uint16_t subdevice;
 244
 245         vendor = pci_get_vendor(dev);
 246         device = pci_get_device(dev);
 247         subvendor = pci_get_subvendor(dev);
 248         subdevice = pci_get_subdevice(dev);
 249
 250         spin_lock(&pci_lock);
 251         list_for_each_entry(pdrv, &pci_drivers, node) {
 252                 for (id = pdrv->id_table; id->vendor != 0; id++) {
 253                         if (vendor == id->vendor &&
 254                             (PCI_ANY_ID == id->device || device == id->device) &&
 255                             (PCI_ANY_ID == id->subvendor || subvendor == id->subvendor) &&
 256                             (PCI_ANY_ID == id->subdevice || subdevice == id->subdevice)) {
 257                                 *idp = id;
 258                                 spin_unlock(&pci_lock);
 259                                 return (pdrv);
 260                         }
 261                 }
 262         }
 263         spin_unlock(&pci_lock);
 264         return (NULL);
 265 }
 266
 267 static void
 268 lkpi_pci_dev_release(struct device *dev)
 269 {
 270
 271         lkpi_devres_release_free_list(dev);
 272         spin_lock_destroy(&dev->devres_lock);
 273 }
 274
 275 static void
 276 lkpifill_pci_dev(device_t dev, struct pci_dev *pdev)
 277 {
 278
 279         pdev->devfn = PCI_DEVFN(pci_get_slot(dev), pci_get_function(dev));
 280         pdev->vendor = pci_get_vendor(dev);
 281         pdev->device = pci_get_device(dev);
 282         pdev->subsystem_vendor = pci_get_subvendor(dev);
 283         pdev->subsystem_device = pci_get_subdevice(dev);
 284         pdev->class = pci_get_class(dev);
 285         pdev->revision = pci_get_revid(dev);
 286         pdev->bus = malloc(sizeof(*pdev->bus), M_DEVBUF, M_WAITOK | M_ZERO);
 287         /*
 288          * This should be the upstream bridge; pci_upstream_bridge()
 289          * handles that case on demand as otherwise we'll shadow the
 290          * entire PCI hierarchy.
 291          */
 292         pdev->bus->self = pdev;
 293         pdev->bus->number = pci_get_bus(dev);
 294         pdev->bus->domain = pci_get_domain(dev);
 295         pdev->dev.bsddev = dev;
 296         pdev->dev.parent = &linux_root_device;
 297         pdev->dev.release = lkpi_pci_dev_release;
 298         INIT_LIST_HEAD(&pdev->dev.irqents);
 299         kobject_init(&pdev->dev.kobj, &linux_dev_ktype);
 300         kobject_set_name(&pdev->dev.kobj, device_get_nameunit(dev));
 301         kobject_add(&pdev->dev.kobj, &linux_root_device.kobj,
 302             kobject_name(&pdev->dev.kobj));
 303         spin_lock_init(&pdev->dev.devres_lock);
 304         INIT_LIST_HEAD(&pdev->dev.devres_head);
 305 }
 306
 307 static void
 308 lkpinew_pci_dev_release(struct device *dev)
 309 {
 310         struct pci_dev *pdev;
 311
 312         pdev = to_pci_dev(dev);
 313         if (pdev->root != NULL)
 314                 pci_dev_put(pdev->root);
 315         if (pdev->bus->self != pdev)
 316                 pci_dev_put(pdev->bus->self);
 317         free(pdev->bus, M_DEVBUF);
 318         free(pdev, M_DEVBUF);
 319 }
 320
 321 struct pci_dev *
 322 lkpinew_pci_dev(device_t dev)
 323 {
 324         struct pci_dev *pdev;
 325
 326         pdev = malloc(sizeof(*pdev), M_DEVBUF, M_WAITOK|M_ZERO);
 327         lkpifill_pci_dev(dev, pdev);
 328         pdev->dev.release = lkpinew_pci_dev_release;
 329
 330         return (pdev);
 331 }
 332
 333 struct pci_dev *
 334 lkpi_pci_get_class(unsigned int class, struct pci_dev *from)
 335 {
 336         device_t dev;
 337         device_t devfrom = NULL;
 338         struct pci_dev *pdev;
 339
 340         if (from != NULL)
 341                 devfrom = from->dev.bsddev;
 342
 343         dev = pci_find_class_from(class >> 16, (class >> 8) & 0xFF, devfrom);
 344         if (dev == NULL)
 345                 return (NULL);
 346
 347         pdev = lkpinew_pci_dev(dev);
 348         return (pdev);
 349 }
 350
 351 struct pci_dev *
 352 lkpi_pci_get_domain_bus_and_slot(int domain, unsigned int bus,
 353     unsigned int devfn)
 354 {
 355         device_t dev;
 356         struct pci_dev *pdev;
 357
 358         dev = pci_find_dbsf(domain, bus, PCI_SLOT(devfn), PCI_FUNC(devfn));
 359         if (dev == NULL)
 360                 return (NULL);
 361
 362         pdev = lkpinew_pci_dev(dev);
 363         return (pdev);
 364 }
 365
 366 static int
 367 linux_pci_probe(device_t dev)
 368 {
 369         const struct pci_device_id *id;
 370         struct pci_driver *pdrv;
 371
 372         if ((pdrv = linux_pci_find(dev, &id)) == NULL)
 373                 return (ENXIO);
 374         if (device_get_driver(dev) != &pdrv->bsddriver)
 375                 return (ENXIO);
 376         device_set_desc(dev, pdrv->name);
 377
 378         /* Assume BSS initialized (should never return BUS_PROBE_SPECIFIC). */
 379         if (pdrv->bsd_probe_return == 0)
 380                 return (BUS_PROBE_DEFAULT);
 381         else
 382                 return (pdrv->bsd_probe_return);
 383 }
 384
 385 static int
 386 linux_pci_attach(device_t dev)
 387 {
 388         const struct pci_device_id *id;
 389         struct pci_driver *pdrv;
 390         struct pci_dev *pdev;
 391
 392         pdrv = linux_pci_find(dev, &id);
 393         pdev = device_get_softc(dev);
 394
 395         MPASS(pdrv != NULL);
 396         MPASS(pdev != NULL);
 397
 398         return (linux_pci_attach_device(dev, pdrv, id, pdev));
 399 }
 400
 401 int
 402 linux_pci_attach_device(device_t dev, struct pci_driver *pdrv,
 403     const struct pci_device_id *id, struct pci_dev *pdev)
 404 {
 405         struct resource_list_entry *rle;
 406         device_t parent;
 407         uintptr_t rid;
 408         int error;
 409         bool isdrm;
 410
 411         linux_set_current(curthread);
 412
 413         parent = device_get_parent(dev);
 414         isdrm = pdrv != NULL && linux_is_drm(pdrv);
 415
 416         if (isdrm) {
 417                 struct pci_devinfo *dinfo;
 418
 419                 dinfo = device_get_ivars(parent);
 420                 device_set_ivars(dev, dinfo);
 421         }
 422
 423         lkpifill_pci_dev(dev, pdev);
 424         if (isdrm)
 425                 PCI_GET_ID(device_get_parent(parent), parent, PCI_ID_RID, &rid);
 426         else
 427                 PCI_GET_ID(parent, dev, PCI_ID_RID, &rid);
 428         pdev->devfn = rid;
 429         pdev->pdrv = pdrv;
 430         rle = linux_pci_get_rle(pdev, SYS_RES_IRQ, 0, false);
 431         if (rle != NULL)
 432                 pdev->dev.irq = rle->start;
 433         else
 434                 pdev->dev.irq = LINUX_IRQ_INVALID;
 435         pdev->irq = pdev->dev.irq;
 436         error = linux_pdev_dma_init(pdev);
 437         if (error)
 438                 goto out_dma_init;
 439
 440         TAILQ_INIT(&pdev->mmio);
 441
 442         spin_lock(&pci_lock);
 443         list_add(&pdev->links, &pci_devices);
 444         spin_unlock(&pci_lock);
 445
 446         if (pdrv != NULL) {
 447                 error = pdrv->probe(pdev, id);
 448                 if (error)
 449                         goto out_probe;
 450         }
 451         return (0);
 452
 453 out_probe:
 454         free(pdev->bus, M_DEVBUF);
 455         linux_pdev_dma_uninit(pdev);
 456 out_dma_init:
 457         spin_lock(&pci_lock);
 458         list_del(&pdev->links);
 459         spin_unlock(&pci_lock);
 460         put_device(&pdev->dev);
 461         return (-error);
 462 }
 463
 464 static int
 465 linux_pci_detach(device_t dev)
 466 {
 467         struct pci_dev *pdev;
 468
 469         pdev = device_get_softc(dev);
 470
 471         MPASS(pdev != NULL);
 472
 473         device_set_desc(dev, NULL);
 474
 475         return (linux_pci_detach_device(pdev));
 476 }
 477
 478 int
 479 linux_pci_detach_device(struct pci_dev *pdev)
 480 {
 481
 482         linux_set_current(curthread);
 483
 484         if (pdev->pdrv != NULL)
 485                 pdev->pdrv->remove(pdev);
 486
 487         if (pdev->root != NULL)
 488                 pci_dev_put(pdev->root);
 489         free(pdev->bus, M_DEVBUF);
 490         linux_pdev_dma_uninit(pdev);
 491
 492         spin_lock(&pci_lock);
 493         list_del(&pdev->links);
 494         spin_unlock(&pci_lock);
 495         put_device(&pdev->dev);
 496
 497         return (0);
 498 }
 499
 500 static int
 501 lkpi_pci_disable_dev(struct device *dev)
 502 {
 503
 504         (void) pci_disable_io(dev->bsddev, SYS_RES_MEMORY);
 505         (void) pci_disable_io(dev->bsddev, SYS_RES_IOPORT);
 506         return (0);
 507 }
 508
 509 struct pci_devres *
 510 lkpi_pci_devres_get_alloc(struct pci_dev *pdev)
 511 {
 512         struct pci_devres *dr;
 513
 514         dr = lkpi_devres_find(&pdev->dev, lkpi_pci_devres_release, NULL, NULL);
 515         if (dr == NULL) {
 516                 dr = lkpi_devres_alloc(lkpi_pci_devres_release, sizeof(*dr),
 517                     GFP_KERNEL | __GFP_ZERO);
 518                 if (dr != NULL)
 519                         lkpi_devres_add(&pdev->dev, dr);
 520         }
 521
 522         return (dr);
 523 }
 524
 525 void
 526 lkpi_pci_devres_release(struct device *dev, void *p)
 527 {
 528         struct pci_devres *dr;
 529         struct pci_dev *pdev;
 530         int bar;
 531
 532         pdev = to_pci_dev(dev);
 533         dr = p;
 534
 535         if (pdev->msix_enabled)
 536                 lkpi_pci_disable_msix(pdev);
 537         if (pdev->msi_enabled)
 538                 lkpi_pci_disable_msi(pdev);
 539
 540         if (dr->enable_io && lkpi_pci_disable_dev(dev) == 0)
 541                 dr->enable_io = false;
 542
 543         if (dr->region_mask == 0)
 544                 return;
 545         for (bar = PCIR_MAX_BAR_0; bar >= 0; bar--) {
 546
 547                 if ((dr->region_mask & (1 << bar)) == 0)
 548                         continue;
 549                 pci_release_region(pdev, bar);
 550         }
 551 }
 552
 553 void
 554 lkpi_pcim_iomap_table_release(struct device *dev, void *p)
 555 {
 556         struct pcim_iomap_devres *dr;
 557         struct pci_dev *pdev;
 558         int bar;
 559
 560         dr = p;
 561         pdev = to_pci_dev(dev);
 562         for (bar = PCIR_MAX_BAR_0; bar >= 0; bar--) {
 563
 564                 if (dr->mmio_table[bar] == NULL)
 565                         continue;
 566
 567                 pci_iounmap(pdev, dr->mmio_table[bar]);
 568         }
 569 }
 570
 571 static int
 572 linux_pci_suspend(device_t dev)
 573 {
 574         const struct dev_pm_ops *pmops;
 575         struct pm_message pm = { };
 576         struct pci_dev *pdev;
 577         int error;
 578
 579         error = 0;
 580         linux_set_current(curthread);
 581         pdev = device_get_softc(dev);
 582         pmops = pdev->pdrv->driver.pm;
 583
 584         if (pdev->pdrv->suspend != NULL)
 585                 error = -pdev->pdrv->suspend(pdev, pm);
 586         else if (pmops != NULL && pmops->suspend != NULL) {
 587                 error = -pmops->suspend(&pdev->dev);
 588                 if (error == 0 && pmops->suspend_late != NULL)
 589                         error = -pmops->suspend_late(&pdev->dev);
 590         }
 591         return (error);
 592 }
 593
 594 static int
 595 linux_pci_resume(device_t dev)
 596 {
 597         const struct dev_pm_ops *pmops;
 598         struct pci_dev *pdev;
 599         int error;
 600
 601         error = 0;
 602         linux_set_current(curthread);
 603         pdev = device_get_softc(dev);
 604         pmops = pdev->pdrv->driver.pm;
 605
 606         if (pdev->pdrv->resume != NULL)
 607                 error = -pdev->pdrv->resume(pdev);
 608         else if (pmops != NULL && pmops->resume != NULL) {
 609                 if (pmops->resume_early != NULL)
 610                         error = -pmops->resume_early(&pdev->dev);
 611                 if (error == 0 && pmops->resume != NULL)
 612                         error = -pmops->resume(&pdev->dev);
 613         }
 614         return (error);
 615 }
 616
 617 static int
 618 linux_pci_shutdown(device_t dev)
 619 {
 620         struct pci_dev *pdev;
 621
 622         linux_set_current(curthread);
 623         pdev = device_get_softc(dev);
 624         if (pdev->pdrv->shutdown != NULL)
 625                 pdev->pdrv->shutdown(pdev);
 626         return (0);
 627 }
 628
 629 static int
 630 linux_pci_iov_init(device_t dev, uint16_t num_vfs, const nvlist_t *pf_config)
 631 {
 632         struct pci_dev *pdev;
 633         int error;
 634
 635         linux_set_current(curthread);
 636         pdev = device_get_softc(dev);
 637         if (pdev->pdrv->bsd_iov_init != NULL)
 638                 error = pdev->pdrv->bsd_iov_init(dev, num_vfs, pf_config);
 639         else
 640                 error = EINVAL;
 641         return (error);
 642 }
 643
 644 static void
 645 linux_pci_iov_uninit(device_t dev)
 646 {
 647         struct pci_dev *pdev;
 648
 649         linux_set_current(curthread);
 650         pdev = device_get_softc(dev);
 651         if (pdev->pdrv->bsd_iov_uninit != NULL)
 652                 pdev->pdrv->bsd_iov_uninit(dev);
 653 }
 654
 655 static int
 656 linux_pci_iov_add_vf(device_t dev, uint16_t vfnum, const nvlist_t *vf_config)
 657 {
 658         struct pci_dev *pdev;
 659         int error;
 660
 661         linux_set_current(curthread);
 662         pdev = device_get_softc(dev);
 663         if (pdev->pdrv->bsd_iov_add_vf != NULL)
 664                 error = pdev->pdrv->bsd_iov_add_vf(dev, vfnum, vf_config);
 665         else
 666                 error = EINVAL;
 667         return (error);
 668 }
 669
 670 static int
 671 _linux_pci_register_driver(struct pci_driver *pdrv, devclass_t dc)
 672 {
 673         int error;
 674
 675         linux_set_current(curthread);
 676         spin_lock(&pci_lock);
 677         list_add(&pdrv->node, &pci_drivers);
 678         spin_unlock(&pci_lock);
 679         if (pdrv->bsddriver.name == NULL)
 680                 pdrv->bsddriver.name = pdrv->name;
 681         pdrv->bsddriver.methods = pci_methods;
 682         pdrv->bsddriver.size = sizeof(struct pci_dev);
 683
 684         mtx_lock(&Giant);
 685         error = devclass_add_driver(dc, &pdrv->bsddriver,
 686             BUS_PASS_DEFAULT, &pdrv->bsdclass);
 687         mtx_unlock(&Giant);
 688         return (-error);
 689 }
 690
 691 int
 692 linux_pci_register_driver(struct pci_driver *pdrv)
 693 {
 694         devclass_t dc;
 695
 696         dc = devclass_find("pci");
 697         if (dc == NULL)
 698                 return (-ENXIO);
 699         return (_linux_pci_register_driver(pdrv, dc));
 700 }
 701
 702 struct resource_list_entry *
 703 linux_pci_reserve_bar(struct pci_dev *pdev, struct resource_list *rl,
 704     int type, int rid)
 705 {
 706         device_t dev;
 707         struct resource *res;
 708
 709         KASSERT(type == SYS_RES_IOPORT || type == SYS_RES_MEMORY,
 710             ("trying to reserve non-BAR type %d", type));
 711
 712         dev = pdev->pdrv != NULL && linux_is_drm(pdev->pdrv) ?
 713             device_get_parent(pdev->dev.bsddev) : pdev->dev.bsddev;
 714         res = pci_reserve_map(device_get_parent(dev), dev, type, &rid, 0, ~0,
 715             1, 1, 0);
 716         if (res == NULL)
 717                 return (NULL);
 718         return (resource_list_find(rl, type, rid));
 719 }
 720
 721 unsigned long
 722 pci_resource_start(struct pci_dev *pdev, int bar)
 723 {
 724         struct resource_list_entry *rle;
 725         rman_res_t newstart;
 726         device_t dev;
 727
 728         if ((rle = linux_pci_get_bar(pdev, bar, true)) == NULL)
 729                 return (0);
 730         dev = pdev->pdrv != NULL && linux_is_drm(pdev->pdrv) ?
 731             device_get_parent(pdev->dev.bsddev) : pdev->dev.bsddev;
 732         if (BUS_TRANSLATE_RESOURCE(dev, rle->type, rle->start, &newstart)) {
 733                 device_printf(pdev->dev.bsddev, "translate of %#jx failed\n",
 734                     (uintmax_t)rle->start);
 735                 return (0);
 736         }
 737         return (newstart);
 738 }
 739
 740 unsigned long
 741 pci_resource_len(struct pci_dev *pdev, int bar)
 742 {
 743         struct resource_list_entry *rle;
 744
 745         if ((rle = linux_pci_get_bar(pdev, bar, true)) == NULL)
 746                 return (0);
 747         return (rle->count);
 748 }
 749
 750 int
 751 pci_request_region(struct pci_dev *pdev, int bar, const char *res_name)
 752 {
 753         struct resource *res;
 754         struct pci_devres *dr;
 755         struct pci_mmio_region *mmio;
 756         int rid;
 757         int type;
 758
 759         type = pci_resource_type(pdev, bar);
 760         if (type < 0)
 761                 return (-ENODEV);
 762         rid = PCIR_BAR(bar);
 763         res = bus_alloc_resource_any(pdev->dev.bsddev, type, &rid,
 764             RF_ACTIVE|RF_SHAREABLE);
 765         if (res == NULL) {
 766                 device_printf(pdev->dev.bsddev, "%s: failed to alloc "
 767                     "bar %d type %d rid %d\n",
 768                     __func__, bar, type, PCIR_BAR(bar));
 769                 return (-ENODEV);
 770         }
 771
 772         /*
 773          * It seems there is an implicit devres tracking on these if the device
 774          * is managed; otherwise the resources are not automatiaclly freed on
 775          * FreeBSD/LinuxKPI tough they should be/are expected to be by Linux
 776          * drivers.
 777          */
 778         dr = lkpi_pci_devres_find(pdev);
 779         if (dr != NULL) {
 780                 dr->region_mask |= (1 << bar);
 781                 dr->region_table[bar] = res;
 782         }
 783
 784         /* Even if the device is not managed we need to track it for iomap. */
 785         mmio = malloc(sizeof(*mmio), M_DEVBUF, M_WAITOK | M_ZERO);
 786         mmio->rid = PCIR_BAR(bar);
 787         mmio->type = type;
 788         mmio->res = res;
 789         TAILQ_INSERT_TAIL(&pdev->mmio, mmio, next);
 790
 791         return (0);
 792 }
 793
 794 struct resource *
 795 _lkpi_pci_iomap(struct pci_dev *pdev, int bar, int mmio_size __unused)
 796 {
 797         struct pci_mmio_region *mmio, *p;
 798         int type;
 799
 800         type = pci_resource_type(pdev, bar);
 801         if (type < 0) {
 802                 device_printf(pdev->dev.bsddev, "%s: bar %d type %d\n",
 803                      __func__, bar, type);
 804                 return (NULL);
 805         }
 806
 807         /*
 808          * Check for duplicate mappings.
 809          * This can happen if a driver calls pci_request_region() first.
 810          */
 811         TAILQ_FOREACH_SAFE(mmio, &pdev->mmio, next, p) {
 812                 if (mmio->type == type && mmio->rid == PCIR_BAR(bar)) {
 813                         return (mmio->res);
 814                 }
 815         }
 816
 817         mmio = malloc(sizeof(*mmio), M_DEVBUF, M_WAITOK | M_ZERO);
 818         mmio->rid = PCIR_BAR(bar);
 819         mmio->type = type;
 820         mmio->res = bus_alloc_resource_any(pdev->dev.bsddev, mmio->type,
 821             &mmio->rid, RF_ACTIVE|RF_SHAREABLE);
 822         if (mmio->res == NULL) {
 823                 device_printf(pdev->dev.bsddev, "%s: failed to alloc "
 824                     "bar %d type %d rid %d\n",
 825                     __func__, bar, type, PCIR_BAR(bar));
 826                 free(mmio, M_DEVBUF);
 827                 return (NULL);
 828         }
 829         TAILQ_INSERT_TAIL(&pdev->mmio, mmio, next);
 830
 831         return (mmio->res);
 832 }
 833
 834 int
 835 linux_pci_register_drm_driver(struct pci_driver *pdrv)
 836 {
 837         devclass_t dc;
 838
 839         dc = devclass_create("vgapci");
 840         if (dc == NULL)
 841                 return (-ENXIO);
 842         pdrv->name = "drmn";
 843         return (_linux_pci_register_driver(pdrv, dc));
 844 }
 845
 846 void
 847 linux_pci_unregister_driver(struct pci_driver *pdrv)
 848 {
 849         devclass_t bus;
 850
 851         bus = devclass_find("pci");
 852
 853         spin_lock(&pci_lock);
 854         list_del(&pdrv->node);
 855         spin_unlock(&pci_lock);
 856         mtx_lock(&Giant);
 857         if (bus != NULL)
 858                 devclass_delete_driver(bus, &pdrv->bsddriver);
 859         mtx_unlock(&Giant);
 860 }
 861
 862 void
 863 linux_pci_unregister_drm_driver(struct pci_driver *pdrv)
 864 {
 865         devclass_t bus;
 866
 867         bus = devclass_find("vgapci");
 868
 869         spin_lock(&pci_lock);
 870         list_del(&pdrv->node);
 871         spin_unlock(&pci_lock);
 872         mtx_lock(&Giant);
 873         if (bus != NULL)
 874                 devclass_delete_driver(bus, &pdrv->bsddriver);
 875         mtx_unlock(&Giant);
 876 }
 877
 878 int
 879 pci_alloc_irq_vectors(struct pci_dev *pdev, int minv, int maxv,
 880     unsigned int flags)
 881 {
 882         int error;
 883
 884         if (flags & PCI_IRQ_MSIX) {
 885                 struct msix_entry *entries;
 886                 int i;
 887
 888                 entries = kcalloc(maxv, sizeof(*entries), GFP_KERNEL);
 889                 if (entries == NULL) {
 890                         error = -ENOMEM;
 891                         goto out;
 892                 }
 893                 for (i = 0; i < maxv; ++i)
 894                         entries[i].entry = i;
 895                 error = pci_enable_msix(pdev, entries, maxv);
 896 out:
 897                 kfree(entries);
 898                 if (error == 0 && pdev->msix_enabled)
 899                         return (pdev->dev.irq_end - pdev->dev.irq_start);
 900         }
 901         if (flags & PCI_IRQ_MSI) {
 902                 error = pci_enable_msi(pdev);
 903                 if (error == 0 && pdev->msi_enabled)
 904                         return (pdev->dev.irq_end - pdev->dev.irq_start);
 905         }
 906         if (flags & PCI_IRQ_LEGACY) {
 907                 if (pdev->irq)
 908                         return (1);
 909         }
 910
 911         return (-EINVAL);
 912 }
 913
 914 CTASSERT(sizeof(dma_addr_t) <= sizeof(uint64_t));
 915
 916 struct linux_dma_obj {
 917         void            *vaddr;
 918         uint64_t        dma_addr;
 919         bus_dmamap_t    dmamap;
 920         bus_dma_tag_t   dmat;
 921 };
 922
 923 static uma_zone_t linux_dma_trie_zone;
 924 static uma_zone_t linux_dma_obj_zone;
 925
 926 static void
 927 linux_dma_init(void *arg)
 928 {
 929
 930         linux_dma_trie_zone = uma_zcreate("linux_dma_pctrie",
 931             pctrie_node_size(), NULL, NULL, pctrie_zone_init, NULL,
 932             UMA_ALIGN_PTR, 0);
 933         linux_dma_obj_zone = uma_zcreate("linux_dma_object",
 934             sizeof(struct linux_dma_obj), NULL, NULL, NULL, NULL,
 935             UMA_ALIGN_PTR, 0);
 936
 937 }
 938 SYSINIT(linux_dma, SI_SUB_DRIVERS, SI_ORDER_THIRD, linux_dma_init, NULL);
 939
 940 static void
 941 linux_dma_uninit(void *arg)
 942 {
 943
 944         uma_zdestroy(linux_dma_obj_zone);
 945         uma_zdestroy(linux_dma_trie_zone);
 946 }
 947 SYSUNINIT(linux_dma, SI_SUB_DRIVERS, SI_ORDER_THIRD, linux_dma_uninit, NULL);
 948
 949 static void *
 950 linux_dma_trie_alloc(struct pctrie *ptree)
 951 {
 952
 953         return (uma_zalloc(linux_dma_trie_zone, M_NOWAIT));
 954 }
 955
 956 static void
 957 linux_dma_trie_free(struct pctrie *ptree, void *node)
 958 {
 959
 960         uma_zfree(linux_dma_trie_zone, node);
 961 }
 962
 963 PCTRIE_DEFINE(LINUX_DMA, linux_dma_obj, dma_addr, linux_dma_trie_alloc,
 964     linux_dma_trie_free);
 965
 966 #if defined(__i386__) || defined(__amd64__) || defined(__aarch64__)
 967 static dma_addr_t
 968 linux_dma_map_phys_common(struct device *dev, vm_paddr_t phys, size_t len,
 969     bus_dma_tag_t dmat)
 970 {
 971         struct linux_dma_priv *priv;
 972         struct linux_dma_obj *obj;
 973         int error, nseg;
 974         bus_dma_segment_t seg;
 975
 976         priv = dev->dma_priv;
 977
 978         /*
 979          * If the resultant mapping will be entirely 1:1 with the
 980          * physical address, short-circuit the remainder of the
 981          * bus_dma API.  This avoids tracking collisions in the pctrie
 982          * with the additional benefit of reducing overhead.
 983          */
 984         if (bus_dma_id_mapped(dmat, phys, len))
 985                 return (phys);
 986
 987         obj = uma_zalloc(linux_dma_obj_zone, M_NOWAIT);
 988         if (obj == NULL) {
 989                 return (0);
 990         }
 991         obj->dmat = dmat;
 992
 993         DMA_PRIV_LOCK(priv);
 994         if (bus_dmamap_create(obj->dmat, 0, &obj->dmamap) != 0) {
 995                 DMA_PRIV_UNLOCK(priv);
 996                 uma_zfree(linux_dma_obj_zone, obj);
 997                 return (0);
 998         }
 999
1000         nseg = -1;
1001         if (_bus_dmamap_load_phys(obj->dmat, obj->dmamap, phys, len,
1002             BUS_DMA_NOWAIT, &seg, &nseg) != 0) {
1003                 bus_dmamap_destroy(obj->dmat, obj->dmamap);
1004                 DMA_PRIV_UNLOCK(priv);
1005                 uma_zfree(linux_dma_obj_zone, obj);
1006                 return (0);
1007         }
1008
1009         KASSERT(++nseg == 1, ("More than one segment (nseg=%d)", nseg));
1010         obj->dma_addr = seg.ds_addr;
1011
1012         error = LINUX_DMA_PCTRIE_INSERT(&priv->ptree, obj);
1013         if (error != 0) {
1014                 bus_dmamap_unload(obj->dmat, obj->dmamap);
1015                 bus_dmamap_destroy(obj->dmat, obj->dmamap);
1016                 DMA_PRIV_UNLOCK(priv);
1017                 uma_zfree(linux_dma_obj_zone, obj);
1018                 return (0);
1019         }
1020         DMA_PRIV_UNLOCK(priv);
1021         return (obj->dma_addr);
1022 }
1023 #else
1024 static dma_addr_t
1025 linux_dma_map_phys_common(struct device *dev __unused, vm_paddr_t phys,
1026     size_t len __unused, bus_dma_tag_t dmat __unused)
1027 {
1028         return (phys);
1029 }
1030 #endif
1031
1032 dma_addr_t
1033 linux_dma_map_phys(struct device *dev, vm_paddr_t phys, size_t len)
1034 {
1035         struct linux_dma_priv *priv;
1036
1037         priv = dev->dma_priv;
1038         return (linux_dma_map_phys_common(dev, phys, len, priv->dmat));
1039 }
1040
1041 #if defined(__i386__) || defined(__amd64__) || defined(__aarch64__)
1042 void
1043 linux_dma_unmap(struct device *dev, dma_addr_t dma_addr, size_t len)
1044 {
1045         struct linux_dma_priv *priv;
1046         struct linux_dma_obj *obj;
1047
1048         priv = dev->dma_priv;
1049
1050         if (pctrie_is_empty(&priv->ptree))
1051                 return;
1052
1053         DMA_PRIV_LOCK(priv);
1054         obj = LINUX_DMA_PCTRIE_LOOKUP(&priv->ptree, dma_addr);
1055         if (obj == NULL) {
1056                 DMA_PRIV_UNLOCK(priv);
1057                 return;
1058         }
1059         LINUX_DMA_PCTRIE_REMOVE(&priv->ptree, dma_addr);
1060         bus_dmamap_unload(obj->dmat, obj->dmamap);
1061         bus_dmamap_destroy(obj->dmat, obj->dmamap);
1062         DMA_PRIV_UNLOCK(priv);
1063
1064         uma_zfree(linux_dma_obj_zone, obj);
1065 }
1066 #else
1067 void
1068 linux_dma_unmap(struct device *dev, dma_addr_t dma_addr, size_t len)
1069 {
1070 }
1071 #endif
1072
1073 void *
1074 linux_dma_alloc_coherent(struct device *dev, size_t size,
1075     dma_addr_t *dma_handle, gfp_t flag)
1076 {
1077         struct linux_dma_priv *priv;
1078         vm_paddr_t high;
1079         size_t align;
1080         void *mem;
1081
1082         if (dev == NULL || dev->dma_priv == NULL) {
1083                 *dma_handle = 0;
1084                 return (NULL);
1085         }
1086         priv = dev->dma_priv;
1087         if (priv->dma_coherent_mask)
1088                 high = priv->dma_coherent_mask;
1089         else
1090                 /* Coherent is lower 32bit only by default in Linux. */
1091                 high = BUS_SPACE_MAXADDR_32BIT;
1092         align = PAGE_SIZE << get_order(size);
1093         /* Always zero the allocation. */
1094         flag |= M_ZERO;
1095         mem = (void *)kmem_alloc_contig(size, flag & GFP_NATIVE_MASK, 0, high,
1096             align, 0, VM_MEMATTR_DEFAULT);
1097         if (mem != NULL) {
1098                 *dma_handle = linux_dma_map_phys_common(dev, vtophys(mem), size,
1099                     priv->dmat_coherent);
1100                 if (*dma_handle == 0) {
1101                         kmem_free((vm_offset_t)mem, size);
1102                         mem = NULL;
1103                 }
1104         } else {
1105                 *dma_handle = 0;
1106         }
1107         return (mem);
1108 }
1109
1110 void
1111 linuxkpi_dma_sync(struct device *dev, dma_addr_t dma_addr, size_t size,
1112     bus_dmasync_op_t op)
1113 {
1114         struct linux_dma_priv *priv;
1115         struct linux_dma_obj *obj;
1116
1117         priv = dev->dma_priv;
1118
1119         if (pctrie_is_empty(&priv->ptree))
1120                 return;
1121
1122         DMA_PRIV_LOCK(priv);
1123         obj = LINUX_DMA_PCTRIE_LOOKUP(&priv->ptree, dma_addr);
1124         if (obj == NULL) {
1125                 DMA_PRIV_UNLOCK(priv);
1126                 return;
1127         }
1128
1129         bus_dmamap_sync(obj->dmat, obj->dmamap, op);
1130         DMA_PRIV_UNLOCK(priv);
1131 }
1132
1133 int
1134 linux_dma_map_sg_attrs(struct device *dev, struct scatterlist *sgl, int nents,
1135     enum dma_data_direction direction, unsigned long attrs __unused)
1136 {
1137         struct linux_dma_priv *priv;
1138         struct scatterlist *sg;
1139         int i, nseg;
1140         bus_dma_segment_t seg;
1141
1142         priv = dev->dma_priv;
1143
1144         DMA_PRIV_LOCK(priv);
1145
1146         /* create common DMA map in the first S/G entry */
1147         if (bus_dmamap_create(priv->dmat, 0, &sgl->dma_map) != 0) {
1148                 DMA_PRIV_UNLOCK(priv);
1149                 return (0);
1150         }
1151
1152         /* load all S/G list entries */
1153         for_each_sg(sgl, sg, nents, i) {
1154                 nseg = -1;
1155                 if (_bus_dmamap_load_phys(priv->dmat, sgl->dma_map,
1156                     sg_phys(sg), sg->length, BUS_DMA_NOWAIT,
1157                     &seg, &nseg) != 0) {
1158                         bus_dmamap_unload(priv->dmat, sgl->dma_map);
1159                         bus_dmamap_destroy(priv->dmat, sgl->dma_map);
1160                         DMA_PRIV_UNLOCK(priv);
1161                         return (0);
1162                 }
1163                 KASSERT(nseg == 0,
1164                     ("More than one segment (nseg=%d)", nseg + 1));
1165
1166                 sg_dma_address(sg) = seg.ds_addr;
1167         }
1168
1169         switch (direction) {
1170         case DMA_BIDIRECTIONAL:
1171                 bus_dmamap_sync(priv->dmat, sgl->dma_map, BUS_DMASYNC_PREWRITE);
1172                 break;
1173         case DMA_TO_DEVICE:
1174                 bus_dmamap_sync(priv->dmat, sgl->dma_map, BUS_DMASYNC_PREREAD);
1175                 break;
1176         case DMA_FROM_DEVICE:
1177                 bus_dmamap_sync(priv->dmat, sgl->dma_map, BUS_DMASYNC_PREWRITE);
1178                 break;
1179         default:
1180                 break;
1181         }
1182
1183         DMA_PRIV_UNLOCK(priv);
1184
1185         return (nents);
1186 }
1187
1188 void
1189 linux_dma_unmap_sg_attrs(struct device *dev, struct scatterlist *sgl,
1190     int nents __unused, enum dma_data_direction direction,
1191     unsigned long attrs __unused)
1192 {
1193         struct linux_dma_priv *priv;
1194
1195         priv = dev->dma_priv;
1196
1197         DMA_PRIV_LOCK(priv);
1198
1199         switch (direction) {
1200         case DMA_BIDIRECTIONAL:
1201                 bus_dmamap_sync(priv->dmat, sgl->dma_map, BUS_DMASYNC_POSTREAD);
1202                 bus_dmamap_sync(priv->dmat, sgl->dma_map, BUS_DMASYNC_PREREAD);
1203                 break;
1204         case DMA_TO_DEVICE:
1205                 bus_dmamap_sync(priv->dmat, sgl->dma_map, BUS_DMASYNC_POSTWRITE);
1206                 break;
1207         case DMA_FROM_DEVICE:
1208                 bus_dmamap_sync(priv->dmat, sgl->dma_map, BUS_DMASYNC_POSTREAD);
1209                 break;
1210         default:
1211                 break;
1212         }
1213
1214         bus_dmamap_unload(priv->dmat, sgl->dma_map);
1215         bus_dmamap_destroy(priv->dmat, sgl->dma_map);
1216         DMA_PRIV_UNLOCK(priv);
1217 }
1218
1219 struct dma_pool {
1220         struct device  *pool_device;
1221         uma_zone_t      pool_zone;
1222         struct mtx      pool_lock;
1223         bus_dma_tag_t   pool_dmat;
1224         size_t          pool_entry_size;
1225         struct pctrie   pool_ptree;
1226 };
1227
1228 #define DMA_POOL_LOCK(pool) mtx_lock(&(pool)->pool_lock)
1229 #define DMA_POOL_UNLOCK(pool) mtx_unlock(&(pool)->pool_lock)
1230
1231 static inline int
1232 dma_pool_obj_ctor(void *mem, int size, void *arg, int flags)
1233 {
1234         struct linux_dma_obj *obj = mem;
1235         struct dma_pool *pool = arg;
1236         int error, nseg;
1237         bus_dma_segment_t seg;
1238
1239         nseg = -1;
1240         DMA_POOL_LOCK(pool);
1241         error = _bus_dmamap_load_phys(pool->pool_dmat, obj->dmamap,
1242             vtophys(obj->vaddr), pool->pool_entry_size, BUS_DMA_NOWAIT,
1243             &seg, &nseg);
1244         DMA_POOL_UNLOCK(pool);
1245         if (error != 0) {
1246                 return (error);
1247         }
1248         KASSERT(++nseg == 1, ("More than one segment (nseg=%d)", nseg));
1249         obj->dma_addr = seg.ds_addr;
1250
1251         return (0);
1252 }
1253
1254 static void
1255 dma_pool_obj_dtor(void *mem, int size, void *arg)
1256 {
1257         struct linux_dma_obj *obj = mem;
1258         struct dma_pool *pool = arg;
1259
1260         DMA_POOL_LOCK(pool);
1261         bus_dmamap_unload(pool->pool_dmat, obj->dmamap);
1262         DMA_POOL_UNLOCK(pool);
1263 }
1264
1265 static int
1266 dma_pool_obj_import(void *arg, void **store, int count, int domain __unused,
1267     int flags)
1268 {
1269         struct dma_pool *pool = arg;
1270         struct linux_dma_obj *obj;
1271         int error, i;
1272
1273         for (i = 0; i < count; i++) {
1274                 obj = uma_zalloc(linux_dma_obj_zone, flags);
1275                 if (obj == NULL)
1276                         break;
1277
1278                 error = bus_dmamem_alloc(pool->pool_dmat, &obj->vaddr,
1279                     BUS_DMA_NOWAIT, &obj->dmamap);
1280                 if (error!= 0) {
1281                         uma_zfree(linux_dma_obj_zone, obj);
1282                         break;
1283                 }
1284
1285                 store[i] = obj;
1286         }
1287
1288         return (i);
1289 }
1290
1291 static void
1292 dma_pool_obj_release(void *arg, void **store, int count)
1293 {
1294         struct dma_pool *pool = arg;
1295         struct linux_dma_obj *obj;
1296         int i;
1297
1298         for (i = 0; i < count; i++) {
1299                 obj = store[i];
1300                 bus_dmamem_free(pool->pool_dmat, obj->vaddr, obj->dmamap);
1301                 uma_zfree(linux_dma_obj_zone, obj);
1302         }
1303 }
1304
1305 struct dma_pool *
1306 linux_dma_pool_create(char *name, struct device *dev, size_t size,
1307     size_t align, size_t boundary)
1308 {
1309         struct linux_dma_priv *priv;
1310         struct dma_pool *pool;
1311
1312         priv = dev->dma_priv;
1313
1314         pool = kzalloc(sizeof(*pool), GFP_KERNEL);
1315         pool->pool_device = dev;
1316         pool->pool_entry_size = size;
1317
1318         if (bus_dma_tag_create(bus_get_dma_tag(dev->bsddev),
1319             align, boundary,            /* alignment, boundary */
1320             priv->dma_mask,             /* lowaddr */
1321             BUS_SPACE_MAXADDR,          /* highaddr */
1322             NULL, NULL,                 /* filtfunc, filtfuncarg */
1323             size,                       /* maxsize */
1324             1,                          /* nsegments */
1325             size,                       /* maxsegsz */
1326             0,                          /* flags */
1327             NULL, NULL,                 /* lockfunc, lockfuncarg */
1328             &pool->pool_dmat)) {
1329                 kfree(pool);
1330                 return (NULL);
1331         }
1332
1333         pool->pool_zone = uma_zcache_create(name, -1, dma_pool_obj_ctor,
1334             dma_pool_obj_dtor, NULL, NULL, dma_pool_obj_import,
1335             dma_pool_obj_release, pool, 0);
1336
1337         mtx_init(&pool->pool_lock, "lkpi-dma-pool", NULL, MTX_DEF);
1338         pctrie_init(&pool->pool_ptree);
1339
1340         return (pool);
1341 }
1342
1343 void
1344 linux_dma_pool_destroy(struct dma_pool *pool)
1345 {
1346
1347         uma_zdestroy(pool->pool_zone);
1348         bus_dma_tag_destroy(pool->pool_dmat);
1349         mtx_destroy(&pool->pool_lock);
1350         kfree(pool);
1351 }
1352
1353 void
1354 lkpi_dmam_pool_destroy(struct device *dev, void *p)
1355 {
1356         struct dma_pool *pool;
1357
1358         pool = *(struct dma_pool **)p;
1359         LINUX_DMA_PCTRIE_RECLAIM(&pool->pool_ptree);
1360         linux_dma_pool_destroy(pool);
1361 }
1362
1363 void *
1364 linux_dma_pool_alloc(struct dma_pool *pool, gfp_t mem_flags,
1365     dma_addr_t *handle)
1366 {
1367         struct linux_dma_obj *obj;
1368
1369         obj = uma_zalloc_arg(pool->pool_zone, pool, mem_flags & GFP_NATIVE_MASK);
1370         if (obj == NULL)
1371                 return (NULL);
1372
1373         DMA_POOL_LOCK(pool);
1374         if (LINUX_DMA_PCTRIE_INSERT(&pool->pool_ptree, obj) != 0) {
1375                 DMA_POOL_UNLOCK(pool);
1376                 uma_zfree_arg(pool->pool_zone, obj, pool);
1377                 return (NULL);
1378         }
1379         DMA_POOL_UNLOCK(pool);
1380
1381         *handle = obj->dma_addr;
1382         return (obj->vaddr);
1383 }
1384
1385 void
1386 linux_dma_pool_free(struct dma_pool *pool, void *vaddr, dma_addr_t dma_addr)
1387 {
1388         struct linux_dma_obj *obj;
1389
1390         DMA_POOL_LOCK(pool);
1391         obj = LINUX_DMA_PCTRIE_LOOKUP(&pool->pool_ptree, dma_addr);
1392         if (obj == NULL) {
1393                 DMA_POOL_UNLOCK(pool);
1394                 return;
1395         }
1396         LINUX_DMA_PCTRIE_REMOVE(&pool->pool_ptree, dma_addr);
1397         DMA_POOL_UNLOCK(pool);
1398
1399         uma_zfree_arg(pool->pool_zone, obj, pool);
1400 }
1401
1402 static int
1403 linux_backlight_get_status(device_t dev, struct backlight_props *props)
1404 {
1405         struct pci_dev *pdev;
1406
1407         linux_set_current(curthread);
1408         pdev = device_get_softc(dev);
1409
1410         props->brightness = pdev->dev.bd->props.brightness;
1411         props->brightness = props->brightness * 100 / pdev->dev.bd->props.max_brightness;
1412         props->nlevels = 0;
1413
1414         return (0);
1415 }
1416
1417 static int
1418 linux_backlight_get_info(device_t dev, struct backlight_info *info)
1419 {
1420         struct pci_dev *pdev;
1421
1422         linux_set_current(curthread);
1423         pdev = device_get_softc(dev);
1424
1425         info->type = BACKLIGHT_TYPE_PANEL;
1426         strlcpy(info->name, pdev->dev.bd->name, BACKLIGHTMAXNAMELENGTH);
1427         return (0);
1428 }
1429
1430 static int
1431 linux_backlight_update_status(device_t dev, struct backlight_props *props)
1432 {
1433         struct pci_dev *pdev;
1434
1435         linux_set_current(curthread);
1436         pdev = device_get_softc(dev);
1437
1438         pdev->dev.bd->props.brightness = pdev->dev.bd->props.max_brightness *
1439                 props->brightness / 100;
1440         pdev->dev.bd->props.power = props->brightness == 0 ?
1441                 4/* FB_BLANK_POWERDOWN */ : 0/* FB_BLANK_UNBLANK */;
1442         return (pdev->dev.bd->ops->update_status(pdev->dev.bd));
1443 }
1444
1445 struct backlight_device *
1446 linux_backlight_device_register(const char *name, struct device *dev,
1447     void *data, const struct backlight_ops *ops, struct backlight_properties *props)
1448 {
1449
1450         dev->bd = malloc(sizeof(*dev->bd), M_DEVBUF, M_WAITOK | M_ZERO);
1451         dev->bd->ops = ops;
1452         dev->bd->props.type = props->type;
1453         dev->bd->props.max_brightness = props->max_brightness;
1454         dev->bd->props.brightness = props->brightness;
1455         dev->bd->props.power = props->power;
1456         dev->bd->data = data;
1457         dev->bd->dev = dev;
1458         dev->bd->name = strdup(name, M_DEVBUF);
1459
1460         dev->backlight_dev = backlight_register(name, dev->bsddev);
1461
1462         return (dev->bd);
1463 }
1464
1465 void
1466 linux_backlight_device_unregister(struct backlight_device *bd)
1467 {
1468
1469         backlight_destroy(bd->dev->backlight_dev);
1470         free(bd->name, M_DEVBUF);
1471         free(bd, M_DEVBUF);
1472 }