1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun * virtio_pmem.c: Virtio pmem Driver
4*4882a593Smuzhiyun *
5*4882a593Smuzhiyun * Discovers persistent memory range information
6*4882a593Smuzhiyun * from host and registers the virtual pmem device
7*4882a593Smuzhiyun * with libnvdimm core.
8*4882a593Smuzhiyun */
9*4882a593Smuzhiyun #include "virtio_pmem.h"
10*4882a593Smuzhiyun #include "nd.h"
11*4882a593Smuzhiyun
12*4882a593Smuzhiyun static struct virtio_device_id id_table[] = {
13*4882a593Smuzhiyun { VIRTIO_ID_PMEM, VIRTIO_DEV_ANY_ID },
14*4882a593Smuzhiyun { 0 },
15*4882a593Smuzhiyun };
16*4882a593Smuzhiyun
17*4882a593Smuzhiyun /* Initialize virt queue */
init_vq(struct virtio_pmem * vpmem)18*4882a593Smuzhiyun static int init_vq(struct virtio_pmem *vpmem)
19*4882a593Smuzhiyun {
20*4882a593Smuzhiyun /* single vq */
21*4882a593Smuzhiyun vpmem->req_vq = virtio_find_single_vq(vpmem->vdev,
22*4882a593Smuzhiyun virtio_pmem_host_ack, "flush_queue");
23*4882a593Smuzhiyun if (IS_ERR(vpmem->req_vq))
24*4882a593Smuzhiyun return PTR_ERR(vpmem->req_vq);
25*4882a593Smuzhiyun
26*4882a593Smuzhiyun spin_lock_init(&vpmem->pmem_lock);
27*4882a593Smuzhiyun INIT_LIST_HEAD(&vpmem->req_list);
28*4882a593Smuzhiyun
29*4882a593Smuzhiyun return 0;
30*4882a593Smuzhiyun };
31*4882a593Smuzhiyun
virtio_pmem_probe(struct virtio_device * vdev)32*4882a593Smuzhiyun static int virtio_pmem_probe(struct virtio_device *vdev)
33*4882a593Smuzhiyun {
34*4882a593Smuzhiyun struct nd_region_desc ndr_desc = {};
35*4882a593Smuzhiyun int nid = dev_to_node(&vdev->dev);
36*4882a593Smuzhiyun struct nd_region *nd_region;
37*4882a593Smuzhiyun struct virtio_pmem *vpmem;
38*4882a593Smuzhiyun struct resource res;
39*4882a593Smuzhiyun int err = 0;
40*4882a593Smuzhiyun
41*4882a593Smuzhiyun if (!vdev->config->get) {
42*4882a593Smuzhiyun dev_err(&vdev->dev, "%s failure: config access disabled\n",
43*4882a593Smuzhiyun __func__);
44*4882a593Smuzhiyun return -EINVAL;
45*4882a593Smuzhiyun }
46*4882a593Smuzhiyun
47*4882a593Smuzhiyun vpmem = devm_kzalloc(&vdev->dev, sizeof(*vpmem), GFP_KERNEL);
48*4882a593Smuzhiyun if (!vpmem) {
49*4882a593Smuzhiyun err = -ENOMEM;
50*4882a593Smuzhiyun goto out_err;
51*4882a593Smuzhiyun }
52*4882a593Smuzhiyun
53*4882a593Smuzhiyun vpmem->vdev = vdev;
54*4882a593Smuzhiyun vdev->priv = vpmem;
55*4882a593Smuzhiyun err = init_vq(vpmem);
56*4882a593Smuzhiyun if (err) {
57*4882a593Smuzhiyun dev_err(&vdev->dev, "failed to initialize virtio pmem vq's\n");
58*4882a593Smuzhiyun goto out_err;
59*4882a593Smuzhiyun }
60*4882a593Smuzhiyun
61*4882a593Smuzhiyun virtio_cread_le(vpmem->vdev, struct virtio_pmem_config,
62*4882a593Smuzhiyun start, &vpmem->start);
63*4882a593Smuzhiyun virtio_cread_le(vpmem->vdev, struct virtio_pmem_config,
64*4882a593Smuzhiyun size, &vpmem->size);
65*4882a593Smuzhiyun
66*4882a593Smuzhiyun res.start = vpmem->start;
67*4882a593Smuzhiyun res.end = vpmem->start + vpmem->size - 1;
68*4882a593Smuzhiyun vpmem->nd_desc.provider_name = "virtio-pmem";
69*4882a593Smuzhiyun vpmem->nd_desc.module = THIS_MODULE;
70*4882a593Smuzhiyun
71*4882a593Smuzhiyun vpmem->nvdimm_bus = nvdimm_bus_register(&vdev->dev,
72*4882a593Smuzhiyun &vpmem->nd_desc);
73*4882a593Smuzhiyun if (!vpmem->nvdimm_bus) {
74*4882a593Smuzhiyun dev_err(&vdev->dev, "failed to register device with nvdimm_bus\n");
75*4882a593Smuzhiyun err = -ENXIO;
76*4882a593Smuzhiyun goto out_vq;
77*4882a593Smuzhiyun }
78*4882a593Smuzhiyun
79*4882a593Smuzhiyun dev_set_drvdata(&vdev->dev, vpmem->nvdimm_bus);
80*4882a593Smuzhiyun
81*4882a593Smuzhiyun ndr_desc.res = &res;
82*4882a593Smuzhiyun ndr_desc.numa_node = nid;
83*4882a593Smuzhiyun ndr_desc.flush = async_pmem_flush;
84*4882a593Smuzhiyun set_bit(ND_REGION_PAGEMAP, &ndr_desc.flags);
85*4882a593Smuzhiyun set_bit(ND_REGION_ASYNC, &ndr_desc.flags);
86*4882a593Smuzhiyun nd_region = nvdimm_pmem_region_create(vpmem->nvdimm_bus, &ndr_desc);
87*4882a593Smuzhiyun if (!nd_region) {
88*4882a593Smuzhiyun dev_err(&vdev->dev, "failed to create nvdimm region\n");
89*4882a593Smuzhiyun err = -ENXIO;
90*4882a593Smuzhiyun goto out_nd;
91*4882a593Smuzhiyun }
92*4882a593Smuzhiyun nd_region->provider_data = dev_to_virtio(nd_region->dev.parent->parent);
93*4882a593Smuzhiyun return 0;
94*4882a593Smuzhiyun out_nd:
95*4882a593Smuzhiyun nvdimm_bus_unregister(vpmem->nvdimm_bus);
96*4882a593Smuzhiyun out_vq:
97*4882a593Smuzhiyun vdev->config->del_vqs(vdev);
98*4882a593Smuzhiyun out_err:
99*4882a593Smuzhiyun return err;
100*4882a593Smuzhiyun }
101*4882a593Smuzhiyun
virtio_pmem_remove(struct virtio_device * vdev)102*4882a593Smuzhiyun static void virtio_pmem_remove(struct virtio_device *vdev)
103*4882a593Smuzhiyun {
104*4882a593Smuzhiyun struct nvdimm_bus *nvdimm_bus = dev_get_drvdata(&vdev->dev);
105*4882a593Smuzhiyun
106*4882a593Smuzhiyun nvdimm_bus_unregister(nvdimm_bus);
107*4882a593Smuzhiyun vdev->config->del_vqs(vdev);
108*4882a593Smuzhiyun vdev->config->reset(vdev);
109*4882a593Smuzhiyun }
110*4882a593Smuzhiyun
111*4882a593Smuzhiyun static struct virtio_driver virtio_pmem_driver = {
112*4882a593Smuzhiyun .driver.name = KBUILD_MODNAME,
113*4882a593Smuzhiyun .driver.owner = THIS_MODULE,
114*4882a593Smuzhiyun .id_table = id_table,
115*4882a593Smuzhiyun .probe = virtio_pmem_probe,
116*4882a593Smuzhiyun .remove = virtio_pmem_remove,
117*4882a593Smuzhiyun };
118*4882a593Smuzhiyun
119*4882a593Smuzhiyun module_virtio_driver(virtio_pmem_driver);
120*4882a593Smuzhiyun MODULE_DEVICE_TABLE(virtio, id_table);
121*4882a593Smuzhiyun MODULE_DESCRIPTION("Virtio pmem driver");
122*4882a593Smuzhiyun MODULE_LICENSE("GPL");
123