1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Raydium touchscreen I2C driver.
4  *
5  * Copyright (C) 2012-2014, Raydium Semiconductor Corporation.
6  *
7  * Raydium reserves the right to make changes without further notice
8  * to the materials described herein. Raydium does not assume any
9  * liability arising out of the application described herein.
10  *
11  * Contact Raydium Semiconductor Corporation at www.rad-ic.com
12  */
13 
14 #include <linux/acpi.h>
15 #include <linux/delay.h>
16 #include <linux/firmware.h>
17 #include <linux/gpio/consumer.h>
18 #include <linux/i2c.h>
19 #include <linux/input.h>
20 #include <linux/input/mt.h>
21 #include <linux/interrupt.h>
22 #include <linux/module.h>
23 #include <linux/of.h>
24 #include <linux/regulator/consumer.h>
25 #include <linux/slab.h>
26 #include <asm/unaligned.h>
27 
28 /* Slave I2C mode */
29 #define RM_BOOT_BLDR		0x02
30 #define RM_BOOT_MAIN		0x03
31 
32 /* I2C bootoloader commands */
33 #define RM_CMD_BOOT_PAGE_WRT	0x0B		/* send bl page write */
34 #define RM_CMD_BOOT_WRT		0x11		/* send bl write */
35 #define RM_CMD_BOOT_ACK		0x22		/* send ack*/
36 #define RM_CMD_BOOT_CHK		0x33		/* send data check */
37 #define RM_CMD_BOOT_READ	0x44		/* send wait bl data ready*/
38 
39 #define RM_BOOT_RDY		0xFF		/* bl data ready */
40 
41 /* I2C main commands */
42 #define RM_CMD_QUERY_BANK	0x2B
43 #define RM_CMD_DATA_BANK	0x4D
44 #define RM_CMD_ENTER_SLEEP	0x4E
45 #define RM_CMD_BANK_SWITCH	0xAA
46 
47 #define RM_RESET_MSG_ADDR	0x40000004
48 
49 #define RM_MAX_READ_SIZE	56
50 #define RM_PACKET_CRC_SIZE	2
51 
52 /* Touch relative info */
53 #define RM_MAX_RETRIES		3
54 #define RM_RETRY_DELAY_MS	20
55 #define RM_MAX_TOUCH_NUM	10
56 #define RM_BOOT_DELAY_MS	100
57 
58 /* Offsets in contact data */
59 #define RM_CONTACT_STATE_POS	0
60 #define RM_CONTACT_X_POS	1
61 #define RM_CONTACT_Y_POS	3
62 #define RM_CONTACT_PRESSURE_POS	5
63 #define RM_CONTACT_WIDTH_X_POS	6
64 #define RM_CONTACT_WIDTH_Y_POS	7
65 
66 /* Bootloader relative info */
67 #define RM_BL_WRT_CMD_SIZE	3	/* bl flash wrt cmd size */
68 #define RM_BL_WRT_PKG_SIZE	32	/* bl wrt pkg size */
69 #define RM_BL_WRT_LEN		(RM_BL_WRT_PKG_SIZE + RM_BL_WRT_CMD_SIZE)
70 #define RM_FW_PAGE_SIZE		128
71 #define RM_MAX_FW_RETRIES	30
72 #define RM_MAX_FW_SIZE		0xD000
73 
74 #define RM_POWERON_DELAY_USEC	500
75 #define RM_RESET_DELAY_MSEC	50
76 
77 enum raydium_bl_cmd {
78 	BL_HEADER = 0,
79 	BL_PAGE_STR,
80 	BL_PKG_IDX,
81 	BL_DATA_STR,
82 };
83 
84 enum raydium_bl_ack {
85 	RAYDIUM_ACK_NULL = 0,
86 	RAYDIUM_WAIT_READY,
87 	RAYDIUM_PATH_READY,
88 };
89 
90 enum raydium_boot_mode {
91 	RAYDIUM_TS_MAIN = 0,
92 	RAYDIUM_TS_BLDR,
93 };
94 
95 /* Response to RM_CMD_DATA_BANK request */
96 struct raydium_data_info {
97 	__le32 data_bank_addr;
98 	u8 pkg_size;
99 	u8 tp_info_size;
100 };
101 
102 struct raydium_info {
103 	__le32 hw_ver;		/*device version */
104 	u8 main_ver;
105 	u8 sub_ver;
106 	__le16 ft_ver;		/* test version */
107 	u8 x_num;
108 	u8 y_num;
109 	__le16 x_max;
110 	__le16 y_max;
111 	u8 x_res;		/* units/mm */
112 	u8 y_res;		/* units/mm */
113 };
114 
115 /* struct raydium_data - represents state of Raydium touchscreen device */
116 struct raydium_data {
117 	struct i2c_client *client;
118 	struct input_dev *input;
119 
120 	struct regulator *avdd;
121 	struct regulator *vccio;
122 	struct gpio_desc *reset_gpio;
123 
124 	struct raydium_info info;
125 
126 	struct mutex sysfs_mutex;
127 
128 	u8 *report_data;
129 
130 	u32 data_bank_addr;
131 	u8 report_size;
132 	u8 contact_size;
133 	u8 pkg_size;
134 
135 	enum raydium_boot_mode boot_mode;
136 
137 	bool wake_irq_enabled;
138 };
139 
140 /*
141  * Header to be sent for RM_CMD_BANK_SWITCH command. This is used by
142  * raydium_i2c_{read|send} below.
143  */
144 struct __packed raydium_bank_switch_header {
145 	u8 cmd;
146 	__be32 be_addr;
147 };
148 
149 static int raydium_i2c_xfer(struct i2c_client *client, u32 addr,
150 			    struct i2c_msg *xfer, size_t xfer_count)
151 {
152 	int ret;
153 	/*
154 	 * If address is greater than 255, then RM_CMD_BANK_SWITCH needs to be
155 	 * sent first. Else, skip the header i.e. xfer[0].
156 	 */
157 	int xfer_start_idx = (addr > 0xff) ? 0 : 1;
158 	xfer_count -= xfer_start_idx;
159 
160 	ret = i2c_transfer(client->adapter, &xfer[xfer_start_idx], xfer_count);
161 	if (likely(ret == xfer_count))
162 		return 0;
163 
164 	return ret < 0 ? ret : -EIO;
165 }
166 
167 static int raydium_i2c_send(struct i2c_client *client,
168 			    u32 addr, const void *data, size_t len)
169 {
170 	int tries = 0;
171 	int error;
172 	u8 *tx_buf;
173 	u8 reg_addr = addr & 0xff;
174 
175 	tx_buf = kmalloc(len + 1, GFP_KERNEL);
176 	if (!tx_buf)
177 		return -ENOMEM;
178 
179 	tx_buf[0] = reg_addr;
180 	memcpy(tx_buf + 1, data, len);
181 
182 	do {
183 		struct raydium_bank_switch_header header = {
184 			.cmd = RM_CMD_BANK_SWITCH,
185 			.be_addr = cpu_to_be32(addr),
186 		};
187 
188 		/*
189 		 * Perform as a single i2c_transfer transaction to ensure that
190 		 * no other I2C transactions are initiated on the bus to any
191 		 * other device in between. Initiating transacations to other
192 		 * devices after RM_CMD_BANK_SWITCH is sent is known to cause
193 		 * issues. This is also why regmap infrastructure cannot be used
194 		 * for this driver. Regmap handles page(bank) switch and reads
195 		 * as separate i2c_transfer() operations. This can result in
196 		 * problems if the Raydium device is on a shared I2C bus.
197 		 */
198 		struct i2c_msg xfer[] = {
199 			{
200 				.addr = client->addr,
201 				.len = sizeof(header),
202 				.buf = (u8 *)&header,
203 			},
204 			{
205 				.addr = client->addr,
206 				.len = len + 1,
207 				.buf = tx_buf,
208 			},
209 		};
210 
211 		error = raydium_i2c_xfer(client, addr, xfer, ARRAY_SIZE(xfer));
212 		if (likely(!error))
213 			return 0;
214 
215 		msleep(RM_RETRY_DELAY_MS);
216 	} while (++tries < RM_MAX_RETRIES);
217 
218 	dev_err(&client->dev, "%s failed: %d\n", __func__, error);
219 	return error;
220 }
221 
222 static int raydium_i2c_read(struct i2c_client *client,
223 			    u32 addr, void *data, size_t len)
224 {
225 	int error;
226 
227 	while (len) {
228 		u8 reg_addr = addr & 0xff;
229 		struct raydium_bank_switch_header header = {
230 			.cmd = RM_CMD_BANK_SWITCH,
231 			.be_addr = cpu_to_be32(addr),
232 		};
233 		size_t xfer_len = min_t(size_t, len, RM_MAX_READ_SIZE);
234 
235 		/*
236 		 * Perform as a single i2c_transfer transaction to ensure that
237 		 * no other I2C transactions are initiated on the bus to any
238 		 * other device in between. Initiating transacations to other
239 		 * devices after RM_CMD_BANK_SWITCH is sent is known to cause
240 		 * issues. This is also why regmap infrastructure cannot be used
241 		 * for this driver. Regmap handles page(bank) switch and writes
242 		 * as separate i2c_transfer() operations. This can result in
243 		 * problems if the Raydium device is on a shared I2C bus.
244 		 */
245 		struct i2c_msg xfer[] = {
246 			{
247 				.addr = client->addr,
248 				.len = sizeof(header),
249 				.buf = (u8 *)&header,
250 			},
251 			{
252 				.addr = client->addr,
253 				.len = 1,
254 				.buf = &reg_addr,
255 			},
256 			{
257 				.addr = client->addr,
258 				.len = xfer_len,
259 				.buf = data,
260 				.flags = I2C_M_RD,
261 			}
262 		};
263 
264 		error = raydium_i2c_xfer(client, addr, xfer, ARRAY_SIZE(xfer));
265 		if (unlikely(error))
266 			return error;
267 
268 		len -= xfer_len;
269 		data += xfer_len;
270 		addr += xfer_len;
271 	}
272 
273 	return 0;
274 }
275 
276 static int raydium_i2c_sw_reset(struct i2c_client *client)
277 {
278 	const u8 soft_rst_cmd = 0x01;
279 	int error;
280 
281 	error = raydium_i2c_send(client, RM_RESET_MSG_ADDR, &soft_rst_cmd,
282 				 sizeof(soft_rst_cmd));
283 	if (error) {
284 		dev_err(&client->dev, "software reset failed: %d\n", error);
285 		return error;
286 	}
287 
288 	msleep(RM_RESET_DELAY_MSEC);
289 
290 	return 0;
291 }
292 
293 static int raydium_i2c_query_ts_info(struct raydium_data *ts)
294 {
295 	struct i2c_client *client = ts->client;
296 	struct raydium_data_info data_info;
297 	__le32 query_bank_addr;
298 
299 	int error, retry_cnt;
300 
301 	for (retry_cnt = 0; retry_cnt < RM_MAX_RETRIES; retry_cnt++) {
302 		error = raydium_i2c_read(client, RM_CMD_DATA_BANK,
303 					 &data_info, sizeof(data_info));
304 		if (error)
305 			continue;
306 
307 		/*
308 		 * Warn user if we already allocated memory for reports and
309 		 * then the size changed (due to firmware update?) and keep
310 		 * old size instead.
311 		 */
312 		if (ts->report_data && ts->pkg_size != data_info.pkg_size) {
313 			dev_warn(&client->dev,
314 				 "report size changes, was: %d, new: %d\n",
315 				 ts->pkg_size, data_info.pkg_size);
316 		} else {
317 			ts->pkg_size = data_info.pkg_size;
318 			ts->report_size = ts->pkg_size - RM_PACKET_CRC_SIZE;
319 		}
320 
321 		ts->contact_size = data_info.tp_info_size;
322 		ts->data_bank_addr = le32_to_cpu(data_info.data_bank_addr);
323 
324 		dev_dbg(&client->dev,
325 			"data_bank_addr: %#08x, report_size: %d, contact_size: %d\n",
326 			ts->data_bank_addr, ts->report_size, ts->contact_size);
327 
328 		error = raydium_i2c_read(client, RM_CMD_QUERY_BANK,
329 					 &query_bank_addr,
330 					 sizeof(query_bank_addr));
331 		if (error)
332 			continue;
333 
334 		error = raydium_i2c_read(client, le32_to_cpu(query_bank_addr),
335 					 &ts->info, sizeof(ts->info));
336 		if (error)
337 			continue;
338 
339 		return 0;
340 	}
341 
342 	dev_err(&client->dev, "failed to query device parameters: %d\n", error);
343 	return error;
344 }
345 
346 static int raydium_i2c_check_fw_status(struct raydium_data *ts)
347 {
348 	struct i2c_client *client = ts->client;
349 	static const u8 bl_ack = 0x62;
350 	static const u8 main_ack = 0x66;
351 	u8 buf[4];
352 	int error;
353 
354 	error = raydium_i2c_read(client, RM_CMD_BOOT_READ, buf, sizeof(buf));
355 	if (!error) {
356 		if (buf[0] == bl_ack)
357 			ts->boot_mode = RAYDIUM_TS_BLDR;
358 		else if (buf[0] == main_ack)
359 			ts->boot_mode = RAYDIUM_TS_MAIN;
360 		return 0;
361 	}
362 
363 	return error;
364 }
365 
366 static int raydium_i2c_initialize(struct raydium_data *ts)
367 {
368 	struct i2c_client *client = ts->client;
369 	int error, retry_cnt;
370 
371 	for (retry_cnt = 0; retry_cnt < RM_MAX_RETRIES; retry_cnt++) {
372 		/* Wait for Hello packet */
373 		msleep(RM_BOOT_DELAY_MS);
374 
375 		error = raydium_i2c_check_fw_status(ts);
376 		if (error) {
377 			dev_err(&client->dev,
378 				"failed to read 'hello' packet: %d\n", error);
379 			continue;
380 		}
381 
382 		if (ts->boot_mode == RAYDIUM_TS_BLDR ||
383 		    ts->boot_mode == RAYDIUM_TS_MAIN) {
384 			break;
385 		}
386 	}
387 
388 	if (error)
389 		ts->boot_mode = RAYDIUM_TS_BLDR;
390 
391 	if (ts->boot_mode == RAYDIUM_TS_BLDR) {
392 		ts->info.hw_ver = cpu_to_le32(0xffffffffUL);
393 		ts->info.main_ver = 0xff;
394 		ts->info.sub_ver = 0xff;
395 	} else {
396 		raydium_i2c_query_ts_info(ts);
397 	}
398 
399 	return error;
400 }
401 
402 static int raydium_i2c_bl_chk_state(struct i2c_client *client,
403 				    enum raydium_bl_ack state)
404 {
405 	static const u8 ack_ok[] = { 0xFF, 0x39, 0x30, 0x30, 0x54 };
406 	u8 rbuf[sizeof(ack_ok)];
407 	u8 retry;
408 	int error;
409 
410 	for (retry = 0; retry < RM_MAX_FW_RETRIES; retry++) {
411 		switch (state) {
412 		case RAYDIUM_ACK_NULL:
413 			return 0;
414 
415 		case RAYDIUM_WAIT_READY:
416 			error = raydium_i2c_read(client, RM_CMD_BOOT_CHK,
417 						 &rbuf[0], 1);
418 			if (!error && rbuf[0] == RM_BOOT_RDY)
419 				return 0;
420 
421 			break;
422 
423 		case RAYDIUM_PATH_READY:
424 			error = raydium_i2c_read(client, RM_CMD_BOOT_CHK,
425 						 rbuf, sizeof(rbuf));
426 			if (!error && !memcmp(rbuf, ack_ok, sizeof(ack_ok)))
427 				return 0;
428 
429 			break;
430 
431 		default:
432 			dev_err(&client->dev, "%s: invalid target state %d\n",
433 				__func__, state);
434 			return -EINVAL;
435 		}
436 
437 		msleep(20);
438 	}
439 
440 	return -ETIMEDOUT;
441 }
442 
443 static int raydium_i2c_write_object(struct i2c_client *client,
444 				    const void *data, size_t len,
445 				    enum raydium_bl_ack state)
446 {
447 	int error;
448 
449 	error = raydium_i2c_send(client, RM_CMD_BOOT_WRT, data, len);
450 	if (error) {
451 		dev_err(&client->dev, "WRT obj command failed: %d\n",
452 			error);
453 		return error;
454 	}
455 
456 	error = raydium_i2c_send(client, RM_CMD_BOOT_ACK, NULL, 0);
457 	if (error) {
458 		dev_err(&client->dev, "Ack obj command failed: %d\n", error);
459 		return error;
460 	}
461 
462 	error = raydium_i2c_bl_chk_state(client, state);
463 	if (error) {
464 		dev_err(&client->dev, "BL check state failed: %d\n", error);
465 		return error;
466 	}
467 	return 0;
468 }
469 
470 static int raydium_i2c_boot_trigger(struct i2c_client *client)
471 {
472 	static const u8 cmd[7][6] = {
473 		{ 0x08, 0x0C, 0x09, 0x00, 0x50, 0xD7 },
474 		{ 0x08, 0x04, 0x09, 0x00, 0x50, 0xA5 },
475 		{ 0x08, 0x04, 0x09, 0x00, 0x50, 0x00 },
476 		{ 0x08, 0x04, 0x09, 0x00, 0x50, 0xA5 },
477 		{ 0x08, 0x0C, 0x09, 0x00, 0x50, 0x00 },
478 		{ 0x06, 0x01, 0x00, 0x00, 0x00, 0x00 },
479 		{ 0x02, 0xA2, 0x00, 0x00, 0x00, 0x00 },
480 	};
481 	int i;
482 	int error;
483 
484 	for (i = 0; i < 7; i++) {
485 		error = raydium_i2c_write_object(client, cmd[i], sizeof(cmd[i]),
486 						 RAYDIUM_WAIT_READY);
487 		if (error) {
488 			dev_err(&client->dev,
489 				"boot trigger failed at step %d: %d\n",
490 				i, error);
491 			return error;
492 		}
493 	}
494 
495 	return 0;
496 }
497 
498 static int raydium_i2c_fw_trigger(struct i2c_client *client)
499 {
500 	static const u8 cmd[5][11] = {
501 		{ 0, 0x09, 0x71, 0x0C, 0x09, 0x00, 0x50, 0xD7, 0, 0, 0 },
502 		{ 0, 0x09, 0x71, 0x04, 0x09, 0x00, 0x50, 0xA5, 0, 0, 0 },
503 		{ 0, 0x09, 0x71, 0x04, 0x09, 0x00, 0x50, 0x00, 0, 0, 0 },
504 		{ 0, 0x09, 0x71, 0x04, 0x09, 0x00, 0x50, 0xA5, 0, 0, 0 },
505 		{ 0, 0x09, 0x71, 0x0C, 0x09, 0x00, 0x50, 0x00, 0, 0, 0 },
506 	};
507 	int i;
508 	int error;
509 
510 	for (i = 0; i < 5; i++) {
511 		error = raydium_i2c_write_object(client, cmd[i], sizeof(cmd[i]),
512 						 RAYDIUM_ACK_NULL);
513 		if (error) {
514 			dev_err(&client->dev,
515 				"fw trigger failed at step %d: %d\n",
516 				i, error);
517 			return error;
518 		}
519 	}
520 
521 	return 0;
522 }
523 
524 static int raydium_i2c_check_path(struct i2c_client *client)
525 {
526 	static const u8 cmd[] = { 0x09, 0x00, 0x09, 0x00, 0x50, 0x10, 0x00 };
527 	int error;
528 
529 	error = raydium_i2c_write_object(client, cmd, sizeof(cmd),
530 					 RAYDIUM_PATH_READY);
531 	if (error) {
532 		dev_err(&client->dev, "check path command failed: %d\n", error);
533 		return error;
534 	}
535 
536 	return 0;
537 }
538 
539 static int raydium_i2c_enter_bl(struct i2c_client *client)
540 {
541 	static const u8 cal_cmd[] = { 0x00, 0x01, 0x52 };
542 	int error;
543 
544 	error = raydium_i2c_write_object(client, cal_cmd, sizeof(cal_cmd),
545 					 RAYDIUM_ACK_NULL);
546 	if (error) {
547 		dev_err(&client->dev, "enter bl command failed: %d\n", error);
548 		return error;
549 	}
550 
551 	msleep(RM_BOOT_DELAY_MS);
552 	return 0;
553 }
554 
555 static int raydium_i2c_leave_bl(struct i2c_client *client)
556 {
557 	static const u8 leave_cmd[] = { 0x05, 0x00 };
558 	int error;
559 
560 	error = raydium_i2c_write_object(client, leave_cmd, sizeof(leave_cmd),
561 					 RAYDIUM_ACK_NULL);
562 	if (error) {
563 		dev_err(&client->dev, "leave bl command failed: %d\n", error);
564 		return error;
565 	}
566 
567 	msleep(RM_BOOT_DELAY_MS);
568 	return 0;
569 }
570 
571 static int raydium_i2c_write_checksum(struct i2c_client *client,
572 				      size_t length, u16 checksum)
573 {
574 	u8 checksum_cmd[] = { 0x00, 0x05, 0x6D, 0x00, 0x00, 0x00, 0x00 };
575 	int error;
576 
577 	put_unaligned_le16(length, &checksum_cmd[3]);
578 	put_unaligned_le16(checksum, &checksum_cmd[5]);
579 
580 	error = raydium_i2c_write_object(client,
581 					 checksum_cmd, sizeof(checksum_cmd),
582 					 RAYDIUM_ACK_NULL);
583 	if (error) {
584 		dev_err(&client->dev, "failed to write checksum: %d\n",
585 			error);
586 		return error;
587 	}
588 
589 	return 0;
590 }
591 
592 static int raydium_i2c_disable_watch_dog(struct i2c_client *client)
593 {
594 	static const u8 cmd[] = { 0x0A, 0xAA };
595 	int error;
596 
597 	error = raydium_i2c_write_object(client, cmd, sizeof(cmd),
598 					 RAYDIUM_WAIT_READY);
599 	if (error) {
600 		dev_err(&client->dev, "disable watchdog command failed: %d\n",
601 			error);
602 		return error;
603 	}
604 
605 	return 0;
606 }
607 
608 static int raydium_i2c_fw_write_page(struct i2c_client *client,
609 				     u16 page_idx, const void *data, size_t len)
610 {
611 	u8 buf[RM_BL_WRT_LEN];
612 	size_t xfer_len;
613 	int error;
614 	int i;
615 
616 	BUILD_BUG_ON((RM_FW_PAGE_SIZE % RM_BL_WRT_PKG_SIZE) != 0);
617 
618 	for (i = 0; i < RM_FW_PAGE_SIZE / RM_BL_WRT_PKG_SIZE; i++) {
619 		buf[BL_HEADER] = RM_CMD_BOOT_PAGE_WRT;
620 		buf[BL_PAGE_STR] = page_idx ? 0xff : 0;
621 		buf[BL_PKG_IDX] = i + 1;
622 
623 		xfer_len = min_t(size_t, len, RM_BL_WRT_PKG_SIZE);
624 		memcpy(&buf[BL_DATA_STR], data, xfer_len);
625 		if (len < RM_BL_WRT_PKG_SIZE)
626 			memset(&buf[BL_DATA_STR + xfer_len], 0xff,
627 				RM_BL_WRT_PKG_SIZE - xfer_len);
628 
629 		error = raydium_i2c_write_object(client, buf, RM_BL_WRT_LEN,
630 						 RAYDIUM_WAIT_READY);
631 		if (error) {
632 			dev_err(&client->dev,
633 				"page write command failed for page %d, chunk %d: %d\n",
634 				page_idx, i, error);
635 			return error;
636 		}
637 
638 		data += xfer_len;
639 		len -= xfer_len;
640 	}
641 
642 	return error;
643 }
644 
645 static u16 raydium_calc_chksum(const u8 *buf, u16 len)
646 {
647 	u16 checksum = 0;
648 	u16 i;
649 
650 	for (i = 0; i < len; i++)
651 		checksum += buf[i];
652 
653 	return checksum;
654 }
655 
656 static int raydium_i2c_do_update_firmware(struct raydium_data *ts,
657 					 const struct firmware *fw)
658 {
659 	struct i2c_client *client = ts->client;
660 	const void *data;
661 	size_t data_len;
662 	size_t len;
663 	int page_nr;
664 	int i;
665 	int error;
666 	u16 fw_checksum;
667 
668 	if (fw->size == 0 || fw->size > RM_MAX_FW_SIZE) {
669 		dev_err(&client->dev, "Invalid firmware length\n");
670 		return -EINVAL;
671 	}
672 
673 	error = raydium_i2c_check_fw_status(ts);
674 	if (error) {
675 		dev_err(&client->dev, "Unable to access IC %d\n", error);
676 		return error;
677 	}
678 
679 	if (ts->boot_mode == RAYDIUM_TS_MAIN) {
680 		for (i = 0; i < RM_MAX_RETRIES; i++) {
681 			error = raydium_i2c_enter_bl(client);
682 			if (!error) {
683 				error = raydium_i2c_check_fw_status(ts);
684 				if (error) {
685 					dev_err(&client->dev,
686 						"unable to access IC: %d\n",
687 						error);
688 					return error;
689 				}
690 
691 				if (ts->boot_mode == RAYDIUM_TS_BLDR)
692 					break;
693 			}
694 		}
695 
696 		if (ts->boot_mode == RAYDIUM_TS_MAIN) {
697 			dev_err(&client->dev,
698 				"failed to jump to boot loader: %d\n",
699 				error);
700 			return -EIO;
701 		}
702 	}
703 
704 	error = raydium_i2c_disable_watch_dog(client);
705 	if (error)
706 		return error;
707 
708 	error = raydium_i2c_check_path(client);
709 	if (error)
710 		return error;
711 
712 	error = raydium_i2c_boot_trigger(client);
713 	if (error) {
714 		dev_err(&client->dev, "send boot trigger fail: %d\n", error);
715 		return error;
716 	}
717 
718 	msleep(RM_BOOT_DELAY_MS);
719 
720 	data = fw->data;
721 	data_len = fw->size;
722 	page_nr = 0;
723 
724 	while (data_len) {
725 		len = min_t(size_t, data_len, RM_FW_PAGE_SIZE);
726 
727 		error = raydium_i2c_fw_write_page(client, page_nr++, data, len);
728 		if (error)
729 			return error;
730 
731 		msleep(20);
732 
733 		data += len;
734 		data_len -= len;
735 	}
736 
737 	error = raydium_i2c_leave_bl(client);
738 	if (error) {
739 		dev_err(&client->dev,
740 			"failed to leave boot loader: %d\n", error);
741 		return error;
742 	}
743 
744 	dev_dbg(&client->dev, "left boot loader mode\n");
745 	msleep(RM_BOOT_DELAY_MS);
746 
747 	error = raydium_i2c_check_fw_status(ts);
748 	if (error) {
749 		dev_err(&client->dev,
750 			"failed to check fw status after write: %d\n",
751 			error);
752 		return error;
753 	}
754 
755 	if (ts->boot_mode != RAYDIUM_TS_MAIN) {
756 		dev_err(&client->dev,
757 			"failed to switch to main fw after writing firmware: %d\n",
758 			error);
759 		return -EINVAL;
760 	}
761 
762 	error = raydium_i2c_fw_trigger(client);
763 	if (error) {
764 		dev_err(&client->dev, "failed to trigger fw: %d\n", error);
765 		return error;
766 	}
767 
768 	fw_checksum = raydium_calc_chksum(fw->data, fw->size);
769 
770 	error = raydium_i2c_write_checksum(client, fw->size, fw_checksum);
771 	if (error)
772 		return error;
773 
774 	return 0;
775 }
776 
777 static int raydium_i2c_fw_update(struct raydium_data *ts)
778 {
779 	struct i2c_client *client = ts->client;
780 	const struct firmware *fw = NULL;
781 	char *fw_file;
782 	int error;
783 
784 	fw_file = kasprintf(GFP_KERNEL, "raydium_%#04x.fw",
785 			    le32_to_cpu(ts->info.hw_ver));
786 	if (!fw_file)
787 		return -ENOMEM;
788 
789 	dev_dbg(&client->dev, "firmware name: %s\n", fw_file);
790 
791 	error = request_firmware(&fw, fw_file, &client->dev);
792 	if (error) {
793 		dev_err(&client->dev, "Unable to open firmware %s\n", fw_file);
794 		goto out_free_fw_file;
795 	}
796 
797 	disable_irq(client->irq);
798 
799 	error = raydium_i2c_do_update_firmware(ts, fw);
800 	if (error) {
801 		dev_err(&client->dev, "firmware update failed: %d\n", error);
802 		ts->boot_mode = RAYDIUM_TS_BLDR;
803 		goto out_enable_irq;
804 	}
805 
806 	error = raydium_i2c_initialize(ts);
807 	if (error) {
808 		dev_err(&client->dev,
809 			"failed to initialize device after firmware update: %d\n",
810 			error);
811 		ts->boot_mode = RAYDIUM_TS_BLDR;
812 		goto out_enable_irq;
813 	}
814 
815 	ts->boot_mode = RAYDIUM_TS_MAIN;
816 
817 out_enable_irq:
818 	enable_irq(client->irq);
819 	msleep(100);
820 
821 	release_firmware(fw);
822 
823 out_free_fw_file:
824 	kfree(fw_file);
825 
826 	return error;
827 }
828 
829 static void raydium_mt_event(struct raydium_data *ts)
830 {
831 	int i;
832 
833 	for (i = 0; i < ts->report_size / ts->contact_size; i++) {
834 		u8 *contact = &ts->report_data[ts->contact_size * i];
835 		bool state = contact[RM_CONTACT_STATE_POS];
836 		u8 wx, wy;
837 
838 		input_mt_slot(ts->input, i);
839 		input_mt_report_slot_state(ts->input, MT_TOOL_FINGER, state);
840 
841 		if (!state)
842 			continue;
843 
844 		input_report_abs(ts->input, ABS_MT_POSITION_X,
845 				get_unaligned_le16(&contact[RM_CONTACT_X_POS]));
846 		input_report_abs(ts->input, ABS_MT_POSITION_Y,
847 				get_unaligned_le16(&contact[RM_CONTACT_Y_POS]));
848 		input_report_abs(ts->input, ABS_MT_PRESSURE,
849 				contact[RM_CONTACT_PRESSURE_POS]);
850 
851 		wx = contact[RM_CONTACT_WIDTH_X_POS];
852 		wy = contact[RM_CONTACT_WIDTH_Y_POS];
853 
854 		input_report_abs(ts->input, ABS_MT_TOUCH_MAJOR, max(wx, wy));
855 		input_report_abs(ts->input, ABS_MT_TOUCH_MINOR, min(wx, wy));
856 	}
857 
858 	input_mt_sync_frame(ts->input);
859 	input_sync(ts->input);
860 }
861 
862 static irqreturn_t raydium_i2c_irq(int irq, void *_dev)
863 {
864 	struct raydium_data *ts = _dev;
865 	int error;
866 	u16 fw_crc;
867 	u16 calc_crc;
868 
869 	if (ts->boot_mode != RAYDIUM_TS_MAIN)
870 		goto out;
871 
872 	error = raydium_i2c_read(ts->client, ts->data_bank_addr,
873 				 ts->report_data, ts->pkg_size);
874 	if (error)
875 		goto out;
876 
877 	fw_crc = get_unaligned_le16(&ts->report_data[ts->report_size]);
878 	calc_crc = raydium_calc_chksum(ts->report_data, ts->report_size);
879 	if (unlikely(fw_crc != calc_crc)) {
880 		dev_warn(&ts->client->dev,
881 			 "%s: invalid packet crc %#04x vs %#04x\n",
882 			 __func__, calc_crc, fw_crc);
883 		goto out;
884 	}
885 
886 	raydium_mt_event(ts);
887 
888 out:
889 	return IRQ_HANDLED;
890 }
891 
892 static ssize_t raydium_i2c_fw_ver_show(struct device *dev,
893 				       struct device_attribute *attr, char *buf)
894 {
895 	struct i2c_client *client = to_i2c_client(dev);
896 	struct raydium_data *ts = i2c_get_clientdata(client);
897 
898 	return sprintf(buf, "%d.%d\n", ts->info.main_ver, ts->info.sub_ver);
899 }
900 
901 static ssize_t raydium_i2c_hw_ver_show(struct device *dev,
902 				       struct device_attribute *attr, char *buf)
903 {
904 	struct i2c_client *client = to_i2c_client(dev);
905 	struct raydium_data *ts = i2c_get_clientdata(client);
906 
907 	return sprintf(buf, "%#04x\n", le32_to_cpu(ts->info.hw_ver));
908 }
909 
910 static ssize_t raydium_i2c_boot_mode_show(struct device *dev,
911 					  struct device_attribute *attr,
912 					  char *buf)
913 {
914 	struct i2c_client *client = to_i2c_client(dev);
915 	struct raydium_data *ts = i2c_get_clientdata(client);
916 
917 	return sprintf(buf, "%s\n",
918 		       ts->boot_mode == RAYDIUM_TS_MAIN ?
919 				"Normal" : "Recovery");
920 }
921 
922 static ssize_t raydium_i2c_update_fw_store(struct device *dev,
923 					   struct device_attribute *attr,
924 					   const char *buf, size_t count)
925 {
926 	struct i2c_client *client = to_i2c_client(dev);
927 	struct raydium_data *ts = i2c_get_clientdata(client);
928 	int error;
929 
930 	error = mutex_lock_interruptible(&ts->sysfs_mutex);
931 	if (error)
932 		return error;
933 
934 	error = raydium_i2c_fw_update(ts);
935 
936 	mutex_unlock(&ts->sysfs_mutex);
937 
938 	return error ?: count;
939 }
940 
941 static ssize_t raydium_i2c_calibrate_store(struct device *dev,
942 					   struct device_attribute *attr,
943 					   const char *buf, size_t count)
944 {
945 	struct i2c_client *client = to_i2c_client(dev);
946 	struct raydium_data *ts = i2c_get_clientdata(client);
947 	static const u8 cal_cmd[] = { 0x00, 0x01, 0x9E };
948 	int error;
949 
950 	error = mutex_lock_interruptible(&ts->sysfs_mutex);
951 	if (error)
952 		return error;
953 
954 	error = raydium_i2c_write_object(client, cal_cmd, sizeof(cal_cmd),
955 					 RAYDIUM_WAIT_READY);
956 	if (error)
957 		dev_err(&client->dev, "calibrate command failed: %d\n", error);
958 
959 	mutex_unlock(&ts->sysfs_mutex);
960 	return error ?: count;
961 }
962 
963 static DEVICE_ATTR(fw_version, S_IRUGO, raydium_i2c_fw_ver_show, NULL);
964 static DEVICE_ATTR(hw_version, S_IRUGO, raydium_i2c_hw_ver_show, NULL);
965 static DEVICE_ATTR(boot_mode, S_IRUGO, raydium_i2c_boot_mode_show, NULL);
966 static DEVICE_ATTR(update_fw, S_IWUSR, NULL, raydium_i2c_update_fw_store);
967 static DEVICE_ATTR(calibrate, S_IWUSR, NULL, raydium_i2c_calibrate_store);
968 
969 static struct attribute *raydium_i2c_attributes[] = {
970 	&dev_attr_update_fw.attr,
971 	&dev_attr_boot_mode.attr,
972 	&dev_attr_fw_version.attr,
973 	&dev_attr_hw_version.attr,
974 	&dev_attr_calibrate.attr,
975 	NULL
976 };
977 
978 static const struct attribute_group raydium_i2c_attribute_group = {
979 	.attrs = raydium_i2c_attributes,
980 };
981 
982 static int raydium_i2c_power_on(struct raydium_data *ts)
983 {
984 	int error;
985 
986 	if (!ts->reset_gpio)
987 		return 0;
988 
989 	gpiod_set_value_cansleep(ts->reset_gpio, 1);
990 
991 	error = regulator_enable(ts->avdd);
992 	if (error) {
993 		dev_err(&ts->client->dev,
994 			"failed to enable avdd regulator: %d\n", error);
995 		goto release_reset_gpio;
996 	}
997 
998 	error = regulator_enable(ts->vccio);
999 	if (error) {
1000 		regulator_disable(ts->avdd);
1001 		dev_err(&ts->client->dev,
1002 			"failed to enable vccio regulator: %d\n", error);
1003 		goto release_reset_gpio;
1004 	}
1005 
1006 	udelay(RM_POWERON_DELAY_USEC);
1007 
1008 release_reset_gpio:
1009 	gpiod_set_value_cansleep(ts->reset_gpio, 0);
1010 
1011 	if (error)
1012 		return error;
1013 
1014 	msleep(RM_RESET_DELAY_MSEC);
1015 
1016 	return 0;
1017 }
1018 
1019 static void raydium_i2c_power_off(void *_data)
1020 {
1021 	struct raydium_data *ts = _data;
1022 
1023 	if (ts->reset_gpio) {
1024 		gpiod_set_value_cansleep(ts->reset_gpio, 1);
1025 		regulator_disable(ts->vccio);
1026 		regulator_disable(ts->avdd);
1027 	}
1028 }
1029 
1030 static int raydium_i2c_probe(struct i2c_client *client,
1031 			     const struct i2c_device_id *id)
1032 {
1033 	union i2c_smbus_data dummy;
1034 	struct raydium_data *ts;
1035 	int error;
1036 
1037 	if (!i2c_check_functionality(client->adapter, I2C_FUNC_I2C)) {
1038 		dev_err(&client->dev,
1039 			"i2c check functionality error (need I2C_FUNC_I2C)\n");
1040 		return -ENXIO;
1041 	}
1042 
1043 	ts = devm_kzalloc(&client->dev, sizeof(*ts), GFP_KERNEL);
1044 	if (!ts)
1045 		return -ENOMEM;
1046 
1047 	mutex_init(&ts->sysfs_mutex);
1048 
1049 	ts->client = client;
1050 	i2c_set_clientdata(client, ts);
1051 
1052 	ts->avdd = devm_regulator_get(&client->dev, "avdd");
1053 	if (IS_ERR(ts->avdd)) {
1054 		error = PTR_ERR(ts->avdd);
1055 		if (error != -EPROBE_DEFER)
1056 			dev_err(&client->dev,
1057 				"Failed to get 'avdd' regulator: %d\n", error);
1058 		return error;
1059 	}
1060 
1061 	ts->vccio = devm_regulator_get(&client->dev, "vccio");
1062 	if (IS_ERR(ts->vccio)) {
1063 		error = PTR_ERR(ts->vccio);
1064 		if (error != -EPROBE_DEFER)
1065 			dev_err(&client->dev,
1066 				"Failed to get 'vccio' regulator: %d\n", error);
1067 		return error;
1068 	}
1069 
1070 	ts->reset_gpio = devm_gpiod_get_optional(&client->dev, "reset",
1071 						 GPIOD_OUT_LOW);
1072 	if (IS_ERR(ts->reset_gpio)) {
1073 		error = PTR_ERR(ts->reset_gpio);
1074 		if (error != -EPROBE_DEFER)
1075 			dev_err(&client->dev,
1076 				"failed to get reset gpio: %d\n", error);
1077 		return error;
1078 	}
1079 
1080 	error = raydium_i2c_power_on(ts);
1081 	if (error)
1082 		return error;
1083 
1084 	error = devm_add_action(&client->dev, raydium_i2c_power_off, ts);
1085 	if (error) {
1086 		dev_err(&client->dev,
1087 			"failed to install power off action: %d\n", error);
1088 		raydium_i2c_power_off(ts);
1089 		return error;
1090 	}
1091 
1092 	/* Make sure there is something at this address */
1093 	if (i2c_smbus_xfer(client->adapter, client->addr, 0,
1094 			   I2C_SMBUS_READ, 0, I2C_SMBUS_BYTE, &dummy) < 0) {
1095 		dev_err(&client->dev, "nothing at this address\n");
1096 		return -ENXIO;
1097 	}
1098 
1099 	error = raydium_i2c_initialize(ts);
1100 	if (error) {
1101 		dev_err(&client->dev, "failed to initialize: %d\n", error);
1102 		return error;
1103 	}
1104 
1105 	ts->report_data = devm_kmalloc(&client->dev,
1106 				       ts->pkg_size, GFP_KERNEL);
1107 	if (!ts->report_data)
1108 		return -ENOMEM;
1109 
1110 	ts->input = devm_input_allocate_device(&client->dev);
1111 	if (!ts->input) {
1112 		dev_err(&client->dev, "Failed to allocate input device\n");
1113 		return -ENOMEM;
1114 	}
1115 
1116 	ts->input->name = "Raydium Touchscreen";
1117 	ts->input->id.bustype = BUS_I2C;
1118 
1119 	input_set_abs_params(ts->input, ABS_MT_POSITION_X,
1120 			     0, le16_to_cpu(ts->info.x_max), 0, 0);
1121 	input_set_abs_params(ts->input, ABS_MT_POSITION_Y,
1122 			     0, le16_to_cpu(ts->info.y_max), 0, 0);
1123 	input_abs_set_res(ts->input, ABS_MT_POSITION_X, ts->info.x_res);
1124 	input_abs_set_res(ts->input, ABS_MT_POSITION_Y, ts->info.y_res);
1125 
1126 	input_set_abs_params(ts->input, ABS_MT_TOUCH_MAJOR, 0, 255, 0, 0);
1127 	input_set_abs_params(ts->input, ABS_MT_PRESSURE, 0, 255, 0, 0);
1128 
1129 	error = input_mt_init_slots(ts->input, RM_MAX_TOUCH_NUM,
1130 				    INPUT_MT_DIRECT | INPUT_MT_DROP_UNUSED);
1131 	if (error) {
1132 		dev_err(&client->dev,
1133 			"failed to initialize MT slots: %d\n", error);
1134 		return error;
1135 	}
1136 
1137 	error = input_register_device(ts->input);
1138 	if (error) {
1139 		dev_err(&client->dev,
1140 			"unable to register input device: %d\n", error);
1141 		return error;
1142 	}
1143 
1144 	error = devm_request_threaded_irq(&client->dev, client->irq,
1145 					  NULL, raydium_i2c_irq,
1146 					  IRQF_ONESHOT, client->name, ts);
1147 	if (error) {
1148 		dev_err(&client->dev, "Failed to register interrupt\n");
1149 		return error;
1150 	}
1151 
1152 	error = devm_device_add_group(&client->dev,
1153 				   &raydium_i2c_attribute_group);
1154 	if (error) {
1155 		dev_err(&client->dev, "failed to create sysfs attributes: %d\n",
1156 			error);
1157 		return error;
1158 	}
1159 
1160 	return 0;
1161 }
1162 
1163 static void __maybe_unused raydium_enter_sleep(struct i2c_client *client)
1164 {
1165 	static const u8 sleep_cmd[] = { 0x5A, 0xff, 0x00, 0x0f };
1166 	int error;
1167 
1168 	error = raydium_i2c_send(client, RM_CMD_ENTER_SLEEP,
1169 				 sleep_cmd, sizeof(sleep_cmd));
1170 	if (error)
1171 		dev_err(&client->dev,
1172 			"sleep command failed: %d\n", error);
1173 }
1174 
1175 static int __maybe_unused raydium_i2c_suspend(struct device *dev)
1176 {
1177 	struct i2c_client *client = to_i2c_client(dev);
1178 	struct raydium_data *ts = i2c_get_clientdata(client);
1179 
1180 	/* Sleep is not available in BLDR recovery mode */
1181 	if (ts->boot_mode != RAYDIUM_TS_MAIN)
1182 		return -EBUSY;
1183 
1184 	disable_irq(client->irq);
1185 
1186 	if (device_may_wakeup(dev)) {
1187 		raydium_enter_sleep(client);
1188 
1189 		ts->wake_irq_enabled = (enable_irq_wake(client->irq) == 0);
1190 	} else {
1191 		raydium_i2c_power_off(ts);
1192 	}
1193 
1194 	return 0;
1195 }
1196 
1197 static int __maybe_unused raydium_i2c_resume(struct device *dev)
1198 {
1199 	struct i2c_client *client = to_i2c_client(dev);
1200 	struct raydium_data *ts = i2c_get_clientdata(client);
1201 
1202 	if (device_may_wakeup(dev)) {
1203 		if (ts->wake_irq_enabled)
1204 			disable_irq_wake(client->irq);
1205 		raydium_i2c_sw_reset(client);
1206 	} else {
1207 		raydium_i2c_power_on(ts);
1208 		raydium_i2c_initialize(ts);
1209 	}
1210 
1211 	enable_irq(client->irq);
1212 
1213 	return 0;
1214 }
1215 
1216 static SIMPLE_DEV_PM_OPS(raydium_i2c_pm_ops,
1217 			 raydium_i2c_suspend, raydium_i2c_resume);
1218 
1219 static const struct i2c_device_id raydium_i2c_id[] = {
1220 	{ "raydium_i2c" , 0 },
1221 	{ "rm32380", 0 },
1222 	{ /* sentinel */ }
1223 };
1224 MODULE_DEVICE_TABLE(i2c, raydium_i2c_id);
1225 
1226 #ifdef CONFIG_ACPI
1227 static const struct acpi_device_id raydium_acpi_id[] = {
1228 	{ "RAYD0001", 0 },
1229 	{ /* sentinel */ }
1230 };
1231 MODULE_DEVICE_TABLE(acpi, raydium_acpi_id);
1232 #endif
1233 
1234 #ifdef CONFIG_OF
1235 static const struct of_device_id raydium_of_match[] = {
1236 	{ .compatible = "raydium,rm32380", },
1237 	{ /* sentinel */ }
1238 };
1239 MODULE_DEVICE_TABLE(of, raydium_of_match);
1240 #endif
1241 
1242 static struct i2c_driver raydium_i2c_driver = {
1243 	.probe = raydium_i2c_probe,
1244 	.id_table = raydium_i2c_id,
1245 	.driver = {
1246 		.name = "raydium_ts",
1247 		.pm = &raydium_i2c_pm_ops,
1248 		.acpi_match_table = ACPI_PTR(raydium_acpi_id),
1249 		.of_match_table = of_match_ptr(raydium_of_match),
1250 	},
1251 };
1252 module_i2c_driver(raydium_i2c_driver);
1253 
1254 MODULE_AUTHOR("Raydium");
1255 MODULE_DESCRIPTION("Raydium I2c Touchscreen driver");
1256 MODULE_LICENSE("GPL v2");
1257