Gitiles
Code Review Sign In
gerrit.devboardsforandroid.linaro.org / platform / external / u-boot / fad5ec5ecd7b97325f5bf686a195760932af67b0 / . / test / cmd / mem_search.c
blob: d57bfad39823966a78bd5b9d6f5ffaa3823fe382 [file] [log] [blame]
Simon Glassbdded202020-06-02 19:26:49 -0600[diff] [blame]1// SPDX-License-Identifier: GPL-2.0+
2/*
3 * Tests for memory commands
4 *
5 * Copyright 2020 Google LLC
6 * Written by Simon Glass <sjg@chromium.org>
7 */
8
9#include <common.h>
10#include <console.h>
11#include <mapmem.h>
12#include <dm/test.h>
13#include <test/ut.h>
14
15#define BUF_SIZE 0x100
16
17/* Test 'ms' command with bytes */
18static int dm_test_ms_b(struct unit_test_state *uts)
19{
20 u8 *buf;
21
22 buf = map_sysmem(0, BUF_SIZE + 1);
23 memset(buf, '\0', BUF_SIZE);
24 buf[0x0] = 0x12;
25 buf[0x31] = 0x12;
26 buf[0xff] = 0x12;
27 buf[0x100] = 0x12;
28 console_record_reset();
29 run_command("ms.b 1 ff 12", 0);
30 ut_assert_nextline("00000030: 00 12 00 00 00 00 00 00 00 00 00 00 00 00 00 00 ................");
31 ut_assert_nextline("--");
32 ut_assert_nextline("000000f0: 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 12 ................");
33 ut_assert_nextline("2 matches");
34 ut_assert_console_end();
35
36 ut_asserteq(2, env_get_hex("memmatches", 0));
37 ut_asserteq(0xff, env_get_hex("memaddr", 0));
38 ut_asserteq(0xfe, env_get_hex("mempos", 0));
39
40 unmap_sysmem(buf);
41
42 return 0;
43}
44DM_TEST(dm_test_ms_b, 0);
45
46/* Test 'ms' command with 16-bit values */
47static int dm_test_ms_w(struct unit_test_state *uts)
48{
49 u16 *buf;
50
51 buf = map_sysmem(0, BUF_SIZE + 2);
52 memset(buf, '\0', BUF_SIZE);
53 buf[0x34 / 2] = 0x1234;
54 buf[BUF_SIZE / 2] = 0x1234;
55 console_record_reset();
56 run_command("ms.w 0 80 1234", 0);
57 ut_assert_nextline("00000030: 0000 0000 1234 0000 0000 0000 0000 0000 ....4...........");
58 ut_assert_nextline("1 match");
59 ut_assert_console_end();
60
61 ut_asserteq(1, env_get_hex("memmatches", 0));
62 ut_asserteq(0x34, env_get_hex("memaddr", 0));
63 ut_asserteq(0x34 / 2, env_get_hex("mempos", 0));
64
65 unmap_sysmem(buf);
66
67 return 0;
68}
69DM_TEST(dm_test_ms_w, 0);
70
71/* Test 'ms' command with 32-bit values */
72static int dm_test_ms_l(struct unit_test_state *uts)
73{
74 u32 *buf;
75
76 buf = map_sysmem(0, BUF_SIZE + 4);
77 memset(buf, '\0', BUF_SIZE);
78 buf[0x38 / 4] = 0x12345678;
79 buf[BUF_SIZE / 4] = 0x12345678;
80 console_record_reset();
81 run_command("ms 0 40 12345678", 0);
82 ut_assert_nextline("00000030: 00000000 00000000 12345678 00000000 ........xV4.....");
83 ut_assert_nextline("1 match");
84 ut_assert_console_end();
85
86 ut_asserteq(1, env_get_hex("memmatches", 0));
87 ut_asserteq(0x38, env_get_hex("memaddr", 0));
88 ut_asserteq(0x38 / 4, env_get_hex("mempos", 0));
89
90 console_record_reset();
91 run_command("ms 0 80 12345679", 0);
92 ut_assert_nextline("0 matches");
93 ut_assert_console_end();
94
95 ut_asserteq(0, env_get_hex("memmatches", 0));
96 ut_asserteq(0, env_get_hex("memaddr", 0));
97 ut_asserteq(0 / 4, env_get_hex("mempos", 0));
98
99 unmap_sysmem(buf);
100
101 return 0;
102}
103DM_TEST(dm_test_ms_l, 0);
104
105/* Test 'ms' command with continuation */
106static int dm_test_ms_cont(struct unit_test_state *uts)
107{
108 char *const args[] = {"ms.b", "0", "100", "34"};
109 int repeatable;
110 u8 *buf;
111 int i;
112
113 buf = map_sysmem(0, BUF_SIZE);
114 memset(buf, '\0', BUF_SIZE);
115 for (i = 5; i < 0x33; i += 3)
116 buf[i] = 0x34;
117 console_record_reset();
118 run_command("ms.b 0 100 34", 0);
119 ut_assert_nextlinen("00000000: 00 00 00 00 00 34 00 00 34 00 00 34 00 00 34 00");
120 ut_assert_nextline("--");
121 ut_assert_nextlinen("00000010: 00 34 00 00 34 00 00 34 00 00 34 00 00 34 00 00");
122 ut_assert_nextline("--");
123 ut_assert_nextlinen("00000020: 34 00 00 34 00 00 34 00 00 34 00 00 34 00 00 34");
124 ut_assert_nextlinen("10 matches (repeat command to check for more)");
125 ut_assert_console_end();
126
127 ut_asserteq(10, env_get_hex("memmatches", 0));
128 ut_asserteq(0x20, env_get_hex("memaddr", 0));
129 ut_asserteq(0x20, env_get_hex("mempos", 0));
130
131 /*
132 * run_command() ignoes the repeatable flag when using hush, so call
133 * cmd_process() directly
134 */
135 console_record_reset();
136 cmd_process(CMD_FLAG_REPEAT, 4, args, &repeatable, NULL);
137 ut_assert_nextlinen("00000020: 34 00 00 34 00 00 34 00 00 34 00 00 34 00 00 34");
138 ut_assert_nextline("--");
139 ut_assert_nextlinen("00000030: 00 00 34 00 00 00 00 00");
140 ut_assert_nextlinen("6 matches");
141 ut_assert_console_end();
142
143 ut_asserteq(6, env_get_hex("memmatches", 0));
144 ut_asserteq(0x32, env_get_hex("memaddr", 0));
145
146 /* 0x32 less 0x21, where the second search started */
147 ut_asserteq(0x11, env_get_hex("mempos", 0));
148
149 unmap_sysmem(buf);
150
151 return 0;
152}
153DM_TEST(dm_test_ms_cont, 0);
154
155/* Test 'ms' command with multiple values */
156static int dm_test_ms_mult(struct unit_test_state *uts)
157{
158 static const char str[] = "hello";
159 char *buf;
160
161 buf = map_sysmem(0, BUF_SIZE + 5);
162 memset(buf, '\0', BUF_SIZE);
163 strcpy(buf + 0x1e, str);
164 strcpy(buf + 0x63, str);
165 strcpy(buf + BUF_SIZE - strlen(str) + 1, str);
166 console_record_reset();
167 run_command("ms.b 0 100 68 65 6c 6c 6f", 0);
168 ut_assert_nextline("00000010: 00 00 00 00 00 00 00 00 00 00 00 00 00 00 68 65 ..............he");
169 ut_assert_nextline("00000020: 6c 6c 6f 00 00 00 00 00 00 00 00 00 00 00 00 00 llo.............");
170 ut_assert_nextline("--");
171 ut_assert_nextline("00000060: 00 00 00 68 65 6c 6c 6f 00 00 00 00 00 00 00 00 ...hello........");
172 ut_assert_nextline("2 matches");
173 ut_assert_console_end();
174 unmap_sysmem(buf);
175
176 ut_asserteq(2, env_get_hex("memmatches", 0));
177 ut_asserteq(0x63, env_get_hex("memaddr", 0));
178 ut_asserteq(0x63, env_get_hex("mempos", 0));
179
180 return 0;
181}
182DM_TEST(dm_test_ms_mult, 0);
183
184/* Test 'ms' command with string */
185static int dm_test_ms_s(struct unit_test_state *uts)
186{
187 static const char str[] = "hello";
188 static const char str2[] = "hellothere";
189 char *buf;
190
191 buf = map_sysmem(0, BUF_SIZE);
192 memset(buf, '\0', BUF_SIZE);
193 strcpy(buf + 0x1e, str);
194 strcpy(buf + 0x63, str);
195 strcpy(buf + 0xa1, str2);
196 console_record_reset();
197 run_command("ms.s 0 100 hello", 0);
198 ut_assert_nextline("00000010: 00 00 00 00 00 00 00 00 00 00 00 00 00 00 68 65 ..............he");
199 ut_assert_nextline("00000020: 6c 6c 6f 00 00 00 00 00 00 00 00 00 00 00 00 00 llo.............");
200 ut_assert_nextline("--");
201 ut_assert_nextline("00000060: 00 00 00 68 65 6c 6c 6f 00 00 00 00 00 00 00 00 ...hello........");
202 ut_assert_nextline("--");
203 ut_assert_nextline("000000a0: 00 68 65 6c 6c 6f 74 68 65 72 65 00 00 00 00 00 .hellothere.....");
204 ut_assert_nextline("3 matches");
205 ut_assert_console_end();
206
207 ut_asserteq(3, env_get_hex("memmatches", 0));
208 ut_asserteq(0xa1, env_get_hex("memaddr", 0));
209 ut_asserteq(0xa1, env_get_hex("mempos", 0));
210
211 console_record_reset();
212 run_command("ms.s 0 100 hello there", 0);
213 ut_assert_nextline("000000a0: 00 68 65 6c 6c 6f 74 68 65 72 65 00 00 00 00 00 .hellothere.....");
214 ut_assert_nextline("1 match");
215 ut_assert_console_end();
216
217 ut_asserteq(1, env_get_hex("memmatches", 0));
218 ut_asserteq(0xa1, env_get_hex("memaddr", 0));
219 ut_asserteq(0xa1, env_get_hex("mempos", 0));
220
221 unmap_sysmem(buf);
222
223 return 0;
224}
225DM_TEST(dm_test_ms_s, 0);
226
227/* Test 'ms' command with limit */
228static int dm_test_ms_limit(struct unit_test_state *uts)
229{
230 u8 *buf;
231
232 buf = map_sysmem(0, BUF_SIZE + 1);
233 memset(buf, '\0', BUF_SIZE);
234 buf[0x0] = 0x12;
235 buf[0x31] = 0x12;
236 buf[0x62] = 0x12;
237 buf[0x76] = 0x12;
238 console_record_reset();
239 run_command("ms.b -l2 1 ff 12", 0);
240 ut_assert_nextline("00000030: 00 12 00 00 00 00 00 00 00 00 00 00 00 00 00 00 ................");
241 ut_assert_nextline("--");
242 ut_assert_nextlinen("00000060: 00 00 12 00 00 00 00 00 00 00 00 00 00 00 00 00");
243 ut_assert_nextline("2 matches (repeat command to check for more)");
244 ut_assert_console_end();
245
246 ut_asserteq(2, env_get_hex("memmatches", 0));
247 ut_asserteq(0x62, env_get_hex("memaddr", 0));
248 ut_asserteq(0x61, env_get_hex("mempos", 0));
249
250 unmap_sysmem(buf);
251
252 return 0;
253}
254DM_TEST(dm_test_ms_limit, 0);
255
256/* Test 'ms' command in quiet mode */
257static int dm_test_ms_quiet(struct unit_test_state *uts)
258{
259 u8 *buf;
260
261 buf = map_sysmem(0, BUF_SIZE + 1);
262 memset(buf, '\0', BUF_SIZE);
263 buf[0x0] = 0x12;
264 buf[0x31] = 0x12;
265 buf[0x62] = 0x12;
266 buf[0x76] = 0x12;
267 console_record_reset();
268 run_command("ms.b -l2 1 ff 12", 0);
269 ut_assert_console_end();
270 unmap_sysmem(buf);
271
272 return 0;
273}
274DM_TEST(dm_test_ms_quiet, 0);
275