vp_vdpa.c

  1// SPDX-License-Identifier: GPL-2.0-only
  2/*
  3 * vDPA bridge driver for modern virtio-pci device
  4 *
  5 * Copyright (c) 2020, Red Hat Inc. All rights reserved.
  6 * Author: Jason Wang <jasowang@redhat.com>
  7 *
  8 * Based on virtio_pci_modern.c.
  9 */
 10
 11#include <linux/interrupt.h>
 12#include <linux/module.h>
 13#include <linux/pci.h>
 14#include <linux/vdpa.h>
 15#include <linux/virtio.h>
 16#include <linux/virtio_config.h>
 17#include <linux/virtio_ring.h>
 18#include <linux/virtio_pci.h>
 19#include <linux/virtio_pci_modern.h>
 20#include <uapi/linux/vdpa.h>
 21
 22#define VP_VDPA_QUEUE_MAX 256
 23#define VP_VDPA_DRIVER_NAME "vp_vdpa"
 24#define VP_VDPA_NAME_SIZE 256
 25
 26struct vp_vring {
 27	void __iomem *notify;
 28	char msix_name[VP_VDPA_NAME_SIZE];
 29	struct vdpa_callback cb;
 30	resource_size_t notify_pa;
 31	int irq;
 32};
 33
 34struct vp_vdpa {
 35	struct vdpa_device vdpa;
 36	struct virtio_pci_modern_device *mdev;
 37	struct vp_vring *vring;
 38	struct vdpa_callback config_cb;
 39	u64 device_features;
 40	char msix_name[VP_VDPA_NAME_SIZE];
 41	int config_irq;
 42	int queues;
 43	int vectors;
 44};
 45
 46struct vp_vdpa_mgmtdev {
 47	struct vdpa_mgmt_dev mgtdev;
 48	struct virtio_pci_modern_device *mdev;
 49	struct vp_vdpa *vp_vdpa;
 50};
 51
 52static struct vp_vdpa *vdpa_to_vp(struct vdpa_device *vdpa)
 53{
 54	return container_of(vdpa, struct vp_vdpa, vdpa);
 55}
 56
 57static struct virtio_pci_modern_device *vdpa_to_mdev(struct vdpa_device *vdpa)
 58{
 59	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
 60
 61	return vp_vdpa->mdev;
 62}
 63
 64static struct virtio_pci_modern_device *vp_vdpa_to_mdev(struct vp_vdpa *vp_vdpa)
 65{
 66	return vp_vdpa->mdev;
 67}
 68
 69static u64 vp_vdpa_get_device_features(struct vdpa_device *vdpa)
 70{
 71	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
 72
 73	return vp_vdpa->device_features;
 74}
 75
 76static int vp_vdpa_set_driver_features(struct vdpa_device *vdpa, u64 features)
 77{
 78	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
 79
 80	vp_modern_set_features(mdev, features);
 81
 82	return 0;
 83}
 84
 85static u64 vp_vdpa_get_driver_features(struct vdpa_device *vdpa)
 86{
 87	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
 88
 89	return vp_modern_get_driver_features(mdev);
 90}
 91
 92static u8 vp_vdpa_get_status(struct vdpa_device *vdpa)
 93{
 94	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
 95
 96	return vp_modern_get_status(mdev);
 97}
 98
 99static int vp_vdpa_get_vq_irq(struct vdpa_device *vdpa, u16 idx)
100{
101	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
102	int irq = vp_vdpa->vring[idx].irq;
103
104	if (irq == VIRTIO_MSI_NO_VECTOR)
105		return -EINVAL;
106
107	return irq;
108}
109
110static void vp_vdpa_free_irq(struct vp_vdpa *vp_vdpa)
111{
112	struct virtio_pci_modern_device *mdev = vp_vdpa_to_mdev(vp_vdpa);
113	struct pci_dev *pdev = mdev->pci_dev;
114	int i;
115
116	for (i = 0; i < vp_vdpa->queues; i++) {
117		if (vp_vdpa->vring[i].irq != VIRTIO_MSI_NO_VECTOR) {
118			vp_modern_queue_vector(mdev, i, VIRTIO_MSI_NO_VECTOR);
119			devm_free_irq(&pdev->dev, vp_vdpa->vring[i].irq,
120				      &vp_vdpa->vring[i]);
121			vp_vdpa->vring[i].irq = VIRTIO_MSI_NO_VECTOR;
122		}
123	}
124
125	if (vp_vdpa->config_irq != VIRTIO_MSI_NO_VECTOR) {
126		vp_modern_config_vector(mdev, VIRTIO_MSI_NO_VECTOR);
127		devm_free_irq(&pdev->dev, vp_vdpa->config_irq, vp_vdpa);
128		vp_vdpa->config_irq = VIRTIO_MSI_NO_VECTOR;
129	}
130
131	if (vp_vdpa->vectors) {
132		pci_free_irq_vectors(pdev);
133		vp_vdpa->vectors = 0;
134	}
135}
136
137static irqreturn_t vp_vdpa_vq_handler(int irq, void *arg)
138{
139	struct vp_vring *vring = arg;
140
141	if (vring->cb.callback)
142		return vring->cb.callback(vring->cb.private);
143
144	return IRQ_HANDLED;
145}
146
147static irqreturn_t vp_vdpa_config_handler(int irq, void *arg)
148{
149	struct vp_vdpa *vp_vdpa = arg;
150
151	if (vp_vdpa->config_cb.callback)
152		return vp_vdpa->config_cb.callback(vp_vdpa->config_cb.private);
153
154	return IRQ_HANDLED;
155}
156
157static int vp_vdpa_request_irq(struct vp_vdpa *vp_vdpa)
158{
159	struct virtio_pci_modern_device *mdev = vp_vdpa_to_mdev(vp_vdpa);
160	struct pci_dev *pdev = mdev->pci_dev;
161	int i, ret, irq;
162	int queues = vp_vdpa->queues;
163	int vectors = 1;
164	int msix_vec = 0;
165
166	for (i = 0; i < queues; i++) {
167		if (vp_vdpa->vring[i].cb.callback)
168			vectors++;
169	}
170
171	ret = pci_alloc_irq_vectors(pdev, vectors, vectors, PCI_IRQ_MSIX);
172	if (ret != vectors) {
173		dev_err(&pdev->dev,
174			"vp_vdpa: fail to allocate irq vectors want %d but %d\n",
175			vectors, ret);
176		return ret;
177	}
178
179	vp_vdpa->vectors = vectors;
180
181	for (i = 0; i < queues; i++) {
182		if (!vp_vdpa->vring[i].cb.callback)
183			continue;
184
185		snprintf(vp_vdpa->vring[i].msix_name, VP_VDPA_NAME_SIZE,
186			"vp-vdpa[%s]-%d\n", pci_name(pdev), i);
187		irq = pci_irq_vector(pdev, msix_vec);
188		ret = devm_request_irq(&pdev->dev, irq,
189				       vp_vdpa_vq_handler,
190				       0, vp_vdpa->vring[i].msix_name,
191				       &vp_vdpa->vring[i]);
192		if (ret) {
193			dev_err(&pdev->dev,
194				"vp_vdpa: fail to request irq for vq %d\n", i);
195			goto err;
196		}
197		vp_modern_queue_vector(mdev, i, msix_vec);
198		vp_vdpa->vring[i].irq = irq;
199		msix_vec++;
200	}
201
202	snprintf(vp_vdpa->msix_name, VP_VDPA_NAME_SIZE, "vp-vdpa[%s]-config\n",
203		 pci_name(pdev));
204	irq = pci_irq_vector(pdev, msix_vec);
205	ret = devm_request_irq(&pdev->dev, irq,	vp_vdpa_config_handler, 0,
206			       vp_vdpa->msix_name, vp_vdpa);
207	if (ret) {
208		dev_err(&pdev->dev,
209			"vp_vdpa: fail to request irq for config: %d\n", ret);
210			goto err;
211	}
212	vp_modern_config_vector(mdev, msix_vec);
213	vp_vdpa->config_irq = irq;
214
215	return 0;
216err:
217	vp_vdpa_free_irq(vp_vdpa);
218	return ret;
219}
220
221static void vp_vdpa_set_status(struct vdpa_device *vdpa, u8 status)
222{
223	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
224	struct virtio_pci_modern_device *mdev = vp_vdpa_to_mdev(vp_vdpa);
225	u8 s = vp_vdpa_get_status(vdpa);
226
227	if (status & VIRTIO_CONFIG_S_DRIVER_OK &&
228	    !(s & VIRTIO_CONFIG_S_DRIVER_OK)) {
229		if (vp_vdpa_request_irq(vp_vdpa)) {
230			WARN_ON(1);
231			return;
232		}
233	}
234
235	vp_modern_set_status(mdev, status);
236}
237
238static int vp_vdpa_reset(struct vdpa_device *vdpa)
239{
240	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
241	struct virtio_pci_modern_device *mdev = vp_vdpa_to_mdev(vp_vdpa);
242	u8 s = vp_vdpa_get_status(vdpa);
243
244	vp_modern_set_status(mdev, 0);
245
246	if (s & VIRTIO_CONFIG_S_DRIVER_OK)
247		vp_vdpa_free_irq(vp_vdpa);
248
249	return 0;
250}
251
252static u16 vp_vdpa_get_vq_num_max(struct vdpa_device *vdpa)
253{
254	return VP_VDPA_QUEUE_MAX;
255}
256
257static int vp_vdpa_get_vq_state(struct vdpa_device *vdpa, u16 qid,
258				struct vdpa_vq_state *state)
259{
260	/* Note that this is not supported by virtio specification, so
261	 * we return -EOPNOTSUPP here. This means we can't support live
262	 * migration, vhost device start/stop.
263	 */
264	return -EOPNOTSUPP;
265}
266
267static int vp_vdpa_set_vq_state_split(struct vdpa_device *vdpa,
268				      const struct vdpa_vq_state *state)
269{
270	const struct vdpa_vq_state_split *split = &state->split;
271
272	if (split->avail_index == 0)
273		return 0;
274
275	return -EOPNOTSUPP;
276}
277
278static int vp_vdpa_set_vq_state_packed(struct vdpa_device *vdpa,
279				       const struct vdpa_vq_state *state)
280{
281	const struct vdpa_vq_state_packed *packed = &state->packed;
282
283	if (packed->last_avail_counter == 1 &&
284	    packed->last_avail_idx == 0 &&
285	    packed->last_used_counter == 1 &&
286	    packed->last_used_idx == 0)
287		return 0;
288
289	return -EOPNOTSUPP;
290}
291
292static int vp_vdpa_set_vq_state(struct vdpa_device *vdpa, u16 qid,
293				const struct vdpa_vq_state *state)
294{
295	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
296
297	/* Note that this is not supported by virtio specification.
298	 * But if the state is by chance equal to the device initial
299	 * state, we can let it go.
300	 */
301	if ((vp_modern_get_status(mdev) & VIRTIO_CONFIG_S_FEATURES_OK) &&
302	    !vp_modern_get_queue_enable(mdev, qid)) {
303		if (vp_modern_get_driver_features(mdev) &
304		    BIT_ULL(VIRTIO_F_RING_PACKED))
305			return vp_vdpa_set_vq_state_packed(vdpa, state);
306		else
307			return vp_vdpa_set_vq_state_split(vdpa,	state);
308	}
309
310	return -EOPNOTSUPP;
311}
312
313static void vp_vdpa_set_vq_cb(struct vdpa_device *vdpa, u16 qid,
314			      struct vdpa_callback *cb)
315{
316	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
317
318	vp_vdpa->vring[qid].cb = *cb;
319}
320
321static void vp_vdpa_set_vq_ready(struct vdpa_device *vdpa,
322				 u16 qid, bool ready)
323{
324	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
325
326	vp_modern_set_queue_enable(mdev, qid, ready);
327}
328
329static bool vp_vdpa_get_vq_ready(struct vdpa_device *vdpa, u16 qid)
330{
331	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
332
333	return vp_modern_get_queue_enable(mdev, qid);
334}
335
336static void vp_vdpa_set_vq_num(struct vdpa_device *vdpa, u16 qid,
337			       u32 num)
338{
339	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
340
341	vp_modern_set_queue_size(mdev, qid, num);
342}
343
344static u16 vp_vdpa_get_vq_size(struct vdpa_device *vdpa, u16 qid)
345{
346	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
347
348	return vp_modern_get_queue_size(mdev, qid);
349}
350
351static int vp_vdpa_set_vq_address(struct vdpa_device *vdpa, u16 qid,
352				  u64 desc_area, u64 driver_area,
353				  u64 device_area)
354{
355	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
356
357	vp_modern_queue_address(mdev, qid, desc_area,
358				driver_area, device_area);
359
360	return 0;
361}
362
363static void vp_vdpa_kick_vq(struct vdpa_device *vdpa, u16 qid)
364{
365	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
366
367	vp_iowrite16(qid, vp_vdpa->vring[qid].notify);
368}
369
370static void vp_vdpa_kick_vq_with_data(struct vdpa_device *vdpa, u32 data)
371{
372	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
373	u16 qid = data & 0xFFFF;
374
375	vp_iowrite32(data, vp_vdpa->vring[qid].notify);
376}
377
378static u32 vp_vdpa_get_generation(struct vdpa_device *vdpa)
379{
380	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
381
382	return vp_modern_generation(mdev);
383}
384
385static u32 vp_vdpa_get_device_id(struct vdpa_device *vdpa)
386{
387	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
388
389	return mdev->id.device;
390}
391
392static u32 vp_vdpa_get_vendor_id(struct vdpa_device *vdpa)
393{
394	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
395
396	return mdev->id.vendor;
397}
398
399static u32 vp_vdpa_get_vq_align(struct vdpa_device *vdpa)
400{
401	return PAGE_SIZE;
402}
403
404static size_t vp_vdpa_get_config_size(struct vdpa_device *vdpa)
405{
406	struct virtio_pci_modern_device *mdev = vdpa_to_mdev(vdpa);
407
408	return mdev->device_len;
409}
410
411static void vp_vdpa_get_config(struct vdpa_device *vdpa,
412			       unsigned int offset,
413			       void *buf, unsigned int len)
414{
415	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
416	struct virtio_pci_modern_device *mdev = vp_vdpa_to_mdev(vp_vdpa);
417	u8 old, new;
418	u8 *p;
419	int i;
420
421	do {
422		old = vp_ioread8(&mdev->common->config_generation);
423		p = buf;
424		for (i = 0; i < len; i++)
425			*p++ = vp_ioread8(mdev->device + offset + i);
426
427		new = vp_ioread8(&mdev->common->config_generation);
428	} while (old != new);
429}
430
431static void vp_vdpa_set_config(struct vdpa_device *vdpa,
432			       unsigned int offset, const void *buf,
433			       unsigned int len)
434{
435	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
436	struct virtio_pci_modern_device *mdev = vp_vdpa_to_mdev(vp_vdpa);
437	const u8 *p = buf;
438	int i;
439
440	for (i = 0; i < len; i++)
441		vp_iowrite8(*p++, mdev->device + offset + i);
442}
443
444static void vp_vdpa_set_config_cb(struct vdpa_device *vdpa,
445				  struct vdpa_callback *cb)
446{
447	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
448
449	vp_vdpa->config_cb = *cb;
450}
451
452static struct vdpa_notification_area
453vp_vdpa_get_vq_notification(struct vdpa_device *vdpa, u16 qid)
454{
455	struct vp_vdpa *vp_vdpa = vdpa_to_vp(vdpa);
456	struct virtio_pci_modern_device *mdev = vp_vdpa_to_mdev(vp_vdpa);
457	struct vdpa_notification_area notify;
458
459	notify.addr = vp_vdpa->vring[qid].notify_pa;
460	notify.size = mdev->notify_offset_multiplier;
461
462	return notify;
463}
464
465static const struct vdpa_config_ops vp_vdpa_ops = {
466	.get_device_features = vp_vdpa_get_device_features,
467	.set_driver_features = vp_vdpa_set_driver_features,
468	.get_driver_features = vp_vdpa_get_driver_features,
469	.get_status	= vp_vdpa_get_status,
470	.set_status	= vp_vdpa_set_status,
471	.reset		= vp_vdpa_reset,
472	.get_vq_num_max	= vp_vdpa_get_vq_num_max,
473	.get_vq_state	= vp_vdpa_get_vq_state,
474	.get_vq_notification = vp_vdpa_get_vq_notification,
475	.set_vq_state	= vp_vdpa_set_vq_state,
476	.set_vq_cb	= vp_vdpa_set_vq_cb,
477	.set_vq_ready	= vp_vdpa_set_vq_ready,
478	.get_vq_ready	= vp_vdpa_get_vq_ready,
479	.set_vq_num	= vp_vdpa_set_vq_num,
480	.get_vq_size	= vp_vdpa_get_vq_size,
481	.set_vq_address	= vp_vdpa_set_vq_address,
482	.kick_vq	= vp_vdpa_kick_vq,
483	.kick_vq_with_data      = vp_vdpa_kick_vq_with_data,
484	.get_generation	= vp_vdpa_get_generation,
485	.get_device_id	= vp_vdpa_get_device_id,
486	.get_vendor_id	= vp_vdpa_get_vendor_id,
487	.get_vq_align	= vp_vdpa_get_vq_align,
488	.get_config_size = vp_vdpa_get_config_size,
489	.get_config	= vp_vdpa_get_config,
490	.set_config	= vp_vdpa_set_config,
491	.set_config_cb  = vp_vdpa_set_config_cb,
492	.get_vq_irq	= vp_vdpa_get_vq_irq,
493};
494
495static void vp_vdpa_free_irq_vectors(void *data)
496{
497	pci_free_irq_vectors(data);
498}
499
500static int vp_vdpa_dev_add(struct vdpa_mgmt_dev *v_mdev, const char *name,
501			   const struct vdpa_dev_set_config *add_config)
502{
503	struct vp_vdpa_mgmtdev *vp_vdpa_mgtdev =
504		container_of(v_mdev, struct vp_vdpa_mgmtdev, mgtdev);
505
506	struct virtio_pci_modern_device *mdev = vp_vdpa_mgtdev->mdev;
507	struct pci_dev *pdev = mdev->pci_dev;
508	struct device *dev = &pdev->dev;
509	struct vp_vdpa *vp_vdpa = NULL;
510	u64 device_features;
511	int ret, i;
512
513	vp_vdpa = vdpa_alloc_device(struct vp_vdpa, vdpa,
514				    dev, &vp_vdpa_ops, NULL,
515				    1, 1, name, false);
516
517	if (IS_ERR(vp_vdpa)) {
518		dev_err(dev, "vp_vdpa: Failed to allocate vDPA structure\n");
519		return PTR_ERR(vp_vdpa);
520	}
521
522	vp_vdpa_mgtdev->vp_vdpa = vp_vdpa;
523
524	vp_vdpa->vdpa.vmap.dma_dev = &pdev->dev;
525	vp_vdpa->queues = vp_modern_get_num_queues(mdev);
526	vp_vdpa->mdev = mdev;
527
528	device_features = vp_modern_get_features(mdev);
529	if (add_config->mask & BIT_ULL(VDPA_ATTR_DEV_FEATURES)) {
530		if (add_config->device_features & ~device_features) {
531			ret = -EINVAL;
532			dev_err(&pdev->dev, "Try to provision features "
533				"that are not supported by the device: "
534				"device_features 0x%llx provisioned 0x%llx\n",
535				device_features, add_config->device_features);
536			goto err;
537		}
538		device_features = add_config->device_features;
539	}
540	vp_vdpa->device_features = device_features;
541
542	ret = devm_add_action_or_reset(dev, vp_vdpa_free_irq_vectors, pdev);
543	if (ret) {
544		dev_err(&pdev->dev,
545			"Failed for adding devres for freeing irq vectors\n");
546		goto err;
547	}
548
549	vp_vdpa->vring = devm_kcalloc(&pdev->dev, vp_vdpa->queues,
550				      sizeof(*vp_vdpa->vring),
551				      GFP_KERNEL);
552	if (!vp_vdpa->vring) {
553		ret = -ENOMEM;
554		dev_err(&pdev->dev, "Fail to allocate virtqueues\n");
555		goto err;
556	}
557
558	for (i = 0; i < vp_vdpa->queues; i++) {
559		vp_vdpa->vring[i].irq = VIRTIO_MSI_NO_VECTOR;
560		vp_vdpa->vring[i].notify =
561			vp_modern_map_vq_notify(mdev, i,
562						&vp_vdpa->vring[i].notify_pa);
563		if (!vp_vdpa->vring[i].notify) {
564			ret = -EINVAL;
565			dev_warn(&pdev->dev, "Fail to map vq notify %d\n", i);
566			goto err;
567		}
568	}
569	vp_vdpa->config_irq = VIRTIO_MSI_NO_VECTOR;
570
571	vp_vdpa->vdpa.mdev = &vp_vdpa_mgtdev->mgtdev;
572	ret = _vdpa_register_device(&vp_vdpa->vdpa, vp_vdpa->queues);
573	if (ret) {
574		dev_err(&pdev->dev, "Failed to register to vdpa bus\n");
575		goto err;
576	}
577
578	return 0;
579
580err:
581	put_device(&vp_vdpa->vdpa.dev);
582	return ret;
583}
584
585static void vp_vdpa_dev_del(struct vdpa_mgmt_dev *v_mdev,
586			    struct vdpa_device *dev)
587{
588	struct vp_vdpa_mgmtdev *vp_vdpa_mgtdev =
589		container_of(v_mdev, struct vp_vdpa_mgmtdev, mgtdev);
590
591	struct vp_vdpa *vp_vdpa = vp_vdpa_mgtdev->vp_vdpa;
592
593	_vdpa_unregister_device(&vp_vdpa->vdpa);
594	vp_vdpa_mgtdev->vp_vdpa = NULL;
595}
596
597static const struct vdpa_mgmtdev_ops vp_vdpa_mdev_ops = {
598	.dev_add = vp_vdpa_dev_add,
599	.dev_del = vp_vdpa_dev_del,
600};
601
602static int vp_vdpa_probe(struct pci_dev *pdev, const struct pci_device_id *id)
603{
604	struct vp_vdpa_mgmtdev *vp_vdpa_mgtdev = NULL;
605	struct vdpa_mgmt_dev *mgtdev;
606	struct device *dev = &pdev->dev;
607	struct virtio_pci_modern_device *mdev = NULL;
608	struct virtio_device_id *mdev_id = NULL;
609	int err;
610
611	vp_vdpa_mgtdev = kzalloc(sizeof(*vp_vdpa_mgtdev), GFP_KERNEL);
612	if (!vp_vdpa_mgtdev)
613		return -ENOMEM;
614
615	mgtdev = &vp_vdpa_mgtdev->mgtdev;
616	mgtdev->ops = &vp_vdpa_mdev_ops;
617	mgtdev->device = dev;
618
619	mdev = kzalloc(sizeof(struct virtio_pci_modern_device), GFP_KERNEL);
620	if (!mdev) {
621		err = -ENOMEM;
622		goto mdev_err;
623	}
624
625	/*
626	 * id_table should be a null terminated array, so allocate one additional
627	 * entry here, see vdpa_mgmtdev_get_classes().
628	 */
629	mdev_id = kcalloc(2, sizeof(struct virtio_device_id), GFP_KERNEL);
630	if (!mdev_id) {
631		err = -ENOMEM;
632		goto mdev_id_err;
633	}
634
635	vp_vdpa_mgtdev->mdev = mdev;
636	mdev->pci_dev = pdev;
637
638	err = pcim_enable_device(pdev);
639	if (err) {
640		goto probe_err;
641	}
642
643	err = vp_modern_probe(mdev);
644	if (err) {
645		dev_err(&pdev->dev, "Failed to probe modern PCI device\n");
646		goto probe_err;
647	}
648
649	mdev_id[0].device = mdev->id.device;
650	mdev_id[0].vendor = mdev->id.vendor;
651	mgtdev->id_table = mdev_id;
652	mgtdev->max_supported_vqs = vp_modern_get_num_queues(mdev);
653	mgtdev->supported_features = vp_modern_get_features(mdev);
654	mgtdev->config_attr_mask = (1 << VDPA_ATTR_DEV_FEATURES);
655	pci_set_master(pdev);
656	pci_set_drvdata(pdev, vp_vdpa_mgtdev);
657
658	err = vdpa_mgmtdev_register(mgtdev);
659	if (err) {
660		dev_err(&pdev->dev, "Failed to register vdpa mgmtdev device\n");
661		goto register_err;
662	}
663
664	return 0;
665
666register_err:
667	vp_modern_remove(vp_vdpa_mgtdev->mdev);
668probe_err:
669	kfree(mdev_id);
670mdev_id_err:
671	kfree(mdev);
672mdev_err:
673	kfree(vp_vdpa_mgtdev);
674	return err;
675}
676
677static void vp_vdpa_remove(struct pci_dev *pdev)
678{
679	struct vp_vdpa_mgmtdev *vp_vdpa_mgtdev = pci_get_drvdata(pdev);
680	struct virtio_pci_modern_device *mdev = NULL;
681
682	mdev = vp_vdpa_mgtdev->mdev;
683	vdpa_mgmtdev_unregister(&vp_vdpa_mgtdev->mgtdev);
684	vp_modern_remove(mdev);
685	kfree(vp_vdpa_mgtdev->mgtdev.id_table);
686	kfree(mdev);
687	kfree(vp_vdpa_mgtdev);
688}
689
690static struct pci_driver vp_vdpa_driver = {
691	.name		= "vp-vdpa",
692	.id_table	= NULL, /* only dynamic ids */
693	.probe		= vp_vdpa_probe,
694	.remove		= vp_vdpa_remove,
695};
696
697module_pci_driver(vp_vdpa_driver);
698
699MODULE_AUTHOR("Jason Wang <jasowang@redhat.com>");
700MODULE_DESCRIPTION("vp-vdpa");
701MODULE_LICENSE("GPL");
702MODULE_VERSION("1");