xref: /kvmtool/virtio/9p.c (revision c81fc2c78c0165ecc9eea36b7bfd7c984ad7cef6)
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 
711c7850f9SSasha Levin /* Warning: Immediately use value returned from this function */
72b4422bf3SAneesh Kumar K.V static const char *rel_to_abs(struct p9_dev *p9dev,
73b4422bf3SAneesh Kumar K.V 			      const char *path, char *abs_path)
741c7850f9SSasha Levin {
75b4422bf3SAneesh Kumar K.V 	sprintf(abs_path, "%s/%s", p9dev->root_dir, path);
761c7850f9SSasha Levin 
771c7850f9SSasha Levin 	return abs_path;
781c7850f9SSasha Levin }
791c7850f9SSasha Levin 
80b4422bf3SAneesh Kumar K.V static bool virtio_p9_dev_in(struct p9_dev *p9dev, void *data,
81b4422bf3SAneesh Kumar K.V 			     unsigned long offset,
82b4422bf3SAneesh Kumar K.V 			     int size, u32 count)
831c7850f9SSasha Levin {
84b4422bf3SAneesh Kumar K.V 	u8 *config_space = (u8 *) p9dev->config;
851c7850f9SSasha Levin 
861c7850f9SSasha Levin 	if (size != 1 || count != 1)
871c7850f9SSasha Levin 		return false;
881c7850f9SSasha Levin 
891c7850f9SSasha Levin 	ioport__write8(data, config_space[offset - VIRTIO_MSI_CONFIG_VECTOR]);
901c7850f9SSasha Levin 
911c7850f9SSasha Levin 	return true;
921c7850f9SSasha Levin }
931c7850f9SSasha Levin 
94b4422bf3SAneesh Kumar K.V static bool virtio_p9_pci_io_in(struct ioport *ioport, struct kvm *kvm,
95b4422bf3SAneesh Kumar K.V 				u16 port, void *data, int size, u32 count)
961c7850f9SSasha Levin {
971c7850f9SSasha Levin 	bool ret = true;
98b4422bf3SAneesh Kumar K.V 	unsigned long offset;
99b4422bf3SAneesh Kumar K.V 	struct p9_dev *p9dev = ioport->priv;
1001c7850f9SSasha Levin 
101b4422bf3SAneesh Kumar K.V 
102b4422bf3SAneesh Kumar K.V 	offset = port - p9dev->base_addr;
1031c7850f9SSasha Levin 
1041c7850f9SSasha Levin 	switch (offset) {
1051c7850f9SSasha Levin 	case VIRTIO_PCI_HOST_FEATURES:
106b4422bf3SAneesh Kumar K.V 		ioport__write32(data, p9dev->features);
1071c7850f9SSasha Levin 		ret = true;
1081c7850f9SSasha Levin 		break;
1091c7850f9SSasha Levin 	case VIRTIO_PCI_GUEST_FEATURES:
1101c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_SEL:
1111c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_NOTIFY:
1121c7850f9SSasha Levin 		ret = false;
1131c7850f9SSasha Levin 		break;
1141c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_PFN:
115b4422bf3SAneesh Kumar K.V 		ioport__write32(data, p9dev->vqs[p9dev->queue_selector].pfn);
1161c7850f9SSasha Levin 		break;
1171c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_NUM:
118612038c5SAneesh Kumar K.V 		ioport__write16(data, VIRTQUEUE_NUM);
1191c7850f9SSasha Levin 		break;
1201c7850f9SSasha Levin 	case VIRTIO_PCI_STATUS:
121b4422bf3SAneesh Kumar K.V 		ioport__write8(data, p9dev->status);
1221c7850f9SSasha Levin 		break;
1231c7850f9SSasha Levin 	case VIRTIO_PCI_ISR:
124b4422bf3SAneesh Kumar K.V 		ioport__write8(data, p9dev->isr);
125b4422bf3SAneesh Kumar K.V 		kvm__irq_line(kvm, p9dev->pci_hdr.irq_line, VIRTIO_IRQ_LOW);
126b4422bf3SAneesh Kumar K.V 		p9dev->isr = VIRTIO_IRQ_LOW;
1271c7850f9SSasha Levin 		break;
1281c7850f9SSasha Levin 	default:
129b4422bf3SAneesh Kumar K.V 		ret = virtio_p9_dev_in(p9dev, data, offset, size, count);
1301c7850f9SSasha Levin 		break;
1311c7850f9SSasha Levin 	};
1321c7850f9SSasha Levin 
1331c7850f9SSasha Levin 	return ret;
1341c7850f9SSasha Levin }
1351c7850f9SSasha Levin 
1361c7850f9SSasha Levin static int omode2uflags(u8 mode)
1371c7850f9SSasha Levin {
1381c7850f9SSasha Levin 	int ret = 0;
1391c7850f9SSasha Levin 
1401c7850f9SSasha Levin 	/* Basic open modes are same as uflags */
1411c7850f9SSasha Levin 	ret = mode & 3;
1421c7850f9SSasha Levin 
1431c7850f9SSasha Levin 	/* Everything else is different */
1441c7850f9SSasha Levin 	if (mode & P9_OTRUNC)
1451c7850f9SSasha Levin 		ret |= O_TRUNC;
1461c7850f9SSasha Levin 
1471c7850f9SSasha Levin 	if (mode & P9_OAPPEND)
1481c7850f9SSasha Levin 		ret |= O_APPEND;
1491c7850f9SSasha Levin 
1501c7850f9SSasha Levin 	if (mode & P9_OEXCL)
1511c7850f9SSasha Levin 		ret |= O_EXCL;
1521c7850f9SSasha Levin 
1531c7850f9SSasha Levin 	return ret;
1541c7850f9SSasha Levin }
1551c7850f9SSasha Levin 
1561c7850f9SSasha Levin static void st2qid(struct stat *st, struct p9_qid *qid)
1571c7850f9SSasha Levin {
1581c7850f9SSasha Levin 	*qid = (struct p9_qid) {
1591c7850f9SSasha Levin 		.path		= st->st_ino,
1601c7850f9SSasha Levin 		.version	= st->st_mtime,
1611c7850f9SSasha Levin 	};
1621c7850f9SSasha Levin 
1631c7850f9SSasha Levin 	if (S_ISDIR(st->st_mode))
1641c7850f9SSasha Levin 		qid->type	|= P9_QTDIR;
1651c7850f9SSasha Levin }
1661c7850f9SSasha Levin 
167b4422bf3SAneesh Kumar K.V static void close_fid(struct p9_dev *p9dev, u32 fid)
1681c7850f9SSasha Levin {
169b4422bf3SAneesh Kumar K.V 	if (p9dev->fids[fid].fd > 0) {
170b4422bf3SAneesh Kumar K.V 		close(p9dev->fids[fid].fd);
171b4422bf3SAneesh Kumar K.V 		p9dev->fids[fid].fd = -1;
1721c7850f9SSasha Levin 	}
173b4422bf3SAneesh Kumar K.V 	if (p9dev->fids[fid].dir) {
174b4422bf3SAneesh Kumar K.V 		closedir(p9dev->fids[fid].dir);
175b4422bf3SAneesh Kumar K.V 		p9dev->fids[fid].dir = NULL;
1761c7850f9SSasha Levin 	}
1771c7850f9SSasha Levin }
1781c7850f9SSasha Levin 
1791c7850f9SSasha Levin static void set_p9msg_hdr(struct p9_msg *msg, u32 size, u8 cmd, u16 tag)
1801c7850f9SSasha Levin {
1811c7850f9SSasha Levin 	*msg = (struct p9_msg) {
1821c7850f9SSasha Levin 		.size	= size,
1831c7850f9SSasha Levin 		.tag	= tag,
1841c7850f9SSasha Levin 		.cmd	= cmd,
1851c7850f9SSasha Levin 	};
1861c7850f9SSasha Levin }
1871c7850f9SSasha Levin 
188b4422bf3SAneesh Kumar K.V static bool virtio_p9_version(struct p9_dev *p9dev, struct p9_msg *msg,
189b4422bf3SAneesh Kumar K.V 			      u32 len, struct iovec *iov,
190b4422bf3SAneesh Kumar K.V 			      int outiovcnt, int iniovcnt, u32 *outlen)
1911c7850f9SSasha Levin {
1921c7850f9SSasha Levin 	struct p9_msg *outmsg = iov[0].iov_base;
1931c7850f9SSasha Levin 	struct p9_rversion *rversion = (struct p9_rversion *)outmsg->msg;
1941c7850f9SSasha Levin 
1951c7850f9SSasha Levin 	rversion->msize		= 4096;
1961c7850f9SSasha Levin 	rversion->version.len	= strlen(VIRTIO_P9_VERSION);
1971c7850f9SSasha Levin 	memcpy(&rversion->version.str, VIRTIO_P9_VERSION, rversion->version.len);
1981c7850f9SSasha Levin 
199b4422bf3SAneesh Kumar K.V 	*outlen = VIRTIO_P9_HDR_LEN +
200b4422bf3SAneesh Kumar K.V 		rversion->version.len + sizeof(u16) + sizeof(u32);
2011c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_RVERSION, msg->tag);
2021c7850f9SSasha Levin 
2031c7850f9SSasha Levin 	return true;
2041c7850f9SSasha Levin }
2051c7850f9SSasha Levin 
206b4422bf3SAneesh Kumar K.V static bool virtio_p9_clunk(struct p9_dev *p9dev, struct p9_msg *msg,
207b4422bf3SAneesh Kumar K.V 			    u32 len, struct iovec *iov,
208b4422bf3SAneesh Kumar K.V 			    int outiovcnt, int iniovcnt, u32 *outlen)
2091c7850f9SSasha Levin {
2101c7850f9SSasha Levin 	struct p9_msg *outmsg = iov[0].iov_base;
2111c7850f9SSasha Levin 	struct p9_tclunk *tclunk = (struct p9_tclunk *)msg->msg;
2121c7850f9SSasha Levin 
213b4422bf3SAneesh Kumar K.V 	close_fid(p9dev, tclunk->fid);
2141c7850f9SSasha Levin 
2151c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN;
2161c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_RCLUNK, msg->tag);
2171c7850f9SSasha Levin 
2181c7850f9SSasha Levin 	return true;
2191c7850f9SSasha Levin }
2201c7850f9SSasha Levin 
221b4422bf3SAneesh Kumar K.V static bool virtio_p9_open(struct p9_dev *p9dev, struct p9_msg *msg,
222b4422bf3SAneesh Kumar K.V 			   u32 len, struct iovec *iov,
223b4422bf3SAneesh Kumar K.V 			   int outiovcnt, int iniovcnt, u32 *outlen)
2241c7850f9SSasha Levin {
2251c7850f9SSasha Levin 	struct p9_msg *outmsg	= iov[0].iov_base;
2261c7850f9SSasha Levin 	struct p9_topen *topen	= (struct p9_topen *)msg->msg;
2271c7850f9SSasha Levin 	struct p9_ropen *ropen	= (struct p9_ropen *)outmsg->msg;
228b4422bf3SAneesh Kumar K.V 	struct p9_fid *new_fid	= &p9dev->fids[topen->fid];
2291c7850f9SSasha Levin 	struct stat st;
2301c7850f9SSasha Levin 
23130204a77SAneesh Kumar K.V 	if (lstat(new_fid->abs_path, &st) < 0)
2321c7850f9SSasha Levin 		return false;
2331c7850f9SSasha Levin 
2341c7850f9SSasha Levin 	st2qid(&st, &ropen->qid);
2351c7850f9SSasha Levin 	ropen->iounit = 0;
2361c7850f9SSasha Levin 
2371c7850f9SSasha Levin 	if (new_fid->is_dir)
2381c7850f9SSasha Levin 		new_fid->dir	= opendir(new_fid->abs_path);
2391c7850f9SSasha Levin 	else
2401c7850f9SSasha Levin 		new_fid->fd	= open(new_fid->abs_path, omode2uflags(topen->mode));
2411c7850f9SSasha Levin 
2421c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(*ropen);
2431c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_ROPEN, msg->tag);
2441c7850f9SSasha Levin 
2451c7850f9SSasha Levin 	return true;
2461c7850f9SSasha Levin }
2471c7850f9SSasha Levin 
248b4422bf3SAneesh Kumar K.V static bool virtio_p9_create(struct p9_dev *p9dev, struct p9_msg *msg,
249b4422bf3SAneesh Kumar K.V 			     u32 len, struct iovec *iov,
250b4422bf3SAneesh Kumar K.V 			     int outiovcnt, int iniovcnt, u32 *outlen)
2511c7850f9SSasha Levin {
2521c7850f9SSasha Levin 	struct p9_msg *outmsg		= iov[0].iov_base;
2531c7850f9SSasha Levin 	struct p9_tcreate *tcreate	= (struct p9_tcreate *)msg->msg;
2541c7850f9SSasha Levin 	struct p9_rcreate *rcreate	= (struct p9_rcreate *)outmsg->msg;
255b4422bf3SAneesh Kumar K.V 	struct p9_fid *fid		= &p9dev->fids[tcreate->fid];
2561c7850f9SSasha Levin 	struct stat st;
2571c7850f9SSasha Levin 	u8 mode;
2581c7850f9SSasha Levin 	u32 perm;
2591c7850f9SSasha Levin 
2601c7850f9SSasha Levin 	rcreate->iounit = 0;
2611c7850f9SSasha Levin 
2621c7850f9SSasha Levin 	/* Get last byte of the variable length struct */
2631c7850f9SSasha Levin 	mode = *((u8 *)msg + msg->size - 1);
2641c7850f9SSasha Levin 	perm = *(u32 *)((u8 *)msg + msg->size - 5);
2651c7850f9SSasha Levin 
2661c7850f9SSasha Levin 	sprintf(fid->path, "%s/%.*s", fid->path, tcreate->name.len, (char *)&tcreate->name.str);
2671c7850f9SSasha Levin 
268b4422bf3SAneesh Kumar K.V 	close_fid(p9dev, tcreate->fid);
2691c7850f9SSasha Levin 
2701c7850f9SSasha Levin 	if (perm & P9_DMDIR) {
2711c7850f9SSasha Levin 		mkdir(fid->abs_path, perm & 0xFFFF);
2721c7850f9SSasha Levin 		fid->dir = opendir(fid->abs_path);
2731c7850f9SSasha Levin 		fid->is_dir = 1;
2741c7850f9SSasha Levin 	} else {
2751c7850f9SSasha Levin 		fid->fd = open(fid->abs_path, omode2uflags(mode) | O_CREAT, 0777);
2761c7850f9SSasha Levin 	}
2771c7850f9SSasha Levin 
27830204a77SAneesh Kumar K.V 	if (lstat(fid->abs_path, &st) < 0)
2791c7850f9SSasha Levin 		return false;
2801c7850f9SSasha Levin 
2811c7850f9SSasha Levin 	st2qid(&st, &rcreate->qid);
2821c7850f9SSasha Levin 
2831c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(*rcreate);
2841c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_RCREATE, msg->tag);
2851c7850f9SSasha Levin 
2861c7850f9SSasha Levin 	return true;
2871c7850f9SSasha Levin }
2881c7850f9SSasha Levin 
289b4422bf3SAneesh Kumar K.V static bool virtio_p9_walk(struct p9_dev *p9dev, struct p9_msg *msg,
290b4422bf3SAneesh Kumar K.V 			   u32 len, struct iovec *iov,
291b4422bf3SAneesh Kumar K.V 			   int outiovcnt, int iniovcnt, u32 *outlen)
2921c7850f9SSasha Levin {
2931c7850f9SSasha Levin 	struct p9_msg *outmsg	= iov[0].iov_base;
2941c7850f9SSasha Levin 	struct p9_twalk *twalk	= (struct p9_twalk *)msg->msg;
2951c7850f9SSasha Levin 	struct p9_rwalk *rwalk	= (struct p9_rwalk *)outmsg->msg;
2961c7850f9SSasha Levin 	struct p9_str *str	= twalk->wnames;
297b4422bf3SAneesh Kumar K.V 	struct p9_fid *new_fid	= &p9dev->fids[twalk->newfid];
2981c7850f9SSasha Levin 	u8 i;
2991c7850f9SSasha Levin 
3001c7850f9SSasha Levin 	rwalk->nwqid = 0;
3011c7850f9SSasha Levin 	if (twalk->nwname) {
302b4422bf3SAneesh Kumar K.V 		struct p9_fid *fid = &p9dev->fids[twalk->fid];
3031c7850f9SSasha Levin 
3041c7850f9SSasha Levin 		for (i = 0; i < twalk->nwname; i++) {
3051c7850f9SSasha Levin 			char tmp[PATH_MAX] = {0};
3061c7850f9SSasha Levin 			char full_path[PATH_MAX];
3071c7850f9SSasha Levin 			struct stat st;
3081c7850f9SSasha Levin 
3091c7850f9SSasha Levin 			/* Format the new path we're 'walk'ing into */
3101c7850f9SSasha Levin 			sprintf(tmp, "%s/%.*s", fid->path, str->len, (char *)&str->str);
3111c7850f9SSasha Levin 
31230204a77SAneesh Kumar K.V 			if (lstat(rel_to_abs(p9dev, tmp, full_path), &st) < 0)
3131c7850f9SSasha Levin 				break;
3141c7850f9SSasha Levin 
3151c7850f9SSasha Levin 			st2qid(&st, &rwalk->wqids[i]);
3161c7850f9SSasha Levin 			new_fid->is_dir = S_ISDIR(st.st_mode);
3171c7850f9SSasha Levin 			strcpy(new_fid->path, tmp);
3181c7850f9SSasha Levin 			new_fid->fid = twalk->newfid;
3191c7850f9SSasha Levin 			rwalk->nwqid++;
3201c7850f9SSasha Levin 		}
3211c7850f9SSasha Levin 	} else {
322b4422bf3SAneesh Kumar K.V 		new_fid->is_dir = p9dev->fids[twalk->fid].is_dir;
323b4422bf3SAneesh Kumar K.V 		strcpy(new_fid->path, p9dev->fids[twalk->fid].path);
3241c7850f9SSasha Levin 		new_fid->fid	= twalk->newfid;
3251c7850f9SSasha Levin 	}
3261c7850f9SSasha Levin 
3271c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(u16) + sizeof(struct p9_qid) * rwalk->nwqid;
3281c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_RWALK, msg->tag);
3291c7850f9SSasha Levin 
3301c7850f9SSasha Levin 	return true;
3311c7850f9SSasha Levin }
3321c7850f9SSasha Levin 
333b4422bf3SAneesh Kumar K.V static bool virtio_p9_attach(struct p9_dev *p9dev, struct p9_msg *msg,
334b4422bf3SAneesh Kumar K.V 			     u32 len, struct iovec *iov,
335b4422bf3SAneesh Kumar K.V 			     int outiovcnt, int iniovcnt, u32 *outlen)
3361c7850f9SSasha Levin {
3371c7850f9SSasha Levin 	struct p9_msg *outmsg = iov[0].iov_base;
3381c7850f9SSasha Levin 	struct p9_rattach *rattach = (struct p9_rattach *)outmsg->msg;
3391c7850f9SSasha Levin 	struct p9_tattach *tattach = (struct p9_tattach *)msg->msg;
3401c7850f9SSasha Levin 	struct stat st;
3411c7850f9SSasha Levin 	struct p9_fid *fid;
3421c7850f9SSasha Levin 	u32 i;
3431c7850f9SSasha Levin 
3441c7850f9SSasha Levin 	/* Reset everything */
3451c7850f9SSasha Levin 	for (i = 0; i < VIRTIO_P9_MAX_FID; i++)
346b4422bf3SAneesh Kumar K.V 		p9dev->fids[i].fid = P9_NOFID;
3471c7850f9SSasha Levin 
34830204a77SAneesh Kumar K.V 	if (lstat(p9dev->root_dir, &st) < 0)
3491c7850f9SSasha Levin 		return false;
3501c7850f9SSasha Levin 
3511c7850f9SSasha Levin 	st2qid(&st, &rattach->qid);
3521c7850f9SSasha Levin 
353b4422bf3SAneesh Kumar K.V 	fid = &p9dev->fids[tattach->fid];
3541c7850f9SSasha Levin 	fid->fid = tattach->fid;
3551c7850f9SSasha Levin 	fid->is_dir = 1;
3561c7850f9SSasha Levin 	strcpy(fid->path, "/");
3571c7850f9SSasha Levin 
3581c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(*rattach);
3591c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_RATTACH, msg->tag);
3601c7850f9SSasha Levin 
3611c7850f9SSasha Levin 	return true;
3621c7850f9SSasha Levin }
3631c7850f9SSasha Levin 
364b4422bf3SAneesh Kumar K.V static u32 virtio_p9_fill_stat(struct p9_dev *p9dev, const char *name,
365b4422bf3SAneesh Kumar K.V 			       struct stat *st, struct p9_rstat *rstat)
3661c7850f9SSasha Levin {
3671c7850f9SSasha Levin 	struct p9_str *str;
3681c7850f9SSasha Levin 
3691c7850f9SSasha Levin 	rstat->stat.type = 0;
3701c7850f9SSasha Levin 	rstat->stat.dev = 0;
3711c7850f9SSasha Levin 	st2qid(st, &rstat->stat.qid);
3721c7850f9SSasha Levin 	rstat->stat.mode = st->st_mode;
3731c7850f9SSasha Levin 	rstat->stat.length = st->st_size;
3741c7850f9SSasha Levin 	if (S_ISDIR(st->st_mode)) {
3751c7850f9SSasha Levin 		rstat->stat.length = 0;
3761c7850f9SSasha Levin 		rstat->stat.mode |= P9_DMDIR;
3771c7850f9SSasha Levin 	}
3781c7850f9SSasha Levin 
3791c7850f9SSasha Levin 	rstat->stat.atime = st->st_atime;
3801c7850f9SSasha Levin 	rstat->stat.mtime = st->st_mtime;
3811c7850f9SSasha Levin 
3821c7850f9SSasha Levin 	str = (struct p9_str *)&rstat->stat.name;
3831c7850f9SSasha Levin 	str->len = strlen(name);
3841c7850f9SSasha Levin 	memcpy(&str->str, name, str->len);
3851c7850f9SSasha Levin 	str = (void *)str + str->len + sizeof(u16);
3861c7850f9SSasha Levin 
3871c7850f9SSasha Levin 	/* TODO: Pass usernames to the client */
3881c7850f9SSasha Levin 	str->len = 0;
3891c7850f9SSasha Levin 	str = (void *)str + sizeof(u16);
3901c7850f9SSasha Levin 	str->len = 0;
3911c7850f9SSasha Levin 	str = (void *)str + sizeof(u16);
3921c7850f9SSasha Levin 	str->len = 0;
3931c7850f9SSasha Levin 	str = (void *)str + sizeof(u16);
3941c7850f9SSasha Levin 
395b4422bf3SAneesh Kumar K.V 	/*
396b4422bf3SAneesh Kumar K.V 	 * We subtract a u16 here because rstat->size
397b4422bf3SAneesh Kumar K.V 	 * doesn't include rstat->size itself
398b4422bf3SAneesh Kumar K.V 	 */
3991c7850f9SSasha Levin 	rstat->stat.size = (void *)str - (void *)&rstat->stat - sizeof(u16);
4001c7850f9SSasha Levin 
4011c7850f9SSasha Levin 	return rstat->stat.size + sizeof(u16);
4021c7850f9SSasha Levin }
4031c7850f9SSasha Levin 
404b4422bf3SAneesh Kumar K.V static bool virtio_p9_read(struct p9_dev *p9dev, struct p9_msg *msg,
405b4422bf3SAneesh Kumar K.V 			   u32 len, struct iovec *iov,
406b4422bf3SAneesh Kumar K.V 			   int outiovcnt, int iniovcnt, u32 *outlen)
4071c7850f9SSasha Levin {
4081c7850f9SSasha Levin 	struct p9_msg *outmsg	= iov[0].iov_base;
4091c7850f9SSasha Levin 	struct p9_tread *tread	= (struct p9_tread *)msg->msg;
4101c7850f9SSasha Levin 	struct p9_rread *rread	= (struct p9_rread *)outmsg->msg;
4111c7850f9SSasha Levin 	struct p9_rstat *rstat	= (struct p9_rstat *)iov[1].iov_base;
412b4422bf3SAneesh Kumar K.V 	struct p9_fid *fid	= &p9dev->fids[tread->fid];
4131c7850f9SSasha Levin 	struct stat st;
4141c7850f9SSasha Levin 
4151c7850f9SSasha Levin 	rread->count = 0;
4161c7850f9SSasha Levin 
4171c7850f9SSasha Levin 	if (fid->is_dir) {
4181c7850f9SSasha Levin 		/* If reading a dir, fill the buffer with p9_stat entries */
4191c7850f9SSasha Levin 		struct dirent *cur = readdir(fid->dir);
4201c7850f9SSasha Levin 		char full_path[PATH_MAX];
4211c7850f9SSasha Levin 
4221c7850f9SSasha Levin 		while (cur) {
4231c7850f9SSasha Levin 			u32 read;
4241c7850f9SSasha Levin 
42530204a77SAneesh Kumar K.V 			lstat(rel_to_abs(p9dev, cur->d_name, full_path), &st);
426b4422bf3SAneesh Kumar K.V 			read = virtio_p9_fill_stat(p9dev, cur->d_name,
427b4422bf3SAneesh Kumar K.V 						   &st, rstat);
4281c7850f9SSasha Levin 			rread->count += read;
4291c7850f9SSasha Levin 			rstat = (void *)rstat + read;
4301c7850f9SSasha Levin 			cur = readdir(fid->dir);
4311c7850f9SSasha Levin 		}
4321c7850f9SSasha Levin 	} else {
4331c7850f9SSasha Levin 		iov[0].iov_base += VIRTIO_P9_HDR_LEN + sizeof(u32);
4341c7850f9SSasha Levin 		iov[0].iov_len -= VIRTIO_P9_HDR_LEN + sizeof(u32);
435b4422bf3SAneesh Kumar K.V 		rread->count = preadv(fid->fd, iov, iniovcnt, tread->offset);
4361c7850f9SSasha Levin 		if (rread->count > tread->count)
4371c7850f9SSasha Levin 			rread->count = tread->count;
4381c7850f9SSasha Levin 	}
4391c7850f9SSasha Levin 
4401c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(u32) + rread->count;
4411c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_RREAD, msg->tag);
4421c7850f9SSasha Levin 
4431c7850f9SSasha Levin 	return true;
4441c7850f9SSasha Levin }
4451c7850f9SSasha Levin 
446b4422bf3SAneesh Kumar K.V static bool virtio_p9_stat(struct p9_dev *p9dev, struct p9_msg *msg,
447b4422bf3SAneesh Kumar K.V 			   u32 len, struct iovec *iov,
448b4422bf3SAneesh Kumar K.V 			   int outiovcnt, int iniovcnt, u32 *outlen)
4491c7850f9SSasha Levin {
4501c7850f9SSasha Levin 	struct p9_msg *outmsg = iov[0].iov_base;
4511c7850f9SSasha Levin 	struct p9_tstat *tstat = (struct p9_tstat *)msg->msg;
4521c7850f9SSasha Levin 	struct p9_rstat *rstat = (struct p9_rstat *)(outmsg->msg + sizeof(u16));
4531c7850f9SSasha Levin 	struct stat st;
454b4422bf3SAneesh Kumar K.V 	struct p9_fid *fid = &p9dev->fids[tstat->fid];
4551c7850f9SSasha Levin 	u32 ret;
4561c7850f9SSasha Levin 
45730204a77SAneesh Kumar K.V 	if (lstat(fid->abs_path, &st) < 0)
4581c7850f9SSasha Levin 		return false;
4591c7850f9SSasha Levin 
460b4422bf3SAneesh Kumar K.V 	ret = virtio_p9_fill_stat(p9dev, fid->path, &st, rstat);
4611c7850f9SSasha Levin 
462*c81fc2c7SAneesh Kumar K.V 	*outlen = VIRTIO_P9_HDR_LEN + ret + sizeof(u16);
4631c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_RSTAT, msg->tag);
4641c7850f9SSasha Levin 	return true;
4651c7850f9SSasha Levin }
4661c7850f9SSasha Levin 
467b4422bf3SAneesh Kumar K.V static bool virtio_p9_wstat(struct p9_dev *p9dev, struct p9_msg *msg,
468b4422bf3SAneesh Kumar K.V 			    u32 len, struct iovec *iov,
469b4422bf3SAneesh Kumar K.V 			    int outiovcnt, int iniovcnt, u32 *outlen)
4701c7850f9SSasha Levin {
4711c7850f9SSasha Levin 	struct p9_msg *outmsg = iov[0].iov_base;
4721c7850f9SSasha Levin 	struct p9_twstat *twstat = (struct p9_twstat *)msg->msg;
4731c7850f9SSasha Levin 	struct p9_str *str;
474b4422bf3SAneesh Kumar K.V 	struct p9_fid *fid = &p9dev->fids[twstat->fid];
475aec426f0SCyrill Gorcunov 	int res = 0;
4761c7850f9SSasha Levin 
4771c7850f9SSasha Levin 	if (twstat->stat.length != -1UL)
4781c7850f9SSasha Levin 		res = ftruncate(fid->fd, twstat->stat.length);
4791c7850f9SSasha Levin 
4801c7850f9SSasha Levin 	if (twstat->stat.mode != -1U)
4811c7850f9SSasha Levin 		chmod(fid->abs_path, twstat->stat.mode & 0xFFFF);
4821c7850f9SSasha Levin 
4831c7850f9SSasha Levin 	str = (void *)&twstat->stat.name + sizeof(u16);
4841c7850f9SSasha Levin 	if (str->len > 0) {
4851c7850f9SSasha Levin 		char new_name[PATH_MAX] = {0};
4861c7850f9SSasha Levin 		char full_path[PATH_MAX];
4871c7850f9SSasha Levin 		char *last_dir = strrchr(fid->path, '/');
4881c7850f9SSasha Levin 
4891c7850f9SSasha Levin 		/* We need to get the full file name out of twstat->name */
4901c7850f9SSasha Levin 		if (last_dir)
4911c7850f9SSasha Levin 			strncpy(new_name, fid->path, last_dir - fid->path + 1);
4921c7850f9SSasha Levin 
4931c7850f9SSasha Levin 		memcpy(new_name + strlen(new_name), &str->str, str->len);
4941c7850f9SSasha Levin 
4951c7850f9SSasha Levin 		/* fid is reused for the new file */
496b4422bf3SAneesh Kumar K.V 		rename(fid->abs_path, rel_to_abs(p9dev, new_name, full_path));
4971c7850f9SSasha Levin 		sprintf(fid->path, "%s", new_name);
4981c7850f9SSasha Levin 	}
4991c7850f9SSasha Levin 
5001c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN;
5011c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_RWSTAT, msg->tag);
502aec426f0SCyrill Gorcunov 
503aec426f0SCyrill Gorcunov 	return res == 0;
5041c7850f9SSasha Levin }
5051c7850f9SSasha Levin 
506b4422bf3SAneesh Kumar K.V static bool virtio_p9_remove(struct p9_dev *p9dev, struct p9_msg *msg,
507b4422bf3SAneesh Kumar K.V 			     u32 len, struct iovec *iov,
508b4422bf3SAneesh Kumar K.V 			     int outiovcnt, int iniovcnt, u32 *outlen)
5091c7850f9SSasha Levin {
5101c7850f9SSasha Levin 	struct p9_msg *outmsg = iov[0].iov_base;
5111c7850f9SSasha Levin 	struct p9_tremove *tremove = (struct p9_tremove *)msg->msg;
512b4422bf3SAneesh Kumar K.V 	struct p9_fid *fid = &p9dev->fids[tremove->fid];
5131c7850f9SSasha Levin 
514b4422bf3SAneesh Kumar K.V 	close_fid(p9dev, tremove->fid);
5151c7850f9SSasha Levin 	if (fid->is_dir)
5161c7850f9SSasha Levin 		rmdir(fid->abs_path);
5171c7850f9SSasha Levin 	else
5181c7850f9SSasha Levin 		unlink(fid->abs_path);
5191c7850f9SSasha Levin 
5201c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN;
5211c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_RREMOVE, msg->tag);
5221c7850f9SSasha Levin 	return true;
5231c7850f9SSasha Levin }
5241c7850f9SSasha Levin 
525b4422bf3SAneesh Kumar K.V static bool virtio_p9_write(struct p9_dev *p9dev, struct p9_msg *msg,
526b4422bf3SAneesh Kumar K.V 			    u32 len, struct iovec *iov,
527b4422bf3SAneesh Kumar K.V 			    int outiovcnt, int iniovcnt, u32 *outlen)
5281c7850f9SSasha Levin {
5291c7850f9SSasha Levin 	struct p9_msg *outmsg;
5301c7850f9SSasha Levin 	struct p9_rwrite *rwrite;
5311c7850f9SSasha Levin 	struct p9_twrite *twrite = (struct p9_twrite *)msg->msg;
532b4422bf3SAneesh Kumar K.V 	struct p9_fid *fid = &p9dev->fids[twrite->fid];
5331c7850f9SSasha Levin 
534b4422bf3SAneesh Kumar K.V 	if (outiovcnt == 1) {
5351c7850f9SSasha Levin 		outmsg = iov[0].iov_base;
5361c7850f9SSasha Levin 		rwrite = (struct p9_rwrite *)outmsg->msg;
537b4422bf3SAneesh Kumar K.V 		rwrite->count = pwrite(fid->fd, &twrite->data,
538b4422bf3SAneesh Kumar K.V 				       twrite->count, twrite->offset);
5391c7850f9SSasha Levin 	} else {
5401c7850f9SSasha Levin 		outmsg = iov[2].iov_base;
5411c7850f9SSasha Levin 		rwrite = (struct p9_rwrite *)outmsg->msg;
542b4422bf3SAneesh Kumar K.V 		rwrite->count = pwrite(fid->fd, iov[1].iov_base,
543b4422bf3SAneesh Kumar K.V 				       twrite->count, twrite->offset);
5441c7850f9SSasha Levin 	}
5451c7850f9SSasha Levin 	*outlen = VIRTIO_P9_HDR_LEN + sizeof(u32);
5461c7850f9SSasha Levin 	set_p9msg_hdr(outmsg, *outlen, P9_RWRITE, msg->tag);
5471c7850f9SSasha Levin 
5481c7850f9SSasha Levin 	return true;
5491c7850f9SSasha Levin }
5501c7850f9SSasha Levin 
551b4422bf3SAneesh Kumar K.V typedef bool p9_handler(struct p9_dev *p9dev, struct p9_msg *msg,
552b4422bf3SAneesh Kumar K.V 			u32 len, struct iovec *iov,
553b4422bf3SAneesh Kumar K.V 			int outiovcnt, int iniovcnt, u32 *outlen);
554b4422bf3SAneesh Kumar K.V 
555b4422bf3SAneesh Kumar K.V static p9_handler *virtio_9p_handler [] = {
556b4422bf3SAneesh Kumar K.V 	[P9_TVERSION] = virtio_p9_version,
557b4422bf3SAneesh Kumar K.V 	[P9_TATTACH]  = virtio_p9_attach,
558b4422bf3SAneesh Kumar K.V 	[P9_TSTAT]    = virtio_p9_stat,
559b4422bf3SAneesh Kumar K.V 	[P9_TCLUNK]   =	virtio_p9_clunk,
560b4422bf3SAneesh Kumar K.V 	[P9_TWALK]    =	virtio_p9_walk,
561b4422bf3SAneesh Kumar K.V 	[P9_TOPEN]    =	virtio_p9_open,
562b4422bf3SAneesh Kumar K.V 	[P9_TREAD]    = virtio_p9_read,
563b4422bf3SAneesh Kumar K.V 	[P9_TCREATE]  =	virtio_p9_create,
564b4422bf3SAneesh Kumar K.V 	[P9_TWSTAT]   =	virtio_p9_wstat,
565b4422bf3SAneesh Kumar K.V 	[P9_TREMOVE]  =	virtio_p9_remove,
566b4422bf3SAneesh Kumar K.V 	[P9_TWRITE]   =	virtio_p9_write,
567b4422bf3SAneesh Kumar K.V };
568b4422bf3SAneesh Kumar K.V 
569b4422bf3SAneesh Kumar K.V static bool virtio_p9_do_io_request(struct kvm *kvm, struct p9_dev_job *job)
5701c7850f9SSasha Levin {
571b4422bf3SAneesh Kumar K.V 	u32 len = 0;
5721c7850f9SSasha Levin 	u16 out, in, head;
5731c7850f9SSasha Levin 	struct p9_msg *msg;
574b4422bf3SAneesh Kumar K.V 	p9_handler *handler;
575b4422bf3SAneesh Kumar K.V 	struct virt_queue *vq;
576b4422bf3SAneesh Kumar K.V 	struct p9_dev *p9dev;
577612038c5SAneesh Kumar K.V 	struct iovec iov[VIRTQUEUE_NUM];
5781c7850f9SSasha Levin 
579b4422bf3SAneesh Kumar K.V 	vq = job->vq;
580b4422bf3SAneesh Kumar K.V 	p9dev = job->p9dev;
581b4422bf3SAneesh Kumar K.V 	head  = virt_queue__get_iov(vq, iov, &out, &in, kvm);
5821c7850f9SSasha Levin 	msg   = iov[0].iov_base;
5831c7850f9SSasha Levin 
584b4422bf3SAneesh Kumar K.V 	if (msg->cmd >= ARRAY_SIZE(virtio_9p_handler) ||
585b4422bf3SAneesh Kumar K.V 	    !virtio_9p_handler[msg->cmd]) {
5861c7850f9SSasha Levin 		printf("Unsupported P9 message type: %u\n", msg->cmd);
5871c7850f9SSasha Levin 
588b4422bf3SAneesh Kumar K.V 	} else {
589b4422bf3SAneesh Kumar K.V 		handler = virtio_9p_handler[msg->cmd];
590b4422bf3SAneesh Kumar K.V 		handler(p9dev, msg, iov[0].iov_len, iov+1, out, in, &len);
591b4422bf3SAneesh Kumar K.V 	}
592b4422bf3SAneesh Kumar K.V 	virt_queue__set_used_elem(vq, head, len);
5931c7850f9SSasha Levin 	return true;
5941c7850f9SSasha Levin }
5951c7850f9SSasha Levin 
5961c7850f9SSasha Levin static void virtio_p9_do_io(struct kvm *kvm, void *param)
5971c7850f9SSasha Levin {
598b4422bf3SAneesh Kumar K.V 	struct p9_dev_job *job = (struct p9_dev_job *)param;
599b4422bf3SAneesh Kumar K.V 	struct p9_dev *p9dev   = job->p9dev;
600b4422bf3SAneesh Kumar K.V 	struct virt_queue *vq  = job->vq;
6011c7850f9SSasha Levin 
6021c7850f9SSasha Levin 	while (virt_queue__available(vq)) {
603b4422bf3SAneesh Kumar K.V 		virtio_p9_do_io_request(kvm, job);
604b4422bf3SAneesh Kumar K.V 		virt_queue__trigger_irq(vq, p9dev->pci_hdr.irq_line,
605b4422bf3SAneesh Kumar K.V 					&p9dev->isr, kvm);
6061c7850f9SSasha Levin 	}
6071c7850f9SSasha Levin }
6081c7850f9SSasha Levin 
60960eb42d5SSasha Levin static void ioevent_callback(struct kvm *kvm, void *param)
61060eb42d5SSasha Levin {
61160eb42d5SSasha Levin 	struct p9_dev_job *job = param;
61260eb42d5SSasha Levin 
61360eb42d5SSasha Levin 	thread_pool__do_job(job->job_id);
61460eb42d5SSasha Levin }
61560eb42d5SSasha Levin 
616b4422bf3SAneesh Kumar K.V static bool virtio_p9_pci_io_out(struct ioport *ioport, struct kvm *kvm,
617b4422bf3SAneesh Kumar K.V 				 u16 port, void *data, int size, u32 count)
6181c7850f9SSasha Levin {
6191c7850f9SSasha Levin 	unsigned long offset;
6201c7850f9SSasha Levin 	bool ret = true;
621b4422bf3SAneesh Kumar K.V 	struct p9_dev  *p9dev;
62260eb42d5SSasha Levin 	struct ioevent ioevent;
6231c7850f9SSasha Levin 
624b4422bf3SAneesh Kumar K.V 	p9dev = ioport->priv;
625b4422bf3SAneesh Kumar K.V 	offset = port - p9dev->base_addr;
6261c7850f9SSasha Levin 
6271c7850f9SSasha Levin 	switch (offset) {
6281c7850f9SSasha Levin 	case VIRTIO_MSI_QUEUE_VECTOR:
6291c7850f9SSasha Levin 	case VIRTIO_PCI_GUEST_FEATURES:
6301c7850f9SSasha Levin 		break;
6311c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_PFN: {
6321c7850f9SSasha Levin 		void *p;
633b4422bf3SAneesh Kumar K.V 		struct p9_dev_job *job;
634b4422bf3SAneesh Kumar K.V 		struct virt_queue *queue;
6351c7850f9SSasha Levin 
636b4422bf3SAneesh Kumar K.V 		job			= &p9dev->jobs[p9dev->queue_selector];
637b4422bf3SAneesh Kumar K.V 		queue			= &p9dev->vqs[p9dev->queue_selector];
6381c7850f9SSasha Levin 		queue->pfn		= ioport__read32(data);
6391c7850f9SSasha Levin 		p			= guest_pfn_to_host(kvm, queue->pfn);
6401c7850f9SSasha Levin 
641612038c5SAneesh Kumar K.V 		vring_init(&queue->vring, VIRTQUEUE_NUM, p,
642b4422bf3SAneesh Kumar K.V 			   VIRTIO_PCI_VRING_ALIGN);
6431c7850f9SSasha Levin 
644b4422bf3SAneesh Kumar K.V 		*job			= (struct p9_dev_job) {
645b4422bf3SAneesh Kumar K.V 			.vq			= queue,
646b4422bf3SAneesh Kumar K.V 			.p9dev			= p9dev,
647b4422bf3SAneesh Kumar K.V 		};
648b4422bf3SAneesh Kumar K.V 		job->job_id = thread_pool__add_job(kvm, virtio_p9_do_io, job);
64960eb42d5SSasha Levin 
65060eb42d5SSasha Levin 		ioevent = (struct ioevent) {
65160eb42d5SSasha Levin 			.io_addr		= p9dev->base_addr + VIRTIO_PCI_QUEUE_NOTIFY,
65260eb42d5SSasha Levin 			.io_len			= sizeof(u16),
65360eb42d5SSasha Levin 			.fn			= ioevent_callback,
65460eb42d5SSasha Levin 			.datamatch		= p9dev->queue_selector,
65560eb42d5SSasha Levin 			.fn_ptr			= &p9dev->jobs[p9dev->queue_selector],
65660eb42d5SSasha Levin 			.fn_kvm			= kvm,
65760eb42d5SSasha Levin 			.fd			= eventfd(0, 0),
65860eb42d5SSasha Levin 		};
65960eb42d5SSasha Levin 
66060eb42d5SSasha Levin 		ioeventfd__add_event(&ioevent);
66160eb42d5SSasha Levin 
6621c7850f9SSasha Levin 		break;
6631c7850f9SSasha Levin 	}
6641c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_SEL:
665b4422bf3SAneesh Kumar K.V 		p9dev->queue_selector	= ioport__read16(data);
6661c7850f9SSasha Levin 		break;
6671c7850f9SSasha Levin 	case VIRTIO_PCI_QUEUE_NOTIFY: {
6681c7850f9SSasha Levin 		u16 queue_index;
66960eb42d5SSasha Levin 
6701c7850f9SSasha Levin 		queue_index		= ioport__read16(data);
671b4422bf3SAneesh Kumar K.V 		thread_pool__do_job(p9dev->jobs[queue_index].job_id);
6721c7850f9SSasha Levin 		break;
6731c7850f9SSasha Levin 	}
6741c7850f9SSasha Levin 	case VIRTIO_PCI_STATUS:
675b4422bf3SAneesh Kumar K.V 		p9dev->status		= ioport__read8(data);
6761c7850f9SSasha Levin 		break;
6771c7850f9SSasha Levin 	case VIRTIO_MSI_CONFIG_VECTOR:
678b4422bf3SAneesh Kumar K.V 		p9dev->config_vector	= VIRTIO_MSI_NO_VECTOR;
6791c7850f9SSasha Levin 		break;
6801c7850f9SSasha Levin 	default:
6811c7850f9SSasha Levin 		ret			= false;
6821c7850f9SSasha Levin 		break;
6831c7850f9SSasha Levin 	};
6841c7850f9SSasha Levin 
6851c7850f9SSasha Levin 	return ret;
6861c7850f9SSasha Levin }
6871c7850f9SSasha Levin 
6881c7850f9SSasha Levin static struct ioport_operations virtio_p9_io_ops = {
6891c7850f9SSasha Levin 	.io_in				= virtio_p9_pci_io_in,
6901c7850f9SSasha Levin 	.io_out				= virtio_p9_pci_io_out,
6911c7850f9SSasha Levin };
6921c7850f9SSasha Levin 
693b4422bf3SAneesh Kumar K.V void virtio_9p__init(struct kvm *kvm, const char *root, const char *tag_name)
6941c7850f9SSasha Levin {
6951c7850f9SSasha Levin 	u8 pin, line, dev;
6961c7850f9SSasha Levin 	u32 i, root_len;
697f884f920SSasha Levin 	u16 p9_base_addr;
698b4422bf3SAneesh Kumar K.V 	struct p9_dev *p9dev;
6991c7850f9SSasha Levin 
700b4422bf3SAneesh Kumar K.V 	p9dev = calloc(1, sizeof(*p9dev));
701b4422bf3SAneesh Kumar K.V 	if (!p9dev)
7021c7850f9SSasha Levin 		return;
703b4422bf3SAneesh Kumar K.V 	if (!tag_name)
704b4422bf3SAneesh Kumar K.V 		tag_name = VIRTIO_P9_DEFAULT_TAG;
705b4422bf3SAneesh Kumar K.V 	p9dev->config = calloc(1, sizeof(*p9dev->config) + strlen(tag_name) + 1);
706b4422bf3SAneesh Kumar K.V 	if (p9dev->config == NULL)
707b4422bf3SAneesh Kumar K.V 		goto free_p9dev;
7081c7850f9SSasha Levin 
709b4422bf3SAneesh Kumar K.V 	strcpy(p9dev->root_dir, root);
7101c7850f9SSasha Levin 	root_len = strlen(root);
7111c7850f9SSasha Levin 	/*
7121c7850f9SSasha Levin 	 * We prefix the full path in all fids, This allows us to get the
7131c7850f9SSasha Levin 	 * absolute path of an fid without playing with strings.
7141c7850f9SSasha Levin 	 */
7151c7850f9SSasha Levin 	for (i = 0; i < VIRTIO_P9_MAX_FID; i++) {
716b4422bf3SAneesh Kumar K.V 		strcpy(p9dev->fids[i].abs_path, root);
717b4422bf3SAneesh Kumar K.V 		p9dev->fids[i].path = p9dev->fids[i].abs_path + root_len;
7181c7850f9SSasha Levin 	}
719b4422bf3SAneesh Kumar K.V 	p9dev->config->tag_len = strlen(tag_name);
720b4422bf3SAneesh Kumar K.V 	if (p9dev->config->tag_len > MAX_TAG_LEN)
721b4422bf3SAneesh Kumar K.V 		goto free_p9dev_config;
7221c7850f9SSasha Levin 
723b4422bf3SAneesh Kumar K.V 	memcpy(p9dev->config->tag, tag_name, strlen(tag_name));
724b4422bf3SAneesh Kumar K.V 	p9dev->features |= 1 << VIRTIO_9P_MOUNT_TAG;
7251c7850f9SSasha Levin 
7261c7850f9SSasha Levin 	if (irq__register_device(VIRTIO_ID_9P, &dev, &pin, &line) < 0)
727b4422bf3SAneesh Kumar K.V 		goto free_p9dev_config;
7281c7850f9SSasha Levin 
729b4422bf3SAneesh Kumar K.V 	p9_base_addr			= ioport__register(IOPORT_EMPTY,
730b4422bf3SAneesh Kumar K.V 							   &virtio_p9_io_ops,
731b4422bf3SAneesh Kumar K.V 							   IOPORT_SIZE, p9dev);
732b4422bf3SAneesh Kumar K.V 	p9dev->base_addr		    = p9_base_addr;
733b4422bf3SAneesh Kumar K.V 	p9dev->pci_hdr = (struct pci_device_header) {
734b4422bf3SAneesh Kumar K.V 		.vendor_id		= PCI_VENDOR_ID_REDHAT_QUMRANET,
735b4422bf3SAneesh Kumar K.V 		.device_id		= PCI_DEVICE_ID_VIRTIO_P9,
736b4422bf3SAneesh Kumar K.V 		.header_type		= PCI_HEADER_TYPE_NORMAL,
737b4422bf3SAneesh Kumar K.V 		.revision_id		= 0,
738b4422bf3SAneesh Kumar K.V 		.class			= 0x010000,
739b4422bf3SAneesh Kumar K.V 		.subsys_vendor_id	= PCI_SUBSYSTEM_VENDOR_ID_REDHAT_QUMRANET,
740b4422bf3SAneesh Kumar K.V 		.subsys_id		= VIRTIO_ID_9P,
741b4422bf3SAneesh Kumar K.V 		.irq_pin		= pin,
742b4422bf3SAneesh Kumar K.V 		.irq_line		= line,
743b4422bf3SAneesh Kumar K.V 		.bar[0]			= p9_base_addr | PCI_BASE_ADDRESS_SPACE_IO,
744b4422bf3SAneesh Kumar K.V 	};
745b4422bf3SAneesh Kumar K.V 	pci__register(&p9dev->pci_hdr, dev);
746b4422bf3SAneesh Kumar K.V 
747b4422bf3SAneesh Kumar K.V 	return;
748b4422bf3SAneesh Kumar K.V free_p9dev_config:
749b4422bf3SAneesh Kumar K.V 	free(p9dev->config);
750b4422bf3SAneesh Kumar K.V free_p9dev:
751b4422bf3SAneesh Kumar K.V 	free(p9dev);
7521c7850f9SSasha Levin }
753