i2c: muxes: fix leaked i2c adapter device node references
[firefly-linux-kernel-4.4.55.git] / drivers / i2c / muxes / i2c-mux-reg.c
1 /*
2  * I2C multiplexer using a single register
3  *
4  * Copyright 2015 Freescale Semiconductor
5  * York Sun  <yorksun@freescale.com>
6  *
7  * This program is free software; you can redistribute it and/or modify it
8  * under the terms of the GNU General Public License as published by the
9  * Free Software Foundation; either version 2 of the License, or (at your
10  * option) any later version.
11  */
12
13 #include <linux/i2c.h>
14 #include <linux/i2c-mux.h>
15 #include <linux/init.h>
16 #include <linux/io.h>
17 #include <linux/module.h>
18 #include <linux/of_address.h>
19 #include <linux/platform_data/i2c-mux-reg.h>
20 #include <linux/platform_device.h>
21 #include <linux/slab.h>
22
23 struct regmux {
24         struct i2c_adapter *parent;
25         struct i2c_adapter **adap; /* child busses */
26         struct i2c_mux_reg_platform_data data;
27 };
28
29 static int i2c_mux_reg_set(const struct regmux *mux, unsigned int chan_id)
30 {
31         if (!mux->data.reg)
32                 return -EINVAL;
33
34         switch (mux->data.reg_size) {
35         case 4:
36                 if (mux->data.little_endian) {
37                         iowrite32(chan_id, mux->data.reg);
38                         if (!mux->data.write_only)
39                                 ioread32(mux->data.reg);
40                 } else {
41                         iowrite32be(chan_id, mux->data.reg);
42                         if (!mux->data.write_only)
43                                 ioread32(mux->data.reg);
44                 }
45                 break;
46         case 2:
47                 if (mux->data.little_endian) {
48                         iowrite16(chan_id, mux->data.reg);
49                         if (!mux->data.write_only)
50                                 ioread16(mux->data.reg);
51                 } else {
52                         iowrite16be(chan_id, mux->data.reg);
53                         if (!mux->data.write_only)
54                                 ioread16be(mux->data.reg);
55                 }
56                 break;
57         case 1:
58                 iowrite8(chan_id, mux->data.reg);
59                 if (!mux->data.write_only)
60                         ioread8(mux->data.reg);
61                 break;
62         default:
63                 pr_err("Invalid register size\n");
64                 return -EINVAL;
65         }
66
67         return 0;
68 }
69
70 static int i2c_mux_reg_select(struct i2c_adapter *adap, void *data,
71                               unsigned int chan)
72 {
73         struct regmux *mux = data;
74
75         return i2c_mux_reg_set(mux, chan);
76 }
77
78 static int i2c_mux_reg_deselect(struct i2c_adapter *adap, void *data,
79                                 unsigned int chan)
80 {
81         struct regmux *mux = data;
82
83         if (mux->data.idle_in_use)
84                 return i2c_mux_reg_set(mux, mux->data.idle);
85
86         return 0;
87 }
88
89 #ifdef CONFIG_OF
90 static int i2c_mux_reg_probe_dt(struct regmux *mux,
91                                         struct platform_device *pdev)
92 {
93         struct device_node *np = pdev->dev.of_node;
94         struct device_node *adapter_np, *child;
95         struct i2c_adapter *adapter;
96         struct resource res;
97         unsigned *values;
98         int i = 0;
99
100         if (!np)
101                 return -ENODEV;
102
103         adapter_np = of_parse_phandle(np, "i2c-parent", 0);
104         if (!adapter_np) {
105                 dev_err(&pdev->dev, "Cannot parse i2c-parent\n");
106                 return -ENODEV;
107         }
108         adapter = of_find_i2c_adapter_by_node(adapter_np);
109         of_node_put(adapter_np);
110         if (!adapter)
111                 return -EPROBE_DEFER;
112
113         mux->parent = adapter;
114         mux->data.parent = i2c_adapter_id(adapter);
115         put_device(&adapter->dev);
116
117         mux->data.n_values = of_get_child_count(np);
118         if (of_find_property(np, "little-endian", NULL)) {
119                 mux->data.little_endian = true;
120         } else if (of_find_property(np, "big-endian", NULL)) {
121                 mux->data.little_endian = false;
122         } else {
123 #if defined(__BYTE_ORDER) ? __BYTE_ORDER == __LITTLE_ENDIAN : \
124         defined(__LITTLE_ENDIAN)
125                 mux->data.little_endian = true;
126 #elif defined(__BYTE_ORDER) ? __BYTE_ORDER == __BIG_ENDIAN : \
127         defined(__BIG_ENDIAN)
128                 mux->data.little_endian = false;
129 #else
130 #error Endianness not defined?
131 #endif
132         }
133         if (of_find_property(np, "write-only", NULL))
134                 mux->data.write_only = true;
135         else
136                 mux->data.write_only = false;
137
138         values = devm_kzalloc(&pdev->dev,
139                               sizeof(*mux->data.values) * mux->data.n_values,
140                               GFP_KERNEL);
141         if (!values) {
142                 dev_err(&pdev->dev, "Cannot allocate values array");
143                 return -ENOMEM;
144         }
145
146         for_each_child_of_node(np, child) {
147                 of_property_read_u32(child, "reg", values + i);
148                 i++;
149         }
150         mux->data.values = values;
151
152         if (!of_property_read_u32(np, "idle-state", &mux->data.idle))
153                 mux->data.idle_in_use = true;
154
155         /* map address from "reg" if exists */
156         if (of_address_to_resource(np, 0, &res)) {
157                 mux->data.reg_size = resource_size(&res);
158                 if (mux->data.reg_size > 4) {
159                         dev_err(&pdev->dev, "Invalid address size\n");
160                         return -EINVAL;
161                 }
162                 mux->data.reg = devm_ioremap_resource(&pdev->dev, &res);
163                 if (IS_ERR(mux->data.reg))
164                         return PTR_ERR(mux->data.reg);
165         }
166
167         return 0;
168 }
169 #else
170 static int i2c_mux_reg_probe_dt(struct gpiomux *mux,
171                                         struct platform_device *pdev)
172 {
173         return 0;
174 }
175 #endif
176
177 static int i2c_mux_reg_probe(struct platform_device *pdev)
178 {
179         struct regmux *mux;
180         struct i2c_adapter *parent;
181         struct resource *res;
182         int (*deselect)(struct i2c_adapter *, void *, u32);
183         unsigned int class;
184         int i, ret, nr;
185
186         mux = devm_kzalloc(&pdev->dev, sizeof(*mux), GFP_KERNEL);
187         if (!mux)
188                 return -ENOMEM;
189
190         platform_set_drvdata(pdev, mux);
191
192         if (dev_get_platdata(&pdev->dev)) {
193                 memcpy(&mux->data, dev_get_platdata(&pdev->dev),
194                         sizeof(mux->data));
195
196                 parent = i2c_get_adapter(mux->data.parent);
197                 if (!parent)
198                         return -EPROBE_DEFER;
199
200                 mux->parent = parent;
201         } else {
202                 ret = i2c_mux_reg_probe_dt(mux, pdev);
203                 if (ret < 0) {
204                         dev_err(&pdev->dev, "Error parsing device tree");
205                         return ret;
206                 }
207         }
208
209         if (!mux->data.reg) {
210                 dev_info(&pdev->dev,
211                         "Register not set, using platform resource\n");
212                 res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
213                 mux->data.reg_size = resource_size(res);
214                 if (mux->data.reg_size > 4) {
215                         dev_err(&pdev->dev, "Invalid resource size\n");
216                         return -EINVAL;
217                 }
218                 mux->data.reg = devm_ioremap_resource(&pdev->dev, res);
219                 if (IS_ERR(mux->data.reg))
220                         return PTR_ERR(mux->data.reg);
221         }
222
223         mux->adap = devm_kzalloc(&pdev->dev,
224                                  sizeof(*mux->adap) * mux->data.n_values,
225                                  GFP_KERNEL);
226         if (!mux->adap) {
227                 dev_err(&pdev->dev, "Cannot allocate i2c_adapter structure");
228                 return -ENOMEM;
229         }
230
231         if (mux->data.idle_in_use)
232                 deselect = i2c_mux_reg_deselect;
233         else
234                 deselect = NULL;
235
236         for (i = 0; i < mux->data.n_values; i++) {
237                 nr = mux->data.base_nr ? (mux->data.base_nr + i) : 0;
238                 class = mux->data.classes ? mux->data.classes[i] : 0;
239
240                 mux->adap[i] = i2c_add_mux_adapter(mux->parent, &pdev->dev, mux,
241                                                    nr, mux->data.values[i],
242                                                    class, i2c_mux_reg_select,
243                                                    deselect);
244                 if (!mux->adap[i]) {
245                         ret = -ENODEV;
246                         dev_err(&pdev->dev, "Failed to add adapter %d\n", i);
247                         goto add_adapter_failed;
248                 }
249         }
250
251         dev_dbg(&pdev->dev, "%d port mux on %s adapter\n",
252                  mux->data.n_values, mux->parent->name);
253
254         return 0;
255
256 add_adapter_failed:
257         for (; i > 0; i--)
258                 i2c_del_mux_adapter(mux->adap[i - 1]);
259
260         return ret;
261 }
262
263 static int i2c_mux_reg_remove(struct platform_device *pdev)
264 {
265         struct regmux *mux = platform_get_drvdata(pdev);
266         int i;
267
268         for (i = 0; i < mux->data.n_values; i++)
269                 i2c_del_mux_adapter(mux->adap[i]);
270
271         i2c_put_adapter(mux->parent);
272
273         return 0;
274 }
275
276 static const struct of_device_id i2c_mux_reg_of_match[] = {
277         { .compatible = "i2c-mux-reg", },
278         {},
279 };
280 MODULE_DEVICE_TABLE(of, i2c_mux_reg_of_match);
281
282 static struct platform_driver i2c_mux_reg_driver = {
283         .probe  = i2c_mux_reg_probe,
284         .remove = i2c_mux_reg_remove,
285         .driver = {
286                 .name   = "i2c-mux-reg",
287         },
288 };
289
290 module_platform_driver(i2c_mux_reg_driver);
291
292 MODULE_DESCRIPTION("Register-based I2C multiplexer driver");
293 MODULE_AUTHOR("York Sun <yorksun@freescale.com>");
294 MODULE_LICENSE("GPL");
295 MODULE_ALIAS("platform:i2c-mux-reg");