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