Merge branch 'master' of git://git.kernel.org/pub/scm/linux/kernel/git/linville/wirel...
[pandora-kernel.git] / drivers / i2c / busses / i2c-at91.c
1 /*
2  *  i2c Support for Atmel's AT91 Two-Wire Interface (TWI)
3  *
4  *  Copyright (C) 2011 Weinmann Medical GmbH
5  *  Author: Nikolaus Voss <n.voss@weinmann.de>
6  *
7  *  Evolved from original work by:
8  *  Copyright (C) 2004 Rick Bronson
9  *  Converted to 2.6 by Andrew Victor <andrew@sanpeople.com>
10  *
11  *  Borrowed heavily from original work by:
12  *  Copyright (C) 2000 Philip Edelbrock <phil@stimpy.netroedge.com>
13  *
14  *  This program is free software; you can redistribute it and/or modify
15  *  it under the terms of the GNU General Public License as published by
16  *  the Free Software Foundation; either version 2 of the License, or
17  *  (at your option) any later version.
18  */
19
20 #include <linux/clk.h>
21 #include <linux/completion.h>
22 #include <linux/dma-mapping.h>
23 #include <linux/dmaengine.h>
24 #include <linux/err.h>
25 #include <linux/i2c.h>
26 #include <linux/interrupt.h>
27 #include <linux/io.h>
28 #include <linux/module.h>
29 #include <linux/of.h>
30 #include <linux/of_device.h>
31 #include <linux/of_i2c.h>
32 #include <linux/platform_device.h>
33 #include <linux/slab.h>
34 #include <linux/platform_data/dma-atmel.h>
35
36 #define TWI_CLK_HZ              100000                  /* max 400 Kbits/s */
37 #define AT91_I2C_TIMEOUT        msecs_to_jiffies(100)   /* transfer timeout */
38 #define AT91_I2C_DMA_THRESHOLD  8                       /* enable DMA if transfer size is bigger than this threshold */
39
40 /* AT91 TWI register definitions */
41 #define AT91_TWI_CR             0x0000  /* Control Register */
42 #define AT91_TWI_START          0x0001  /* Send a Start Condition */
43 #define AT91_TWI_STOP           0x0002  /* Send a Stop Condition */
44 #define AT91_TWI_MSEN           0x0004  /* Master Transfer Enable */
45 #define AT91_TWI_SVDIS          0x0020  /* Slave Transfer Disable */
46 #define AT91_TWI_QUICK          0x0040  /* SMBus quick command */
47 #define AT91_TWI_SWRST          0x0080  /* Software Reset */
48
49 #define AT91_TWI_MMR            0x0004  /* Master Mode Register */
50 #define AT91_TWI_IADRSZ_1       0x0100  /* Internal Device Address Size */
51 #define AT91_TWI_MREAD          0x1000  /* Master Read Direction */
52
53 #define AT91_TWI_IADR           0x000c  /* Internal Address Register */
54
55 #define AT91_TWI_CWGR           0x0010  /* Clock Waveform Generator Reg */
56
57 #define AT91_TWI_SR             0x0020  /* Status Register */
58 #define AT91_TWI_TXCOMP         0x0001  /* Transmission Complete */
59 #define AT91_TWI_RXRDY          0x0002  /* Receive Holding Register Ready */
60 #define AT91_TWI_TXRDY          0x0004  /* Transmit Holding Register Ready */
61
62 #define AT91_TWI_OVRE           0x0040  /* Overrun Error */
63 #define AT91_TWI_UNRE           0x0080  /* Underrun Error */
64 #define AT91_TWI_NACK           0x0100  /* Not Acknowledged */
65
66 #define AT91_TWI_IER            0x0024  /* Interrupt Enable Register */
67 #define AT91_TWI_IDR            0x0028  /* Interrupt Disable Register */
68 #define AT91_TWI_IMR            0x002c  /* Interrupt Mask Register */
69 #define AT91_TWI_RHR            0x0030  /* Receive Holding Register */
70 #define AT91_TWI_THR            0x0034  /* Transmit Holding Register */
71
72 struct at91_twi_pdata {
73         unsigned clk_max_div;
74         unsigned clk_offset;
75         bool has_unre_flag;
76         bool has_dma_support;
77         struct at_dma_slave dma_slave;
78 };
79
80 struct at91_twi_dma {
81         struct dma_chan *chan_rx;
82         struct dma_chan *chan_tx;
83         struct scatterlist sg;
84         struct dma_async_tx_descriptor *data_desc;
85         enum dma_data_direction direction;
86         bool buf_mapped;
87         bool xfer_in_progress;
88 };
89
90 struct at91_twi_dev {
91         struct device *dev;
92         void __iomem *base;
93         struct completion cmd_complete;
94         struct clk *clk;
95         u8 *buf;
96         size_t buf_len;
97         struct i2c_msg *msg;
98         int irq;
99         unsigned imr;
100         unsigned transfer_status;
101         struct i2c_adapter adapter;
102         unsigned twi_cwgr_reg;
103         struct at91_twi_pdata *pdata;
104         bool use_dma;
105         struct at91_twi_dma dma;
106 };
107
108 static unsigned at91_twi_read(struct at91_twi_dev *dev, unsigned reg)
109 {
110         return readl_relaxed(dev->base + reg);
111 }
112
113 static void at91_twi_write(struct at91_twi_dev *dev, unsigned reg, unsigned val)
114 {
115         writel_relaxed(val, dev->base + reg);
116 }
117
118 static void at91_disable_twi_interrupts(struct at91_twi_dev *dev)
119 {
120         at91_twi_write(dev, AT91_TWI_IDR,
121                        AT91_TWI_TXCOMP | AT91_TWI_RXRDY | AT91_TWI_TXRDY);
122 }
123
124 static void at91_twi_irq_save(struct at91_twi_dev *dev)
125 {
126         dev->imr = at91_twi_read(dev, AT91_TWI_IMR) & 0x7;
127         at91_disable_twi_interrupts(dev);
128 }
129
130 static void at91_twi_irq_restore(struct at91_twi_dev *dev)
131 {
132         at91_twi_write(dev, AT91_TWI_IER, dev->imr);
133 }
134
135 static void at91_init_twi_bus(struct at91_twi_dev *dev)
136 {
137         at91_disable_twi_interrupts(dev);
138         at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_SWRST);
139         at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_MSEN);
140         at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_SVDIS);
141         at91_twi_write(dev, AT91_TWI_CWGR, dev->twi_cwgr_reg);
142 }
143
144 /*
145  * Calculate symmetric clock as stated in datasheet:
146  * twi_clk = F_MAIN / (2 * (cdiv * (1 << ckdiv) + offset))
147  */
148 static void at91_calc_twi_clock(struct at91_twi_dev *dev, int twi_clk)
149 {
150         int ckdiv, cdiv, div;
151         struct at91_twi_pdata *pdata = dev->pdata;
152         int offset = pdata->clk_offset;
153         int max_ckdiv = pdata->clk_max_div;
154
155         div = max(0, (int)DIV_ROUND_UP(clk_get_rate(dev->clk),
156                                        2 * twi_clk) - offset);
157         ckdiv = fls(div >> 8);
158         cdiv = div >> ckdiv;
159
160         if (ckdiv > max_ckdiv) {
161                 dev_warn(dev->dev, "%d exceeds ckdiv max value which is %d.\n",
162                          ckdiv, max_ckdiv);
163                 ckdiv = max_ckdiv;
164                 cdiv = 255;
165         }
166
167         dev->twi_cwgr_reg = (ckdiv << 16) | (cdiv << 8) | cdiv;
168         dev_dbg(dev->dev, "cdiv %d ckdiv %d\n", cdiv, ckdiv);
169 }
170
171 static void at91_twi_dma_cleanup(struct at91_twi_dev *dev)
172 {
173         struct at91_twi_dma *dma = &dev->dma;
174
175         at91_twi_irq_save(dev);
176
177         if (dma->xfer_in_progress) {
178                 if (dma->direction == DMA_FROM_DEVICE)
179                         dmaengine_terminate_all(dma->chan_rx);
180                 else
181                         dmaengine_terminate_all(dma->chan_tx);
182                 dma->xfer_in_progress = false;
183         }
184         if (dma->buf_mapped) {
185                 dma_unmap_single(dev->dev, sg_dma_address(&dma->sg),
186                                  dev->buf_len, dma->direction);
187                 dma->buf_mapped = false;
188         }
189
190         at91_twi_irq_restore(dev);
191 }
192
193 static void at91_twi_write_next_byte(struct at91_twi_dev *dev)
194 {
195         if (dev->buf_len <= 0)
196                 return;
197
198         at91_twi_write(dev, AT91_TWI_THR, *dev->buf);
199
200         /* send stop when last byte has been written */
201         if (--dev->buf_len == 0)
202                 at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_STOP);
203
204         dev_dbg(dev->dev, "wrote 0x%x, to go %d\n", *dev->buf, dev->buf_len);
205
206         ++dev->buf;
207 }
208
209 static void at91_twi_write_data_dma_callback(void *data)
210 {
211         struct at91_twi_dev *dev = (struct at91_twi_dev *)data;
212
213         dma_unmap_single(dev->dev, sg_dma_address(&dev->dma.sg),
214                          dev->buf_len, DMA_MEM_TO_DEV);
215
216         at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_STOP);
217 }
218
219 static void at91_twi_write_data_dma(struct at91_twi_dev *dev)
220 {
221         dma_addr_t dma_addr;
222         struct dma_async_tx_descriptor *txdesc;
223         struct at91_twi_dma *dma = &dev->dma;
224         struct dma_chan *chan_tx = dma->chan_tx;
225
226         if (dev->buf_len <= 0)
227                 return;
228
229         dma->direction = DMA_TO_DEVICE;
230
231         at91_twi_irq_save(dev);
232         dma_addr = dma_map_single(dev->dev, dev->buf, dev->buf_len,
233                                   DMA_TO_DEVICE);
234         if (dma_mapping_error(dev->dev, dma_addr)) {
235                 dev_err(dev->dev, "dma map failed\n");
236                 return;
237         }
238         dma->buf_mapped = true;
239         at91_twi_irq_restore(dev);
240         sg_dma_len(&dma->sg) = dev->buf_len;
241         sg_dma_address(&dma->sg) = dma_addr;
242
243         txdesc = dmaengine_prep_slave_sg(chan_tx, &dma->sg, 1, DMA_MEM_TO_DEV,
244                                          DMA_PREP_INTERRUPT | DMA_CTRL_ACK);
245         if (!txdesc) {
246                 dev_err(dev->dev, "dma prep slave sg failed\n");
247                 goto error;
248         }
249
250         txdesc->callback = at91_twi_write_data_dma_callback;
251         txdesc->callback_param = dev;
252
253         dma->xfer_in_progress = true;
254         dmaengine_submit(txdesc);
255         dma_async_issue_pending(chan_tx);
256
257         return;
258
259 error:
260         at91_twi_dma_cleanup(dev);
261 }
262
263 static void at91_twi_read_next_byte(struct at91_twi_dev *dev)
264 {
265         if (dev->buf_len <= 0)
266                 return;
267
268         *dev->buf = at91_twi_read(dev, AT91_TWI_RHR) & 0xff;
269         --dev->buf_len;
270
271         /* handle I2C_SMBUS_BLOCK_DATA */
272         if (unlikely(dev->msg->flags & I2C_M_RECV_LEN)) {
273                 dev->msg->flags &= ~I2C_M_RECV_LEN;
274                 dev->buf_len += *dev->buf;
275                 dev->msg->len = dev->buf_len + 1;
276                 dev_dbg(dev->dev, "received block length %d\n", dev->buf_len);
277         }
278
279         /* send stop if second but last byte has been read */
280         if (dev->buf_len == 1)
281                 at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_STOP);
282
283         dev_dbg(dev->dev, "read 0x%x, to go %d\n", *dev->buf, dev->buf_len);
284
285         ++dev->buf;
286 }
287
288 static void at91_twi_read_data_dma_callback(void *data)
289 {
290         struct at91_twi_dev *dev = (struct at91_twi_dev *)data;
291
292         dma_unmap_single(dev->dev, sg_dma_address(&dev->dma.sg),
293                          dev->buf_len, DMA_DEV_TO_MEM);
294
295         /* The last two bytes have to be read without using dma */
296         dev->buf += dev->buf_len - 2;
297         dev->buf_len = 2;
298         at91_twi_write(dev, AT91_TWI_IER, AT91_TWI_RXRDY);
299 }
300
301 static void at91_twi_read_data_dma(struct at91_twi_dev *dev)
302 {
303         dma_addr_t dma_addr;
304         struct dma_async_tx_descriptor *rxdesc;
305         struct at91_twi_dma *dma = &dev->dma;
306         struct dma_chan *chan_rx = dma->chan_rx;
307
308         dma->direction = DMA_FROM_DEVICE;
309
310         /* Keep in mind that we won't use dma to read the last two bytes */
311         at91_twi_irq_save(dev);
312         dma_addr = dma_map_single(dev->dev, dev->buf, dev->buf_len - 2,
313                                   DMA_FROM_DEVICE);
314         if (dma_mapping_error(dev->dev, dma_addr)) {
315                 dev_err(dev->dev, "dma map failed\n");
316                 return;
317         }
318         dma->buf_mapped = true;
319         at91_twi_irq_restore(dev);
320         dma->sg.dma_address = dma_addr;
321         sg_dma_len(&dma->sg) = dev->buf_len - 2;
322
323         rxdesc = dmaengine_prep_slave_sg(chan_rx, &dma->sg, 1, DMA_DEV_TO_MEM,
324                                          DMA_PREP_INTERRUPT | DMA_CTRL_ACK);
325         if (!rxdesc) {
326                 dev_err(dev->dev, "dma prep slave sg failed\n");
327                 goto error;
328         }
329
330         rxdesc->callback = at91_twi_read_data_dma_callback;
331         rxdesc->callback_param = dev;
332
333         dma->xfer_in_progress = true;
334         dmaengine_submit(rxdesc);
335         dma_async_issue_pending(dma->chan_rx);
336
337         return;
338
339 error:
340         at91_twi_dma_cleanup(dev);
341 }
342
343 static irqreturn_t atmel_twi_interrupt(int irq, void *dev_id)
344 {
345         struct at91_twi_dev *dev = dev_id;
346         const unsigned status = at91_twi_read(dev, AT91_TWI_SR);
347         const unsigned irqstatus = status & at91_twi_read(dev, AT91_TWI_IMR);
348
349         if (!irqstatus)
350                 return IRQ_NONE;
351         else if (irqstatus & AT91_TWI_RXRDY)
352                 at91_twi_read_next_byte(dev);
353         else if (irqstatus & AT91_TWI_TXRDY)
354                 at91_twi_write_next_byte(dev);
355
356         /* catch error flags */
357         dev->transfer_status |= status;
358
359         if (irqstatus & AT91_TWI_TXCOMP) {
360                 at91_disable_twi_interrupts(dev);
361                 complete(&dev->cmd_complete);
362         }
363
364         return IRQ_HANDLED;
365 }
366
367 static int at91_do_twi_transfer(struct at91_twi_dev *dev)
368 {
369         int ret;
370         bool has_unre_flag = dev->pdata->has_unre_flag;
371
372         dev_dbg(dev->dev, "transfer: %s %d bytes.\n",
373                 (dev->msg->flags & I2C_M_RD) ? "read" : "write", dev->buf_len);
374
375         INIT_COMPLETION(dev->cmd_complete);
376         dev->transfer_status = 0;
377
378         if (!dev->buf_len) {
379                 at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_QUICK);
380                 at91_twi_write(dev, AT91_TWI_IER, AT91_TWI_TXCOMP);
381         } else if (dev->msg->flags & I2C_M_RD) {
382                 unsigned start_flags = AT91_TWI_START;
383
384                 if (at91_twi_read(dev, AT91_TWI_SR) & AT91_TWI_RXRDY) {
385                         dev_err(dev->dev, "RXRDY still set!");
386                         at91_twi_read(dev, AT91_TWI_RHR);
387                 }
388
389                 /* if only one byte is to be read, immediately stop transfer */
390                 if (dev->buf_len <= 1 && !(dev->msg->flags & I2C_M_RECV_LEN))
391                         start_flags |= AT91_TWI_STOP;
392                 at91_twi_write(dev, AT91_TWI_CR, start_flags);
393                 /*
394                  * When using dma, the last byte has to be read manually in
395                  * order to not send the stop command too late and then
396                  * to receive extra data. In practice, there are some issues
397                  * if you use the dma to read n-1 bytes because of latency.
398                  * Reading n-2 bytes with dma and the two last ones manually
399                  * seems to be the best solution.
400                  */
401                 if (dev->use_dma && (dev->buf_len > AT91_I2C_DMA_THRESHOLD)) {
402                         at91_twi_read_data_dma(dev);
403                         /*
404                          * It is important to enable TXCOMP irq here because
405                          * doing it only when transferring the last two bytes
406                          * will mask NACK errors since TXCOMP is set when a
407                          * NACK occurs.
408                          */
409                         at91_twi_write(dev, AT91_TWI_IER,
410                                AT91_TWI_TXCOMP);
411                 } else
412                         at91_twi_write(dev, AT91_TWI_IER,
413                                AT91_TWI_TXCOMP | AT91_TWI_RXRDY);
414         } else {
415                 if (dev->use_dma && (dev->buf_len > AT91_I2C_DMA_THRESHOLD)) {
416                         at91_twi_write_data_dma(dev);
417                         at91_twi_write(dev, AT91_TWI_IER, AT91_TWI_TXCOMP);
418                 } else {
419                         at91_twi_write_next_byte(dev);
420                         at91_twi_write(dev, AT91_TWI_IER,
421                                 AT91_TWI_TXCOMP | AT91_TWI_TXRDY);
422                 }
423         }
424
425         ret = wait_for_completion_interruptible_timeout(&dev->cmd_complete,
426                                                         dev->adapter.timeout);
427         if (ret == 0) {
428                 dev_err(dev->dev, "controller timed out\n");
429                 at91_init_twi_bus(dev);
430                 ret = -ETIMEDOUT;
431                 goto error;
432         }
433         if (dev->transfer_status & AT91_TWI_NACK) {
434                 dev_dbg(dev->dev, "received nack\n");
435                 ret = -EREMOTEIO;
436                 goto error;
437         }
438         if (dev->transfer_status & AT91_TWI_OVRE) {
439                 dev_err(dev->dev, "overrun while reading\n");
440                 ret = -EIO;
441                 goto error;
442         }
443         if (has_unre_flag && dev->transfer_status & AT91_TWI_UNRE) {
444                 dev_err(dev->dev, "underrun while writing\n");
445                 ret = -EIO;
446                 goto error;
447         }
448         dev_dbg(dev->dev, "transfer complete\n");
449
450         return 0;
451
452 error:
453         at91_twi_dma_cleanup(dev);
454         return ret;
455 }
456
457 static int at91_twi_xfer(struct i2c_adapter *adap, struct i2c_msg *msg, int num)
458 {
459         struct at91_twi_dev *dev = i2c_get_adapdata(adap);
460         int ret;
461         unsigned int_addr_flag = 0;
462         struct i2c_msg *m_start = msg;
463
464         dev_dbg(&adap->dev, "at91_xfer: processing %d messages:\n", num);
465
466         /*
467          * The hardware can handle at most two messages concatenated by a
468          * repeated start via it's internal address feature.
469          */
470         if (num > 2) {
471                 dev_err(dev->dev,
472                         "cannot handle more than two concatenated messages.\n");
473                 return 0;
474         } else if (num == 2) {
475                 int internal_address = 0;
476                 int i;
477
478                 if (msg->flags & I2C_M_RD) {
479                         dev_err(dev->dev, "first transfer must be write.\n");
480                         return -EINVAL;
481                 }
482                 if (msg->len > 3) {
483                         dev_err(dev->dev, "first message size must be <= 3.\n");
484                         return -EINVAL;
485                 }
486
487                 /* 1st msg is put into the internal address, start with 2nd */
488                 m_start = &msg[1];
489                 for (i = 0; i < msg->len; ++i) {
490                         const unsigned addr = msg->buf[msg->len - 1 - i];
491
492                         internal_address |= addr << (8 * i);
493                         int_addr_flag += AT91_TWI_IADRSZ_1;
494                 }
495                 at91_twi_write(dev, AT91_TWI_IADR, internal_address);
496         }
497
498         at91_twi_write(dev, AT91_TWI_MMR, (m_start->addr << 16) | int_addr_flag
499                        | ((m_start->flags & I2C_M_RD) ? AT91_TWI_MREAD : 0));
500
501         dev->buf_len = m_start->len;
502         dev->buf = m_start->buf;
503         dev->msg = m_start;
504
505         ret = at91_do_twi_transfer(dev);
506
507         return (ret < 0) ? ret : num;
508 }
509
510 static u32 at91_twi_func(struct i2c_adapter *adapter)
511 {
512         return I2C_FUNC_I2C | I2C_FUNC_SMBUS_EMUL
513                 | I2C_FUNC_SMBUS_READ_BLOCK_DATA;
514 }
515
516 static struct i2c_algorithm at91_twi_algorithm = {
517         .master_xfer    = at91_twi_xfer,
518         .functionality  = at91_twi_func,
519 };
520
521 static struct at91_twi_pdata at91rm9200_config = {
522         .clk_max_div = 5,
523         .clk_offset = 3,
524         .has_unre_flag = true,
525         .has_dma_support = false,
526 };
527
528 static struct at91_twi_pdata at91sam9261_config = {
529         .clk_max_div = 5,
530         .clk_offset = 4,
531         .has_unre_flag = false,
532         .has_dma_support = false,
533 };
534
535 static struct at91_twi_pdata at91sam9260_config = {
536         .clk_max_div = 7,
537         .clk_offset = 4,
538         .has_unre_flag = false,
539         .has_dma_support = false,
540 };
541
542 static struct at91_twi_pdata at91sam9g20_config = {
543         .clk_max_div = 7,
544         .clk_offset = 4,
545         .has_unre_flag = false,
546         .has_dma_support = false,
547 };
548
549 static struct at91_twi_pdata at91sam9g10_config = {
550         .clk_max_div = 7,
551         .clk_offset = 4,
552         .has_unre_flag = false,
553         .has_dma_support = false,
554 };
555
556 static struct at91_twi_pdata at91sam9x5_config = {
557         .clk_max_div = 7,
558         .clk_offset = 4,
559         .has_unre_flag = false,
560         .has_dma_support = true,
561 };
562
563 static const struct platform_device_id at91_twi_devtypes[] = {
564         {
565                 .name = "i2c-at91rm9200",
566                 .driver_data = (unsigned long) &at91rm9200_config,
567         }, {
568                 .name = "i2c-at91sam9261",
569                 .driver_data = (unsigned long) &at91sam9261_config,
570         }, {
571                 .name = "i2c-at91sam9260",
572                 .driver_data = (unsigned long) &at91sam9260_config,
573         }, {
574                 .name = "i2c-at91sam9g20",
575                 .driver_data = (unsigned long) &at91sam9g20_config,
576         }, {
577                 .name = "i2c-at91sam9g10",
578                 .driver_data = (unsigned long) &at91sam9g10_config,
579         }, {
580                 /* sentinel */
581         }
582 };
583
584 #if defined(CONFIG_OF)
585 static const struct of_device_id atmel_twi_dt_ids[] = {
586         {
587                 .compatible = "atmel,at91sam9260-i2c",
588                 .data = &at91sam9260_config,
589         } , {
590                 .compatible = "atmel,at91sam9g20-i2c",
591                 .data = &at91sam9g20_config,
592         } , {
593                 .compatible = "atmel,at91sam9g10-i2c",
594                 .data = &at91sam9g10_config,
595         }, {
596                 .compatible = "atmel,at91sam9x5-i2c",
597                 .data = &at91sam9x5_config,
598         }, {
599                 /* sentinel */
600         }
601 };
602 MODULE_DEVICE_TABLE(of, atmel_twi_dt_ids);
603 #else
604 #define atmel_twi_dt_ids NULL
605 #endif
606
607 static bool filter(struct dma_chan *chan, void *slave)
608 {
609         struct at_dma_slave *sl = slave;
610
611         if (sl->dma_dev == chan->device->dev) {
612                 chan->private = sl;
613                 return true;
614         } else {
615                 return false;
616         }
617 }
618
619 static int at91_twi_configure_dma(struct at91_twi_dev *dev, u32 phy_addr)
620 {
621         int ret = 0;
622         struct at_dma_slave *sdata;
623         struct dma_slave_config slave_config;
624         struct at91_twi_dma *dma = &dev->dma;
625
626         sdata = &dev->pdata->dma_slave;
627
628         memset(&slave_config, 0, sizeof(slave_config));
629         slave_config.src_addr = (dma_addr_t)phy_addr + AT91_TWI_RHR;
630         slave_config.src_addr_width = DMA_SLAVE_BUSWIDTH_1_BYTE;
631         slave_config.src_maxburst = 1;
632         slave_config.dst_addr = (dma_addr_t)phy_addr + AT91_TWI_THR;
633         slave_config.dst_addr_width = DMA_SLAVE_BUSWIDTH_1_BYTE;
634         slave_config.dst_maxburst = 1;
635         slave_config.device_fc = false;
636
637         if (sdata && sdata->dma_dev) {
638                 dma_cap_mask_t mask;
639
640                 dma_cap_zero(mask);
641                 dma_cap_set(DMA_SLAVE, mask);
642                 dma->chan_tx = dma_request_channel(mask, filter, sdata);
643                 if (!dma->chan_tx) {
644                         dev_err(dev->dev, "no DMA channel available for tx\n");
645                         ret = -EBUSY;
646                         goto error;
647                 }
648                 dma->chan_rx = dma_request_channel(mask, filter, sdata);
649                 if (!dma->chan_rx) {
650                         dev_err(dev->dev, "no DMA channel available for rx\n");
651                         ret = -EBUSY;
652                         goto error;
653                 }
654         } else {
655                 ret = -EINVAL;
656                 goto error;
657         }
658
659         slave_config.direction = DMA_MEM_TO_DEV;
660         if (dmaengine_slave_config(dma->chan_tx, &slave_config)) {
661                 dev_err(dev->dev, "failed to configure tx channel\n");
662                 ret = -EINVAL;
663                 goto error;
664         }
665
666         slave_config.direction = DMA_DEV_TO_MEM;
667         if (dmaengine_slave_config(dma->chan_rx, &slave_config)) {
668                 dev_err(dev->dev, "failed to configure rx channel\n");
669                 ret = -EINVAL;
670                 goto error;
671         }
672
673         sg_init_table(&dma->sg, 1);
674         dma->buf_mapped = false;
675         dma->xfer_in_progress = false;
676
677         dev_info(dev->dev, "using %s (tx) and %s (rx) for DMA transfers\n",
678                  dma_chan_name(dma->chan_tx), dma_chan_name(dma->chan_rx));
679
680         return ret;
681
682 error:
683         dev_info(dev->dev, "can't use DMA\n");
684         if (dma->chan_rx)
685                 dma_release_channel(dma->chan_rx);
686         if (dma->chan_tx)
687                 dma_release_channel(dma->chan_tx);
688         return ret;
689 }
690
691 static struct at91_twi_pdata *at91_twi_get_driver_data(
692                                         struct platform_device *pdev)
693 {
694         if (pdev->dev.of_node) {
695                 const struct of_device_id *match;
696                 match = of_match_node(atmel_twi_dt_ids, pdev->dev.of_node);
697                 if (!match)
698                         return NULL;
699                 return (struct at91_twi_pdata *)match->data;
700         }
701         return (struct at91_twi_pdata *) platform_get_device_id(pdev)->driver_data;
702 }
703
704 static int at91_twi_probe(struct platform_device *pdev)
705 {
706         struct at91_twi_dev *dev;
707         struct resource *mem;
708         int rc;
709         u32 phy_addr;
710
711         dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_KERNEL);
712         if (!dev)
713                 return -ENOMEM;
714         init_completion(&dev->cmd_complete);
715         dev->dev = &pdev->dev;
716
717         mem = platform_get_resource(pdev, IORESOURCE_MEM, 0);
718         if (!mem)
719                 return -ENODEV;
720         phy_addr = mem->start;
721
722         dev->pdata = at91_twi_get_driver_data(pdev);
723         if (!dev->pdata)
724                 return -ENODEV;
725
726         dev->base = devm_request_and_ioremap(&pdev->dev, mem);
727         if (!dev->base)
728                 return -EBUSY;
729
730         dev->irq = platform_get_irq(pdev, 0);
731         if (dev->irq < 0)
732                 return dev->irq;
733
734         rc = devm_request_irq(&pdev->dev, dev->irq, atmel_twi_interrupt, 0,
735                          dev_name(dev->dev), dev);
736         if (rc) {
737                 dev_err(dev->dev, "Cannot get irq %d: %d\n", dev->irq, rc);
738                 return rc;
739         }
740
741         platform_set_drvdata(pdev, dev);
742
743         dev->clk = devm_clk_get(dev->dev, NULL);
744         if (IS_ERR(dev->clk)) {
745                 dev_err(dev->dev, "no clock defined\n");
746                 return -ENODEV;
747         }
748         clk_prepare_enable(dev->clk);
749
750         if (dev->pdata->has_dma_support) {
751                 if (at91_twi_configure_dma(dev, phy_addr) == 0)
752                         dev->use_dma = true;
753         }
754
755         at91_calc_twi_clock(dev, TWI_CLK_HZ);
756         at91_init_twi_bus(dev);
757
758         snprintf(dev->adapter.name, sizeof(dev->adapter.name), "AT91");
759         i2c_set_adapdata(&dev->adapter, dev);
760         dev->adapter.owner = THIS_MODULE;
761         dev->adapter.class = I2C_CLASS_HWMON;
762         dev->adapter.algo = &at91_twi_algorithm;
763         dev->adapter.dev.parent = dev->dev;
764         dev->adapter.nr = pdev->id;
765         dev->adapter.timeout = AT91_I2C_TIMEOUT;
766         dev->adapter.dev.of_node = pdev->dev.of_node;
767
768         rc = i2c_add_numbered_adapter(&dev->adapter);
769         if (rc) {
770                 dev_err(dev->dev, "Adapter %s registration failed\n",
771                         dev->adapter.name);
772                 clk_disable_unprepare(dev->clk);
773                 return rc;
774         }
775
776         of_i2c_register_devices(&dev->adapter);
777
778         dev_info(dev->dev, "AT91 i2c bus driver.\n");
779         return 0;
780 }
781
782 static int at91_twi_remove(struct platform_device *pdev)
783 {
784         struct at91_twi_dev *dev = platform_get_drvdata(pdev);
785         int rc;
786
787         rc = i2c_del_adapter(&dev->adapter);
788         clk_disable_unprepare(dev->clk);
789
790         return rc;
791 }
792
793 #ifdef CONFIG_PM
794
795 static int at91_twi_runtime_suspend(struct device *dev)
796 {
797         struct at91_twi_dev *twi_dev = dev_get_drvdata(dev);
798
799         clk_disable(twi_dev->clk);
800
801         return 0;
802 }
803
804 static int at91_twi_runtime_resume(struct device *dev)
805 {
806         struct at91_twi_dev *twi_dev = dev_get_drvdata(dev);
807
808         return clk_enable(twi_dev->clk);
809 }
810
811 static const struct dev_pm_ops at91_twi_pm = {
812         .runtime_suspend        = at91_twi_runtime_suspend,
813         .runtime_resume         = at91_twi_runtime_resume,
814 };
815
816 #define at91_twi_pm_ops (&at91_twi_pm)
817 #else
818 #define at91_twi_pm_ops NULL
819 #endif
820
821 static struct platform_driver at91_twi_driver = {
822         .probe          = at91_twi_probe,
823         .remove         = at91_twi_remove,
824         .id_table       = at91_twi_devtypes,
825         .driver         = {
826                 .name   = "at91_i2c",
827                 .owner  = THIS_MODULE,
828                 .of_match_table = atmel_twi_dt_ids,
829                 .pm     = at91_twi_pm_ops,
830         },
831 };
832
833 static int __init at91_twi_init(void)
834 {
835         return platform_driver_register(&at91_twi_driver);
836 }
837
838 static void __exit at91_twi_exit(void)
839 {
840         platform_driver_unregister(&at91_twi_driver);
841 }
842
843 subsys_initcall(at91_twi_init);
844 module_exit(at91_twi_exit);
845
846 MODULE_AUTHOR("Nikolaus Voss <n.voss@weinmann.de>");
847 MODULE_DESCRIPTION("I2C (TWI) driver for Atmel AT91");
848 MODULE_LICENSE("GPL");
849 MODULE_ALIAS("platform:at91_i2c");