1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * (C) Copyright 2014 Google, Inc
4 */
5
6 #include <common.h>
7 #include <command.h>
8 #include <log.h>
9 #include <asm/msr.h>
10 #include <asm/mp.h>
11 #include <asm/mtrr.h>
12
13 static const char *const mtrr_type_name[MTRR_TYPE_COUNT] = {
14 "Uncacheable",
15 "Combine",
16 "2",
17 "3",
18 "Through",
19 "Protect",
20 "Back",
21 };
22
read_mtrrs(void * arg)23 static void read_mtrrs(void *arg)
24 {
25 struct mtrr_info *info = arg;
26
27 mtrr_read_all(info);
28 }
29
do_mtrr_list(int reg_count,int cpu_select)30 static int do_mtrr_list(int reg_count, int cpu_select)
31 {
32 struct mtrr_info info;
33 int ret;
34 int i;
35
36 printf("Reg Valid Write-type %-16s %-16s %-16s\n", "Base ||",
37 "Mask ||", "Size ||");
38 memset(&info, '\0', sizeof(info));
39 ret = mp_run_on_cpus(cpu_select, read_mtrrs, &info);
40 if (ret)
41 return log_msg_ret("run", ret);
42 for (i = 0; i < reg_count; i++) {
43 const char *type = "Invalid";
44 uint64_t base, mask, size;
45 bool valid;
46
47 base = info.mtrr[i].base;
48 mask = info.mtrr[i].mask;
49 size = ~mask & ((1ULL << CONFIG_CPU_ADDR_BITS) - 1);
50 size |= (1 << 12) - 1;
51 size += 1;
52 valid = mask & MTRR_PHYS_MASK_VALID;
53 type = mtrr_type_name[base & MTRR_BASE_TYPE_MASK];
54 printf("%d %-5s %-12s %016llx %016llx %016llx\n", i,
55 valid ? "Y" : "N", type, base & ~MTRR_BASE_TYPE_MASK,
56 mask & ~MTRR_PHYS_MASK_VALID, size);
57 }
58
59 return 0;
60 }
61
do_mtrr_set(int cpu_select,uint reg,int argc,char * const argv[])62 static int do_mtrr_set(int cpu_select, uint reg, int argc, char *const argv[])
63 {
64 const char *typename = argv[0];
65 uint32_t start, size;
66 uint64_t base, mask;
67 int i, type = -1;
68 bool valid;
69 int ret;
70
71 if (argc < 3)
72 return CMD_RET_USAGE;
73 for (i = 0; i < MTRR_TYPE_COUNT; i++) {
74 if (*typename == *mtrr_type_name[i])
75 type = i;
76 }
77 if (type == -1) {
78 printf("Invalid type name %s\n", typename);
79 return CMD_RET_USAGE;
80 }
81 start = simple_strtoul(argv[1], NULL, 16);
82 size = simple_strtoul(argv[2], NULL, 16);
83
84 base = start | type;
85 valid = native_read_msr(MTRR_PHYS_MASK_MSR(reg)) & MTRR_PHYS_MASK_VALID;
86 mask = ~((uint64_t)size - 1);
87 mask &= (1ULL << CONFIG_CPU_ADDR_BITS) - 1;
88 if (valid)
89 mask |= MTRR_PHYS_MASK_VALID;
90
91 ret = mtrr_set(cpu_select, reg, base, mask);
92 if (ret)
93 return CMD_RET_FAILURE;
94
95 return 0;
96 }
97
do_mtrr(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])98 static int do_mtrr(struct cmd_tbl *cmdtp, int flag, int argc,
99 char *const argv[])
100 {
101 int reg_count = mtrr_get_var_count();
102 int cmd;
103 int cpu_select;
104 uint reg;
105 int ret;
106
107 cpu_select = MP_SELECT_BSP;
108 if (argc >= 3 && !strcmp("-c", argv[1])) {
109 const char *cpustr;
110
111 cpustr = argv[2];
112 if (*cpustr == 'a')
113 cpu_select = MP_SELECT_ALL;
114 else
115 cpu_select = simple_strtol(cpustr, NULL, 16);
116 argc -= 2;
117 argv += 2;
118 }
119 argc--;
120 argv++;
121 cmd = argv[0] ? *argv[0] : 0;
122 if (argc < 1 || !cmd) {
123 cmd = 'l';
124 reg = 0;
125 }
126 if (cmd != 'l') {
127 if (argc < 2)
128 return CMD_RET_USAGE;
129 reg = simple_strtoul(argv[1], NULL, 16);
130 if (reg >= reg_count) {
131 printf("Invalid register number\n");
132 return CMD_RET_USAGE;
133 }
134 }
135 if (cmd == 'l') {
136 bool first;
137 int i;
138
139 i = mp_first_cpu(cpu_select);
140 if (i < 0) {
141 printf("Invalid CPU (err=%d)\n", i);
142 return CMD_RET_FAILURE;
143 }
144 first = true;
145 for (; i >= 0; i = mp_next_cpu(cpu_select, i)) {
146 if (!first)
147 printf("\n");
148 printf("CPU %d:\n", i);
149 ret = do_mtrr_list(reg_count, i);
150 if (ret) {
151 printf("Failed to read CPU %d (err=%d)\n", i,
152 ret);
153 return CMD_RET_FAILURE;
154 }
155 first = false;
156 }
157 } else {
158 switch (cmd) {
159 case 'e':
160 ret = mtrr_set_valid(cpu_select, reg, true);
161 break;
162 case 'd':
163 ret = mtrr_set_valid(cpu_select, reg, false);
164 break;
165 case 's':
166 ret = do_mtrr_set(cpu_select, reg, argc - 2, argv + 2);
167 break;
168 default:
169 return CMD_RET_USAGE;
170 }
171 if (ret) {
172 printf("Operation failed (err=%d)\n", ret);
173 return CMD_RET_FAILURE;
174 }
175 }
176
177 return 0;
178 }
179
180 U_BOOT_CMD(
181 mtrr, 8, 1, do_mtrr,
182 "Use x86 memory type range registers (32-bit only)",
183 "[list] - list current registers\n"
184 "set <reg> <type> <start> <size> - set a register\n"
185 "\t<type> is Uncacheable, Combine, Through, Protect, Back\n"
186 "disable <reg> - disable a register\n"
187 "enable <reg> - enable a register\n"
188 "\n"
189 "Precede command with '-c <n>|all' to access a particular hex CPU, e.g.\n"
190 " mtrr -c all list; mtrr -c 2e list"
191 );
192