xref: /kvmtool/virtio/9p.c (revision ead43b0101e99bc6a9e14b290c71366aacdd1844)
11c7850f9SSasha Levin #include "kvm/virtio-9p.h"
21c7850f9SSasha Levin #include "kvm/virtio-pci-dev.h"
31c7850f9SSasha Levin #include "kvm/virtio.h"
41c7850f9SSasha Levin #include "kvm/ioport.h"
51c7850f9SSasha Levin #include "kvm/mutex.h"
61c7850f9SSasha Levin #include "kvm/util.h"
71c7850f9SSasha Levin #include "kvm/kvm.h"
81c7850f9SSasha Levin #include "kvm/pci.h"
91c7850f9SSasha Levin #include "kvm/threadpool.h"
101c7850f9SSasha Levin #include "kvm/irq.h"
1160eb42d5SSasha Levin #include "kvm/ioeventfd.h"
121c7850f9SSasha Levin 
131c7850f9SSasha Levin #include <fcntl.h>
141c7850f9SSasha Levin #include <sys/types.h>
151c7850f9SSasha Levin #include <sys/stat.h>
161c7850f9SSasha Levin #include <pthread.h>
171c7850f9SSasha Levin #include <dirent.h>
181c7850f9SSasha Levin 
192daa28d4SAneesh Kumar K.V #include <linux/virtio_ring.h>
202daa28d4SAneesh Kumar K.V #include <linux/virtio_9p.h>
212daa28d4SAneesh Kumar K.V #include <net/9p/9p.h>
222daa28d4SAneesh Kumar K.V 
231c7850f9SSasha Levin #define NUM_VIRT_QUEUES		1
24612038c5SAneesh Kumar K.V #define VIRTQUEUE_NUM		128
25b4422bf3SAneesh Kumar K.V #define	VIRTIO_P9_DEFAULT_TAG	"kvm_9p"
261c7850f9SSasha Levin #define VIRTIO_P9_HDR_LEN	(sizeof(u32)+sizeof(u8)+sizeof(u16))
271c7850f9SSasha Levin #define VIRTIO_P9_MAX_FID	128
281c7850f9SSasha Levin #define VIRTIO_P9_VERSION	"9P2000"
29b4422bf3SAneesh Kumar K.V #define MAX_TAG_LEN		32
30b4422bf3SAneesh Kumar K.V 
311c7850f9SSasha Levin 
321c7850f9SSasha Levin struct p9_msg {
331c7850f9SSasha Levin 	u32			size;
341c7850f9SSasha Levin 	u8			cmd;
351c7850f9SSasha Levin 	u16			tag;
361c7850f9SSasha Levin 	u8			msg[0];
371c7850f9SSasha Levin } __attribute__((packed));
381c7850f9SSasha Levin 
391c7850f9SSasha Levin struct p9_fid {
401c7850f9SSasha Levin 	u32			fid;
411c7850f9SSasha Levin 	u8			is_dir;
421c7850f9SSasha Levin 	char			abs_path[PATH_MAX];
431c7850f9SSasha Levin 	char			*path;
441c7850f9SSasha Levin 	DIR			*dir;
451c7850f9SSasha Levin 	int			fd;
461c7850f9SSasha Levin };
471c7850f9SSasha Levin 
48b4422bf3SAneesh Kumar K.V struct p9_dev_job {
49b4422bf3SAneesh Kumar K.V 	struct virt_queue		*vq;
50b4422bf3SAneesh Kumar K.V 	struct p9_dev			*p9dev;
51b4422bf3SAneesh Kumar K.V 	void				*job_id;
521c7850f9SSasha Levin };
531c7850f9SSasha Levin 
541c7850f9SSasha Levin struct p9_dev {
551c7850f9SSasha Levin 	u8			status;
561c7850f9SSasha Levin 	u8			isr;
571c7850f9SSasha Levin 	u16			config_vector;
581c7850f9SSasha Levin 	u32			features;
591c7850f9SSasha Levin 	struct virtio_9p_config	*config;
60f884f920SSasha Levin 	u16			base_addr;
611c7850f9SSasha Levin 
621c7850f9SSasha Levin 	/* virtio queue */
631c7850f9SSasha Levin 	u16			queue_selector;
641c7850f9SSasha Levin 	struct virt_queue	vqs[NUM_VIRT_QUEUES];
65b4422bf3SAneesh Kumar K.V 	struct p9_dev_job	jobs[NUM_VIRT_QUEUES];
661c7850f9SSasha Levin 	struct p9_fid		fids[VIRTIO_P9_MAX_FID];
671c7850f9SSasha Levin 	char			root_dir[PATH_MAX];
68b4422bf3SAneesh Kumar K.V 	struct pci_device_header pci_hdr;
691c7850f9SSasha Levin };
701c7850f9SSasha Levin 
71af045e53SAneesh Kumar K.V struct p9_pdu {
72af045e53SAneesh Kumar K.V 	u32 queue_head;
73af045e53SAneesh Kumar K.V 	int offset;
74af045e53SAneesh Kumar K.V 	u16 out_iov_cnt;
75af045e53SAneesh Kumar K.V 	u16 in_iov_cnt;
76af045e53SAneesh Kumar K.V 	struct iovec in_iov[VIRTQUEUE_NUM];
77af045e53SAneesh Kumar K.V 	struct iovec out_iov[VIRTQUEUE_NUM];
78af045e53SAneesh Kumar K.V };
79af045e53SAneesh Kumar K.V 
801c7850f9SSasha Levin /* Warning: Immediately use value returned from this function */
81b4422bf3SAneesh Kumar K.V static const char *rel_to_abs(struct p9_dev *p9dev,
82b4422bf3SAneesh Kumar K.V 			      const char *path, char *abs_path)
831c7850f9SSasha Levin {
84b4422bf3SAneesh Kumar K.V 	sprintf(abs_path, "%s/%s", p9dev->root_dir, path);
851c7850f9SSasha Levin 
861c7850f9SSasha Levin 	return abs_path;
871c7850f9SSasha Levin }
881c7850f9SSasha Levin 
89b4422bf3SAneesh Kumar K.V static bool virtio_p9_dev_in(struct p9_dev *p9dev, void *data,
90b4422bf3SAneesh Kumar K.V 			     unsigned long offset,
91b4422bf3SAneesh Kumar K.V 			     int size, u32 count)
921c7850f9SSasha Levin {
93b4422bf3SAneesh Kumar K.V 	u8 *config_space = (u8 *) p9dev->config;
941c7850f9SSasha Levin 
951c7850f9SSasha Levin 	if (size != 1 || count != 1)
961c7850f9SSasha Levin 		return false;
971c7850f9SSasha Levin 
981c7850f9SSasha Levin 	ioport__write8(data, config_space[offset - VIRTIO_MSI_CONFIG_VECTOR]);
991c7850f9SSasha Levin 
1001c7850f9SSasha Levin 	return true;
1011c7850f9SSasha Levin }
1021c7850f9SSasha Levin 
103b4422bf3SAneesh Kumar K.V static bool virtio_p9_pci_io_in(struct ioport *ioport, struct kvm *kvm,
104b4422bf3SAneesh Kumar K.V 				u16 port, void *data, int size, u32 count)
1051c7850f9SSasha Levin {
1061c7850f9SSasha Levin 	bool ret = true;
107b4422bf3SAneesh Kumar K.V 	unsigned long offset;
108b4422bf3SAneesh Kumar K.V 	struct p9_dev *p9dev = ioport->priv;
1091c7850f9SSasha Levin 
110b4422bf3SAneesh Kumar K.V 
111b4422bf3SAneesh Kumar K.V 	offset = port - p9dev->base_addr;
1121c7850f9SSasha Levin 
1131c7850f9SSasha Levin 	switch (offset) {
1141c7850f9SSasha Levin 	case VIRTIO_PCI_HOST_FEATURES:
115b4422bf3SAneesh Kumar K.V 		ioport__write32(data, p9dev->features);
1161c7850f9SSasha Levin 		ret = true;
1171c7850f9SSasha Levin 		break;
1181c7850f9SSasha Levin 	case VIRTIO_PCI_GUEST_FEATURES:
1191c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_SEL:
1201c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_NOTIFY:
1211c7850f9SSasha Levin 		ret = false;
1221c7850f9SSasha Levin 		break;
1231c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_PFN:
124b4422bf3SAneesh Kumar K.V 		ioport__write32(data, p9dev->vqs[p9dev->queue_selector].pfn);
1251c7850f9SSasha Levin 		break;
1261c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_NUM:
127612038c5SAneesh Kumar K.V 		ioport__write16(data, VIRTQUEUE_NUM);
1281c7850f9SSasha Levin 		break;
1291c7850f9SSasha Levin 	case VIRTIO_PCI_STATUS:
130b4422bf3SAneesh Kumar K.V 		ioport__write8(data, p9dev->status);
1311c7850f9SSasha Levin 		break;
1321c7850f9SSasha Levin 	case VIRTIO_PCI_ISR:
133b4422bf3SAneesh Kumar K.V 		ioport__write8(data, p9dev->isr);
134b4422bf3SAneesh Kumar K.V 		kvm__irq_line(kvm, p9dev->pci_hdr.irq_line, VIRTIO_IRQ_LOW);
135b4422bf3SAneesh Kumar K.V 		p9dev->isr = VIRTIO_IRQ_LOW;
1361c7850f9SSasha Levin 		break;
1371c7850f9SSasha Levin 	default:
138b4422bf3SAneesh Kumar K.V 		ret = virtio_p9_dev_in(p9dev, data, offset, size, count);
1391c7850f9SSasha Levin 		break;
1401c7850f9SSasha Levin 	};
1411c7850f9SSasha Levin 
1421c7850f9SSasha Levin 	return ret;
1431c7850f9SSasha Levin }
1441c7850f9SSasha Levin 
1451c7850f9SSasha Levin static int omode2uflags(u8 mode)
1461c7850f9SSasha Levin {
1471c7850f9SSasha Levin 	int ret = 0;
1481c7850f9SSasha Levin 
1491c7850f9SSasha Levin 	/* Basic open modes are same as uflags */
1501c7850f9SSasha Levin 	ret = mode & 3;
1511c7850f9SSasha Levin 
1521c7850f9SSasha Levin 	/* Everything else is different */
1531c7850f9SSasha Levin 	if (mode & P9_OTRUNC)
1541c7850f9SSasha Levin 		ret |= O_TRUNC;
1551c7850f9SSasha Levin 
1561c7850f9SSasha Levin 	if (mode & P9_OAPPEND)
1571c7850f9SSasha Levin 		ret |= O_APPEND;
1581c7850f9SSasha Levin 
1591c7850f9SSasha Levin 	if (mode & P9_OEXCL)
1601c7850f9SSasha Levin 		ret |= O_EXCL;
1611c7850f9SSasha Levin 
1621c7850f9SSasha Levin 	return ret;
1631c7850f9SSasha Levin }
1641c7850f9SSasha Levin 
1651c7850f9SSasha Levin static void st2qid(struct stat *st, struct p9_qid *qid)
1661c7850f9SSasha Levin {
1671c7850f9SSasha Levin 	*qid = (struct p9_qid) {
1681c7850f9SSasha Levin 		.path		= st->st_ino,
1691c7850f9SSasha Levin 		.version	= st->st_mtime,
1701c7850f9SSasha Levin 	};
1711c7850f9SSasha Levin 
1721c7850f9SSasha Levin 	if (S_ISDIR(st->st_mode))
1731c7850f9SSasha Levin 		qid->type	|= P9_QTDIR;
1741c7850f9SSasha Levin }
1751c7850f9SSasha Levin 
176b4422bf3SAneesh Kumar K.V static void close_fid(struct p9_dev *p9dev, u32 fid)
1771c7850f9SSasha Levin {
178b4422bf3SAneesh Kumar K.V 	if (p9dev->fids[fid].fd > 0) {
179b4422bf3SAneesh Kumar K.V 		close(p9dev->fids[fid].fd);
180b4422bf3SAneesh Kumar K.V 		p9dev->fids[fid].fd = -1;
1811c7850f9SSasha Levin 	}
182b4422bf3SAneesh Kumar K.V 	if (p9dev->fids[fid].dir) {
183b4422bf3SAneesh Kumar K.V 		closedir(p9dev->fids[fid].dir);
184b4422bf3SAneesh Kumar K.V 		p9dev->fids[fid].dir = NULL;
1851c7850f9SSasha Levin 	}
1861c7850f9SSasha Levin }
1871c7850f9SSasha Levin 
1881c7850f9SSasha Levin static void set_p9msg_hdr(struct p9_msg *msg, u32 size, u8 cmd, u16 tag)
1891c7850f9SSasha Levin {
1901c7850f9SSasha Levin 	*msg = (struct p9_msg) {
1911c7850f9SSasha Levin 		.size	= size,
1921c7850f9SSasha Levin 		.tag	= tag,
1931c7850f9SSasha Levin 		.cmd	= cmd,
1941c7850f9SSasha Levin 	};
1951c7850f9SSasha Levin }
1961c7850f9SSasha Levin 
1976b163a87SAneesh Kumar K.V static u16 virtio_p9_update_iov_cnt(struct iovec iov[], u32 count, int iov_cnt)
1986b163a87SAneesh Kumar K.V {
1996b163a87SAneesh Kumar K.V 	int i;
2006b163a87SAneesh Kumar K.V 	u32 total = 0;
2016b163a87SAneesh Kumar K.V 	for (i = 0; (i < iov_cnt) && (total < count); i++) {
2026b163a87SAneesh Kumar K.V 		if (total + iov[i].iov_len > count) {
2036b163a87SAneesh Kumar K.V 			/* we don't need this iov fully */
2046b163a87SAneesh Kumar K.V 			iov[i].iov_len -= ((total + iov[i].iov_len) - count);
2056b163a87SAneesh Kumar K.V 			i++;
2066b163a87SAneesh Kumar K.V 			break;
2076b163a87SAneesh Kumar K.V 		}
2086b163a87SAneesh Kumar K.V 		total += iov[i].iov_len;
2096b163a87SAneesh Kumar K.V 	}
2106b163a87SAneesh Kumar K.V 	return i;
2116b163a87SAneesh Kumar K.V }
2126b163a87SAneesh Kumar K.V 
213eee1ba8eSAneesh Kumar K.V static void virtio_p9_error_reply(struct p9_dev *p9dev,
214eee1ba8eSAneesh Kumar K.V 				  struct p9_pdu *pdu, int err, u32 *outlen)
215eee1ba8eSAneesh Kumar K.V {
216eee1ba8eSAneesh Kumar K.V 	char *err_str;
217eee1ba8eSAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
218eee1ba8eSAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
219eee1ba8eSAneesh Kumar K.V 	struct p9_rerror *rerror  = (struct p9_rerror *)inmsg->msg;
220eee1ba8eSAneesh Kumar K.V 
221eee1ba8eSAneesh Kumar K.V 	err_str = strerror(err);
222eee1ba8eSAneesh Kumar K.V 	rerror->error.len = strlen(err_str);
223eee1ba8eSAneesh Kumar K.V 	memcpy(&rerror->error.str, err_str, rerror->error.len);
224eee1ba8eSAneesh Kumar K.V 
225eee1ba8eSAneesh Kumar K.V 	*outlen = VIRTIO_P9_HDR_LEN + rerror->error.len + sizeof(u16);
226eee1ba8eSAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RERROR, outmsg->tag);
227eee1ba8eSAneesh Kumar K.V }
228eee1ba8eSAneesh Kumar K.V 
229*ead43b01SAneesh Kumar K.V static void virtio_p9_version(struct p9_dev *p9dev,
230af045e53SAneesh Kumar K.V 			      struct p9_pdu *pdu, u32 *outlen)
2311c7850f9SSasha Levin {
232af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
233af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
234af045e53SAneesh Kumar K.V 	struct p9_rversion *rversion = (struct p9_rversion *)inmsg->msg;
2351c7850f9SSasha Levin 
2361c7850f9SSasha Levin 	rversion->msize		= 4096;
2371c7850f9SSasha Levin 	rversion->version.len	= strlen(VIRTIO_P9_VERSION);
2381c7850f9SSasha Levin 	memcpy(&rversion->version.str, VIRTIO_P9_VERSION, rversion->version.len);
2391c7850f9SSasha Levin 
240b4422bf3SAneesh Kumar K.V 	*outlen = VIRTIO_P9_HDR_LEN +
241b4422bf3SAneesh Kumar K.V 		rversion->version.len + sizeof(u16) + sizeof(u32);
242af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RVERSION, outmsg->tag);
2431c7850f9SSasha Levin 
244*ead43b01SAneesh Kumar K.V 	return;
2451c7850f9SSasha Levin }
2461c7850f9SSasha Levin 
247*ead43b01SAneesh Kumar K.V static void virtio_p9_clunk(struct p9_dev *p9dev,
248af045e53SAneesh Kumar K.V 			    struct p9_pdu *pdu, u32 *outlen)
2491c7850f9SSasha Levin {
250af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
251af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
252af045e53SAneesh Kumar K.V 	struct p9_tclunk *tclunk = (struct p9_tclunk *)outmsg->msg;
2531c7850f9SSasha Levin 
254b4422bf3SAneesh Kumar K.V 	close_fid(p9dev, tclunk->fid);
2551c7850f9SSasha Levin 
2561c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN;
257af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RCLUNK, outmsg->tag);
2581c7850f9SSasha Levin 
259*ead43b01SAneesh Kumar K.V 	return;
2601c7850f9SSasha Levin }
2611c7850f9SSasha Levin 
262*ead43b01SAneesh Kumar K.V static void virtio_p9_open(struct p9_dev *p9dev,
263af045e53SAneesh Kumar K.V 			   struct p9_pdu *pdu, u32 *outlen)
2641c7850f9SSasha Levin {
2651c7850f9SSasha Levin 	struct stat st;
266af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
267af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
268af045e53SAneesh Kumar K.V 	struct p9_topen *topen	= (struct p9_topen *)outmsg->msg;
269af045e53SAneesh Kumar K.V 	struct p9_ropen *ropen	= (struct p9_ropen *)inmsg->msg;
270af045e53SAneesh Kumar K.V 	struct p9_fid *new_fid	= &p9dev->fids[topen->fid];
2711c7850f9SSasha Levin 
27230204a77SAneesh Kumar K.V 	if (lstat(new_fid->abs_path, &st) < 0)
273eee1ba8eSAneesh Kumar K.V 		goto err_out;
2741c7850f9SSasha Levin 
2751c7850f9SSasha Levin 	st2qid(&st, &ropen->qid);
2761c7850f9SSasha Levin 	ropen->iounit = 0;
2771c7850f9SSasha Levin 
278eee1ba8eSAneesh Kumar K.V 	if (new_fid->is_dir) {
2791c7850f9SSasha Levin 		new_fid->dir = opendir(new_fid->abs_path);
280eee1ba8eSAneesh Kumar K.V 		if (!new_fid->dir)
281eee1ba8eSAneesh Kumar K.V 			goto err_out;
282eee1ba8eSAneesh Kumar K.V 	} else {
283eee1ba8eSAneesh Kumar K.V 		new_fid->fd  = open(new_fid->abs_path,
284eee1ba8eSAneesh Kumar K.V 				   omode2uflags(topen->mode) | O_NOFOLLOW);
285eee1ba8eSAneesh Kumar K.V 		if (new_fid->fd < 0)
286eee1ba8eSAneesh Kumar K.V 			goto err_out;
287eee1ba8eSAneesh Kumar K.V 	}
2881c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(*ropen);
289af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_ROPEN, outmsg->tag);
290*ead43b01SAneesh Kumar K.V 	return;
291eee1ba8eSAneesh Kumar K.V err_out:
292eee1ba8eSAneesh Kumar K.V 	virtio_p9_error_reply(p9dev, pdu, errno, outlen);
293*ead43b01SAneesh Kumar K.V 	return;
2941c7850f9SSasha Levin }
2951c7850f9SSasha Levin 
296*ead43b01SAneesh Kumar K.V static void virtio_p9_create(struct p9_dev *p9dev,
297af045e53SAneesh Kumar K.V 			     struct p9_pdu *pdu, u32 *outlen)
2981c7850f9SSasha Levin {
2991c7850f9SSasha Levin 	u8 mode;
3001c7850f9SSasha Levin 	u32 perm;
301af045e53SAneesh Kumar K.V 	struct stat st;
302af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
303af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
304af045e53SAneesh Kumar K.V 	struct p9_tcreate *tcreate	= (struct p9_tcreate *)outmsg->msg;
305af045e53SAneesh Kumar K.V 	struct p9_rcreate *rcreate	= (struct p9_rcreate *)inmsg->msg;
306af045e53SAneesh Kumar K.V 	struct p9_fid *fid		= &p9dev->fids[tcreate->fid];
307af045e53SAneesh Kumar K.V 
3081c7850f9SSasha Levin 
3091c7850f9SSasha Levin 	rcreate->iounit = 0;
3101c7850f9SSasha Levin 
3111c7850f9SSasha Levin 	/* Get last byte of the variable length struct */
312af045e53SAneesh Kumar K.V 	mode = *((u8 *)outmsg + outmsg->size - 1);
313af045e53SAneesh Kumar K.V 	perm = *(u32 *)((u8 *)outmsg + outmsg->size - 5);
3141c7850f9SSasha Levin 
3151c7850f9SSasha Levin 	sprintf(fid->path, "%s/%.*s", fid->path, tcreate->name.len, (char *)&tcreate->name.str);
3161c7850f9SSasha Levin 
317b4422bf3SAneesh Kumar K.V 	close_fid(p9dev, tcreate->fid);
3181c7850f9SSasha Levin 
3191c7850f9SSasha Levin 	if (perm & P9_DMDIR) {
3201c7850f9SSasha Levin 		mkdir(fid->abs_path, perm & 0xFFFF);
3211c7850f9SSasha Levin 		fid->dir = opendir(fid->abs_path);
3221c7850f9SSasha Levin 		fid->is_dir = 1;
3231c7850f9SSasha Levin 	} else {
3241c7850f9SSasha Levin 		fid->fd = open(fid->abs_path, omode2uflags(mode) | O_CREAT, 0777);
3251c7850f9SSasha Levin 	}
3261c7850f9SSasha Levin 
32730204a77SAneesh Kumar K.V 	if (lstat(fid->abs_path, &st) < 0)
328*ead43b01SAneesh Kumar K.V 		return;
3291c7850f9SSasha Levin 
3301c7850f9SSasha Levin 	st2qid(&st, &rcreate->qid);
3311c7850f9SSasha Levin 
3321c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(*rcreate);
333af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RCREATE, outmsg->tag);
3341c7850f9SSasha Levin 
335*ead43b01SAneesh Kumar K.V 	return;
3361c7850f9SSasha Levin }
3371c7850f9SSasha Levin 
338*ead43b01SAneesh Kumar K.V static void virtio_p9_walk(struct p9_dev *p9dev,
339af045e53SAneesh Kumar K.V 			   struct p9_pdu *pdu, u32 *outlen)
3401c7850f9SSasha Levin {
341af045e53SAneesh Kumar K.V 	u8 i;
342af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
343af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
344af045e53SAneesh Kumar K.V 	struct p9_twalk *twalk	= (struct p9_twalk *)outmsg->msg;
345af045e53SAneesh Kumar K.V 	struct p9_rwalk *rwalk	= (struct p9_rwalk *)inmsg->msg;
3461c7850f9SSasha Levin 	struct p9_str *str	= twalk->wnames;
347b4422bf3SAneesh Kumar K.V 	struct p9_fid *new_fid	= &p9dev->fids[twalk->newfid];
348af045e53SAneesh Kumar K.V 
3491c7850f9SSasha Levin 
3501c7850f9SSasha Levin 	rwalk->nwqid = 0;
3511c7850f9SSasha Levin 	if (twalk->nwname) {
352b4422bf3SAneesh Kumar K.V 		struct p9_fid *fid = &p9dev->fids[twalk->fid];
3531c7850f9SSasha Levin 
3541c7850f9SSasha Levin 		for (i = 0; i < twalk->nwname; i++) {
3551c7850f9SSasha Levin 			char tmp[PATH_MAX] = {0};
3561c7850f9SSasha Levin 			char full_path[PATH_MAX];
3571c7850f9SSasha Levin 			struct stat st;
3581c7850f9SSasha Levin 
3591c7850f9SSasha Levin 			/* Format the new path we're 'walk'ing into */
360af045e53SAneesh Kumar K.V 			sprintf(tmp, "%s/%.*s", fid->path,
361af045e53SAneesh Kumar K.V 				str->len, (char *)&str->str);
3621c7850f9SSasha Levin 
36330204a77SAneesh Kumar K.V 			if (lstat(rel_to_abs(p9dev, tmp, full_path), &st) < 0)
3641c7850f9SSasha Levin 				break;
3651c7850f9SSasha Levin 
3661c7850f9SSasha Levin 			st2qid(&st, &rwalk->wqids[i]);
3671c7850f9SSasha Levin 			new_fid->is_dir = S_ISDIR(st.st_mode);
3681c7850f9SSasha Levin 			strcpy(new_fid->path, tmp);
3691c7850f9SSasha Levin 			new_fid->fid = twalk->newfid;
3701c7850f9SSasha Levin 			rwalk->nwqid++;
3711c7850f9SSasha Levin 		}
3721c7850f9SSasha Levin 	} else {
373b4422bf3SAneesh Kumar K.V 		new_fid->is_dir = p9dev->fids[twalk->fid].is_dir;
374b4422bf3SAneesh Kumar K.V 		strcpy(new_fid->path, p9dev->fids[twalk->fid].path);
3751c7850f9SSasha Levin 		new_fid->fid	= twalk->newfid;
3761c7850f9SSasha Levin 	}
3771c7850f9SSasha Levin 
378af045e53SAneesh Kumar K.V 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(u16) +
379af045e53SAneesh Kumar K.V 		sizeof(struct p9_qid)*rwalk->nwqid;
380af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RWALK, outmsg->tag);
3811c7850f9SSasha Levin 
382*ead43b01SAneesh Kumar K.V 	return;
3831c7850f9SSasha Levin }
3841c7850f9SSasha Levin 
385*ead43b01SAneesh Kumar K.V static void virtio_p9_attach(struct p9_dev *p9dev,
386af045e53SAneesh Kumar K.V 			     struct p9_pdu *pdu, u32 *outlen)
3871c7850f9SSasha Levin {
388af045e53SAneesh Kumar K.V 	u32 i;
3891c7850f9SSasha Levin 	struct stat st;
3901c7850f9SSasha Levin 	struct p9_fid *fid;
391af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
392af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
393af045e53SAneesh Kumar K.V 	struct p9_rattach *rattach = (struct p9_rattach *)inmsg->msg;
394af045e53SAneesh Kumar K.V 	struct p9_tattach *tattach = (struct p9_tattach *)outmsg->msg;
3951c7850f9SSasha Levin 
3961c7850f9SSasha Levin 	/* Reset everything */
3971c7850f9SSasha Levin 	for (i = 0; i < VIRTIO_P9_MAX_FID; i++)
398b4422bf3SAneesh Kumar K.V 		p9dev->fids[i].fid = P9_NOFID;
3991c7850f9SSasha Levin 
40030204a77SAneesh Kumar K.V 	if (lstat(p9dev->root_dir, &st) < 0)
401*ead43b01SAneesh Kumar K.V 		return;
4021c7850f9SSasha Levin 
4031c7850f9SSasha Levin 	st2qid(&st, &rattach->qid);
4041c7850f9SSasha Levin 
405b4422bf3SAneesh Kumar K.V 	fid = &p9dev->fids[tattach->fid];
4061c7850f9SSasha Levin 	fid->fid = tattach->fid;
4071c7850f9SSasha Levin 	fid->is_dir = 1;
4081c7850f9SSasha Levin 	strcpy(fid->path, "/");
4091c7850f9SSasha Levin 
4101c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(*rattach);
411af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RATTACH, outmsg->tag);
4121c7850f9SSasha Levin 
413*ead43b01SAneesh Kumar K.V 	return;
4141c7850f9SSasha Levin }
4151c7850f9SSasha Levin 
416b4422bf3SAneesh Kumar K.V static u32 virtio_p9_fill_stat(struct p9_dev *p9dev, const char *name,
417b4422bf3SAneesh Kumar K.V 			       struct stat *st, struct p9_rstat *rstat)
4181c7850f9SSasha Levin {
4191c7850f9SSasha Levin 	struct p9_str *str;
4201c7850f9SSasha Levin 
4211c7850f9SSasha Levin 	rstat->stat.type = 0;
4221c7850f9SSasha Levin 	rstat->stat.dev = 0;
4231c7850f9SSasha Levin 	st2qid(st, &rstat->stat.qid);
4241c7850f9SSasha Levin 	rstat->stat.mode = st->st_mode;
4251c7850f9SSasha Levin 	rstat->stat.length = st->st_size;
4261c7850f9SSasha Levin 	if (S_ISDIR(st->st_mode)) {
4271c7850f9SSasha Levin 		rstat->stat.length = 0;
4281c7850f9SSasha Levin 		rstat->stat.mode |= P9_DMDIR;
4291c7850f9SSasha Levin 	}
4301c7850f9SSasha Levin 
4311c7850f9SSasha Levin 	rstat->stat.atime = st->st_atime;
4321c7850f9SSasha Levin 	rstat->stat.mtime = st->st_mtime;
4331c7850f9SSasha Levin 
4341c7850f9SSasha Levin 	str = (struct p9_str *)&rstat->stat.name;
4351c7850f9SSasha Levin 	str->len = strlen(name);
4361c7850f9SSasha Levin 	memcpy(&str->str, name, str->len);
4371c7850f9SSasha Levin 	str = (void *)str + str->len + sizeof(u16);
4381c7850f9SSasha Levin 
4391c7850f9SSasha Levin 	/* TODO: Pass usernames to the client */
4401c7850f9SSasha Levin 	str->len = 0;
4411c7850f9SSasha Levin 	str = (void *)str + sizeof(u16);
4421c7850f9SSasha Levin 	str->len = 0;
4431c7850f9SSasha Levin 	str = (void *)str + sizeof(u16);
4441c7850f9SSasha Levin 	str->len = 0;
4451c7850f9SSasha Levin 	str = (void *)str + sizeof(u16);
4461c7850f9SSasha Levin 
447b4422bf3SAneesh Kumar K.V 	/*
448b4422bf3SAneesh Kumar K.V 	 * We subtract a u16 here because rstat->size
449b4422bf3SAneesh Kumar K.V 	 * doesn't include rstat->size itself
450b4422bf3SAneesh Kumar K.V 	 */
4511c7850f9SSasha Levin 	rstat->stat.size = (void *)str - (void *)&rstat->stat - sizeof(u16);
4521c7850f9SSasha Levin 
4531c7850f9SSasha Levin 	return rstat->stat.size + sizeof(u16);
4541c7850f9SSasha Levin }
4551c7850f9SSasha Levin 
456*ead43b01SAneesh Kumar K.V static void virtio_p9_read(struct p9_dev *p9dev,
457af045e53SAneesh Kumar K.V 			   struct p9_pdu *pdu, u32 *outlen)
4581c7850f9SSasha Levin {
459af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
460af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
461af045e53SAneesh Kumar K.V 	struct p9_tread *tread	= (struct p9_tread *)outmsg->msg;
462af045e53SAneesh Kumar K.V 	struct p9_rread *rread	= (struct p9_rread *)inmsg->msg;
463af045e53SAneesh Kumar K.V 	struct p9_rstat *rstat	= (struct p9_rstat *)pdu->in_iov[1].iov_base;
464b4422bf3SAneesh Kumar K.V 	struct p9_fid *fid	= &p9dev->fids[tread->fid];
4651c7850f9SSasha Levin 	struct stat st;
4661c7850f9SSasha Levin 
4671c7850f9SSasha Levin 	rread->count = 0;
4681c7850f9SSasha Levin 
4691c7850f9SSasha Levin 	if (fid->is_dir) {
4701c7850f9SSasha Levin 		/* If reading a dir, fill the buffer with p9_stat entries */
4711c7850f9SSasha Levin 		struct dirent *cur = readdir(fid->dir);
4721c7850f9SSasha Levin 		char full_path[PATH_MAX];
4731c7850f9SSasha Levin 
4741c7850f9SSasha Levin 		while (cur) {
4751c7850f9SSasha Levin 			u32 read;
4761c7850f9SSasha Levin 
47730204a77SAneesh Kumar K.V 			lstat(rel_to_abs(p9dev, cur->d_name, full_path), &st);
478b4422bf3SAneesh Kumar K.V 			read = virtio_p9_fill_stat(p9dev, cur->d_name,
479b4422bf3SAneesh Kumar K.V 						   &st, rstat);
4801c7850f9SSasha Levin 			rread->count += read;
4811c7850f9SSasha Levin 			rstat = (void *)rstat + read;
4821c7850f9SSasha Levin 			cur = readdir(fid->dir);
4831c7850f9SSasha Levin 		}
4841c7850f9SSasha Levin 	} else {
485af045e53SAneesh Kumar K.V 		pdu->in_iov[0].iov_base += VIRTIO_P9_HDR_LEN + sizeof(u32);
486af045e53SAneesh Kumar K.V 		pdu->in_iov[0].iov_len -= VIRTIO_P9_HDR_LEN + sizeof(u32);
4876b163a87SAneesh Kumar K.V 		pdu->in_iov_cnt = virtio_p9_update_iov_cnt(pdu->in_iov,
4886b163a87SAneesh Kumar K.V 							    tread->count,
4896b163a87SAneesh Kumar K.V 							    pdu->in_iov_cnt);
490af045e53SAneesh Kumar K.V 		rread->count = preadv(fid->fd, pdu->in_iov,
491af045e53SAneesh Kumar K.V 				      pdu->in_iov_cnt, tread->offset);
4921c7850f9SSasha Levin 		if (rread->count > tread->count)
4931c7850f9SSasha Levin 			rread->count = tread->count;
4941c7850f9SSasha Levin 	}
4951c7850f9SSasha Levin 
4961c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(u32) + rread->count;
497af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RREAD, outmsg->tag);
4981c7850f9SSasha Levin 
499*ead43b01SAneesh Kumar K.V 	return;
5001c7850f9SSasha Levin }
5011c7850f9SSasha Levin 
502*ead43b01SAneesh Kumar K.V static void virtio_p9_stat(struct p9_dev *p9dev,
503af045e53SAneesh Kumar K.V 			   struct p9_pdu *pdu, u32 *outlen)
5041c7850f9SSasha Levin {
5051c7850f9SSasha Levin 	u32 ret;
506af045e53SAneesh Kumar K.V 	struct stat st;
507af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
508af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
509af045e53SAneesh Kumar K.V 	struct p9_tstat *tstat = (struct p9_tstat *)outmsg->msg;
510af045e53SAneesh Kumar K.V 	struct p9_rstat *rstat = (struct p9_rstat *)(inmsg->msg + sizeof(u16));
511af045e53SAneesh Kumar K.V 	struct p9_fid *fid = &p9dev->fids[tstat->fid];
5121c7850f9SSasha Levin 
51330204a77SAneesh Kumar K.V 	if (lstat(fid->abs_path, &st) < 0)
514*ead43b01SAneesh Kumar K.V 		return;
5151c7850f9SSasha Levin 
516b4422bf3SAneesh Kumar K.V 	ret = virtio_p9_fill_stat(p9dev, fid->path, &st, rstat);
5171c7850f9SSasha Levin 
518c81fc2c7SAneesh Kumar K.V 	*outlen = VIRTIO_P9_HDR_LEN + ret + sizeof(u16);
519af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RSTAT, outmsg->tag);
520*ead43b01SAneesh Kumar K.V 	return;
5211c7850f9SSasha Levin }
5221c7850f9SSasha Levin 
523*ead43b01SAneesh Kumar K.V static void virtio_p9_wstat(struct p9_dev *p9dev,
524af045e53SAneesh Kumar K.V 			    struct p9_pdu *pdu, u32 *outlen)
5251c7850f9SSasha Levin {
526aec426f0SCyrill Gorcunov 	int res = 0;
527af045e53SAneesh Kumar K.V 	struct p9_str *str;
528af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
529af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
530af045e53SAneesh Kumar K.V 	struct p9_twstat *twstat = (struct p9_twstat *)outmsg->msg;
531af045e53SAneesh Kumar K.V 	struct p9_fid *fid = &p9dev->fids[twstat->fid];
532af045e53SAneesh Kumar K.V 
5331c7850f9SSasha Levin 
5341c7850f9SSasha Levin 	if (twstat->stat.length != -1UL)
5351c7850f9SSasha Levin 		res = ftruncate(fid->fd, twstat->stat.length);
5361c7850f9SSasha Levin 
5371c7850f9SSasha Levin 	if (twstat->stat.mode != -1U)
5381c7850f9SSasha Levin 		chmod(fid->abs_path, twstat->stat.mode & 0xFFFF);
5391c7850f9SSasha Levin 
5401c7850f9SSasha Levin 	str = (void *)&twstat->stat.name + sizeof(u16);
5411c7850f9SSasha Levin 	if (str->len > 0) {
5421c7850f9SSasha Levin 		char new_name[PATH_MAX] = {0};
5431c7850f9SSasha Levin 		char full_path[PATH_MAX];
5441c7850f9SSasha Levin 		char *last_dir = strrchr(fid->path, '/');
5451c7850f9SSasha Levin 
5461c7850f9SSasha Levin 		/* We need to get the full file name out of twstat->name */
5471c7850f9SSasha Levin 		if (last_dir)
5481c7850f9SSasha Levin 			strncpy(new_name, fid->path, last_dir - fid->path + 1);
5491c7850f9SSasha Levin 
5501c7850f9SSasha Levin 		memcpy(new_name + strlen(new_name), &str->str, str->len);
5511c7850f9SSasha Levin 
5521c7850f9SSasha Levin 		/* fid is reused for the new file */
553b4422bf3SAneesh Kumar K.V 		rename(fid->abs_path, rel_to_abs(p9dev, new_name, full_path));
5541c7850f9SSasha Levin 		sprintf(fid->path, "%s", new_name);
5551c7850f9SSasha Levin 	}
5561c7850f9SSasha Levin 
5571c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN;
558af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RWSTAT, outmsg->tag);
559aec426f0SCyrill Gorcunov 
560*ead43b01SAneesh Kumar K.V 	return;
5611c7850f9SSasha Levin }
5621c7850f9SSasha Levin 
563*ead43b01SAneesh Kumar K.V static void virtio_p9_remove(struct p9_dev *p9dev,
564af045e53SAneesh Kumar K.V 			     struct p9_pdu *pdu, u32 *outlen)
5651c7850f9SSasha Levin {
566af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
567af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
568af045e53SAneesh Kumar K.V 	struct p9_tremove *tremove = (struct p9_tremove *)outmsg->msg;
569b4422bf3SAneesh Kumar K.V 	struct p9_fid *fid = &p9dev->fids[tremove->fid];
5701c7850f9SSasha Levin 
571b4422bf3SAneesh Kumar K.V 	close_fid(p9dev, tremove->fid);
5721c7850f9SSasha Levin 	if (fid->is_dir)
5731c7850f9SSasha Levin 		rmdir(fid->abs_path);
5741c7850f9SSasha Levin 	else
5751c7850f9SSasha Levin 		unlink(fid->abs_path);
5761c7850f9SSasha Levin 
5771c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN;
578af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RREMOVE, outmsg->tag);
579*ead43b01SAneesh Kumar K.V 	return;
5801c7850f9SSasha Levin }
5811c7850f9SSasha Levin 
582*ead43b01SAneesh Kumar K.V static void virtio_p9_write(struct p9_dev *p9dev,
583af045e53SAneesh Kumar K.V 			    struct p9_pdu *pdu, u32 *outlen)
5841c7850f9SSasha Levin {
585af045e53SAneesh Kumar K.V 	struct p9_msg *inmsg  = pdu->in_iov[0].iov_base;
586af045e53SAneesh Kumar K.V 	struct p9_msg *outmsg = pdu->out_iov[0].iov_base;
587af045e53SAneesh Kumar K.V 	struct p9_twrite *twrite = (struct p9_twrite *)outmsg->msg;
588af045e53SAneesh Kumar K.V 	struct p9_rwrite *rwrite = (struct p9_rwrite *)inmsg->msg;
589b4422bf3SAneesh Kumar K.V 	struct p9_fid *fid = &p9dev->fids[twrite->fid];
5901c7850f9SSasha Levin 
591af045e53SAneesh Kumar K.V 
592af045e53SAneesh Kumar K.V 	pdu->out_iov[0].iov_base += (sizeof(*outmsg) + sizeof(*twrite));
593af045e53SAneesh Kumar K.V 	pdu->out_iov[0].iov_len -= (sizeof(*outmsg) + sizeof(*twrite));
5946b163a87SAneesh Kumar K.V 	pdu->out_iov_cnt = virtio_p9_update_iov_cnt(pdu->out_iov, twrite->count,
5956b163a87SAneesh Kumar K.V 						    pdu->out_iov_cnt);
596af045e53SAneesh Kumar K.V 	rwrite->count = pwritev(fid->fd, pdu->out_iov,
597af045e53SAneesh Kumar K.V 				pdu->out_iov_cnt, twrite->offset);
5981c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(u32);
599af045e53SAneesh Kumar K.V 	set_p9msg_hdr(inmsg, *outlen, P9_RWRITE, outmsg->tag);
6001c7850f9SSasha Levin 
601*ead43b01SAneesh Kumar K.V 	return;
6021c7850f9SSasha Levin }
6031c7850f9SSasha Levin 
604*ead43b01SAneesh Kumar K.V typedef void p9_handler(struct p9_dev *p9dev,
605af045e53SAneesh Kumar K.V 			struct p9_pdu *pdu, u32 *outlen);
606b4422bf3SAneesh Kumar K.V 
607b4422bf3SAneesh Kumar K.V static p9_handler *virtio_9p_handler [] = {
608b4422bf3SAneesh Kumar K.V 	[P9_TVERSION] = virtio_p9_version,
609b4422bf3SAneesh Kumar K.V 	[P9_TATTACH]  = virtio_p9_attach,
610b4422bf3SAneesh Kumar K.V 	[P9_TSTAT]    = virtio_p9_stat,
611b4422bf3SAneesh Kumar K.V 	[P9_TCLUNK]   =	virtio_p9_clunk,
612b4422bf3SAneesh Kumar K.V 	[P9_TWALK]    =	virtio_p9_walk,
613b4422bf3SAneesh Kumar K.V 	[P9_TOPEN]    =	virtio_p9_open,
614b4422bf3SAneesh Kumar K.V 	[P9_TREAD]    = virtio_p9_read,
615b4422bf3SAneesh Kumar K.V 	[P9_TCREATE]  =	virtio_p9_create,
616b4422bf3SAneesh Kumar K.V 	[P9_TWSTAT]   =	virtio_p9_wstat,
617b4422bf3SAneesh Kumar K.V 	[P9_TREMOVE]  =	virtio_p9_remove,
618b4422bf3SAneesh Kumar K.V 	[P9_TWRITE]   =	virtio_p9_write,
619b4422bf3SAneesh Kumar K.V };
620b4422bf3SAneesh Kumar K.V 
621af045e53SAneesh Kumar K.V static struct p9_pdu *virtio_p9_pdu_init(struct kvm *kvm, struct virt_queue *vq)
622af045e53SAneesh Kumar K.V {
623af045e53SAneesh Kumar K.V 	struct p9_pdu *pdu = calloc(1, sizeof(*pdu));
624af045e53SAneesh Kumar K.V 	if (!pdu)
625af045e53SAneesh Kumar K.V 		return NULL;
626af045e53SAneesh Kumar K.V 
627af045e53SAneesh Kumar K.V 	pdu->queue_head  = virt_queue__get_inout_iov(kvm, vq, pdu->in_iov,
628af045e53SAneesh Kumar K.V 						     pdu->out_iov,
629af045e53SAneesh Kumar K.V 						     &pdu->in_iov_cnt,
630af045e53SAneesh Kumar K.V 						     &pdu->out_iov_cnt);
631af045e53SAneesh Kumar K.V 	return pdu;
632af045e53SAneesh Kumar K.V }
633af045e53SAneesh Kumar K.V 
634af045e53SAneesh Kumar K.V static u8 virtio_p9_get_cmd(struct p9_pdu *pdu)
635af045e53SAneesh Kumar K.V {
636af045e53SAneesh Kumar K.V 	struct p9_msg *msg;
637af045e53SAneesh Kumar K.V 	/*
638af045e53SAneesh Kumar K.V 	 * we can peek directly into pdu for a u8
639af045e53SAneesh Kumar K.V 	 * value. The host endianess won't be an issue
640af045e53SAneesh Kumar K.V 	 */
641af045e53SAneesh Kumar K.V 	msg = pdu->out_iov[0].iov_base;
642af045e53SAneesh Kumar K.V 	return msg->cmd;
643af045e53SAneesh Kumar K.V }
644af045e53SAneesh Kumar K.V 
645b4422bf3SAneesh Kumar K.V static bool virtio_p9_do_io_request(struct kvm *kvm, struct p9_dev_job *job)
6461c7850f9SSasha Levin {
647af045e53SAneesh Kumar K.V 	u8 cmd;
648b4422bf3SAneesh Kumar K.V 	u32 len = 0;
649b4422bf3SAneesh Kumar K.V 	p9_handler *handler;
650b4422bf3SAneesh Kumar K.V 	struct p9_dev *p9dev;
651af045e53SAneesh Kumar K.V 	struct virt_queue *vq;
652af045e53SAneesh Kumar K.V 	struct p9_pdu *p9pdu;
6531c7850f9SSasha Levin 
654b4422bf3SAneesh Kumar K.V 	vq = job->vq;
655b4422bf3SAneesh Kumar K.V 	p9dev = job->p9dev;
6561c7850f9SSasha Levin 
657af045e53SAneesh Kumar K.V 	p9pdu = virtio_p9_pdu_init(kvm, vq);
658af045e53SAneesh Kumar K.V 	cmd = virtio_p9_get_cmd(p9pdu);
659af045e53SAneesh Kumar K.V 
660af045e53SAneesh Kumar K.V 	if (cmd >= ARRAY_SIZE(virtio_9p_handler) ||
661af045e53SAneesh Kumar K.V 	    !virtio_9p_handler[cmd]) {
662af045e53SAneesh Kumar K.V 		printf("Unsupported P9 message type: %u\n", cmd);
6631c7850f9SSasha Levin 
664b4422bf3SAneesh Kumar K.V 	} else {
665af045e53SAneesh Kumar K.V 		handler = virtio_9p_handler[cmd];
666af045e53SAneesh Kumar K.V 		handler(p9dev, p9pdu, &len);
667b4422bf3SAneesh Kumar K.V 	}
668af045e53SAneesh Kumar K.V 	virt_queue__set_used_elem(vq, p9pdu->queue_head, len);
669af045e53SAneesh Kumar K.V 	free(p9pdu);
6701c7850f9SSasha Levin 	return true;
6711c7850f9SSasha Levin }
6721c7850f9SSasha Levin 
6731c7850f9SSasha Levin static void virtio_p9_do_io(struct kvm *kvm, void *param)
6741c7850f9SSasha Levin {
675b4422bf3SAneesh Kumar K.V 	struct p9_dev_job *job = (struct p9_dev_job *)param;
676b4422bf3SAneesh Kumar K.V 	struct p9_dev *p9dev   = job->p9dev;
677b4422bf3SAneesh Kumar K.V 	struct virt_queue *vq  = job->vq;
6781c7850f9SSasha Levin 
6791c7850f9SSasha Levin 	while (virt_queue__available(vq)) {
680b4422bf3SAneesh Kumar K.V 		virtio_p9_do_io_request(kvm, job);
681b4422bf3SAneesh Kumar K.V 		virt_queue__trigger_irq(vq, p9dev->pci_hdr.irq_line,
682b4422bf3SAneesh Kumar K.V 					&p9dev->isr, kvm);
6831c7850f9SSasha Levin 	}
6841c7850f9SSasha Levin }
6851c7850f9SSasha Levin 
68660eb42d5SSasha Levin static void ioevent_callback(struct kvm *kvm, void *param)
68760eb42d5SSasha Levin {
68860eb42d5SSasha Levin 	struct p9_dev_job *job = param;
68960eb42d5SSasha Levin 
69060eb42d5SSasha Levin 	thread_pool__do_job(job->job_id);
69160eb42d5SSasha Levin }
69260eb42d5SSasha Levin 
693b4422bf3SAneesh Kumar K.V static bool virtio_p9_pci_io_out(struct ioport *ioport, struct kvm *kvm,
694b4422bf3SAneesh Kumar K.V 				 u16 port, void *data, int size, u32 count)
6951c7850f9SSasha Levin {
6961c7850f9SSasha Levin 	unsigned long offset;
6971c7850f9SSasha Levin 	bool ret = true;
698b4422bf3SAneesh Kumar K.V 	struct p9_dev  *p9dev;
69960eb42d5SSasha Levin 	struct ioevent ioevent;
7001c7850f9SSasha Levin 
701b4422bf3SAneesh Kumar K.V 	p9dev = ioport->priv;
702b4422bf3SAneesh Kumar K.V 	offset = port - p9dev->base_addr;
7031c7850f9SSasha Levin 
7041c7850f9SSasha Levin 	switch (offset) {
7051c7850f9SSasha Levin 	case VIRTIO_MSI_QUEUE_VECTOR:
7061c7850f9SSasha Levin 	case VIRTIO_PCI_GUEST_FEATURES:
7071c7850f9SSasha Levin 		break;
7081c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_PFN: {
7091c7850f9SSasha Levin 		void *p;
710b4422bf3SAneesh Kumar K.V 		struct p9_dev_job *job;
711b4422bf3SAneesh Kumar K.V 		struct virt_queue *queue;
7121c7850f9SSasha Levin 
713b4422bf3SAneesh Kumar K.V 		job			= &p9dev->jobs[p9dev->queue_selector];
714b4422bf3SAneesh Kumar K.V 		queue			= &p9dev->vqs[p9dev->queue_selector];
7151c7850f9SSasha Levin 		queue->pfn		= ioport__read32(data);
7161c7850f9SSasha Levin 		p			= guest_pfn_to_host(kvm, queue->pfn);
7171c7850f9SSasha Levin 
718612038c5SAneesh Kumar K.V 		vring_init(&queue->vring, VIRTQUEUE_NUM, p,
719b4422bf3SAneesh Kumar K.V 			   VIRTIO_PCI_VRING_ALIGN);
7201c7850f9SSasha Levin 
721b4422bf3SAneesh Kumar K.V 		*job			= (struct p9_dev_job) {
722b4422bf3SAneesh Kumar K.V 			.vq			= queue,
723b4422bf3SAneesh Kumar K.V 			.p9dev			= p9dev,
724b4422bf3SAneesh Kumar K.V 		};
725b4422bf3SAneesh Kumar K.V 		job->job_id = thread_pool__add_job(kvm, virtio_p9_do_io, job);
72660eb42d5SSasha Levin 
72760eb42d5SSasha Levin 		ioevent = (struct ioevent) {
72860eb42d5SSasha Levin 			.io_addr		= p9dev->base_addr + VIRTIO_PCI_QUEUE_NOTIFY,
72960eb42d5SSasha Levin 			.io_len			= sizeof(u16),
73060eb42d5SSasha Levin 			.fn			= ioevent_callback,
73160eb42d5SSasha Levin 			.datamatch		= p9dev->queue_selector,
73260eb42d5SSasha Levin 			.fn_ptr			= &p9dev->jobs[p9dev->queue_selector],
73360eb42d5SSasha Levin 			.fn_kvm			= kvm,
73460eb42d5SSasha Levin 			.fd			= eventfd(0, 0),
73560eb42d5SSasha Levin 		};
73660eb42d5SSasha Levin 
73760eb42d5SSasha Levin 		ioeventfd__add_event(&ioevent);
73860eb42d5SSasha Levin 
7391c7850f9SSasha Levin 		break;
7401c7850f9SSasha Levin 	}
7411c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_SEL:
742b4422bf3SAneesh Kumar K.V 		p9dev->queue_selector	= ioport__read16(data);
7431c7850f9SSasha Levin 		break;
7441c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_NOTIFY: {
7451c7850f9SSasha Levin 		u16 queue_index;
74660eb42d5SSasha Levin 
7471c7850f9SSasha Levin 		queue_index		= ioport__read16(data);
748b4422bf3SAneesh Kumar K.V 		thread_pool__do_job(p9dev->jobs[queue_index].job_id);
7491c7850f9SSasha Levin 		break;
7501c7850f9SSasha Levin 	}
7511c7850f9SSasha Levin 	case VIRTIO_PCI_STATUS:
752b4422bf3SAneesh Kumar K.V 		p9dev->status		= ioport__read8(data);
7531c7850f9SSasha Levin 		break;
7541c7850f9SSasha Levin 	case VIRTIO_MSI_CONFIG_VECTOR:
755b4422bf3SAneesh Kumar K.V 		p9dev->config_vector	= VIRTIO_MSI_NO_VECTOR;
7561c7850f9SSasha Levin 		break;
7571c7850f9SSasha Levin 	default:
7581c7850f9SSasha Levin 		ret			= false;
7591c7850f9SSasha Levin 		break;
7601c7850f9SSasha Levin 	};
7611c7850f9SSasha Levin 
7621c7850f9SSasha Levin 	return ret;
7631c7850f9SSasha Levin }
7641c7850f9SSasha Levin 
7651c7850f9SSasha Levin static struct ioport_operations virtio_p9_io_ops = {
7661c7850f9SSasha Levin 	.io_in				= virtio_p9_pci_io_in,
7671c7850f9SSasha Levin 	.io_out				= virtio_p9_pci_io_out,
7681c7850f9SSasha Levin };
7691c7850f9SSasha Levin 
770b4422bf3SAneesh Kumar K.V void virtio_9p__init(struct kvm *kvm, const char *root, const char *tag_name)
7711c7850f9SSasha Levin {
7721c7850f9SSasha Levin 	u8 pin, line, dev;
7731c7850f9SSasha Levin 	u32 i, root_len;
774f884f920SSasha Levin 	u16 p9_base_addr;
775b4422bf3SAneesh Kumar K.V 	struct p9_dev *p9dev;
7761c7850f9SSasha Levin 
777b4422bf3SAneesh Kumar K.V 	p9dev = calloc(1, sizeof(*p9dev));
778b4422bf3SAneesh Kumar K.V 	if (!p9dev)
7791c7850f9SSasha Levin 		return;
780b4422bf3SAneesh Kumar K.V 	if (!tag_name)
781b4422bf3SAneesh Kumar K.V 		tag_name = VIRTIO_P9_DEFAULT_TAG;
782b4422bf3SAneesh Kumar K.V 	p9dev->config = calloc(1, sizeof(*p9dev->config) + strlen(tag_name) + 1);
783b4422bf3SAneesh Kumar K.V 	if (p9dev->config == NULL)
784b4422bf3SAneesh Kumar K.V 		goto free_p9dev;
7851c7850f9SSasha Levin 
786b4422bf3SAneesh Kumar K.V 	strcpy(p9dev->root_dir, root);
7871c7850f9SSasha Levin 	root_len = strlen(root);
7881c7850f9SSasha Levin 	/*
7891c7850f9SSasha Levin 	 * We prefix the full path in all fids, This allows us to get the
7901c7850f9SSasha Levin 	 * absolute path of an fid without playing with strings.
7911c7850f9SSasha Levin 	 */
7921c7850f9SSasha Levin 	for (i = 0; i < VIRTIO_P9_MAX_FID; i++) {
793b4422bf3SAneesh Kumar K.V 		strcpy(p9dev->fids[i].abs_path, root);
794b4422bf3SAneesh Kumar K.V 		p9dev->fids[i].path = p9dev->fids[i].abs_path + root_len;
7951c7850f9SSasha Levin 	}
796b4422bf3SAneesh Kumar K.V 	p9dev->config->tag_len = strlen(tag_name);
797b4422bf3SAneesh Kumar K.V 	if (p9dev->config->tag_len > MAX_TAG_LEN)
798b4422bf3SAneesh Kumar K.V 		goto free_p9dev_config;
7991c7850f9SSasha Levin 
800b4422bf3SAneesh Kumar K.V 	memcpy(p9dev->config->tag, tag_name, strlen(tag_name));
801b4422bf3SAneesh Kumar K.V 	p9dev->features |= 1 << VIRTIO_9P_MOUNT_TAG;
8021c7850f9SSasha Levin 
8031c7850f9SSasha Levin 	if (irq__register_device(VIRTIO_ID_9P, &dev, &pin, &line) < 0)
804b4422bf3SAneesh Kumar K.V 		goto free_p9dev_config;
8051c7850f9SSasha Levin 
806b4422bf3SAneesh Kumar K.V 	p9_base_addr			= ioport__register(IOPORT_EMPTY,
807b4422bf3SAneesh Kumar K.V 							   &virtio_p9_io_ops,
808b4422bf3SAneesh Kumar K.V 							   IOPORT_SIZE, p9dev);
809b4422bf3SAneesh Kumar K.V 	p9dev->base_addr		    = p9_base_addr;
810b4422bf3SAneesh Kumar K.V 	p9dev->pci_hdr = (struct pci_device_header) {
811b4422bf3SAneesh Kumar K.V 		.vendor_id		= PCI_VENDOR_ID_REDHAT_QUMRANET,
812b4422bf3SAneesh Kumar K.V 		.device_id		= PCI_DEVICE_ID_VIRTIO_P9,
813b4422bf3SAneesh Kumar K.V 		.header_type		= PCI_HEADER_TYPE_NORMAL,
814b4422bf3SAneesh Kumar K.V 		.revision_id		= 0,
815b4422bf3SAneesh Kumar K.V 		.class			= 0x010000,
816b4422bf3SAneesh Kumar K.V 		.subsys_vendor_id	= PCI_SUBSYSTEM_VENDOR_ID_REDHAT_QUMRANET,
817b4422bf3SAneesh Kumar K.V 		.subsys_id		= VIRTIO_ID_9P,
818b4422bf3SAneesh Kumar K.V 		.irq_pin		= pin,
819b4422bf3SAneesh Kumar K.V 		.irq_line		= line,
820b4422bf3SAneesh Kumar K.V 		.bar[0]			= p9_base_addr | PCI_BASE_ADDRESS_SPACE_IO,
821b4422bf3SAneesh Kumar K.V 	};
822b4422bf3SAneesh Kumar K.V 	pci__register(&p9dev->pci_hdr, dev);
823b4422bf3SAneesh Kumar K.V 
824b4422bf3SAneesh Kumar K.V 	return;
825b4422bf3SAneesh Kumar K.V free_p9dev_config:
826b4422bf3SAneesh Kumar K.V 	free(p9dev->config);
827b4422bf3SAneesh Kumar K.V free_p9dev:
828b4422bf3SAneesh Kumar K.V 	free(p9dev);
8291c7850f9SSasha Levin }
830