do_mmap() checks in with the FS
[akaros.git] / kern / src / mm.c
1 /* Copyright (c) 2009, 2010 The Regents of the University of California
2  * Barret Rhoden <brho@cs.berkeley.edu>
3  * See LICENSE for details.
4  *
5  * Virtual memory management functions.  Creation, modification, etc, of virtual
6  * memory regions (VMRs) as well as mmap(), mprotect(), and munmap().
7  *
8  * In general, error checking / bounds checks are done in the main function
9  * (e.g. mmap()), and the work is done in a do_ function (e.g. do_mmap()).
10  * Versions of those functions that are called when the memory lock (proc_lock
11  * for now) is already held begin with __ (e.g. __do_munmap()).  */
12
13 #include <frontend.h>
14 #include <ros/common.h>
15 #include <ros/mman.h>
16 #include <pmap.h>
17 #include <mm.h>
18 #include <process.h>
19 #include <stdio.h>
20 #include <syscall.h>
21 #include <slab.h>
22 #include <kmalloc.h>
23 #include <vfs.h>
24
25 struct kmem_cache *vmr_kcache;
26
27 void vmr_init(void)
28 {
29         vmr_kcache = kmem_cache_create("vm_regions", sizeof(struct vm_region),
30                                        __alignof__(struct dentry), 0, 0, 0);
31 }
32
33 /* For now, the caller will set the prot, flags, file, and offset.  In the
34  * future, we may put those in here, to do clever things with merging vm_regions
35  * that are the same.
36  *
37  * TODO: take a look at solari's vmem alloc.  And consider keeping these in a
38  * tree of some sort for easier lookups. */
39 struct vm_region *create_vmr(struct proc *p, uintptr_t va, size_t len)
40 {
41         struct vm_region *vmr = 0, *vm_i, *vm_next;
42         uintptr_t gap_end;
43
44         assert(!PGOFF(va));
45         assert(!PGOFF(len));
46         assert(va + len <= UMAPTOP);
47
48         /* Is there room before the first one: */
49         vm_i = TAILQ_FIRST(&p->vm_regions);
50         if (!vm_i || (va + len < vm_i->vm_base)) {
51                 vmr = kmem_cache_alloc(vmr_kcache, 0);
52                 if (!vmr)
53                         panic("EOM!");
54                 vmr->vm_base = va;
55                 TAILQ_INSERT_HEAD(&p->vm_regions, vmr, vm_link);
56         } else {
57                 TAILQ_FOREACH(vm_i, &p->vm_regions, vm_link) {
58                         vm_next = TAILQ_NEXT(vm_i, vm_link);
59                         gap_end = vm_next ? vm_next->vm_base : UMAPTOP;
60                         /* skip til we get past the 'hint' va */
61                         if (va >= gap_end)
62                                 continue;
63                         /* Find a gap that is big enough */
64                         if (gap_end - vm_i->vm_end >= len) {
65                                 vmr = kmem_cache_alloc(vmr_kcache, 0);
66                                 /* if we can put it at va, let's do that.  o/w, put it so it
67                                  * fits */
68                                 if ((gap_end >= va + len) && (va >= vm_i->vm_end))
69                                         vmr->vm_base = va;
70                                 else
71                                         vmr->vm_base = vm_i->vm_end;
72                                 TAILQ_INSERT_AFTER(&p->vm_regions, vm_i, vmr, vm_link);
73                                 break;
74                         }
75                 }
76         }
77         /* Finalize the creation, if we got one */
78         if (vmr) {
79                 vmr->vm_proc = p;
80                 vmr->vm_end = vmr->vm_base + len;
81         }
82         if (!vmr)
83                 warn("Not making a VMR, wanted %08p, + %p = %p", va, len, va + len);
84         return vmr;
85 }
86
87 /* Split a VMR at va, returning the new VMR.  It is set up the same way, with
88  * file offsets fixed accordingly.  'va' is the beginning of the new one, and
89  * must be page aligned. */
90 struct vm_region *split_vmr(struct vm_region *old_vmr, uintptr_t va)
91 {
92         struct vm_region *new_vmr;
93
94         assert(!PGOFF(va));
95         if ((old_vmr->vm_base >= va) || (old_vmr->vm_end <= va))
96                 return 0;
97         new_vmr = kmem_cache_alloc(vmr_kcache, 0);
98         TAILQ_INSERT_AFTER(&old_vmr->vm_proc->vm_regions, old_vmr, new_vmr,
99                            vm_link);
100         new_vmr->vm_proc = old_vmr->vm_proc;
101         new_vmr->vm_base = va;
102         new_vmr->vm_end = old_vmr->vm_end;
103         old_vmr->vm_end = va;
104         new_vmr->vm_prot = old_vmr->vm_prot;
105         new_vmr->vm_flags = old_vmr->vm_flags;
106         if (old_vmr->vm_file) {
107                 new_vmr->vm_file = old_vmr->vm_file;
108                 atomic_inc(&new_vmr->vm_file->f_refcnt);
109                 new_vmr->vm_foff = old_vmr->vm_foff +
110                                       old_vmr->vm_end - old_vmr->vm_base;
111         } else {
112                 new_vmr->vm_file = 0;
113                 new_vmr->vm_foff = 0;
114         }
115         return new_vmr;
116 }
117
118 /* Merges two vm regions.  For now, it will check to make sure they are the
119  * same.  The second one will be destroyed. */
120 int merge_vmr(struct vm_region *first, struct vm_region *second)
121 {
122         assert(first->vm_proc == second->vm_proc);
123         if ((first->vm_end != second->vm_base) ||
124             (first->vm_prot != second->vm_prot) ||
125             (first->vm_flags != second->vm_flags) ||
126             (first->vm_file != second->vm_file))
127                 return -1;
128         if ((first->vm_file) && (second->vm_foff != first->vm_foff +
129                                  first->vm_end - first->vm_base))
130                 return -1;
131         first->vm_end = second->vm_end;
132         destroy_vmr(second);
133         return 0;
134 }
135
136 /* Attempts to merge vmr with adjacent VMRs, returning a ptr to be used for vmr.
137  * It could be the same struct vmr, or possibly another one (usually lower in
138  * the address space. */
139 struct vm_region *merge_me(struct vm_region *vmr)
140 {
141         struct vm_region *vmr_temp;
142         /* Merge will fail if it cannot do it.  If it succeeds, the second VMR is
143          * destroyed, so we need to be a bit careful. */
144         vmr_temp = TAILQ_PREV(vmr, vmr_tailq, vm_link);
145         if (vmr_temp)
146                 if (!merge_vmr(vmr_temp, vmr))
147                         vmr = vmr_temp;
148         vmr_temp = TAILQ_NEXT(vmr, vm_link);
149         if (vmr_temp)
150                 merge_vmr(vmr, vmr_temp);
151         return vmr;
152 }
153
154 /* Grows the vm region up to (and not including) va.  Fails if another is in the
155  * way, etc. */
156 int grow_vmr(struct vm_region *vmr, uintptr_t va)
157 {
158         assert(!PGOFF(va));
159         struct vm_region *next = TAILQ_NEXT(vmr, vm_link);
160         if (next && next->vm_base < va)
161                 return -1;
162         if (va <= vmr->vm_end)
163                 return -1;
164         vmr->vm_end = va;
165         return 0;
166 }
167
168 /* Shrinks the vm region down to (and not including) va.  Whoever calls this
169  * will need to sort out the page table entries. */
170 int shrink_vmr(struct vm_region *vmr, uintptr_t va)
171 {
172         assert(!PGOFF(va));
173         if ((va < vmr->vm_base) || (va > vmr->vm_end))
174                 return -1;
175         vmr->vm_end = va;
176         return 0;
177 }
178
179 /* Called by the unmapper, just cleans up.  Whoever calls this will need to sort
180  * out the page table entries. */
181 void destroy_vmr(struct vm_region *vmr)
182 {
183         if (vmr->vm_file)
184                 atomic_dec(&vmr->vm_file->f_refcnt);
185         TAILQ_REMOVE(&vmr->vm_proc->vm_regions, vmr, vm_link);
186         kmem_cache_free(vmr_kcache, vmr);
187 }
188
189 /* Given a va and a proc (later an mm, possibly), returns the owning vmr, or 0
190  * if there is none. */
191 struct vm_region *find_vmr(struct proc *p, uintptr_t va)
192 {
193         struct vm_region *vmr;
194         /* ugly linear seach */
195         TAILQ_FOREACH(vmr, &p->vm_regions, vm_link) {
196                 if ((vmr->vm_base <= va) && (vmr->vm_end > va))
197                         return vmr;
198         }
199         return 0;
200 }
201
202 /* Finds the first vmr after va (including the one holding va), or 0 if there is
203  * none. */
204 struct vm_region *find_first_vmr(struct proc *p, uintptr_t va)
205 {
206         struct vm_region *vmr;
207         /* ugly linear seach */
208         TAILQ_FOREACH(vmr, &p->vm_regions, vm_link) {
209                 if ((vmr->vm_base <= va) && (vmr->vm_end > va))
210                         return vmr;
211                 if (vmr->vm_base > va)
212                         return vmr;
213         }
214         return 0;
215 }
216
217 /* Makes sure that no VMRs cross either the start or end of the given region
218  * [va, va + len), splitting any VMRs that are on the endpoints. */
219 void isolate_vmrs(struct proc *p, uintptr_t va, size_t len)
220 {
221         struct vm_region *vmr;
222         if ((vmr = find_vmr(p, va)))
223                 split_vmr(vmr, va);
224         /* TODO: don't want to do another find (linear search) */
225         if ((vmr = find_vmr(p, va + len)))
226                 split_vmr(vmr, va + len);
227 }
228
229 /* This will make new_p have the same VMRs as p, though it does nothing to
230  * ensure the physical pages or whatever are shared/mapped/copied/whatever.
231  * This is used by fork().
232  *
233  * Note that if you are working on a VMR that is a file, you'll want to be
234  * careful about how it is mapped (SHARED, PRIVATE, etc). */
235 void duplicate_vmrs(struct proc *p, struct proc *new_p)
236 {
237         struct vm_region *vmr, *vm_i;
238         TAILQ_FOREACH(vm_i, &p->vm_regions, vm_link) {
239                 vmr = kmem_cache_alloc(vmr_kcache, 0);
240                 if (!vmr)
241                         panic("EOM!");
242                 vmr->vm_proc = new_p;
243                 vmr->vm_base = vm_i->vm_base;
244                 vmr->vm_end = vm_i->vm_end;
245                 vmr->vm_prot = vm_i->vm_prot;   
246                 vmr->vm_flags = vm_i->vm_flags; 
247                 vmr->vm_file = vm_i->vm_file;
248                 if (vmr->vm_file)
249                         atomic_inc(&vmr->vm_file->f_refcnt);
250                 vmr->vm_foff = vm_i->vm_foff;
251                 TAILQ_INSERT_TAIL(&new_p->vm_regions, vmr, vm_link);
252         }
253 }
254
255 void print_vmrs(struct proc *p)
256 {
257         int count = 0;
258         struct vm_region *vmr;
259         printk("VM Regions for proc %d\n", p->pid);
260         TAILQ_FOREACH(vmr, &p->vm_regions, vm_link)
261                 printk("%02d: (0x%08x - 0x%08x): %08p, %08p, %08p\n", count++,
262                        vmr->vm_base, vmr->vm_end, vmr->vm_prot, vmr->vm_flags,
263                        vmr->vm_file);
264 }
265
266
267 /* Error values aren't quite comprehensive - check man mmap() once we do better
268  * with the FS */
269 void *mmap(struct proc *p, uintptr_t addr, size_t len, int prot, int flags,
270            int fd, size_t offset)
271 {
272         struct file *file = NULL;
273         printd("mmap(addr %x, len %x, prot %x, flags %x, fd %x, off %x)\n", addr,
274                len, prot, flags, fd, offset);
275         if (fd >= 0 && (flags & MAP_SHARED)) {
276                 printk("[kernel] mmap() for files requires !MAP_SHARED.\n");
277                 set_errno(current_tf, EACCES);
278                 return MAP_FAILED;
279         }
280         if (fd >= 0 && (flags & MAP_ANON)) {
281                 set_errno(current_tf, EBADF);
282                 return MAP_FAILED;
283         }
284         if ((addr + len > UMAPTOP) || (PGOFF(addr))) {
285                 set_errno(current_tf, EINVAL);
286                 return MAP_FAILED;
287         }
288         if (!len) {
289                 set_errno(current_tf, EINVAL);
290                 return MAP_FAILED;
291         }
292         if (fd != -1) {
293                 file = file_open_from_fd(p, fd);
294                 if (!file) {
295                         set_errno(current_tf, EBADF);
296                         return MAP_FAILED;
297                 }
298         }
299         addr = MAX(addr, MMAP_LOWEST_VA);
300         void *result = do_mmap(p, addr, len, prot, flags, file, offset);
301         if (file)
302                 file_decref(file);
303         return result;
304 }
305
306 void *do_mmap(struct proc *p, uintptr_t addr, size_t len, int prot, int flags,
307               struct file *file, size_t offset)
308 {
309         // TODO: grab the appropriate mm_lock
310         spin_lock(&p->proc_lock);
311         void *ret = __do_mmap(p, addr, len, prot, flags, file, offset);
312         spin_unlock(&p->proc_lock);
313         return ret;
314 }
315
316 void *__do_mmap(struct proc *p, uintptr_t addr, size_t len, int prot, int flags,
317                 struct file *file, size_t offset)
318 {
319         len = ROUNDUP(len, PGSIZE);
320         int num_pages = len / PGSIZE;
321
322         struct vm_region *vmr, *vmr_temp;
323
324 #ifndef __CONFIG_DEMAND_PAGING__
325         flags |= MAP_POPULATE;
326 #endif
327         /* Need to make sure nothing is in our way when we want a FIXED location.
328          * We just need to split on the end points (if they exist), and then remove
329          * everything in between.  __do_munmap() will do this. */
330         if (flags & MAP_FIXED)
331                 __do_munmap(p, addr, len);
332         vmr = create_vmr(p, addr, len);
333         if (!vmr) {
334                 printk("[kernel] do_mmap() aborted for %08p + %d!\n", addr, len);
335                 set_errno(current_tf, ENOMEM);
336                 return MAP_FAILED;              /* TODO: error propagation for mmap() */
337         }
338         vmr->vm_prot = prot;
339         vmr->vm_flags = flags;
340         vmr->vm_file = file;
341         vmr->vm_foff = offset;
342         /* Prep the FS to make sure it can mmap the file.  Slightly weird semantics:
343          * they will have a hole in their VM now. */
344         if (file && file->f_op->mmap(file, vmr)) {
345                 destroy_vmr(vmr);
346                 set_errno(current_tf, EACCES);  /* not quite */
347                 return MAP_FAILED;
348         }
349         addr = vmr->vm_base;            /* so we know which pages to populate later */
350         vmr = merge_me(vmr);            /* attempts to merge with neighbors */
351         /* Fault in pages now if MAP_POPULATE - die on failure.  We want to populate
352          * the region requested, but we need to be careful and only populate the
353          * requested length and not any merged regions, which is why we set addr
354          * above and use it here. */
355         if (flags & MAP_POPULATE)
356                 for (int i = 0; i < num_pages; i++)
357                         if (__handle_page_fault(p, addr + i*PGSIZE, vmr->vm_prot)) {
358                                 spin_unlock(&p->proc_lock);
359                                 proc_destroy(p);
360                         }
361         return (void*SAFE)TC(addr);
362 }
363
364 int mprotect(struct proc *p, uintptr_t addr, size_t len, int prot)
365 {
366         printd("mprotect(addr %x, len %x, prot %x)\n", addr, len, prot);
367         if (!len)
368                 return 0;
369         if ((addr % PGSIZE) || (addr < MMAP_LOWEST_VA)) {
370                 set_errno(current_tf, EINVAL);
371                 return -1;
372         }
373         uintptr_t end = ROUNDUP(addr + len, PGSIZE);
374         if (end > UMAPTOP || addr > end) {
375                 set_errno(current_tf, ENOMEM);
376                 return -1;
377         }
378         spin_lock(&p->proc_lock);
379         int ret = __do_mprotect(p, addr, len, prot);
380         spin_unlock(&p->proc_lock);
381         return ret;
382 }
383
384 /* This does not care if the region is not mapped.  POSIX says you should return
385  * ENOMEM if any part of it is unmapped.  Can do this later if we care, based on
386  * the VMRs, not the actual page residency. */
387 int __do_mprotect(struct proc *p, uintptr_t addr, size_t len, int prot)
388 {
389         struct vm_region *vmr, *next_vmr;
390         pte_t *pte;
391         bool shootdown_needed = FALSE;
392         int pte_prot = (prot & PROT_WRITE) ? PTE_USER_RW :
393                        (prot & (PROT_READ|PROT_EXEC)) ? PTE_USER_RO : 0;
394         /* TODO: this is aggressively splitting, when we might not need to if the
395          * prots are the same as the previous.  Plus, there are three excessive
396          * scans.  Finally, we might be able to merge when we are done. */
397         isolate_vmrs(p, addr, addr + len);
398         vmr = find_first_vmr(p, addr);
399         while (vmr && vmr->vm_base < addr + len) {
400                 if (vmr->vm_prot == prot)
401                         continue;
402                 /* if vmr maps a file, then we need to make sure the protection change
403                  * is in compliance with the open mode of the file.  At least for any
404                  * mapping that is write-backed to a file.  For now, we just do it for
405                  * all file mappings.  And this hasn't been tested */
406                 if (vmr->vm_file && (prot & PROT_WRITE)) {
407                         if (!(vmr->vm_file->f_mode & PROT_WRITE)) {
408                                 set_errno(current_tf, EACCES);
409                                 return -1;
410                         }
411                 }
412                 vmr->vm_prot = prot;
413                 for (uintptr_t va = vmr->vm_base; va < vmr->vm_end; va += PGSIZE) { 
414                         pte = pgdir_walk(p->env_pgdir, (void*)va, 0);
415                         if (pte && PAGE_PRESENT(*pte)) {
416                                 *pte = (*pte & ~PTE_PERM) | pte_prot;
417                                 shootdown_needed = TRUE;
418                         }
419                 }
420                 next_vmr = TAILQ_NEXT(vmr, vm_link);
421                 vmr = next_vmr;
422         }
423         if (shootdown_needed)
424                 __proc_tlbshootdown(p, addr, addr + len);
425         return 0;
426 }
427
428 int munmap(struct proc *p, uintptr_t addr, size_t len)
429 {
430         printd("munmap(addr %x, len %x, prot %x)\n", addr, len, prot);
431         if (!len)
432                 return 0;
433         if ((addr % PGSIZE) || (addr < MMAP_LOWEST_VA)) {
434                 set_errno(current_tf, EINVAL);
435                 return -1;
436         }
437         uintptr_t end = ROUNDUP(addr + len, PGSIZE);
438         if (end > UMAPTOP || addr > end) {
439                 set_errno(current_tf, EINVAL);
440                 return -1;
441         }
442         spin_lock(&p->proc_lock);
443         int ret = __do_munmap(p, addr, len);
444         spin_unlock(&p->proc_lock);
445         return ret;
446 }
447
448 int __do_munmap(struct proc *p, uintptr_t addr, size_t len)
449 {
450         struct vm_region *vmr, *next_vmr;
451         pte_t *pte;
452         bool shootdown_needed = FALSE;
453
454         /* TODO: this will be a bit slow, since we end up doing three linear
455          * searches (two in isolate, one in find_first). */
456         isolate_vmrs(p, addr, addr + len);
457         vmr = find_first_vmr(p, addr);
458         while (vmr && vmr->vm_base < addr + len) {
459                 for (uintptr_t va = vmr->vm_base; va < vmr->vm_end; va += PGSIZE) { 
460                         pte = pgdir_walk(p->env_pgdir, (void*)va, 0);
461                         if (!pte)
462                                 continue;
463                         if (PAGE_PRESENT(*pte)) {
464                                 /* TODO: (TLB) race here, where the page can be given out before
465                                  * the shootdown happened.  Need to put it on a temp list. */
466                                 page_t *page = ppn2page(PTE2PPN(*pte));
467                                 *pte = 0;
468                                 page_decref(page);
469                                 shootdown_needed = TRUE;
470                         } else if (PAGE_PAGED_OUT(*pte)) {
471                                 /* TODO: (SWAP) mark free in the swapfile or whatever.  For now,
472                                  * PAGED_OUT is also being used to mean "hasn't been mapped
473                                  * yet".  Note we now allow PAGE_UNMAPPED, unlike older
474                                  * versions of mmap(). */
475                                 panic("Swapping not supported!");
476                                 *pte = 0;
477                         }
478                 }
479                 next_vmr = TAILQ_NEXT(vmr, vm_link);
480                 destroy_vmr(vmr);
481                 vmr = next_vmr;
482         }
483         if (shootdown_needed)
484                 __proc_tlbshootdown(p, addr, addr + len);
485         return 0;
486 }
487
488 int handle_page_fault(struct proc* p, uintptr_t va, int prot)
489 {
490         va = ROUNDDOWN(va,PGSIZE);
491
492         if (prot != PROT_READ && prot != PROT_WRITE && prot != PROT_EXEC)
493                 panic("bad prot!");
494
495         spin_lock(&p->proc_lock);
496         int ret = __handle_page_fault(p, va, prot);
497         spin_unlock(&p->proc_lock);
498         return ret;
499 }
500
501 /* Returns 0 on success, or an appropriate -error code.  Assumes you hold the
502  * appropriate lock.
503  *
504  * Notes: if your TLB caches negative results, you'll need to flush the
505  * appropriate tlb entry.  Also, you could have a weird race where a present PTE
506  * faulted for a different reason (was mprotected on another core), and the
507  * shootdown is on its way.  Userspace should have waited for the mprotect to
508  * return before trying to write (or whatever), so we don't care and will fault
509  * them. */
510 int __handle_page_fault(struct proc* p, uintptr_t va, int prot)
511 {
512         struct vm_region *vmr;
513         struct page *a_page;
514         unsigned int f_idx;     /* index of the missing page in the file */
515         int retval = 0;
516         /* Check the vmr's protection */
517         vmr = find_vmr(p, va);
518         if (!vmr)                                                       /* not mapped at all */
519                 return -EFAULT;
520         if (!(vmr->vm_prot & prot))                     /* wrong prots for this vmr */
521                 return -EFAULT;
522         /* find offending PTE (prob don't read this in).  This might alloc an
523          * intermediate page table page. */
524         pte_t *pte = pgdir_walk(p->env_pgdir, (void*)va, 1);
525         if (!pte)
526                 return -ENOMEM;
527         /* a spurious, valid PF is possible due to a legit race: the page might have
528          * been faulted in by another core already (and raced on the memory lock),
529          * in which case we should just return. */
530         if (PAGE_PRESENT(*pte)) {
531                 return 0;
532         } else if (PAGE_PAGED_OUT(*pte)) {
533                 /* TODO: (SWAP) bring in the paged out frame. (BLK) */
534                 panic("Swapping not supported!");
535                 return 0;
536         }
537         if (!vmr->vm_file) {
538                 /* No file - just want anonymous memory */
539                 if (upage_alloc(p, &a_page, TRUE))
540                         return -ENOMEM;
541         } else {
542                 /* Load the file's page in the page cache.
543                  * TODO: (BLK) Note, we are holding the mem lock!  We need to rewrite
544                  * this stuff so we aren't hold the lock as excessively as we are, and
545                  * such that we can block and resume later. */
546                 f_idx = (va - vmr->vm_base + vmr->vm_foff) >> PGSHIFT;
547                 retval = file_load_page(vmr->vm_file, f_idx, &a_page);
548                 if (retval)
549                         return retval;
550                 /* if this is an executable page, we might have to flush the instruction
551                  * cache if our HW requires it. */
552                 if (vmr->vm_prot & PROT_EXEC)
553                         icache_flush_page((void*)va, page2kva(a_page));
554         }
555         /* update the page table TODO: careful with MAP_PRIVATE etc.  might do this
556          * separately (file, no file) */
557         int pte_prot = (vmr->vm_prot & PROT_WRITE) ? PTE_USER_RW :
558                        (vmr->vm_prot & (PROT_READ|PROT_EXEC)) ? PTE_USER_RO : 0;
559         page_incref(a_page);    /* incref, since we manually insert in the pgdir */
560         *pte = PTE(page2ppn(a_page), PTE_P | pte_prot);
561         return 0;
562 }