regulator: core: Fix regualtor_ena_gpio_free not to access pin after freeing
[cascardo/linux.git] / drivers / i2c / busses / i2c-imx.c
1 /*
2  *      Copyright (C) 2002 Motorola GSG-China
3  *
4  *      This program is free software; you can redistribute it and/or
5  *      modify it under the terms of the GNU General Public License
6  *      as published by the Free Software Foundation; either version 2
7  *      of the License, or (at your option) any later version.
8  *
9  *      This program is distributed in the hope that it will be useful,
10  *      but WITHOUT ANY WARRANTY; without even the implied warranty of
11  *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  *      GNU General Public License for more details.
13  *
14  *      You should have received a copy of the GNU General Public License
15  *      along with this program; if not, write to the Free Software
16  *      Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,
17  *      USA.
18  *
19  * Author:
20  *      Darius Augulis, Teltonika Inc.
21  *
22  * Desc.:
23  *      Implementation of I2C Adapter/Algorithm Driver
24  *      for I2C Bus integrated in Freescale i.MX/MXC processors
25  *
26  *      Derived from Motorola GSG China I2C example driver
27  *
28  *      Copyright (C) 2005 Torsten Koschorrek <koschorrek at synertronixx.de
29  *      Copyright (C) 2005 Matthias Blaschke <blaschke at synertronixx.de
30  *      Copyright (C) 2007 RightHand Technologies, Inc.
31  *      Copyright (C) 2008 Darius Augulis <darius.augulis at teltonika.lt>
32  *
33  *      Copyright 2013 Freescale Semiconductor, Inc.
34  *
35  */
36
37 /** Includes *******************************************************************
38 *******************************************************************************/
39
40 #include <linux/init.h>
41 #include <linux/kernel.h>
42 #include <linux/module.h>
43 #include <linux/errno.h>
44 #include <linux/err.h>
45 #include <linux/interrupt.h>
46 #include <linux/delay.h>
47 #include <linux/i2c.h>
48 #include <linux/io.h>
49 #include <linux/sched.h>
50 #include <linux/platform_device.h>
51 #include <linux/clk.h>
52 #include <linux/slab.h>
53 #include <linux/of.h>
54 #include <linux/of_device.h>
55 #include <linux/platform_data/i2c-imx.h>
56
57 /** Defines ********************************************************************
58 *******************************************************************************/
59
60 /* This will be the driver name the kernel reports */
61 #define DRIVER_NAME "imx-i2c"
62
63 /* Default value */
64 #define IMX_I2C_BIT_RATE        100000  /* 100kHz */
65
66 /* IMX I2C registers:
67  * the I2C register offset is different between SoCs,
68  * to provid support for all these chips, split the
69  * register offset into a fixed base address and a
70  * variable shift value, then the full register offset
71  * will be calculated by
72  * reg_off = ( reg_base_addr << reg_shift)
73  */
74 #define IMX_I2C_IADR    0x00    /* i2c slave address */
75 #define IMX_I2C_IFDR    0x01    /* i2c frequency divider */
76 #define IMX_I2C_I2CR    0x02    /* i2c control */
77 #define IMX_I2C_I2SR    0x03    /* i2c status */
78 #define IMX_I2C_I2DR    0x04    /* i2c transfer data */
79
80 #define IMX_I2C_REGSHIFT        2
81 #define VF610_I2C_REGSHIFT      0
82
83 /* Bits of IMX I2C registers */
84 #define I2SR_RXAK       0x01
85 #define I2SR_IIF        0x02
86 #define I2SR_SRW        0x04
87 #define I2SR_IAL        0x10
88 #define I2SR_IBB        0x20
89 #define I2SR_IAAS       0x40
90 #define I2SR_ICF        0x80
91 #define I2CR_RSTA       0x04
92 #define I2CR_TXAK       0x08
93 #define I2CR_MTX        0x10
94 #define I2CR_MSTA       0x20
95 #define I2CR_IIEN       0x40
96 #define I2CR_IEN        0x80
97
98 /* register bits different operating codes definition:
99  * 1) I2SR: Interrupt flags clear operation differ between SoCs:
100  * - write zero to clear(w0c) INT flag on i.MX,
101  * - but write one to clear(w1c) INT flag on Vybrid.
102  * 2) I2CR: I2C module enable operation also differ between SoCs:
103  * - set I2CR_IEN bit enable the module on i.MX,
104  * - but clear I2CR_IEN bit enable the module on Vybrid.
105  */
106 #define I2SR_CLR_OPCODE_W0C     0x0
107 #define I2SR_CLR_OPCODE_W1C     (I2SR_IAL | I2SR_IIF)
108 #define I2CR_IEN_OPCODE_0       0x0
109 #define I2CR_IEN_OPCODE_1       I2CR_IEN
110
111 /** Variables ******************************************************************
112 *******************************************************************************/
113
114 /*
115  * sorted list of clock divider, register value pairs
116  * taken from table 26-5, p.26-9, Freescale i.MX
117  * Integrated Portable System Processor Reference Manual
118  * Document Number: MC9328MXLRM, Rev. 5.1, 06/2007
119  *
120  * Duplicated divider values removed from list
121  */
122 struct imx_i2c_clk_pair {
123         u16     div;
124         u16     val;
125 };
126
127 static struct imx_i2c_clk_pair imx_i2c_clk_div[] = {
128         { 22,   0x20 }, { 24,   0x21 }, { 26,   0x22 }, { 28,   0x23 },
129         { 30,   0x00 }, { 32,   0x24 }, { 36,   0x25 }, { 40,   0x26 },
130         { 42,   0x03 }, { 44,   0x27 }, { 48,   0x28 }, { 52,   0x05 },
131         { 56,   0x29 }, { 60,   0x06 }, { 64,   0x2A }, { 72,   0x2B },
132         { 80,   0x2C }, { 88,   0x09 }, { 96,   0x2D }, { 104,  0x0A },
133         { 112,  0x2E }, { 128,  0x2F }, { 144,  0x0C }, { 160,  0x30 },
134         { 192,  0x31 }, { 224,  0x32 }, { 240,  0x0F }, { 256,  0x33 },
135         { 288,  0x10 }, { 320,  0x34 }, { 384,  0x35 }, { 448,  0x36 },
136         { 480,  0x13 }, { 512,  0x37 }, { 576,  0x14 }, { 640,  0x38 },
137         { 768,  0x39 }, { 896,  0x3A }, { 960,  0x17 }, { 1024, 0x3B },
138         { 1152, 0x18 }, { 1280, 0x3C }, { 1536, 0x3D }, { 1792, 0x3E },
139         { 1920, 0x1B }, { 2048, 0x3F }, { 2304, 0x1C }, { 2560, 0x1D },
140         { 3072, 0x1E }, { 3840, 0x1F }
141 };
142
143 /* Vybrid VF610 clock divider, register value pairs */
144 static struct imx_i2c_clk_pair vf610_i2c_clk_div[] = {
145         { 20,   0x00 }, { 22,   0x01 }, { 24,   0x02 }, { 26,   0x03 },
146         { 28,   0x04 }, { 30,   0x05 }, { 32,   0x09 }, { 34,   0x06 },
147         { 36,   0x0A }, { 40,   0x07 }, { 44,   0x0C }, { 48,   0x0D },
148         { 52,   0x43 }, { 56,   0x0E }, { 60,   0x45 }, { 64,   0x12 },
149         { 68,   0x0F }, { 72,   0x13 }, { 80,   0x14 }, { 88,   0x15 },
150         { 96,   0x19 }, { 104,  0x16 }, { 112,  0x1A }, { 128,  0x17 },
151         { 136,  0x4F }, { 144,  0x1C }, { 160,  0x1D }, { 176,  0x55 },
152         { 192,  0x1E }, { 208,  0x56 }, { 224,  0x22 }, { 228,  0x24 },
153         { 240,  0x1F }, { 256,  0x23 }, { 288,  0x5C }, { 320,  0x25 },
154         { 384,  0x26 }, { 448,  0x2A }, { 480,  0x27 }, { 512,  0x2B },
155         { 576,  0x2C }, { 640,  0x2D }, { 768,  0x31 }, { 896,  0x32 },
156         { 960,  0x2F }, { 1024, 0x33 }, { 1152, 0x34 }, { 1280, 0x35 },
157         { 1536, 0x36 }, { 1792, 0x3A }, { 1920, 0x37 }, { 2048, 0x3B },
158         { 2304, 0x3C }, { 2560, 0x3D }, { 3072, 0x3E }, { 3584, 0x7A },
159         { 3840, 0x3F }, { 4096, 0x7B }, { 5120, 0x7D }, { 6144, 0x7E },
160 };
161
162 enum imx_i2c_type {
163         IMX1_I2C,
164         IMX21_I2C,
165         VF610_I2C,
166 };
167
168 struct imx_i2c_hwdata {
169         enum imx_i2c_type       devtype;
170         unsigned                regshift;
171         struct imx_i2c_clk_pair *clk_div;
172         unsigned                ndivs;
173         unsigned                i2sr_clr_opcode;
174         unsigned                i2cr_ien_opcode;
175 };
176
177 struct imx_i2c_struct {
178         struct i2c_adapter      adapter;
179         struct clk              *clk;
180         void __iomem            *base;
181         wait_queue_head_t       queue;
182         unsigned long           i2csr;
183         unsigned int            disable_delay;
184         int                     stopped;
185         unsigned int            ifdr; /* IMX_I2C_IFDR */
186         unsigned int            cur_clk;
187         unsigned int            bitrate;
188         const struct imx_i2c_hwdata     *hwdata;
189 };
190
191 static const struct imx_i2c_hwdata imx1_i2c_hwdata  = {
192         .devtype                = IMX1_I2C,
193         .regshift               = IMX_I2C_REGSHIFT,
194         .clk_div                = imx_i2c_clk_div,
195         .ndivs                  = ARRAY_SIZE(imx_i2c_clk_div),
196         .i2sr_clr_opcode        = I2SR_CLR_OPCODE_W0C,
197         .i2cr_ien_opcode        = I2CR_IEN_OPCODE_1,
198
199 };
200
201 static const struct imx_i2c_hwdata imx21_i2c_hwdata  = {
202         .devtype                = IMX21_I2C,
203         .regshift               = IMX_I2C_REGSHIFT,
204         .clk_div                = imx_i2c_clk_div,
205         .ndivs                  = ARRAY_SIZE(imx_i2c_clk_div),
206         .i2sr_clr_opcode        = I2SR_CLR_OPCODE_W0C,
207         .i2cr_ien_opcode        = I2CR_IEN_OPCODE_1,
208
209 };
210
211 static struct imx_i2c_hwdata vf610_i2c_hwdata = {
212         .devtype                = VF610_I2C,
213         .regshift               = VF610_I2C_REGSHIFT,
214         .clk_div                = vf610_i2c_clk_div,
215         .ndivs                  = ARRAY_SIZE(vf610_i2c_clk_div),
216         .i2sr_clr_opcode        = I2SR_CLR_OPCODE_W1C,
217         .i2cr_ien_opcode        = I2CR_IEN_OPCODE_0,
218
219 };
220
221 static struct platform_device_id imx_i2c_devtype[] = {
222         {
223                 .name = "imx1-i2c",
224                 .driver_data = (kernel_ulong_t)&imx1_i2c_hwdata,
225         }, {
226                 .name = "imx21-i2c",
227                 .driver_data = (kernel_ulong_t)&imx21_i2c_hwdata,
228         }, {
229                 /* sentinel */
230         }
231 };
232 MODULE_DEVICE_TABLE(platform, imx_i2c_devtype);
233
234 static const struct of_device_id i2c_imx_dt_ids[] = {
235         { .compatible = "fsl,imx1-i2c", .data = &imx1_i2c_hwdata, },
236         { .compatible = "fsl,imx21-i2c", .data = &imx21_i2c_hwdata, },
237         { .compatible = "fsl,vf610-i2c", .data = &vf610_i2c_hwdata, },
238         { /* sentinel */ }
239 };
240 MODULE_DEVICE_TABLE(of, i2c_imx_dt_ids);
241
242 static inline int is_imx1_i2c(struct imx_i2c_struct *i2c_imx)
243 {
244         return i2c_imx->hwdata->devtype == IMX1_I2C;
245 }
246
247 static inline void imx_i2c_write_reg(unsigned int val,
248                 struct imx_i2c_struct *i2c_imx, unsigned int reg)
249 {
250         writeb(val, i2c_imx->base + (reg << i2c_imx->hwdata->regshift));
251 }
252
253 static inline unsigned char imx_i2c_read_reg(struct imx_i2c_struct *i2c_imx,
254                 unsigned int reg)
255 {
256         return readb(i2c_imx->base + (reg << i2c_imx->hwdata->regshift));
257 }
258
259 /** Functions for IMX I2C adapter driver ***************************************
260 *******************************************************************************/
261
262 static int i2c_imx_bus_busy(struct imx_i2c_struct *i2c_imx, int for_busy)
263 {
264         unsigned long orig_jiffies = jiffies;
265         unsigned int temp;
266
267         dev_dbg(&i2c_imx->adapter.dev, "<%s>\n", __func__);
268
269         while (1) {
270                 temp = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2SR);
271
272                 /* check for arbitration lost */
273                 if (temp & I2SR_IAL) {
274                         temp &= ~I2SR_IAL;
275                         imx_i2c_write_reg(temp, i2c_imx, IMX_I2C_I2SR);
276                         return -EAGAIN;
277                 }
278
279                 if (for_busy && (temp & I2SR_IBB))
280                         break;
281                 if (!for_busy && !(temp & I2SR_IBB))
282                         break;
283                 if (time_after(jiffies, orig_jiffies + msecs_to_jiffies(500))) {
284                         dev_dbg(&i2c_imx->adapter.dev,
285                                 "<%s> I2C bus is busy\n", __func__);
286                         return -ETIMEDOUT;
287                 }
288                 schedule();
289         }
290
291         return 0;
292 }
293
294 static int i2c_imx_trx_complete(struct imx_i2c_struct *i2c_imx)
295 {
296         wait_event_timeout(i2c_imx->queue, i2c_imx->i2csr & I2SR_IIF, HZ / 10);
297
298         if (unlikely(!(i2c_imx->i2csr & I2SR_IIF))) {
299                 dev_dbg(&i2c_imx->adapter.dev, "<%s> Timeout\n", __func__);
300                 return -ETIMEDOUT;
301         }
302         dev_dbg(&i2c_imx->adapter.dev, "<%s> TRX complete\n", __func__);
303         i2c_imx->i2csr = 0;
304         return 0;
305 }
306
307 static int i2c_imx_acked(struct imx_i2c_struct *i2c_imx)
308 {
309         if (imx_i2c_read_reg(i2c_imx, IMX_I2C_I2SR) & I2SR_RXAK) {
310                 dev_dbg(&i2c_imx->adapter.dev, "<%s> No ACK\n", __func__);
311                 return -EIO;  /* No ACK */
312         }
313
314         dev_dbg(&i2c_imx->adapter.dev, "<%s> ACK received\n", __func__);
315         return 0;
316 }
317
318 static void i2c_imx_set_clk(struct imx_i2c_struct *i2c_imx)
319 {
320         struct imx_i2c_clk_pair *i2c_clk_div = i2c_imx->hwdata->clk_div;
321         unsigned int i2c_clk_rate;
322         unsigned int div;
323         int i;
324
325         /* Divider value calculation */
326         i2c_clk_rate = clk_get_rate(i2c_imx->clk);
327         if (i2c_imx->cur_clk == i2c_clk_rate)
328                 return;
329         else
330                 i2c_imx->cur_clk = i2c_clk_rate;
331
332         div = (i2c_clk_rate + i2c_imx->bitrate - 1) / i2c_imx->bitrate;
333         if (div < i2c_clk_div[0].div)
334                 i = 0;
335         else if (div > i2c_clk_div[i2c_imx->hwdata->ndivs - 1].div)
336                 i = i2c_imx->hwdata->ndivs - 1;
337         else
338                 for (i = 0; i2c_clk_div[i].div < div; i++);
339
340         /* Store divider value */
341         i2c_imx->ifdr = i2c_clk_div[i].val;
342
343         /*
344          * There dummy delay is calculated.
345          * It should be about one I2C clock period long.
346          * This delay is used in I2C bus disable function
347          * to fix chip hardware bug.
348          */
349         i2c_imx->disable_delay = (500000U * i2c_clk_div[i].div
350                 + (i2c_clk_rate / 2) - 1) / (i2c_clk_rate / 2);
351
352 #ifdef CONFIG_I2C_DEBUG_BUS
353         dev_dbg(&i2c_imx->adapter.dev, "I2C_CLK=%d, REQ DIV=%d\n",
354                 i2c_clk_rate, div);
355         dev_dbg(&i2c_imx->adapter.dev, "IFDR[IC]=0x%x, REAL DIV=%d\n",
356                 i2c_clk_div[i].val, i2c_clk_div[i].div);
357 #endif
358 }
359
360 static int i2c_imx_start(struct imx_i2c_struct *i2c_imx)
361 {
362         unsigned int temp = 0;
363         int result;
364
365         dev_dbg(&i2c_imx->adapter.dev, "<%s>\n", __func__);
366
367         i2c_imx_set_clk(i2c_imx);
368
369         result = clk_prepare_enable(i2c_imx->clk);
370         if (result)
371                 return result;
372         imx_i2c_write_reg(i2c_imx->ifdr, i2c_imx, IMX_I2C_IFDR);
373         /* Enable I2C controller */
374         imx_i2c_write_reg(i2c_imx->hwdata->i2sr_clr_opcode, i2c_imx, IMX_I2C_I2SR);
375         imx_i2c_write_reg(i2c_imx->hwdata->i2cr_ien_opcode, i2c_imx, IMX_I2C_I2CR);
376
377         /* Wait controller to be stable */
378         udelay(50);
379
380         /* Start I2C transaction */
381         temp = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2CR);
382         temp |= I2CR_MSTA;
383         imx_i2c_write_reg(temp, i2c_imx, IMX_I2C_I2CR);
384         result = i2c_imx_bus_busy(i2c_imx, 1);
385         if (result)
386                 return result;
387         i2c_imx->stopped = 0;
388
389         temp |= I2CR_IIEN | I2CR_MTX | I2CR_TXAK;
390         imx_i2c_write_reg(temp, i2c_imx, IMX_I2C_I2CR);
391         return result;
392 }
393
394 static void i2c_imx_stop(struct imx_i2c_struct *i2c_imx)
395 {
396         unsigned int temp = 0;
397
398         if (!i2c_imx->stopped) {
399                 /* Stop I2C transaction */
400                 dev_dbg(&i2c_imx->adapter.dev, "<%s>\n", __func__);
401                 temp = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2CR);
402                 temp &= ~(I2CR_MSTA | I2CR_MTX);
403                 imx_i2c_write_reg(temp, i2c_imx, IMX_I2C_I2CR);
404         }
405         if (is_imx1_i2c(i2c_imx)) {
406                 /*
407                  * This delay caused by an i.MXL hardware bug.
408                  * If no (or too short) delay, no "STOP" bit will be generated.
409                  */
410                 udelay(i2c_imx->disable_delay);
411         }
412
413         if (!i2c_imx->stopped) {
414                 i2c_imx_bus_busy(i2c_imx, 0);
415                 i2c_imx->stopped = 1;
416         }
417
418         /* Disable I2C controller */
419         temp = i2c_imx->hwdata->i2cr_ien_opcode ^ I2CR_IEN,
420         imx_i2c_write_reg(temp, i2c_imx, IMX_I2C_I2CR);
421         clk_disable_unprepare(i2c_imx->clk);
422 }
423
424 static irqreturn_t i2c_imx_isr(int irq, void *dev_id)
425 {
426         struct imx_i2c_struct *i2c_imx = dev_id;
427         unsigned int temp;
428
429         temp = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2SR);
430         if (temp & I2SR_IIF) {
431                 /* save status register */
432                 i2c_imx->i2csr = temp;
433                 temp &= ~I2SR_IIF;
434                 temp |= (i2c_imx->hwdata->i2sr_clr_opcode & I2SR_IIF);
435                 imx_i2c_write_reg(temp, i2c_imx, IMX_I2C_I2SR);
436                 wake_up(&i2c_imx->queue);
437                 return IRQ_HANDLED;
438         }
439
440         return IRQ_NONE;
441 }
442
443 static int i2c_imx_write(struct imx_i2c_struct *i2c_imx, struct i2c_msg *msgs)
444 {
445         int i, result;
446
447         dev_dbg(&i2c_imx->adapter.dev, "<%s> write slave address: addr=0x%x\n",
448                 __func__, msgs->addr << 1);
449
450         /* write slave address */
451         imx_i2c_write_reg(msgs->addr << 1, i2c_imx, IMX_I2C_I2DR);
452         result = i2c_imx_trx_complete(i2c_imx);
453         if (result)
454                 return result;
455         result = i2c_imx_acked(i2c_imx);
456         if (result)
457                 return result;
458         dev_dbg(&i2c_imx->adapter.dev, "<%s> write data\n", __func__);
459
460         /* write data */
461         for (i = 0; i < msgs->len; i++) {
462                 dev_dbg(&i2c_imx->adapter.dev,
463                         "<%s> write byte: B%d=0x%X\n",
464                         __func__, i, msgs->buf[i]);
465                 imx_i2c_write_reg(msgs->buf[i], i2c_imx, IMX_I2C_I2DR);
466                 result = i2c_imx_trx_complete(i2c_imx);
467                 if (result)
468                         return result;
469                 result = i2c_imx_acked(i2c_imx);
470                 if (result)
471                         return result;
472         }
473         return 0;
474 }
475
476 static int i2c_imx_read(struct imx_i2c_struct *i2c_imx, struct i2c_msg *msgs, bool is_lastmsg)
477 {
478         int i, result;
479         unsigned int temp;
480         int block_data = msgs->flags & I2C_M_RECV_LEN;
481
482         dev_dbg(&i2c_imx->adapter.dev,
483                 "<%s> write slave address: addr=0x%x\n",
484                 __func__, (msgs->addr << 1) | 0x01);
485
486         /* write slave address */
487         imx_i2c_write_reg((msgs->addr << 1) | 0x01, i2c_imx, IMX_I2C_I2DR);
488         result = i2c_imx_trx_complete(i2c_imx);
489         if (result)
490                 return result;
491         result = i2c_imx_acked(i2c_imx);
492         if (result)
493                 return result;
494
495         dev_dbg(&i2c_imx->adapter.dev, "<%s> setup bus\n", __func__);
496
497         /* setup bus to read data */
498         temp = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2CR);
499         temp &= ~I2CR_MTX;
500
501         /*
502          * Reset the I2CR_TXAK flag initially for SMBus block read since the
503          * length is unknown
504          */
505         if ((msgs->len - 1) || block_data)
506                 temp &= ~I2CR_TXAK;
507         imx_i2c_write_reg(temp, i2c_imx, IMX_I2C_I2CR);
508         imx_i2c_read_reg(i2c_imx, IMX_I2C_I2DR); /* dummy read */
509
510         dev_dbg(&i2c_imx->adapter.dev, "<%s> read data\n", __func__);
511
512         /* read data */
513         for (i = 0; i < msgs->len; i++) {
514                 u8 len = 0;
515                 result = i2c_imx_trx_complete(i2c_imx);
516                 if (result)
517                         return result;
518                 /*
519                  * First byte is the length of remaining packet
520                  * in the SMBus block data read. Add it to
521                  * msgs->len.
522                  */
523                 if ((!i) && block_data) {
524                         len = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2DR);
525                         if ((len == 0) || (len > I2C_SMBUS_BLOCK_MAX))
526                                 return -EPROTO;
527                         dev_dbg(&i2c_imx->adapter.dev,
528                                 "<%s> read length: 0x%X\n",
529                                 __func__, len);
530                         msgs->len += len;
531                 }
532                 if (i == (msgs->len - 1)) {
533                         if (is_lastmsg) {
534                                 /*
535                                  * It must generate STOP before read I2DR to prevent
536                                  * controller from generating another clock cycle
537                                  */
538                                 dev_dbg(&i2c_imx->adapter.dev,
539                                         "<%s> clear MSTA\n", __func__);
540                                 temp = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2CR);
541                                 temp &= ~(I2CR_MSTA | I2CR_MTX);
542                                 imx_i2c_write_reg(temp, i2c_imx, IMX_I2C_I2CR);
543                                 i2c_imx_bus_busy(i2c_imx, 0);
544                                 i2c_imx->stopped = 1;
545                         } else {
546                                 /*
547                                  * For i2c master receiver repeat restart operation like:
548                                  * read -> repeat MSTA -> read/write
549                                  * The controller must set MTX before read the last byte in
550                                  * the first read operation, otherwise the first read cost
551                                  * one extra clock cycle.
552                                  */
553                                 temp = readb(i2c_imx->base + IMX_I2C_I2CR);
554                                 temp |= I2CR_MTX;
555                                 writeb(temp, i2c_imx->base + IMX_I2C_I2CR);
556                         }
557                 } else if (i == (msgs->len - 2)) {
558                         dev_dbg(&i2c_imx->adapter.dev,
559                                 "<%s> set TXAK\n", __func__);
560                         temp = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2CR);
561                         temp |= I2CR_TXAK;
562                         imx_i2c_write_reg(temp, i2c_imx, IMX_I2C_I2CR);
563                 }
564                 if ((!i) && block_data)
565                         msgs->buf[0] = len;
566                 else
567                         msgs->buf[i] =  imx_i2c_read_reg(i2c_imx, IMX_I2C_I2DR);
568                 dev_dbg(&i2c_imx->adapter.dev,
569                         "<%s> read byte: B%d=0x%X\n",
570                         __func__, i, msgs->buf[i]);
571         }
572         return 0;
573 }
574
575 static int i2c_imx_xfer(struct i2c_adapter *adapter,
576                                                 struct i2c_msg *msgs, int num)
577 {
578         unsigned int i, temp;
579         int result;
580         bool is_lastmsg = false;
581         struct imx_i2c_struct *i2c_imx = i2c_get_adapdata(adapter);
582
583         dev_dbg(&i2c_imx->adapter.dev, "<%s>\n", __func__);
584
585         /* Start I2C transfer */
586         result = i2c_imx_start(i2c_imx);
587         if (result)
588                 goto fail0;
589
590         /* read/write data */
591         for (i = 0; i < num; i++) {
592                 if (i == num - 1)
593                         is_lastmsg = true;
594
595                 if (i) {
596                         dev_dbg(&i2c_imx->adapter.dev,
597                                 "<%s> repeated start\n", __func__);
598                         temp = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2CR);
599                         temp |= I2CR_RSTA;
600                         imx_i2c_write_reg(temp, i2c_imx, IMX_I2C_I2CR);
601                         result =  i2c_imx_bus_busy(i2c_imx, 1);
602                         if (result)
603                                 goto fail0;
604                 }
605                 dev_dbg(&i2c_imx->adapter.dev,
606                         "<%s> transfer message: %d\n", __func__, i);
607                 /* write/read data */
608 #ifdef CONFIG_I2C_DEBUG_BUS
609                 temp = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2CR);
610                 dev_dbg(&i2c_imx->adapter.dev, "<%s> CONTROL: IEN=%d, IIEN=%d, "
611                         "MSTA=%d, MTX=%d, TXAK=%d, RSTA=%d\n", __func__,
612                         (temp & I2CR_IEN ? 1 : 0), (temp & I2CR_IIEN ? 1 : 0),
613                         (temp & I2CR_MSTA ? 1 : 0), (temp & I2CR_MTX ? 1 : 0),
614                         (temp & I2CR_TXAK ? 1 : 0), (temp & I2CR_RSTA ? 1 : 0));
615                 temp = imx_i2c_read_reg(i2c_imx, IMX_I2C_I2SR);
616                 dev_dbg(&i2c_imx->adapter.dev,
617                         "<%s> STATUS: ICF=%d, IAAS=%d, IBB=%d, "
618                         "IAL=%d, SRW=%d, IIF=%d, RXAK=%d\n", __func__,
619                         (temp & I2SR_ICF ? 1 : 0), (temp & I2SR_IAAS ? 1 : 0),
620                         (temp & I2SR_IBB ? 1 : 0), (temp & I2SR_IAL ? 1 : 0),
621                         (temp & I2SR_SRW ? 1 : 0), (temp & I2SR_IIF ? 1 : 0),
622                         (temp & I2SR_RXAK ? 1 : 0));
623 #endif
624                 if (msgs[i].flags & I2C_M_RD)
625                         result = i2c_imx_read(i2c_imx, &msgs[i], is_lastmsg);
626                 else
627                         result = i2c_imx_write(i2c_imx, &msgs[i]);
628                 if (result)
629                         goto fail0;
630         }
631
632 fail0:
633         /* Stop I2C transfer */
634         i2c_imx_stop(i2c_imx);
635
636         dev_dbg(&i2c_imx->adapter.dev, "<%s> exit with: %s: %d\n", __func__,
637                 (result < 0) ? "error" : "success msg",
638                         (result < 0) ? result : num);
639         return (result < 0) ? result : num;
640 }
641
642 static u32 i2c_imx_func(struct i2c_adapter *adapter)
643 {
644         return I2C_FUNC_I2C | I2C_FUNC_SMBUS_EMUL
645                 | I2C_FUNC_SMBUS_READ_BLOCK_DATA;
646 }
647
648 static struct i2c_algorithm i2c_imx_algo = {
649         .master_xfer    = i2c_imx_xfer,
650         .functionality  = i2c_imx_func,
651 };
652
653 static int i2c_imx_probe(struct platform_device *pdev)
654 {
655         const struct of_device_id *of_id = of_match_device(i2c_imx_dt_ids,
656                                                            &pdev->dev);
657         struct imx_i2c_struct *i2c_imx;
658         struct resource *res;
659         struct imxi2c_platform_data *pdata = dev_get_platdata(&pdev->dev);
660         void __iomem *base;
661         int irq, ret;
662
663         dev_dbg(&pdev->dev, "<%s>\n", __func__);
664
665         irq = platform_get_irq(pdev, 0);
666         if (irq < 0) {
667                 dev_err(&pdev->dev, "can't get irq number\n");
668                 return irq;
669         }
670
671         res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
672         base = devm_ioremap_resource(&pdev->dev, res);
673         if (IS_ERR(base))
674                 return PTR_ERR(base);
675
676         i2c_imx = devm_kzalloc(&pdev->dev, sizeof(struct imx_i2c_struct),
677                                 GFP_KERNEL);
678         if (!i2c_imx)
679                 return -ENOMEM;
680
681         if (of_id)
682                 i2c_imx->hwdata = of_id->data;
683         else
684                 i2c_imx->hwdata = (struct imx_i2c_hwdata *)
685                                 platform_get_device_id(pdev)->driver_data;
686
687         /* Setup i2c_imx driver structure */
688         strlcpy(i2c_imx->adapter.name, pdev->name, sizeof(i2c_imx->adapter.name));
689         i2c_imx->adapter.owner          = THIS_MODULE;
690         i2c_imx->adapter.algo           = &i2c_imx_algo;
691         i2c_imx->adapter.dev.parent     = &pdev->dev;
692         i2c_imx->adapter.nr             = pdev->id;
693         i2c_imx->adapter.dev.of_node    = pdev->dev.of_node;
694         i2c_imx->base                   = base;
695
696         /* Get I2C clock */
697         i2c_imx->clk = devm_clk_get(&pdev->dev, NULL);
698         if (IS_ERR(i2c_imx->clk)) {
699                 dev_err(&pdev->dev, "can't get I2C clock\n");
700                 return PTR_ERR(i2c_imx->clk);
701         }
702
703         ret = clk_prepare_enable(i2c_imx->clk);
704         if (ret) {
705                 dev_err(&pdev->dev, "can't enable I2C clock\n");
706                 return ret;
707         }
708         /* Request IRQ */
709         ret = devm_request_irq(&pdev->dev, irq, i2c_imx_isr, 0,
710                                 pdev->name, i2c_imx);
711         if (ret) {
712                 dev_err(&pdev->dev, "can't claim irq %d\n", irq);
713                 goto clk_disable;
714         }
715
716         /* Init queue */
717         init_waitqueue_head(&i2c_imx->queue);
718
719         /* Set up adapter data */
720         i2c_set_adapdata(&i2c_imx->adapter, i2c_imx);
721
722         /* Set up clock divider */
723         i2c_imx->bitrate = IMX_I2C_BIT_RATE;
724         ret = of_property_read_u32(pdev->dev.of_node,
725                                    "clock-frequency", &i2c_imx->bitrate);
726         if (ret < 0 && pdata && pdata->bitrate)
727                 i2c_imx->bitrate = pdata->bitrate;
728
729         /* Set up chip registers to defaults */
730         imx_i2c_write_reg(i2c_imx->hwdata->i2cr_ien_opcode ^ I2CR_IEN,
731                         i2c_imx, IMX_I2C_I2CR);
732         imx_i2c_write_reg(i2c_imx->hwdata->i2sr_clr_opcode, i2c_imx, IMX_I2C_I2SR);
733
734         /* Add I2C adapter */
735         ret = i2c_add_numbered_adapter(&i2c_imx->adapter);
736         if (ret < 0) {
737                 dev_err(&pdev->dev, "registration failed\n");
738                 goto clk_disable;
739         }
740
741         /* Set up platform driver data */
742         platform_set_drvdata(pdev, i2c_imx);
743         clk_disable_unprepare(i2c_imx->clk);
744
745         dev_dbg(&i2c_imx->adapter.dev, "claimed irq %d\n", irq);
746         dev_dbg(&i2c_imx->adapter.dev, "device resources: %pR\n", res);
747         dev_dbg(&i2c_imx->adapter.dev, "adapter name: \"%s\"\n",
748                 i2c_imx->adapter.name);
749         dev_info(&i2c_imx->adapter.dev, "IMX I2C adapter registered\n");
750
751         return 0;   /* Return OK */
752
753 clk_disable:
754         clk_disable_unprepare(i2c_imx->clk);
755         return ret;
756 }
757
758 static int i2c_imx_remove(struct platform_device *pdev)
759 {
760         struct imx_i2c_struct *i2c_imx = platform_get_drvdata(pdev);
761
762         /* remove adapter */
763         dev_dbg(&i2c_imx->adapter.dev, "adapter removed\n");
764         i2c_del_adapter(&i2c_imx->adapter);
765
766         /* setup chip registers to defaults */
767         imx_i2c_write_reg(0, i2c_imx, IMX_I2C_IADR);
768         imx_i2c_write_reg(0, i2c_imx, IMX_I2C_IFDR);
769         imx_i2c_write_reg(0, i2c_imx, IMX_I2C_I2CR);
770         imx_i2c_write_reg(0, i2c_imx, IMX_I2C_I2SR);
771
772         return 0;
773 }
774
775 static struct platform_driver i2c_imx_driver = {
776         .probe = i2c_imx_probe,
777         .remove = i2c_imx_remove,
778         .driver = {
779                 .name   = DRIVER_NAME,
780                 .owner  = THIS_MODULE,
781                 .of_match_table = i2c_imx_dt_ids,
782         },
783         .id_table       = imx_i2c_devtype,
784 };
785
786 static int __init i2c_adap_imx_init(void)
787 {
788         return platform_driver_register(&i2c_imx_driver);
789 }
790 subsys_initcall(i2c_adap_imx_init);
791
792 static void __exit i2c_adap_imx_exit(void)
793 {
794         platform_driver_unregister(&i2c_imx_driver);
795 }
796 module_exit(i2c_adap_imx_exit);
797
798 MODULE_LICENSE("GPL");
799 MODULE_AUTHOR("Darius Augulis");
800 MODULE_DESCRIPTION("I2C adapter driver for IMX I2C bus");
801 MODULE_ALIAS("platform:" DRIVER_NAME);