ring-buffer: Prevent overflow of size in ring_buffer_resize()
[cascardo/linux.git] / drivers / input / misc / twl6040-vibra.c
1 /*
2  * twl6040-vibra.c - TWL6040 Vibrator driver
3  *
4  * Author:      Jorge Eduardo Candelaria <jorge.candelaria@ti.com>
5  * Author:      Misael Lopez Cruz <misael.lopez@ti.com>
6  *
7  * Copyright:   (C) 2011 Texas Instruments, Inc.
8  *
9  * Based on twl4030-vibra.c by Henrik Saari <henrik.saari@nokia.com>
10  *                              Felipe Balbi <felipe.balbi@nokia.com>
11  *                              Jari Vanhala <ext-javi.vanhala@nokia.com>
12  *
13  * This program is free software; you can redistribute it and/or modify
14  * it under the terms of the GNU General Public License version 2 as
15  * published by the Free Software Foundation.
16  *
17  * This program is distributed in the hope that it will be useful, but
18  * WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
20  * General Public License for more details.
21  *
22  * You should have received a copy of the GNU General Public License
23  * along with this program; if not, write to the Free Software
24  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
25  * 02110-1301 USA
26  *
27  */
28 #include <linux/module.h>
29 #include <linux/platform_device.h>
30 #include <linux/of.h>
31 #include <linux/workqueue.h>
32 #include <linux/input.h>
33 #include <linux/mfd/twl6040.h>
34 #include <linux/slab.h>
35 #include <linux/delay.h>
36 #include <linux/regulator/consumer.h>
37
38 #define EFFECT_DIR_180_DEG      0x8000
39
40 /* Recommended modulation index 85% */
41 #define TWL6040_VIBRA_MOD       85
42
43 #define TWL6040_NUM_SUPPLIES 2
44
45 struct vibra_info {
46         struct device *dev;
47         struct input_dev *input_dev;
48         struct work_struct play_work;
49         struct mutex mutex;
50         int irq;
51
52         bool enabled;
53         int weak_speed;
54         int strong_speed;
55         int direction;
56
57         unsigned int vibldrv_res;
58         unsigned int vibrdrv_res;
59         unsigned int viblmotor_res;
60         unsigned int vibrmotor_res;
61
62         struct regulator_bulk_data supplies[TWL6040_NUM_SUPPLIES];
63
64         struct twl6040 *twl6040;
65 };
66
67 static irqreturn_t twl6040_vib_irq_handler(int irq, void *data)
68 {
69         struct vibra_info *info = data;
70         struct twl6040 *twl6040 = info->twl6040;
71         u8 status;
72
73         status = twl6040_reg_read(twl6040, TWL6040_REG_STATUS);
74         if (status & TWL6040_VIBLOCDET) {
75                 dev_warn(info->dev, "Left Vibrator overcurrent detected\n");
76                 twl6040_clear_bits(twl6040, TWL6040_REG_VIBCTLL,
77                                    TWL6040_VIBENA);
78         }
79         if (status & TWL6040_VIBROCDET) {
80                 dev_warn(info->dev, "Right Vibrator overcurrent detected\n");
81                 twl6040_clear_bits(twl6040, TWL6040_REG_VIBCTLR,
82                                    TWL6040_VIBENA);
83         }
84
85         return IRQ_HANDLED;
86 }
87
88 static void twl6040_vibra_enable(struct vibra_info *info)
89 {
90         struct twl6040 *twl6040 = info->twl6040;
91         int ret;
92
93         ret = regulator_bulk_enable(ARRAY_SIZE(info->supplies), info->supplies);
94         if (ret) {
95                 dev_err(info->dev, "failed to enable regulators %d\n", ret);
96                 return;
97         }
98
99         twl6040_power(info->twl6040, 1);
100         if (twl6040_get_revid(twl6040) <= TWL6040_REV_ES1_1) {
101                 /*
102                  * ERRATA: Disable overcurrent protection for at least
103                  * 3ms when enabling vibrator drivers to avoid false
104                  * overcurrent detection
105                  */
106                 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLL,
107                                   TWL6040_VIBENA | TWL6040_VIBCTRL);
108                 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLR,
109                                   TWL6040_VIBENA | TWL6040_VIBCTRL);
110                 usleep_range(3000, 3500);
111         }
112
113         twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLL,
114                           TWL6040_VIBENA);
115         twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLR,
116                           TWL6040_VIBENA);
117
118         info->enabled = true;
119 }
120
121 static void twl6040_vibra_disable(struct vibra_info *info)
122 {
123         struct twl6040 *twl6040 = info->twl6040;
124
125         twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLL, 0x00);
126         twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLR, 0x00);
127         twl6040_power(info->twl6040, 0);
128
129         regulator_bulk_disable(ARRAY_SIZE(info->supplies), info->supplies);
130
131         info->enabled = false;
132 }
133
134 static u8 twl6040_vibra_code(int vddvib, int vibdrv_res, int motor_res,
135                              int speed, int direction)
136 {
137         int vpk, max_code;
138         u8 vibdat;
139
140         /* output swing */
141         vpk = (vddvib * motor_res * TWL6040_VIBRA_MOD) /
142                 (100 * (vibdrv_res + motor_res));
143
144         /* 50mV per VIBDAT code step */
145         max_code = vpk / 50;
146         if (max_code > TWL6040_VIBDAT_MAX)
147                 max_code = TWL6040_VIBDAT_MAX;
148
149         /* scale speed to max allowed code */
150         vibdat = (u8)((speed * max_code) / USHRT_MAX);
151
152         /* 2's complement for direction > 180 degrees */
153         vibdat *= direction;
154
155         return vibdat;
156 }
157
158 static void twl6040_vibra_set_effect(struct vibra_info *info)
159 {
160         struct twl6040 *twl6040 = info->twl6040;
161         u8 vibdatl, vibdatr;
162         int volt;
163
164         /* weak motor */
165         volt = regulator_get_voltage(info->supplies[0].consumer) / 1000;
166         vibdatl = twl6040_vibra_code(volt, info->vibldrv_res,
167                                      info->viblmotor_res,
168                                      info->weak_speed, info->direction);
169
170         /* strong motor */
171         volt = regulator_get_voltage(info->supplies[1].consumer) / 1000;
172         vibdatr = twl6040_vibra_code(volt, info->vibrdrv_res,
173                                      info->vibrmotor_res,
174                                      info->strong_speed, info->direction);
175
176         twl6040_reg_write(twl6040, TWL6040_REG_VIBDATL, vibdatl);
177         twl6040_reg_write(twl6040, TWL6040_REG_VIBDATR, vibdatr);
178 }
179
180 static void vibra_play_work(struct work_struct *work)
181 {
182         struct vibra_info *info = container_of(work,
183                                 struct vibra_info, play_work);
184
185         mutex_lock(&info->mutex);
186
187         if (info->weak_speed || info->strong_speed) {
188                 if (!info->enabled)
189                         twl6040_vibra_enable(info);
190
191                 twl6040_vibra_set_effect(info);
192         } else if (info->enabled)
193                 twl6040_vibra_disable(info);
194
195         mutex_unlock(&info->mutex);
196 }
197
198 static int vibra_play(struct input_dev *input, void *data,
199                       struct ff_effect *effect)
200 {
201         struct vibra_info *info = input_get_drvdata(input);
202         int ret;
203
204         /* Do not allow effect, while the routing is set to use audio */
205         ret = twl6040_get_vibralr_status(info->twl6040);
206         if (ret & TWL6040_VIBSEL) {
207                 dev_info(&input->dev, "Vibra is configured for audio\n");
208                 return -EBUSY;
209         }
210
211         info->weak_speed = effect->u.rumble.weak_magnitude;
212         info->strong_speed = effect->u.rumble.strong_magnitude;
213         info->direction = effect->direction < EFFECT_DIR_180_DEG ? 1 : -1;
214
215         schedule_work(&info->play_work);
216
217         return 0;
218 }
219
220 static void twl6040_vibra_close(struct input_dev *input)
221 {
222         struct vibra_info *info = input_get_drvdata(input);
223
224         cancel_work_sync(&info->play_work);
225
226         mutex_lock(&info->mutex);
227
228         if (info->enabled)
229                 twl6040_vibra_disable(info);
230
231         mutex_unlock(&info->mutex);
232 }
233
234 static int __maybe_unused twl6040_vibra_suspend(struct device *dev)
235 {
236         struct platform_device *pdev = to_platform_device(dev);
237         struct vibra_info *info = platform_get_drvdata(pdev);
238
239         mutex_lock(&info->mutex);
240
241         if (info->enabled)
242                 twl6040_vibra_disable(info);
243
244         mutex_unlock(&info->mutex);
245
246         return 0;
247 }
248
249 static SIMPLE_DEV_PM_OPS(twl6040_vibra_pm_ops, twl6040_vibra_suspend, NULL);
250
251 static int twl6040_vibra_probe(struct platform_device *pdev)
252 {
253         struct device *twl6040_core_dev = pdev->dev.parent;
254         struct device_node *twl6040_core_node;
255         struct vibra_info *info;
256         int vddvibl_uV = 0;
257         int vddvibr_uV = 0;
258         int error;
259
260         twl6040_core_node = of_find_node_by_name(twl6040_core_dev->of_node,
261                                                  "vibra");
262         if (!twl6040_core_node) {
263                 dev_err(&pdev->dev, "parent of node is missing?\n");
264                 return -EINVAL;
265         }
266
267         info = devm_kzalloc(&pdev->dev, sizeof(*info), GFP_KERNEL);
268         if (!info) {
269                 of_node_put(twl6040_core_node);
270                 dev_err(&pdev->dev, "couldn't allocate memory\n");
271                 return -ENOMEM;
272         }
273
274         info->dev = &pdev->dev;
275
276         info->twl6040 = dev_get_drvdata(pdev->dev.parent);
277
278         of_property_read_u32(twl6040_core_node, "ti,vibldrv-res",
279                              &info->vibldrv_res);
280         of_property_read_u32(twl6040_core_node, "ti,vibrdrv-res",
281                              &info->vibrdrv_res);
282         of_property_read_u32(twl6040_core_node, "ti,viblmotor-res",
283                              &info->viblmotor_res);
284         of_property_read_u32(twl6040_core_node, "ti,vibrmotor-res",
285                              &info->vibrmotor_res);
286         of_property_read_u32(twl6040_core_node, "ti,vddvibl-uV", &vddvibl_uV);
287         of_property_read_u32(twl6040_core_node, "ti,vddvibr-uV", &vddvibr_uV);
288
289         of_node_put(twl6040_core_node);
290
291         if ((!info->vibldrv_res && !info->viblmotor_res) ||
292             (!info->vibrdrv_res && !info->vibrmotor_res)) {
293                 dev_err(info->dev, "invalid vibra driver/motor resistance\n");
294                 return -EINVAL;
295         }
296
297         info->irq = platform_get_irq(pdev, 0);
298         if (info->irq < 0) {
299                 dev_err(info->dev, "invalid irq\n");
300                 return -EINVAL;
301         }
302
303         mutex_init(&info->mutex);
304
305         error = devm_request_threaded_irq(&pdev->dev, info->irq, NULL,
306                                           twl6040_vib_irq_handler,
307                                           IRQF_ONESHOT,
308                                           "twl6040_irq_vib", info);
309         if (error) {
310                 dev_err(info->dev, "VIB IRQ request failed: %d\n", error);
311                 return error;
312         }
313
314         info->supplies[0].supply = "vddvibl";
315         info->supplies[1].supply = "vddvibr";
316         /*
317          * When booted with Device tree the regulators are attached to the
318          * parent device (twl6040 MFD core)
319          */
320         error = devm_regulator_bulk_get(twl6040_core_dev,
321                                         ARRAY_SIZE(info->supplies),
322                                         info->supplies);
323         if (error) {
324                 dev_err(info->dev, "couldn't get regulators %d\n", error);
325                 return error;
326         }
327
328         if (vddvibl_uV) {
329                 error = regulator_set_voltage(info->supplies[0].consumer,
330                                               vddvibl_uV, vddvibl_uV);
331                 if (error) {
332                         dev_err(info->dev, "failed to set VDDVIBL volt %d\n",
333                                 error);
334                         return error;
335                 }
336         }
337
338         if (vddvibr_uV) {
339                 error = regulator_set_voltage(info->supplies[1].consumer,
340                                               vddvibr_uV, vddvibr_uV);
341                 if (error) {
342                         dev_err(info->dev, "failed to set VDDVIBR volt %d\n",
343                                 error);
344                         return error;
345                 }
346         }
347
348         INIT_WORK(&info->play_work, vibra_play_work);
349
350         info->input_dev = devm_input_allocate_device(&pdev->dev);
351         if (!info->input_dev) {
352                 dev_err(info->dev, "couldn't allocate input device\n");
353                 return -ENOMEM;
354         }
355
356         input_set_drvdata(info->input_dev, info);
357
358         info->input_dev->name = "twl6040:vibrator";
359         info->input_dev->id.version = 1;
360         info->input_dev->close = twl6040_vibra_close;
361         __set_bit(FF_RUMBLE, info->input_dev->ffbit);
362
363         error = input_ff_create_memless(info->input_dev, NULL, vibra_play);
364         if (error) {
365                 dev_err(info->dev, "couldn't register vibrator to FF\n");
366                 return error;
367         }
368
369         error = input_register_device(info->input_dev);
370         if (error) {
371                 dev_err(info->dev, "couldn't register input device\n");
372                 return error;
373         }
374
375         platform_set_drvdata(pdev, info);
376
377         return 0;
378 }
379
380 static struct platform_driver twl6040_vibra_driver = {
381         .probe          = twl6040_vibra_probe,
382         .driver         = {
383                 .name   = "twl6040-vibra",
384                 .pm     = &twl6040_vibra_pm_ops,
385         },
386 };
387 module_platform_driver(twl6040_vibra_driver);
388
389 MODULE_ALIAS("platform:twl6040-vibra");
390 MODULE_DESCRIPTION("TWL6040 Vibra driver");
391 MODULE_LICENSE("GPL");
392 MODULE_AUTHOR("Jorge Eduardo Candelaria <jorge.candelaria@ti.com>");
393 MODULE_AUTHOR("Misael Lopez Cruz <misael.lopez@ti.com>");