*/
 int unpoison_memory(unsigned long pfn)
 {
-       struct page *page;
        struct folio *folio;
        struct page *p;
        int ret = -EBUSY;
                return -ENXIO;
 
        p = pfn_to_page(pfn);
-       page = compound_head(p);
        folio = page_folio(p);
 
        mutex_lock(&mf_mutex);
                goto unlock_mutex;
        }
 
-       if (!PageHWPoison(p)) {
+       if (!folio_test_hwpoison(folio)) {
                unpoison_pr_info("Unpoison: Page was already unpoisoned %#lx\n",
                                 pfn, &unpoison_rs);
                goto unlock_mutex;
        }
 
-       if (page_count(page) > 1) {
+       if (folio_ref_count(folio) > 1) {
                unpoison_pr_info("Unpoison: Someone grabs the hwpoison page %#lx\n",
                                 pfn, &unpoison_rs);
                goto unlock_mutex;
        }
 
-       if (page_mapped(page)) {
+       if (folio_mapped(folio)) {
                unpoison_pr_info("Unpoison: Someone maps the hwpoison page %#lx\n",
                                 pfn, &unpoison_rs);
                goto unlock_mutex;
        }
 
-       if (page_mapping(page)) {
+       if (folio_mapping(folio)) {
                unpoison_pr_info("Unpoison: the hwpoison page has non-NULL mapping %#lx\n",
                                 pfn, &unpoison_rs);
                goto unlock_mutex;
        }
 
-       if (PageSlab(page) || PageTable(page) || PageReserved(page))
+       if (folio_test_slab(folio) || PageTable(&folio->page) || folio_test_reserved(folio))
                goto unlock_mutex;
 
        ret = get_hwpoison_page(p, MF_UNPOISON);
                                goto unlock_mutex;
                        }
                }
-               ret = TestClearPageHWPoison(page) ? 0 : -EBUSY;
+               ret = folio_test_clear_hwpoison(folio) ? 0 : -EBUSY;
        } else if (ret < 0) {
                if (ret == -EHWPOISON) {
                        ret = put_page_back_buddy(p) ? 0 : -EBUSY;
                        count = folio_free_raw_hwp(folio, false);
                        if (count == 0) {
                                ret = -EBUSY;
-                               put_page(page);
+                               folio_put(folio);
                                goto unlock_mutex;
                        }
                }
 
-               put_page(page);
+               folio_put(folio);
                if (TestClearPageHWPoison(p)) {
-                       put_page(page);
+                       folio_put(folio);
                        ret = 0;
                }
        }