142 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			142 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * Set up the VMAs to tell the VM about the vDSO.
 | |
|  * Copyright 2007 Andi Kleen, SUSE Labs.
 | |
|  * Subject to the GPL, v.2
 | |
|  */
 | |
| #include <linux/mm.h>
 | |
| #include <linux/err.h>
 | |
| #include <linux/sched.h>
 | |
| #include <linux/init.h>
 | |
| #include <linux/random.h>
 | |
| #include <linux/elf.h>
 | |
| #include <asm/vsyscall.h>
 | |
| #include <asm/vgtod.h>
 | |
| #include <asm/proto.h>
 | |
| #include <asm/vdso.h>
 | |
| 
 | |
| #include "vextern.h"		/* Just for VMAGIC.  */
 | |
| #undef VEXTERN
 | |
| 
 | |
| unsigned int __read_mostly vdso_enabled = 1;
 | |
| 
 | |
| extern char vdso_start[], vdso_end[];
 | |
| extern unsigned short vdso_sync_cpuid;
 | |
| 
 | |
| static struct page **vdso_pages;
 | |
| static unsigned vdso_size;
 | |
| 
 | |
| static inline void *var_ref(void *p, char *name)
 | |
| {
 | |
| 	if (*(void **)p != (void *)VMAGIC) {
 | |
| 		printk("VDSO: variable %s broken\n", name);
 | |
| 		vdso_enabled = 0;
 | |
| 	}
 | |
| 	return p;
 | |
| }
 | |
| 
 | |
| static int __init init_vdso_vars(void)
 | |
| {
 | |
| 	int npages = (vdso_end - vdso_start + PAGE_SIZE - 1) / PAGE_SIZE;
 | |
| 	int i;
 | |
| 	char *vbase;
 | |
| 
 | |
| 	vdso_size = npages << PAGE_SHIFT;
 | |
| 	vdso_pages = kmalloc(sizeof(struct page *) * npages, GFP_KERNEL);
 | |
| 	if (!vdso_pages)
 | |
| 		goto oom;
 | |
| 	for (i = 0; i < npages; i++) {
 | |
| 		struct page *p;
 | |
| 		p = alloc_page(GFP_KERNEL);
 | |
| 		if (!p)
 | |
| 			goto oom;
 | |
| 		vdso_pages[i] = p;
 | |
| 		copy_page(page_address(p), vdso_start + i*PAGE_SIZE);
 | |
| 	}
 | |
| 
 | |
| 	vbase = vmap(vdso_pages, npages, 0, PAGE_KERNEL);
 | |
| 	if (!vbase)
 | |
| 		goto oom;
 | |
| 
 | |
| 	if (memcmp(vbase, "\177ELF", 4)) {
 | |
| 		printk("VDSO: I'm broken; not ELF\n");
 | |
| 		vdso_enabled = 0;
 | |
| 	}
 | |
| 
 | |
| #define VEXTERN(x) \
 | |
| 	*(typeof(__ ## x) **) var_ref(VDSO64_SYMBOL(vbase, x), #x) = &__ ## x;
 | |
| #include "vextern.h"
 | |
| #undef VEXTERN
 | |
| 	return 0;
 | |
| 
 | |
|  oom:
 | |
| 	printk("Cannot allocate vdso\n");
 | |
| 	vdso_enabled = 0;
 | |
| 	return -ENOMEM;
 | |
| }
 | |
| __initcall(init_vdso_vars);
 | |
| 
 | |
| struct linux_binprm;
 | |
| 
 | |
| /* Put the vdso above the (randomized) stack with another randomized offset.
 | |
|    This way there is no hole in the middle of address space.
 | |
|    To save memory make sure it is still in the same PTE as the stack top.
 | |
|    This doesn't give that many random bits */
 | |
| static unsigned long vdso_addr(unsigned long start, unsigned len)
 | |
| {
 | |
| 	unsigned long addr, end;
 | |
| 	unsigned offset;
 | |
| 	end = (start + PMD_SIZE - 1) & PMD_MASK;
 | |
| 	if (end >= TASK_SIZE_MAX)
 | |
| 		end = TASK_SIZE_MAX;
 | |
| 	end -= len;
 | |
| 	/* This loses some more bits than a modulo, but is cheaper */
 | |
| 	offset = get_random_int() & (PTRS_PER_PTE - 1);
 | |
| 	addr = start + (offset << PAGE_SHIFT);
 | |
| 	if (addr >= end)
 | |
| 		addr = end;
 | |
| 	return addr;
 | |
| }
 | |
| 
 | |
| /* Setup a VMA at program startup for the vsyscall page.
 | |
|    Not called for compat tasks */
 | |
| int arch_setup_additional_pages(struct linux_binprm *bprm, int uses_interp)
 | |
| {
 | |
| 	struct mm_struct *mm = current->mm;
 | |
| 	unsigned long addr;
 | |
| 	int ret;
 | |
| 
 | |
| 	if (!vdso_enabled)
 | |
| 		return 0;
 | |
| 
 | |
| 	down_write(&mm->mmap_sem);
 | |
| 	addr = vdso_addr(mm->start_stack, vdso_size);
 | |
| 	addr = get_unmapped_area(NULL, addr, vdso_size, 0, 0);
 | |
| 	if (IS_ERR_VALUE(addr)) {
 | |
| 		ret = addr;
 | |
| 		goto up_fail;
 | |
| 	}
 | |
| 
 | |
| 	current->mm->context.vdso = (void *)addr;
 | |
| 
 | |
| 	ret = install_special_mapping(mm, addr, vdso_size,
 | |
| 				      VM_READ|VM_EXEC|
 | |
| 				      VM_MAYREAD|VM_MAYWRITE|VM_MAYEXEC|
 | |
| 				      VM_ALWAYSDUMP,
 | |
| 				      vdso_pages);
 | |
| 	if (ret) {
 | |
| 		current->mm->context.vdso = NULL;
 | |
| 		goto up_fail;
 | |
| 	}
 | |
| 
 | |
| up_fail:
 | |
| 	up_write(&mm->mmap_sem);
 | |
| 	return ret;
 | |
| }
 | |
| 
 | |
| static __init int vdso_setup(char *s)
 | |
| {
 | |
| 	vdso_enabled = simple_strtoul(s, NULL, 0);
 | |
| 	return 0;
 | |
| }
 | |
| __setup("vdso=", vdso_setup);
 |