183d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
209d71aacSSimon Glass /*
309d71aacSSimon Glass * Copyright (C) 2016 Google, Inc
409d71aacSSimon Glass * Written by Simon Glass <sjg@chromium.org>
509d71aacSSimon Glass */
609d71aacSSimon Glass
709d71aacSSimon Glass #include <common.h>
809d71aacSSimon Glass #include <blk.h>
909d71aacSSimon Glass #include <dm.h>
1009d71aacSSimon Glass #include <dm/device-internal.h>
1109d71aacSSimon Glass #include <dm/lists.h>
128a5cbc06SStefan Roese #include <dm/uclass-internal.h>
1309d71aacSSimon Glass
14d508c82bSSimon Glass static const char *if_typename_str[IF_TYPE_COUNT] = {
15d508c82bSSimon Glass [IF_TYPE_IDE] = "ide",
16d508c82bSSimon Glass [IF_TYPE_SCSI] = "scsi",
17d508c82bSSimon Glass [IF_TYPE_ATAPI] = "atapi",
18d508c82bSSimon Glass [IF_TYPE_USB] = "usb",
19d508c82bSSimon Glass [IF_TYPE_DOC] = "doc",
20d508c82bSSimon Glass [IF_TYPE_MMC] = "mmc",
21d508c82bSSimon Glass [IF_TYPE_SD] = "sd",
22d508c82bSSimon Glass [IF_TYPE_SATA] = "sata",
23d508c82bSSimon Glass [IF_TYPE_HOST] = "host",
24ffab6945SZhikang Zhang [IF_TYPE_NVME] = "nvme",
2505ef48a2SHeinrich Schuchardt [IF_TYPE_EFI] = "efi",
264ad54ec4STuomas Tynkkynen [IF_TYPE_VIRTIO] = "virtio",
27d508c82bSSimon Glass };
28d508c82bSSimon Glass
29d508c82bSSimon Glass static enum uclass_id if_type_uclass_id[IF_TYPE_COUNT] = {
3068e6f221SBin Meng [IF_TYPE_IDE] = UCLASS_IDE,
31e8a016b5SMichal Simek [IF_TYPE_SCSI] = UCLASS_SCSI,
32d508c82bSSimon Glass [IF_TYPE_ATAPI] = UCLASS_INVALID,
33d508c82bSSimon Glass [IF_TYPE_USB] = UCLASS_MASS_STORAGE,
34d508c82bSSimon Glass [IF_TYPE_DOC] = UCLASS_INVALID,
35d508c82bSSimon Glass [IF_TYPE_MMC] = UCLASS_MMC,
36d508c82bSSimon Glass [IF_TYPE_SD] = UCLASS_INVALID,
37d508c82bSSimon Glass [IF_TYPE_SATA] = UCLASS_AHCI,
38d508c82bSSimon Glass [IF_TYPE_HOST] = UCLASS_ROOT,
3905ef48a2SHeinrich Schuchardt [IF_TYPE_NVME] = UCLASS_NVME,
4005ef48a2SHeinrich Schuchardt [IF_TYPE_EFI] = UCLASS_EFI,
414ad54ec4STuomas Tynkkynen [IF_TYPE_VIRTIO] = UCLASS_VIRTIO,
42d508c82bSSimon Glass };
43d508c82bSSimon Glass
if_typename_to_iftype(const char * if_typename)44d508c82bSSimon Glass static enum if_type if_typename_to_iftype(const char *if_typename)
45d508c82bSSimon Glass {
46d508c82bSSimon Glass int i;
47d508c82bSSimon Glass
48d508c82bSSimon Glass for (i = 0; i < IF_TYPE_COUNT; i++) {
49d508c82bSSimon Glass if (if_typename_str[i] &&
50d508c82bSSimon Glass !strcmp(if_typename, if_typename_str[i]))
51d508c82bSSimon Glass return i;
52d508c82bSSimon Glass }
53d508c82bSSimon Glass
54d508c82bSSimon Glass return IF_TYPE_UNKNOWN;
55d508c82bSSimon Glass }
56d508c82bSSimon Glass
if_type_to_uclass_id(enum if_type if_type)57d508c82bSSimon Glass static enum uclass_id if_type_to_uclass_id(enum if_type if_type)
58d508c82bSSimon Glass {
59d508c82bSSimon Glass return if_type_uclass_id[if_type];
60d508c82bSSimon Glass }
61d508c82bSSimon Glass
blk_get_if_type_name(enum if_type if_type)626faa4ed7SSimon Glass const char *blk_get_if_type_name(enum if_type if_type)
636faa4ed7SSimon Glass {
646faa4ed7SSimon Glass return if_typename_str[if_type];
656faa4ed7SSimon Glass }
666faa4ed7SSimon Glass
blk_get_devnum_by_type(enum if_type if_type,int devnum)67d508c82bSSimon Glass struct blk_desc *blk_get_devnum_by_type(enum if_type if_type, int devnum)
68d508c82bSSimon Glass {
69d508c82bSSimon Glass struct blk_desc *desc;
70d508c82bSSimon Glass struct udevice *dev;
71d508c82bSSimon Glass int ret;
72d508c82bSSimon Glass
73d508c82bSSimon Glass ret = blk_get_device(if_type, devnum, &dev);
74d508c82bSSimon Glass if (ret)
75d508c82bSSimon Glass return NULL;
76d508c82bSSimon Glass desc = dev_get_uclass_platdata(dev);
77d508c82bSSimon Glass
78d508c82bSSimon Glass return desc;
79d508c82bSSimon Glass }
80d508c82bSSimon Glass
81d508c82bSSimon Glass /*
82d508c82bSSimon Glass * This function is complicated with driver model. We look up the interface
83d508c82bSSimon Glass * name in a local table. This gives us an interface type which we can match
84d508c82bSSimon Glass * against the uclass of the block device's parent.
85d508c82bSSimon Glass */
blk_get_devnum_by_typename(const char * if_typename,int devnum)86d508c82bSSimon Glass struct blk_desc *blk_get_devnum_by_typename(const char *if_typename, int devnum)
87d508c82bSSimon Glass {
88d508c82bSSimon Glass enum uclass_id uclass_id;
89d508c82bSSimon Glass enum if_type if_type;
90d508c82bSSimon Glass struct udevice *dev;
91d508c82bSSimon Glass struct uclass *uc;
92d508c82bSSimon Glass int ret;
93d508c82bSSimon Glass
94d508c82bSSimon Glass if_type = if_typename_to_iftype(if_typename);
95d508c82bSSimon Glass if (if_type == IF_TYPE_UNKNOWN) {
96d508c82bSSimon Glass debug("%s: Unknown interface type '%s'\n", __func__,
97d508c82bSSimon Glass if_typename);
98d508c82bSSimon Glass return NULL;
99d508c82bSSimon Glass }
100d508c82bSSimon Glass uclass_id = if_type_to_uclass_id(if_type);
101d508c82bSSimon Glass if (uclass_id == UCLASS_INVALID) {
102d508c82bSSimon Glass debug("%s: Unknown uclass for interface type'\n",
103d508c82bSSimon Glass if_typename_str[if_type]);
104d508c82bSSimon Glass return NULL;
105d508c82bSSimon Glass }
106d508c82bSSimon Glass
107d508c82bSSimon Glass ret = uclass_get(UCLASS_BLK, &uc);
108d508c82bSSimon Glass if (ret)
109d508c82bSSimon Glass return NULL;
110d508c82bSSimon Glass uclass_foreach_dev(dev, uc) {
111d508c82bSSimon Glass struct blk_desc *desc = dev_get_uclass_platdata(dev);
112d508c82bSSimon Glass
113d508c82bSSimon Glass debug("%s: if_type=%d, devnum=%d: %s, %d, %d\n", __func__,
114d508c82bSSimon Glass if_type, devnum, dev->name, desc->if_type, desc->devnum);
115d508c82bSSimon Glass if (desc->devnum != devnum)
116d508c82bSSimon Glass continue;
117d508c82bSSimon Glass
118d508c82bSSimon Glass /* Find out the parent device uclass */
119d508c82bSSimon Glass if (device_get_uclass_id(dev->parent) != uclass_id) {
120d508c82bSSimon Glass debug("%s: parent uclass %d, this dev %d\n", __func__,
121d508c82bSSimon Glass device_get_uclass_id(dev->parent), uclass_id);
122d508c82bSSimon Glass continue;
123d508c82bSSimon Glass }
124d508c82bSSimon Glass
125d508c82bSSimon Glass if (device_probe(dev))
126d508c82bSSimon Glass return NULL;
127d508c82bSSimon Glass
128d508c82bSSimon Glass debug("%s: Device desc %p\n", __func__, desc);
129d508c82bSSimon Glass return desc;
130d508c82bSSimon Glass }
131d508c82bSSimon Glass debug("%s: No device found\n", __func__);
132d508c82bSSimon Glass
133d508c82bSSimon Glass return NULL;
134d508c82bSSimon Glass }
135d508c82bSSimon Glass
136d508c82bSSimon Glass /**
137bc53d263STien Fong Chee * blk_get_by_device() - Get the block device descriptor for the given device
138bc53d263STien Fong Chee * @dev: Instance of a storage device
139bc53d263STien Fong Chee *
140bc53d263STien Fong Chee * Return: With block device descriptor on success , NULL if there is no such
141bc53d263STien Fong Chee * block device.
142bc53d263STien Fong Chee */
blk_get_by_device(struct udevice * dev)143bc53d263STien Fong Chee struct blk_desc *blk_get_by_device(struct udevice *dev)
144bc53d263STien Fong Chee {
145bc53d263STien Fong Chee struct udevice *child_dev, *next;
146bc53d263STien Fong Chee
147bc53d263STien Fong Chee device_foreach_child_safe(child_dev, next, dev) {
148bc53d263STien Fong Chee if (device_get_uclass_id(child_dev) != UCLASS_BLK)
149bc53d263STien Fong Chee continue;
150bc53d263STien Fong Chee
151bc53d263STien Fong Chee return dev_get_uclass_platdata(child_dev);
152bc53d263STien Fong Chee }
153bc53d263STien Fong Chee
154bc53d263STien Fong Chee debug("%s: No block device found\n", __func__);
155bc53d263STien Fong Chee
156bc53d263STien Fong Chee return NULL;
157bc53d263STien Fong Chee }
158bc53d263STien Fong Chee
159bc53d263STien Fong Chee /**
160d508c82bSSimon Glass * get_desc() - Get the block device descriptor for the given device number
161d508c82bSSimon Glass *
162d508c82bSSimon Glass * @if_type: Interface type
163d508c82bSSimon Glass * @devnum: Device number (0 = first)
164d508c82bSSimon Glass * @descp: Returns block device descriptor on success
165d508c82bSSimon Glass * @return 0 on success, -ENODEV if there is no such device and no device
166d508c82bSSimon Glass * with a higher device number, -ENOENT if there is no such device but there
167d508c82bSSimon Glass * is one with a higher number, or other -ve on other error.
168d508c82bSSimon Glass */
get_desc(enum if_type if_type,int devnum,struct blk_desc ** descp)169d508c82bSSimon Glass static int get_desc(enum if_type if_type, int devnum, struct blk_desc **descp)
170d508c82bSSimon Glass {
171d508c82bSSimon Glass bool found_more = false;
172d508c82bSSimon Glass struct udevice *dev;
173d508c82bSSimon Glass struct uclass *uc;
174d508c82bSSimon Glass int ret;
175d508c82bSSimon Glass
176d508c82bSSimon Glass *descp = NULL;
177d508c82bSSimon Glass ret = uclass_get(UCLASS_BLK, &uc);
178d508c82bSSimon Glass if (ret)
179d508c82bSSimon Glass return ret;
180d508c82bSSimon Glass uclass_foreach_dev(dev, uc) {
181d508c82bSSimon Glass struct blk_desc *desc = dev_get_uclass_platdata(dev);
182d508c82bSSimon Glass
183d508c82bSSimon Glass debug("%s: if_type=%d, devnum=%d: %s, %d, %d\n", __func__,
184d508c82bSSimon Glass if_type, devnum, dev->name, desc->if_type, desc->devnum);
185d508c82bSSimon Glass if (desc->if_type == if_type) {
186d508c82bSSimon Glass if (desc->devnum == devnum) {
187d508c82bSSimon Glass ret = device_probe(dev);
188d508c82bSSimon Glass if (ret)
189d508c82bSSimon Glass return ret;
190d508c82bSSimon Glass
1914408f6f4SMichal Simek *descp = desc;
1924408f6f4SMichal Simek return 0;
193d508c82bSSimon Glass } else if (desc->devnum > devnum) {
194d508c82bSSimon Glass found_more = true;
195d508c82bSSimon Glass }
196d508c82bSSimon Glass }
197d508c82bSSimon Glass }
198d508c82bSSimon Glass
199d508c82bSSimon Glass return found_more ? -ENOENT : -ENODEV;
200d508c82bSSimon Glass }
201d508c82bSSimon Glass
blk_select_hwpart_devnum(enum if_type if_type,int devnum,int hwpart)202cd0fb55bSSimon Glass int blk_select_hwpart_devnum(enum if_type if_type, int devnum, int hwpart)
203cd0fb55bSSimon Glass {
204cd0fb55bSSimon Glass struct udevice *dev;
205cd0fb55bSSimon Glass int ret;
206cd0fb55bSSimon Glass
207cd0fb55bSSimon Glass ret = blk_get_device(if_type, devnum, &dev);
208cd0fb55bSSimon Glass if (ret)
209cd0fb55bSSimon Glass return ret;
210cd0fb55bSSimon Glass
211cd0fb55bSSimon Glass return blk_select_hwpart(dev, hwpart);
212cd0fb55bSSimon Glass }
213cd0fb55bSSimon Glass
blk_list_part(enum if_type if_type)214d508c82bSSimon Glass int blk_list_part(enum if_type if_type)
215d508c82bSSimon Glass {
216d508c82bSSimon Glass struct blk_desc *desc;
217d508c82bSSimon Glass int devnum, ok;
218d508c82bSSimon Glass int ret;
219d508c82bSSimon Glass
220d508c82bSSimon Glass for (ok = 0, devnum = 0;; ++devnum) {
221d508c82bSSimon Glass ret = get_desc(if_type, devnum, &desc);
222d508c82bSSimon Glass if (ret == -ENODEV)
223d508c82bSSimon Glass break;
224d508c82bSSimon Glass else if (ret)
225d508c82bSSimon Glass continue;
226d508c82bSSimon Glass if (desc->part_type != PART_TYPE_UNKNOWN) {
227d508c82bSSimon Glass ++ok;
228d508c82bSSimon Glass if (devnum)
229d508c82bSSimon Glass putc('\n');
230d508c82bSSimon Glass part_print(desc);
231d508c82bSSimon Glass }
232d508c82bSSimon Glass }
233d508c82bSSimon Glass if (!ok)
234d508c82bSSimon Glass return -ENODEV;
235d508c82bSSimon Glass
236d508c82bSSimon Glass return 0;
237d508c82bSSimon Glass }
238d508c82bSSimon Glass
blk_print_part_devnum(enum if_type if_type,int devnum)239d508c82bSSimon Glass int blk_print_part_devnum(enum if_type if_type, int devnum)
240d508c82bSSimon Glass {
241d508c82bSSimon Glass struct blk_desc *desc;
242d508c82bSSimon Glass int ret;
243d508c82bSSimon Glass
244d508c82bSSimon Glass ret = get_desc(if_type, devnum, &desc);
245d508c82bSSimon Glass if (ret)
246d508c82bSSimon Glass return ret;
247d508c82bSSimon Glass if (desc->type == DEV_TYPE_UNKNOWN)
248d508c82bSSimon Glass return -ENOENT;
249d508c82bSSimon Glass part_print(desc);
250d508c82bSSimon Glass
251d508c82bSSimon Glass return 0;
252d508c82bSSimon Glass }
253d508c82bSSimon Glass
blk_list_devices(enum if_type if_type)254d508c82bSSimon Glass void blk_list_devices(enum if_type if_type)
255d508c82bSSimon Glass {
256d508c82bSSimon Glass struct blk_desc *desc;
257d508c82bSSimon Glass int ret;
258d508c82bSSimon Glass int i;
259d508c82bSSimon Glass
260d508c82bSSimon Glass for (i = 0;; ++i) {
261d508c82bSSimon Glass ret = get_desc(if_type, i, &desc);
262d508c82bSSimon Glass if (ret == -ENODEV)
263d508c82bSSimon Glass break;
264d508c82bSSimon Glass else if (ret)
265d508c82bSSimon Glass continue;
266d508c82bSSimon Glass if (desc->type == DEV_TYPE_UNKNOWN)
267d508c82bSSimon Glass continue; /* list only known devices */
268d508c82bSSimon Glass printf("Device %d: ", i);
269d508c82bSSimon Glass dev_print(desc);
270d508c82bSSimon Glass }
271d508c82bSSimon Glass }
272d508c82bSSimon Glass
blk_print_device_num(enum if_type if_type,int devnum)273d508c82bSSimon Glass int blk_print_device_num(enum if_type if_type, int devnum)
274d508c82bSSimon Glass {
275d508c82bSSimon Glass struct blk_desc *desc;
276d508c82bSSimon Glass int ret;
277d508c82bSSimon Glass
278d508c82bSSimon Glass ret = get_desc(if_type, devnum, &desc);
279d508c82bSSimon Glass if (ret)
280d508c82bSSimon Glass return ret;
281d508c82bSSimon Glass printf("\nIDE device %d: ", devnum);
282d508c82bSSimon Glass dev_print(desc);
283d508c82bSSimon Glass
284d508c82bSSimon Glass return 0;
285d508c82bSSimon Glass }
286d508c82bSSimon Glass
blk_show_device(enum if_type if_type,int devnum)287d508c82bSSimon Glass int blk_show_device(enum if_type if_type, int devnum)
288d508c82bSSimon Glass {
289d508c82bSSimon Glass struct blk_desc *desc;
290d508c82bSSimon Glass int ret;
291d508c82bSSimon Glass
292d508c82bSSimon Glass printf("\nDevice %d: ", devnum);
293d508c82bSSimon Glass ret = get_desc(if_type, devnum, &desc);
294d508c82bSSimon Glass if (ret == -ENODEV || ret == -ENOENT) {
295d508c82bSSimon Glass printf("unknown device\n");
296d508c82bSSimon Glass return -ENODEV;
297d508c82bSSimon Glass }
298d508c82bSSimon Glass if (ret)
299d508c82bSSimon Glass return ret;
300d508c82bSSimon Glass dev_print(desc);
301d508c82bSSimon Glass
302d508c82bSSimon Glass if (desc->type == DEV_TYPE_UNKNOWN)
303d508c82bSSimon Glass return -ENOENT;
304d508c82bSSimon Glass
305d508c82bSSimon Glass return 0;
306d508c82bSSimon Glass }
307d508c82bSSimon Glass
blk_read_devnum(enum if_type if_type,int devnum,lbaint_t start,lbaint_t blkcnt,void * buffer)308d508c82bSSimon Glass ulong blk_read_devnum(enum if_type if_type, int devnum, lbaint_t start,
309d508c82bSSimon Glass lbaint_t blkcnt, void *buffer)
310d508c82bSSimon Glass {
311d508c82bSSimon Glass struct blk_desc *desc;
312d508c82bSSimon Glass ulong n;
313d508c82bSSimon Glass int ret;
314d508c82bSSimon Glass
315d508c82bSSimon Glass ret = get_desc(if_type, devnum, &desc);
316d508c82bSSimon Glass if (ret)
317d508c82bSSimon Glass return ret;
318d508c82bSSimon Glass n = blk_dread(desc, start, blkcnt, buffer);
319d508c82bSSimon Glass if (IS_ERR_VALUE(n))
320d508c82bSSimon Glass return n;
321d508c82bSSimon Glass
322d508c82bSSimon Glass return n;
323d508c82bSSimon Glass }
324d508c82bSSimon Glass
blk_write_devnum(enum if_type if_type,int devnum,lbaint_t start,lbaint_t blkcnt,const void * buffer)325d508c82bSSimon Glass ulong blk_write_devnum(enum if_type if_type, int devnum, lbaint_t start,
326d508c82bSSimon Glass lbaint_t blkcnt, const void *buffer)
327d508c82bSSimon Glass {
328d508c82bSSimon Glass struct blk_desc *desc;
329d508c82bSSimon Glass int ret;
330d508c82bSSimon Glass
331d508c82bSSimon Glass ret = get_desc(if_type, devnum, &desc);
332d508c82bSSimon Glass if (ret)
333d508c82bSSimon Glass return ret;
334d508c82bSSimon Glass return blk_dwrite(desc, start, blkcnt, buffer);
335d508c82bSSimon Glass }
336d508c82bSSimon Glass
blk_select_hwpart(struct udevice * dev,int hwpart)337cd0fb55bSSimon Glass int blk_select_hwpart(struct udevice *dev, int hwpart)
338cd0fb55bSSimon Glass {
339cd0fb55bSSimon Glass const struct blk_ops *ops = blk_get_ops(dev);
340cd0fb55bSSimon Glass
341cd0fb55bSSimon Glass if (!ops)
342cd0fb55bSSimon Glass return -ENOSYS;
343cd0fb55bSSimon Glass if (!ops->select_hwpart)
344cd0fb55bSSimon Glass return 0;
345cd0fb55bSSimon Glass
346cd0fb55bSSimon Glass return ops->select_hwpart(dev, hwpart);
347cd0fb55bSSimon Glass }
348cd0fb55bSSimon Glass
blk_dselect_hwpart(struct blk_desc * desc,int hwpart)349cd0fb55bSSimon Glass int blk_dselect_hwpart(struct blk_desc *desc, int hwpart)
350cd0fb55bSSimon Glass {
351cd0fb55bSSimon Glass return blk_select_hwpart(desc->bdev, hwpart);
352cd0fb55bSSimon Glass }
353cd0fb55bSSimon Glass
blk_first_device(int if_type,struct udevice ** devp)35409d71aacSSimon Glass int blk_first_device(int if_type, struct udevice **devp)
35509d71aacSSimon Glass {
35609d71aacSSimon Glass struct blk_desc *desc;
35709d71aacSSimon Glass int ret;
35809d71aacSSimon Glass
3598a5cbc06SStefan Roese ret = uclass_find_first_device(UCLASS_BLK, devp);
36009d71aacSSimon Glass if (ret)
36109d71aacSSimon Glass return ret;
36209d71aacSSimon Glass if (!*devp)
36309d71aacSSimon Glass return -ENODEV;
36409d71aacSSimon Glass do {
36509d71aacSSimon Glass desc = dev_get_uclass_platdata(*devp);
36609d71aacSSimon Glass if (desc->if_type == if_type)
36709d71aacSSimon Glass return 0;
3688a5cbc06SStefan Roese ret = uclass_find_next_device(devp);
36909d71aacSSimon Glass if (ret)
37009d71aacSSimon Glass return ret;
37109d71aacSSimon Glass } while (*devp);
37209d71aacSSimon Glass
37309d71aacSSimon Glass return -ENODEV;
37409d71aacSSimon Glass }
37509d71aacSSimon Glass
blk_next_device(struct udevice ** devp)37609d71aacSSimon Glass int blk_next_device(struct udevice **devp)
37709d71aacSSimon Glass {
37809d71aacSSimon Glass struct blk_desc *desc;
37909d71aacSSimon Glass int ret, if_type;
38009d71aacSSimon Glass
38109d71aacSSimon Glass desc = dev_get_uclass_platdata(*devp);
38209d71aacSSimon Glass if_type = desc->if_type;
38309d71aacSSimon Glass do {
3848a5cbc06SStefan Roese ret = uclass_find_next_device(devp);
38509d71aacSSimon Glass if (ret)
38609d71aacSSimon Glass return ret;
38709d71aacSSimon Glass if (!*devp)
38809d71aacSSimon Glass return -ENODEV;
38909d71aacSSimon Glass desc = dev_get_uclass_platdata(*devp);
39009d71aacSSimon Glass if (desc->if_type == if_type)
39109d71aacSSimon Glass return 0;
39209d71aacSSimon Glass } while (1);
39309d71aacSSimon Glass }
39409d71aacSSimon Glass
blk_find_device(int if_type,int devnum,struct udevice ** devp)3956139281aSSimon Glass int blk_find_device(int if_type, int devnum, struct udevice **devp)
39609d71aacSSimon Glass {
39709d71aacSSimon Glass struct uclass *uc;
39809d71aacSSimon Glass struct udevice *dev;
39909d71aacSSimon Glass int ret;
40009d71aacSSimon Glass
40109d71aacSSimon Glass ret = uclass_get(UCLASS_BLK, &uc);
40209d71aacSSimon Glass if (ret)
40309d71aacSSimon Glass return ret;
40409d71aacSSimon Glass uclass_foreach_dev(dev, uc) {
40509d71aacSSimon Glass struct blk_desc *desc = dev_get_uclass_platdata(dev);
40609d71aacSSimon Glass
40709d71aacSSimon Glass debug("%s: if_type=%d, devnum=%d: %s, %d, %d\n", __func__,
40809d71aacSSimon Glass if_type, devnum, dev->name, desc->if_type, desc->devnum);
40909d71aacSSimon Glass if (desc->if_type == if_type && desc->devnum == devnum) {
41009d71aacSSimon Glass *devp = dev;
4116139281aSSimon Glass return 0;
41209d71aacSSimon Glass }
41309d71aacSSimon Glass }
41409d71aacSSimon Glass
41509d71aacSSimon Glass return -ENODEV;
41609d71aacSSimon Glass }
41709d71aacSSimon Glass
blk_get_device(int if_type,int devnum,struct udevice ** devp)4186139281aSSimon Glass int blk_get_device(int if_type, int devnum, struct udevice **devp)
4196139281aSSimon Glass {
4206139281aSSimon Glass int ret;
4216139281aSSimon Glass
4226139281aSSimon Glass ret = blk_find_device(if_type, devnum, devp);
4236139281aSSimon Glass if (ret)
4246139281aSSimon Glass return ret;
4256139281aSSimon Glass
4266139281aSSimon Glass return device_probe(*devp);
4276139281aSSimon Glass }
4286139281aSSimon Glass
blk_dread(struct blk_desc * block_dev,lbaint_t start,lbaint_t blkcnt,void * buffer)42909d71aacSSimon Glass unsigned long blk_dread(struct blk_desc *block_dev, lbaint_t start,
43009d71aacSSimon Glass lbaint_t blkcnt, void *buffer)
43109d71aacSSimon Glass {
43209d71aacSSimon Glass struct udevice *dev = block_dev->bdev;
43309d71aacSSimon Glass const struct blk_ops *ops = blk_get_ops(dev);
434e40cf34aSEric Nelson ulong blks_read;
43509d71aacSSimon Glass
43609d71aacSSimon Glass if (!ops->read)
43709d71aacSSimon Glass return -ENOSYS;
43809d71aacSSimon Glass
439e40cf34aSEric Nelson if (blkcache_read(block_dev->if_type, block_dev->devnum,
440e40cf34aSEric Nelson start, blkcnt, block_dev->blksz, buffer))
441e40cf34aSEric Nelson return blkcnt;
442e40cf34aSEric Nelson blks_read = ops->read(dev, start, blkcnt, buffer);
443e40cf34aSEric Nelson if (blks_read == blkcnt)
444e40cf34aSEric Nelson blkcache_fill(block_dev->if_type, block_dev->devnum,
445e40cf34aSEric Nelson start, blkcnt, block_dev->blksz, buffer);
446e40cf34aSEric Nelson
447e40cf34aSEric Nelson return blks_read;
44809d71aacSSimon Glass }
44909d71aacSSimon Glass
blk_dwrite(struct blk_desc * block_dev,lbaint_t start,lbaint_t blkcnt,const void * buffer)45009d71aacSSimon Glass unsigned long blk_dwrite(struct blk_desc *block_dev, lbaint_t start,
45109d71aacSSimon Glass lbaint_t blkcnt, const void *buffer)
45209d71aacSSimon Glass {
45309d71aacSSimon Glass struct udevice *dev = block_dev->bdev;
45409d71aacSSimon Glass const struct blk_ops *ops = blk_get_ops(dev);
45509d71aacSSimon Glass
45609d71aacSSimon Glass if (!ops->write)
45709d71aacSSimon Glass return -ENOSYS;
45809d71aacSSimon Glass
459e40cf34aSEric Nelson blkcache_invalidate(block_dev->if_type, block_dev->devnum);
46009d71aacSSimon Glass return ops->write(dev, start, blkcnt, buffer);
46109d71aacSSimon Glass }
46209d71aacSSimon Glass
blk_derase(struct blk_desc * block_dev,lbaint_t start,lbaint_t blkcnt)46309d71aacSSimon Glass unsigned long blk_derase(struct blk_desc *block_dev, lbaint_t start,
46409d71aacSSimon Glass lbaint_t blkcnt)
46509d71aacSSimon Glass {
46609d71aacSSimon Glass struct udevice *dev = block_dev->bdev;
46709d71aacSSimon Glass const struct blk_ops *ops = blk_get_ops(dev);
46809d71aacSSimon Glass
46909d71aacSSimon Glass if (!ops->erase)
47009d71aacSSimon Glass return -ENOSYS;
47109d71aacSSimon Glass
472e40cf34aSEric Nelson blkcache_invalidate(block_dev->if_type, block_dev->devnum);
47309d71aacSSimon Glass return ops->erase(dev, start, blkcnt);
47409d71aacSSimon Glass }
47509d71aacSSimon Glass
blk_get_from_parent(struct udevice * parent,struct udevice ** devp)4769f103b9cSSimon Glass int blk_get_from_parent(struct udevice *parent, struct udevice **devp)
4779f103b9cSSimon Glass {
4789f103b9cSSimon Glass struct udevice *dev;
4799f103b9cSSimon Glass enum uclass_id id;
4809f103b9cSSimon Glass int ret;
4819f103b9cSSimon Glass
4829f103b9cSSimon Glass device_find_first_child(parent, &dev);
4839f103b9cSSimon Glass if (!dev) {
4849f103b9cSSimon Glass debug("%s: No block device found for parent '%s'\n", __func__,
4859f103b9cSSimon Glass parent->name);
4869f103b9cSSimon Glass return -ENODEV;
4879f103b9cSSimon Glass }
4889f103b9cSSimon Glass id = device_get_uclass_id(dev);
4899f103b9cSSimon Glass if (id != UCLASS_BLK) {
4909f103b9cSSimon Glass debug("%s: Incorrect uclass %s for block device '%s'\n",
4919f103b9cSSimon Glass __func__, uclass_get_name(id), dev->name);
4929f103b9cSSimon Glass return -ENOTBLK;
4939f103b9cSSimon Glass }
4949f103b9cSSimon Glass ret = device_probe(dev);
4959f103b9cSSimon Glass if (ret)
4969f103b9cSSimon Glass return ret;
4979f103b9cSSimon Glass *devp = dev;
4989f103b9cSSimon Glass
4999f103b9cSSimon Glass return 0;
5009f103b9cSSimon Glass }
5019f103b9cSSimon Glass
blk_find_max_devnum(enum if_type if_type)50252138fd4SSimon Glass int blk_find_max_devnum(enum if_type if_type)
50352138fd4SSimon Glass {
50452138fd4SSimon Glass struct udevice *dev;
50552138fd4SSimon Glass int max_devnum = -ENODEV;
50652138fd4SSimon Glass struct uclass *uc;
50752138fd4SSimon Glass int ret;
50852138fd4SSimon Glass
50952138fd4SSimon Glass ret = uclass_get(UCLASS_BLK, &uc);
51052138fd4SSimon Glass if (ret)
51152138fd4SSimon Glass return ret;
51252138fd4SSimon Glass uclass_foreach_dev(dev, uc) {
51352138fd4SSimon Glass struct blk_desc *desc = dev_get_uclass_platdata(dev);
51452138fd4SSimon Glass
51552138fd4SSimon Glass if (desc->if_type == if_type && desc->devnum > max_devnum)
51652138fd4SSimon Glass max_devnum = desc->devnum;
51752138fd4SSimon Glass }
51852138fd4SSimon Glass
51952138fd4SSimon Glass return max_devnum;
52052138fd4SSimon Glass }
52152138fd4SSimon Glass
blk_next_free_devnum(enum if_type if_type)522c879eeb7SBin Meng int blk_next_free_devnum(enum if_type if_type)
523e8abbb53SSimon Glass {
524e8abbb53SSimon Glass int ret;
525e8abbb53SSimon Glass
526e8abbb53SSimon Glass ret = blk_find_max_devnum(if_type);
527e8abbb53SSimon Glass if (ret == -ENODEV)
528e8abbb53SSimon Glass return 0;
529e8abbb53SSimon Glass if (ret < 0)
530e8abbb53SSimon Glass return ret;
531e8abbb53SSimon Glass
532e8abbb53SSimon Glass return ret + 1;
533e8abbb53SSimon Glass }
534e8abbb53SSimon Glass
blk_claim_devnum(enum if_type if_type,int devnum)535e48eeb9eSSimon Glass static int blk_claim_devnum(enum if_type if_type, int devnum)
536e48eeb9eSSimon Glass {
537e48eeb9eSSimon Glass struct udevice *dev;
538e48eeb9eSSimon Glass struct uclass *uc;
539e48eeb9eSSimon Glass int ret;
540e48eeb9eSSimon Glass
541e48eeb9eSSimon Glass ret = uclass_get(UCLASS_BLK, &uc);
542e48eeb9eSSimon Glass if (ret)
543e48eeb9eSSimon Glass return ret;
544e48eeb9eSSimon Glass uclass_foreach_dev(dev, uc) {
545e48eeb9eSSimon Glass struct blk_desc *desc = dev_get_uclass_platdata(dev);
546e48eeb9eSSimon Glass
547e48eeb9eSSimon Glass if (desc->if_type == if_type && desc->devnum == devnum) {
548e48eeb9eSSimon Glass int next = blk_next_free_devnum(if_type);
549e48eeb9eSSimon Glass
550e48eeb9eSSimon Glass if (next < 0)
551e48eeb9eSSimon Glass return next;
552e48eeb9eSSimon Glass desc->devnum = next;
553e48eeb9eSSimon Glass return 0;
554e48eeb9eSSimon Glass }
555e48eeb9eSSimon Glass }
556e48eeb9eSSimon Glass
557e48eeb9eSSimon Glass return -ENOENT;
558e48eeb9eSSimon Glass }
559e48eeb9eSSimon Glass
blk_create_device(struct udevice * parent,const char * drv_name,const char * name,int if_type,int devnum,int blksz,lbaint_t lba,struct udevice ** devp)56009d71aacSSimon Glass int blk_create_device(struct udevice *parent, const char *drv_name,
56109d71aacSSimon Glass const char *name, int if_type, int devnum, int blksz,
5625fe7702eSJean-Jacques Hiblot lbaint_t lba, struct udevice **devp)
56309d71aacSSimon Glass {
56409d71aacSSimon Glass struct blk_desc *desc;
56509d71aacSSimon Glass struct udevice *dev;
56609d71aacSSimon Glass int ret;
56709d71aacSSimon Glass
56852138fd4SSimon Glass if (devnum == -1) {
569e48eeb9eSSimon Glass devnum = blk_next_free_devnum(if_type);
570e48eeb9eSSimon Glass } else {
571e48eeb9eSSimon Glass ret = blk_claim_devnum(if_type, devnum);
572e48eeb9eSSimon Glass if (ret < 0 && ret != -ENOENT)
57352138fd4SSimon Glass return ret;
57452138fd4SSimon Glass }
575e48eeb9eSSimon Glass if (devnum < 0)
576e48eeb9eSSimon Glass return devnum;
57772a85c0dSSimon Glass ret = device_bind_driver(parent, drv_name, name, &dev);
57872a85c0dSSimon Glass if (ret)
57972a85c0dSSimon Glass return ret;
58072a85c0dSSimon Glass desc = dev_get_uclass_platdata(dev);
58172a85c0dSSimon Glass desc->if_type = if_type;
58272a85c0dSSimon Glass desc->blksz = blksz;
5835fe7702eSJean-Jacques Hiblot desc->lba = lba;
58472a85c0dSSimon Glass desc->part_type = PART_TYPE_UNKNOWN;
58572a85c0dSSimon Glass desc->bdev = dev;
58609d71aacSSimon Glass desc->devnum = devnum;
58709d71aacSSimon Glass *devp = dev;
58809d71aacSSimon Glass
58909d71aacSSimon Glass return 0;
59009d71aacSSimon Glass }
59109d71aacSSimon Glass
blk_create_devicef(struct udevice * parent,const char * drv_name,const char * name,int if_type,int devnum,int blksz,lbaint_t lba,struct udevice ** devp)5929107c973SSimon Glass int blk_create_devicef(struct udevice *parent, const char *drv_name,
5939107c973SSimon Glass const char *name, int if_type, int devnum, int blksz,
5945fe7702eSJean-Jacques Hiblot lbaint_t lba, struct udevice **devp)
5959107c973SSimon Glass {
5969107c973SSimon Glass char dev_name[30], *str;
597d0773524SSimon Glass int ret;
5989107c973SSimon Glass
5999107c973SSimon Glass snprintf(dev_name, sizeof(dev_name), "%s.%s", parent->name, name);
6009107c973SSimon Glass str = strdup(dev_name);
6019107c973SSimon Glass if (!str)
6029107c973SSimon Glass return -ENOMEM;
6039107c973SSimon Glass
604d0773524SSimon Glass ret = blk_create_device(parent, drv_name, str, if_type, devnum,
6055fe7702eSJean-Jacques Hiblot blksz, lba, devp);
606d0773524SSimon Glass if (ret) {
607d0773524SSimon Glass free(str);
608d0773524SSimon Glass return ret;
609d0773524SSimon Glass }
610d0773524SSimon Glass device_set_name_alloced(*devp);
611d0773524SSimon Glass
6127074b2a3SSimon Glass return 0;
6139107c973SSimon Glass }
6149107c973SSimon Glass
blk_unbind_all(int if_type)61509d71aacSSimon Glass int blk_unbind_all(int if_type)
61609d71aacSSimon Glass {
61709d71aacSSimon Glass struct uclass *uc;
61809d71aacSSimon Glass struct udevice *dev, *next;
61909d71aacSSimon Glass int ret;
62009d71aacSSimon Glass
62109d71aacSSimon Glass ret = uclass_get(UCLASS_BLK, &uc);
62209d71aacSSimon Glass if (ret)
62309d71aacSSimon Glass return ret;
62409d71aacSSimon Glass uclass_foreach_dev_safe(dev, next, uc) {
62509d71aacSSimon Glass struct blk_desc *desc = dev_get_uclass_platdata(dev);
62609d71aacSSimon Glass
62709d71aacSSimon Glass if (desc->if_type == if_type) {
628706865afSStefan Roese ret = device_remove(dev, DM_REMOVE_NORMAL);
62909d71aacSSimon Glass if (ret)
63009d71aacSSimon Glass return ret;
63109d71aacSSimon Glass ret = device_unbind(dev);
63209d71aacSSimon Glass if (ret)
63309d71aacSSimon Glass return ret;
63409d71aacSSimon Glass }
63509d71aacSSimon Glass }
63609d71aacSSimon Glass
63709d71aacSSimon Glass return 0;
63809d71aacSSimon Glass }
63909d71aacSSimon Glass
blk_post_probe(struct udevice * dev)640d0851c89SBin Meng static int blk_post_probe(struct udevice *dev)
641d0851c89SBin Meng {
642*91ff6865STom Rini #if defined(CONFIG_PARTITIONS) && defined(CONFIG_HAVE_BLOCK_DEVICE)
643d0851c89SBin Meng struct blk_desc *desc = dev_get_uclass_platdata(dev);
644d0851c89SBin Meng
645d0851c89SBin Meng part_init(desc);
646d0851c89SBin Meng #endif
647d0851c89SBin Meng
648d0851c89SBin Meng return 0;
649d0851c89SBin Meng }
650d0851c89SBin Meng
65109d71aacSSimon Glass UCLASS_DRIVER(blk) = {
65209d71aacSSimon Glass .id = UCLASS_BLK,
65309d71aacSSimon Glass .name = "blk",
654d0851c89SBin Meng .post_probe = blk_post_probe,
65509d71aacSSimon Glass .per_device_platdata_auto_alloc_size = sizeof(struct blk_desc),
65609d71aacSSimon Glass };
657