return ret;
 }
 
-/* Mark all sections offline and remove all free pages from the buddy. */
-static int
-offline_isolated_pages_cb(unsigned long start, unsigned long nr_pages,
-                       void *data)
-{
-       unsigned long *offlined_pages = (unsigned long *)data;
-
-       *offlined_pages += __offline_isolated_pages(start, start + nr_pages);
-       return 0;
-}
-
-/*
- * Check all pages in range, recorded as memory resource, are isolated.
- */
-static int
-check_pages_isolated_cb(unsigned long start_pfn, unsigned long nr_pages,
-                       void *data)
-{
-       return test_pages_isolated(start_pfn, start_pfn + nr_pages,
-                                  MEMORY_OFFLINE);
-}
-
 static int __init cmdline_parse_movable_node(char *p)
 {
        movable_node_enabled = true;
 int __ref offline_pages(unsigned long start_pfn, unsigned long nr_pages)
 {
        const unsigned long end_pfn = start_pfn + nr_pages;
-       unsigned long pfn, system_ram_pages = 0, offlined_pages = 0;
+       unsigned long pfn, system_ram_pages = 0;
        int ret, node, nr_isolate_pageblock;
        unsigned long flags;
        struct zone *zone;
                        reason = "failure to dissolve huge pages";
                        goto failed_removal_isolated;
                }
-               /* check again */
-               ret = walk_system_ram_range(start_pfn, end_pfn - start_pfn,
-                                           NULL, check_pages_isolated_cb);
+
                /*
                 * per-cpu pages are drained in start_isolate_page_range, but if
                 * there are still pages that are not free, make sure that we
                 * because has_unmovable_pages explicitly checks for
                 * PageBuddy on freed pages on other zones.
                 */
+               ret = test_pages_isolated(start_pfn, end_pfn, MEMORY_OFFLINE);
                if (ret)
                        drain_all_pages(zone);
        } while (ret);
 
-       /* Ok, all of our target is isolated.
-          We cannot do rollback at this point. */
-       walk_system_ram_range(start_pfn, end_pfn - start_pfn,
-                             &offlined_pages, offline_isolated_pages_cb);
-       pr_info("Offlined Pages %ld\n", offlined_pages);
+       /* Mark all sections offline and remove free pages from the buddy. */
+       __offline_isolated_pages(start_pfn, end_pfn);
+       pr_info("Offlined Pages %ld\n", nr_pages);
+
        /*
         * Onlining will reset pagetype flags and makes migrate type
         * MOVABLE, so just need to decrease the number of isolated
        spin_unlock_irqrestore(&zone->lock, flags);
 
        /* removal success */
-       adjust_managed_page_count(pfn_to_page(start_pfn), -offlined_pages);
-       zone->present_pages -= offlined_pages;
+       adjust_managed_page_count(pfn_to_page(start_pfn), -nr_pages);
+       zone->present_pages -= nr_pages;
 
        pgdat_resize_lock(zone->zone_pgdat, &flags);
-       zone->zone_pgdat->node_present_pages -= offlined_pages;
+       zone->zone_pgdat->node_present_pages -= nr_pages;
        pgdat_resize_unlock(zone->zone_pgdat, &flags);
 
        init_per_zone_wmark_min();