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