1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 *
4 * Driver interface derived from:
5 * /cmd/host.c
6 * Copyright (c) 2012, Google Inc.
7 *
8 * Copyright (C) 2023 Johan Jonker <jbx6244@gmail.com>
9 */
10
11 #include <blk.h>
12 #include <command.h>
13 #include <dm.h>
14 #include <rkmtd.h>
15 #include <stdio.h>
16 #include <dm/device-internal.h>
17 #include <dm/uclass-internal.h>
18
do_rkmtd_bind(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])19 static int do_rkmtd_bind(struct cmd_tbl *cmdtp, int flag, int argc,
20 char *const argv[])
21 {
22 struct udevice *dev;
23 const char *label;
24 int ret;
25
26 argc--;
27 argv++;
28
29 if (argc < 1)
30 return CMD_RET_USAGE;
31
32 if (argc > 1)
33 return CMD_RET_USAGE;
34
35 label = argv[0];
36 ret = rkmtd_create_attach_mtd(label, &dev);
37 if (ret) {
38 printf("Cannot create device / bind mtd\n");
39 return CMD_RET_FAILURE;
40 }
41
42 return 0;
43 }
44
parse_rkmtd_label(const char * label)45 static struct udevice *parse_rkmtd_label(const char *label)
46 {
47 struct udevice *dev;
48
49 dev = rkmtd_find_by_label(label);
50 if (!dev) {
51 int devnum;
52 char *ep;
53
54 devnum = hextoul(label, &ep);
55 if (*ep ||
56 uclass_find_device_by_seq(UCLASS_RKMTD, devnum, &dev)) {
57 printf("No such device '%s'\n", label);
58 return NULL;
59 }
60 }
61
62 return dev;
63 }
64
do_rkmtd_unbind(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])65 static int do_rkmtd_unbind(struct cmd_tbl *cmdtp, int flag, int argc,
66 char *const argv[])
67 {
68 struct udevice *dev;
69 const char *label;
70 int ret;
71
72 if (argc < 2)
73 return CMD_RET_USAGE;
74
75 label = argv[1];
76 dev = parse_rkmtd_label(label);
77 if (!dev)
78 return CMD_RET_FAILURE;
79
80 ret = rkmtd_detach(dev);
81 if (ret) {
82 printf("Cannot detach mtd\n");
83 return CMD_RET_FAILURE;
84 }
85
86 ret = device_unbind(dev);
87 if (ret) {
88 printf("Cannot unbind device '%s'\n", dev->name);
89 return CMD_RET_FAILURE;
90 }
91
92 return 0;
93 }
94
show_rkmtd_dev(struct udevice * dev)95 static void show_rkmtd_dev(struct udevice *dev)
96 {
97 struct rkmtd_dev *plat = dev_get_plat(dev);
98 struct blk_desc *desc;
99 struct udevice *blk;
100 int ret;
101
102 printf("%3d ", dev_seq(dev));
103
104 ret = blk_get_from_parent(dev, &blk);
105 if (ret)
106 return;
107
108 desc = dev_get_uclass_plat(blk);
109 printf("%12lu %-15s\n", (unsigned long)desc->lba, plat->label);
110 }
111
do_rkmtd_info(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])112 static int do_rkmtd_info(struct cmd_tbl *cmdtp, int flag, int argc,
113 char *const argv[])
114 {
115 struct udevice *dev;
116
117 if (argc < 1)
118 return CMD_RET_USAGE;
119
120 dev = NULL;
121 if (argc >= 2) {
122 dev = parse_rkmtd_label(argv[1]);
123 if (!dev)
124 return CMD_RET_FAILURE;
125 }
126
127 printf("%3s %12s %-15s\n", "dev", "blocks", "label");
128 if (dev) {
129 show_rkmtd_dev(dev);
130 } else {
131 struct uclass *uc;
132
133 uclass_id_foreach_dev(UCLASS_RKMTD, dev, uc)
134 show_rkmtd_dev(dev);
135 }
136
137 return 0;
138 }
139
do_rkmtd_dev(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])140 static int do_rkmtd_dev(struct cmd_tbl *cmdtp, int flag, int argc,
141 char *const argv[])
142 {
143 struct udevice *dev;
144 const char *label;
145
146 if (argc < 1 || argc > 3)
147 return CMD_RET_USAGE;
148
149 if (argc == 1) {
150 struct rkmtd_dev *plat;
151
152 dev = rkmtd_get_cur_dev();
153 if (!dev) {
154 printf("No current rkmtd device\n");
155 return CMD_RET_FAILURE;
156 }
157 plat = dev_get_plat(dev);
158 printf("Current rkmtd device: %d: %s\n", dev_seq(dev),
159 plat->label);
160 return 0;
161 }
162
163 label = argv[1];
164 dev = parse_rkmtd_label(argv[1]);
165 if (!dev)
166 return CMD_RET_FAILURE;
167
168 rkmtd_set_cur_dev(dev);
169
170 return 0;
171 }
172
173 static struct cmd_tbl cmd_rkmtd_sub[] = {
174 U_BOOT_CMD_MKENT(bind, 4, 0, do_rkmtd_bind, "", ""),
175 U_BOOT_CMD_MKENT(unbind, 4, 0, do_rkmtd_unbind, "", ""),
176 U_BOOT_CMD_MKENT(info, 3, 0, do_rkmtd_info, "", ""),
177 U_BOOT_CMD_MKENT(dev, 0, 1, do_rkmtd_dev, "", ""),
178 };
179
do_rkmtd(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])180 static int do_rkmtd(struct cmd_tbl *cmdtp, int flag, int argc,
181 char *const argv[])
182 {
183 struct cmd_tbl *c;
184
185 argc--;
186 argv++;
187
188 c = find_cmd_tbl(argv[0], cmd_rkmtd_sub, ARRAY_SIZE(cmd_rkmtd_sub));
189
190 if (c)
191 return c->cmd(cmdtp, flag, argc, argv);
192 else
193 return CMD_RET_USAGE;
194 }
195
196 U_BOOT_CMD(
197 rkmtd, 8, 1, do_rkmtd,
198 "Rockchip MTD sub-system",
199 "bind <label> - bind RKMTD device\n"
200 "rkmtd unbind <label> - unbind RKMTD device\n"
201 "rkmtd info [<label>] - show all available RKMTD devices\n"
202 "rkmtd dev [<label>] - show or set current RKMTD device\n"
203 );
204