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
14void __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