blob: c4be8a12d71a5518cdc735f821c6e459e25370bd [file] [log] [blame]
Simon Glass2e7d35d2014-02-26 15:59:21 -07001/*
2 * Copyright (c) 2013 Google, Inc
3 *
4 * (C) Copyright 2012
5 * Pavel Herrmann <morpheus.ibis@gmail.com>
6 *
7 * SPDX-License-Identifier: GPL-2.0+
8 */
9
10#include <common.h>
11#include <dm.h>
12#include <errno.h>
13#include <malloc.h>
14#include <dm/test.h>
15#include <dm/ut.h>
16#include <asm/io.h>
17
18int dm_testdrv_op_count[DM_TEST_OP_COUNT];
19static struct dm_test_state *dms = &global_test_state;
20
21static int testdrv_ping(struct device *dev, int pingval, int *pingret)
22{
23 const struct dm_test_pdata *pdata = dev_get_platdata(dev);
24 struct dm_test_priv *priv = dev_get_priv(dev);
25
26 *pingret = pingval + pdata->ping_add;
27 priv->ping_total += *pingret;
28
29 return 0;
30}
31
32static const struct test_ops test_ops = {
33 .ping = testdrv_ping,
34};
35
36static int test_bind(struct device *dev)
37{
38 /* Private data should not be allocated */
39 ut_assert(!dev_get_priv(dev));
40
41 dm_testdrv_op_count[DM_TEST_OP_BIND]++;
42 return 0;
43}
44
45static int test_probe(struct device *dev)
46{
47 struct dm_test_priv *priv = dev_get_priv(dev);
48
49 /* Private data should be allocated */
50 ut_assert(priv);
51
52 dm_testdrv_op_count[DM_TEST_OP_PROBE]++;
53 priv->ping_total += DM_TEST_START_TOTAL;
54 return 0;
55}
56
57static int test_remove(struct device *dev)
58{
59 /* Private data should still be allocated */
60 ut_assert(dev_get_priv(dev));
61
62 dm_testdrv_op_count[DM_TEST_OP_REMOVE]++;
63 return 0;
64}
65
66static int test_unbind(struct device *dev)
67{
68 /* Private data should not be allocated */
69 ut_assert(!dev->priv);
70
71 dm_testdrv_op_count[DM_TEST_OP_UNBIND]++;
72 return 0;
73}
74
75U_BOOT_DRIVER(test_drv) = {
76 .name = "test_drv",
77 .id = UCLASS_TEST,
78 .ops = &test_ops,
79 .bind = test_bind,
80 .probe = test_probe,
81 .remove = test_remove,
82 .unbind = test_unbind,
83 .priv_auto_alloc_size = sizeof(struct dm_test_priv),
84};
85
86U_BOOT_DRIVER(test2_drv) = {
87 .name = "test2_drv",
88 .id = UCLASS_TEST,
89 .ops = &test_ops,
90 .bind = test_bind,
91 .probe = test_probe,
92 .remove = test_remove,
93 .unbind = test_unbind,
94 .priv_auto_alloc_size = sizeof(struct dm_test_priv),
95};
96
97static int test_manual_drv_ping(struct device *dev, int pingval, int *pingret)
98{
99 *pingret = pingval + 2;
100
101 return 0;
102}
103
104static const struct test_ops test_manual_ops = {
105 .ping = test_manual_drv_ping,
106};
107
108static int test_manual_bind(struct device *dev)
109{
110 dm_testdrv_op_count[DM_TEST_OP_BIND]++;
111
112 return 0;
113}
114
115static int test_manual_probe(struct device *dev)
116{
117 dm_testdrv_op_count[DM_TEST_OP_PROBE]++;
118 if (!dms->force_fail_alloc)
119 dev->priv = calloc(1, sizeof(struct dm_test_priv));
120 if (!dev->priv)
121 return -ENOMEM;
122
123 return 0;
124}
125
126static int test_manual_remove(struct device *dev)
127{
128 dm_testdrv_op_count[DM_TEST_OP_REMOVE]++;
129 return 0;
130}
131
132static int test_manual_unbind(struct device *dev)
133{
134 dm_testdrv_op_count[DM_TEST_OP_UNBIND]++;
135 return 0;
136}
137
138U_BOOT_DRIVER(test_manual_drv) = {
139 .name = "test_manual_drv",
140 .id = UCLASS_TEST,
141 .ops = &test_manual_ops,
142 .bind = test_manual_bind,
143 .probe = test_manual_probe,
144 .remove = test_manual_remove,
145 .unbind = test_manual_unbind,
146};