drivers/i2c/busses/i2c-at91-master.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  *  i2c Support for Atmel's AT91 Two-Wire Interface (TWI)
   4  *
   5  *  Copyright (C) 2011 Weinmann Medical GmbH
   6  *  Author: Nikolaus Voss <n.voss@weinmann.de>
   7  *
   8  *  Evolved from original work by:
   9  *  Copyright (C) 2004 Rick Bronson
  10  *  Converted to 2.6 by Andrew Victor <andrew@sanpeople.com>
  11  *
  12  *  Borrowed heavily from original work by:
  13  *  Copyright (C) 2000 Philip Edelbrock <phil@stimpy.netroedge.com>
  14  */
  15
  16 #include <linux/clk.h>
  17 #include <linux/completion.h>
  18 #include <linux/dma-mapping.h>
  19 #include <linux/dmaengine.h>
  20 #include <linux/err.h>
  21 #include <linux/i2c.h>
  22 #include <linux/interrupt.h>
  23 #include <linux/io.h>
  24 #include <linux/of.h>
  25 #include <linux/of_device.h>
  26 #include <linux/platform_device.h>
  27 #include <linux/platform_data/dma-atmel.h>
  28 #include <linux/pm_runtime.h>
  29
  30 #include "i2c-at91.h"
  31
  32 void at91_init_twi_bus_master(struct at91_twi_dev *dev)
  33 {
  34         /* FIFO should be enabled immediately after the software reset */
  35         if (dev->fifo_size)
  36                 at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_FIFOEN);
  37         at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_MSEN);
  38         at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_SVDIS);
  39         at91_twi_write(dev, AT91_TWI_CWGR, dev->twi_cwgr_reg);
  40 }
  41
  42 /*
  43  * Calculate symmetric clock as stated in datasheet:
  44  * twi_clk = F_MAIN / (2 * (cdiv * (1 << ckdiv) + offset))
  45  */
  46 static void at91_calc_twi_clock(struct at91_twi_dev *dev)
  47 {
  48         int ckdiv, cdiv, div, hold = 0;
  49         struct at91_twi_pdata *pdata = dev->pdata;
  50         int offset = pdata->clk_offset;
  51         int max_ckdiv = pdata->clk_max_div;
  52         struct i2c_timings timings, *t = &timings;
  53
  54         i2c_parse_fw_timings(dev->dev, t, true);
  55
  56         div = max(0, (int)DIV_ROUND_UP(clk_get_rate(dev->clk),
  57                                        2 * t->bus_freq_hz) - offset);
  58         ckdiv = fls(div >> 8);
  59         cdiv = div >> ckdiv;
  60
  61         if (ckdiv > max_ckdiv) {
  62                 dev_warn(dev->dev, "%d exceeds ckdiv max value which is %d.\n",
  63                          ckdiv, max_ckdiv);
  64                 ckdiv = max_ckdiv;
  65                 cdiv = 255;
  66         }
  67
  68         if (pdata->has_hold_field) {
  69                 /*
  70                  * hold time = HOLD + 3 x T_peripheral_clock
  71                  * Use clk rate in kHz to prevent overflows when computing
  72                  * hold.
  73                  */
  74                 hold = DIV_ROUND_UP(t->sda_hold_ns
  75                                     * (clk_get_rate(dev->clk) / 1000), 1000000);
  76                 hold -= 3;
  77                 if (hold < 0)
  78                         hold = 0;
  79                 if (hold > AT91_TWI_CWGR_HOLD_MAX) {
  80                         dev_warn(dev->dev,
  81                                  "HOLD field set to its maximum value (%d instead of %d)\n",
  82                                  AT91_TWI_CWGR_HOLD_MAX, hold);
  83                         hold = AT91_TWI_CWGR_HOLD_MAX;
  84                 }
  85         }
  86
  87         dev->twi_cwgr_reg = (ckdiv << 16) | (cdiv << 8) | cdiv
  88                             | AT91_TWI_CWGR_HOLD(hold);
  89
  90         dev_dbg(dev->dev, "cdiv %d ckdiv %d hold %d (%d ns)\n",
  91                 cdiv, ckdiv, hold, t->sda_hold_ns);
  92 }
  93
  94 static void at91_twi_dma_cleanup(struct at91_twi_dev *dev)
  95 {
  96         struct at91_twi_dma *dma = &dev->dma;
  97
  98         at91_twi_irq_save(dev);
  99
 100         if (dma->xfer_in_progress) {
 101                 if (dma->direction == DMA_FROM_DEVICE)
 102                         dmaengine_terminate_all(dma->chan_rx);
 103                 else
 104                         dmaengine_terminate_all(dma->chan_tx);
 105                 dma->xfer_in_progress = false;
 106         }
 107         if (dma->buf_mapped) {
 108                 dma_unmap_single(dev->dev, sg_dma_address(&dma->sg[0]),
 109                                  dev->buf_len, dma->direction);
 110                 dma->buf_mapped = false;
 111         }
 112
 113         at91_twi_irq_restore(dev);
 114 }
 115
 116 static void at91_twi_write_next_byte(struct at91_twi_dev *dev)
 117 {
 118         if (!dev->buf_len)
 119                 return;
 120
 121         /* 8bit write works with and without FIFO */
 122         writeb_relaxed(*dev->buf, dev->base + AT91_TWI_THR);
 123
 124         /* send stop when last byte has been written */
 125         if (--dev->buf_len == 0) {
 126                 if (!dev->use_alt_cmd)
 127                         at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_STOP);
 128                 at91_twi_write(dev, AT91_TWI_IDR, AT91_TWI_TXRDY);
 129         }
 130
 131         dev_dbg(dev->dev, "wrote 0x%x, to go %zu\n", *dev->buf, dev->buf_len);
 132
 133         ++dev->buf;
 134 }
 135
 136 static void at91_twi_write_data_dma_callback(void *data)
 137 {
 138         struct at91_twi_dev *dev = (struct at91_twi_dev *)data;
 139
 140         dma_unmap_single(dev->dev, sg_dma_address(&dev->dma.sg[0]),
 141                          dev->buf_len, DMA_TO_DEVICE);
 142
 143         /*
 144          * When this callback is called, THR/TX FIFO is likely not to be empty
 145          * yet. So we have to wait for TXCOMP or NACK bits to be set into the
 146          * Status Register to be sure that the STOP bit has been sent and the
 147          * transfer is completed. The NACK interrupt has already been enabled,
 148          * we just have to enable TXCOMP one.
 149          */
 150         at91_twi_write(dev, AT91_TWI_IER, AT91_TWI_TXCOMP);
 151         if (!dev->use_alt_cmd)
 152                 at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_STOP);
 153 }
 154
 155 static void at91_twi_write_data_dma(struct at91_twi_dev *dev)
 156 {
 157         dma_addr_t dma_addr;
 158         struct dma_async_tx_descriptor *txdesc;
 159         struct at91_twi_dma *dma = &dev->dma;
 160         struct dma_chan *chan_tx = dma->chan_tx;
 161         unsigned int sg_len = 1;
 162
 163         if (!dev->buf_len)
 164                 return;
 165
 166         dma->direction = DMA_TO_DEVICE;
 167
 168         at91_twi_irq_save(dev);
 169         dma_addr = dma_map_single(dev->dev, dev->buf, dev->buf_len,
 170                                   DMA_TO_DEVICE);
 171         if (dma_mapping_error(dev->dev, dma_addr)) {
 172                 dev_err(dev->dev, "dma map failed\n");
 173                 return;
 174         }
 175         dma->buf_mapped = true;
 176         at91_twi_irq_restore(dev);
 177
 178         if (dev->fifo_size) {
 179                 size_t part1_len, part2_len;
 180                 struct scatterlist *sg;
 181                 unsigned fifo_mr;
 182
 183                 sg_len = 0;
 184
 185                 part1_len = dev->buf_len & ~0x3;
 186                 if (part1_len) {
 187                         sg = &dma->sg[sg_len++];
 188                         sg_dma_len(sg) = part1_len;
 189                         sg_dma_address(sg) = dma_addr;
 190                 }
 191
 192                 part2_len = dev->buf_len & 0x3;
 193                 if (part2_len) {
 194                         sg = &dma->sg[sg_len++];
 195                         sg_dma_len(sg) = part2_len;
 196                         sg_dma_address(sg) = dma_addr + part1_len;
 197                 }
 198
 199                 /*
 200                  * DMA controller is triggered when at least 4 data can be
 201                  * written into the TX FIFO
 202                  */
 203                 fifo_mr = at91_twi_read(dev, AT91_TWI_FMR);
 204                 fifo_mr &= ~AT91_TWI_FMR_TXRDYM_MASK;
 205                 fifo_mr |= AT91_TWI_FMR_TXRDYM(AT91_TWI_FOUR_DATA);
 206                 at91_twi_write(dev, AT91_TWI_FMR, fifo_mr);
 207         } else {
 208                 sg_dma_len(&dma->sg[0]) = dev->buf_len;
 209                 sg_dma_address(&dma->sg[0]) = dma_addr;
 210         }
 211
 212         txdesc = dmaengine_prep_slave_sg(chan_tx, dma->sg, sg_len,
 213                                          DMA_MEM_TO_DEV,
 214                                          DMA_PREP_INTERRUPT | DMA_CTRL_ACK);
 215         if (!txdesc) {
 216                 dev_err(dev->dev, "dma prep slave sg failed\n");
 217                 goto error;
 218         }
 219
 220         txdesc->callback = at91_twi_write_data_dma_callback;
 221         txdesc->callback_param = dev;
 222
 223         dma->xfer_in_progress = true;
 224         dmaengine_submit(txdesc);
 225         dma_async_issue_pending(chan_tx);
 226
 227         return;
 228
 229 error:
 230         at91_twi_dma_cleanup(dev);
 231 }
 232
 233 static void at91_twi_read_next_byte(struct at91_twi_dev *dev)
 234 {
 235         /*
 236          * If we are in this case, it means there is garbage data in RHR, so
 237          * delete them.
 238          */
 239         if (!dev->buf_len) {
 240                 at91_twi_read(dev, AT91_TWI_RHR);
 241                 return;
 242         }
 243
 244         /* 8bit read works with and without FIFO */
 245         *dev->buf = readb_relaxed(dev->base + AT91_TWI_RHR);
 246         --dev->buf_len;
 247
 248         /* return if aborting, we only needed to read RHR to clear RXRDY*/
 249         if (dev->recv_len_abort)
 250                 return;
 251
 252         /* handle I2C_SMBUS_BLOCK_DATA */
 253         if (unlikely(dev->msg->flags & I2C_M_RECV_LEN)) {
 254                 /* ensure length byte is a valid value */
 255                 if (*dev->buf <= I2C_SMBUS_BLOCK_MAX && *dev->buf > 0) {
 256                         dev->msg->flags &= ~I2C_M_RECV_LEN;
 257                         dev->buf_len += *dev->buf;
 258                         dev->msg->len = dev->buf_len + 1;
 259                         dev_dbg(dev->dev, "received block length %zu\n",
 260                                          dev->buf_len);
 261                 } else {
 262                         /* abort and send the stop by reading one more byte */
 263                         dev->recv_len_abort = true;
 264                         dev->buf_len = 1;
 265                 }
 266         }
 267
 268         /* send stop if second but last byte has been read */
 269         if (!dev->use_alt_cmd && dev->buf_len == 1)
 270                 at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_STOP);
 271
 272         dev_dbg(dev->dev, "read 0x%x, to go %zu\n", *dev->buf, dev->buf_len);
 273
 274         ++dev->buf;
 275 }
 276
 277 static void at91_twi_read_data_dma_callback(void *data)
 278 {
 279         struct at91_twi_dev *dev = (struct at91_twi_dev *)data;
 280         unsigned ier = AT91_TWI_TXCOMP;
 281
 282         dma_unmap_single(dev->dev, sg_dma_address(&dev->dma.sg[0]),
 283                          dev->buf_len, DMA_FROM_DEVICE);
 284
 285         if (!dev->use_alt_cmd) {
 286                 /* The last two bytes have to be read without using dma */
 287                 dev->buf += dev->buf_len - 2;
 288                 dev->buf_len = 2;
 289                 ier |= AT91_TWI_RXRDY;
 290         }
 291         at91_twi_write(dev, AT91_TWI_IER, ier);
 292 }
 293
 294 static void at91_twi_read_data_dma(struct at91_twi_dev *dev)
 295 {
 296         dma_addr_t dma_addr;
 297         struct dma_async_tx_descriptor *rxdesc;
 298         struct at91_twi_dma *dma = &dev->dma;
 299         struct dma_chan *chan_rx = dma->chan_rx;
 300         size_t buf_len;
 301
 302         buf_len = (dev->use_alt_cmd) ? dev->buf_len : dev->buf_len - 2;
 303         dma->direction = DMA_FROM_DEVICE;
 304
 305         /* Keep in mind that we won't use dma to read the last two bytes */
 306         at91_twi_irq_save(dev);
 307         dma_addr = dma_map_single(dev->dev, dev->buf, buf_len, DMA_FROM_DEVICE);
 308         if (dma_mapping_error(dev->dev, dma_addr)) {
 309                 dev_err(dev->dev, "dma map failed\n");
 310                 return;
 311         }
 312         dma->buf_mapped = true;
 313         at91_twi_irq_restore(dev);
 314
 315         if (dev->fifo_size && IS_ALIGNED(buf_len, 4)) {
 316                 unsigned fifo_mr;
 317
 318                 /*
 319                  * DMA controller is triggered when at least 4 data can be
 320                  * read from the RX FIFO
 321                  */
 322                 fifo_mr = at91_twi_read(dev, AT91_TWI_FMR);
 323                 fifo_mr &= ~AT91_TWI_FMR_RXRDYM_MASK;
 324                 fifo_mr |= AT91_TWI_FMR_RXRDYM(AT91_TWI_FOUR_DATA);
 325                 at91_twi_write(dev, AT91_TWI_FMR, fifo_mr);
 326         }
 327
 328         sg_dma_len(&dma->sg[0]) = buf_len;
 329         sg_dma_address(&dma->sg[0]) = dma_addr;
 330
 331         rxdesc = dmaengine_prep_slave_sg(chan_rx, dma->sg, 1, DMA_DEV_TO_MEM,
 332                                          DMA_PREP_INTERRUPT | DMA_CTRL_ACK);
 333         if (!rxdesc) {
 334                 dev_err(dev->dev, "dma prep slave sg failed\n");
 335                 goto error;
 336         }
 337
 338         rxdesc->callback = at91_twi_read_data_dma_callback;
 339         rxdesc->callback_param = dev;
 340
 341         dma->xfer_in_progress = true;
 342         dmaengine_submit(rxdesc);
 343         dma_async_issue_pending(dma->chan_rx);
 344
 345         return;
 346
 347 error:
 348         at91_twi_dma_cleanup(dev);
 349 }
 350
 351 static irqreturn_t atmel_twi_interrupt(int irq, void *dev_id)
 352 {
 353         struct at91_twi_dev *dev = dev_id;
 354         const unsigned status = at91_twi_read(dev, AT91_TWI_SR);
 355         const unsigned irqstatus = status & at91_twi_read(dev, AT91_TWI_IMR);
 356
 357         if (!irqstatus)
 358                 return IRQ_NONE;
 359         /*
 360          * In reception, the behavior of the twi device (before sama5d2) is
 361          * weird. There is some magic about RXRDY flag! When a data has been
 362          * almost received, the reception of a new one is anticipated if there
 363          * is no stop command to send. That is the reason why ask for sending
 364          * the stop command not on the last data but on the second last one.
 365          *
 366          * Unfortunately, we could still have the RXRDY flag set even if the
 367          * transfer is done and we have read the last data. It might happen
 368          * when the i2c slave device sends too quickly data after receiving the
 369          * ack from the master. The data has been almost received before having
 370          * the order to send stop. In this case, sending the stop command could
 371          * cause a RXRDY interrupt with a TXCOMP one. It is better to manage
 372          * the RXRDY interrupt first in order to not keep garbage data in the
 373          * Receive Holding Register for the next transfer.
 374          */
 375         if (irqstatus & AT91_TWI_RXRDY) {
 376                 /*
 377                  * Read all available bytes at once by polling RXRDY usable w/
 378                  * and w/o FIFO. With FIFO enabled we could also read RXFL and
 379                  * avoid polling RXRDY.
 380                  */
 381                 do {
 382                         at91_twi_read_next_byte(dev);
 383                 } while (at91_twi_read(dev, AT91_TWI_SR) & AT91_TWI_RXRDY);
 384         }
 385
 386         /*
 387          * When a NACK condition is detected, the I2C controller sets the NACK,
 388          * TXCOMP and TXRDY bits all together in the Status Register (SR).
 389          *
 390          * 1 - Handling NACK errors with CPU write transfer.
 391          *
 392          * In such case, we should not write the next byte into the Transmit
 393          * Holding Register (THR) otherwise the I2C controller would start a new
 394          * transfer and the I2C slave is likely to reply by another NACK.
 395          *
 396          * 2 - Handling NACK errors with DMA write transfer.
 397          *
 398          * By setting the TXRDY bit in the SR, the I2C controller also triggers
 399          * the DMA controller to write the next data into the THR. Then the
 400          * result depends on the hardware version of the I2C controller.
 401          *
 402          * 2a - Without support of the Alternative Command mode.
 403          *
 404          * This is the worst case: the DMA controller is triggered to write the
 405          * next data into the THR, hence starting a new transfer: the I2C slave
 406          * is likely to reply by another NACK.
 407          * Concurrently, this interrupt handler is likely to be called to manage
 408          * the first NACK before the I2C controller detects the second NACK and
 409          * sets once again the NACK bit into the SR.
 410          * When handling the first NACK, this interrupt handler disables the I2C
 411          * controller interruptions, especially the NACK interrupt.
 412          * Hence, the NACK bit is pending into the SR. This is why we should
 413          * read the SR to clear all pending interrupts at the beginning of
 414          * at91_do_twi_transfer() before actually starting a new transfer.
 415          *
 416          * 2b - With support of the Alternative Command mode.
 417          *
 418          * When a NACK condition is detected, the I2C controller also locks the
 419          * THR (and sets the LOCK bit in the SR): even though the DMA controller
 420          * is triggered by the TXRDY bit to write the next data into the THR,
 421          * this data actually won't go on the I2C bus hence a second NACK is not
 422          * generated.
 423          */
 424         if (irqstatus & (AT91_TWI_TXCOMP | AT91_TWI_NACK)) {
 425                 at91_disable_twi_interrupts(dev);
 426                 complete(&dev->cmd_complete);
 427         } else if (irqstatus & AT91_TWI_TXRDY) {
 428                 at91_twi_write_next_byte(dev);
 429         }
 430
 431         /* catch error flags */
 432         dev->transfer_status |= status;
 433
 434         return IRQ_HANDLED;
 435 }
 436
 437 static int at91_do_twi_transfer(struct at91_twi_dev *dev)
 438 {
 439         int ret;
 440         unsigned long time_left;
 441         bool has_unre_flag = dev->pdata->has_unre_flag;
 442         bool has_alt_cmd = dev->pdata->has_alt_cmd;
 443
 444         /*
 445          * WARNING: the TXCOMP bit in the Status Register is NOT a clear on
 446          * read flag but shows the state of the transmission at the time the
 447          * Status Register is read. According to the programmer datasheet,
 448          * TXCOMP is set when both holding register and internal shifter are
 449          * empty and STOP condition has been sent.
 450          * Consequently, we should enable NACK interrupt rather than TXCOMP to
 451          * detect transmission failure.
 452          * Indeed let's take the case of an i2c write command using DMA.
 453          * Whenever the slave doesn't acknowledge a byte, the LOCK, NACK and
 454          * TXCOMP bits are set together into the Status Register.
 455          * LOCK is a clear on write bit, which is set to prevent the DMA
 456          * controller from sending new data on the i2c bus after a NACK
 457          * condition has happened. Once locked, this i2c peripheral stops
 458          * triggering the DMA controller for new data but it is more than
 459          * likely that a new DMA transaction is already in progress, writing
 460          * into the Transmit Holding Register. Since the peripheral is locked,
 461          * these new data won't be sent to the i2c bus but they will remain
 462          * into the Transmit Holding Register, so TXCOMP bit is cleared.
 463          * Then when the interrupt handler is called, the Status Register is
 464          * read: the TXCOMP bit is clear but NACK bit is still set. The driver
 465          * manage the error properly, without waiting for timeout.
 466          * This case can be reproduced easyly when writing into an at24 eeprom.
 467          *
 468          * Besides, the TXCOMP bit is already set before the i2c transaction
 469          * has been started. For read transactions, this bit is cleared when
 470          * writing the START bit into the Control Register. So the
 471          * corresponding interrupt can safely be enabled just after.
 472          * However for write transactions managed by the CPU, we first write
 473          * into THR, so TXCOMP is cleared. Then we can safely enable TXCOMP
 474          * interrupt. If TXCOMP interrupt were enabled before writing into THR,
 475          * the interrupt handler would be called immediately and the i2c command
 476          * would be reported as completed.
 477          * Also when a write transaction is managed by the DMA controller,
 478          * enabling the TXCOMP interrupt in this function may lead to a race
 479          * condition since we don't know whether the TXCOMP interrupt is enabled
 480          * before or after the DMA has started to write into THR. So the TXCOMP
 481          * interrupt is enabled later by at91_twi_write_data_dma_callback().
 482          * Immediately after in that DMA callback, if the alternative command
 483          * mode is not used, we still need to send the STOP condition manually
 484          * writing the corresponding bit into the Control Register.
 485          */
 486
 487         dev_dbg(dev->dev, "transfer: %s %zu bytes.\n",
 488                 (dev->msg->flags & I2C_M_RD) ? "read" : "write", dev->buf_len);
 489
 490         reinit_completion(&dev->cmd_complete);
 491         dev->transfer_status = 0;
 492
 493         /* Clear pending interrupts, such as NACK. */
 494         at91_twi_read(dev, AT91_TWI_SR);
 495
 496         if (dev->fifo_size) {
 497                 unsigned fifo_mr = at91_twi_read(dev, AT91_TWI_FMR);
 498
 499                 /* Reset FIFO mode register */
 500                 fifo_mr &= ~(AT91_TWI_FMR_TXRDYM_MASK |
 501                              AT91_TWI_FMR_RXRDYM_MASK);
 502                 fifo_mr |= AT91_TWI_FMR_TXRDYM(AT91_TWI_ONE_DATA);
 503                 fifo_mr |= AT91_TWI_FMR_RXRDYM(AT91_TWI_ONE_DATA);
 504                 at91_twi_write(dev, AT91_TWI_FMR, fifo_mr);
 505
 506                 /* Flush FIFOs */
 507                 at91_twi_write(dev, AT91_TWI_CR,
 508                                AT91_TWI_THRCLR | AT91_TWI_RHRCLR);
 509         }
 510
 511         if (!dev->buf_len) {
 512                 at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_QUICK);
 513                 at91_twi_write(dev, AT91_TWI_IER, AT91_TWI_TXCOMP);
 514         } else if (dev->msg->flags & I2C_M_RD) {
 515                 unsigned start_flags = AT91_TWI_START;
 516
 517                 /* if only one byte is to be read, immediately stop transfer */
 518                 if (!dev->use_alt_cmd && dev->buf_len <= 1 &&
 519                     !(dev->msg->flags & I2C_M_RECV_LEN))
 520                         start_flags |= AT91_TWI_STOP;
 521                 at91_twi_write(dev, AT91_TWI_CR, start_flags);
 522                 /*
 523                  * When using dma without alternative command mode, the last
 524                  * byte has to be read manually in order to not send the stop
 525                  * command too late and then to receive extra data.
 526                  * In practice, there are some issues if you use the dma to
 527                  * read n-1 bytes because of latency.
 528                  * Reading n-2 bytes with dma and the two last ones manually
 529                  * seems to be the best solution.
 530                  */
 531                 if (dev->use_dma && (dev->buf_len > AT91_I2C_DMA_THRESHOLD)) {
 532                         at91_twi_write(dev, AT91_TWI_IER, AT91_TWI_NACK);
 533                         at91_twi_read_data_dma(dev);
 534                 } else {
 535                         at91_twi_write(dev, AT91_TWI_IER,
 536                                        AT91_TWI_TXCOMP |
 537                                        AT91_TWI_NACK |
 538                                        AT91_TWI_RXRDY);
 539                 }
 540         } else {
 541                 if (dev->use_dma && (dev->buf_len > AT91_I2C_DMA_THRESHOLD)) {
 542                         at91_twi_write(dev, AT91_TWI_IER, AT91_TWI_NACK);
 543                         at91_twi_write_data_dma(dev);
 544                 } else {
 545                         at91_twi_write_next_byte(dev);
 546                         at91_twi_write(dev, AT91_TWI_IER,
 547                                        AT91_TWI_TXCOMP | AT91_TWI_NACK |
 548                                        (dev->buf_len ? AT91_TWI_TXRDY : 0));
 549                 }
 550         }
 551
 552         time_left = wait_for_completion_timeout(&dev->cmd_complete,
 553                                               dev->adapter.timeout);
 554         if (time_left == 0) {
 555                 dev->transfer_status |= at91_twi_read(dev, AT91_TWI_SR);
 556                 dev_err(dev->dev, "controller timed out\n");
 557                 at91_init_twi_bus(dev);
 558                 ret = -ETIMEDOUT;
 559                 goto error;
 560         }
 561         if (dev->transfer_status & AT91_TWI_NACK) {
 562                 dev_dbg(dev->dev, "received nack\n");
 563                 ret = -EREMOTEIO;
 564                 goto error;
 565         }
 566         if (dev->transfer_status & AT91_TWI_OVRE) {
 567                 dev_err(dev->dev, "overrun while reading\n");
 568                 ret = -EIO;
 569                 goto error;
 570         }
 571         if (has_unre_flag && dev->transfer_status & AT91_TWI_UNRE) {
 572                 dev_err(dev->dev, "underrun while writing\n");
 573                 ret = -EIO;
 574                 goto error;
 575         }
 576         if ((has_alt_cmd || dev->fifo_size) &&
 577             (dev->transfer_status & AT91_TWI_LOCK)) {
 578                 dev_err(dev->dev, "tx locked\n");
 579                 ret = -EIO;
 580                 goto error;
 581         }
 582         if (dev->recv_len_abort) {
 583                 dev_err(dev->dev, "invalid smbus block length recvd\n");
 584                 ret = -EPROTO;
 585                 goto error;
 586         }
 587
 588         dev_dbg(dev->dev, "transfer complete\n");
 589
 590         return 0;
 591
 592 error:
 593         /* first stop DMA transfer if still in progress */
 594         at91_twi_dma_cleanup(dev);
 595         /* then flush THR/FIFO and unlock TX if locked */
 596         if ((has_alt_cmd || dev->fifo_size) &&
 597             (dev->transfer_status & AT91_TWI_LOCK)) {
 598                 dev_dbg(dev->dev, "unlock tx\n");
 599                 at91_twi_write(dev, AT91_TWI_CR,
 600                                AT91_TWI_THRCLR | AT91_TWI_LOCKCLR);
 601         }
 602         return ret;
 603 }
 604
 605 static int at91_twi_xfer(struct i2c_adapter *adap, struct i2c_msg *msg, int num)
 606 {
 607         struct at91_twi_dev *dev = i2c_get_adapdata(adap);
 608         int ret;
 609         unsigned int_addr_flag = 0;
 610         struct i2c_msg *m_start = msg;
 611         bool is_read;
 612
 613         dev_dbg(&adap->dev, "at91_xfer: processing %d messages:\n", num);
 614
 615         ret = pm_runtime_get_sync(dev->dev);
 616         if (ret < 0)
 617                 goto out;
 618
 619         if (num == 2) {
 620                 int internal_address = 0;
 621                 int i;
 622
 623                 /* 1st msg is put into the internal address, start with 2nd */
 624                 m_start = &msg[1];
 625                 for (i = 0; i < msg->len; ++i) {
 626                         const unsigned addr = msg->buf[msg->len - 1 - i];
 627
 628                         internal_address |= addr << (8 * i);
 629                         int_addr_flag += AT91_TWI_IADRSZ_1;
 630                 }
 631                 at91_twi_write(dev, AT91_TWI_IADR, internal_address);
 632         }
 633
 634         dev->use_alt_cmd = false;
 635         is_read = (m_start->flags & I2C_M_RD);
 636         if (dev->pdata->has_alt_cmd) {
 637                 if (m_start->len > 0 &&
 638                     m_start->len < AT91_I2C_MAX_ALT_CMD_DATA_SIZE) {
 639                         at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_ACMEN);
 640                         at91_twi_write(dev, AT91_TWI_ACR,
 641                                        AT91_TWI_ACR_DATAL(m_start->len) |
 642                                        ((is_read) ? AT91_TWI_ACR_DIR : 0));
 643                         dev->use_alt_cmd = true;
 644                 } else {
 645                         at91_twi_write(dev, AT91_TWI_CR, AT91_TWI_ACMDIS);
 646                 }
 647         }
 648
 649         at91_twi_write(dev, AT91_TWI_MMR,
 650                        (m_start->addr << 16) |
 651                        int_addr_flag |
 652                        ((!dev->use_alt_cmd && is_read) ? AT91_TWI_MREAD : 0));
 653
 654         dev->buf_len = m_start->len;
 655         dev->buf = m_start->buf;
 656         dev->msg = m_start;
 657         dev->recv_len_abort = false;
 658
 659         ret = at91_do_twi_transfer(dev);
 660
 661         ret = (ret < 0) ? ret : num;
 662 out:
 663         pm_runtime_mark_last_busy(dev->dev);
 664         pm_runtime_put_autosuspend(dev->dev);
 665
 666         return ret;
 667 }
 668
 669 /*
 670  * The hardware can handle at most two messages concatenated by a
 671  * repeated start via it's internal address feature.
 672  */
 673 static const struct i2c_adapter_quirks at91_twi_quirks = {
 674         .flags = I2C_AQ_COMB | I2C_AQ_COMB_WRITE_FIRST | I2C_AQ_COMB_SAME_ADDR,
 675         .max_comb_1st_msg_len = 3,
 676 };
 677
 678 static u32 at91_twi_func(struct i2c_adapter *adapter)
 679 {
 680         return I2C_FUNC_I2C | I2C_FUNC_SMBUS_EMUL
 681                 | I2C_FUNC_SMBUS_READ_BLOCK_DATA;
 682 }
 683
 684 static const struct i2c_algorithm at91_twi_algorithm = {
 685         .master_xfer    = at91_twi_xfer,
 686         .functionality  = at91_twi_func,
 687 };
 688
 689 static int at91_twi_configure_dma(struct at91_twi_dev *dev, u32 phy_addr)
 690 {
 691         int ret = 0;
 692         struct dma_slave_config slave_config;
 693         struct at91_twi_dma *dma = &dev->dma;
 694         enum dma_slave_buswidth addr_width = DMA_SLAVE_BUSWIDTH_1_BYTE;
 695
 696         /*
 697          * The actual width of the access will be chosen in
 698          * dmaengine_prep_slave_sg():
 699          * for each buffer in the scatter-gather list, if its size is aligned
 700          * to addr_width then addr_width accesses will be performed to transfer
 701          * the buffer. On the other hand, if the buffer size is not aligned to
 702          * addr_width then the buffer is transferred using single byte accesses.
 703          * Please refer to the Atmel eXtended DMA controller driver.
 704          * When FIFOs are used, the TXRDYM threshold can always be set to
 705          * trigger the XDMAC when at least 4 data can be written into the TX
 706          * FIFO, even if single byte accesses are performed.
 707          * However the RXRDYM threshold must be set to fit the access width,
 708          * deduced from buffer length, so the XDMAC is triggered properly to
 709          * read data from the RX FIFO.
 710          */
 711         if (dev->fifo_size)
 712                 addr_width = DMA_SLAVE_BUSWIDTH_4_BYTES;
 713
 714         memset(&slave_config, 0, sizeof(slave_config));
 715         slave_config.src_addr = (dma_addr_t)phy_addr + AT91_TWI_RHR;
 716         slave_config.src_addr_width = addr_width;
 717         slave_config.src_maxburst = 1;
 718         slave_config.dst_addr = (dma_addr_t)phy_addr + AT91_TWI_THR;
 719         slave_config.dst_addr_width = addr_width;
 720         slave_config.dst_maxburst = 1;
 721         slave_config.device_fc = false;
 722
 723         dma->chan_tx = dma_request_slave_channel_reason(dev->dev, "tx");
 724         if (IS_ERR(dma->chan_tx)) {
 725                 ret = PTR_ERR(dma->chan_tx);
 726                 dma->chan_tx = NULL;
 727                 goto error;
 728         }
 729
 730         dma->chan_rx = dma_request_slave_channel_reason(dev->dev, "rx");
 731         if (IS_ERR(dma->chan_rx)) {
 732                 ret = PTR_ERR(dma->chan_rx);
 733                 dma->chan_rx = NULL;
 734                 goto error;
 735         }
 736
 737         slave_config.direction = DMA_MEM_TO_DEV;
 738         if (dmaengine_slave_config(dma->chan_tx, &slave_config)) {
 739                 dev_err(dev->dev, "failed to configure tx channel\n");
 740                 ret = -EINVAL;
 741                 goto error;
 742         }
 743
 744         slave_config.direction = DMA_DEV_TO_MEM;
 745         if (dmaengine_slave_config(dma->chan_rx, &slave_config)) {
 746                 dev_err(dev->dev, "failed to configure rx channel\n");
 747                 ret = -EINVAL;
 748                 goto error;
 749         }
 750
 751         sg_init_table(dma->sg, 2);
 752         dma->buf_mapped = false;
 753         dma->xfer_in_progress = false;
 754         dev->use_dma = true;
 755
 756         dev_info(dev->dev, "using %s (tx) and %s (rx) for DMA transfers\n",
 757                  dma_chan_name(dma->chan_tx), dma_chan_name(dma->chan_rx));
 758
 759         return ret;
 760
 761 error:
 762         if (ret != -EPROBE_DEFER)
 763                 dev_info(dev->dev, "can't get DMA channel, continue without DMA support\n");
 764         if (dma->chan_rx)
 765                 dma_release_channel(dma->chan_rx);
 766         if (dma->chan_tx)
 767                 dma_release_channel(dma->chan_tx);
 768         return ret;
 769 }
 770
 771 int at91_twi_probe_master(struct platform_device *pdev,
 772                           u32 phy_addr, struct at91_twi_dev *dev)
 773 {
 774         int rc;
 775
 776         init_completion(&dev->cmd_complete);
 777
 778         rc = devm_request_irq(&pdev->dev, dev->irq, atmel_twi_interrupt, 0,
 779                               dev_name(dev->dev), dev);
 780         if (rc) {
 781                 dev_err(dev->dev, "Cannot get irq %d: %d\n", dev->irq, rc);
 782                 return rc;
 783         }
 784
 785         if (dev->dev->of_node) {
 786                 rc = at91_twi_configure_dma(dev, phy_addr);
 787                 if (rc == -EPROBE_DEFER)
 788                         return rc;
 789         }
 790
 791         if (!of_property_read_u32(pdev->dev.of_node, "atmel,fifo-size",
 792                                   &dev->fifo_size)) {
 793                 dev_info(dev->dev, "Using FIFO (%u data)\n", dev->fifo_size);
 794         }
 795
 796         at91_calc_twi_clock(dev);
 797
 798         dev->adapter.algo = &at91_twi_algorithm;
 799         dev->adapter.quirks = &at91_twi_quirks;
 800
 801         return 0;
 802 }