*/
 unsigned long restore_jump_address;
 
+/*
+ * Value of the cr3 register from before the hibernation (this value is passed
+ * in the image header).
+ */
+unsigned long restore_cr3;
+
 pgd_t *temp_level4_pgt;
 
 void *relocated_restore_code;
 
 struct restore_data_record {
        unsigned long jump_address;
-       unsigned long control;
+       unsigned long cr3;
+       unsigned long magic;
 };
 
 #define RESTORE_MAGIC  0x0123456789ABCDEFUL
        if (max_size < sizeof(struct restore_data_record))
                return -EOVERFLOW;
        rdr->jump_address = restore_jump_address;
-       rdr->control = (restore_jump_address ^ RESTORE_MAGIC);
+       rdr->cr3 = restore_cr3;
+       rdr->magic = RESTORE_MAGIC;
        return 0;
 }
 
        struct restore_data_record *rdr = addr;
 
        restore_jump_address = rdr->jump_address;
-       return (rdr->control == (restore_jump_address ^ RESTORE_MAGIC)) ?
-                       0 : -EINVAL;
+       restore_cr3 = rdr->cr3;
+       return (rdr->magic == RESTORE_MAGIC) ? 0 : -EINVAL;
 }
 #endif /* CONFIG_HIBERNATION */
 
        /* save the address of restore_registers */
        movq    $restore_registers, %rax
        movq    %rax, restore_jump_address(%rip)
+       /* save cr3 */
+       movq    %cr3, %rax
+       movq    %rax, restore_cr3(%rip)
 
        call swsusp_save
        ret
 
        /* prepare to jump to the image kernel */
        movq    restore_jump_address(%rip), %rax
+       movq    restore_cr3(%rip), %rbx
 
        /* prepare to copy image data to their original locations */
        movq    restore_pblist(%rip), %rdx
 
 ENTRY(restore_registers)
        /* go back to the original page tables */
-       movq    $(init_level4_pgt - __START_KERNEL_map), %rax
-       addq    phys_base(%rip), %rax
-       movq    %rax, %cr3
+       movq    %rbx, %cr3
 
        /* Flush TLB, including "global" things (vmalloc) */
        movq    mmu_cr4_features(%rip), %rax