1 files changed, 4 insertions, 2 deletions

diff --git a/drivers/pci/pcie/portdrv_core.c b/drivers/pci/pcie/portdrv_core.c
index 079bbc3ed4f8..52c4e6fd6fd4 100644
--- a/drivers/pci/pcie/portdrv_core.c
+++ b/drivers/pci/pcie/portdrv_core.c
@@ -297,6 +297,10 @@ int pcie_port_device_register(struct pci_dev *dev)
         int status, capabilities, irq_mode, i, nr_serv;
         int vectors[PCIE_PORT_DEVICE_MAXSERVICES];
 
+        capabilities = get_port_device_capability(dev);
+        if (!capabilities)
+                return -ENODEV;
+
         port_data = kzalloc(sizeof(*port_data), GFP_KERNEL);
         if (!port_data)
                 return -ENOMEM;
@@ -304,8 +308,6 @@ int pcie_port_device_register(struct pci_dev *dev)
 
         port_data->port_type = dev->pcie_type;
 
-        capabilities = get_port_device_capability(dev);
-
         irq_mode = assign_interrupt_mode(dev, vectors, capabilities);
         if (irq_mode == PCIE_PORT_NO_IRQ) {
                 /*