vserver 2.0 rc7
[linux-2.6.git] / mm / mprotect.c
index 0427711..e9fbd01 100644 (file)
 #include <asm/cacheflush.h>
 #include <asm/tlbflush.h>
 
-static inline void
-change_pte_range(pmd_t *pmd, unsigned long address,
-               unsigned long size, pgprot_t newprot)
+static void change_pte_range(struct mm_struct *mm, pmd_t *pmd,
+               unsigned long addr, unsigned long end, pgprot_t newprot)
 {
-       pte_t * pte;
-       unsigned long end;
-
-       if (pmd_none(*pmd))
-               return;
-       if (pmd_bad(*pmd)) {
-               pmd_ERROR(*pmd);
-               pmd_clear(pmd);
-               return;
-       }
-       pte = pte_offset_map(pmd, address);
-       address &= ~PMD_MASK;
-       end = address + size;
-       if (end > PMD_SIZE)
-               end = PMD_SIZE;
+       pte_t *pte;
+
+       pte = pte_offset_map(pmd, addr);
        do {
                if (pte_present(*pte)) {
-                       pte_t entry;
+                       pte_t ptent;
 
                        /* Avoid an SMP race with hardware updated dirty/clean
                         * bits by wiping the pte and then setting the new pte
                         * into place.
                         */
-                       entry = ptep_get_and_clear(pte);
-                       set_pte(pte, pte_modify(entry, newprot));
+                       ptent = pte_modify(ptep_get_and_clear(mm, addr, pte), newprot);
+                       set_pte_at(mm, addr, pte, ptent);
+                       lazy_mmu_prot_update(ptent);
                }
-               address += PAGE_SIZE;
-               pte++;
-       } while (address && (address < end));
+       } while (pte++, addr += PAGE_SIZE, addr != end);
        pte_unmap(pte - 1);
 }
 
-static inline void
-change_pmd_range(pud_t *pud, unsigned long address,
-               unsigned long size, pgprot_t newprot)
+static inline void change_pmd_range(struct mm_struct *mm, pud_t *pud,
+               unsigned long addr, unsigned long end, pgprot_t newprot)
 {
-       pmd_t * pmd;
-       unsigned long end;
-
-       if (pud_none(*pud))
-               return;
-       if (pud_bad(*pud)) {
-               pud_ERROR(*pud);
-               pud_clear(pud);
-               return;
-       }
-       pmd = pmd_offset(pud, address);
-       address &= ~PUD_MASK;
-       end = address + size;
-       if (end > PUD_SIZE)
-               end = PUD_SIZE;
+       pmd_t *pmd;
+       unsigned long next;
+
+       pmd = pmd_offset(pud, addr);
        do {
-               change_pte_range(pmd, address, end - address, newprot);
-               address = (address + PMD_SIZE) & PMD_MASK;
-               pmd++;
-       } while (address && (address < end));
+               next = pmd_addr_end(addr, end);
+               if (pmd_none_or_clear_bad(pmd))
+                       continue;
+               change_pte_range(mm, pmd, addr, next, newprot);
+       } while (pmd++, addr = next, addr != end);
 }
 
-static inline void
-change_pud_range(pgd_t *pgd, unsigned long address,
-               unsigned long size, pgprot_t newprot)
+static inline void change_pud_range(struct mm_struct *mm, pgd_t *pgd,
+               unsigned long addr, unsigned long end, pgprot_t newprot)
 {
-       pud_t * pud;
-       unsigned long end;
-
-       if (pgd_none(*pgd))
-               return;
-       if (pgd_bad(*pgd)) {
-               pgd_ERROR(*pgd);
-               pgd_clear(pgd);
-               return;
-       }
-       pud = pud_offset(pgd, address);
-       address &= ~PGDIR_MASK;
-       end = address + size;
-       if (end > PGDIR_SIZE)
-               end = PGDIR_SIZE;
+       pud_t *pud;
+       unsigned long next;
+
+       pud = pud_offset(pgd, addr);
        do {
-               change_pmd_range(pud, address, end - address, newprot);
-               address = (address + PUD_SIZE) & PUD_MASK;
-               pud++;
-       } while (address && (address < end));
+               next = pud_addr_end(addr, end);
+               if (pud_none_or_clear_bad(pud))
+                       continue;
+               change_pmd_range(mm, pud, addr, next, newprot);
+       } while (pud++, addr = next, addr != end);
 }
 
-static void
-change_protection(struct vm_area_struct *vma, unsigned long start,
-               unsigned long end, pgprot_t newprot)
+static void change_protection(struct vm_area_struct *vma,
+               unsigned long addr, unsigned long end, pgprot_t newprot)
 {
-       struct mm_struct *mm = current->mm;
+       struct mm_struct *mm = vma->vm_mm;
        pgd_t *pgd;
-       unsigned long beg = start, next;
-       int i;
+       unsigned long next;
+       unsigned long start = addr;
 
-       pgd = pgd_offset(mm, start);
-       flush_cache_range(vma, beg, end);
-       BUG_ON(start >= end);
+       BUG_ON(addr >= end);
+       pgd = pgd_offset(mm, addr);
+       flush_cache_range(vma, addr, end);
        spin_lock(&mm->page_table_lock);
-       for (i = pgd_index(start); i <= pgd_index(end-1); i++) {
-               next = (start + PGDIR_SIZE) & PGDIR_MASK;
-               if (next <= start || next > end)
-                       next = end;
-               change_pud_range(pgd, start, next - start, newprot);
-               start = next;
-               pgd++;
-       }
-       flush_tlb_range(vma, beg, end);
+       do {
+               next = pgd_addr_end(addr, end);
+               if (pgd_none_or_clear_bad(pgd))
+                       continue;
+               change_pud_range(mm, pgd, addr, next, newprot);
+       } while (pgd++, addr = next, addr != end);
+       flush_tlb_range(vma, start, end);
        spin_unlock(&mm->page_table_lock);
 }
 
@@ -142,7 +103,7 @@ static int
 mprotect_fixup(struct vm_area_struct *vma, struct vm_area_struct **pprev,
        unsigned long start, unsigned long end, unsigned long newflags)
 {
-       struct mm_struct * mm = vma->vm_mm;
+       struct mm_struct *mm = vma->vm_mm;
        unsigned long oldflags = vma->vm_flags;
        long nrpages = (end - start) >> PAGE_SHIFT;
        unsigned long charged = 0;
@@ -185,16 +146,13 @@ mprotect_fixup(struct vm_area_struct *vma, struct vm_area_struct **pprev,
                goto success;
        }
 
+       *pprev = vma;
+
        if (start != vma->vm_start) {
                error = split_vma(mm, vma, start, 1);
                if (error)
                        goto fail;
        }
-       /*
-        * Unless it returns an error, this function always sets *pprev to
-        * the first vma for which vma->vm_end >= end.
-        */
-       *pprev = vma;
 
        if (end != vma->vm_end) {
                error = split_vma(mm, vma, end, 0);
@@ -222,7 +180,7 @@ fail:
 asmlinkage long
 sys_mprotect(unsigned long start, size_t len, unsigned long prot)
 {
-       unsigned long vm_flags, nstart, end, tmp;
+       unsigned long vm_flags, nstart, end, tmp, reqprot;
        struct vm_area_struct *vma, *prev;
        int error = -EINVAL;
        const int grows = prot & (PROT_GROWSDOWN|PROT_GROWSUP);
@@ -232,14 +190,16 @@ sys_mprotect(unsigned long start, size_t len, unsigned long prot)
 
        if (start & ~PAGE_MASK)
                return -EINVAL;
+       if (!len)
+               return 0;
        len = PAGE_ALIGN(len);
        end = start + len;
-       if (end < start)
+       if (end <= start)
                return -ENOMEM;
        if (prot & ~(PROT_READ | PROT_WRITE | PROT_EXEC | PROT_SEM))
                return -EINVAL;
-       if (end == start)
-               return 0;
+
+       reqprot = prot;
        /*
         * Does the application expect PROT_READ to imply PROT_EXEC:
         */
@@ -293,7 +253,7 @@ sys_mprotect(unsigned long start, size_t len, unsigned long prot)
                        goto out;
                }
 
-               error = security_file_mprotect(vma, prot);
+               error = security_file_mprotect(vma, reqprot, prot);
                if (error)
                        goto out;