dm: core: Add a way to count the devices in a uclass
[platform/kernel/u-boot.git] / drivers / core / uclass.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (c) 2013 Google, Inc
4  *
5  * (C) Copyright 2012
6  * Pavel Herrmann <morpheus.ibis@gmail.com>
7  */
8
9 #define LOG_CATEGORY LOGC_DM
10
11 #include <common.h>
12 #include <dm.h>
13 #include <errno.h>
14 #include <log.h>
15 #include <malloc.h>
16 #include <asm/global_data.h>
17 #include <dm/device.h>
18 #include <dm/device-internal.h>
19 #include <dm/lists.h>
20 #include <dm/uclass.h>
21 #include <dm/uclass-internal.h>
22 #include <dm/util.h>
23
24 DECLARE_GLOBAL_DATA_PTR;
25
26 struct uclass *uclass_find(enum uclass_id key)
27 {
28         struct uclass *uc;
29
30         if (!gd->dm_root)
31                 return NULL;
32         /*
33          * TODO(sjg@chromium.org): Optimise this, perhaps moving the found
34          * node to the start of the list, or creating a linear array mapping
35          * id to node.
36          */
37         list_for_each_entry(uc, gd->uclass_root, sibling_node) {
38                 if (uc->uc_drv->id == key)
39                         return uc;
40         }
41
42         return NULL;
43 }
44
45 /**
46  * uclass_add() - Create new uclass in list
47  * @id: Id number to create
48  * @ucp: Returns pointer to uclass, or NULL on error
49  * @return 0 on success, -ve on error
50  *
51  * The new uclass is added to the list. There must be only one uclass for
52  * each id.
53  */
54 static int uclass_add(enum uclass_id id, struct uclass **ucp)
55 {
56         struct uclass_driver *uc_drv;
57         struct uclass *uc;
58         int ret;
59
60         *ucp = NULL;
61         uc_drv = lists_uclass_lookup(id);
62         if (!uc_drv) {
63                 debug("Cannot find uclass for id %d: please add the UCLASS_DRIVER() declaration for this UCLASS_... id\n",
64                       id);
65                 /*
66                  * Use a strange error to make this case easier to find. When
67                  * a uclass is not available it can prevent driver model from
68                  * starting up and this failure is otherwise hard to debug.
69                  */
70                 return -EPFNOSUPPORT;
71         }
72         uc = calloc(1, sizeof(*uc));
73         if (!uc)
74                 return -ENOMEM;
75         if (uc_drv->priv_auto) {
76                 void *ptr;
77
78                 ptr = calloc(1, uc_drv->priv_auto);
79                 if (!ptr) {
80                         ret = -ENOMEM;
81                         goto fail_mem;
82                 }
83                 uclass_set_priv(uc, ptr);
84         }
85         uc->uc_drv = uc_drv;
86         INIT_LIST_HEAD(&uc->sibling_node);
87         INIT_LIST_HEAD(&uc->dev_head);
88         list_add(&uc->sibling_node, DM_UCLASS_ROOT_NON_CONST);
89
90         if (uc_drv->init) {
91                 ret = uc_drv->init(uc);
92                 if (ret)
93                         goto fail;
94         }
95
96         *ucp = uc;
97
98         return 0;
99 fail:
100         if (uc_drv->priv_auto) {
101                 free(uclass_get_priv(uc));
102                 uclass_set_priv(uc, NULL);
103         }
104         list_del(&uc->sibling_node);
105 fail_mem:
106         free(uc);
107
108         return ret;
109 }
110
111 int uclass_destroy(struct uclass *uc)
112 {
113         struct uclass_driver *uc_drv;
114         struct udevice *dev;
115         int ret;
116
117         /*
118          * We cannot use list_for_each_entry_safe() here. If a device in this
119          * uclass has a child device also in this uclass, it will be also be
120          * unbound (by the recursion in the call to device_unbind() below).
121          * We can loop until the list is empty.
122          */
123         while (!list_empty(&uc->dev_head)) {
124                 dev = list_first_entry(&uc->dev_head, struct udevice,
125                                        uclass_node);
126                 ret = device_remove(dev, DM_REMOVE_NORMAL | DM_REMOVE_NO_PD);
127                 if (ret)
128                         return log_msg_ret("remove", ret);
129                 ret = device_unbind(dev);
130                 if (ret)
131                         return log_msg_ret("unbind", ret);
132         }
133
134         uc_drv = uc->uc_drv;
135         if (uc_drv->destroy)
136                 uc_drv->destroy(uc);
137         list_del(&uc->sibling_node);
138         if (uc_drv->priv_auto)
139                 free(uclass_get_priv(uc));
140         free(uc);
141
142         return 0;
143 }
144
145 int uclass_get(enum uclass_id id, struct uclass **ucp)
146 {
147         struct uclass *uc;
148
149         /* Immediately fail if driver model is not set up */
150         if (!gd->uclass_root)
151                 return -EDEADLK;
152         *ucp = NULL;
153         uc = uclass_find(id);
154         if (!uc) {
155                 if (CONFIG_IS_ENABLED(OF_PLATDATA_INST))
156                         return -ENOENT;
157                 return uclass_add(id, ucp);
158         }
159         *ucp = uc;
160
161         return 0;
162 }
163
164 const char *uclass_get_name(enum uclass_id id)
165 {
166         struct uclass *uc;
167
168         if (uclass_get(id, &uc))
169                 return NULL;
170         return uc->uc_drv->name;
171 }
172
173 void *uclass_get_priv(const struct uclass *uc)
174 {
175         return uc->priv_;
176 }
177
178 void uclass_set_priv(struct uclass *uc, void *priv)
179 {
180         uc->priv_ = priv;
181 }
182
183 enum uclass_id uclass_get_by_name_len(const char *name, int len)
184 {
185         int i;
186
187         for (i = 0; i < UCLASS_COUNT; i++) {
188                 struct uclass_driver *uc_drv = lists_uclass_lookup(i);
189
190                 if (uc_drv && !strncmp(uc_drv->name, name, len))
191                         return i;
192         }
193
194         return UCLASS_INVALID;
195 }
196
197 enum uclass_id uclass_get_by_name(const char *name)
198 {
199         return uclass_get_by_name_len(name, strlen(name));
200 }
201
202 int dev_get_uclass_index(struct udevice *dev, struct uclass **ucp)
203 {
204         struct udevice *iter;
205         struct uclass *uc = dev->uclass;
206         int i = 0;
207
208         if (list_empty(&uc->dev_head))
209                 return -ENODEV;
210
211         uclass_foreach_dev(iter, uc) {
212                 if (iter == dev) {
213                         if (ucp)
214                                 *ucp = uc;
215                         return i;
216                 }
217                 i++;
218         }
219
220         return -ENODEV;
221 }
222
223 int uclass_find_device(enum uclass_id id, int index, struct udevice **devp)
224 {
225         struct uclass *uc;
226         struct udevice *dev;
227         int ret;
228
229         *devp = NULL;
230         ret = uclass_get(id, &uc);
231         if (ret)
232                 return ret;
233         if (list_empty(&uc->dev_head))
234                 return -ENODEV;
235
236         uclass_foreach_dev(dev, uc) {
237                 if (!index--) {
238                         *devp = dev;
239                         return 0;
240                 }
241         }
242
243         return -ENODEV;
244 }
245
246 int uclass_find_first_device(enum uclass_id id, struct udevice **devp)
247 {
248         struct uclass *uc;
249         int ret;
250
251         *devp = NULL;
252         ret = uclass_get(id, &uc);
253         if (ret)
254                 return ret;
255         if (list_empty(&uc->dev_head))
256                 return 0;
257
258         *devp = list_first_entry(&uc->dev_head, struct udevice, uclass_node);
259
260         return 0;
261 }
262
263 int uclass_find_next_device(struct udevice **devp)
264 {
265         struct udevice *dev = *devp;
266
267         *devp = NULL;
268         if (list_is_last(&dev->uclass_node, &dev->uclass->dev_head))
269                 return 0;
270
271         *devp = list_entry(dev->uclass_node.next, struct udevice, uclass_node);
272
273         return 0;
274 }
275
276 int uclass_find_device_by_name(enum uclass_id id, const char *name,
277                                struct udevice **devp)
278 {
279         struct uclass *uc;
280         struct udevice *dev;
281         int ret;
282
283         *devp = NULL;
284         if (!name)
285                 return -EINVAL;
286         ret = uclass_get(id, &uc);
287         if (ret)
288                 return ret;
289
290         uclass_foreach_dev(dev, uc) {
291                 if (!strcmp(dev->name, name)) {
292                         *devp = dev;
293                         return 0;
294                 }
295         }
296
297         return -ENODEV;
298 }
299
300 int uclass_find_next_free_seq(struct uclass *uc)
301 {
302         struct udevice *dev;
303         int max = -1;
304
305         /* If using aliases, start with the highest alias value */
306         if (CONFIG_IS_ENABLED(DM_SEQ_ALIAS) &&
307             (uc->uc_drv->flags & DM_UC_FLAG_SEQ_ALIAS))
308                 max = dev_read_alias_highest_id(uc->uc_drv->name);
309
310         /* Avoid conflict with existing devices */
311         list_for_each_entry(dev, &uc->dev_head, uclass_node) {
312                 if (dev->seq_ > max)
313                         max = dev->seq_;
314         }
315         /*
316          * At this point, max will be -1 if there are no existing aliases or
317          * devices
318          */
319
320         return max + 1;
321 }
322
323 int uclass_find_device_by_seq(enum uclass_id id, int seq, struct udevice **devp)
324 {
325         struct uclass *uc;
326         struct udevice *dev;
327         int ret;
328
329         *devp = NULL;
330         log_debug("%d\n", seq);
331         if (seq == -1)
332                 return -ENODEV;
333         ret = uclass_get(id, &uc);
334         if (ret)
335                 return ret;
336
337         uclass_foreach_dev(dev, uc) {
338                 log_debug("   - %d '%s'\n", dev->seq_, dev->name);
339                 if (dev->seq_ == seq) {
340                         *devp = dev;
341                         log_debug("   - found\n");
342                         return 0;
343                 }
344         }
345         log_debug("   - not found\n");
346
347         return -ENODEV;
348 }
349
350 int uclass_find_device_by_of_offset(enum uclass_id id, int node,
351                                     struct udevice **devp)
352 {
353         struct uclass *uc;
354         struct udevice *dev;
355         int ret;
356
357         *devp = NULL;
358         if (node < 0)
359                 return -ENODEV;
360         ret = uclass_get(id, &uc);
361         if (ret)
362                 return ret;
363
364         uclass_foreach_dev(dev, uc) {
365                 if (dev_of_offset(dev) == node) {
366                         *devp = dev;
367                         return 0;
368                 }
369         }
370
371         return -ENODEV;
372 }
373
374 int uclass_find_device_by_ofnode(enum uclass_id id, ofnode node,
375                                  struct udevice **devp)
376 {
377         struct uclass *uc;
378         struct udevice *dev;
379         int ret;
380
381         log(LOGC_DM, LOGL_DEBUG, "Looking for %s\n", ofnode_get_name(node));
382         *devp = NULL;
383         if (!ofnode_valid(node))
384                 return -ENODEV;
385         ret = uclass_get(id, &uc);
386         if (ret)
387                 return ret;
388
389         uclass_foreach_dev(dev, uc) {
390                 log(LOGC_DM, LOGL_DEBUG_CONTENT, "      - checking %s\n",
391                     dev->name);
392                 if (ofnode_equal(dev_ofnode(dev), node)) {
393                         *devp = dev;
394                         goto done;
395                 }
396         }
397         ret = -ENODEV;
398
399 done:
400         log(LOGC_DM, LOGL_DEBUG, "   - result for %s: %s (ret=%d)\n",
401             ofnode_get_name(node), *devp ? (*devp)->name : "(none)", ret);
402         return ret;
403 }
404
405 #if CONFIG_IS_ENABLED(OF_REAL)
406 int uclass_find_device_by_phandle(enum uclass_id id, struct udevice *parent,
407                                   const char *name, struct udevice **devp)
408 {
409         struct udevice *dev;
410         struct uclass *uc;
411         int find_phandle;
412         int ret;
413
414         *devp = NULL;
415         find_phandle = dev_read_u32_default(parent, name, -1);
416         if (find_phandle <= 0)
417                 return -ENOENT;
418         ret = uclass_get(id, &uc);
419         if (ret)
420                 return ret;
421
422         uclass_foreach_dev(dev, uc) {
423                 uint phandle;
424
425                 phandle = dev_read_phandle(dev);
426
427                 if (phandle == find_phandle) {
428                         *devp = dev;
429                         return 0;
430                 }
431         }
432
433         return -ENODEV;
434 }
435 #endif
436
437 int uclass_get_device_by_driver(enum uclass_id id,
438                                 const struct driver *find_drv,
439                                 struct udevice **devp)
440 {
441         struct udevice *dev;
442         struct uclass *uc;
443         int ret;
444
445         ret = uclass_get(id, &uc);
446         if (ret)
447                 return ret;
448
449         uclass_foreach_dev(dev, uc) {
450                 if (dev->driver == find_drv)
451                         return uclass_get_device_tail(dev, 0, devp);
452         }
453
454         return -ENODEV;
455 }
456
457 int uclass_get_device_tail(struct udevice *dev, int ret, struct udevice **devp)
458 {
459         if (ret)
460                 return ret;
461
462         assert(dev);
463         ret = device_probe(dev);
464         if (ret)
465                 return ret;
466
467         *devp = dev;
468
469         return 0;
470 }
471
472 int uclass_get_device(enum uclass_id id, int index, struct udevice **devp)
473 {
474         struct udevice *dev;
475         int ret;
476
477         *devp = NULL;
478         ret = uclass_find_device(id, index, &dev);
479         return uclass_get_device_tail(dev, ret, devp);
480 }
481
482 int uclass_get_device_by_name(enum uclass_id id, const char *name,
483                               struct udevice **devp)
484 {
485         struct udevice *dev;
486         int ret;
487
488         *devp = NULL;
489         ret = uclass_find_device_by_name(id, name, &dev);
490         return uclass_get_device_tail(dev, ret, devp);
491 }
492
493 int uclass_get_device_by_seq(enum uclass_id id, int seq, struct udevice **devp)
494 {
495         struct udevice *dev;
496         int ret;
497
498         *devp = NULL;
499         ret = uclass_find_device_by_seq(id, seq, &dev);
500
501         return uclass_get_device_tail(dev, ret, devp);
502 }
503
504 int uclass_get_device_by_of_offset(enum uclass_id id, int node,
505                                    struct udevice **devp)
506 {
507         struct udevice *dev;
508         int ret;
509
510         *devp = NULL;
511         ret = uclass_find_device_by_of_offset(id, node, &dev);
512         return uclass_get_device_tail(dev, ret, devp);
513 }
514
515 int uclass_get_device_by_ofnode(enum uclass_id id, ofnode node,
516                                 struct udevice **devp)
517 {
518         struct udevice *dev;
519         int ret;
520
521         log(LOGC_DM, LOGL_DEBUG, "Looking for %s\n", ofnode_get_name(node));
522         *devp = NULL;
523         ret = uclass_find_device_by_ofnode(id, node, &dev);
524         log(LOGC_DM, LOGL_DEBUG, "   - result for %s: %s (ret=%d)\n",
525             ofnode_get_name(node), dev ? dev->name : "(none)", ret);
526
527         return uclass_get_device_tail(dev, ret, devp);
528 }
529
530 #if CONFIG_IS_ENABLED(OF_CONTROL)
531 int uclass_get_device_by_phandle_id(enum uclass_id id, uint phandle_id,
532                                     struct udevice **devp)
533 {
534         struct udevice *dev;
535         struct uclass *uc;
536         int ret;
537
538         *devp = NULL;
539         ret = uclass_get(id, &uc);
540         if (ret)
541                 return ret;
542
543         uclass_foreach_dev(dev, uc) {
544                 uint phandle;
545
546                 phandle = dev_read_phandle(dev);
547
548                 if (phandle == phandle_id) {
549                         *devp = dev;
550                         return uclass_get_device_tail(dev, ret, devp);
551                 }
552         }
553
554         return -ENODEV;
555 }
556
557 int uclass_get_device_by_phandle(enum uclass_id id, struct udevice *parent,
558                                  const char *name, struct udevice **devp)
559 {
560         struct udevice *dev;
561         int ret;
562
563         *devp = NULL;
564         ret = uclass_find_device_by_phandle(id, parent, name, &dev);
565         return uclass_get_device_tail(dev, ret, devp);
566 }
567 #endif
568
569 int uclass_first_device(enum uclass_id id, struct udevice **devp)
570 {
571         struct udevice *dev;
572         int ret;
573
574         *devp = NULL;
575         ret = uclass_find_first_device(id, &dev);
576         if (!dev)
577                 return 0;
578         return uclass_get_device_tail(dev, ret, devp);
579 }
580
581 int uclass_first_device_err(enum uclass_id id, struct udevice **devp)
582 {
583         int ret;
584
585         ret = uclass_first_device(id, devp);
586         if (ret)
587                 return ret;
588         else if (!*devp)
589                 return -ENODEV;
590
591         return 0;
592 }
593
594 int uclass_next_device(struct udevice **devp)
595 {
596         struct udevice *dev = *devp;
597         int ret;
598
599         *devp = NULL;
600         ret = uclass_find_next_device(&dev);
601         if (!dev)
602                 return 0;
603         return uclass_get_device_tail(dev, ret, devp);
604 }
605
606 int uclass_next_device_err(struct udevice **devp)
607 {
608         int ret;
609
610         ret = uclass_next_device(devp);
611         if (ret)
612                 return ret;
613         else if (!*devp)
614                 return -ENODEV;
615
616         return 0;
617 }
618
619 int uclass_first_device_check(enum uclass_id id, struct udevice **devp)
620 {
621         int ret;
622
623         *devp = NULL;
624         ret = uclass_find_first_device(id, devp);
625         if (ret)
626                 return ret;
627         if (!*devp)
628                 return 0;
629
630         return device_probe(*devp);
631 }
632
633 int uclass_next_device_check(struct udevice **devp)
634 {
635         int ret;
636
637         ret = uclass_find_next_device(devp);
638         if (ret)
639                 return ret;
640         if (!*devp)
641                 return 0;
642
643         return device_probe(*devp);
644 }
645
646 int uclass_first_device_drvdata(enum uclass_id id, ulong driver_data,
647                                 struct udevice **devp)
648 {
649         struct udevice *dev;
650         struct uclass *uc;
651
652         uclass_id_foreach_dev(id, dev, uc) {
653                 if (dev_get_driver_data(dev) == driver_data) {
654                         *devp = dev;
655
656                         return device_probe(dev);
657                 }
658         }
659
660         return -ENODEV;
661 }
662
663 int uclass_bind_device(struct udevice *dev)
664 {
665         struct uclass *uc;
666         int ret;
667
668         uc = dev->uclass;
669         list_add_tail(&dev->uclass_node, &uc->dev_head);
670
671         if (dev->parent) {
672                 struct uclass_driver *uc_drv = dev->parent->uclass->uc_drv;
673
674                 if (uc_drv->child_post_bind) {
675                         ret = uc_drv->child_post_bind(dev);
676                         if (ret)
677                                 goto err;
678                 }
679         }
680
681         return 0;
682 err:
683         /* There is no need to undo the parent's post_bind call */
684         list_del(&dev->uclass_node);
685
686         return ret;
687 }
688
689 #if CONFIG_IS_ENABLED(DM_DEVICE_REMOVE)
690 int uclass_pre_unbind_device(struct udevice *dev)
691 {
692         struct uclass *uc;
693         int ret;
694
695         uc = dev->uclass;
696         if (uc->uc_drv->pre_unbind) {
697                 ret = uc->uc_drv->pre_unbind(dev);
698                 if (ret)
699                         return ret;
700         }
701
702         return 0;
703 }
704
705 int uclass_unbind_device(struct udevice *dev)
706 {
707         list_del(&dev->uclass_node);
708
709         return 0;
710 }
711 #endif
712
713 int uclass_pre_probe_device(struct udevice *dev)
714 {
715         struct uclass_driver *uc_drv;
716         int ret;
717
718         uc_drv = dev->uclass->uc_drv;
719         if (uc_drv->pre_probe) {
720                 ret = uc_drv->pre_probe(dev);
721                 if (ret)
722                         return ret;
723         }
724
725         if (!dev->parent)
726                 return 0;
727         uc_drv = dev->parent->uclass->uc_drv;
728         if (uc_drv->child_pre_probe) {
729                 ret = uc_drv->child_pre_probe(dev);
730                 if (ret)
731                         return ret;
732         }
733
734         return 0;
735 }
736
737 int uclass_post_probe_device(struct udevice *dev)
738 {
739         struct uclass_driver *uc_drv;
740         int ret;
741
742         if (dev->parent) {
743                 uc_drv = dev->parent->uclass->uc_drv;
744                 if (uc_drv->child_post_probe) {
745                         ret = uc_drv->child_post_probe(dev);
746                         if (ret)
747                                 return ret;
748                 }
749         }
750
751         uc_drv = dev->uclass->uc_drv;
752         if (uc_drv->post_probe) {
753                 ret = uc_drv->post_probe(dev);
754                 if (ret)
755                         return ret;
756         }
757
758         return 0;
759 }
760
761 #if CONFIG_IS_ENABLED(DM_DEVICE_REMOVE)
762 int uclass_pre_remove_device(struct udevice *dev)
763 {
764         struct uclass *uc;
765         int ret;
766
767         uc = dev->uclass;
768         if (uc->uc_drv->pre_remove) {
769                 ret = uc->uc_drv->pre_remove(dev);
770                 if (ret)
771                         return ret;
772         }
773
774         return 0;
775 }
776 #endif
777
778 int uclass_probe_all(enum uclass_id id)
779 {
780         struct udevice *dev;
781         int ret;
782
783         ret = uclass_first_device(id, &dev);
784         if (ret || !dev)
785                 return ret;
786
787         /* Scanning uclass to probe all devices */
788         while (dev) {
789                 ret = uclass_next_device(&dev);
790                 if (ret)
791                         return ret;
792         }
793
794         return 0;
795 }
796
797 int uclass_id_count(enum uclass_id id)
798 {
799         struct udevice *dev;
800         struct uclass *uc;
801         int count = 0;
802
803         uclass_id_foreach_dev(id, dev, uc)
804                 count++;
805
806         return count;
807 }
808
809 UCLASS_DRIVER(nop) = {
810         .id             = UCLASS_NOP,
811         .name           = "nop",
812 };