static vm_fault_t do_page_mkwrite(struct vm_fault *vmf)
 {
        vm_fault_t ret;
-       struct page *page = vmf->page;
+       struct folio *folio = page_folio(vmf->page);
        unsigned int old_flags = vmf->flags;
 
        vmf->flags = FAULT_FLAG_WRITE|FAULT_FLAG_MKWRITE;
        if (unlikely(ret & (VM_FAULT_ERROR | VM_FAULT_NOPAGE)))
                return ret;
        if (unlikely(!(ret & VM_FAULT_LOCKED))) {
-               lock_page(page);
-               if (!page->mapping) {
-                       unlock_page(page);
+               folio_lock(folio);
+               if (!folio->mapping) {
+                       folio_unlock(folio);
                        return 0; /* retry */
                }
                ret |= VM_FAULT_LOCKED;
        } else
-               VM_BUG_ON_PAGE(!PageLocked(page), page);
+               VM_BUG_ON_FOLIO(!folio_test_locked(folio), folio);
        return ret;
 }