Mapping precision fixes on l4_unmap and do_munmap()

l4_unmap now returns -1 if given range was only partially unmapped.

do_munmap() now only unmaps address ranges that have correspondence in
the unmapped vmas. Trying to unmap regions with no correspondent vmas
causes problems in corner cases, e.g. mm0 that tries to mmap its own
address space during initialisation would unmap its whole address space
and fail to execute.

tasks/mm0/include/mmap.h

@@ -24,7 +24,7 @@
 struct vm_area *vma_new(unsigned long pfn_start, unsigned long npages,
 			unsigned int flags, unsigned long file_offset);
 
-int do_munmap(struct tcb *task, void *vaddr, unsigned long size);
+int do_munmap(struct tcb *task, unsigned long vaddr, unsigned long size);
 
 void *do_mmap(struct vm_file *mapfile, unsigned long f_offset, struct tcb *t,
 	      unsigned long map_address, unsigned int flags, unsigned int pages);

tasks/mm0/src/mmap.c

@@ -146,10 +146,9 @@ void *do_mmap(struct vm_file *mapfile, unsigned long file_offset,
 	      struct tcb *task, unsigned long map_address,
 	      unsigned int flags, unsigned int npages)
 {
-	unsigned long map_pfn = __pfn(map_address);
-	unsigned long file_npages;
-	struct vm_area *new, *mapped;
 	struct vm_obj_link *vmo_link, *vmo_link2;
+	unsigned long file_npages;
+	struct vm_area *new;
 	int err;
 
 	/* Set up devzero if none given */

tasks/mm0/src/munmap.c

@@ -9,12 +9,14 @@
 #include <l4/lib/math.h>
 #include <l4lib/arch/syslib.h>
 #include <vm_area.h>
+#include <lib/malloc.h>
 
 /* This splits a vma, splitter region must be in the *middle* of original vma */
 int vma_split(struct vm_area *vma, struct tcb *task,
 	      const unsigned long pfn_start, const unsigned long pfn_end)
 {
 	struct vm_area *new;
+	unsigned long unmap_start = pfn_start, unmap_end = pfn_end;
 
 	/* Allocate an uninitialised vma first */
 	if (!(new = vma_new(0, 0, 0, 0)))
@@ -43,6 +45,10 @@ int vma_split(struct vm_area *vma, struct tcb *task,
 	/* Add new one next to original vma */
 	list_add_tail(&new->list, &vma->list);
 
+	/* Unmap the removed portion */
+	BUG_ON(l4_unmap((void *)__pfn_to_addr(unmap_start),
+	       unmap_end - unmap_start, task->tid) < 0);
+
 	return 0;
 }
 
@@ -50,20 +56,49 @@ int vma_split(struct vm_area *vma, struct tcb *task,
 int vma_shrink(struct vm_area *vma, struct tcb *task,
 	       const unsigned long pfn_start, const unsigned long pfn_end)
 {
-	unsigned long diff;
+	unsigned long diff, unmap_start, unmap_end;
 
 	/* Shrink from the end */
 	if (vma->pfn_start < pfn_start) {
 		BUG_ON(pfn_start >= vma->pfn_end);
+		unmap_start = pfn_start;
+		unmap_end = vma->pfn_end;
 		vma->pfn_end = pfn_start;
 
 	/* Shrink from the beginning */
 	} else if (vma->pfn_end > pfn_end) {
 		BUG_ON(pfn_end <= vma->pfn_start);
+		unmap_start = vma->pfn_start;
+		unmap_end = pfn_end;
 		diff = pfn_end - vma->pfn_start;
 		vma->file_offset += diff;
 		vma->pfn_start = pfn_end;
-	} else BUG();
+	} else
+		BUG();
+
+	/* Unmap the shrinked portion */
+	BUG_ON(l4_unmap((void *)__pfn_to_addr(unmap_start),
+	       unmap_end - unmap_start, task->tid) < 0);
+
+	return 0;
+}
+
+/* Destroys a single vma from a task and unmaps its range from task space */
+int vma_destroy_single(struct tcb *task, struct vm_area *vma)
+{
+	int ret;
+
+	/* Release all object links */
+	if ((ret = vma_drop_merge_delete_all(vma)) < 0)
+		return ret;
+
+	/* Unmap the whole vma address range */
+	BUG_ON(l4_unmap((void *)__pfn_to_addr(vma->pfn_start),
+	       vma->pfn_end - vma->pfn_start, task->tid) < 0);
+
+	/* Unlink and delete vma */
+	list_del(&vma->list);
+	kfree(vma);
 
 	return 0;
 }
@@ -84,7 +119,7 @@ int vma_unmap(struct vm_area *vma, struct tcb *task,
 		return vma_shrink(vma, task, pfn_start, pfn_end);
 	/* Destroy needed? */
 	else if ((vma->pfn_start >= pfn_start) && (vma->pfn_end <= pfn_end))
-		return vma_drop_merge_delete_all(vma);
+		return vma_destroy_single(task, vma);
 	else
 		BUG();
 
@@ -158,9 +193,6 @@ int do_munmap(struct tcb *task, unsigned long vaddr, unsigned long npages)
 		}
 	}
 
-	/* Unmap those pages from the task address space */
-	l4_unmap(vaddr, npages, task->tid);
-
 	return 0;
 }
 

tasks/mm0/src/shm.c

@@ -146,7 +146,7 @@ int do_shmdt(struct tcb *task, struct vm_file *shm)
 	int err;
 
 	if ((err = do_munmap(task,
-			     shm_file_to_desc(shm)->shm_addr,
+			     (unsigned long)shm_file_to_desc(shm)->shm_addr,
 			     shm_file_to_desc(shm)->npages)) < 0)
 		return err;