drivers/pci/pci-uclass.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Copyright (c) 2014 Google, Inc
   4  * Written by Simon Glass <sjg@chromium.org>
   5  */
   6
   7 #include <common.h>
   8 #include <dm.h>
   9 #include <errno.h>
  10 #include <pci.h>
  11 #include <asm/io.h>
  12 #include <dm/device-internal.h>
  13 #include <dm/lists.h>
  14 #if defined(CONFIG_X86) && defined(CONFIG_HAVE_FSP)
  15 #include <asm/fsp/fsp_support.h>
  16 #endif
  17 #include "pci_internal.h"
  18
  19 DECLARE_GLOBAL_DATA_PTR;
  20
  21 int pci_get_bus(int busnum, struct udevice **busp)
  22 {
  23         int ret;
  24
  25         ret = uclass_get_device_by_seq(UCLASS_PCI, busnum, busp);
  26
  27         /* Since buses may not be numbered yet try a little harder with bus 0 */
  28         if (ret == -ENODEV) {
  29                 ret = uclass_first_device_err(UCLASS_PCI, busp);
  30                 if (ret)
  31                         return ret;
  32                 ret = uclass_get_device_by_seq(UCLASS_PCI, busnum, busp);
  33         }
  34
  35         return ret;
  36 }
  37
  38 struct udevice *pci_get_controller(struct udevice *dev)
  39 {
  40         while (device_is_on_pci_bus(dev))
  41                 dev = dev->parent;
  42
  43         return dev;
  44 }
  45
  46 pci_dev_t dm_pci_get_bdf(struct udevice *dev)
  47 {
  48         struct pci_child_platdata *pplat = dev_get_parent_platdata(dev);
  49         struct udevice *bus = dev->parent;
  50
  51         return PCI_ADD_BUS(bus->seq, pplat->devfn);
  52 }
  53
  54 /**
  55  * pci_get_bus_max() - returns the bus number of the last active bus
  56  *
  57  * @return last bus number, or -1 if no active buses
  58  */
  59 static int pci_get_bus_max(void)
  60 {
  61         struct udevice *bus;
  62         struct uclass *uc;
  63         int ret = -1;
  64
  65         ret = uclass_get(UCLASS_PCI, &uc);
  66         uclass_foreach_dev(bus, uc) {
  67                 if (bus->seq > ret)
  68                         ret = bus->seq;
  69         }
  70
  71         debug("%s: ret=%d\n", __func__, ret);
  72
  73         return ret;
  74 }
  75
  76 int pci_last_busno(void)
  77 {
  78         return pci_get_bus_max();
  79 }
  80
  81 int pci_get_ff(enum pci_size_t size)
  82 {
  83         switch (size) {
  84         case PCI_SIZE_8:
  85                 return 0xff;
  86         case PCI_SIZE_16:
  87                 return 0xffff;
  88         default:
  89                 return 0xffffffff;
  90         }
  91 }
  92
  93 static void pci_dev_find_ofnode(struct udevice *bus, phys_addr_t bdf,
  94                                 ofnode *rnode)
  95 {
  96         struct fdt_pci_addr addr;
  97         ofnode node;
  98         int ret;
  99
 100         dev_for_each_subnode(node, bus) {
 101                 ret = ofnode_read_pci_addr(node, FDT_PCI_SPACE_CONFIG, "reg",
 102                                            &addr);
 103                 if (ret)
 104                         continue;
 105
 106                 if (PCI_MASK_BUS(addr.phys_hi) != PCI_MASK_BUS(bdf))
 107                         continue;
 108
 109                 *rnode = node;
 110                 break;
 111         }
 112 };
 113
 114 int pci_bus_find_devfn(struct udevice *bus, pci_dev_t find_devfn,
 115                        struct udevice **devp)
 116 {
 117         struct udevice *dev;
 118
 119         for (device_find_first_child(bus, &dev);
 120              dev;
 121              device_find_next_child(&dev)) {
 122                 struct pci_child_platdata *pplat;
 123
 124                 pplat = dev_get_parent_platdata(dev);
 125                 if (pplat && pplat->devfn == find_devfn) {
 126                         *devp = dev;
 127                         return 0;
 128                 }
 129         }
 130
 131         return -ENODEV;
 132 }
 133
 134 int dm_pci_bus_find_bdf(pci_dev_t bdf, struct udevice **devp)
 135 {
 136         struct udevice *bus;
 137         int ret;
 138
 139         ret = pci_get_bus(PCI_BUS(bdf), &bus);
 140         if (ret)
 141                 return ret;
 142         return pci_bus_find_devfn(bus, PCI_MASK_BUS(bdf), devp);
 143 }
 144
 145 static int pci_device_matches_ids(struct udevice *dev,
 146                                   struct pci_device_id *ids)
 147 {
 148         struct pci_child_platdata *pplat;
 149         int i;
 150
 151         pplat = dev_get_parent_platdata(dev);
 152         if (!pplat)
 153                 return -EINVAL;
 154         for (i = 0; ids[i].vendor != 0; i++) {
 155                 if (pplat->vendor == ids[i].vendor &&
 156                     pplat->device == ids[i].device)
 157                         return i;
 158         }
 159
 160         return -EINVAL;
 161 }
 162
 163 int pci_bus_find_devices(struct udevice *bus, struct pci_device_id *ids,
 164                          int *indexp, struct udevice **devp)
 165 {
 166         struct udevice *dev;
 167
 168         /* Scan all devices on this bus */
 169         for (device_find_first_child(bus, &dev);
 170              dev;
 171              device_find_next_child(&dev)) {
 172                 if (pci_device_matches_ids(dev, ids) >= 0) {
 173                         if ((*indexp)-- <= 0) {
 174                                 *devp = dev;
 175                                 return 0;
 176                         }
 177                 }
 178         }
 179
 180         return -ENODEV;
 181 }
 182
 183 int pci_find_device_id(struct pci_device_id *ids, int index,
 184                        struct udevice **devp)
 185 {
 186         struct udevice *bus;
 187
 188         /* Scan all known buses */
 189         for (uclass_first_device(UCLASS_PCI, &bus);
 190              bus;
 191              uclass_next_device(&bus)) {
 192                 if (!pci_bus_find_devices(bus, ids, &index, devp))
 193                         return 0;
 194         }
 195         *devp = NULL;
 196
 197         return -ENODEV;
 198 }
 199
 200 static int dm_pci_bus_find_device(struct udevice *bus, unsigned int vendor,
 201                                   unsigned int device, int *indexp,
 202                                   struct udevice **devp)
 203 {
 204         struct pci_child_platdata *pplat;
 205         struct udevice *dev;
 206
 207         for (device_find_first_child(bus, &dev);
 208              dev;
 209              device_find_next_child(&dev)) {
 210                 pplat = dev_get_parent_platdata(dev);
 211                 if (pplat->vendor == vendor && pplat->device == device) {
 212                         if (!(*indexp)--) {
 213                                 *devp = dev;
 214                                 return 0;
 215                         }
 216                 }
 217         }
 218
 219         return -ENODEV;
 220 }
 221
 222 int dm_pci_find_device(unsigned int vendor, unsigned int device, int index,
 223                        struct udevice **devp)
 224 {
 225         struct udevice *bus;
 226
 227         /* Scan all known buses */
 228         for (uclass_first_device(UCLASS_PCI, &bus);
 229              bus;
 230              uclass_next_device(&bus)) {
 231                 if (!dm_pci_bus_find_device(bus, vendor, device, &index, devp))
 232                         return device_probe(*devp);
 233         }
 234         *devp = NULL;
 235
 236         return -ENODEV;
 237 }
 238
 239 int dm_pci_find_class(uint find_class, int index, struct udevice **devp)
 240 {
 241         struct udevice *dev;
 242
 243         /* Scan all known buses */
 244         for (pci_find_first_device(&dev);
 245              dev;
 246              pci_find_next_device(&dev)) {
 247                 struct pci_child_platdata *pplat = dev_get_parent_platdata(dev);
 248
 249                 if (pplat->class == find_class && !index--) {
 250                         *devp = dev;
 251                         return device_probe(*devp);
 252                 }
 253         }
 254         *devp = NULL;
 255
 256         return -ENODEV;
 257 }
 258
 259 int pci_bus_write_config(struct udevice *bus, pci_dev_t bdf, int offset,
 260                          unsigned long value, enum pci_size_t size)
 261 {
 262         struct dm_pci_ops *ops;
 263
 264         ops = pci_get_ops(bus);
 265         if (!ops->write_config)
 266                 return -ENOSYS;
 267         return ops->write_config(bus, bdf, offset, value, size);
 268 }
 269
 270 int pci_bus_clrset_config32(struct udevice *bus, pci_dev_t bdf, int offset,
 271                             u32 clr, u32 set)
 272 {
 273         ulong val;
 274         int ret;
 275
 276         ret = pci_bus_read_config(bus, bdf, offset, &val, PCI_SIZE_32);
 277         if (ret)
 278                 return ret;
 279         val &= ~clr;
 280         val |= set;
 281
 282         return pci_bus_write_config(bus, bdf, offset, val, PCI_SIZE_32);
 283 }
 284
 285 int pci_write_config(pci_dev_t bdf, int offset, unsigned long value,
 286                      enum pci_size_t size)
 287 {
 288         struct udevice *bus;
 289         int ret;
 290
 291         ret = pci_get_bus(PCI_BUS(bdf), &bus);
 292         if (ret)
 293                 return ret;
 294
 295         return pci_bus_write_config(bus, bdf, offset, value, size);
 296 }
 297
 298 int dm_pci_write_config(struct udevice *dev, int offset, unsigned long value,
 299                         enum pci_size_t size)
 300 {
 301         struct udevice *bus;
 302
 303         for (bus = dev; device_is_on_pci_bus(bus);)
 304                 bus = bus->parent;
 305         return pci_bus_write_config(bus, dm_pci_get_bdf(dev), offset, value,
 306                                     size);
 307 }
 308
 309 int pci_write_config32(pci_dev_t bdf, int offset, u32 value)
 310 {
 311         return pci_write_config(bdf, offset, value, PCI_SIZE_32);
 312 }
 313
 314 int pci_write_config16(pci_dev_t bdf, int offset, u16 value)
 315 {
 316         return pci_write_config(bdf, offset, value, PCI_SIZE_16);
 317 }
 318
 319 int pci_write_config8(pci_dev_t bdf, int offset, u8 value)
 320 {
 321         return pci_write_config(bdf, offset, value, PCI_SIZE_8);
 322 }
 323
 324 int dm_pci_write_config8(struct udevice *dev, int offset, u8 value)
 325 {
 326         return dm_pci_write_config(dev, offset, value, PCI_SIZE_8);
 327 }
 328
 329 int dm_pci_write_config16(struct udevice *dev, int offset, u16 value)
 330 {
 331         return dm_pci_write_config(dev, offset, value, PCI_SIZE_16);
 332 }
 333
 334 int dm_pci_write_config32(struct udevice *dev, int offset, u32 value)
 335 {
 336         return dm_pci_write_config(dev, offset, value, PCI_SIZE_32);
 337 }
 338
 339 int pci_bus_read_config(struct udevice *bus, pci_dev_t bdf, int offset,
 340                         unsigned long *valuep, enum pci_size_t size)
 341 {
 342         struct dm_pci_ops *ops;
 343
 344         ops = pci_get_ops(bus);
 345         if (!ops->read_config)
 346                 return -ENOSYS;
 347         return ops->read_config(bus, bdf, offset, valuep, size);
 348 }
 349
 350 int pci_read_config(pci_dev_t bdf, int offset, unsigned long *valuep,
 351                     enum pci_size_t size)
 352 {
 353         struct udevice *bus;
 354         int ret;
 355
 356         ret = pci_get_bus(PCI_BUS(bdf), &bus);
 357         if (ret)
 358                 return ret;
 359
 360         return pci_bus_read_config(bus, bdf, offset, valuep, size);
 361 }
 362
 363 int dm_pci_read_config(struct udevice *dev, int offset, unsigned long *valuep,
 364                        enum pci_size_t size)
 365 {
 366         struct udevice *bus;
 367
 368         for (bus = dev; device_is_on_pci_bus(bus);)
 369                 bus = bus->parent;
 370         return pci_bus_read_config(bus, dm_pci_get_bdf(dev), offset, valuep,
 371                                    size);
 372 }
 373
 374 int pci_read_config32(pci_dev_t bdf, int offset, u32 *valuep)
 375 {
 376         unsigned long value;
 377         int ret;
 378
 379         ret = pci_read_config(bdf, offset, &value, PCI_SIZE_32);
 380         if (ret)
 381                 return ret;
 382         *valuep = value;
 383
 384         return 0;
 385 }
 386
 387 int pci_read_config16(pci_dev_t bdf, int offset, u16 *valuep)
 388 {
 389         unsigned long value;
 390         int ret;
 391
 392         ret = pci_read_config(bdf, offset, &value, PCI_SIZE_16);
 393         if (ret)
 394                 return ret;
 395         *valuep = value;
 396
 397         return 0;
 398 }
 399
 400 int pci_read_config8(pci_dev_t bdf, int offset, u8 *valuep)
 401 {
 402         unsigned long value;
 403         int ret;
 404
 405         ret = pci_read_config(bdf, offset, &value, PCI_SIZE_8);
 406         if (ret)
 407                 return ret;
 408         *valuep = value;
 409
 410         return 0;
 411 }
 412
 413 int dm_pci_read_config8(struct udevice *dev, int offset, u8 *valuep)
 414 {
 415         unsigned long value;
 416         int ret;
 417
 418         ret = dm_pci_read_config(dev, offset, &value, PCI_SIZE_8);
 419         if (ret)
 420                 return ret;
 421         *valuep = value;
 422
 423         return 0;
 424 }
 425
 426 int dm_pci_read_config16(struct udevice *dev, int offset, u16 *valuep)
 427 {
 428         unsigned long value;
 429         int ret;
 430
 431         ret = dm_pci_read_config(dev, offset, &value, PCI_SIZE_16);
 432         if (ret)
 433                 return ret;
 434         *valuep = value;
 435
 436         return 0;
 437 }
 438
 439 int dm_pci_read_config32(struct udevice *dev, int offset, u32 *valuep)
 440 {
 441         unsigned long value;
 442         int ret;
 443
 444         ret = dm_pci_read_config(dev, offset, &value, PCI_SIZE_32);
 445         if (ret)
 446                 return ret;
 447         *valuep = value;
 448
 449         return 0;
 450 }
 451
 452 int dm_pci_clrset_config8(struct udevice *dev, int offset, u32 clr, u32 set)
 453 {
 454         u8 val;
 455         int ret;
 456
 457         ret = dm_pci_read_config8(dev, offset, &val);
 458         if (ret)
 459                 return ret;
 460         val &= ~clr;
 461         val |= set;
 462
 463         return dm_pci_write_config8(dev, offset, val);
 464 }
 465
 466 int dm_pci_clrset_config16(struct udevice *dev, int offset, u32 clr, u32 set)
 467 {
 468         u16 val;
 469         int ret;
 470
 471         ret = dm_pci_read_config16(dev, offset, &val);
 472         if (ret)
 473                 return ret;
 474         val &= ~clr;
 475         val |= set;
 476
 477         return dm_pci_write_config16(dev, offset, val);
 478 }
 479
 480 int dm_pci_clrset_config32(struct udevice *dev, int offset, u32 clr, u32 set)
 481 {
 482         u32 val;
 483         int ret;
 484
 485         ret = dm_pci_read_config32(dev, offset, &val);
 486         if (ret)
 487                 return ret;
 488         val &= ~clr;
 489         val |= set;
 490
 491         return dm_pci_write_config32(dev, offset, val);
 492 }
 493
 494 static void set_vga_bridge_bits(struct udevice *dev)
 495 {
 496         struct udevice *parent = dev->parent;
 497         u16 bc;
 498
 499         while (parent->seq != 0) {
 500                 dm_pci_read_config16(parent, PCI_BRIDGE_CONTROL, &bc);
 501                 bc |= PCI_BRIDGE_CTL_VGA;
 502                 dm_pci_write_config16(parent, PCI_BRIDGE_CONTROL, bc);
 503                 parent = parent->parent;
 504         }
 505 }
 506
 507 int pci_auto_config_devices(struct udevice *bus)
 508 {
 509         struct pci_controller *hose = bus->uclass_priv;
 510         struct pci_child_platdata *pplat;
 511         unsigned int sub_bus;
 512         struct udevice *dev;
 513         int ret;
 514
 515         sub_bus = bus->seq;
 516         debug("%s: start\n", __func__);
 517         pciauto_config_init(hose);
 518         for (ret = device_find_first_child(bus, &dev);
 519              !ret && dev;
 520              ret = device_find_next_child(&dev)) {
 521                 unsigned int max_bus;
 522                 int ret;
 523
 524                 debug("%s: device %s\n", __func__, dev->name);
 525                 ret = dm_pciauto_config_device(dev);
 526                 if (ret < 0)
 527                         return ret;
 528                 max_bus = ret;
 529                 sub_bus = max(sub_bus, max_bus);
 530
 531                 pplat = dev_get_parent_platdata(dev);
 532                 if (pplat->class == (PCI_CLASS_DISPLAY_VGA << 8))
 533                         set_vga_bridge_bits(dev);
 534         }
 535         debug("%s: done\n", __func__);
 536
 537         return sub_bus;
 538 }
 539
 540 int pci_generic_mmap_write_config(
 541         struct udevice *bus,
 542         int (*addr_f)(struct udevice *bus, pci_dev_t bdf, uint offset, void **addrp),
 543         pci_dev_t bdf,
 544         uint offset,
 545         ulong value,
 546         enum pci_size_t size)
 547 {
 548         void *address;
 549
 550         if (addr_f(bus, bdf, offset, &address) < 0)
 551                 return 0;
 552
 553         switch (size) {
 554         case PCI_SIZE_8:
 555                 writeb(value, address);
 556                 return 0;
 557         case PCI_SIZE_16:
 558                 writew(value, address);
 559                 return 0;
 560         case PCI_SIZE_32:
 561                 writel(value, address);
 562                 return 0;
 563         default:
 564                 return -EINVAL;
 565         }
 566 }
 567
 568 int pci_generic_mmap_read_config(
 569         struct udevice *bus,
 570         int (*addr_f)(struct udevice *bus, pci_dev_t bdf, uint offset, void **addrp),
 571         pci_dev_t bdf,
 572         uint offset,
 573         ulong *valuep,
 574         enum pci_size_t size)
 575 {
 576         void *address;
 577
 578         if (addr_f(bus, bdf, offset, &address) < 0) {
 579                 *valuep = pci_get_ff(size);
 580                 return 0;
 581         }
 582
 583         switch (size) {
 584         case PCI_SIZE_8:
 585                 *valuep = readb(address);
 586                 return 0;
 587         case PCI_SIZE_16:
 588                 *valuep = readw(address);
 589                 return 0;
 590         case PCI_SIZE_32:
 591                 *valuep = readl(address);
 592                 return 0;
 593         default:
 594                 return -EINVAL;
 595         }
 596 }
 597
 598 int dm_pci_hose_probe_bus(struct udevice *bus)
 599 {
 600         int sub_bus;
 601         int ret;
 602
 603         debug("%s\n", __func__);
 604
 605         sub_bus = pci_get_bus_max() + 1;
 606         debug("%s: bus = %d/%s\n", __func__, sub_bus, bus->name);
 607         dm_pciauto_prescan_setup_bridge(bus, sub_bus);
 608
 609         ret = device_probe(bus);
 610         if (ret) {
 611                 debug("%s: Cannot probe bus %s: %d\n", __func__, bus->name,
 612                       ret);
 613                 return ret;
 614         }
 615         if (sub_bus != bus->seq) {
 616                 printf("%s: Internal error, bus '%s' got seq %d, expected %d\n",
 617                        __func__, bus->name, bus->seq, sub_bus);
 618                 return -EPIPE;
 619         }
 620         sub_bus = pci_get_bus_max();
 621         dm_pciauto_postscan_setup_bridge(bus, sub_bus);
 622
 623         return sub_bus;
 624 }
 625
 626 /**
 627  * pci_match_one_device - Tell if a PCI device structure has a matching
 628  *                        PCI device id structure
 629  * @id: single PCI device id structure to match
 630  * @find: the PCI device id structure to match against
 631  *
 632  * Returns true if the finding pci_device_id structure matched or false if
 633  * there is no match.
 634  */
 635 static bool pci_match_one_id(const struct pci_device_id *id,
 636                              const struct pci_device_id *find)
 637 {
 638         if ((id->vendor == PCI_ANY_ID || id->vendor == find->vendor) &&
 639             (id->device == PCI_ANY_ID || id->device == find->device) &&
 640             (id->subvendor == PCI_ANY_ID || id->subvendor == find->subvendor) &&
 641             (id->subdevice == PCI_ANY_ID || id->subdevice == find->subdevice) &&
 642             !((id->class ^ find->class) & id->class_mask))
 643                 return true;
 644
 645         return false;
 646 }
 647
 648 /**
 649  * pci_find_and_bind_driver() - Find and bind the right PCI driver
 650  *
 651  * This only looks at certain fields in the descriptor.
 652  *
 653  * @parent:     Parent bus
 654  * @find_id:    Specification of the driver to find
 655  * @bdf:        Bus/device/function addreess - see PCI_BDF()
 656  * @devp:       Returns a pointer to the device created
 657  * @return 0 if OK, -EPERM if the device is not needed before relocation and
 658  *         therefore was not created, other -ve value on error
 659  */
 660 static int pci_find_and_bind_driver(struct udevice *parent,
 661                                     struct pci_device_id *find_id,
 662                                     pci_dev_t bdf, struct udevice **devp)
 663 {
 664         struct pci_driver_entry *start, *entry;
 665         ofnode node = ofnode_null();
 666         const char *drv;
 667         int n_ents;
 668         int ret;
 669         char name[30], *str;
 670         bool bridge;
 671
 672         *devp = NULL;
 673
 674         debug("%s: Searching for driver: vendor=%x, device=%x\n", __func__,
 675               find_id->vendor, find_id->device);
 676
 677         /* Determine optional OF node */
 678         pci_dev_find_ofnode(parent, bdf, &node);
 679
 680         start = ll_entry_start(struct pci_driver_entry, pci_driver_entry);
 681         n_ents = ll_entry_count(struct pci_driver_entry, pci_driver_entry);
 682         for (entry = start; entry != start + n_ents; entry++) {
 683                 const struct pci_device_id *id;
 684                 struct udevice *dev;
 685                 const struct driver *drv;
 686
 687                 for (id = entry->match;
 688                      id->vendor || id->subvendor || id->class_mask;
 689                      id++) {
 690                         if (!pci_match_one_id(id, find_id))
 691                                 continue;
 692
 693                         drv = entry->driver;
 694
 695                         /*
 696                          * In the pre-relocation phase, we only bind devices
 697                          * whose driver has the DM_FLAG_PRE_RELOC set, to save
 698                          * precious memory space as on some platforms as that
 699                          * space is pretty limited (ie: using Cache As RAM).
 700                          */
 701                         if (!(gd->flags & GD_FLG_RELOC) &&
 702                             !(drv->flags & DM_FLAG_PRE_RELOC))
 703                                 return -EPERM;
 704
 705                         /*
 706                          * We could pass the descriptor to the driver as
 707                          * platdata (instead of NULL) and allow its bind()
 708                          * method to return -ENOENT if it doesn't support this
 709                          * device. That way we could continue the search to
 710                          * find another driver. For now this doesn't seem
 711                          * necesssary, so just bind the first match.
 712                          */
 713                         ret = device_bind_ofnode(parent, drv, drv->name, NULL,
 714                                                  node, &dev);
 715                         if (ret)
 716                                 goto error;
 717                         debug("%s: Match found: %s\n", __func__, drv->name);
 718                         dev->driver_data = id->driver_data;
 719                         *devp = dev;
 720                         return 0;
 721                 }
 722         }
 723
 724         bridge = (find_id->class >> 8) == PCI_CLASS_BRIDGE_PCI;
 725         /*
 726          * In the pre-relocation phase, we only bind bridge devices to save
 727          * precious memory space as on some platforms as that space is pretty
 728          * limited (ie: using Cache As RAM).
 729          */
 730         if (!(gd->flags & GD_FLG_RELOC) && !bridge)
 731                 return -EPERM;
 732
 733         /* Bind a generic driver so that the device can be used */
 734         sprintf(name, "pci_%x:%x.%x", parent->seq, PCI_DEV(bdf),
 735                 PCI_FUNC(bdf));
 736         str = strdup(name);
 737         if (!str)
 738                 return -ENOMEM;
 739         drv = bridge ? "pci_bridge_drv" : "pci_generic_drv";
 740
 741         ret = device_bind_driver_to_node(parent, drv, str, node, devp);
 742         if (ret) {
 743                 debug("%s: Failed to bind generic driver: %d\n", __func__, ret);
 744                 free(str);
 745                 return ret;
 746         }
 747         debug("%s: No match found: bound generic driver instead\n", __func__);
 748
 749         return 0;
 750
 751 error:
 752         debug("%s: No match found: error %d\n", __func__, ret);
 753         return ret;
 754 }
 755
 756 int pci_bind_bus_devices(struct udevice *bus)
 757 {
 758         ulong vendor, device;
 759         ulong header_type;
 760         pci_dev_t bdf, end;
 761         bool found_multi;
 762         int ret;
 763
 764         found_multi = false;
 765         end = PCI_BDF(bus->seq, PCI_MAX_PCI_DEVICES - 1,
 766                       PCI_MAX_PCI_FUNCTIONS - 1);
 767         for (bdf = PCI_BDF(bus->seq, 0, 0); bdf <= end;
 768              bdf += PCI_BDF(0, 0, 1)) {
 769                 struct pci_child_platdata *pplat;
 770                 struct udevice *dev;
 771                 ulong class;
 772
 773                 if (!PCI_FUNC(bdf))
 774                         found_multi = false;
 775                 if (PCI_FUNC(bdf) && !found_multi)
 776                         continue;
 777
 778                 /* Check only the first access, we don't expect problems */
 779                 ret = pci_bus_read_config(bus, bdf, PCI_VENDOR_ID, &vendor,
 780                                           PCI_SIZE_16);
 781                 if (ret)
 782                         goto error;
 783
 784                 if (vendor == 0xffff || vendor == 0x0000)
 785                         continue;
 786
 787                 pci_bus_read_config(bus, bdf, PCI_HEADER_TYPE,
 788                                     &header_type, PCI_SIZE_8);
 789
 790                 if (!PCI_FUNC(bdf))
 791                         found_multi = header_type & 0x80;
 792
 793                 debug("%s: bus %d/%s: found device %x, function %d\n", __func__,
 794                       bus->seq, bus->name, PCI_DEV(bdf), PCI_FUNC(bdf));
 795                 pci_bus_read_config(bus, bdf, PCI_DEVICE_ID, &device,
 796                                     PCI_SIZE_16);
 797                 pci_bus_read_config(bus, bdf, PCI_CLASS_REVISION, &class,
 798                                     PCI_SIZE_32);
 799                 class >>= 8;
 800
 801                 /* Find this device in the device tree */
 802                 ret = pci_bus_find_devfn(bus, PCI_MASK_BUS(bdf), &dev);
 803
 804                 /* If nothing in the device tree, bind a device */
 805                 if (ret == -ENODEV) {
 806                         struct pci_device_id find_id;
 807                         ulong val;
 808
 809                         memset(&find_id, '\0', sizeof(find_id));
 810                         find_id.vendor = vendor;
 811                         find_id.device = device;
 812                         find_id.class = class;
 813                         if ((header_type & 0x7f) == PCI_HEADER_TYPE_NORMAL) {
 814                                 pci_bus_read_config(bus, bdf,
 815                                                     PCI_SUBSYSTEM_VENDOR_ID,
 816                                                     &val, PCI_SIZE_32);
 817                                 find_id.subvendor = val & 0xffff;
 818                                 find_id.subdevice = val >> 16;
 819                         }
 820                         ret = pci_find_and_bind_driver(bus, &find_id, bdf,
 821                                                        &dev);
 822                 }
 823                 if (ret == -EPERM)
 824                         continue;
 825                 else if (ret)
 826                         return ret;
 827
 828                 /* Update the platform data */
 829                 pplat = dev_get_parent_platdata(dev);
 830                 pplat->devfn = PCI_MASK_BUS(bdf);
 831                 pplat->vendor = vendor;
 832                 pplat->device = device;
 833                 pplat->class = class;
 834         }
 835
 836         return 0;
 837 error:
 838         printf("Cannot read bus configuration: %d\n", ret);
 839
 840         return ret;
 841 }
 842
 843 static void decode_regions(struct pci_controller *hose, ofnode parent_node,
 844                            ofnode node)
 845 {
 846         int pci_addr_cells, addr_cells, size_cells;
 847         int cells_per_record;
 848         const u32 *prop;
 849         int len;
 850         int i;
 851
 852         prop = ofnode_get_property(node, "ranges", &len);
 853         if (!prop) {
 854                 debug("%s: Cannot decode regions\n", __func__);
 855                 return;
 856         }
 857
 858         pci_addr_cells = ofnode_read_simple_addr_cells(node);
 859         addr_cells = ofnode_read_simple_addr_cells(parent_node);
 860         size_cells = ofnode_read_simple_size_cells(node);
 861
 862         /* PCI addresses are always 3-cells */
 863         len /= sizeof(u32);
 864         cells_per_record = pci_addr_cells + addr_cells + size_cells;
 865         hose->region_count = 0;
 866         debug("%s: len=%d, cells_per_record=%d\n", __func__, len,
 867               cells_per_record);
 868         for (i = 0; i < MAX_PCI_REGIONS; i++, len -= cells_per_record) {
 869                 u64 pci_addr, addr, size;
 870                 int space_code;
 871                 u32 flags;
 872                 int type;
 873                 int pos;
 874
 875                 if (len < cells_per_record)
 876                         break;
 877                 flags = fdt32_to_cpu(prop[0]);
 878                 space_code = (flags >> 24) & 3;
 879                 pci_addr = fdtdec_get_number(prop + 1, 2);
 880                 prop += pci_addr_cells;
 881                 addr = fdtdec_get_number(prop, addr_cells);
 882                 prop += addr_cells;
 883                 size = fdtdec_get_number(prop, size_cells);
 884                 prop += size_cells;
 885                 debug("%s: region %d, pci_addr=%llx, addr=%llx, size=%llx, space_code=%d\n",
 886                       __func__, hose->region_count, pci_addr, addr, size, space_code);
 887                 if (space_code & 2) {
 888                         type = flags & (1U << 30) ? PCI_REGION_PREFETCH :
 889                                         PCI_REGION_MEM;
 890                 } else if (space_code & 1) {
 891                         type = PCI_REGION_IO;
 892                 } else {
 893                         continue;
 894                 }
 895
 896                 if (!IS_ENABLED(CONFIG_SYS_PCI_64BIT) &&
 897                     type == PCI_REGION_MEM && upper_32_bits(pci_addr)) {
 898                         debug(" - beyond the 32-bit boundary, ignoring\n");
 899                         continue;
 900                 }
 901
 902                 pos = -1;
 903                 for (i = 0; i < hose->region_count; i++) {
 904                         if (hose->regions[i].flags == type)
 905                                 pos = i;
 906                 }
 907                 if (pos == -1)
 908                         pos = hose->region_count++;
 909                 debug(" - type=%d, pos=%d\n", type, pos);
 910                 pci_set_region(hose->regions + pos, pci_addr, addr, size, type);
 911         }
 912
 913         /* Add a region for our local memory */
 914 #ifdef CONFIG_NR_DRAM_BANKS
 915         bd_t *bd = gd->bd;
 916
 917         if (!bd)
 918                 return;
 919
 920         for (i = 0; i < CONFIG_NR_DRAM_BANKS; ++i) {
 921                 if (hose->region_count == MAX_PCI_REGIONS) {
 922                         pr_err("maximum number of regions parsed, aborting\n");
 923                         break;
 924                 }
 925
 926                 if (bd->bi_dram[i].size) {
 927                         pci_set_region(hose->regions + hose->region_count++,
 928                                        bd->bi_dram[i].start,
 929                                        bd->bi_dram[i].start,
 930                                        bd->bi_dram[i].size,
 931                                        PCI_REGION_MEM | PCI_REGION_SYS_MEMORY);
 932                 }
 933         }
 934 #else
 935         phys_addr_t base = 0, size;
 936
 937         size = gd->ram_size;
 938 #ifdef CONFIG_SYS_SDRAM_BASE
 939         base = CONFIG_SYS_SDRAM_BASE;
 940 #endif
 941         if (gd->pci_ram_top && gd->pci_ram_top < base + size)
 942                 size = gd->pci_ram_top - base;
 943         if (size)
 944                 pci_set_region(hose->regions + hose->region_count++, base,
 945                         base, size, PCI_REGION_MEM | PCI_REGION_SYS_MEMORY);
 946 #endif
 947
 948         return;
 949 }
 950
 951 static int pci_uclass_pre_probe(struct udevice *bus)
 952 {
 953         struct pci_controller *hose;
 954
 955         debug("%s, bus=%d/%s, parent=%s\n", __func__, bus->seq, bus->name,
 956               bus->parent->name);
 957         hose = bus->uclass_priv;
 958
 959         /* For bridges, use the top-level PCI controller */
 960         if (!device_is_on_pci_bus(bus)) {
 961                 hose->ctlr = bus;
 962                 decode_regions(hose, dev_ofnode(bus->parent), dev_ofnode(bus));
 963         } else {
 964                 struct pci_controller *parent_hose;
 965
 966                 parent_hose = dev_get_uclass_priv(bus->parent);
 967                 hose->ctlr = parent_hose->bus;
 968         }
 969         hose->bus = bus;
 970         hose->first_busno = bus->seq;
 971         hose->last_busno = bus->seq;
 972
 973         return 0;
 974 }
 975
 976 static int pci_uclass_post_probe(struct udevice *bus)
 977 {
 978         int ret;
 979
 980         debug("%s: probing bus %d\n", __func__, bus->seq);
 981         ret = pci_bind_bus_devices(bus);
 982         if (ret)
 983                 return ret;
 984
 985 #ifdef CONFIG_PCI_PNP
 986         ret = pci_auto_config_devices(bus);
 987         if (ret < 0)
 988                 return ret;
 989 #endif
 990
 991 #if defined(CONFIG_X86) && defined(CONFIG_HAVE_FSP)
 992         /*
 993          * Per Intel FSP specification, we should call FSP notify API to
 994          * inform FSP that PCI enumeration has been done so that FSP will
 995          * do any necessary initialization as required by the chipset's
 996          * BIOS Writer's Guide (BWG).
 997          *
 998          * Unfortunately we have to put this call here as with driver model,
 999          * the enumeration is all done on a lazy basis as needed, so until
1000          * something is touched on PCI it won't happen.
1001          *
1002          * Note we only call this 1) after U-Boot is relocated, and 2)
1003          * root bus has finished probing.
1004          */
1005         if ((gd->flags & GD_FLG_RELOC) && (bus->seq == 0)) {
1006                 ret = fsp_init_phase_pci();
1007                 if (ret)
1008                         return ret;
1009         }
1010 #endif
1011
1012         return 0;
1013 }
1014
1015 int pci_get_devfn(struct udevice *dev)
1016 {
1017         struct fdt_pci_addr addr;
1018         int ret;
1019
1020         /* Extract the devfn from fdt_pci_addr */
1021         ret = ofnode_read_pci_addr(dev_ofnode(dev), FDT_PCI_SPACE_CONFIG,
1022                                    "reg", &addr);
1023         if (ret) {
1024                 if (ret != -ENOENT)
1025                         return -EINVAL;
1026         }
1027
1028         return addr.phys_hi & 0xff00;
1029 }
1030
1031 static int pci_uclass_child_post_bind(struct udevice *dev)
1032 {
1033         struct pci_child_platdata *pplat;
1034
1035         if (!dev_of_valid(dev))
1036                 return 0;
1037
1038         pplat = dev_get_parent_platdata(dev);
1039
1040         /* Extract vendor id and device id if available */
1041         ofnode_read_pci_vendev(dev_ofnode(dev), &pplat->vendor, &pplat->device);
1042
1043         /* Extract the devfn from fdt_pci_addr */
1044         pplat->devfn = pci_get_devfn(dev);
1045
1046         return 0;
1047 }
1048
1049 static int pci_bridge_read_config(struct udevice *bus, pci_dev_t bdf,
1050                                   uint offset, ulong *valuep,
1051                                   enum pci_size_t size)
1052 {
1053         struct pci_controller *hose = bus->uclass_priv;
1054
1055         return pci_bus_read_config(hose->ctlr, bdf, offset, valuep, size);
1056 }
1057
1058 static int pci_bridge_write_config(struct udevice *bus, pci_dev_t bdf,
1059                                    uint offset, ulong value,
1060                                    enum pci_size_t size)
1061 {
1062         struct pci_controller *hose = bus->uclass_priv;
1063
1064         return pci_bus_write_config(hose->ctlr, bdf, offset, value, size);
1065 }
1066
1067 static int skip_to_next_device(struct udevice *bus, struct udevice **devp)
1068 {
1069         struct udevice *dev;
1070         int ret = 0;
1071
1072         /*
1073          * Scan through all the PCI controllers. On x86 there will only be one
1074          * but that is not necessarily true on other hardware.
1075          */
1076         do {
1077                 device_find_first_child(bus, &dev);
1078                 if (dev) {
1079                         *devp = dev;
1080                         return 0;
1081                 }
1082                 ret = uclass_next_device(&bus);
1083                 if (ret)
1084                         return ret;
1085         } while (bus);
1086
1087         return 0;
1088 }
1089
1090 int pci_find_next_device(struct udevice **devp)
1091 {
1092         struct udevice *child = *devp;
1093         struct udevice *bus = child->parent;
1094         int ret;
1095
1096         /* First try all the siblings */
1097         *devp = NULL;
1098         while (child) {
1099                 device_find_next_child(&child);
1100                 if (child) {
1101                         *devp = child;
1102                         return 0;
1103                 }
1104         }
1105
1106         /* We ran out of siblings. Try the next bus */
1107         ret = uclass_next_device(&bus);
1108         if (ret)
1109                 return ret;
1110
1111         return bus ? skip_to_next_device(bus, devp) : 0;
1112 }
1113
1114 int pci_find_first_device(struct udevice **devp)
1115 {
1116         struct udevice *bus;
1117         int ret;
1118
1119         *devp = NULL;
1120         ret = uclass_first_device(UCLASS_PCI, &bus);
1121         if (ret)
1122                 return ret;
1123
1124         return skip_to_next_device(bus, devp);
1125 }
1126
1127 ulong pci_conv_32_to_size(ulong value, uint offset, enum pci_size_t size)
1128 {
1129         switch (size) {
1130         case PCI_SIZE_8:
1131                 return (value >> ((offset & 3) * 8)) & 0xff;
1132         case PCI_SIZE_16:
1133                 return (value >> ((offset & 2) * 8)) & 0xffff;
1134         default:
1135                 return value;
1136         }
1137 }
1138
1139 ulong pci_conv_size_to_32(ulong old, ulong value, uint offset,
1140                           enum pci_size_t size)
1141 {
1142         uint off_mask;
1143         uint val_mask, shift;
1144         ulong ldata, mask;
1145
1146         switch (size) {
1147         case PCI_SIZE_8:
1148                 off_mask = 3;
1149                 val_mask = 0xff;
1150                 break;
1151         case PCI_SIZE_16:
1152                 off_mask = 2;
1153                 val_mask = 0xffff;
1154                 break;
1155         default:
1156                 return value;
1157         }
1158         shift = (offset & off_mask) * 8;
1159         ldata = (value & val_mask) << shift;
1160         mask = val_mask << shift;
1161         value = (old & ~mask) | ldata;
1162
1163         return value;
1164 }
1165
1166 int pci_get_regions(struct udevice *dev, struct pci_region **iop,
1167                     struct pci_region **memp, struct pci_region **prefp)
1168 {
1169         struct udevice *bus = pci_get_controller(dev);
1170         struct pci_controller *hose = dev_get_uclass_priv(bus);
1171         int i;
1172
1173         *iop = NULL;
1174         *memp = NULL;
1175         *prefp = NULL;
1176         for (i = 0; i < hose->region_count; i++) {
1177                 switch (hose->regions[i].flags) {
1178                 case PCI_REGION_IO:
1179                         if (!*iop || (*iop)->size < hose->regions[i].size)
1180                                 *iop = hose->regions + i;
1181                         break;
1182                 case PCI_REGION_MEM:
1183                         if (!*memp || (*memp)->size < hose->regions[i].size)
1184                                 *memp = hose->regions + i;
1185                         break;
1186                 case (PCI_REGION_MEM | PCI_REGION_PREFETCH):
1187                         if (!*prefp || (*prefp)->size < hose->regions[i].size)
1188                                 *prefp = hose->regions + i;
1189                         break;
1190                 }
1191         }
1192
1193         return (*iop != NULL) + (*memp != NULL) + (*prefp != NULL);
1194 }
1195
1196 u32 dm_pci_read_bar32(struct udevice *dev, int barnum)
1197 {
1198         u32 addr;
1199         int bar;
1200
1201         bar = PCI_BASE_ADDRESS_0 + barnum * 4;
1202         dm_pci_read_config32(dev, bar, &addr);
1203         if (addr & PCI_BASE_ADDRESS_SPACE_IO)
1204                 return addr & PCI_BASE_ADDRESS_IO_MASK;
1205         else
1206                 return addr & PCI_BASE_ADDRESS_MEM_MASK;
1207 }
1208
1209 void dm_pci_write_bar32(struct udevice *dev, int barnum, u32 addr)
1210 {
1211         int bar;
1212
1213         bar = PCI_BASE_ADDRESS_0 + barnum * 4;
1214         dm_pci_write_config32(dev, bar, addr);
1215 }
1216
1217 static int _dm_pci_bus_to_phys(struct udevice *ctlr,
1218                                pci_addr_t bus_addr, unsigned long flags,
1219                                unsigned long skip_mask, phys_addr_t *pa)
1220 {
1221         struct pci_controller *hose = dev_get_uclass_priv(ctlr);
1222         struct pci_region *res;
1223         int i;
1224
1225         if (hose->region_count == 0) {
1226                 *pa = bus_addr;
1227                 return 0;
1228         }
1229
1230         for (i = 0; i < hose->region_count; i++) {
1231                 res = &hose->regions[i];
1232
1233                 if (((res->flags ^ flags) & PCI_REGION_TYPE) != 0)
1234                         continue;
1235
1236                 if (res->flags & skip_mask)
1237                         continue;
1238
1239                 if (bus_addr >= res->bus_start &&
1240                     (bus_addr - res->bus_start) < res->size) {
1241                         *pa = (bus_addr - res->bus_start + res->phys_start);
1242                         return 0;
1243                 }
1244         }
1245
1246         return 1;
1247 }
1248
1249 phys_addr_t dm_pci_bus_to_phys(struct udevice *dev, pci_addr_t bus_addr,
1250                                unsigned long flags)
1251 {
1252         phys_addr_t phys_addr = 0;
1253         struct udevice *ctlr;
1254         int ret;
1255
1256         /* The root controller has the region information */
1257         ctlr = pci_get_controller(dev);
1258
1259         /*
1260          * if PCI_REGION_MEM is set we do a two pass search with preference
1261          * on matches that don't have PCI_REGION_SYS_MEMORY set
1262          */
1263         if ((flags & PCI_REGION_TYPE) == PCI_REGION_MEM) {
1264                 ret = _dm_pci_bus_to_phys(ctlr, bus_addr,
1265                                           flags, PCI_REGION_SYS_MEMORY,
1266                                           &phys_addr);
1267                 if (!ret)
1268                         return phys_addr;
1269         }
1270
1271         ret = _dm_pci_bus_to_phys(ctlr, bus_addr, flags, 0, &phys_addr);
1272
1273         if (ret)
1274                 puts("pci_hose_bus_to_phys: invalid physical address\n");
1275
1276         return phys_addr;
1277 }
1278
1279 int _dm_pci_phys_to_bus(struct udevice *dev, phys_addr_t phys_addr,
1280                         unsigned long flags, unsigned long skip_mask,
1281                         pci_addr_t *ba)
1282 {
1283         struct pci_region *res;
1284         struct udevice *ctlr;
1285         pci_addr_t bus_addr;
1286         int i;
1287         struct pci_controller *hose;
1288
1289         /* The root controller has the region information */
1290         ctlr = pci_get_controller(dev);
1291         hose = dev_get_uclass_priv(ctlr);
1292
1293         if (hose->region_count == 0) {
1294                 *ba = phys_addr;
1295                 return 0;
1296         }
1297
1298         for (i = 0; i < hose->region_count; i++) {
1299                 res = &hose->regions[i];
1300
1301                 if (((res->flags ^ flags) & PCI_REGION_TYPE) != 0)
1302                         continue;
1303
1304                 if (res->flags & skip_mask)
1305                         continue;
1306
1307                 bus_addr = phys_addr - res->phys_start + res->bus_start;
1308
1309                 if (bus_addr >= res->bus_start &&
1310                     (bus_addr - res->bus_start) < res->size) {
1311                         *ba = bus_addr;
1312                         return 0;
1313                 }
1314         }
1315
1316         return 1;
1317 }
1318
1319 pci_addr_t dm_pci_phys_to_bus(struct udevice *dev, phys_addr_t phys_addr,
1320                               unsigned long flags)
1321 {
1322         pci_addr_t bus_addr = 0;
1323         int ret;
1324
1325         /*
1326          * if PCI_REGION_MEM is set we do a two pass search with preference
1327          * on matches that don't have PCI_REGION_SYS_MEMORY set
1328          */
1329         if ((flags & PCI_REGION_TYPE) == PCI_REGION_MEM) {
1330                 ret = _dm_pci_phys_to_bus(dev, phys_addr, flags,
1331                                           PCI_REGION_SYS_MEMORY, &bus_addr);
1332                 if (!ret)
1333                         return bus_addr;
1334         }
1335
1336         ret = _dm_pci_phys_to_bus(dev, phys_addr, flags, 0, &bus_addr);
1337
1338         if (ret)
1339                 puts("pci_hose_phys_to_bus: invalid physical address\n");
1340
1341         return bus_addr;
1342 }
1343
1344 static void *dm_pci_map_ea_bar(struct udevice *dev, int bar, int flags,
1345                                int ea_off)
1346 {
1347         int ea_cnt, i, entry_size;
1348         int bar_id = (bar - PCI_BASE_ADDRESS_0) >> 2;
1349         u32 ea_entry;
1350         phys_addr_t addr;
1351
1352         /* EA capability structure header */
1353         dm_pci_read_config32(dev, ea_off, &ea_entry);
1354         ea_cnt = (ea_entry >> 16) & PCI_EA_NUM_ENT_MASK;
1355         ea_off += PCI_EA_FIRST_ENT;
1356
1357         for (i = 0; i < ea_cnt; i++, ea_off += entry_size) {
1358                 /* Entry header */
1359                 dm_pci_read_config32(dev, ea_off, &ea_entry);
1360                 entry_size = ((ea_entry & PCI_EA_ES) + 1) << 2;
1361
1362                 if (((ea_entry & PCI_EA_BEI) >> 4) != bar_id)
1363                         continue;
1364
1365                 /* Base address, 1st DW */
1366                 dm_pci_read_config32(dev, ea_off + 4, &ea_entry);
1367                 addr = ea_entry & PCI_EA_FIELD_MASK;
1368                 if (ea_entry & PCI_EA_IS_64) {
1369                         /* Base address, 2nd DW, skip over 4B MaxOffset */
1370                         dm_pci_read_config32(dev, ea_off + 12, &ea_entry);
1371                         addr |= ((u64)ea_entry) << 32;
1372                 }
1373
1374                 /* size ignored for now */
1375                 return map_physmem(addr, flags, 0);
1376         }
1377
1378         return 0;
1379 }
1380
1381 void *dm_pci_map_bar(struct udevice *dev, int bar, int flags)
1382 {
1383         pci_addr_t pci_bus_addr;
1384         u32 bar_response;
1385         int ea_off;
1386
1387         /*
1388          * if the function supports Enhanced Allocation use that instead of
1389          * BARs
1390          */
1391         ea_off = dm_pci_find_capability(dev, PCI_CAP_ID_EA);
1392         if (ea_off)
1393                 return dm_pci_map_ea_bar(dev, bar, flags, ea_off);
1394
1395         /* read BAR address */
1396         dm_pci_read_config32(dev, bar, &bar_response);
1397         pci_bus_addr = (pci_addr_t)(bar_response & ~0xf);
1398
1399         /*
1400          * Pass "0" as the length argument to pci_bus_to_virt.  The arg
1401          * isn't actualy used on any platform because u-boot assumes a static
1402          * linear mapping.  In the future, this could read the BAR size
1403          * and pass that as the size if needed.
1404          */
1405         return dm_pci_bus_to_virt(dev, pci_bus_addr, flags, 0, MAP_NOCACHE);
1406 }
1407
1408 static int _dm_pci_find_next_capability(struct udevice *dev, u8 pos, int cap)
1409 {
1410         int ttl = PCI_FIND_CAP_TTL;
1411         u8 id;
1412         u16 ent;
1413
1414         dm_pci_read_config8(dev, pos, &pos);
1415
1416         while (ttl--) {
1417                 if (pos < PCI_STD_HEADER_SIZEOF)
1418                         break;
1419                 pos &= ~3;
1420                 dm_pci_read_config16(dev, pos, &ent);
1421
1422                 id = ent & 0xff;
1423                 if (id == 0xff)
1424                         break;
1425                 if (id == cap)
1426                         return pos;
1427                 pos = (ent >> 8);
1428         }
1429
1430         return 0;
1431 }
1432
1433 int dm_pci_find_next_capability(struct udevice *dev, u8 start, int cap)
1434 {
1435         return _dm_pci_find_next_capability(dev, start + PCI_CAP_LIST_NEXT,
1436                                             cap);
1437 }
1438
1439 int dm_pci_find_capability(struct udevice *dev, int cap)
1440 {
1441         u16 status;
1442         u8 header_type;
1443         u8 pos;
1444
1445         dm_pci_read_config16(dev, PCI_STATUS, &status);
1446         if (!(status & PCI_STATUS_CAP_LIST))
1447                 return 0;
1448
1449         dm_pci_read_config8(dev, PCI_HEADER_TYPE, &header_type);
1450         if ((header_type & 0x7f) == PCI_HEADER_TYPE_CARDBUS)
1451                 pos = PCI_CB_CAPABILITY_LIST;
1452         else
1453                 pos = PCI_CAPABILITY_LIST;
1454
1455         return _dm_pci_find_next_capability(dev, pos, cap);
1456 }
1457
1458 int dm_pci_find_next_ext_capability(struct udevice *dev, int start, int cap)
1459 {
1460         u32 header;
1461         int ttl;
1462         int pos = PCI_CFG_SPACE_SIZE;
1463
1464         /* minimum 8 bytes per capability */
1465         ttl = (PCI_CFG_SPACE_EXP_SIZE - PCI_CFG_SPACE_SIZE) / 8;
1466
1467         if (start)
1468                 pos = start;
1469
1470         dm_pci_read_config32(dev, pos, &header);
1471         /*
1472          * If we have no capabilities, this is indicated by cap ID,
1473          * cap version and next pointer all being 0.
1474          */
1475         if (header == 0)
1476                 return 0;
1477
1478         while (ttl--) {
1479                 if (PCI_EXT_CAP_ID(header) == cap)
1480                         return pos;
1481
1482                 pos = PCI_EXT_CAP_NEXT(header);
1483                 if (pos < PCI_CFG_SPACE_SIZE)
1484                         break;
1485
1486                 dm_pci_read_config32(dev, pos, &header);
1487         }
1488
1489         return 0;
1490 }
1491
1492 int dm_pci_find_ext_capability(struct udevice *dev, int cap)
1493 {
1494         return dm_pci_find_next_ext_capability(dev, 0, cap);
1495 }
1496
1497 int dm_pci_flr(struct udevice *dev)
1498 {
1499         int pcie_off;
1500         u32 cap;
1501
1502         /* look for PCI Express Capability */
1503         pcie_off = dm_pci_find_capability(dev, PCI_CAP_ID_EXP);
1504         if (!pcie_off)
1505                 return -ENOENT;
1506
1507         /* check FLR capability */
1508         dm_pci_read_config32(dev, pcie_off + PCI_EXP_DEVCAP, &cap);
1509         if (!(cap & PCI_EXP_DEVCAP_FLR))
1510                 return -ENOENT;
1511
1512         dm_pci_clrset_config16(dev, pcie_off + PCI_EXP_DEVCTL, 0,
1513                                PCI_EXP_DEVCTL_BCR_FLR);
1514
1515         /* wait 100ms, per PCI spec */
1516         mdelay(100);
1517
1518         return 0;
1519 }
1520
1521 UCLASS_DRIVER(pci) = {
1522         .id             = UCLASS_PCI,
1523         .name           = "pci",
1524         .flags          = DM_UC_FLAG_SEQ_ALIAS,
1525         .post_bind      = dm_scan_fdt_dev,
1526         .pre_probe      = pci_uclass_pre_probe,
1527         .post_probe     = pci_uclass_post_probe,
1528         .child_post_bind = pci_uclass_child_post_bind,
1529         .per_device_auto_alloc_size = sizeof(struct pci_controller),
1530         .per_child_platdata_auto_alloc_size =
1531                         sizeof(struct pci_child_platdata),
1532 };
1533
1534 static const struct dm_pci_ops pci_bridge_ops = {
1535         .read_config    = pci_bridge_read_config,
1536         .write_config   = pci_bridge_write_config,
1537 };
1538
1539 static const struct udevice_id pci_bridge_ids[] = {
1540         { .compatible = "pci-bridge" },
1541         { }
1542 };
1543
1544 U_BOOT_DRIVER(pci_bridge_drv) = {
1545         .name           = "pci_bridge_drv",
1546         .id             = UCLASS_PCI,
1547         .of_match       = pci_bridge_ids,
1548         .ops            = &pci_bridge_ops,
1549 };
1550
1551 UCLASS_DRIVER(pci_generic) = {
1552         .id             = UCLASS_PCI_GENERIC,
1553         .name           = "pci_generic",
1554 };
1555
1556 static const struct udevice_id pci_generic_ids[] = {
1557         { .compatible = "pci-generic" },
1558         { }
1559 };
1560
1561 U_BOOT_DRIVER(pci_generic_drv) = {
1562         .name           = "pci_generic_drv",
1563         .id             = UCLASS_PCI_GENERIC,
1564         .of_match       = pci_generic_ids,
1565 };
1566
1567 void pci_init(void)
1568 {
1569         struct udevice *bus;
1570
1571         /*
1572          * Enumerate all known controller devices. Enumeration has the side-
1573          * effect of probing them, so PCIe devices will be enumerated too.
1574          */
1575         for (uclass_first_device_check(UCLASS_PCI, &bus);
1576              bus;
1577              uclass_next_device_check(&bus)) {
1578                 ;
1579         }
1580 }