NTB: add pci shutdown handler for AMD NTB

The PCI shutdown handler is invoked in response
to system reboot or shutdown. A data transfer
might still be in flight when this happens. So
the very first action we take here is to send
a link down notification, so that any pending
data transfer is terminated. Rest of the actions
are same as that of PCI remove handler.

Signed-off-by: Arindam Nath <arindam.nath@amd.com>
Signed-off-by: Jon Mason <jdmason@kudzu.us>

diff --git a/drivers/ntb/hw/amd/ntb_hw_amd.c b/drivers/ntb/hw/amd/ntb_hw_amd.c
index c6cea00..9e310e1 100644 (file)
--- a/drivers/ntb/hw/amd/ntb_hw_amd.c
+++ b/drivers/ntb/hw/amd/ntb_hw_amd.c
@@ -1296,6 +1296,22 @@ static void amd_ntb_pci_remove(struct pci_dev *pdev)
        kfree(ndev);
 }
 
+static void amd_ntb_pci_shutdown(struct pci_dev *pdev)
+{
+       struct amd_ntb_dev *ndev = pci_get_drvdata(pdev);
+
+       /* Send link down notification */
+       ntb_link_event(&ndev->ntb);
+
+       amd_deinit_side_info(ndev);
+       ntb_peer_db_set(&ndev->ntb, BIT_ULL(ndev->db_last_bit));
+       ntb_unregister_device(&ndev->ntb);
+       ndev_deinit_debugfs(ndev);
+       amd_deinit_dev(ndev);
+       amd_ntb_deinit_pci(ndev);
+       kfree(ndev);
+}
+
 static const struct file_operations amd_ntb_debugfs_info = {
        .owner = THIS_MODULE,
        .open = simple_open,
@@ -1326,6 +1342,7 @@ static struct pci_driver amd_ntb_pci_driver = {
        .id_table       = amd_ntb_pci_tbl,
        .probe          = amd_ntb_pci_probe,
        .remove         = amd_ntb_pci_remove,
+       .shutdown       = amd_ntb_pci_shutdown,
 };
 
 static int __init amd_ntb_pci_driver_init(void)