mm/hugetlb: expand restore_reserve_on_error functionality