1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Copyright (C) 2020-2022 Loongson Technology Corporation Limited
4  */
5 #include <linux/efi.h>
6 #include <linux/initrd.h>
7 #include <linux/memblock.h>
8 
9 #include <asm/bootinfo.h>
10 #include <asm/sections.h>
11 
12 #include <loongson.h>
13 
memblock_init(void)14 void __init memblock_init(void)
15 {
16 	u32 mem_type;
17 	u64 mem_start, mem_end, mem_size;
18 	efi_memory_desc_t *md;
19 
20 	/* Parse memory information */
21 	for_each_efi_memory_desc(md) {
22 		mem_type = md->type;
23 		mem_start = md->phys_addr;
24 		mem_size = md->num_pages << EFI_PAGE_SHIFT;
25 		mem_end = mem_start + mem_size;
26 
27 		switch (mem_type) {
28 		case EFI_LOADER_CODE:
29 		case EFI_LOADER_DATA:
30 		case EFI_BOOT_SERVICES_CODE:
31 		case EFI_BOOT_SERVICES_DATA:
32 		case EFI_PERSISTENT_MEMORY:
33 		case EFI_CONVENTIONAL_MEMORY:
34 			memblock_add(mem_start, mem_size);
35 			if (max_low_pfn < (mem_end >> PAGE_SHIFT))
36 				max_low_pfn = mem_end >> PAGE_SHIFT;
37 			break;
38 		case EFI_PAL_CODE:
39 		case EFI_UNUSABLE_MEMORY:
40 		case EFI_ACPI_RECLAIM_MEMORY:
41 			memblock_add(mem_start, mem_size);
42 			fallthrough;
43 		case EFI_RESERVED_TYPE:
44 		case EFI_RUNTIME_SERVICES_CODE:
45 		case EFI_RUNTIME_SERVICES_DATA:
46 		case EFI_MEMORY_MAPPED_IO:
47 		case EFI_MEMORY_MAPPED_IO_PORT_SPACE:
48 			memblock_reserve(mem_start, mem_size);
49 			break;
50 		}
51 	}
52 
53 	memblock_set_current_limit(PFN_PHYS(max_low_pfn));
54 
55 	/* Reserve the first 2MB */
56 	memblock_reserve(PHYS_OFFSET, 0x200000);
57 
58 	/* Reserve the kernel text/data/bss */
59 	memblock_reserve(__pa_symbol(&_text),
60 			 __pa_symbol(&_end) - __pa_symbol(&_text));
61 
62 	memblock_set_node(0, PHYS_ADDR_MAX, &memblock.memory, 0);
63 	memblock_set_node(0, PHYS_ADDR_MAX, &memblock.reserved, 0);
64 }
65