xref: /openbmc/linux/drivers/xen/privcmd.c (revision d2c43ff1)
1 /******************************************************************************
2  * privcmd.c
3  *
4  * Interface to privileged domain-0 commands.
5  *
6  * Copyright (c) 2002-2004, K A Fraser, B Dragovic
7  */
8 
9 #define pr_fmt(fmt) "xen:" KBUILD_MODNAME ": " fmt
10 
11 #include <linux/kernel.h>
12 #include <linux/module.h>
13 #include <linux/sched.h>
14 #include <linux/slab.h>
15 #include <linux/string.h>
16 #include <linux/errno.h>
17 #include <linux/mm.h>
18 #include <linux/mman.h>
19 #include <linux/uaccess.h>
20 #include <linux/swap.h>
21 #include <linux/highmem.h>
22 #include <linux/pagemap.h>
23 #include <linux/seq_file.h>
24 #include <linux/miscdevice.h>
25 #include <linux/moduleparam.h>
26 
27 #include <asm/pgalloc.h>
28 #include <asm/pgtable.h>
29 #include <asm/tlb.h>
30 #include <asm/xen/hypervisor.h>
31 #include <asm/xen/hypercall.h>
32 
33 #include <xen/xen.h>
34 #include <xen/privcmd.h>
35 #include <xen/interface/xen.h>
36 #include <xen/interface/hvm/dm_op.h>
37 #include <xen/features.h>
38 #include <xen/page.h>
39 #include <xen/xen-ops.h>
40 #include <xen/balloon.h>
41 
42 #include "privcmd.h"
43 
44 MODULE_LICENSE("GPL");
45 
46 #define PRIV_VMA_LOCKED ((void *)1)
47 
48 static unsigned int privcmd_dm_op_max_num = 16;
49 module_param_named(dm_op_max_nr_bufs, privcmd_dm_op_max_num, uint, 0644);
50 MODULE_PARM_DESC(dm_op_max_nr_bufs,
51 		 "Maximum number of buffers per dm_op hypercall");
52 
53 static unsigned int privcmd_dm_op_buf_max_size = 4096;
54 module_param_named(dm_op_buf_max_size, privcmd_dm_op_buf_max_size, uint,
55 		   0644);
56 MODULE_PARM_DESC(dm_op_buf_max_size,
57 		 "Maximum size of a dm_op hypercall buffer");
58 
59 struct privcmd_data {
60 	domid_t domid;
61 };
62 
63 static int privcmd_vma_range_is_mapped(
64                struct vm_area_struct *vma,
65                unsigned long addr,
66                unsigned long nr_pages);
67 
68 static long privcmd_ioctl_hypercall(struct file *file, void __user *udata)
69 {
70 	struct privcmd_data *data = file->private_data;
71 	struct privcmd_hypercall hypercall;
72 	long ret;
73 
74 	/* Disallow arbitrary hypercalls if restricted */
75 	if (data->domid != DOMID_INVALID)
76 		return -EPERM;
77 
78 	if (copy_from_user(&hypercall, udata, sizeof(hypercall)))
79 		return -EFAULT;
80 
81 	xen_preemptible_hcall_begin();
82 	ret = privcmd_call(hypercall.op,
83 			   hypercall.arg[0], hypercall.arg[1],
84 			   hypercall.arg[2], hypercall.arg[3],
85 			   hypercall.arg[4]);
86 	xen_preemptible_hcall_end();
87 
88 	return ret;
89 }
90 
91 static void free_page_list(struct list_head *pages)
92 {
93 	struct page *p, *n;
94 
95 	list_for_each_entry_safe(p, n, pages, lru)
96 		__free_page(p);
97 
98 	INIT_LIST_HEAD(pages);
99 }
100 
101 /*
102  * Given an array of items in userspace, return a list of pages
103  * containing the data.  If copying fails, either because of memory
104  * allocation failure or a problem reading user memory, return an
105  * error code; its up to the caller to dispose of any partial list.
106  */
107 static int gather_array(struct list_head *pagelist,
108 			unsigned nelem, size_t size,
109 			const void __user *data)
110 {
111 	unsigned pageidx;
112 	void *pagedata;
113 	int ret;
114 
115 	if (size > PAGE_SIZE)
116 		return 0;
117 
118 	pageidx = PAGE_SIZE;
119 	pagedata = NULL;	/* quiet, gcc */
120 	while (nelem--) {
121 		if (pageidx > PAGE_SIZE-size) {
122 			struct page *page = alloc_page(GFP_KERNEL);
123 
124 			ret = -ENOMEM;
125 			if (page == NULL)
126 				goto fail;
127 
128 			pagedata = page_address(page);
129 
130 			list_add_tail(&page->lru, pagelist);
131 			pageidx = 0;
132 		}
133 
134 		ret = -EFAULT;
135 		if (copy_from_user(pagedata + pageidx, data, size))
136 			goto fail;
137 
138 		data += size;
139 		pageidx += size;
140 	}
141 
142 	ret = 0;
143 
144 fail:
145 	return ret;
146 }
147 
148 /*
149  * Call function "fn" on each element of the array fragmented
150  * over a list of pages.
151  */
152 static int traverse_pages(unsigned nelem, size_t size,
153 			  struct list_head *pos,
154 			  int (*fn)(void *data, void *state),
155 			  void *state)
156 {
157 	void *pagedata;
158 	unsigned pageidx;
159 	int ret = 0;
160 
161 	BUG_ON(size > PAGE_SIZE);
162 
163 	pageidx = PAGE_SIZE;
164 	pagedata = NULL;	/* hush, gcc */
165 
166 	while (nelem--) {
167 		if (pageidx > PAGE_SIZE-size) {
168 			struct page *page;
169 			pos = pos->next;
170 			page = list_entry(pos, struct page, lru);
171 			pagedata = page_address(page);
172 			pageidx = 0;
173 		}
174 
175 		ret = (*fn)(pagedata + pageidx, state);
176 		if (ret)
177 			break;
178 		pageidx += size;
179 	}
180 
181 	return ret;
182 }
183 
184 /*
185  * Similar to traverse_pages, but use each page as a "block" of
186  * data to be processed as one unit.
187  */
188 static int traverse_pages_block(unsigned nelem, size_t size,
189 				struct list_head *pos,
190 				int (*fn)(void *data, int nr, void *state),
191 				void *state)
192 {
193 	void *pagedata;
194 	unsigned pageidx;
195 	int ret = 0;
196 
197 	BUG_ON(size > PAGE_SIZE);
198 
199 	pageidx = PAGE_SIZE;
200 
201 	while (nelem) {
202 		int nr = (PAGE_SIZE/size);
203 		struct page *page;
204 		if (nr > nelem)
205 			nr = nelem;
206 		pos = pos->next;
207 		page = list_entry(pos, struct page, lru);
208 		pagedata = page_address(page);
209 		ret = (*fn)(pagedata, nr, state);
210 		if (ret)
211 			break;
212 		nelem -= nr;
213 	}
214 
215 	return ret;
216 }
217 
218 struct mmap_gfn_state {
219 	unsigned long va;
220 	struct vm_area_struct *vma;
221 	domid_t domain;
222 };
223 
224 static int mmap_gfn_range(void *data, void *state)
225 {
226 	struct privcmd_mmap_entry *msg = data;
227 	struct mmap_gfn_state *st = state;
228 	struct vm_area_struct *vma = st->vma;
229 	int rc;
230 
231 	/* Do not allow range to wrap the address space. */
232 	if ((msg->npages > (LONG_MAX >> PAGE_SHIFT)) ||
233 	    ((unsigned long)(msg->npages << PAGE_SHIFT) >= -st->va))
234 		return -EINVAL;
235 
236 	/* Range chunks must be contiguous in va space. */
237 	if ((msg->va != st->va) ||
238 	    ((msg->va+(msg->npages<<PAGE_SHIFT)) > vma->vm_end))
239 		return -EINVAL;
240 
241 	rc = xen_remap_domain_gfn_range(vma,
242 					msg->va & PAGE_MASK,
243 					msg->mfn, msg->npages,
244 					vma->vm_page_prot,
245 					st->domain, NULL);
246 	if (rc < 0)
247 		return rc;
248 
249 	st->va += msg->npages << PAGE_SHIFT;
250 
251 	return 0;
252 }
253 
254 static long privcmd_ioctl_mmap(struct file *file, void __user *udata)
255 {
256 	struct privcmd_data *data = file->private_data;
257 	struct privcmd_mmap mmapcmd;
258 	struct mm_struct *mm = current->mm;
259 	struct vm_area_struct *vma;
260 	int rc;
261 	LIST_HEAD(pagelist);
262 	struct mmap_gfn_state state;
263 
264 	/* We only support privcmd_ioctl_mmap_batch for auto translated. */
265 	if (xen_feature(XENFEAT_auto_translated_physmap))
266 		return -ENOSYS;
267 
268 	if (copy_from_user(&mmapcmd, udata, sizeof(mmapcmd)))
269 		return -EFAULT;
270 
271 	/* If restriction is in place, check the domid matches */
272 	if (data->domid != DOMID_INVALID && data->domid != mmapcmd.dom)
273 		return -EPERM;
274 
275 	rc = gather_array(&pagelist,
276 			  mmapcmd.num, sizeof(struct privcmd_mmap_entry),
277 			  mmapcmd.entry);
278 
279 	if (rc || list_empty(&pagelist))
280 		goto out;
281 
282 	down_write(&mm->mmap_sem);
283 
284 	{
285 		struct page *page = list_first_entry(&pagelist,
286 						     struct page, lru);
287 		struct privcmd_mmap_entry *msg = page_address(page);
288 
289 		vma = find_vma(mm, msg->va);
290 		rc = -EINVAL;
291 
292 		if (!vma || (msg->va != vma->vm_start) || vma->vm_private_data)
293 			goto out_up;
294 		vma->vm_private_data = PRIV_VMA_LOCKED;
295 	}
296 
297 	state.va = vma->vm_start;
298 	state.vma = vma;
299 	state.domain = mmapcmd.dom;
300 
301 	rc = traverse_pages(mmapcmd.num, sizeof(struct privcmd_mmap_entry),
302 			    &pagelist,
303 			    mmap_gfn_range, &state);
304 
305 
306 out_up:
307 	up_write(&mm->mmap_sem);
308 
309 out:
310 	free_page_list(&pagelist);
311 
312 	return rc;
313 }
314 
315 struct mmap_batch_state {
316 	domid_t domain;
317 	unsigned long va;
318 	struct vm_area_struct *vma;
319 	int index;
320 	/* A tristate:
321 	 *      0 for no errors
322 	 *      1 if at least one error has happened (and no
323 	 *          -ENOENT errors have happened)
324 	 *      -ENOENT if at least 1 -ENOENT has happened.
325 	 */
326 	int global_error;
327 	int version;
328 
329 	/* User-space gfn array to store errors in the second pass for V1. */
330 	xen_pfn_t __user *user_gfn;
331 	/* User-space int array to store errors in the second pass for V2. */
332 	int __user *user_err;
333 };
334 
335 /* auto translated dom0 note: if domU being created is PV, then gfn is
336  * mfn(addr on bus). If it's auto xlated, then gfn is pfn (input to HAP).
337  */
338 static int mmap_batch_fn(void *data, int nr, void *state)
339 {
340 	xen_pfn_t *gfnp = data;
341 	struct mmap_batch_state *st = state;
342 	struct vm_area_struct *vma = st->vma;
343 	struct page **pages = vma->vm_private_data;
344 	struct page **cur_pages = NULL;
345 	int ret;
346 
347 	if (xen_feature(XENFEAT_auto_translated_physmap))
348 		cur_pages = &pages[st->index];
349 
350 	BUG_ON(nr < 0);
351 	ret = xen_remap_domain_gfn_array(st->vma, st->va & PAGE_MASK, gfnp, nr,
352 					 (int *)gfnp, st->vma->vm_page_prot,
353 					 st->domain, cur_pages);
354 
355 	/* Adjust the global_error? */
356 	if (ret != nr) {
357 		if (ret == -ENOENT)
358 			st->global_error = -ENOENT;
359 		else {
360 			/* Record that at least one error has happened. */
361 			if (st->global_error == 0)
362 				st->global_error = 1;
363 		}
364 	}
365 	st->va += XEN_PAGE_SIZE * nr;
366 	st->index += nr / XEN_PFN_PER_PAGE;
367 
368 	return 0;
369 }
370 
371 static int mmap_return_error(int err, struct mmap_batch_state *st)
372 {
373 	int ret;
374 
375 	if (st->version == 1) {
376 		if (err) {
377 			xen_pfn_t gfn;
378 
379 			ret = get_user(gfn, st->user_gfn);
380 			if (ret < 0)
381 				return ret;
382 			/*
383 			 * V1 encodes the error codes in the 32bit top
384 			 * nibble of the gfn (with its known
385 			 * limitations vis-a-vis 64 bit callers).
386 			 */
387 			gfn |= (err == -ENOENT) ?
388 				PRIVCMD_MMAPBATCH_PAGED_ERROR :
389 				PRIVCMD_MMAPBATCH_MFN_ERROR;
390 			return __put_user(gfn, st->user_gfn++);
391 		} else
392 			st->user_gfn++;
393 	} else { /* st->version == 2 */
394 		if (err)
395 			return __put_user(err, st->user_err++);
396 		else
397 			st->user_err++;
398 	}
399 
400 	return 0;
401 }
402 
403 static int mmap_return_errors(void *data, int nr, void *state)
404 {
405 	struct mmap_batch_state *st = state;
406 	int *errs = data;
407 	int i;
408 	int ret;
409 
410 	for (i = 0; i < nr; i++) {
411 		ret = mmap_return_error(errs[i], st);
412 		if (ret < 0)
413 			return ret;
414 	}
415 	return 0;
416 }
417 
418 /* Allocate pfns that are then mapped with gfns from foreign domid. Update
419  * the vma with the page info to use later.
420  * Returns: 0 if success, otherwise -errno
421  */
422 static int alloc_empty_pages(struct vm_area_struct *vma, int numpgs)
423 {
424 	int rc;
425 	struct page **pages;
426 
427 	pages = kcalloc(numpgs, sizeof(pages[0]), GFP_KERNEL);
428 	if (pages == NULL)
429 		return -ENOMEM;
430 
431 	rc = alloc_xenballooned_pages(numpgs, pages);
432 	if (rc != 0) {
433 		pr_warn("%s Could not alloc %d pfns rc:%d\n", __func__,
434 			numpgs, rc);
435 		kfree(pages);
436 		return -ENOMEM;
437 	}
438 	BUG_ON(vma->vm_private_data != NULL);
439 	vma->vm_private_data = pages;
440 
441 	return 0;
442 }
443 
444 static const struct vm_operations_struct privcmd_vm_ops;
445 
446 static long privcmd_ioctl_mmap_batch(
447 	struct file *file, void __user *udata, int version)
448 {
449 	struct privcmd_data *data = file->private_data;
450 	int ret;
451 	struct privcmd_mmapbatch_v2 m;
452 	struct mm_struct *mm = current->mm;
453 	struct vm_area_struct *vma;
454 	unsigned long nr_pages;
455 	LIST_HEAD(pagelist);
456 	struct mmap_batch_state state;
457 
458 	switch (version) {
459 	case 1:
460 		if (copy_from_user(&m, udata, sizeof(struct privcmd_mmapbatch)))
461 			return -EFAULT;
462 		/* Returns per-frame error in m.arr. */
463 		m.err = NULL;
464 		if (!access_ok(VERIFY_WRITE, m.arr, m.num * sizeof(*m.arr)))
465 			return -EFAULT;
466 		break;
467 	case 2:
468 		if (copy_from_user(&m, udata, sizeof(struct privcmd_mmapbatch_v2)))
469 			return -EFAULT;
470 		/* Returns per-frame error code in m.err. */
471 		if (!access_ok(VERIFY_WRITE, m.err, m.num * (sizeof(*m.err))))
472 			return -EFAULT;
473 		break;
474 	default:
475 		return -EINVAL;
476 	}
477 
478 	/* If restriction is in place, check the domid matches */
479 	if (data->domid != DOMID_INVALID && data->domid != m.dom)
480 		return -EPERM;
481 
482 	nr_pages = DIV_ROUND_UP(m.num, XEN_PFN_PER_PAGE);
483 	if ((m.num <= 0) || (nr_pages > (LONG_MAX >> PAGE_SHIFT)))
484 		return -EINVAL;
485 
486 	ret = gather_array(&pagelist, m.num, sizeof(xen_pfn_t), m.arr);
487 
488 	if (ret)
489 		goto out;
490 	if (list_empty(&pagelist)) {
491 		ret = -EINVAL;
492 		goto out;
493 	}
494 
495 	if (version == 2) {
496 		/* Zero error array now to only copy back actual errors. */
497 		if (clear_user(m.err, sizeof(int) * m.num)) {
498 			ret = -EFAULT;
499 			goto out;
500 		}
501 	}
502 
503 	down_write(&mm->mmap_sem);
504 
505 	vma = find_vma(mm, m.addr);
506 	if (!vma ||
507 	    vma->vm_ops != &privcmd_vm_ops) {
508 		ret = -EINVAL;
509 		goto out_unlock;
510 	}
511 
512 	/*
513 	 * Caller must either:
514 	 *
515 	 * Map the whole VMA range, which will also allocate all the
516 	 * pages required for the auto_translated_physmap case.
517 	 *
518 	 * Or
519 	 *
520 	 * Map unmapped holes left from a previous map attempt (e.g.,
521 	 * because those foreign frames were previously paged out).
522 	 */
523 	if (vma->vm_private_data == NULL) {
524 		if (m.addr != vma->vm_start ||
525 		    m.addr + (nr_pages << PAGE_SHIFT) != vma->vm_end) {
526 			ret = -EINVAL;
527 			goto out_unlock;
528 		}
529 		if (xen_feature(XENFEAT_auto_translated_physmap)) {
530 			ret = alloc_empty_pages(vma, nr_pages);
531 			if (ret < 0)
532 				goto out_unlock;
533 		} else
534 			vma->vm_private_data = PRIV_VMA_LOCKED;
535 	} else {
536 		if (m.addr < vma->vm_start ||
537 		    m.addr + (nr_pages << PAGE_SHIFT) > vma->vm_end) {
538 			ret = -EINVAL;
539 			goto out_unlock;
540 		}
541 		if (privcmd_vma_range_is_mapped(vma, m.addr, nr_pages)) {
542 			ret = -EINVAL;
543 			goto out_unlock;
544 		}
545 	}
546 
547 	state.domain        = m.dom;
548 	state.vma           = vma;
549 	state.va            = m.addr;
550 	state.index         = 0;
551 	state.global_error  = 0;
552 	state.version       = version;
553 
554 	BUILD_BUG_ON(((PAGE_SIZE / sizeof(xen_pfn_t)) % XEN_PFN_PER_PAGE) != 0);
555 	/* mmap_batch_fn guarantees ret == 0 */
556 	BUG_ON(traverse_pages_block(m.num, sizeof(xen_pfn_t),
557 				    &pagelist, mmap_batch_fn, &state));
558 
559 	up_write(&mm->mmap_sem);
560 
561 	if (state.global_error) {
562 		/* Write back errors in second pass. */
563 		state.user_gfn = (xen_pfn_t *)m.arr;
564 		state.user_err = m.err;
565 		ret = traverse_pages_block(m.num, sizeof(xen_pfn_t),
566 					   &pagelist, mmap_return_errors, &state);
567 	} else
568 		ret = 0;
569 
570 	/* If we have not had any EFAULT-like global errors then set the global
571 	 * error to -ENOENT if necessary. */
572 	if ((ret == 0) && (state.global_error == -ENOENT))
573 		ret = -ENOENT;
574 
575 out:
576 	free_page_list(&pagelist);
577 	return ret;
578 
579 out_unlock:
580 	up_write(&mm->mmap_sem);
581 	goto out;
582 }
583 
584 static int lock_pages(
585 	struct privcmd_dm_op_buf kbufs[], unsigned int num,
586 	struct page *pages[], unsigned int nr_pages)
587 {
588 	unsigned int i;
589 
590 	for (i = 0; i < num; i++) {
591 		unsigned int requested;
592 		int pinned;
593 
594 		requested = DIV_ROUND_UP(
595 			offset_in_page(kbufs[i].uptr) + kbufs[i].size,
596 			PAGE_SIZE);
597 		if (requested > nr_pages)
598 			return -ENOSPC;
599 
600 		pinned = get_user_pages_fast(
601 			(unsigned long) kbufs[i].uptr,
602 			requested, FOLL_WRITE, pages);
603 		if (pinned < 0)
604 			return pinned;
605 
606 		nr_pages -= pinned;
607 		pages += pinned;
608 	}
609 
610 	return 0;
611 }
612 
613 static void unlock_pages(struct page *pages[], unsigned int nr_pages)
614 {
615 	unsigned int i;
616 
617 	if (!pages)
618 		return;
619 
620 	for (i = 0; i < nr_pages; i++) {
621 		if (pages[i])
622 			put_page(pages[i]);
623 	}
624 }
625 
626 static long privcmd_ioctl_dm_op(struct file *file, void __user *udata)
627 {
628 	struct privcmd_data *data = file->private_data;
629 	struct privcmd_dm_op kdata;
630 	struct privcmd_dm_op_buf *kbufs;
631 	unsigned int nr_pages = 0;
632 	struct page **pages = NULL;
633 	struct xen_dm_op_buf *xbufs = NULL;
634 	unsigned int i;
635 	long rc;
636 
637 	if (copy_from_user(&kdata, udata, sizeof(kdata)))
638 		return -EFAULT;
639 
640 	/* If restriction is in place, check the domid matches */
641 	if (data->domid != DOMID_INVALID && data->domid != kdata.dom)
642 		return -EPERM;
643 
644 	if (kdata.num == 0)
645 		return 0;
646 
647 	if (kdata.num > privcmd_dm_op_max_num)
648 		return -E2BIG;
649 
650 	kbufs = kcalloc(kdata.num, sizeof(*kbufs), GFP_KERNEL);
651 	if (!kbufs)
652 		return -ENOMEM;
653 
654 	if (copy_from_user(kbufs, kdata.ubufs,
655 			   sizeof(*kbufs) * kdata.num)) {
656 		rc = -EFAULT;
657 		goto out;
658 	}
659 
660 	for (i = 0; i < kdata.num; i++) {
661 		if (kbufs[i].size > privcmd_dm_op_buf_max_size) {
662 			rc = -E2BIG;
663 			goto out;
664 		}
665 
666 		if (!access_ok(VERIFY_WRITE, kbufs[i].uptr,
667 			       kbufs[i].size)) {
668 			rc = -EFAULT;
669 			goto out;
670 		}
671 
672 		nr_pages += DIV_ROUND_UP(
673 			offset_in_page(kbufs[i].uptr) + kbufs[i].size,
674 			PAGE_SIZE);
675 	}
676 
677 	pages = kcalloc(nr_pages, sizeof(*pages), GFP_KERNEL);
678 	if (!pages) {
679 		rc = -ENOMEM;
680 		goto out;
681 	}
682 
683 	xbufs = kcalloc(kdata.num, sizeof(*xbufs), GFP_KERNEL);
684 	if (!xbufs) {
685 		rc = -ENOMEM;
686 		goto out;
687 	}
688 
689 	rc = lock_pages(kbufs, kdata.num, pages, nr_pages);
690 	if (rc)
691 		goto out;
692 
693 	for (i = 0; i < kdata.num; i++) {
694 		set_xen_guest_handle(xbufs[i].h, kbufs[i].uptr);
695 		xbufs[i].size = kbufs[i].size;
696 	}
697 
698 	xen_preemptible_hcall_begin();
699 	rc = HYPERVISOR_dm_op(kdata.dom, kdata.num, xbufs);
700 	xen_preemptible_hcall_end();
701 
702 out:
703 	unlock_pages(pages, nr_pages);
704 	kfree(xbufs);
705 	kfree(pages);
706 	kfree(kbufs);
707 
708 	return rc;
709 }
710 
711 static long privcmd_ioctl_restrict(struct file *file, void __user *udata)
712 {
713 	struct privcmd_data *data = file->private_data;
714 	domid_t dom;
715 
716 	if (copy_from_user(&dom, udata, sizeof(dom)))
717 		return -EFAULT;
718 
719 	/* Set restriction to the specified domain, or check it matches */
720 	if (data->domid == DOMID_INVALID)
721 		data->domid = dom;
722 	else if (data->domid != dom)
723 		return -EINVAL;
724 
725 	return 0;
726 }
727 
728 static long privcmd_ioctl(struct file *file,
729 			  unsigned int cmd, unsigned long data)
730 {
731 	int ret = -ENOTTY;
732 	void __user *udata = (void __user *) data;
733 
734 	switch (cmd) {
735 	case IOCTL_PRIVCMD_HYPERCALL:
736 		ret = privcmd_ioctl_hypercall(file, udata);
737 		break;
738 
739 	case IOCTL_PRIVCMD_MMAP:
740 		ret = privcmd_ioctl_mmap(file, udata);
741 		break;
742 
743 	case IOCTL_PRIVCMD_MMAPBATCH:
744 		ret = privcmd_ioctl_mmap_batch(file, udata, 1);
745 		break;
746 
747 	case IOCTL_PRIVCMD_MMAPBATCH_V2:
748 		ret = privcmd_ioctl_mmap_batch(file, udata, 2);
749 		break;
750 
751 	case IOCTL_PRIVCMD_DM_OP:
752 		ret = privcmd_ioctl_dm_op(file, udata);
753 		break;
754 
755 	case IOCTL_PRIVCMD_RESTRICT:
756 		ret = privcmd_ioctl_restrict(file, udata);
757 		break;
758 
759 	default:
760 		break;
761 	}
762 
763 	return ret;
764 }
765 
766 static int privcmd_open(struct inode *ino, struct file *file)
767 {
768 	struct privcmd_data *data = kzalloc(sizeof(*data), GFP_KERNEL);
769 
770 	if (!data)
771 		return -ENOMEM;
772 
773 	/* DOMID_INVALID implies no restriction */
774 	data->domid = DOMID_INVALID;
775 
776 	file->private_data = data;
777 	return 0;
778 }
779 
780 static int privcmd_release(struct inode *ino, struct file *file)
781 {
782 	struct privcmd_data *data = file->private_data;
783 
784 	kfree(data);
785 	return 0;
786 }
787 
788 static void privcmd_close(struct vm_area_struct *vma)
789 {
790 	struct page **pages = vma->vm_private_data;
791 	int numpgs = vma_pages(vma);
792 	int numgfns = (vma->vm_end - vma->vm_start) >> XEN_PAGE_SHIFT;
793 	int rc;
794 
795 	if (!xen_feature(XENFEAT_auto_translated_physmap) || !numpgs || !pages)
796 		return;
797 
798 	rc = xen_unmap_domain_gfn_range(vma, numgfns, pages);
799 	if (rc == 0)
800 		free_xenballooned_pages(numpgs, pages);
801 	else
802 		pr_crit("unable to unmap MFN range: leaking %d pages. rc=%d\n",
803 			numpgs, rc);
804 	kfree(pages);
805 }
806 
807 static int privcmd_fault(struct vm_fault *vmf)
808 {
809 	printk(KERN_DEBUG "privcmd_fault: vma=%p %lx-%lx, pgoff=%lx, uv=%p\n",
810 	       vmf->vma, vmf->vma->vm_start, vmf->vma->vm_end,
811 	       vmf->pgoff, (void *)vmf->address);
812 
813 	return VM_FAULT_SIGBUS;
814 }
815 
816 static const struct vm_operations_struct privcmd_vm_ops = {
817 	.close = privcmd_close,
818 	.fault = privcmd_fault
819 };
820 
821 static int privcmd_mmap(struct file *file, struct vm_area_struct *vma)
822 {
823 	/* DONTCOPY is essential for Xen because copy_page_range doesn't know
824 	 * how to recreate these mappings */
825 	vma->vm_flags |= VM_IO | VM_PFNMAP | VM_DONTCOPY |
826 			 VM_DONTEXPAND | VM_DONTDUMP;
827 	vma->vm_ops = &privcmd_vm_ops;
828 	vma->vm_private_data = NULL;
829 
830 	return 0;
831 }
832 
833 /*
834  * For MMAPBATCH*. This allows asserting the singleshot mapping
835  * on a per pfn/pte basis. Mapping calls that fail with ENOENT
836  * can be then retried until success.
837  */
838 static int is_mapped_fn(pte_t *pte, struct page *pmd_page,
839 	                unsigned long addr, void *data)
840 {
841 	return pte_none(*pte) ? 0 : -EBUSY;
842 }
843 
844 static int privcmd_vma_range_is_mapped(
845 	           struct vm_area_struct *vma,
846 	           unsigned long addr,
847 	           unsigned long nr_pages)
848 {
849 	return apply_to_page_range(vma->vm_mm, addr, nr_pages << PAGE_SHIFT,
850 				   is_mapped_fn, NULL) != 0;
851 }
852 
853 const struct file_operations xen_privcmd_fops = {
854 	.owner = THIS_MODULE,
855 	.unlocked_ioctl = privcmd_ioctl,
856 	.open = privcmd_open,
857 	.release = privcmd_release,
858 	.mmap = privcmd_mmap,
859 };
860 EXPORT_SYMBOL_GPL(xen_privcmd_fops);
861 
862 static struct miscdevice privcmd_dev = {
863 	.minor = MISC_DYNAMIC_MINOR,
864 	.name = "xen/privcmd",
865 	.fops = &xen_privcmd_fops,
866 };
867 
868 static int __init privcmd_init(void)
869 {
870 	int err;
871 
872 	if (!xen_domain())
873 		return -ENODEV;
874 
875 	err = misc_register(&privcmd_dev);
876 	if (err != 0) {
877 		pr_err("Could not register Xen privcmd device\n");
878 		return err;
879 	}
880 	return 0;
881 }
882 
883 static void __exit privcmd_exit(void)
884 {
885 	misc_deregister(&privcmd_dev);
886 }
887 
888 module_init(privcmd_init);
889 module_exit(privcmd_exit);
890