Merge tag 'for_linus' of git://git.kernel.org/pub/scm/linux/kernel/git/mst/vhost

[thirdparty/linux.git] / drivers / virtio / virtio_pci_common.c

diff --git a/drivers/virtio/virtio_pci_common.c b/drivers/virtio/virtio_pci_common.c
index 9f93330fc2cc0b3929b39574216c5ae2a66a305b..b655fccaf77330b08b23a57d7e2fe1d012409615 100644 (file)
--- a/drivers/virtio/virtio_pci_common.c
+++ b/drivers/virtio/virtio_pci_common.c
@@ -495,8 +495,40 @@ static int virtio_pci_restore(struct device *dev)
        return virtio_device_restore(&vp_dev->vdev);
 }
 
+static bool vp_supports_pm_no_reset(struct device *dev)
+{
+       struct pci_dev *pci_dev = to_pci_dev(dev);
+       u16 pmcsr;
+
+       if (!pci_dev->pm_cap)
+               return false;
+
+       pci_read_config_word(pci_dev, pci_dev->pm_cap + PCI_PM_CTRL, &pmcsr);
+       if (PCI_POSSIBLE_ERROR(pmcsr)) {
+               dev_err(dev, "Unable to query pmcsr");
+               return false;
+       }
+
+       return pmcsr & PCI_PM_CTRL_NO_SOFT_RESET;
+}
+
+static int virtio_pci_suspend(struct device *dev)
+{
+       return vp_supports_pm_no_reset(dev) ? 0 : virtio_pci_freeze(dev);
+}
+
+static int virtio_pci_resume(struct device *dev)
+{
+       return vp_supports_pm_no_reset(dev) ? 0 : virtio_pci_restore(dev);
+}
+
 static const struct dev_pm_ops virtio_pci_pm_ops = {
-       SET_SYSTEM_SLEEP_PM_OPS(virtio_pci_freeze, virtio_pci_restore)
+       .suspend = virtio_pci_suspend,
+       .resume = virtio_pci_resume,
+       .freeze = virtio_pci_freeze,
+       .thaw = virtio_pci_restore,
+       .poweroff = virtio_pci_freeze,
+       .restore = virtio_pci_restore,
 };
 #endif