drivers/spi/spi-mt65xx.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Copyright (c) 2015 MediaTek Inc.
   4  * Author: Leilk Liu <leilk.liu@mediatek.com>
   5  */
   6
   7 #include <linux/clk.h>
   8 #include <linux/device.h>
   9 #include <linux/err.h>
  10 #include <linux/interrupt.h>
  11 #include <linux/io.h>
  12 #include <linux/ioport.h>
  13 #include <linux/module.h>
  14 #include <linux/of.h>
  15 #include <linux/of_gpio.h>
  16 #include <linux/platform_device.h>
  17 #include <linux/platform_data/spi-mt65xx.h>
  18 #include <linux/pm_runtime.h>
  19 #include <linux/spi/spi.h>
  20 #include <linux/dma-mapping.h>
  21
  22 #define SPI_CFG0_REG                      0x0000
  23 #define SPI_CFG1_REG                      0x0004
  24 #define SPI_TX_SRC_REG                    0x0008
  25 #define SPI_RX_DST_REG                    0x000c
  26 #define SPI_TX_DATA_REG                   0x0010
  27 #define SPI_RX_DATA_REG                   0x0014
  28 #define SPI_CMD_REG                       0x0018
  29 #define SPI_STATUS0_REG                   0x001c
  30 #define SPI_PAD_SEL_REG                   0x0024
  31 #define SPI_CFG2_REG                      0x0028
  32 #define SPI_TX_SRC_REG_64                 0x002c
  33 #define SPI_RX_DST_REG_64                 0x0030
  34
  35 #define SPI_CFG0_SCK_HIGH_OFFSET          0
  36 #define SPI_CFG0_SCK_LOW_OFFSET           8
  37 #define SPI_CFG0_CS_HOLD_OFFSET           16
  38 #define SPI_CFG0_CS_SETUP_OFFSET          24
  39 #define SPI_ADJUST_CFG0_CS_HOLD_OFFSET    0
  40 #define SPI_ADJUST_CFG0_CS_SETUP_OFFSET   16
  41
  42 #define SPI_CFG1_CS_IDLE_OFFSET           0
  43 #define SPI_CFG1_PACKET_LOOP_OFFSET       8
  44 #define SPI_CFG1_PACKET_LENGTH_OFFSET     16
  45 #define SPI_CFG1_GET_TICK_DLY_OFFSET      30
  46
  47 #define SPI_CFG1_CS_IDLE_MASK             0xff
  48 #define SPI_CFG1_PACKET_LOOP_MASK         0xff00
  49 #define SPI_CFG1_PACKET_LENGTH_MASK       0x3ff0000
  50 #define SPI_CFG2_SCK_HIGH_OFFSET          0
  51 #define SPI_CFG2_SCK_LOW_OFFSET           16
  52
  53 #define SPI_CMD_ACT                  BIT(0)
  54 #define SPI_CMD_RESUME               BIT(1)
  55 #define SPI_CMD_RST                  BIT(2)
  56 #define SPI_CMD_PAUSE_EN             BIT(4)
  57 #define SPI_CMD_DEASSERT             BIT(5)
  58 #define SPI_CMD_SAMPLE_SEL           BIT(6)
  59 #define SPI_CMD_CS_POL               BIT(7)
  60 #define SPI_CMD_CPHA                 BIT(8)
  61 #define SPI_CMD_CPOL                 BIT(9)
  62 #define SPI_CMD_RX_DMA               BIT(10)
  63 #define SPI_CMD_TX_DMA               BIT(11)
  64 #define SPI_CMD_TXMSBF               BIT(12)
  65 #define SPI_CMD_RXMSBF               BIT(13)
  66 #define SPI_CMD_RX_ENDIAN            BIT(14)
  67 #define SPI_CMD_TX_ENDIAN            BIT(15)
  68 #define SPI_CMD_FINISH_IE            BIT(16)
  69 #define SPI_CMD_PAUSE_IE             BIT(17)
  70
  71 #define MT8173_SPI_MAX_PAD_SEL 3
  72
  73 #define MTK_SPI_PAUSE_INT_STATUS 0x2
  74
  75 #define MTK_SPI_IDLE 0
  76 #define MTK_SPI_PAUSED 1
  77
  78 #define MTK_SPI_MAX_FIFO_SIZE 32U
  79 #define MTK_SPI_PACKET_SIZE 1024
  80 #define MTK_SPI_32BITS_MASK  (0xffffffff)
  81
  82 #define DMA_ADDR_EXT_BITS (36)
  83 #define DMA_ADDR_DEF_BITS (32)
  84
  85 struct mtk_spi_compatible {
  86         bool need_pad_sel;
  87         /* Must explicitly send dummy Tx bytes to do Rx only transfer */
  88         bool must_tx;
  89         /* some IC design adjust cfg register to enhance time accuracy */
  90         bool enhance_timing;
  91         /* some IC support DMA addr extension */
  92         bool dma_ext;
  93 };
  94
  95 struct mtk_spi {
  96         void __iomem *base;
  97         u32 state;
  98         int pad_num;
  99         u32 *pad_sel;
 100         struct clk *parent_clk, *sel_clk, *spi_clk;
 101         struct spi_transfer *cur_transfer;
 102         u32 xfer_len;
 103         u32 num_xfered;
 104         struct scatterlist *tx_sgl, *rx_sgl;
 105         u32 tx_sgl_len, rx_sgl_len;
 106         const struct mtk_spi_compatible *dev_comp;
 107 };
 108
 109 static const struct mtk_spi_compatible mtk_common_compat;
 110
 111 static const struct mtk_spi_compatible mt2712_compat = {
 112         .must_tx = true,
 113 };
 114
 115 static const struct mtk_spi_compatible mt6765_compat = {
 116         .need_pad_sel = true,
 117         .must_tx = true,
 118         .enhance_timing = true,
 119         .dma_ext = true,
 120 };
 121
 122 static const struct mtk_spi_compatible mt7622_compat = {
 123         .must_tx = true,
 124         .enhance_timing = true,
 125 };
 126
 127 static const struct mtk_spi_compatible mt8173_compat = {
 128         .need_pad_sel = true,
 129         .must_tx = true,
 130 };
 131
 132 static const struct mtk_spi_compatible mt8183_compat = {
 133         .need_pad_sel = true,
 134         .must_tx = true,
 135         .enhance_timing = true,
 136 };
 137
 138 /*
 139  * A piece of default chip info unless the platform
 140  * supplies it.
 141  */
 142 static const struct mtk_chip_config mtk_default_chip_info = {
 143         .sample_sel = 0,
 144 };
 145
 146 static const struct of_device_id mtk_spi_of_match[] = {
 147         { .compatible = "mediatek,mt2701-spi",
 148                 .data = (void *)&mtk_common_compat,
 149         },
 150         { .compatible = "mediatek,mt2712-spi",
 151                 .data = (void *)&mt2712_compat,
 152         },
 153         { .compatible = "mediatek,mt6589-spi",
 154                 .data = (void *)&mtk_common_compat,
 155         },
 156         { .compatible = "mediatek,mt6765-spi",
 157                 .data = (void *)&mt6765_compat,
 158         },
 159         { .compatible = "mediatek,mt7622-spi",
 160                 .data = (void *)&mt7622_compat,
 161         },
 162         { .compatible = "mediatek,mt7629-spi",
 163                 .data = (void *)&mt7622_compat,
 164         },
 165         { .compatible = "mediatek,mt8135-spi",
 166                 .data = (void *)&mtk_common_compat,
 167         },
 168         { .compatible = "mediatek,mt8173-spi",
 169                 .data = (void *)&mt8173_compat,
 170         },
 171         { .compatible = "mediatek,mt8183-spi",
 172                 .data = (void *)&mt8183_compat,
 173         },
 174         { .compatible = "mediatek,mt8192-spi",
 175                 .data = (void *)&mt6765_compat,
 176         },
 177         {}
 178 };
 179 MODULE_DEVICE_TABLE(of, mtk_spi_of_match);
 180
 181 static void mtk_spi_reset(struct mtk_spi *mdata)
 182 {
 183         u32 reg_val;
 184
 185         /* set the software reset bit in SPI_CMD_REG. */
 186         reg_val = readl(mdata->base + SPI_CMD_REG);
 187         reg_val |= SPI_CMD_RST;
 188         writel(reg_val, mdata->base + SPI_CMD_REG);
 189
 190         reg_val = readl(mdata->base + SPI_CMD_REG);
 191         reg_val &= ~SPI_CMD_RST;
 192         writel(reg_val, mdata->base + SPI_CMD_REG);
 193 }
 194
 195 static int mtk_spi_prepare_message(struct spi_master *master,
 196                                    struct spi_message *msg)
 197 {
 198         u16 cpha, cpol;
 199         u32 reg_val;
 200         struct spi_device *spi = msg->spi;
 201         struct mtk_chip_config *chip_config = spi->controller_data;
 202         struct mtk_spi *mdata = spi_master_get_devdata(master);
 203
 204         cpha = spi->mode & SPI_CPHA ? 1 : 0;
 205         cpol = spi->mode & SPI_CPOL ? 1 : 0;
 206
 207         reg_val = readl(mdata->base + SPI_CMD_REG);
 208         if (cpha)
 209                 reg_val |= SPI_CMD_CPHA;
 210         else
 211                 reg_val &= ~SPI_CMD_CPHA;
 212         if (cpol)
 213                 reg_val |= SPI_CMD_CPOL;
 214         else
 215                 reg_val &= ~SPI_CMD_CPOL;
 216
 217         /* set the mlsbx and mlsbtx */
 218         if (spi->mode & SPI_LSB_FIRST) {
 219                 reg_val &= ~SPI_CMD_TXMSBF;
 220                 reg_val &= ~SPI_CMD_RXMSBF;
 221         } else {
 222                 reg_val |= SPI_CMD_TXMSBF;
 223                 reg_val |= SPI_CMD_RXMSBF;
 224         }
 225
 226         /* set the tx/rx endian */
 227 #ifdef __LITTLE_ENDIAN
 228         reg_val &= ~SPI_CMD_TX_ENDIAN;
 229         reg_val &= ~SPI_CMD_RX_ENDIAN;
 230 #else
 231         reg_val |= SPI_CMD_TX_ENDIAN;
 232         reg_val |= SPI_CMD_RX_ENDIAN;
 233 #endif
 234
 235         if (mdata->dev_comp->enhance_timing) {
 236                 /* set CS polarity */
 237                 if (spi->mode & SPI_CS_HIGH)
 238                         reg_val |= SPI_CMD_CS_POL;
 239                 else
 240                         reg_val &= ~SPI_CMD_CS_POL;
 241
 242                 if (chip_config->sample_sel)
 243                         reg_val |= SPI_CMD_SAMPLE_SEL;
 244                 else
 245                         reg_val &= ~SPI_CMD_SAMPLE_SEL;
 246         }
 247
 248         /* set finish and pause interrupt always enable */
 249         reg_val |= SPI_CMD_FINISH_IE | SPI_CMD_PAUSE_IE;
 250
 251         /* disable dma mode */
 252         reg_val &= ~(SPI_CMD_TX_DMA | SPI_CMD_RX_DMA);
 253
 254         /* disable deassert mode */
 255         reg_val &= ~SPI_CMD_DEASSERT;
 256
 257         writel(reg_val, mdata->base + SPI_CMD_REG);
 258
 259         /* pad select */
 260         if (mdata->dev_comp->need_pad_sel)
 261                 writel(mdata->pad_sel[spi->chip_select],
 262                        mdata->base + SPI_PAD_SEL_REG);
 263
 264         return 0;
 265 }
 266
 267 static void mtk_spi_set_cs(struct spi_device *spi, bool enable)
 268 {
 269         u32 reg_val;
 270         struct mtk_spi *mdata = spi_master_get_devdata(spi->master);
 271
 272         if (spi->mode & SPI_CS_HIGH)
 273                 enable = !enable;
 274
 275         reg_val = readl(mdata->base + SPI_CMD_REG);
 276         if (!enable) {
 277                 reg_val |= SPI_CMD_PAUSE_EN;
 278                 writel(reg_val, mdata->base + SPI_CMD_REG);
 279         } else {
 280                 reg_val &= ~SPI_CMD_PAUSE_EN;
 281                 writel(reg_val, mdata->base + SPI_CMD_REG);
 282                 mdata->state = MTK_SPI_IDLE;
 283                 mtk_spi_reset(mdata);
 284         }
 285 }
 286
 287 static void mtk_spi_prepare_transfer(struct spi_master *master,
 288                                      struct spi_transfer *xfer)
 289 {
 290         u32 spi_clk_hz, div, sck_time, cs_time, reg_val;
 291         struct mtk_spi *mdata = spi_master_get_devdata(master);
 292
 293         spi_clk_hz = clk_get_rate(mdata->spi_clk);
 294         if (xfer->speed_hz < spi_clk_hz / 2)
 295                 div = DIV_ROUND_UP(spi_clk_hz, xfer->speed_hz);
 296         else
 297                 div = 1;
 298
 299         sck_time = (div + 1) / 2;
 300         cs_time = sck_time * 2;
 301
 302         if (mdata->dev_comp->enhance_timing) {
 303                 reg_val = (((sck_time - 1) & 0xffff)
 304                            << SPI_CFG2_SCK_HIGH_OFFSET);
 305                 reg_val |= (((sck_time - 1) & 0xffff)
 306                            << SPI_CFG2_SCK_LOW_OFFSET);
 307                 writel(reg_val, mdata->base + SPI_CFG2_REG);
 308                 reg_val = (((cs_time - 1) & 0xffff)
 309                            << SPI_ADJUST_CFG0_CS_HOLD_OFFSET);
 310                 reg_val |= (((cs_time - 1) & 0xffff)
 311                            << SPI_ADJUST_CFG0_CS_SETUP_OFFSET);
 312                 writel(reg_val, mdata->base + SPI_CFG0_REG);
 313         } else {
 314                 reg_val = (((sck_time - 1) & 0xff)
 315                            << SPI_CFG0_SCK_HIGH_OFFSET);
 316                 reg_val |= (((sck_time - 1) & 0xff) << SPI_CFG0_SCK_LOW_OFFSET);
 317                 reg_val |= (((cs_time - 1) & 0xff) << SPI_CFG0_CS_HOLD_OFFSET);
 318                 reg_val |= (((cs_time - 1) & 0xff) << SPI_CFG0_CS_SETUP_OFFSET);
 319                 writel(reg_val, mdata->base + SPI_CFG0_REG);
 320         }
 321
 322         reg_val = readl(mdata->base + SPI_CFG1_REG);
 323         reg_val &= ~SPI_CFG1_CS_IDLE_MASK;
 324         reg_val |= (((cs_time - 1) & 0xff) << SPI_CFG1_CS_IDLE_OFFSET);
 325         writel(reg_val, mdata->base + SPI_CFG1_REG);
 326 }
 327
 328 static void mtk_spi_setup_packet(struct spi_master *master)
 329 {
 330         u32 packet_size, packet_loop, reg_val;
 331         struct mtk_spi *mdata = spi_master_get_devdata(master);
 332
 333         packet_size = min_t(u32, mdata->xfer_len, MTK_SPI_PACKET_SIZE);
 334         packet_loop = mdata->xfer_len / packet_size;
 335
 336         reg_val = readl(mdata->base + SPI_CFG1_REG);
 337         reg_val &= ~(SPI_CFG1_PACKET_LENGTH_MASK | SPI_CFG1_PACKET_LOOP_MASK);
 338         reg_val |= (packet_size - 1) << SPI_CFG1_PACKET_LENGTH_OFFSET;
 339         reg_val |= (packet_loop - 1) << SPI_CFG1_PACKET_LOOP_OFFSET;
 340         writel(reg_val, mdata->base + SPI_CFG1_REG);
 341 }
 342
 343 static void mtk_spi_enable_transfer(struct spi_master *master)
 344 {
 345         u32 cmd;
 346         struct mtk_spi *mdata = spi_master_get_devdata(master);
 347
 348         cmd = readl(mdata->base + SPI_CMD_REG);
 349         if (mdata->state == MTK_SPI_IDLE)
 350                 cmd |= SPI_CMD_ACT;
 351         else
 352                 cmd |= SPI_CMD_RESUME;
 353         writel(cmd, mdata->base + SPI_CMD_REG);
 354 }
 355
 356 static int mtk_spi_get_mult_delta(u32 xfer_len)
 357 {
 358         u32 mult_delta;
 359
 360         if (xfer_len > MTK_SPI_PACKET_SIZE)
 361                 mult_delta = xfer_len % MTK_SPI_PACKET_SIZE;
 362         else
 363                 mult_delta = 0;
 364
 365         return mult_delta;
 366 }
 367
 368 static void mtk_spi_update_mdata_len(struct spi_master *master)
 369 {
 370         int mult_delta;
 371         struct mtk_spi *mdata = spi_master_get_devdata(master);
 372
 373         if (mdata->tx_sgl_len && mdata->rx_sgl_len) {
 374                 if (mdata->tx_sgl_len > mdata->rx_sgl_len) {
 375                         mult_delta = mtk_spi_get_mult_delta(mdata->rx_sgl_len);
 376                         mdata->xfer_len = mdata->rx_sgl_len - mult_delta;
 377                         mdata->rx_sgl_len = mult_delta;
 378                         mdata->tx_sgl_len -= mdata->xfer_len;
 379                 } else {
 380                         mult_delta = mtk_spi_get_mult_delta(mdata->tx_sgl_len);
 381                         mdata->xfer_len = mdata->tx_sgl_len - mult_delta;
 382                         mdata->tx_sgl_len = mult_delta;
 383                         mdata->rx_sgl_len -= mdata->xfer_len;
 384                 }
 385         } else if (mdata->tx_sgl_len) {
 386                 mult_delta = mtk_spi_get_mult_delta(mdata->tx_sgl_len);
 387                 mdata->xfer_len = mdata->tx_sgl_len - mult_delta;
 388                 mdata->tx_sgl_len = mult_delta;
 389         } else if (mdata->rx_sgl_len) {
 390                 mult_delta = mtk_spi_get_mult_delta(mdata->rx_sgl_len);
 391                 mdata->xfer_len = mdata->rx_sgl_len - mult_delta;
 392                 mdata->rx_sgl_len = mult_delta;
 393         }
 394 }
 395
 396 static void mtk_spi_setup_dma_addr(struct spi_master *master,
 397                                    struct spi_transfer *xfer)
 398 {
 399         struct mtk_spi *mdata = spi_master_get_devdata(master);
 400
 401         if (mdata->tx_sgl) {
 402                 writel((u32)(xfer->tx_dma & MTK_SPI_32BITS_MASK),
 403                        mdata->base + SPI_TX_SRC_REG);
 404 #ifdef CONFIG_ARCH_DMA_ADDR_T_64BIT
 405                 if (mdata->dev_comp->dma_ext)
 406                         writel((u32)(xfer->tx_dma >> 32),
 407                                mdata->base + SPI_TX_SRC_REG_64);
 408 #endif
 409         }
 410
 411         if (mdata->rx_sgl) {
 412                 writel((u32)(xfer->rx_dma & MTK_SPI_32BITS_MASK),
 413                        mdata->base + SPI_RX_DST_REG);
 414 #ifdef CONFIG_ARCH_DMA_ADDR_T_64BIT
 415                 if (mdata->dev_comp->dma_ext)
 416                         writel((u32)(xfer->rx_dma >> 32),
 417                                mdata->base + SPI_RX_DST_REG_64);
 418 #endif
 419         }
 420 }
 421
 422 static int mtk_spi_fifo_transfer(struct spi_master *master,
 423                                  struct spi_device *spi,
 424                                  struct spi_transfer *xfer)
 425 {
 426         int cnt, remainder;
 427         u32 reg_val;
 428         struct mtk_spi *mdata = spi_master_get_devdata(master);
 429
 430         mdata->cur_transfer = xfer;
 431         mdata->xfer_len = min(MTK_SPI_MAX_FIFO_SIZE, xfer->len);
 432         mdata->num_xfered = 0;
 433         mtk_spi_prepare_transfer(master, xfer);
 434         mtk_spi_setup_packet(master);
 435
 436         if (xfer->tx_buf) {
 437                 cnt = xfer->len / 4;
 438                 iowrite32_rep(mdata->base + SPI_TX_DATA_REG, xfer->tx_buf, cnt);
 439                 remainder = xfer->len % 4;
 440                 if (remainder > 0) {
 441                         reg_val = 0;
 442                         memcpy(&reg_val, xfer->tx_buf + (cnt * 4), remainder);
 443                         writel(reg_val, mdata->base + SPI_TX_DATA_REG);
 444                 }
 445         }
 446
 447         mtk_spi_enable_transfer(master);
 448
 449         return 1;
 450 }
 451
 452 static int mtk_spi_dma_transfer(struct spi_master *master,
 453                                 struct spi_device *spi,
 454                                 struct spi_transfer *xfer)
 455 {
 456         int cmd;
 457         struct mtk_spi *mdata = spi_master_get_devdata(master);
 458
 459         mdata->tx_sgl = NULL;
 460         mdata->rx_sgl = NULL;
 461         mdata->tx_sgl_len = 0;
 462         mdata->rx_sgl_len = 0;
 463         mdata->cur_transfer = xfer;
 464         mdata->num_xfered = 0;
 465
 466         mtk_spi_prepare_transfer(master, xfer);
 467
 468         cmd = readl(mdata->base + SPI_CMD_REG);
 469         if (xfer->tx_buf)
 470                 cmd |= SPI_CMD_TX_DMA;
 471         if (xfer->rx_buf)
 472                 cmd |= SPI_CMD_RX_DMA;
 473         writel(cmd, mdata->base + SPI_CMD_REG);
 474
 475         if (xfer->tx_buf)
 476                 mdata->tx_sgl = xfer->tx_sg.sgl;
 477         if (xfer->rx_buf)
 478                 mdata->rx_sgl = xfer->rx_sg.sgl;
 479
 480         if (mdata->tx_sgl) {
 481                 xfer->tx_dma = sg_dma_address(mdata->tx_sgl);
 482                 mdata->tx_sgl_len = sg_dma_len(mdata->tx_sgl);
 483         }
 484         if (mdata->rx_sgl) {
 485                 xfer->rx_dma = sg_dma_address(mdata->rx_sgl);
 486                 mdata->rx_sgl_len = sg_dma_len(mdata->rx_sgl);
 487         }
 488
 489         mtk_spi_update_mdata_len(master);
 490         mtk_spi_setup_packet(master);
 491         mtk_spi_setup_dma_addr(master, xfer);
 492         mtk_spi_enable_transfer(master);
 493
 494         return 1;
 495 }
 496
 497 static int mtk_spi_transfer_one(struct spi_master *master,
 498                                 struct spi_device *spi,
 499                                 struct spi_transfer *xfer)
 500 {
 501         if (master->can_dma(master, spi, xfer))
 502                 return mtk_spi_dma_transfer(master, spi, xfer);
 503         else
 504                 return mtk_spi_fifo_transfer(master, spi, xfer);
 505 }
 506
 507 static bool mtk_spi_can_dma(struct spi_master *master,
 508                             struct spi_device *spi,
 509                             struct spi_transfer *xfer)
 510 {
 511         /* Buffers for DMA transactions must be 4-byte aligned */
 512         return (xfer->len > MTK_SPI_MAX_FIFO_SIZE &&
 513                 (unsigned long)xfer->tx_buf % 4 == 0 &&
 514                 (unsigned long)xfer->rx_buf % 4 == 0);
 515 }
 516
 517 static int mtk_spi_setup(struct spi_device *spi)
 518 {
 519         struct mtk_spi *mdata = spi_master_get_devdata(spi->master);
 520
 521         if (!spi->controller_data)
 522                 spi->controller_data = (void *)&mtk_default_chip_info;
 523
 524         if (mdata->dev_comp->need_pad_sel && gpio_is_valid(spi->cs_gpio))
 525                 gpio_direction_output(spi->cs_gpio, !(spi->mode & SPI_CS_HIGH));
 526
 527         return 0;
 528 }
 529
 530 static irqreturn_t mtk_spi_interrupt(int irq, void *dev_id)
 531 {
 532         u32 cmd, reg_val, cnt, remainder, len;
 533         struct spi_master *master = dev_id;
 534         struct mtk_spi *mdata = spi_master_get_devdata(master);
 535         struct spi_transfer *trans = mdata->cur_transfer;
 536
 537         reg_val = readl(mdata->base + SPI_STATUS0_REG);
 538         if (reg_val & MTK_SPI_PAUSE_INT_STATUS)
 539                 mdata->state = MTK_SPI_PAUSED;
 540         else
 541                 mdata->state = MTK_SPI_IDLE;
 542
 543         if (!master->can_dma(master, NULL, trans)) {
 544                 if (trans->rx_buf) {
 545                         cnt = mdata->xfer_len / 4;
 546                         ioread32_rep(mdata->base + SPI_RX_DATA_REG,
 547                                      trans->rx_buf + mdata->num_xfered, cnt);
 548                         remainder = mdata->xfer_len % 4;
 549                         if (remainder > 0) {
 550                                 reg_val = readl(mdata->base + SPI_RX_DATA_REG);
 551                                 memcpy(trans->rx_buf +
 552                                         mdata->num_xfered +
 553                                         (cnt * 4),
 554                                         &reg_val,
 555                                         remainder);
 556                         }
 557                 }
 558
 559                 mdata->num_xfered += mdata->xfer_len;
 560                 if (mdata->num_xfered == trans->len) {
 561                         spi_finalize_current_transfer(master);
 562                         return IRQ_HANDLED;
 563                 }
 564
 565                 len = trans->len - mdata->num_xfered;
 566                 mdata->xfer_len = min(MTK_SPI_MAX_FIFO_SIZE, len);
 567                 mtk_spi_setup_packet(master);
 568
 569                 if (trans->tx_buf) {
 570                         cnt = mdata->xfer_len / 4;
 571                         iowrite32_rep(mdata->base + SPI_TX_DATA_REG,
 572                                         trans->tx_buf + mdata->num_xfered, cnt);
 573
 574                         remainder = mdata->xfer_len % 4;
 575                         if (remainder > 0) {
 576                                 reg_val = 0;
 577                                 memcpy(&reg_val,
 578                                         trans->tx_buf + (cnt * 4) + mdata->num_xfered,
 579                                         remainder);
 580                                 writel(reg_val, mdata->base + SPI_TX_DATA_REG);
 581                         }
 582                 }
 583
 584                 mtk_spi_enable_transfer(master);
 585
 586                 return IRQ_HANDLED;
 587         }
 588
 589         if (mdata->tx_sgl)
 590                 trans->tx_dma += mdata->xfer_len;
 591         if (mdata->rx_sgl)
 592                 trans->rx_dma += mdata->xfer_len;
 593
 594         if (mdata->tx_sgl && (mdata->tx_sgl_len == 0)) {
 595                 mdata->tx_sgl = sg_next(mdata->tx_sgl);
 596                 if (mdata->tx_sgl) {
 597                         trans->tx_dma = sg_dma_address(mdata->tx_sgl);
 598                         mdata->tx_sgl_len = sg_dma_len(mdata->tx_sgl);
 599                 }
 600         }
 601         if (mdata->rx_sgl && (mdata->rx_sgl_len == 0)) {
 602                 mdata->rx_sgl = sg_next(mdata->rx_sgl);
 603                 if (mdata->rx_sgl) {
 604                         trans->rx_dma = sg_dma_address(mdata->rx_sgl);
 605                         mdata->rx_sgl_len = sg_dma_len(mdata->rx_sgl);
 606                 }
 607         }
 608
 609         if (!mdata->tx_sgl && !mdata->rx_sgl) {
 610                 /* spi disable dma */
 611                 cmd = readl(mdata->base + SPI_CMD_REG);
 612                 cmd &= ~SPI_CMD_TX_DMA;
 613                 cmd &= ~SPI_CMD_RX_DMA;
 614                 writel(cmd, mdata->base + SPI_CMD_REG);
 615
 616                 spi_finalize_current_transfer(master);
 617                 return IRQ_HANDLED;
 618         }
 619
 620         mtk_spi_update_mdata_len(master);
 621         mtk_spi_setup_packet(master);
 622         mtk_spi_setup_dma_addr(master, trans);
 623         mtk_spi_enable_transfer(master);
 624
 625         return IRQ_HANDLED;
 626 }
 627
 628 static int mtk_spi_probe(struct platform_device *pdev)
 629 {
 630         struct spi_master *master;
 631         struct mtk_spi *mdata;
 632         const struct of_device_id *of_id;
 633         int i, irq, ret, addr_bits;
 634
 635         master = spi_alloc_master(&pdev->dev, sizeof(*mdata));
 636         if (!master) {
 637                 dev_err(&pdev->dev, "failed to alloc spi master\n");
 638                 return -ENOMEM;
 639         }
 640
 641         master->auto_runtime_pm = true;
 642         master->dev.of_node = pdev->dev.of_node;
 643         master->mode_bits = SPI_CPOL | SPI_CPHA | SPI_LSB_FIRST;
 644
 645         master->set_cs = mtk_spi_set_cs;
 646         master->prepare_message = mtk_spi_prepare_message;
 647         master->transfer_one = mtk_spi_transfer_one;
 648         master->can_dma = mtk_spi_can_dma;
 649         master->setup = mtk_spi_setup;
 650
 651         of_id = of_match_node(mtk_spi_of_match, pdev->dev.of_node);
 652         if (!of_id) {
 653                 dev_err(&pdev->dev, "failed to probe of_node\n");
 654                 ret = -EINVAL;
 655                 goto err_put_master;
 656         }
 657
 658         mdata = spi_master_get_devdata(master);
 659         mdata->dev_comp = of_id->data;
 660
 661         if (mdata->dev_comp->enhance_timing)
 662                 master->mode_bits |= SPI_CS_HIGH;
 663
 664         if (mdata->dev_comp->must_tx)
 665                 master->flags = SPI_MASTER_MUST_TX;
 666
 667         if (mdata->dev_comp->need_pad_sel) {
 668                 mdata->pad_num = of_property_count_u32_elems(
 669                         pdev->dev.of_node,
 670                         "mediatek,pad-select");
 671                 if (mdata->pad_num < 0) {
 672                         dev_err(&pdev->dev,
 673                                 "No 'mediatek,pad-select' property\n");
 674                         ret = -EINVAL;
 675                         goto err_put_master;
 676                 }
 677
 678                 mdata->pad_sel = devm_kmalloc_array(&pdev->dev, mdata->pad_num,
 679                                                     sizeof(u32), GFP_KERNEL);
 680                 if (!mdata->pad_sel) {
 681                         ret = -ENOMEM;
 682                         goto err_put_master;
 683                 }
 684
 685                 for (i = 0; i < mdata->pad_num; i++) {
 686                         of_property_read_u32_index(pdev->dev.of_node,
 687                                                    "mediatek,pad-select",
 688                                                    i, &mdata->pad_sel[i]);
 689                         if (mdata->pad_sel[i] > MT8173_SPI_MAX_PAD_SEL) {
 690                                 dev_err(&pdev->dev, "wrong pad-sel[%d]: %u\n",
 691                                         i, mdata->pad_sel[i]);
 692                                 ret = -EINVAL;
 693                                 goto err_put_master;
 694                         }
 695                 }
 696         }
 697
 698         platform_set_drvdata(pdev, master);
 699         mdata->base = devm_platform_ioremap_resource(pdev, 0);
 700         if (IS_ERR(mdata->base)) {
 701                 ret = PTR_ERR(mdata->base);
 702                 goto err_put_master;
 703         }
 704
 705         irq = platform_get_irq(pdev, 0);
 706         if (irq < 0) {
 707                 ret = irq;
 708                 goto err_put_master;
 709         }
 710
 711         if (!pdev->dev.dma_mask)
 712                 pdev->dev.dma_mask = &pdev->dev.coherent_dma_mask;
 713
 714         ret = devm_request_irq(&pdev->dev, irq, mtk_spi_interrupt,
 715                                IRQF_TRIGGER_NONE, dev_name(&pdev->dev), master);
 716         if (ret) {
 717                 dev_err(&pdev->dev, "failed to register irq (%d)\n", ret);
 718                 goto err_put_master;
 719         }
 720
 721         mdata->parent_clk = devm_clk_get(&pdev->dev, "parent-clk");
 722         if (IS_ERR(mdata->parent_clk)) {
 723                 ret = PTR_ERR(mdata->parent_clk);
 724                 dev_err(&pdev->dev, "failed to get parent-clk: %d\n", ret);
 725                 goto err_put_master;
 726         }
 727
 728         mdata->sel_clk = devm_clk_get(&pdev->dev, "sel-clk");
 729         if (IS_ERR(mdata->sel_clk)) {
 730                 ret = PTR_ERR(mdata->sel_clk);
 731                 dev_err(&pdev->dev, "failed to get sel-clk: %d\n", ret);
 732                 goto err_put_master;
 733         }
 734
 735         mdata->spi_clk = devm_clk_get(&pdev->dev, "spi-clk");
 736         if (IS_ERR(mdata->spi_clk)) {
 737                 ret = PTR_ERR(mdata->spi_clk);
 738                 dev_err(&pdev->dev, "failed to get spi-clk: %d\n", ret);
 739                 goto err_put_master;
 740         }
 741
 742         ret = clk_prepare_enable(mdata->spi_clk);
 743         if (ret < 0) {
 744                 dev_err(&pdev->dev, "failed to enable spi_clk (%d)\n", ret);
 745                 goto err_put_master;
 746         }
 747
 748         ret = clk_set_parent(mdata->sel_clk, mdata->parent_clk);
 749         if (ret < 0) {
 750                 dev_err(&pdev->dev, "failed to clk_set_parent (%d)\n", ret);
 751                 clk_disable_unprepare(mdata->spi_clk);
 752                 goto err_put_master;
 753         }
 754
 755         clk_disable_unprepare(mdata->spi_clk);
 756
 757         pm_runtime_enable(&pdev->dev);
 758
 759         ret = devm_spi_register_master(&pdev->dev, master);
 760         if (ret) {
 761                 dev_err(&pdev->dev, "failed to register master (%d)\n", ret);
 762                 goto err_disable_runtime_pm;
 763         }
 764
 765         if (mdata->dev_comp->need_pad_sel) {
 766                 if (mdata->pad_num != master->num_chipselect) {
 767                         dev_err(&pdev->dev,
 768                                 "pad_num does not match num_chipselect(%d != %d)\n",
 769                                 mdata->pad_num, master->num_chipselect);
 770                         ret = -EINVAL;
 771                         goto err_disable_runtime_pm;
 772                 }
 773
 774                 if (!master->cs_gpios && master->num_chipselect > 1) {
 775                         dev_err(&pdev->dev,
 776                                 "cs_gpios not specified and num_chipselect > 1\n");
 777                         ret = -EINVAL;
 778                         goto err_disable_runtime_pm;
 779                 }
 780
 781                 if (master->cs_gpios) {
 782                         for (i = 0; i < master->num_chipselect; i++) {
 783                                 ret = devm_gpio_request(&pdev->dev,
 784                                                         master->cs_gpios[i],
 785                                                         dev_name(&pdev->dev));
 786                                 if (ret) {
 787                                         dev_err(&pdev->dev,
 788                                                 "can't get CS GPIO %i\n", i);
 789                                         goto err_disable_runtime_pm;
 790                                 }
 791                         }
 792                 }
 793         }
 794
 795         if (mdata->dev_comp->dma_ext)
 796                 addr_bits = DMA_ADDR_EXT_BITS;
 797         else
 798                 addr_bits = DMA_ADDR_DEF_BITS;
 799         ret = dma_set_mask(&pdev->dev, DMA_BIT_MASK(addr_bits));
 800         if (ret)
 801                 dev_notice(&pdev->dev, "SPI dma_set_mask(%d) failed, ret:%d\n",
 802                            addr_bits, ret);
 803
 804         return 0;
 805
 806 err_disable_runtime_pm:
 807         pm_runtime_disable(&pdev->dev);
 808 err_put_master:
 809         spi_master_put(master);
 810
 811         return ret;
 812 }
 813
 814 static int mtk_spi_remove(struct platform_device *pdev)
 815 {
 816         struct spi_master *master = platform_get_drvdata(pdev);
 817         struct mtk_spi *mdata = spi_master_get_devdata(master);
 818
 819         pm_runtime_disable(&pdev->dev);
 820
 821         mtk_spi_reset(mdata);
 822
 823         return 0;
 824 }
 825
 826 #ifdef CONFIG_PM_SLEEP
 827 static int mtk_spi_suspend(struct device *dev)
 828 {
 829         int ret;
 830         struct spi_master *master = dev_get_drvdata(dev);
 831         struct mtk_spi *mdata = spi_master_get_devdata(master);
 832
 833         ret = spi_master_suspend(master);
 834         if (ret)
 835                 return ret;
 836
 837         if (!pm_runtime_suspended(dev))
 838                 clk_disable_unprepare(mdata->spi_clk);
 839
 840         return ret;
 841 }
 842
 843 static int mtk_spi_resume(struct device *dev)
 844 {
 845         int ret;
 846         struct spi_master *master = dev_get_drvdata(dev);
 847         struct mtk_spi *mdata = spi_master_get_devdata(master);
 848
 849         if (!pm_runtime_suspended(dev)) {
 850                 ret = clk_prepare_enable(mdata->spi_clk);
 851                 if (ret < 0) {
 852                         dev_err(dev, "failed to enable spi_clk (%d)\n", ret);
 853                         return ret;
 854                 }
 855         }
 856
 857         ret = spi_master_resume(master);
 858         if (ret < 0)
 859                 clk_disable_unprepare(mdata->spi_clk);
 860
 861         return ret;
 862 }
 863 #endif /* CONFIG_PM_SLEEP */
 864
 865 #ifdef CONFIG_PM
 866 static int mtk_spi_runtime_suspend(struct device *dev)
 867 {
 868         struct spi_master *master = dev_get_drvdata(dev);
 869         struct mtk_spi *mdata = spi_master_get_devdata(master);
 870
 871         clk_disable_unprepare(mdata->spi_clk);
 872
 873         return 0;
 874 }
 875
 876 static int mtk_spi_runtime_resume(struct device *dev)
 877 {
 878         struct spi_master *master = dev_get_drvdata(dev);
 879         struct mtk_spi *mdata = spi_master_get_devdata(master);
 880         int ret;
 881
 882         ret = clk_prepare_enable(mdata->spi_clk);
 883         if (ret < 0) {
 884                 dev_err(dev, "failed to enable spi_clk (%d)\n", ret);
 885                 return ret;
 886         }
 887
 888         return 0;
 889 }
 890 #endif /* CONFIG_PM */
 891
 892 static const struct dev_pm_ops mtk_spi_pm = {
 893         SET_SYSTEM_SLEEP_PM_OPS(mtk_spi_suspend, mtk_spi_resume)
 894         SET_RUNTIME_PM_OPS(mtk_spi_runtime_suspend,
 895                            mtk_spi_runtime_resume, NULL)
 896 };
 897
 898 static struct platform_driver mtk_spi_driver = {
 899         .driver = {
 900                 .name = "mtk-spi",
 901                 .pm     = &mtk_spi_pm,
 902                 .of_match_table = mtk_spi_of_match,
 903         },
 904         .probe = mtk_spi_probe,
 905         .remove = mtk_spi_remove,
 906 };
 907
 908 module_platform_driver(mtk_spi_driver);
 909
 910 MODULE_DESCRIPTION("MTK SPI Controller driver");
 911 MODULE_AUTHOR("Leilk Liu <leilk.liu@mediatek.com>");
 912 MODULE_LICENSE("GPL v2");
 913 MODULE_ALIAS("platform:mtk-spi");