1 /*
2 * Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
3 * Use of this source code is governed by a BSD-style license that can be
4 * found in the LICENSE file.
5 *
6 * Alternatively, this software may be distributed under the terms of the
7 * GNU General Public License ("GPL") version 2 as published by the Free
8 * Software Foundation.
9 */
10
11 #include <command.h>
12 #include <mapmem.h>
13 #include <part.h>
14 #include <vsprintf.h>
15
16 static int
do_rw(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])17 do_rw(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
18 {
19 struct blk_desc *dev_desc = NULL;
20 struct disk_partition part_info;
21 ulong offset, limit;
22 uint blk, cnt, res;
23 void *ptr;
24 int part;
25
26 if (argc != 6) {
27 cmd_usage(cmdtp);
28 return 1;
29 }
30
31 part = part_get_info_by_dev_and_name_or_num(argv[1], argv[2],
32 &dev_desc, &part_info, 1);
33 if (part < 0)
34 return 1;
35
36 ptr = map_sysmem(hextoul(argv[3], NULL), 0);
37 blk = hextoul(argv[4], NULL);
38 cnt = hextoul(argv[5], NULL);
39
40 if (part > 0) {
41 offset = part_info.start;
42 limit = part_info.size;
43 } else {
44 /* Largest address not available in struct blk_desc. */
45 offset = 0;
46 limit = ~0;
47 }
48
49 if (cnt + blk > limit) {
50 printf("%s out of range\n", cmdtp->name);
51 unmap_sysmem(ptr);
52 return 1;
53 }
54
55 if (IS_ENABLED(CONFIG_CMD_WRITE) && !strcmp(cmdtp->name, "write"))
56 res = blk_dwrite(dev_desc, offset + blk, cnt, ptr);
57 else
58 res = blk_dread(dev_desc, offset + blk, cnt, ptr);
59 unmap_sysmem(ptr);
60
61 if (res != cnt) {
62 printf("%s error\n", cmdtp->name);
63 return 1;
64 }
65
66 return 0;
67 }
68
69 #ifdef CONFIG_CMD_READ
70 U_BOOT_CMD(
71 read, 6, 0, do_rw,
72 "Load binary data from a partition",
73 "<interface> <dev[:part|#partname]> addr blk# cnt"
74 );
75 #endif
76
77 #ifdef CONFIG_CMD_WRITE
78 U_BOOT_CMD(
79 write, 6, 0, do_rw,
80 "Store binary data to a partition",
81 "<interface> <dev[:part|#partname]> addr blk# cnt"
82 );
83 #endif
84