xref: /kernel/linux/linux-6.6/drivers/xen/privcmd.c (revision 62306a36)
1// SPDX-License-Identifier: GPL-2.0-only
2/******************************************************************************
3 * privcmd.c
4 *
5 * Interface to privileged domain-0 commands.
6 *
7 * Copyright (c) 2002-2004, K A Fraser, B Dragovic
8 */
9
10#define pr_fmt(fmt) "xen:" KBUILD_MODNAME ": " fmt
11
12#include <linux/eventfd.h>
13#include <linux/file.h>
14#include <linux/kernel.h>
15#include <linux/module.h>
16#include <linux/mutex.h>
17#include <linux/poll.h>
18#include <linux/sched.h>
19#include <linux/slab.h>
20#include <linux/string.h>
21#include <linux/workqueue.h>
22#include <linux/errno.h>
23#include <linux/mm.h>
24#include <linux/mman.h>
25#include <linux/uaccess.h>
26#include <linux/swap.h>
27#include <linux/highmem.h>
28#include <linux/pagemap.h>
29#include <linux/seq_file.h>
30#include <linux/miscdevice.h>
31#include <linux/moduleparam.h>
32
33#include <asm/xen/hypervisor.h>
34#include <asm/xen/hypercall.h>
35
36#include <xen/xen.h>
37#include <xen/privcmd.h>
38#include <xen/interface/xen.h>
39#include <xen/interface/memory.h>
40#include <xen/interface/hvm/dm_op.h>
41#include <xen/features.h>
42#include <xen/page.h>
43#include <xen/xen-ops.h>
44#include <xen/balloon.h>
45
46#include "privcmd.h"
47
48MODULE_LICENSE("GPL");
49
50#define PRIV_VMA_LOCKED ((void *)1)
51
52static unsigned int privcmd_dm_op_max_num = 16;
53module_param_named(dm_op_max_nr_bufs, privcmd_dm_op_max_num, uint, 0644);
54MODULE_PARM_DESC(dm_op_max_nr_bufs,
55		 "Maximum number of buffers per dm_op hypercall");
56
57static unsigned int privcmd_dm_op_buf_max_size = 4096;
58module_param_named(dm_op_buf_max_size, privcmd_dm_op_buf_max_size, uint,
59		   0644);
60MODULE_PARM_DESC(dm_op_buf_max_size,
61		 "Maximum size of a dm_op hypercall buffer");
62
63struct privcmd_data {
64	domid_t domid;
65};
66
67static int privcmd_vma_range_is_mapped(
68               struct vm_area_struct *vma,
69               unsigned long addr,
70               unsigned long nr_pages);
71
72static long privcmd_ioctl_hypercall(struct file *file, void __user *udata)
73{
74	struct privcmd_data *data = file->private_data;
75	struct privcmd_hypercall hypercall;
76	long ret;
77
78	/* Disallow arbitrary hypercalls if restricted */
79	if (data->domid != DOMID_INVALID)
80		return -EPERM;
81
82	if (copy_from_user(&hypercall, udata, sizeof(hypercall)))
83		return -EFAULT;
84
85	xen_preemptible_hcall_begin();
86	ret = privcmd_call(hypercall.op,
87			   hypercall.arg[0], hypercall.arg[1],
88			   hypercall.arg[2], hypercall.arg[3],
89			   hypercall.arg[4]);
90	xen_preemptible_hcall_end();
91
92	return ret;
93}
94
95static void free_page_list(struct list_head *pages)
96{
97	struct page *p, *n;
98
99	list_for_each_entry_safe(p, n, pages, lru)
100		__free_page(p);
101
102	INIT_LIST_HEAD(pages);
103}
104
105/*
106 * Given an array of items in userspace, return a list of pages
107 * containing the data.  If copying fails, either because of memory
108 * allocation failure or a problem reading user memory, return an
109 * error code; its up to the caller to dispose of any partial list.
110 */
111static int gather_array(struct list_head *pagelist,
112			unsigned nelem, size_t size,
113			const void __user *data)
114{
115	unsigned pageidx;
116	void *pagedata;
117	int ret;
118
119	if (size > PAGE_SIZE)
120		return 0;
121
122	pageidx = PAGE_SIZE;
123	pagedata = NULL;	/* quiet, gcc */
124	while (nelem--) {
125		if (pageidx > PAGE_SIZE-size) {
126			struct page *page = alloc_page(GFP_KERNEL);
127
128			ret = -ENOMEM;
129			if (page == NULL)
130				goto fail;
131
132			pagedata = page_address(page);
133
134			list_add_tail(&page->lru, pagelist);
135			pageidx = 0;
136		}
137
138		ret = -EFAULT;
139		if (copy_from_user(pagedata + pageidx, data, size))
140			goto fail;
141
142		data += size;
143		pageidx += size;
144	}
145
146	ret = 0;
147
148fail:
149	return ret;
150}
151
152/*
153 * Call function "fn" on each element of the array fragmented
154 * over a list of pages.
155 */
156static int traverse_pages(unsigned nelem, size_t size,
157			  struct list_head *pos,
158			  int (*fn)(void *data, void *state),
159			  void *state)
160{
161	void *pagedata;
162	unsigned pageidx;
163	int ret = 0;
164
165	BUG_ON(size > PAGE_SIZE);
166
167	pageidx = PAGE_SIZE;
168	pagedata = NULL;	/* hush, gcc */
169
170	while (nelem--) {
171		if (pageidx > PAGE_SIZE-size) {
172			struct page *page;
173			pos = pos->next;
174			page = list_entry(pos, struct page, lru);
175			pagedata = page_address(page);
176			pageidx = 0;
177		}
178
179		ret = (*fn)(pagedata + pageidx, state);
180		if (ret)
181			break;
182		pageidx += size;
183	}
184
185	return ret;
186}
187
188/*
189 * Similar to traverse_pages, but use each page as a "block" of
190 * data to be processed as one unit.
191 */
192static int traverse_pages_block(unsigned nelem, size_t size,
193				struct list_head *pos,
194				int (*fn)(void *data, int nr, void *state),
195				void *state)
196{
197	void *pagedata;
198	int ret = 0;
199
200	BUG_ON(size > PAGE_SIZE);
201
202	while (nelem) {
203		int nr = (PAGE_SIZE/size);
204		struct page *page;
205		if (nr > nelem)
206			nr = nelem;
207		pos = pos->next;
208		page = list_entry(pos, struct page, lru);
209		pagedata = page_address(page);
210		ret = (*fn)(pagedata, nr, state);
211		if (ret)
212			break;
213		nelem -= nr;
214	}
215
216	return ret;
217}
218
219struct mmap_gfn_state {
220	unsigned long va;
221	struct vm_area_struct *vma;
222	domid_t domain;
223};
224
225static int mmap_gfn_range(void *data, void *state)
226{
227	struct privcmd_mmap_entry *msg = data;
228	struct mmap_gfn_state *st = state;
229	struct vm_area_struct *vma = st->vma;
230	int rc;
231
232	/* Do not allow range to wrap the address space. */
233	if ((msg->npages > (LONG_MAX >> PAGE_SHIFT)) ||
234	    ((unsigned long)(msg->npages << PAGE_SHIFT) >= -st->va))
235		return -EINVAL;
236
237	/* Range chunks must be contiguous in va space. */
238	if ((msg->va != st->va) ||
239	    ((msg->va+(msg->npages<<PAGE_SHIFT)) > vma->vm_end))
240		return -EINVAL;
241
242	rc = xen_remap_domain_gfn_range(vma,
243					msg->va & PAGE_MASK,
244					msg->mfn, msg->npages,
245					vma->vm_page_prot,
246					st->domain, NULL);
247	if (rc < 0)
248		return rc;
249
250	st->va += msg->npages << PAGE_SHIFT;
251
252	return 0;
253}
254
255static long privcmd_ioctl_mmap(struct file *file, void __user *udata)
256{
257	struct privcmd_data *data = file->private_data;
258	struct privcmd_mmap mmapcmd;
259	struct mm_struct *mm = current->mm;
260	struct vm_area_struct *vma;
261	int rc;
262	LIST_HEAD(pagelist);
263	struct mmap_gfn_state state;
264
265	/* We only support privcmd_ioctl_mmap_batch for non-auto-translated. */
266	if (xen_feature(XENFEAT_auto_translated_physmap))
267		return -ENOSYS;
268
269	if (copy_from_user(&mmapcmd, udata, sizeof(mmapcmd)))
270		return -EFAULT;
271
272	/* If restriction is in place, check the domid matches */
273	if (data->domid != DOMID_INVALID && data->domid != mmapcmd.dom)
274		return -EPERM;
275
276	rc = gather_array(&pagelist,
277			  mmapcmd.num, sizeof(struct privcmd_mmap_entry),
278			  mmapcmd.entry);
279
280	if (rc || list_empty(&pagelist))
281		goto out;
282
283	mmap_write_lock(mm);
284
285	{
286		struct page *page = list_first_entry(&pagelist,
287						     struct page, lru);
288		struct privcmd_mmap_entry *msg = page_address(page);
289
290		vma = vma_lookup(mm, msg->va);
291		rc = -EINVAL;
292
293		if (!vma || (msg->va != vma->vm_start) || vma->vm_private_data)
294			goto out_up;
295		vma->vm_private_data = PRIV_VMA_LOCKED;
296	}
297
298	state.va = vma->vm_start;
299	state.vma = vma;
300	state.domain = mmapcmd.dom;
301
302	rc = traverse_pages(mmapcmd.num, sizeof(struct privcmd_mmap_entry),
303			    &pagelist,
304			    mmap_gfn_range, &state);
305
306
307out_up:
308	mmap_write_unlock(mm);
309
310out:
311	free_page_list(&pagelist);
312
313	return rc;
314}
315
316struct mmap_batch_state {
317	domid_t domain;
318	unsigned long va;
319	struct vm_area_struct *vma;
320	int index;
321	/* A tristate:
322	 *      0 for no errors
323	 *      1 if at least one error has happened (and no
324	 *          -ENOENT errors have happened)
325	 *      -ENOENT if at least 1 -ENOENT has happened.
326	 */
327	int global_error;
328	int version;
329
330	/* User-space gfn array to store errors in the second pass for V1. */
331	xen_pfn_t __user *user_gfn;
332	/* User-space int array to store errors in the second pass for V2. */
333	int __user *user_err;
334};
335
336/* auto translated dom0 note: if domU being created is PV, then gfn is
337 * mfn(addr on bus). If it's auto xlated, then gfn is pfn (input to HAP).
338 */
339static int mmap_batch_fn(void *data, int nr, void *state)
340{
341	xen_pfn_t *gfnp = data;
342	struct mmap_batch_state *st = state;
343	struct vm_area_struct *vma = st->vma;
344	struct page **pages = vma->vm_private_data;
345	struct page **cur_pages = NULL;
346	int ret;
347
348	if (xen_feature(XENFEAT_auto_translated_physmap))
349		cur_pages = &pages[st->index];
350
351	BUG_ON(nr < 0);
352	ret = xen_remap_domain_gfn_array(st->vma, st->va & PAGE_MASK, gfnp, nr,
353					 (int *)gfnp, st->vma->vm_page_prot,
354					 st->domain, cur_pages);
355
356	/* Adjust the global_error? */
357	if (ret != nr) {
358		if (ret == -ENOENT)
359			st->global_error = -ENOENT;
360		else {
361			/* Record that at least one error has happened. */
362			if (st->global_error == 0)
363				st->global_error = 1;
364		}
365	}
366	st->va += XEN_PAGE_SIZE * nr;
367	st->index += nr / XEN_PFN_PER_PAGE;
368
369	return 0;
370}
371
372static int mmap_return_error(int err, struct mmap_batch_state *st)
373{
374	int ret;
375
376	if (st->version == 1) {
377		if (err) {
378			xen_pfn_t gfn;
379
380			ret = get_user(gfn, st->user_gfn);
381			if (ret < 0)
382				return ret;
383			/*
384			 * V1 encodes the error codes in the 32bit top
385			 * nibble of the gfn (with its known
386			 * limitations vis-a-vis 64 bit callers).
387			 */
388			gfn |= (err == -ENOENT) ?
389				PRIVCMD_MMAPBATCH_PAGED_ERROR :
390				PRIVCMD_MMAPBATCH_MFN_ERROR;
391			return __put_user(gfn, st->user_gfn++);
392		} else
393			st->user_gfn++;
394	} else { /* st->version == 2 */
395		if (err)
396			return __put_user(err, st->user_err++);
397		else
398			st->user_err++;
399	}
400
401	return 0;
402}
403
404static int mmap_return_errors(void *data, int nr, void *state)
405{
406	struct mmap_batch_state *st = state;
407	int *errs = data;
408	int i;
409	int ret;
410
411	for (i = 0; i < nr; i++) {
412		ret = mmap_return_error(errs[i], st);
413		if (ret < 0)
414			return ret;
415	}
416	return 0;
417}
418
419/* Allocate pfns that are then mapped with gfns from foreign domid. Update
420 * the vma with the page info to use later.
421 * Returns: 0 if success, otherwise -errno
422 */
423static int alloc_empty_pages(struct vm_area_struct *vma, int numpgs)
424{
425	int rc;
426	struct page **pages;
427
428	pages = kvcalloc(numpgs, sizeof(pages[0]), GFP_KERNEL);
429	if (pages == NULL)
430		return -ENOMEM;
431
432	rc = xen_alloc_unpopulated_pages(numpgs, pages);
433	if (rc != 0) {
434		pr_warn("%s Could not alloc %d pfns rc:%d\n", __func__,
435			numpgs, rc);
436		kvfree(pages);
437		return -ENOMEM;
438	}
439	BUG_ON(vma->vm_private_data != NULL);
440	vma->vm_private_data = pages;
441
442	return 0;
443}
444
445static const struct vm_operations_struct privcmd_vm_ops;
446
447static long privcmd_ioctl_mmap_batch(
448	struct file *file, void __user *udata, int version)
449{
450	struct privcmd_data *data = file->private_data;
451	int ret;
452	struct privcmd_mmapbatch_v2 m;
453	struct mm_struct *mm = current->mm;
454	struct vm_area_struct *vma;
455	unsigned long nr_pages;
456	LIST_HEAD(pagelist);
457	struct mmap_batch_state state;
458
459	switch (version) {
460	case 1:
461		if (copy_from_user(&m, udata, sizeof(struct privcmd_mmapbatch)))
462			return -EFAULT;
463		/* Returns per-frame error in m.arr. */
464		m.err = NULL;
465		if (!access_ok(m.arr, m.num * sizeof(*m.arr)))
466			return -EFAULT;
467		break;
468	case 2:
469		if (copy_from_user(&m, udata, sizeof(struct privcmd_mmapbatch_v2)))
470			return -EFAULT;
471		/* Returns per-frame error code in m.err. */
472		if (!access_ok(m.err, m.num * (sizeof(*m.err))))
473			return -EFAULT;
474		break;
475	default:
476		return -EINVAL;
477	}
478
479	/* If restriction is in place, check the domid matches */
480	if (data->domid != DOMID_INVALID && data->domid != m.dom)
481		return -EPERM;
482
483	nr_pages = DIV_ROUND_UP(m.num, XEN_PFN_PER_PAGE);
484	if ((m.num <= 0) || (nr_pages > (LONG_MAX >> PAGE_SHIFT)))
485		return -EINVAL;
486
487	ret = gather_array(&pagelist, m.num, sizeof(xen_pfn_t), m.arr);
488
489	if (ret)
490		goto out;
491	if (list_empty(&pagelist)) {
492		ret = -EINVAL;
493		goto out;
494	}
495
496	if (version == 2) {
497		/* Zero error array now to only copy back actual errors. */
498		if (clear_user(m.err, sizeof(int) * m.num)) {
499			ret = -EFAULT;
500			goto out;
501		}
502	}
503
504	mmap_write_lock(mm);
505
506	vma = find_vma(mm, m.addr);
507	if (!vma ||
508	    vma->vm_ops != &privcmd_vm_ops) {
509		ret = -EINVAL;
510		goto out_unlock;
511	}
512
513	/*
514	 * Caller must either:
515	 *
516	 * Map the whole VMA range, which will also allocate all the
517	 * pages required for the auto_translated_physmap case.
518	 *
519	 * Or
520	 *
521	 * Map unmapped holes left from a previous map attempt (e.g.,
522	 * because those foreign frames were previously paged out).
523	 */
524	if (vma->vm_private_data == NULL) {
525		if (m.addr != vma->vm_start ||
526		    m.addr + (nr_pages << PAGE_SHIFT) != vma->vm_end) {
527			ret = -EINVAL;
528			goto out_unlock;
529		}
530		if (xen_feature(XENFEAT_auto_translated_physmap)) {
531			ret = alloc_empty_pages(vma, nr_pages);
532			if (ret < 0)
533				goto out_unlock;
534		} else
535			vma->vm_private_data = PRIV_VMA_LOCKED;
536	} else {
537		if (m.addr < vma->vm_start ||
538		    m.addr + (nr_pages << PAGE_SHIFT) > vma->vm_end) {
539			ret = -EINVAL;
540			goto out_unlock;
541		}
542		if (privcmd_vma_range_is_mapped(vma, m.addr, nr_pages)) {
543			ret = -EINVAL;
544			goto out_unlock;
545		}
546	}
547
548	state.domain        = m.dom;
549	state.vma           = vma;
550	state.va            = m.addr;
551	state.index         = 0;
552	state.global_error  = 0;
553	state.version       = version;
554
555	BUILD_BUG_ON(((PAGE_SIZE / sizeof(xen_pfn_t)) % XEN_PFN_PER_PAGE) != 0);
556	/* mmap_batch_fn guarantees ret == 0 */
557	BUG_ON(traverse_pages_block(m.num, sizeof(xen_pfn_t),
558				    &pagelist, mmap_batch_fn, &state));
559
560	mmap_write_unlock(mm);
561
562	if (state.global_error) {
563		/* Write back errors in second pass. */
564		state.user_gfn = (xen_pfn_t *)m.arr;
565		state.user_err = m.err;
566		ret = traverse_pages_block(m.num, sizeof(xen_pfn_t),
567					   &pagelist, mmap_return_errors, &state);
568	} else
569		ret = 0;
570
571	/* If we have not had any EFAULT-like global errors then set the global
572	 * error to -ENOENT if necessary. */
573	if ((ret == 0) && (state.global_error == -ENOENT))
574		ret = -ENOENT;
575
576out:
577	free_page_list(&pagelist);
578	return ret;
579
580out_unlock:
581	mmap_write_unlock(mm);
582	goto out;
583}
584
585static int lock_pages(
586	struct privcmd_dm_op_buf kbufs[], unsigned int num,
587	struct page *pages[], unsigned int nr_pages, unsigned int *pinned)
588{
589	unsigned int i, off = 0;
590
591	for (i = 0; i < num; ) {
592		unsigned int requested;
593		int page_count;
594
595		requested = DIV_ROUND_UP(
596			offset_in_page(kbufs[i].uptr) + kbufs[i].size,
597			PAGE_SIZE) - off;
598		if (requested > nr_pages)
599			return -ENOSPC;
600
601		page_count = pin_user_pages_fast(
602			(unsigned long)kbufs[i].uptr + off * PAGE_SIZE,
603			requested, FOLL_WRITE, pages);
604		if (page_count <= 0)
605			return page_count ? : -EFAULT;
606
607		*pinned += page_count;
608		nr_pages -= page_count;
609		pages += page_count;
610
611		off = (requested == page_count) ? 0 : off + page_count;
612		i += !off;
613	}
614
615	return 0;
616}
617
618static void unlock_pages(struct page *pages[], unsigned int nr_pages)
619{
620	unpin_user_pages_dirty_lock(pages, nr_pages, true);
621}
622
623static long privcmd_ioctl_dm_op(struct file *file, void __user *udata)
624{
625	struct privcmd_data *data = file->private_data;
626	struct privcmd_dm_op kdata;
627	struct privcmd_dm_op_buf *kbufs;
628	unsigned int nr_pages = 0;
629	struct page **pages = NULL;
630	struct xen_dm_op_buf *xbufs = NULL;
631	unsigned int i;
632	long rc;
633	unsigned int pinned = 0;
634
635	if (copy_from_user(&kdata, udata, sizeof(kdata)))
636		return -EFAULT;
637
638	/* If restriction is in place, check the domid matches */
639	if (data->domid != DOMID_INVALID && data->domid != kdata.dom)
640		return -EPERM;
641
642	if (kdata.num == 0)
643		return 0;
644
645	if (kdata.num > privcmd_dm_op_max_num)
646		return -E2BIG;
647
648	kbufs = kcalloc(kdata.num, sizeof(*kbufs), GFP_KERNEL);
649	if (!kbufs)
650		return -ENOMEM;
651
652	if (copy_from_user(kbufs, kdata.ubufs,
653			   sizeof(*kbufs) * kdata.num)) {
654		rc = -EFAULT;
655		goto out;
656	}
657
658	for (i = 0; i < kdata.num; i++) {
659		if (kbufs[i].size > privcmd_dm_op_buf_max_size) {
660			rc = -E2BIG;
661			goto out;
662		}
663
664		if (!access_ok(kbufs[i].uptr,
665			       kbufs[i].size)) {
666			rc = -EFAULT;
667			goto out;
668		}
669
670		nr_pages += DIV_ROUND_UP(
671			offset_in_page(kbufs[i].uptr) + kbufs[i].size,
672			PAGE_SIZE);
673	}
674
675	pages = kcalloc(nr_pages, sizeof(*pages), GFP_KERNEL);
676	if (!pages) {
677		rc = -ENOMEM;
678		goto out;
679	}
680
681	xbufs = kcalloc(kdata.num, sizeof(*xbufs), GFP_KERNEL);
682	if (!xbufs) {
683		rc = -ENOMEM;
684		goto out;
685	}
686
687	rc = lock_pages(kbufs, kdata.num, pages, nr_pages, &pinned);
688	if (rc < 0)
689		goto out;
690
691	for (i = 0; i < kdata.num; i++) {
692		set_xen_guest_handle(xbufs[i].h, kbufs[i].uptr);
693		xbufs[i].size = kbufs[i].size;
694	}
695
696	xen_preemptible_hcall_begin();
697	rc = HYPERVISOR_dm_op(kdata.dom, kdata.num, xbufs);
698	xen_preemptible_hcall_end();
699
700out:
701	unlock_pages(pages, pinned);
702	kfree(xbufs);
703	kfree(pages);
704	kfree(kbufs);
705
706	return rc;
707}
708
709static long privcmd_ioctl_restrict(struct file *file, void __user *udata)
710{
711	struct privcmd_data *data = file->private_data;
712	domid_t dom;
713
714	if (copy_from_user(&dom, udata, sizeof(dom)))
715		return -EFAULT;
716
717	/* Set restriction to the specified domain, or check it matches */
718	if (data->domid == DOMID_INVALID)
719		data->domid = dom;
720	else if (data->domid != dom)
721		return -EINVAL;
722
723	return 0;
724}
725
726static long privcmd_ioctl_mmap_resource(struct file *file,
727				struct privcmd_mmap_resource __user *udata)
728{
729	struct privcmd_data *data = file->private_data;
730	struct mm_struct *mm = current->mm;
731	struct vm_area_struct *vma;
732	struct privcmd_mmap_resource kdata;
733	xen_pfn_t *pfns = NULL;
734	struct xen_mem_acquire_resource xdata = { };
735	int rc;
736
737	if (copy_from_user(&kdata, udata, sizeof(kdata)))
738		return -EFAULT;
739
740	/* If restriction is in place, check the domid matches */
741	if (data->domid != DOMID_INVALID && data->domid != kdata.dom)
742		return -EPERM;
743
744	/* Both fields must be set or unset */
745	if (!!kdata.addr != !!kdata.num)
746		return -EINVAL;
747
748	xdata.domid = kdata.dom;
749	xdata.type = kdata.type;
750	xdata.id = kdata.id;
751
752	if (!kdata.addr && !kdata.num) {
753		/* Query the size of the resource. */
754		rc = HYPERVISOR_memory_op(XENMEM_acquire_resource, &xdata);
755		if (rc)
756			return rc;
757		return __put_user(xdata.nr_frames, &udata->num);
758	}
759
760	mmap_write_lock(mm);
761
762	vma = find_vma(mm, kdata.addr);
763	if (!vma || vma->vm_ops != &privcmd_vm_ops) {
764		rc = -EINVAL;
765		goto out;
766	}
767
768	pfns = kcalloc(kdata.num, sizeof(*pfns), GFP_KERNEL | __GFP_NOWARN);
769	if (!pfns) {
770		rc = -ENOMEM;
771		goto out;
772	}
773
774	if (IS_ENABLED(CONFIG_XEN_AUTO_XLATE) &&
775	    xen_feature(XENFEAT_auto_translated_physmap)) {
776		unsigned int nr = DIV_ROUND_UP(kdata.num, XEN_PFN_PER_PAGE);
777		struct page **pages;
778		unsigned int i;
779
780		rc = alloc_empty_pages(vma, nr);
781		if (rc < 0)
782			goto out;
783
784		pages = vma->vm_private_data;
785		for (i = 0; i < kdata.num; i++) {
786			xen_pfn_t pfn =
787				page_to_xen_pfn(pages[i / XEN_PFN_PER_PAGE]);
788
789			pfns[i] = pfn + (i % XEN_PFN_PER_PAGE);
790		}
791	} else
792		vma->vm_private_data = PRIV_VMA_LOCKED;
793
794	xdata.frame = kdata.idx;
795	xdata.nr_frames = kdata.num;
796	set_xen_guest_handle(xdata.frame_list, pfns);
797
798	xen_preemptible_hcall_begin();
799	rc = HYPERVISOR_memory_op(XENMEM_acquire_resource, &xdata);
800	xen_preemptible_hcall_end();
801
802	if (rc)
803		goto out;
804
805	if (IS_ENABLED(CONFIG_XEN_AUTO_XLATE) &&
806	    xen_feature(XENFEAT_auto_translated_physmap)) {
807		rc = xen_remap_vma_range(vma, kdata.addr, kdata.num << PAGE_SHIFT);
808	} else {
809		unsigned int domid =
810			(xdata.flags & XENMEM_rsrc_acq_caller_owned) ?
811			DOMID_SELF : kdata.dom;
812		int num, *errs = (int *)pfns;
813
814		BUILD_BUG_ON(sizeof(*errs) > sizeof(*pfns));
815		num = xen_remap_domain_mfn_array(vma,
816						 kdata.addr & PAGE_MASK,
817						 pfns, kdata.num, errs,
818						 vma->vm_page_prot,
819						 domid);
820		if (num < 0)
821			rc = num;
822		else if (num != kdata.num) {
823			unsigned int i;
824
825			for (i = 0; i < num; i++) {
826				rc = errs[i];
827				if (rc < 0)
828					break;
829			}
830		} else
831			rc = 0;
832	}
833
834out:
835	mmap_write_unlock(mm);
836	kfree(pfns);
837
838	return rc;
839}
840
841#ifdef CONFIG_XEN_PRIVCMD_IRQFD
842/* Irqfd support */
843static struct workqueue_struct *irqfd_cleanup_wq;
844static DEFINE_MUTEX(irqfds_lock);
845static LIST_HEAD(irqfds_list);
846
847struct privcmd_kernel_irqfd {
848	struct xen_dm_op_buf xbufs;
849	domid_t dom;
850	bool error;
851	struct eventfd_ctx *eventfd;
852	struct work_struct shutdown;
853	wait_queue_entry_t wait;
854	struct list_head list;
855	poll_table pt;
856};
857
858static void irqfd_deactivate(struct privcmd_kernel_irqfd *kirqfd)
859{
860	lockdep_assert_held(&irqfds_lock);
861
862	list_del_init(&kirqfd->list);
863	queue_work(irqfd_cleanup_wq, &kirqfd->shutdown);
864}
865
866static void irqfd_shutdown(struct work_struct *work)
867{
868	struct privcmd_kernel_irqfd *kirqfd =
869		container_of(work, struct privcmd_kernel_irqfd, shutdown);
870	u64 cnt;
871
872	eventfd_ctx_remove_wait_queue(kirqfd->eventfd, &kirqfd->wait, &cnt);
873	eventfd_ctx_put(kirqfd->eventfd);
874	kfree(kirqfd);
875}
876
877static void irqfd_inject(struct privcmd_kernel_irqfd *kirqfd)
878{
879	u64 cnt;
880	long rc;
881
882	eventfd_ctx_do_read(kirqfd->eventfd, &cnt);
883
884	xen_preemptible_hcall_begin();
885	rc = HYPERVISOR_dm_op(kirqfd->dom, 1, &kirqfd->xbufs);
886	xen_preemptible_hcall_end();
887
888	/* Don't repeat the error message for consecutive failures */
889	if (rc && !kirqfd->error) {
890		pr_err("Failed to configure irq for guest domain: %d\n",
891		       kirqfd->dom);
892	}
893
894	kirqfd->error = rc;
895}
896
897static int
898irqfd_wakeup(wait_queue_entry_t *wait, unsigned int mode, int sync, void *key)
899{
900	struct privcmd_kernel_irqfd *kirqfd =
901		container_of(wait, struct privcmd_kernel_irqfd, wait);
902	__poll_t flags = key_to_poll(key);
903
904	if (flags & EPOLLIN)
905		irqfd_inject(kirqfd);
906
907	if (flags & EPOLLHUP) {
908		mutex_lock(&irqfds_lock);
909		irqfd_deactivate(kirqfd);
910		mutex_unlock(&irqfds_lock);
911	}
912
913	return 0;
914}
915
916static void
917irqfd_poll_func(struct file *file, wait_queue_head_t *wqh, poll_table *pt)
918{
919	struct privcmd_kernel_irqfd *kirqfd =
920		container_of(pt, struct privcmd_kernel_irqfd, pt);
921
922	add_wait_queue_priority(wqh, &kirqfd->wait);
923}
924
925static int privcmd_irqfd_assign(struct privcmd_irqfd *irqfd)
926{
927	struct privcmd_kernel_irqfd *kirqfd, *tmp;
928	__poll_t events;
929	struct fd f;
930	void *dm_op;
931	int ret;
932
933	kirqfd = kzalloc(sizeof(*kirqfd) + irqfd->size, GFP_KERNEL);
934	if (!kirqfd)
935		return -ENOMEM;
936	dm_op = kirqfd + 1;
937
938	if (copy_from_user(dm_op, u64_to_user_ptr(irqfd->dm_op), irqfd->size)) {
939		ret = -EFAULT;
940		goto error_kfree;
941	}
942
943	kirqfd->xbufs.size = irqfd->size;
944	set_xen_guest_handle(kirqfd->xbufs.h, dm_op);
945	kirqfd->dom = irqfd->dom;
946	INIT_WORK(&kirqfd->shutdown, irqfd_shutdown);
947
948	f = fdget(irqfd->fd);
949	if (!f.file) {
950		ret = -EBADF;
951		goto error_kfree;
952	}
953
954	kirqfd->eventfd = eventfd_ctx_fileget(f.file);
955	if (IS_ERR(kirqfd->eventfd)) {
956		ret = PTR_ERR(kirqfd->eventfd);
957		goto error_fd_put;
958	}
959
960	/*
961	 * Install our own custom wake-up handling so we are notified via a
962	 * callback whenever someone signals the underlying eventfd.
963	 */
964	init_waitqueue_func_entry(&kirqfd->wait, irqfd_wakeup);
965	init_poll_funcptr(&kirqfd->pt, irqfd_poll_func);
966
967	mutex_lock(&irqfds_lock);
968
969	list_for_each_entry(tmp, &irqfds_list, list) {
970		if (kirqfd->eventfd == tmp->eventfd) {
971			ret = -EBUSY;
972			mutex_unlock(&irqfds_lock);
973			goto error_eventfd;
974		}
975	}
976
977	list_add_tail(&kirqfd->list, &irqfds_list);
978	mutex_unlock(&irqfds_lock);
979
980	/*
981	 * Check if there was an event already pending on the eventfd before we
982	 * registered, and trigger it as if we didn't miss it.
983	 */
984	events = vfs_poll(f.file, &kirqfd->pt);
985	if (events & EPOLLIN)
986		irqfd_inject(kirqfd);
987
988	/*
989	 * Do not drop the file until the kirqfd is fully initialized, otherwise
990	 * we might race against the EPOLLHUP.
991	 */
992	fdput(f);
993	return 0;
994
995error_eventfd:
996	eventfd_ctx_put(kirqfd->eventfd);
997
998error_fd_put:
999	fdput(f);
1000
1001error_kfree:
1002	kfree(kirqfd);
1003	return ret;
1004}
1005
1006static int privcmd_irqfd_deassign(struct privcmd_irqfd *irqfd)
1007{
1008	struct privcmd_kernel_irqfd *kirqfd;
1009	struct eventfd_ctx *eventfd;
1010
1011	eventfd = eventfd_ctx_fdget(irqfd->fd);
1012	if (IS_ERR(eventfd))
1013		return PTR_ERR(eventfd);
1014
1015	mutex_lock(&irqfds_lock);
1016
1017	list_for_each_entry(kirqfd, &irqfds_list, list) {
1018		if (kirqfd->eventfd == eventfd) {
1019			irqfd_deactivate(kirqfd);
1020			break;
1021		}
1022	}
1023
1024	mutex_unlock(&irqfds_lock);
1025
1026	eventfd_ctx_put(eventfd);
1027
1028	/*
1029	 * Block until we know all outstanding shutdown jobs have completed so
1030	 * that we guarantee there will not be any more interrupts once this
1031	 * deassign function returns.
1032	 */
1033	flush_workqueue(irqfd_cleanup_wq);
1034
1035	return 0;
1036}
1037
1038static long privcmd_ioctl_irqfd(struct file *file, void __user *udata)
1039{
1040	struct privcmd_data *data = file->private_data;
1041	struct privcmd_irqfd irqfd;
1042
1043	if (copy_from_user(&irqfd, udata, sizeof(irqfd)))
1044		return -EFAULT;
1045
1046	/* No other flags should be set */
1047	if (irqfd.flags & ~PRIVCMD_IRQFD_FLAG_DEASSIGN)
1048		return -EINVAL;
1049
1050	/* If restriction is in place, check the domid matches */
1051	if (data->domid != DOMID_INVALID && data->domid != irqfd.dom)
1052		return -EPERM;
1053
1054	if (irqfd.flags & PRIVCMD_IRQFD_FLAG_DEASSIGN)
1055		return privcmd_irqfd_deassign(&irqfd);
1056
1057	return privcmd_irqfd_assign(&irqfd);
1058}
1059
1060static int privcmd_irqfd_init(void)
1061{
1062	irqfd_cleanup_wq = alloc_workqueue("privcmd-irqfd-cleanup", 0, 0);
1063	if (!irqfd_cleanup_wq)
1064		return -ENOMEM;
1065
1066	return 0;
1067}
1068
1069static void privcmd_irqfd_exit(void)
1070{
1071	struct privcmd_kernel_irqfd *kirqfd, *tmp;
1072
1073	mutex_lock(&irqfds_lock);
1074
1075	list_for_each_entry_safe(kirqfd, tmp, &irqfds_list, list)
1076		irqfd_deactivate(kirqfd);
1077
1078	mutex_unlock(&irqfds_lock);
1079
1080	destroy_workqueue(irqfd_cleanup_wq);
1081}
1082#else
1083static inline long privcmd_ioctl_irqfd(struct file *file, void __user *udata)
1084{
1085	return -EOPNOTSUPP;
1086}
1087
1088static inline int privcmd_irqfd_init(void)
1089{
1090	return 0;
1091}
1092
1093static inline void privcmd_irqfd_exit(void)
1094{
1095}
1096#endif /* CONFIG_XEN_PRIVCMD_IRQFD */
1097
1098static long privcmd_ioctl(struct file *file,
1099			  unsigned int cmd, unsigned long data)
1100{
1101	int ret = -ENOTTY;
1102	void __user *udata = (void __user *) data;
1103
1104	switch (cmd) {
1105	case IOCTL_PRIVCMD_HYPERCALL:
1106		ret = privcmd_ioctl_hypercall(file, udata);
1107		break;
1108
1109	case IOCTL_PRIVCMD_MMAP:
1110		ret = privcmd_ioctl_mmap(file, udata);
1111		break;
1112
1113	case IOCTL_PRIVCMD_MMAPBATCH:
1114		ret = privcmd_ioctl_mmap_batch(file, udata, 1);
1115		break;
1116
1117	case IOCTL_PRIVCMD_MMAPBATCH_V2:
1118		ret = privcmd_ioctl_mmap_batch(file, udata, 2);
1119		break;
1120
1121	case IOCTL_PRIVCMD_DM_OP:
1122		ret = privcmd_ioctl_dm_op(file, udata);
1123		break;
1124
1125	case IOCTL_PRIVCMD_RESTRICT:
1126		ret = privcmd_ioctl_restrict(file, udata);
1127		break;
1128
1129	case IOCTL_PRIVCMD_MMAP_RESOURCE:
1130		ret = privcmd_ioctl_mmap_resource(file, udata);
1131		break;
1132
1133	case IOCTL_PRIVCMD_IRQFD:
1134		ret = privcmd_ioctl_irqfd(file, udata);
1135		break;
1136
1137	default:
1138		break;
1139	}
1140
1141	return ret;
1142}
1143
1144static int privcmd_open(struct inode *ino, struct file *file)
1145{
1146	struct privcmd_data *data = kzalloc(sizeof(*data), GFP_KERNEL);
1147
1148	if (!data)
1149		return -ENOMEM;
1150
1151	/* DOMID_INVALID implies no restriction */
1152	data->domid = DOMID_INVALID;
1153
1154	file->private_data = data;
1155	return 0;
1156}
1157
1158static int privcmd_release(struct inode *ino, struct file *file)
1159{
1160	struct privcmd_data *data = file->private_data;
1161
1162	kfree(data);
1163	return 0;
1164}
1165
1166static void privcmd_close(struct vm_area_struct *vma)
1167{
1168	struct page **pages = vma->vm_private_data;
1169	int numpgs = vma_pages(vma);
1170	int numgfns = (vma->vm_end - vma->vm_start) >> XEN_PAGE_SHIFT;
1171	int rc;
1172
1173	if (!xen_feature(XENFEAT_auto_translated_physmap) || !numpgs || !pages)
1174		return;
1175
1176	rc = xen_unmap_domain_gfn_range(vma, numgfns, pages);
1177	if (rc == 0)
1178		xen_free_unpopulated_pages(numpgs, pages);
1179	else
1180		pr_crit("unable to unmap MFN range: leaking %d pages. rc=%d\n",
1181			numpgs, rc);
1182	kvfree(pages);
1183}
1184
1185static vm_fault_t privcmd_fault(struct vm_fault *vmf)
1186{
1187	printk(KERN_DEBUG "privcmd_fault: vma=%p %lx-%lx, pgoff=%lx, uv=%p\n",
1188	       vmf->vma, vmf->vma->vm_start, vmf->vma->vm_end,
1189	       vmf->pgoff, (void *)vmf->address);
1190
1191	return VM_FAULT_SIGBUS;
1192}
1193
1194static const struct vm_operations_struct privcmd_vm_ops = {
1195	.close = privcmd_close,
1196	.fault = privcmd_fault
1197};
1198
1199static int privcmd_mmap(struct file *file, struct vm_area_struct *vma)
1200{
1201	/* DONTCOPY is essential for Xen because copy_page_range doesn't know
1202	 * how to recreate these mappings */
1203	vm_flags_set(vma, VM_IO | VM_PFNMAP | VM_DONTCOPY |
1204			 VM_DONTEXPAND | VM_DONTDUMP);
1205	vma->vm_ops = &privcmd_vm_ops;
1206	vma->vm_private_data = NULL;
1207
1208	return 0;
1209}
1210
1211/*
1212 * For MMAPBATCH*. This allows asserting the singleshot mapping
1213 * on a per pfn/pte basis. Mapping calls that fail with ENOENT
1214 * can be then retried until success.
1215 */
1216static int is_mapped_fn(pte_t *pte, unsigned long addr, void *data)
1217{
1218	return pte_none(ptep_get(pte)) ? 0 : -EBUSY;
1219}
1220
1221static int privcmd_vma_range_is_mapped(
1222	           struct vm_area_struct *vma,
1223	           unsigned long addr,
1224	           unsigned long nr_pages)
1225{
1226	return apply_to_page_range(vma->vm_mm, addr, nr_pages << PAGE_SHIFT,
1227				   is_mapped_fn, NULL) != 0;
1228}
1229
1230const struct file_operations xen_privcmd_fops = {
1231	.owner = THIS_MODULE,
1232	.unlocked_ioctl = privcmd_ioctl,
1233	.open = privcmd_open,
1234	.release = privcmd_release,
1235	.mmap = privcmd_mmap,
1236};
1237EXPORT_SYMBOL_GPL(xen_privcmd_fops);
1238
1239static struct miscdevice privcmd_dev = {
1240	.minor = MISC_DYNAMIC_MINOR,
1241	.name = "xen/privcmd",
1242	.fops = &xen_privcmd_fops,
1243};
1244
1245static int __init privcmd_init(void)
1246{
1247	int err;
1248
1249	if (!xen_domain())
1250		return -ENODEV;
1251
1252	err = misc_register(&privcmd_dev);
1253	if (err != 0) {
1254		pr_err("Could not register Xen privcmd device\n");
1255		return err;
1256	}
1257
1258	err = misc_register(&xen_privcmdbuf_dev);
1259	if (err != 0) {
1260		pr_err("Could not register Xen hypercall-buf device\n");
1261		goto err_privcmdbuf;
1262	}
1263
1264	err = privcmd_irqfd_init();
1265	if (err != 0) {
1266		pr_err("irqfd init failed\n");
1267		goto err_irqfd;
1268	}
1269
1270	return 0;
1271
1272err_irqfd:
1273	misc_deregister(&xen_privcmdbuf_dev);
1274err_privcmdbuf:
1275	misc_deregister(&privcmd_dev);
1276	return err;
1277}
1278
1279static void __exit privcmd_exit(void)
1280{
1281	privcmd_irqfd_exit();
1282	misc_deregister(&privcmd_dev);
1283	misc_deregister(&xen_privcmdbuf_dev);
1284}
1285
1286module_init(privcmd_init);
1287module_exit(privcmd_exit);
1288