xref: /openbmc/u-boot/drivers/spi/mt7621_spi.c (revision ef64e782)
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (C) 2018 Stefan Roese <sr@denx.de>
4  *
5  * Derived from the Linux driver version drivers/spi/spi-mt7621.c
6  *   Copyright (C) 2011 Sergiy <piratfm@gmail.com>
7  *   Copyright (C) 2011-2013 Gabor Juhos <juhosg@openwrt.org>
8  *   Copyright (C) 2014-2015 Felix Fietkau <nbd@nbd.name>
9  */
10 
11 #include <common.h>
12 #include <dm.h>
13 #include <spi.h>
14 #include <wait_bit.h>
15 #include <linux/io.h>
16 
17 #define SPI_MSG_SIZE_MAX	32	/* SPI message chunk size */
18 /* Enough for SPI NAND page read / write with page size 2048 bytes */
19 #define SPI_MSG_SIZE_OVERALL	(2048 + 16)
20 
21 #define MT7621_SPI_TRANS	0x00
22 #define MT7621_SPI_TRANS_START	BIT(8)
23 #define MT7621_SPI_TRANS_BUSY	BIT(16)
24 
25 #define MT7621_SPI_OPCODE	0x04
26 #define MT7621_SPI_DATA0	0x08
27 #define MT7621_SPI_DATA4	0x18
28 #define MT7621_SPI_MASTER	0x28
29 #define MT7621_SPI_MOREBUF	0x2c
30 #define MT7621_SPI_POLAR	0x38
31 
32 #define MT7621_LSB_FIRST	BIT(3)
33 #define MT7621_CPOL		BIT(4)
34 #define MT7621_CPHA		BIT(5)
35 
36 #define MASTER_MORE_BUFMODE	BIT(2)
37 #define MASTER_RS_CLK_SEL	GENMASK(27, 16)
38 #define MASTER_RS_CLK_SEL_SHIFT	16
39 #define MASTER_RS_SLAVE_SEL	GENMASK(31, 29)
40 
41 struct mt7621_spi {
42 	void __iomem *base;
43 	unsigned int sys_freq;
44 	u32 data[(SPI_MSG_SIZE_OVERALL / 4) + 1];
45 	int tx_len;
46 };
47 
48 static void mt7621_spi_reset(struct mt7621_spi *rs, int duplex)
49 {
50 	setbits_le32(rs->base + MT7621_SPI_MASTER,
51 		     MASTER_RS_SLAVE_SEL | MASTER_MORE_BUFMODE);
52 }
53 
54 static void mt7621_spi_set_cs(struct mt7621_spi *rs, int cs, int enable)
55 {
56 	u32 val = 0;
57 
58 	debug("%s: cs#%d -> %s\n", __func__, cs, enable ? "enable" : "disable");
59 	if (enable)
60 		val = BIT(cs);
61 	iowrite32(val, rs->base + MT7621_SPI_POLAR);
62 }
63 
64 static int mt7621_spi_set_mode(struct udevice *bus, uint mode)
65 {
66 	struct mt7621_spi *rs = dev_get_priv(bus);
67 	u32 reg;
68 
69 	debug("%s: mode=0x%08x\n", __func__, mode);
70 	reg = ioread32(rs->base + MT7621_SPI_MASTER);
71 
72 	reg &= ~MT7621_LSB_FIRST;
73 	if (mode & SPI_LSB_FIRST)
74 		reg |= MT7621_LSB_FIRST;
75 
76 	reg &= ~(MT7621_CPHA | MT7621_CPOL);
77 	switch (mode & (SPI_CPOL | SPI_CPHA)) {
78 	case SPI_MODE_0:
79 		break;
80 	case SPI_MODE_1:
81 		reg |= MT7621_CPHA;
82 		break;
83 	case SPI_MODE_2:
84 		reg |= MT7621_CPOL;
85 		break;
86 	case SPI_MODE_3:
87 		reg |= MT7621_CPOL | MT7621_CPHA;
88 		break;
89 	}
90 	iowrite32(reg, rs->base + MT7621_SPI_MASTER);
91 
92 	return 0;
93 }
94 
95 static int mt7621_spi_set_speed(struct udevice *bus, uint speed)
96 {
97 	struct mt7621_spi *rs = dev_get_priv(bus);
98 	u32 rate;
99 	u32 reg;
100 
101 	debug("%s: speed=%d\n", __func__, speed);
102 	rate = DIV_ROUND_UP(rs->sys_freq, speed);
103 	debug("rate:%u\n", rate);
104 
105 	if (rate > 4097)
106 		return -EINVAL;
107 
108 	if (rate < 2)
109 		rate = 2;
110 
111 	reg = ioread32(rs->base + MT7621_SPI_MASTER);
112 	reg &= ~MASTER_RS_CLK_SEL;
113 	reg |= (rate - 2) << MASTER_RS_CLK_SEL_SHIFT;
114 	iowrite32(reg, rs->base + MT7621_SPI_MASTER);
115 
116 	return 0;
117 }
118 
119 static inline int mt7621_spi_wait_till_ready(struct mt7621_spi *rs)
120 {
121 	int ret;
122 
123 	ret =  wait_for_bit_le32(rs->base + MT7621_SPI_TRANS,
124 				 MT7621_SPI_TRANS_BUSY, 0, 10, 0);
125 	if (ret)
126 		pr_err("Timeout in %s!\n", __func__);
127 
128 	return ret;
129 }
130 
131 static int mt7621_spi_xfer(struct udevice *dev, unsigned int bitlen,
132 			   const void *dout, void *din, unsigned long flags)
133 {
134 	struct udevice *bus = dev->parent;
135 	struct mt7621_spi *rs = dev_get_priv(bus);
136 	const u8 *tx_buf = dout;
137 	u8 *ptr = (u8 *)dout;
138 	u8 *rx_buf = din;
139 	int total_size = bitlen >> 3;
140 	int chunk_size;
141 	int rx_len = 0;
142 	u32 data[(SPI_MSG_SIZE_MAX / 4) + 1] = { 0 };
143 	u32 val;
144 	int i;
145 
146 	debug("%s: dout=%p, din=%p, len=%x, flags=%lx\n", __func__, dout, din,
147 	      total_size, flags);
148 
149 	/*
150 	 * This driver only supports half-duplex, so complain and bail out
151 	 * upon full-duplex messages
152 	 */
153 	if (dout && din) {
154 		printf("Only half-duplex SPI transfer supported\n");
155 		return -EIO;
156 	}
157 
158 	if (dout) {
159 		debug("TX-DATA: ");
160 		for (i = 0; i < total_size; i++)
161 			debug("%02x ", *ptr++);
162 		debug("\n");
163 	}
164 
165 	mt7621_spi_wait_till_ready(rs);
166 
167 	/*
168 	 * Set CS active upon start of SPI message. This message can
169 	 * be split upon multiple calls to this xfer function
170 	 */
171 	if (flags & SPI_XFER_BEGIN)
172 		mt7621_spi_set_cs(rs, spi_chip_select(dev), 1);
173 
174 	while (total_size > 0) {
175 		/* Don't exceed the max xfer size */
176 		chunk_size = min_t(int, total_size, SPI_MSG_SIZE_MAX);
177 
178 		/*
179 		 * We might have some TX data buffered from the last xfer
180 		 * message. Make sure, that this does not exceed the max
181 		 * xfer size
182 		 */
183 		if (rs->tx_len > 4)
184 			chunk_size -= rs->tx_len;
185 		if (din)
186 			rx_len = chunk_size;
187 
188 		if (tx_buf) {
189 			/* Check if this message does not exceed the buffer */
190 			if ((chunk_size + rs->tx_len) > SPI_MSG_SIZE_OVERALL) {
191 				printf("TX message size too big (%d)\n",
192 				       chunk_size + rs->tx_len);
193 				return -EMSGSIZE;
194 			}
195 
196 			/*
197 			 * Write all TX data into internal buffer to collect
198 			 * all TX messages into one buffer (might be split into
199 			 * multiple calls to this function)
200 			 */
201 			for (i = 0; i < chunk_size; i++, rs->tx_len++) {
202 				rs->data[rs->tx_len / 4] |=
203 					tx_buf[i] << (8 * (rs->tx_len & 3));
204 			}
205 		}
206 
207 		if (flags & SPI_XFER_END) {
208 			/* Write TX data into controller */
209 			if (rs->tx_len) {
210 				rs->data[0] = swab32(rs->data[0]);
211 				if (rs->tx_len < 4)
212 					rs->data[0] >>= (4 - rs->tx_len) * 8;
213 
214 				for (i = 0; i < rs->tx_len; i += 4) {
215 					iowrite32(rs->data[i / 4], rs->base +
216 						  MT7621_SPI_OPCODE + i);
217 				}
218 			}
219 
220 			/* Write length into controller */
221 			val = (min_t(int, rs->tx_len, 4) * 8) << 24;
222 			if (rs->tx_len > 4)
223 				val |= (rs->tx_len - 4) * 8;
224 			val |= (rx_len * 8) << 12;
225 			iowrite32(val, rs->base + MT7621_SPI_MOREBUF);
226 
227 			/* Start the xfer */
228 			setbits_le32(rs->base + MT7621_SPI_TRANS,
229 				     MT7621_SPI_TRANS_START);
230 
231 			/* Wait until xfer is finished on bus */
232 			mt7621_spi_wait_till_ready(rs);
233 
234 			/* Reset TX length and TX buffer for next xfer */
235 			rs->tx_len = 0;
236 			memset(rs->data, 0, sizeof(rs->data));
237 		}
238 
239 		for (i = 0; i < rx_len; i += 4)
240 			data[i / 4] = ioread32(rs->base + MT7621_SPI_DATA0 + i);
241 
242 		if (rx_len) {
243 			debug("RX-DATA: ");
244 			for (i = 0; i < rx_len; i++) {
245 				rx_buf[i] = data[i / 4] >> (8 * (i & 3));
246 				debug("%02x ", rx_buf[i]);
247 			}
248 			debug("\n");
249 		}
250 
251 		if (tx_buf)
252 			tx_buf += chunk_size;
253 		if (rx_buf)
254 			rx_buf += chunk_size;
255 		total_size -= chunk_size;
256 	}
257 
258 	/* Wait until xfer is finished on bus and de-assert CS */
259 	mt7621_spi_wait_till_ready(rs);
260 	if (flags & SPI_XFER_END)
261 		mt7621_spi_set_cs(rs, spi_chip_select(dev), 0);
262 
263 	return 0;
264 }
265 
266 static int mt7621_spi_probe(struct udevice *dev)
267 {
268 	struct mt7621_spi *rs = dev_get_priv(dev);
269 
270 	rs->base = dev_remap_addr(dev);
271 	if (!rs->base)
272 		return -EINVAL;
273 
274 	/*
275 	 * Read input clock via DT for now. At some point this should be
276 	 * replaced by implementing a clock driver for this SoC and getting
277 	 * the SPI frequency via this clock driver.
278 	 */
279 	rs->sys_freq = dev_read_u32_default(dev, "clock-frequency", 0);
280 	if (!rs->sys_freq) {
281 		printf("Please provide clock-frequency!\n");
282 		return -EINVAL;
283 	}
284 
285 	mt7621_spi_reset(rs, 0);
286 
287 	return 0;
288 }
289 
290 static const struct dm_spi_ops mt7621_spi_ops = {
291 	.set_mode = mt7621_spi_set_mode,
292 	.set_speed = mt7621_spi_set_speed,
293 	.xfer = mt7621_spi_xfer,
294 	/*
295 	 * cs_info is not needed, since we require all chip selects to be
296 	 * in the device tree explicitly
297 	 */
298 };
299 
300 static const struct udevice_id mt7621_spi_ids[] = {
301 	{ .compatible = "ralink,mt7621-spi" },
302 	{ }
303 };
304 
305 U_BOOT_DRIVER(mt7621_spi) = {
306 	.name = "mt7621_spi",
307 	.id = UCLASS_SPI,
308 	.of_match = mt7621_spi_ids,
309 	.ops = &mt7621_spi_ops,
310 	.priv_auto_alloc_size = sizeof(struct mt7621_spi),
311 	.probe = mt7621_spi_probe,
312 };
313