1
2
3
4
5
6
7
8
9
10
11
12
13
14
15#include <linux/clk.h>
16#include <linux/delay.h>
17#include <linux/device.h>
18#include <linux/init.h>
19#include <linux/io.h>
20#include <linux/iopoll.h>
21#include <linux/ioport.h>
22#include <linux/math64.h>
23#include <linux/module.h>
24#include <linux/mutex.h>
25#include <linux/of.h>
26#include <linux/of_device.h>
27#include <linux/platform_device.h>
28#include <linux/slab.h>
29#include <linux/mtd/mtd.h>
30#include <linux/mtd/partitions.h>
31#include <linux/mtd/spi-nor.h>
32
33#define MTK_NOR_CMD_REG 0x00
34#define MTK_NOR_CNT_REG 0x04
35#define MTK_NOR_RDSR_REG 0x08
36#define MTK_NOR_RDATA_REG 0x0c
37#define MTK_NOR_RADR0_REG 0x10
38#define MTK_NOR_RADR1_REG 0x14
39#define MTK_NOR_RADR2_REG 0x18
40#define MTK_NOR_WDATA_REG 0x1c
41#define MTK_NOR_PRGDATA0_REG 0x20
42#define MTK_NOR_PRGDATA1_REG 0x24
43#define MTK_NOR_PRGDATA2_REG 0x28
44#define MTK_NOR_PRGDATA3_REG 0x2c
45#define MTK_NOR_PRGDATA4_REG 0x30
46#define MTK_NOR_PRGDATA5_REG 0x34
47#define MTK_NOR_SHREG0_REG 0x38
48#define MTK_NOR_SHREG1_REG 0x3c
49#define MTK_NOR_SHREG2_REG 0x40
50#define MTK_NOR_SHREG3_REG 0x44
51#define MTK_NOR_SHREG4_REG 0x48
52#define MTK_NOR_SHREG5_REG 0x4c
53#define MTK_NOR_SHREG6_REG 0x50
54#define MTK_NOR_SHREG7_REG 0x54
55#define MTK_NOR_SHREG8_REG 0x58
56#define MTK_NOR_SHREG9_REG 0x5c
57#define MTK_NOR_CFG1_REG 0x60
58#define MTK_NOR_CFG2_REG 0x64
59#define MTK_NOR_CFG3_REG 0x68
60#define MTK_NOR_STATUS0_REG 0x70
61#define MTK_NOR_STATUS1_REG 0x74
62#define MTK_NOR_STATUS2_REG 0x78
63#define MTK_NOR_STATUS3_REG 0x7c
64#define MTK_NOR_FLHCFG_REG 0x84
65#define MTK_NOR_TIME_REG 0x94
66#define MTK_NOR_PP_DATA_REG 0x98
67#define MTK_NOR_PREBUF_STUS_REG 0x9c
68#define MTK_NOR_DELSEL0_REG 0xa0
69#define MTK_NOR_DELSEL1_REG 0xa4
70#define MTK_NOR_INTRSTUS_REG 0xa8
71#define MTK_NOR_INTREN_REG 0xac
72#define MTK_NOR_CHKSUM_CTL_REG 0xb8
73#define MTK_NOR_CHKSUM_REG 0xbc
74#define MTK_NOR_CMD2_REG 0xc0
75#define MTK_NOR_WRPROT_REG 0xc4
76#define MTK_NOR_RADR3_REG 0xc8
77#define MTK_NOR_DUAL_REG 0xcc
78#define MTK_NOR_DELSEL2_REG 0xd0
79#define MTK_NOR_DELSEL3_REG 0xd4
80#define MTK_NOR_DELSEL4_REG 0xd8
81
82
83#define MTK_NOR_READ_CMD 0x0
84#define MTK_NOR_RDSR_CMD 0x2
85#define MTK_NOR_PRG_CMD 0x4
86#define MTK_NOR_WR_CMD 0x10
87#define MTK_NOR_PIO_WR_CMD 0x90
88#define MTK_NOR_WRSR_CMD 0x20
89#define MTK_NOR_PIO_READ_CMD 0x81
90#define MTK_NOR_WR_BUF_ENABLE 0x1
91#define MTK_NOR_WR_BUF_DISABLE 0x0
92#define MTK_NOR_ENABLE_SF_CMD 0x30
93#define MTK_NOR_DUAD_ADDR_EN 0x8
94#define MTK_NOR_QUAD_READ_EN 0x4
95#define MTK_NOR_DUAL_ADDR_EN 0x2
96#define MTK_NOR_DUAL_READ_EN 0x1
97#define MTK_NOR_DUAL_DISABLE 0x0
98#define MTK_NOR_FAST_READ 0x1
99
100#define SFLASH_WRBUF_SIZE 128
101
102
103#define MTK_NOR_MAX_RX_TX_SHIFT 6
104
105#define MTK_NOR_MAX_SHIFT 7
106
107#define MTK_NOR_4B_ADDR_EN BIT(4)
108
109
110#define MTK_NOR_PRG_REG(n) (MTK_NOR_PRGDATA0_REG + 4 * (n))
111#define MTK_NOR_SHREG(n) (MTK_NOR_SHREG0_REG + 4 * (n))
112
113struct mtk_nor {
114 struct spi_nor nor;
115 struct device *dev;
116 void __iomem *base;
117 struct clk *spi_clk;
118 struct clk *nor_clk;
119};
120
121static void mtk_nor_set_read_mode(struct mtk_nor *mtk_nor)
122{
123 struct spi_nor *nor = &mtk_nor->nor;
124
125 switch (nor->read_proto) {
126 case SNOR_PROTO_1_1_1:
127 writeb(nor->read_opcode, mtk_nor->base +
128 MTK_NOR_PRGDATA3_REG);
129 writeb(MTK_NOR_FAST_READ, mtk_nor->base +
130 MTK_NOR_CFG1_REG);
131 break;
132 case SNOR_PROTO_1_1_2:
133 writeb(nor->read_opcode, mtk_nor->base +
134 MTK_NOR_PRGDATA3_REG);
135 writeb(MTK_NOR_DUAL_READ_EN, mtk_nor->base +
136 MTK_NOR_DUAL_REG);
137 break;
138 case SNOR_PROTO_1_1_4:
139 writeb(nor->read_opcode, mtk_nor->base +
140 MTK_NOR_PRGDATA4_REG);
141 writeb(MTK_NOR_QUAD_READ_EN, mtk_nor->base +
142 MTK_NOR_DUAL_REG);
143 break;
144 default:
145 writeb(MTK_NOR_DUAL_DISABLE, mtk_nor->base +
146 MTK_NOR_DUAL_REG);
147 break;
148 }
149}
150
151static int mtk_nor_execute_cmd(struct mtk_nor *mtk_nor, u8 cmdval)
152{
153 int reg;
154 u8 val = cmdval & 0x1f;
155
156 writeb(cmdval, mtk_nor->base + MTK_NOR_CMD_REG);
157 return readl_poll_timeout(mtk_nor->base + MTK_NOR_CMD_REG, reg,
158 !(reg & val), 100, 10000);
159}
160
161static int mtk_nor_do_tx_rx(struct mtk_nor *mtk_nor, u8 op,
162 u8 *tx, int txlen, u8 *rx, int rxlen)
163{
164 int len = 1 + txlen + rxlen;
165 int i, ret, idx;
166
167 if (len > MTK_NOR_MAX_SHIFT)
168 return -EINVAL;
169
170 writeb(len * 8, mtk_nor->base + MTK_NOR_CNT_REG);
171
172
173 idx = MTK_NOR_MAX_RX_TX_SHIFT - 1;
174
175
176 writeb(op, mtk_nor->base + MTK_NOR_PRG_REG(idx));
177 idx--;
178
179
180 for (i = 0; i < txlen; i++, idx--)
181 writeb(tx[i], mtk_nor->base + MTK_NOR_PRG_REG(idx));
182
183
184 while (idx >= 0) {
185 writeb(0, mtk_nor->base + MTK_NOR_PRG_REG(idx));
186 idx--;
187 }
188
189 ret = mtk_nor_execute_cmd(mtk_nor, MTK_NOR_PRG_CMD);
190 if (ret)
191 return ret;
192
193
194 idx = rxlen - 1;
195
196
197 for (i = 0; i < rxlen; i++, idx--)
198 rx[i] = readb(mtk_nor->base + MTK_NOR_SHREG(idx));
199
200 return 0;
201}
202
203
204static int mtk_nor_wr_sr(struct mtk_nor *mtk_nor, u8 sr)
205{
206 writeb(sr, mtk_nor->base + MTK_NOR_PRGDATA5_REG);
207 writeb(8, mtk_nor->base + MTK_NOR_CNT_REG);
208 return mtk_nor_execute_cmd(mtk_nor, MTK_NOR_WRSR_CMD);
209}
210
211static int mtk_nor_write_buffer_enable(struct mtk_nor *mtk_nor)
212{
213 u8 reg;
214
215
216
217
218
219 writel(MTK_NOR_WR_BUF_ENABLE, mtk_nor->base + MTK_NOR_CFG2_REG);
220 return readb_poll_timeout(mtk_nor->base + MTK_NOR_CFG2_REG, reg,
221 0x01 == (reg & 0x01), 100, 10000);
222}
223
224static int mtk_nor_write_buffer_disable(struct mtk_nor *mtk_nor)
225{
226 u8 reg;
227
228 writel(MTK_NOR_WR_BUF_DISABLE, mtk_nor->base + MTK_NOR_CFG2_REG);
229 return readb_poll_timeout(mtk_nor->base + MTK_NOR_CFG2_REG, reg,
230 MTK_NOR_WR_BUF_DISABLE == (reg & 0x1), 100,
231 10000);
232}
233
234static void mtk_nor_set_addr_width(struct mtk_nor *mtk_nor)
235{
236 u8 val;
237 struct spi_nor *nor = &mtk_nor->nor;
238
239 val = readb(mtk_nor->base + MTK_NOR_DUAL_REG);
240
241 switch (nor->addr_width) {
242 case 3:
243 val &= ~MTK_NOR_4B_ADDR_EN;
244 break;
245 case 4:
246 val |= MTK_NOR_4B_ADDR_EN;
247 break;
248 default:
249 dev_warn(mtk_nor->dev, "Unexpected address width %u.\n",
250 nor->addr_width);
251 break;
252 }
253
254 writeb(val, mtk_nor->base + MTK_NOR_DUAL_REG);
255}
256
257static void mtk_nor_set_addr(struct mtk_nor *mtk_nor, u32 addr)
258{
259 int i;
260
261 mtk_nor_set_addr_width(mtk_nor);
262
263 for (i = 0; i < 3; i++) {
264 writeb(addr & 0xff, mtk_nor->base + MTK_NOR_RADR0_REG + i * 4);
265 addr >>= 8;
266 }
267
268 writeb(addr & 0xff, mtk_nor->base + MTK_NOR_RADR3_REG);
269}
270
271static ssize_t mtk_nor_read(struct spi_nor *nor, loff_t from, size_t length,
272 u_char *buffer)
273{
274 int i, ret;
275 int addr = (int)from;
276 u8 *buf = (u8 *)buffer;
277 struct mtk_nor *mtk_nor = nor->priv;
278
279
280 mtk_nor_set_read_mode(mtk_nor);
281 mtk_nor_set_addr(mtk_nor, addr);
282
283 for (i = 0; i < length; i++) {
284 ret = mtk_nor_execute_cmd(mtk_nor, MTK_NOR_PIO_READ_CMD);
285 if (ret < 0)
286 return ret;
287 buf[i] = readb(mtk_nor->base + MTK_NOR_RDATA_REG);
288 }
289 return length;
290}
291
292static int mtk_nor_write_single_byte(struct mtk_nor *mtk_nor,
293 int addr, int length, u8 *data)
294{
295 int i, ret;
296
297 mtk_nor_set_addr(mtk_nor, addr);
298
299 for (i = 0; i < length; i++) {
300 writeb(*data++, mtk_nor->base + MTK_NOR_WDATA_REG);
301 ret = mtk_nor_execute_cmd(mtk_nor, MTK_NOR_PIO_WR_CMD);
302 if (ret < 0)
303 return ret;
304 }
305 return 0;
306}
307
308static int mtk_nor_write_buffer(struct mtk_nor *mtk_nor, int addr,
309 const u8 *buf)
310{
311 int i, bufidx, data;
312
313 mtk_nor_set_addr(mtk_nor, addr);
314
315 bufidx = 0;
316 for (i = 0; i < SFLASH_WRBUF_SIZE; i += 4) {
317 data = buf[bufidx + 3]<<24 | buf[bufidx + 2]<<16 |
318 buf[bufidx + 1]<<8 | buf[bufidx];
319 bufidx += 4;
320 writel(data, mtk_nor->base + MTK_NOR_PP_DATA_REG);
321 }
322 return mtk_nor_execute_cmd(mtk_nor, MTK_NOR_WR_CMD);
323}
324
325static ssize_t mtk_nor_write(struct spi_nor *nor, loff_t to, size_t len,
326 const u_char *buf)
327{
328 int ret;
329 struct mtk_nor *mtk_nor = nor->priv;
330 size_t i;
331
332 ret = mtk_nor_write_buffer_enable(mtk_nor);
333 if (ret < 0) {
334 dev_warn(mtk_nor->dev, "write buffer enable failed!\n");
335 return ret;
336 }
337
338 for (i = 0; i + SFLASH_WRBUF_SIZE <= len; i += SFLASH_WRBUF_SIZE) {
339 ret = mtk_nor_write_buffer(mtk_nor, to, buf);
340 if (ret < 0) {
341 dev_err(mtk_nor->dev, "write buffer failed!\n");
342 return ret;
343 }
344 to += SFLASH_WRBUF_SIZE;
345 buf += SFLASH_WRBUF_SIZE;
346 }
347 ret = mtk_nor_write_buffer_disable(mtk_nor);
348 if (ret < 0) {
349 dev_warn(mtk_nor->dev, "write buffer disable failed!\n");
350 return ret;
351 }
352
353 if (i < len) {
354 ret = mtk_nor_write_single_byte(mtk_nor, to,
355 (int)(len - i), (u8 *)buf);
356 if (ret < 0) {
357 dev_err(mtk_nor->dev, "write single byte failed!\n");
358 return ret;
359 }
360 }
361
362 return len;
363}
364
365static int mtk_nor_read_reg(struct spi_nor *nor, u8 opcode, u8 *buf, int len)
366{
367 int ret;
368 struct mtk_nor *mtk_nor = nor->priv;
369
370 switch (opcode) {
371 case SPINOR_OP_RDSR:
372 ret = mtk_nor_execute_cmd(mtk_nor, MTK_NOR_RDSR_CMD);
373 if (ret < 0)
374 return ret;
375 if (len == 1)
376 *buf = readb(mtk_nor->base + MTK_NOR_RDSR_REG);
377 else
378 dev_err(mtk_nor->dev, "len should be 1 for read status!\n");
379 break;
380 default:
381 ret = mtk_nor_do_tx_rx(mtk_nor, opcode, NULL, 0, buf, len);
382 break;
383 }
384 return ret;
385}
386
387static int mtk_nor_write_reg(struct spi_nor *nor, u8 opcode, u8 *buf,
388 int len)
389{
390 int ret;
391 struct mtk_nor *mtk_nor = nor->priv;
392
393 switch (opcode) {
394 case SPINOR_OP_WRSR:
395
396 ret = mtk_nor_wr_sr(mtk_nor, *buf);
397 break;
398 default:
399 ret = mtk_nor_do_tx_rx(mtk_nor, opcode, buf, len, NULL, 0);
400 if (ret)
401 dev_warn(mtk_nor->dev, "write reg failure!\n");
402 break;
403 }
404 return ret;
405}
406
407static void mtk_nor_disable_clk(struct mtk_nor *mtk_nor)
408{
409 clk_disable_unprepare(mtk_nor->spi_clk);
410 clk_disable_unprepare(mtk_nor->nor_clk);
411}
412
413static int mtk_nor_enable_clk(struct mtk_nor *mtk_nor)
414{
415 int ret;
416
417 ret = clk_prepare_enable(mtk_nor->spi_clk);
418 if (ret)
419 return ret;
420
421 ret = clk_prepare_enable(mtk_nor->nor_clk);
422 if (ret) {
423 clk_disable_unprepare(mtk_nor->spi_clk);
424 return ret;
425 }
426
427 return 0;
428}
429
430static int mtk_nor_init(struct mtk_nor *mtk_nor,
431 struct device_node *flash_node)
432{
433 const struct spi_nor_hwcaps hwcaps = {
434 .mask = SNOR_HWCAPS_READ_FAST |
435 SNOR_HWCAPS_READ_1_1_2 |
436 SNOR_HWCAPS_PP,
437 };
438 int ret;
439 struct spi_nor *nor;
440
441
442 writel(MTK_NOR_ENABLE_SF_CMD, mtk_nor->base + MTK_NOR_WRPROT_REG);
443
444 nor = &mtk_nor->nor;
445 nor->dev = mtk_nor->dev;
446 nor->priv = mtk_nor;
447 spi_nor_set_flash_node(nor, flash_node);
448
449
450 nor->read = mtk_nor_read;
451 nor->read_reg = mtk_nor_read_reg;
452 nor->write = mtk_nor_write;
453 nor->write_reg = mtk_nor_write_reg;
454 nor->mtd.name = "mtk_nor";
455
456 ret = spi_nor_scan(nor, NULL, &hwcaps);
457 if (ret)
458 return ret;
459
460 return mtd_device_register(&nor->mtd, NULL, 0);
461}
462
463static int mtk_nor_drv_probe(struct platform_device *pdev)
464{
465 struct device_node *flash_np;
466 struct resource *res;
467 int ret;
468 struct mtk_nor *mtk_nor;
469
470 if (!pdev->dev.of_node) {
471 dev_err(&pdev->dev, "No DT found\n");
472 return -EINVAL;
473 }
474
475 mtk_nor = devm_kzalloc(&pdev->dev, sizeof(*mtk_nor), GFP_KERNEL);
476 if (!mtk_nor)
477 return -ENOMEM;
478 platform_set_drvdata(pdev, mtk_nor);
479
480 res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
481 mtk_nor->base = devm_ioremap_resource(&pdev->dev, res);
482 if (IS_ERR(mtk_nor->base))
483 return PTR_ERR(mtk_nor->base);
484
485 mtk_nor->spi_clk = devm_clk_get(&pdev->dev, "spi");
486 if (IS_ERR(mtk_nor->spi_clk))
487 return PTR_ERR(mtk_nor->spi_clk);
488
489 mtk_nor->nor_clk = devm_clk_get(&pdev->dev, "sf");
490 if (IS_ERR(mtk_nor->nor_clk))
491 return PTR_ERR(mtk_nor->nor_clk);
492
493 mtk_nor->dev = &pdev->dev;
494
495 ret = mtk_nor_enable_clk(mtk_nor);
496 if (ret)
497 return ret;
498
499
500 flash_np = of_get_next_available_child(pdev->dev.of_node, NULL);
501 if (!flash_np) {
502 dev_err(&pdev->dev, "no SPI flash device to configure\n");
503 ret = -ENODEV;
504 goto nor_free;
505 }
506 ret = mtk_nor_init(mtk_nor, flash_np);
507
508nor_free:
509 if (ret)
510 mtk_nor_disable_clk(mtk_nor);
511
512 return ret;
513}
514
515static int mtk_nor_drv_remove(struct platform_device *pdev)
516{
517 struct mtk_nor *mtk_nor = platform_get_drvdata(pdev);
518
519 mtk_nor_disable_clk(mtk_nor);
520
521 return 0;
522}
523
524#ifdef CONFIG_PM_SLEEP
525static int mtk_nor_suspend(struct device *dev)
526{
527 struct mtk_nor *mtk_nor = dev_get_drvdata(dev);
528
529 mtk_nor_disable_clk(mtk_nor);
530
531 return 0;
532}
533
534static int mtk_nor_resume(struct device *dev)
535{
536 struct mtk_nor *mtk_nor = dev_get_drvdata(dev);
537
538 return mtk_nor_enable_clk(mtk_nor);
539}
540
541static const struct dev_pm_ops mtk_nor_dev_pm_ops = {
542 .suspend = mtk_nor_suspend,
543 .resume = mtk_nor_resume,
544};
545
546#define MTK_NOR_DEV_PM_OPS (&mtk_nor_dev_pm_ops)
547#else
548#define MTK_NOR_DEV_PM_OPS NULL
549#endif
550
551static const struct of_device_id mtk_nor_of_ids[] = {
552 { .compatible = "mediatek,mt8173-nor"},
553 { }
554};
555MODULE_DEVICE_TABLE(of, mtk_nor_of_ids);
556
557static struct platform_driver mtk_nor_driver = {
558 .probe = mtk_nor_drv_probe,
559 .remove = mtk_nor_drv_remove,
560 .driver = {
561 .name = "mtk-nor",
562 .pm = MTK_NOR_DEV_PM_OPS,
563 .of_match_table = mtk_nor_of_ids,
564 },
565};
566
567module_platform_driver(mtk_nor_driver);
568MODULE_LICENSE("GPL v2");
569MODULE_DESCRIPTION("MediaTek SPI NOR Flash Driver");
570
