2 files changed, 9 insertions, 6 deletions

diff --git a/arch/x86/mm/fault.c b/arch/x86/mm/fault.c
index d8ed4006b3d2..621afb6343dc 100644
--- a/arch/x86/mm/fault.c
+++ b/arch/x86/mm/fault.c
@@ -958,11 +958,12 @@ void vmalloc_sync_all(void)
         for (address = start; address <= VMALLOC_END; address += PGDIR_SIZE) {
                 if (!test_bit(pgd_index(address), insync)) {
                         const pgd_t *pgd_ref = pgd_offset_k(address);
+                        unsigned long flags;
                         struct page *page;
 
                         if (pgd_none(*pgd_ref))
                                 continue;
-                        spin_lock(&pgd_lock);
+                        spin_lock_irqsave(&pgd_lock, flags);
                         list_for_each_entry(page, &pgd_list, lru) {
                                 pgd_t *pgd;
                                 pgd = (pgd_t *)page_address(page) + pgd_index(address);
@@ -971,7 +972,7 @@ void vmalloc_sync_all(void)
                                 else
                                         BUG_ON(pgd_page_vaddr(*pgd) != pgd_page_vaddr(*pgd_ref));
                         }
-                        spin_unlock(&pgd_lock);
+                        spin_unlock_irqrestore(&pgd_lock, flags);
                         set_bit(pgd_index(address), insync);
                 }
                 if (address == start)
diff --git a/include/asm-x86/pgalloc_64.h b/include/asm-x86/pgalloc_64.h
index 315314ce4bfb..4f6220db22b1 100644
--- a/include/asm-x86/pgalloc_64.h
+++ b/include/asm-x86/pgalloc_64.h
@@ -42,19 +42,21 @@ static inline void pud_free(struct mm_struct *mm, pud_t *pud)
 static inline void pgd_list_add(pgd_t *pgd)
 {
         struct page *page = virt_to_page(pgd);
+        unsigned long flags;
 
-        spin_lock(&pgd_lock);
+        spin_lock_irqsave(&pgd_lock, flags);
         list_add(&page->lru, &pgd_list);
-        spin_unlock(&pgd_lock);
+        spin_unlock_irqrestore(&pgd_lock, flags);
 }
 
 static inline void pgd_list_del(pgd_t *pgd)
 {
         struct page *page = virt_to_page(pgd);
+        unsigned long flags;
 
-        spin_lock(&pgd_lock);
+        spin_lock_irqsave(&pgd_lock, flags);
         list_del(&page->lru);
-        spin_unlock(&pgd_lock);
+        spin_unlock_irqrestore(&pgd_lock, flags);
 }
 
 static inline pgd_t *pgd_alloc(struct mm_struct *mm)