[Lkw] [PATCH] nvme-cmb: add the basic PCI boilerplate code

[Mauricio J Llugdar]

---
 drivers/block/nvme-cmb.c | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/drivers/block/nvme-cmb.c b/drivers/block/nvme-cmb.c
index 56f0e06dc..c78c647d0 100644
--- a/drivers/block/nvme-cmb.c
+++ b/drivers/block/nvme-cmb.c
@@ -7,18 +7,36 @@ MODULE_LICENSE("GPL");
  
 static int nvme_cmb_probe(struct pci_dev *pdev, const struct pci_device_id *id)
 {
+	
+        int error;		
 	dev_info(&pdev->dev, "found NVMe device\n");
+	
+	pci_set_master(pdev);
+	error = pci_enable_device_mem(pdev);
+	if(error)
+		return error;
+	error = pci_request_mem_regions(pdev, "nvme-cmb");
+	if(error)
+		goto out_disable_device;
 	return 0;
+
+out_disable_device:
+	pci_disable_device(pdev);
+	return error;
 }
 
 static void nvme_cmb_remove(struct pci_dev *pdev)
 {
 	dev_info(&pdev->dev, "unbinding NVMe device\n");
+
+	pci_release_mem_regions(pdev);
+	pci_disable_device(pdev);
 }
 
 static const struct pci_device_id nvme_cmb_id_table[] = {
 	{ PCI_DEVICE_CLASS(PCI_CLASS_STORAGE_EXPRESS, 0xffffff) },
 	{ 0, }
+
 };
 
 static struct pci_driver nvme_cmb_driver = {
-- 
2.39.2