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 <common.h>
12#include <command.h>
13#include <mapmem.h>
14#include <part.h>
15
16static int
17do_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 *addr;
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	addr = 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		return 1;
52	}
53
54	if (IS_ENABLED(CONFIG_CMD_WRITE) && !strcmp(cmdtp->name, "write"))
55		res = blk_dwrite(dev_desc, offset + blk, cnt, addr);
56	else
57		res = blk_dread(dev_desc, offset + blk, cnt, addr);
58
59	if (res != cnt) {
60		printf("%s error\n", cmdtp->name);
61		return 1;
62	}
63
64	return 0;
65}
66
67#ifdef CONFIG_CMD_READ
68U_BOOT_CMD(
69	read,	6,	0,	do_rw,
70	"Load binary data from a partition",
71	"<interface> <dev[:part|#partname]> addr blk# cnt"
72);
73#endif
74
75#ifdef CONFIG_CMD_WRITE
76U_BOOT_CMD(
77	write,	6,	0,	do_rw,
78	"Store binary data to a partition",
79	"<interface> <dev[:part|#partname]> addr blk# cnt"
80);
81#endif
82