1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright(c) 2013-2015 Intel Corporation. All rights reserved.
4  */
5 #include <linux/vmalloc.h>
6 #include <linux/module.h>
7 #include <linux/device.h>
8 #include <linux/sizes.h>
9 #include <linux/ndctl.h>
10 #include <linux/slab.h>
11 #include <linux/mm.h>
12 #include <linux/nd.h>
13 #include "label.h"
14 #include "nd.h"
15 
nvdimm_probe(struct device * dev)16 static int nvdimm_probe(struct device *dev)
17 {
18 	struct nvdimm_drvdata *ndd;
19 	int rc;
20 
21 	rc = nvdimm_security_setup_events(dev);
22 	if (rc < 0) {
23 		dev_err(dev, "security event setup failed: %d\n", rc);
24 		return rc;
25 	}
26 
27 	rc = nvdimm_check_config_data(dev);
28 	if (rc) {
29 		/* not required for non-aliased nvdimm, ex. NVDIMM-N */
30 		if (rc == -ENOTTY)
31 			rc = 0;
32 		return rc;
33 	}
34 
35 	/*
36 	 * The locked status bit reflects explicit status codes from the
37 	 * label reading commands, revalidate it each time the driver is
38 	 * activated and re-reads the label area.
39 	 */
40 	nvdimm_clear_locked(dev);
41 
42 	ndd = kzalloc(sizeof(*ndd), GFP_KERNEL);
43 	if (!ndd)
44 		return -ENOMEM;
45 
46 	dev_set_drvdata(dev, ndd);
47 	ndd->dpa.name = dev_name(dev);
48 	ndd->ns_current = -1;
49 	ndd->ns_next = -1;
50 	ndd->dpa.start = 0;
51 	ndd->dpa.end = -1;
52 	ndd->dev = dev;
53 	get_device(dev);
54 	kref_init(&ndd->kref);
55 
56 	/*
57 	 * Attempt to unlock, if the DIMM supports security commands,
58 	 * otherwise the locked indication is determined by explicit
59 	 * status codes from the label reading commands.
60 	 */
61 	rc = nvdimm_security_unlock(dev);
62 	if (rc < 0)
63 		dev_dbg(dev, "failed to unlock dimm: %d\n", rc);
64 
65 
66 	/*
67 	 * EACCES failures reading the namespace label-area-properties
68 	 * are interpreted as the DIMM capacity being locked but the
69 	 * namespace labels themselves being accessible.
70 	 */
71 	rc = nvdimm_init_nsarea(ndd);
72 	if (rc == -EACCES) {
73 		/*
74 		 * See nvdimm_namespace_common_probe() where we fail to
75 		 * allow namespaces to probe while the DIMM is locked,
76 		 * but we do allow for namespace enumeration.
77 		 */
78 		nvdimm_set_locked(dev);
79 		rc = 0;
80 	}
81 	if (rc)
82 		goto err;
83 
84 	/*
85 	 * EACCES failures reading the namespace label-data are
86 	 * interpreted as the label area being locked in addition to the
87 	 * DIMM capacity. We fail the dimm probe to prevent regions from
88 	 * attempting to parse the label area.
89 	 */
90 	rc = nd_label_data_init(ndd);
91 	if (rc == -EACCES)
92 		nvdimm_set_locked(dev);
93 	if (rc)
94 		goto err;
95 
96 	dev_dbg(dev, "config data size: %d\n", ndd->nsarea.config_size);
97 
98 	nvdimm_bus_lock(dev);
99 	if (ndd->ns_current >= 0) {
100 		rc = nd_label_reserve_dpa(ndd);
101 		if (rc == 0)
102 			nvdimm_set_labeling(dev);
103 	}
104 	nvdimm_bus_unlock(dev);
105 
106 	if (rc)
107 		goto err;
108 
109 	return 0;
110 
111  err:
112 	put_ndd(ndd);
113 	return rc;
114 }
115 
nvdimm_remove(struct device * dev)116 static void nvdimm_remove(struct device *dev)
117 {
118 	struct nvdimm_drvdata *ndd = dev_get_drvdata(dev);
119 
120 	nvdimm_bus_lock(dev);
121 	dev_set_drvdata(dev, NULL);
122 	nvdimm_bus_unlock(dev);
123 	put_ndd(ndd);
124 }
125 
126 static struct nd_device_driver nvdimm_driver = {
127 	.probe = nvdimm_probe,
128 	.remove = nvdimm_remove,
129 	.drv = {
130 		.name = "nvdimm",
131 	},
132 	.type = ND_DRIVER_DIMM,
133 };
134 
nvdimm_init(void)135 int __init nvdimm_init(void)
136 {
137 	return nd_driver_register(&nvdimm_driver);
138 }
139 
nvdimm_exit(void)140 void nvdimm_exit(void)
141 {
142 	driver_unregister(&nvdimm_driver.drv);
143 }
144 
145 MODULE_ALIAS_ND_DEVICE(ND_DEVICE_DIMM);
146