static void pcie_pme_mark_devices(struct pci_dev *port)
 {
        pcie_pme_can_wakeup(port, NULL);
-       if (port->subordinate)
+
+       if (pci_pcie_type(port) == PCI_EXP_TYPE_RC_EC)
+               pcie_walk_rcec(port, pcie_pme_can_wakeup, NULL);
+       else if (port->subordinate)
                pci_walk_bus(port->subordinate, pcie_pme_can_wakeup, NULL);
 }
 
  */
 static int pcie_pme_probe(struct pcie_device *srv)
 {
-       struct pci_dev *port;
+       struct pci_dev *port = srv->port;
        struct pcie_pme_service_data *data;
+       int type = pci_pcie_type(port);
        int ret;
 
+       /* Limit to Root Ports or Root Complex Event Collectors */
+       if (type != PCI_EXP_TYPE_RC_EC &&
+           type != PCI_EXP_TYPE_ROOT_PORT)
+               return -ENODEV;
+
        data = kzalloc(sizeof(*data), GFP_KERNEL);
        if (!data)
                return -ENOMEM;
        data->srv = srv;
        set_service_data(srv, data);
 
-       port = srv->port;
        pcie_pme_interrupt_enable(port, false);
        pcie_clear_root_pme_status(port);
 
 
 static struct pcie_port_service_driver pcie_pme_driver = {
        .name           = "pcie_pme",
-       .port_type      = PCI_EXP_TYPE_ROOT_PORT,
+       .port_type      = PCIE_ANY_PORT,
        .service        = PCIE_PORT_SERVICE_PME,
 
        .probe          = pcie_pme_probe,
 
        }
 #endif
 
-       /*
-        * Root ports are capable of generating PME too.  Root Complex
-        * Event Collectors can also generate PMEs, but we don't handle
-        * those yet.
-        */
-       if (pci_pcie_type(dev) == PCI_EXP_TYPE_ROOT_PORT &&
+       /* Root Ports and Root Complex Event Collectors may generate PMEs */
+       if ((pci_pcie_type(dev) == PCI_EXP_TYPE_ROOT_PORT ||
+            pci_pcie_type(dev) == PCI_EXP_TYPE_RC_EC) &&
            (pcie_ports_native || host->native_pme)) {
                services |= PCIE_PORT_SERVICE_PME;