* Check that no O_DIRECT or similar I/O is in progress on the
                 * page
                 */
-               if ((page_mapcount(page) + 2 + swapped) != page_count(page)) {
+               if (page_mapcount(page) + 1 + swapped != page_count(page)) {
                        set_pte_at_notify(mm, addr, ptep, entry);
                        goto out_unlock;
                }
        pte_t *ptep;
        spinlock_t *ptl;
        unsigned long addr;
-       pgprot_t prot;
        int err = -EFAULT;
 
-       prot = vm_get_page_prot(vma->vm_flags & ~VM_WRITE);
-
        addr = page_address_in_vma(oldpage, vma);
        if (addr == -EFAULT)
                goto out;
 
        flush_cache_page(vma, addr, pte_pfn(*ptep));
        ptep_clear_flush(vma, addr, ptep);
-       set_pte_at_notify(mm, addr, ptep, mk_pte(newpage, prot));
+       set_pte_at_notify(mm, addr, ptep, mk_pte(newpage, vma->vm_page_prot));
 
        page_remove_rmap(oldpage);
        put_page(oldpage);
 
        if (!(vma->vm_flags & VM_MERGEABLE))
                goto out;
-
        if (!PageAnon(oldpage))
                goto out;
 
-       get_page(newpage);
-       get_page(oldpage);
-
        /*
         * We need the page lock to read a stable PageSwapCache in
         * write_protect_page().  We use trylock_page() instead of
         * then come back to this page when it is unlocked.
         */
        if (!trylock_page(oldpage))
-               goto out_putpage;
+               goto out;
        /*
         * If this anonymous page is mapped only here, its pte may need
         * to be write-protected.  If it's mapped elsewhere, all of its
         * ptes are necessarily already write-protected.  But in either
         * case, we need to lock and check page_count is not raised.
         */
-       if (write_protect_page(vma, oldpage, &orig_pte)) {
-               unlock_page(oldpage);
-               goto out_putpage;
-       }
-       unlock_page(oldpage);
-
-       if (pages_identical(oldpage, newpage))
+       if (write_protect_page(vma, oldpage, &orig_pte) == 0 &&
+           pages_identical(oldpage, newpage))
                err = replace_page(vma, oldpage, newpage, orig_pte);
 
-out_putpage:
-       put_page(oldpage);
-       put_page(newpage);
+       unlock_page(oldpage);
 out:
        return err;
 }