[Lkw] [PATCH 02/10] nvme-cmb: Introducing PIC-driver for NVMe

[Damian Gonzalez]

Signed-off-by: Damian Gonzalez <damian.gonzalez at mi.unc.edu.ar>
---
 drivers/block/nvme-cmb.c | 37 ++++++++++++++++++++++++++++++++-----
 1 file changed, 32 insertions(+), 5 deletions(-)

diff --git a/drivers/block/nvme-cmb.c b/drivers/block/nvme-cmb.c
index e11aae5a8..b57c86ba4 100644
--- a/drivers/block/nvme-cmb.c
+++ b/drivers/block/nvme-cmb.c
@@ -1,17 +1,44 @@
-#include <linux/module.h>
-MODULE_LICENSE("GPL");
 #include <linux/init.h>
+#include <linux/module.h>
 #include <linux/kernel.h>
+#include <linux/pci.h>
 
-static int __init nvme_cmb_init(void)
+MODULE_LICENSE("GPL");
+
+static int nvme_cmb_probe(struct pci_dev *pdev, const struct pci_device_id *id)
 {
-	pr_info("loading nvme-cmb module\n");
+	dev_info(&pdev->dev, "found NVMe device\n");
 	return 0;
 }
 
+static void nvme_cmb_remove(struct pci_dev *pdev)
+{
+	dev_info(&pdev->dev, "unbinding NVMe device\n");
+}
+
+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 = {
+	.name = "nvme_cmb",
+	.id_table = nvme_cmb_id_table,
+	.probe = nvme_cmb_probe,
+	.remove = nvme_cmb_remove,
+};
+
+
+static int __init nvme_cmb_init(void)
+{
+      pr_info("loading nvme-cmb module\n");
+      return pci_register_driver(&nvme_cmb_driver);
+}
+
 static void __exit nvme_cmb_exit(void)
 {
-	pr_info("unloading nvme-cmb module\n");
+       pr_info("unloading nvme-cmb module\n");
+       pci_unregister_driver(&nvme_cmb_driver);
 }
 
 module_init(nvme_cmb_init);
-- 
2.39.2