1 /* 2 * Copyright(c) 2013-2015 Intel Corporation. All rights reserved. 3 * 4 * This program is free software; you can redistribute it and/or modify 5 * it under the terms of version 2 of the GNU General Public License as 6 * published by the Free Software Foundation. 7 * 8 * This program is distributed in the hope that it will be useful, but 9 * WITHOUT ANY WARRANTY; without even the implied warranty of 10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 11 * General Public License for more details. 12 */ 13 #include <linux/vmalloc.h> 14 #include <linux/module.h> 15 #include <linux/device.h> 16 #include <linux/sizes.h> 17 #include <linux/ndctl.h> 18 #include <linux/slab.h> 19 #include <linux/mm.h> 20 #include <linux/nd.h> 21 #include "label.h" 22 #include "nd.h" 23 24 static int nvdimm_probe(struct device *dev) 25 { 26 struct nvdimm_drvdata *ndd; 27 int rc; 28 29 rc = nvdimm_check_config_data(dev); 30 if (rc) { 31 /* not required for non-aliased nvdimm, ex. NVDIMM-N */ 32 if (rc == -ENOTTY) 33 rc = 0; 34 return rc; 35 } 36 37 ndd = kzalloc(sizeof(*ndd), GFP_KERNEL); 38 if (!ndd) 39 return -ENOMEM; 40 41 dev_set_drvdata(dev, ndd); 42 ndd->dpa.name = dev_name(dev); 43 ndd->ns_current = -1; 44 ndd->ns_next = -1; 45 ndd->dpa.start = 0; 46 ndd->dpa.end = -1; 47 ndd->dev = dev; 48 get_device(dev); 49 kref_init(&ndd->kref); 50 51 rc = nvdimm_init_nsarea(ndd); 52 if (rc == -EACCES) 53 nvdimm_set_locked(dev); 54 if (rc) 55 goto err; 56 57 rc = nvdimm_init_config_data(ndd); 58 if (rc == -EACCES) 59 nvdimm_set_locked(dev); 60 if (rc) 61 goto err; 62 63 dev_dbg(dev, "config data size: %d\n", ndd->nsarea.config_size); 64 65 nvdimm_bus_lock(dev); 66 ndd->ns_current = nd_label_validate(ndd); 67 ndd->ns_next = nd_label_next_nsindex(ndd->ns_current); 68 nd_label_copy(ndd, to_next_namespace_index(ndd), 69 to_current_namespace_index(ndd)); 70 if (ndd->ns_current >= 0) { 71 rc = nd_label_reserve_dpa(ndd); 72 if (rc == 0) 73 nvdimm_set_aliasing(dev); 74 } 75 nvdimm_clear_locked(dev); 76 nvdimm_bus_unlock(dev); 77 78 if (rc) 79 goto err; 80 81 return 0; 82 83 err: 84 put_ndd(ndd); 85 return rc; 86 } 87 88 static int nvdimm_remove(struct device *dev) 89 { 90 struct nvdimm_drvdata *ndd = dev_get_drvdata(dev); 91 92 if (!ndd) 93 return 0; 94 95 nvdimm_bus_lock(dev); 96 dev_set_drvdata(dev, NULL); 97 nvdimm_bus_unlock(dev); 98 put_ndd(ndd); 99 100 return 0; 101 } 102 103 static struct nd_device_driver nvdimm_driver = { 104 .probe = nvdimm_probe, 105 .remove = nvdimm_remove, 106 .drv = { 107 .name = "nvdimm", 108 }, 109 .type = ND_DRIVER_DIMM, 110 }; 111 112 int __init nvdimm_init(void) 113 { 114 return nd_driver_register(&nvdimm_driver); 115 } 116 117 void nvdimm_exit(void) 118 { 119 driver_unregister(&nvdimm_driver.drv); 120 } 121 122 MODULE_ALIAS_ND_DEVICE(ND_DEVICE_DIMM); 123