arc: handle pgtable_page_ctor() fail

Signed-off-by: Kirill A. Shutemov <kirill.shutemov@linux.intel.com>
Acked-by: Vineet Gupta <vgupta@synopsys.com> [for arch/arc bits]
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>

1 files changed, 8 insertions, 3 deletions

diff --git a/arch/arc/include/asm/pgalloc.h b/arch/arc/include/asm/pgalloc.h
index 36a9f20c21a3..81208bfd9dcb 100644
--- a/arch/arc/include/asm/pgalloc.h
+++ b/arch/arc/include/asm/pgalloc.h
@@ -105,11 +105,16 @@ static inline pgtable_t
 pte_alloc_one(struct mm_struct *mm, unsigned long address)
 {
         pgtable_t pte_pg;
+        struct page *page;
 
         pte_pg = __get_free_pages(GFP_KERNEL | __GFP_REPEAT, __get_order_pte());
-        if (pte_pg) {
-                memzero((void *)pte_pg, PTRS_PER_PTE * 4);
-                pgtable_page_ctor(virt_to_page(pte_pg));
+        if (!pte_pg)
+                return 0;
+        memzero((void *)pte_pg, PTRS_PER_PTE * 4);
+        page = virt_to_page(pte_pg);
+        if (!pgtable_page_ctor(page)) {
+                __free_page(page);
+                return 0;
         }
 
         return pte_pg;