}
 __setup("pcie_pme=", pcie_pme_setup);
 
-enum pme_suspend_level {
-       PME_SUSPEND_NONE = 0,
-       PME_SUSPEND_WAKEUP,
-       PME_SUSPEND_NOIRQ,
-};
-
 struct pcie_pme_service_data {
        spinlock_t lock;
        struct pcie_device *srv;
        struct work_struct work;
-       enum pme_suspend_level suspend_level;
+       bool noirq; /* If set, keep the PME interrupt disabled. */
 };
 
 /**
        spin_lock_irq(&data->lock);
 
        for (;;) {
-               if (data->suspend_level != PME_SUSPEND_NONE)
+               if (data->noirq)
                        break;
 
                pcie_capability_read_dword(port, PCI_EXP_RTSTA, &rtsta);
                spin_lock_irq(&data->lock);
        }
 
-       if (data->suspend_level == PME_SUSPEND_NONE)
+       if (!data->noirq)
                pcie_pme_interrupt_enable(port, true);
 
        spin_unlock_irq(&data->lock);
 {
        struct pcie_pme_service_data *data = get_service_data(srv);
        struct pci_dev *port = srv->port;
-       bool wakeup, wake_irq_enabled = false;
+       bool wakeup;
        int ret;
 
        if (device_may_wakeup(&port->dev)) {
                wakeup = pcie_pme_check_wakeup(port->subordinate);
                up_read(&pci_bus_sem);
        }
-       spin_lock_irq(&data->lock);
        if (wakeup) {
                ret = enable_irq_wake(srv->irq);
-               if (ret == 0) {
-                       data->suspend_level = PME_SUSPEND_WAKEUP;
-                       wake_irq_enabled = true;
-               }
-       }
-       if (!wake_irq_enabled) {
-               pcie_pme_interrupt_enable(port, false);
-               pcie_clear_root_pme_status(port);
-               data->suspend_level = PME_SUSPEND_NOIRQ;
+               if (!ret)
+                       return 0;
        }
+
+       spin_lock_irq(&data->lock);
+       pcie_pme_interrupt_enable(port, false);
+       pcie_clear_root_pme_status(port);
+       data->noirq = true;
        spin_unlock_irq(&data->lock);
 
        synchronize_irq(srv->irq);
        struct pcie_pme_service_data *data = get_service_data(srv);
 
        spin_lock_irq(&data->lock);
-       if (data->suspend_level == PME_SUSPEND_NOIRQ) {
+       if (data->noirq) {
                struct pci_dev *port = srv->port;
 
                pcie_clear_root_pme_status(port);
                pcie_pme_interrupt_enable(port, true);
+               data->noirq = false;
        } else {
                disable_irq_wake(srv->irq);
        }
-       data->suspend_level = PME_SUSPEND_NONE;
        spin_unlock_irq(&data->lock);
 
        return 0;