162306a36Sopenharmony_ci/* SPDX-License-Identifier: GPL-2.0 */
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (c) 2020, Red Hat Inc. All rights reserved.
462306a36Sopenharmony_ci */
562306a36Sopenharmony_ci
662306a36Sopenharmony_ci#ifndef _VDPA_SIM_H
762306a36Sopenharmony_ci#define _VDPA_SIM_H
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/iova.h>
1062306a36Sopenharmony_ci#include <linux/vringh.h>
1162306a36Sopenharmony_ci#include <linux/vdpa.h>
1262306a36Sopenharmony_ci#include <linux/virtio_byteorder.h>
1362306a36Sopenharmony_ci#include <linux/vhost_iotlb.h>
1462306a36Sopenharmony_ci#include <uapi/linux/virtio_config.h>
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ci#define VDPASIM_FEATURES	((1ULL << VIRTIO_F_ANY_LAYOUT) | \
1762306a36Sopenharmony_ci				 (1ULL << VIRTIO_F_VERSION_1)  | \
1862306a36Sopenharmony_ci				 (1ULL << VIRTIO_F_ACCESS_PLATFORM))
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_cistruct vdpasim;
2162306a36Sopenharmony_ci
2262306a36Sopenharmony_cistruct vdpasim_virtqueue {
2362306a36Sopenharmony_ci	struct vringh vring;
2462306a36Sopenharmony_ci	struct vringh_kiov in_iov;
2562306a36Sopenharmony_ci	struct vringh_kiov out_iov;
2662306a36Sopenharmony_ci	unsigned short head;
2762306a36Sopenharmony_ci	bool ready;
2862306a36Sopenharmony_ci	u64 desc_addr;
2962306a36Sopenharmony_ci	u64 device_addr;
3062306a36Sopenharmony_ci	u64 driver_addr;
3162306a36Sopenharmony_ci	u32 num;
3262306a36Sopenharmony_ci	void *private;
3362306a36Sopenharmony_ci	irqreturn_t (*cb)(void *data);
3462306a36Sopenharmony_ci};
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_cistruct vdpasim_dev_attr {
3762306a36Sopenharmony_ci	struct vdpa_mgmt_dev *mgmt_dev;
3862306a36Sopenharmony_ci	const char *name;
3962306a36Sopenharmony_ci	u64 supported_features;
4062306a36Sopenharmony_ci	size_t alloc_size;
4162306a36Sopenharmony_ci	size_t config_size;
4262306a36Sopenharmony_ci	int nvqs;
4362306a36Sopenharmony_ci	u32 id;
4462306a36Sopenharmony_ci	u32 ngroups;
4562306a36Sopenharmony_ci	u32 nas;
4662306a36Sopenharmony_ci
4762306a36Sopenharmony_ci	void (*work_fn)(struct vdpasim *vdpasim);
4862306a36Sopenharmony_ci	void (*get_config)(struct vdpasim *vdpasim, void *config);
4962306a36Sopenharmony_ci	void (*set_config)(struct vdpasim *vdpasim, const void *config);
5062306a36Sopenharmony_ci	int (*get_stats)(struct vdpasim *vdpasim, u16 idx,
5162306a36Sopenharmony_ci			 struct sk_buff *msg,
5262306a36Sopenharmony_ci			 struct netlink_ext_ack *extack);
5362306a36Sopenharmony_ci	void (*free)(struct vdpasim *vdpasim);
5462306a36Sopenharmony_ci};
5562306a36Sopenharmony_ci
5662306a36Sopenharmony_ci/* State of each vdpasim device */
5762306a36Sopenharmony_cistruct vdpasim {
5862306a36Sopenharmony_ci	struct vdpa_device vdpa;
5962306a36Sopenharmony_ci	struct vdpasim_virtqueue *vqs;
6062306a36Sopenharmony_ci	struct kthread_worker *worker;
6162306a36Sopenharmony_ci	struct kthread_work work;
6262306a36Sopenharmony_ci	struct mm_struct *mm_bound;
6362306a36Sopenharmony_ci	struct vdpasim_dev_attr dev_attr;
6462306a36Sopenharmony_ci	/* mutex to synchronize virtqueue state */
6562306a36Sopenharmony_ci	struct mutex mutex;
6662306a36Sopenharmony_ci	/* virtio config according to device type */
6762306a36Sopenharmony_ci	void *config;
6862306a36Sopenharmony_ci	struct vhost_iotlb *iommu;
6962306a36Sopenharmony_ci	bool *iommu_pt;
7062306a36Sopenharmony_ci	u32 status;
7162306a36Sopenharmony_ci	u32 generation;
7262306a36Sopenharmony_ci	u64 features;
7362306a36Sopenharmony_ci	u32 groups;
7462306a36Sopenharmony_ci	bool running;
7562306a36Sopenharmony_ci	bool pending_kick;
7662306a36Sopenharmony_ci	/* spinlock to synchronize iommu table */
7762306a36Sopenharmony_ci	spinlock_t iommu_lock;
7862306a36Sopenharmony_ci};
7962306a36Sopenharmony_ci
8062306a36Sopenharmony_cistruct vdpasim *vdpasim_create(struct vdpasim_dev_attr *attr,
8162306a36Sopenharmony_ci			       const struct vdpa_dev_set_config *config);
8262306a36Sopenharmony_civoid vdpasim_schedule_work(struct vdpasim *vdpasim);
8362306a36Sopenharmony_ci
8462306a36Sopenharmony_ci/* TODO: cross-endian support */
8562306a36Sopenharmony_cistatic inline bool vdpasim_is_little_endian(struct vdpasim *vdpasim)
8662306a36Sopenharmony_ci{
8762306a36Sopenharmony_ci	return virtio_legacy_is_little_endian() ||
8862306a36Sopenharmony_ci		(vdpasim->features & (1ULL << VIRTIO_F_VERSION_1));
8962306a36Sopenharmony_ci}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_cistatic inline u16 vdpasim16_to_cpu(struct vdpasim *vdpasim, __virtio16 val)
9262306a36Sopenharmony_ci{
9362306a36Sopenharmony_ci	return __virtio16_to_cpu(vdpasim_is_little_endian(vdpasim), val);
9462306a36Sopenharmony_ci}
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_cistatic inline __virtio16 cpu_to_vdpasim16(struct vdpasim *vdpasim, u16 val)
9762306a36Sopenharmony_ci{
9862306a36Sopenharmony_ci	return __cpu_to_virtio16(vdpasim_is_little_endian(vdpasim), val);
9962306a36Sopenharmony_ci}
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_cistatic inline u32 vdpasim32_to_cpu(struct vdpasim *vdpasim, __virtio32 val)
10262306a36Sopenharmony_ci{
10362306a36Sopenharmony_ci	return __virtio32_to_cpu(vdpasim_is_little_endian(vdpasim), val);
10462306a36Sopenharmony_ci}
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_cistatic inline __virtio32 cpu_to_vdpasim32(struct vdpasim *vdpasim, u32 val)
10762306a36Sopenharmony_ci{
10862306a36Sopenharmony_ci	return __cpu_to_virtio32(vdpasim_is_little_endian(vdpasim), val);
10962306a36Sopenharmony_ci}
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_cistatic inline u64 vdpasim64_to_cpu(struct vdpasim *vdpasim, __virtio64 val)
11262306a36Sopenharmony_ci{
11362306a36Sopenharmony_ci	return __virtio64_to_cpu(vdpasim_is_little_endian(vdpasim), val);
11462306a36Sopenharmony_ci}
11562306a36Sopenharmony_ci
11662306a36Sopenharmony_cistatic inline __virtio64 cpu_to_vdpasim64(struct vdpasim *vdpasim, u64 val)
11762306a36Sopenharmony_ci{
11862306a36Sopenharmony_ci	return __cpu_to_virtio64(vdpasim_is_little_endian(vdpasim), val);
11962306a36Sopenharmony_ci}
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_ci#endif
122