void sclp_early_setup_buffer(void);
 void print_pgm_check_info(void);
 unsigned long get_random_base(unsigned long safe_addr);
-void setup_vmem(unsigned long online_end, unsigned long asce_limit);
+void setup_vmem(unsigned long ident_map_size, unsigned long asce_limit);
 void __printf(1, 2) decompressor_printk(const char *fmt, ...);
 void error(char *m);
 
 
        unsigned long random_lma;
        unsigned long safe_addr;
        unsigned long asce_limit;
-       unsigned long online_end;
        void *img;
        psw_t psw;
 
        setup_ident_map_size(detect_memory());
        setup_vmalloc_size();
        asce_limit = setup_kernel_memory_layout();
-       online_end = min(get_mem_detect_end(), ident_map_size);
 
        if (IS_ENABLED(CONFIG_RANDOMIZE_BASE) && kaslr_enabled) {
                random_lma = get_random_base(safe_addr);
         */
        clear_bss_section();
        handle_relocs(__kaslr_offset);
-       setup_vmem(online_end, asce_limit);
+       setup_vmem(ident_map_size, asce_limit);
        copy_bootdata();
 
        if (__kaslr_offset) {
 
                error("out of memory on boot\n");
 }
 
-static void pgtable_populate_begin(unsigned long online_end)
+static void pgtable_populate_begin(unsigned long ident_map_size)
 {
        unsigned long initrd_end;
        unsigned long kernel_end;
                pgalloc_low = max(pgalloc_low, initrd_end);
        }
 
-       pgalloc_end = round_down(online_end, PAGE_SIZE);
+       pgalloc_end = round_down(min(ident_map_size, get_mem_detect_end()), PAGE_SIZE);
        pgalloc_pos = pgalloc_end;
 
        boot_check_oom();
        } while (pgalloc_pos < pgalloc_pos_prev);
 }
 
-void setup_vmem(unsigned long online_end, unsigned long asce_limit)
+void setup_vmem(unsigned long ident_map_size, unsigned long asce_limit)
 {
+       unsigned long start, end;
        unsigned long asce_type;
        unsigned long asce_bits;
+       int i;
 
        if (asce_limit == _REGION1_SIZE) {
                asce_type = _REGION2_ENTRY_EMPTY;
         * No further pgtable_populate() calls are allowed after the value
         * of pgalloc_pos finalized with a call to pgtable_populate_end().
         */
-       pgtable_populate_begin(online_end);
+       pgtable_populate_begin(ident_map_size);
        pgtable_populate(0, sizeof(struct lowcore), POPULATE_ONE2ONE);
-       pgtable_populate(0, online_end, POPULATE_ONE2ONE);
+       for_each_mem_detect_block(i, &start, &end) {
+               if (start >= ident_map_size)
+                       break;
+               pgtable_populate(start, min(end, ident_map_size), POPULATE_ONE2ONE);
+       }
        pgtable_populate(__abs_lowcore, __abs_lowcore + sizeof(struct lowcore),
                         POPULATE_ABS_LOWCORE);
        pgtable_populate(__memcpy_real_area, __memcpy_real_area + PAGE_SIZE,