1// SPDX-License-Identifier: GPL-2.0-or-later
2/*
3 * Copyright (C) 2010 Loongson Inc. & Lemote Inc. &
4 *                    Institute of Computing Technology
5 * Author:  Xiang Gao, gaoxiang@ict.ac.cn
6 *          Huacai Chen, chenhc@lemote.com
7 *          Xiaofu Meng, Shuangshuang Zhang
8 */
9#include <linux/init.h>
10#include <linux/kernel.h>
11#include <linux/mm.h>
12#include <linux/mmzone.h>
13#include <linux/export.h>
14#include <linux/nodemask.h>
15#include <linux/swap.h>
16#include <linux/memblock.h>
17#include <linux/pfn.h>
18#include <linux/highmem.h>
19#include <asm/page.h>
20#include <asm/pgalloc.h>
21#include <asm/sections.h>
22#include <linux/irq.h>
23#include <asm/bootinfo.h>
24#include <asm/mc146818-time.h>
25#include <asm/time.h>
26#include <asm/wbflush.h>
27#include <boot_param.h>
28#include <loongson.h>
29
30unsigned char __node_distances[MAX_NUMNODES][MAX_NUMNODES];
31EXPORT_SYMBOL(__node_distances);
32struct pglist_data *__node_data[MAX_NUMNODES];
33EXPORT_SYMBOL(__node_data);
34
35cpumask_t __node_cpumask[MAX_NUMNODES];
36EXPORT_SYMBOL(__node_cpumask);
37
38static void cpu_node_probe(void)
39{
40	int i;
41
42	nodes_clear(node_possible_map);
43	nodes_clear(node_online_map);
44	for (i = 0; i < loongson_sysconf.nr_nodes; i++) {
45		node_set_state(num_online_nodes(), N_POSSIBLE);
46		node_set_online(num_online_nodes());
47	}
48
49	pr_info("NUMA: Discovered %d cpus on %d nodes\n",
50		loongson_sysconf.nr_cpus, num_online_nodes());
51}
52
53static int __init compute_node_distance(int row, int col)
54{
55	int package_row = row * loongson_sysconf.cores_per_node /
56				loongson_sysconf.cores_per_package;
57	int package_col = col * loongson_sysconf.cores_per_node /
58				loongson_sysconf.cores_per_package;
59
60	if (col == row)
61		return LOCAL_DISTANCE;
62	else if (package_row == package_col)
63		return 40;
64	else
65		return 100;
66}
67
68static void __init init_topology_matrix(void)
69{
70	int row, col;
71
72	for (row = 0; row < MAX_NUMNODES; row++)
73		for (col = 0; col < MAX_NUMNODES; col++)
74			__node_distances[row][col] = -1;
75
76	for_each_online_node(row) {
77		for_each_online_node(col) {
78			__node_distances[row][col] =
79				compute_node_distance(row, col);
80		}
81	}
82}
83
84static void __init node_mem_init(unsigned int node)
85{
86	struct pglist_data *nd;
87	unsigned long node_addrspace_offset;
88	unsigned long start_pfn, end_pfn;
89	unsigned long nd_pa;
90	int tnid;
91	const size_t nd_size = roundup(sizeof(pg_data_t), SMP_CACHE_BYTES);
92
93	node_addrspace_offset = nid_to_addrbase(node);
94	pr_info("Node%d's addrspace_offset is 0x%lx\n",
95			node, node_addrspace_offset);
96
97	get_pfn_range_for_nid(node, &start_pfn, &end_pfn);
98	pr_info("Node%d: start_pfn=0x%lx, end_pfn=0x%lx\n",
99		node, start_pfn, end_pfn);
100
101	nd_pa = memblock_phys_alloc_try_nid(nd_size, SMP_CACHE_BYTES, node);
102	if (!nd_pa)
103		panic("Cannot allocate %zu bytes for node %d data\n",
104		      nd_size, node);
105	nd = __va(nd_pa);
106	memset(nd, 0, sizeof(struct pglist_data));
107	tnid = early_pfn_to_nid(nd_pa >> PAGE_SHIFT);
108	if (tnid != node)
109		pr_info("NODE_DATA(%d) on node %d\n", node, tnid);
110	__node_data[node] = nd;
111	NODE_DATA(node)->node_start_pfn = start_pfn;
112	NODE_DATA(node)->node_spanned_pages = end_pfn - start_pfn;
113
114	if (node == 0) {
115		/* kernel start address */
116		unsigned long kernel_start_pfn = PFN_DOWN(__pa_symbol(&_text));
117
118		/* kernel end address */
119		unsigned long kernel_end_pfn = PFN_UP(__pa_symbol(&_end));
120
121		/* used by finalize_initrd() */
122		max_low_pfn = end_pfn;
123
124		/* Reserve the kernel text/data/bss */
125		memblock_reserve(kernel_start_pfn << PAGE_SHIFT,
126				 ((kernel_end_pfn - kernel_start_pfn) << PAGE_SHIFT));
127
128		/* Reserve 0xfe000000~0xffffffff for RS780E integrated GPU */
129		if (node_end_pfn(0) >= (0xffffffff >> PAGE_SHIFT))
130			memblock_reserve((node_addrspace_offset | 0xfe000000),
131					 32 << 20);
132
133		/* Reserve pfn range 0~node[0]->node_start_pfn */
134		memblock_reserve(0, PAGE_SIZE * start_pfn);
135		/* set nid for reserved memory on node 0 */
136		memblock_set_node(0, 1ULL << 44, &memblock.reserved, 0);
137	}
138}
139
140static __init void prom_meminit(void)
141{
142	unsigned int node, cpu, active_cpu = 0;
143
144	cpu_node_probe();
145	init_topology_matrix();
146
147	for (node = 0; node < loongson_sysconf.nr_nodes; node++) {
148		if (node_online(node)) {
149			szmem(node);
150			node_mem_init(node);
151			cpumask_clear(&__node_cpumask[node]);
152		}
153	}
154	max_low_pfn = PHYS_PFN(memblock_end_of_DRAM());
155
156	for (cpu = 0; cpu < loongson_sysconf.nr_cpus; cpu++) {
157		node = cpu / loongson_sysconf.cores_per_node;
158		if (node >= num_online_nodes())
159			node = 0;
160
161		if (loongson_sysconf.reserved_cpus_mask & (1<<cpu))
162			continue;
163
164		cpumask_set_cpu(active_cpu, &__node_cpumask[node]);
165		pr_info("NUMA: set cpumask cpu %d on node %d\n", active_cpu, node);
166
167		active_cpu++;
168	}
169}
170
171void __init paging_init(void)
172{
173	unsigned long zones_size[MAX_NR_ZONES] = {0, };
174
175	pagetable_init();
176	zones_size[ZONE_DMA32] = MAX_DMA32_PFN;
177	zones_size[ZONE_NORMAL] = max_low_pfn;
178	free_area_init(zones_size);
179}
180
181void __init mem_init(void)
182{
183	high_memory = (void *) __va(get_num_physpages() << PAGE_SHIFT);
184	memblock_free_all();
185	setup_zero_pages();	/* This comes from node 0 */
186}
187
188/* All PCI device belongs to logical Node-0 */
189int pcibus_to_node(struct pci_bus *bus)
190{
191	return 0;
192}
193EXPORT_SYMBOL(pcibus_to_node);
194
195void __init prom_init_numa_memory(void)
196{
197	pr_info("CP0_Config3: CP0 16.3 (0x%x)\n", read_c0_config3());
198	pr_info("CP0_PageGrain: CP0 5.1 (0x%x)\n", read_c0_pagegrain());
199	prom_meminit();
200}
201
202pg_data_t * __init arch_alloc_nodedata(int nid)
203{
204	return memblock_alloc(sizeof(pg_data_t), SMP_CACHE_BYTES);
205}
206
207void arch_refresh_nodedata(int nid, pg_data_t *pgdat)
208{
209	__node_data[nid] = pgdat;
210}
211