xref: /OK3568_Linux_fs/kernel/drivers/input/mouse/cyapa_gen3.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun  * Cypress APA trackpad with I2C interface
3*4882a593Smuzhiyun  *
4*4882a593Smuzhiyun  * Author: Dudley Du <dudl@cypress.com>
5*4882a593Smuzhiyun  * Further cleanup and restructuring by:
6*4882a593Smuzhiyun  *   Daniel Kurtz <djkurtz@chromium.org>
7*4882a593Smuzhiyun  *   Benson Leung <bleung@chromium.org>
8*4882a593Smuzhiyun  *
9*4882a593Smuzhiyun  * Copyright (C) 2011-2015 Cypress Semiconductor, Inc.
10*4882a593Smuzhiyun  * Copyright (C) 2011-2012 Google, Inc.
11*4882a593Smuzhiyun  *
12*4882a593Smuzhiyun  * This file is subject to the terms and conditions of the GNU General Public
13*4882a593Smuzhiyun  * License.  See the file COPYING in the main directory of this archive for
14*4882a593Smuzhiyun  * more details.
15*4882a593Smuzhiyun  */
16*4882a593Smuzhiyun 
17*4882a593Smuzhiyun #include <linux/delay.h>
18*4882a593Smuzhiyun #include <linux/i2c.h>
19*4882a593Smuzhiyun #include <linux/input.h>
20*4882a593Smuzhiyun #include <linux/input/mt.h>
21*4882a593Smuzhiyun #include <linux/module.h>
22*4882a593Smuzhiyun #include <linux/slab.h>
23*4882a593Smuzhiyun #include <asm/unaligned.h>
24*4882a593Smuzhiyun #include "cyapa.h"
25*4882a593Smuzhiyun 
26*4882a593Smuzhiyun 
27*4882a593Smuzhiyun #define GEN3_MAX_FINGERS 5
28*4882a593Smuzhiyun #define GEN3_FINGER_NUM(x) (((x) >> 4) & 0x07)
29*4882a593Smuzhiyun 
30*4882a593Smuzhiyun #define BLK_HEAD_BYTES 32
31*4882a593Smuzhiyun 
32*4882a593Smuzhiyun /* Macro for register map group offset. */
33*4882a593Smuzhiyun #define PRODUCT_ID_SIZE  16
34*4882a593Smuzhiyun #define QUERY_DATA_SIZE  27
35*4882a593Smuzhiyun #define REG_PROTOCOL_GEN_QUERY_OFFSET  20
36*4882a593Smuzhiyun 
37*4882a593Smuzhiyun #define REG_OFFSET_DATA_BASE     0x0000
38*4882a593Smuzhiyun #define REG_OFFSET_COMMAND_BASE  0x0028
39*4882a593Smuzhiyun #define REG_OFFSET_QUERY_BASE    0x002a
40*4882a593Smuzhiyun 
41*4882a593Smuzhiyun #define CYAPA_OFFSET_SOFT_RESET  REG_OFFSET_COMMAND_BASE
42*4882a593Smuzhiyun #define OP_RECALIBRATION_MASK    0x80
43*4882a593Smuzhiyun #define OP_REPORT_BASELINE_MASK  0x40
44*4882a593Smuzhiyun #define REG_OFFSET_MAX_BASELINE  0x0026
45*4882a593Smuzhiyun #define REG_OFFSET_MIN_BASELINE  0x0027
46*4882a593Smuzhiyun 
47*4882a593Smuzhiyun #define REG_OFFSET_POWER_MODE (REG_OFFSET_COMMAND_BASE + 1)
48*4882a593Smuzhiyun #define SET_POWER_MODE_DELAY   10000  /* Unit: us */
49*4882a593Smuzhiyun #define SET_POWER_MODE_TRIES   5
50*4882a593Smuzhiyun 
51*4882a593Smuzhiyun #define GEN3_BL_CMD_CHECKSUM_SEED 0xff
52*4882a593Smuzhiyun #define GEN3_BL_CMD_INITIATE_BL   0x38
53*4882a593Smuzhiyun #define GEN3_BL_CMD_WRITE_BLOCK   0x39
54*4882a593Smuzhiyun #define GEN3_BL_CMD_VERIFY_BLOCK  0x3a
55*4882a593Smuzhiyun #define GEN3_BL_CMD_TERMINATE_BL  0x3b
56*4882a593Smuzhiyun #define GEN3_BL_CMD_LAUNCH_APP    0xa5
57*4882a593Smuzhiyun 
58*4882a593Smuzhiyun /*
59*4882a593Smuzhiyun  * CYAPA trackpad device states.
60*4882a593Smuzhiyun  * Used in register 0x00, bit1-0, DeviceStatus field.
61*4882a593Smuzhiyun  * Other values indicate device is in an abnormal state and must be reset.
62*4882a593Smuzhiyun  */
63*4882a593Smuzhiyun #define CYAPA_DEV_NORMAL  0x03
64*4882a593Smuzhiyun #define CYAPA_DEV_BUSY    0x01
65*4882a593Smuzhiyun 
66*4882a593Smuzhiyun #define CYAPA_FW_BLOCK_SIZE	64
67*4882a593Smuzhiyun #define CYAPA_FW_READ_SIZE	16
68*4882a593Smuzhiyun #define CYAPA_FW_HDR_START	0x0780
69*4882a593Smuzhiyun #define CYAPA_FW_HDR_BLOCK_COUNT  2
70*4882a593Smuzhiyun #define CYAPA_FW_HDR_BLOCK_START  (CYAPA_FW_HDR_START / CYAPA_FW_BLOCK_SIZE)
71*4882a593Smuzhiyun #define CYAPA_FW_HDR_SIZE	  (CYAPA_FW_HDR_BLOCK_COUNT * \
72*4882a593Smuzhiyun 					CYAPA_FW_BLOCK_SIZE)
73*4882a593Smuzhiyun #define CYAPA_FW_DATA_START	0x0800
74*4882a593Smuzhiyun #define CYAPA_FW_DATA_BLOCK_COUNT  480
75*4882a593Smuzhiyun #define CYAPA_FW_DATA_BLOCK_START  (CYAPA_FW_DATA_START / CYAPA_FW_BLOCK_SIZE)
76*4882a593Smuzhiyun #define CYAPA_FW_DATA_SIZE	(CYAPA_FW_DATA_BLOCK_COUNT * \
77*4882a593Smuzhiyun 				 CYAPA_FW_BLOCK_SIZE)
78*4882a593Smuzhiyun #define CYAPA_FW_SIZE		(CYAPA_FW_HDR_SIZE + CYAPA_FW_DATA_SIZE)
79*4882a593Smuzhiyun #define CYAPA_CMD_LEN		16
80*4882a593Smuzhiyun 
81*4882a593Smuzhiyun #define GEN3_BL_IDLE_FW_MAJ_VER_OFFSET 0x0b
82*4882a593Smuzhiyun #define GEN3_BL_IDLE_FW_MIN_VER_OFFSET (GEN3_BL_IDLE_FW_MAJ_VER_OFFSET + 1)
83*4882a593Smuzhiyun 
84*4882a593Smuzhiyun 
85*4882a593Smuzhiyun struct cyapa_touch {
86*4882a593Smuzhiyun 	/*
87*4882a593Smuzhiyun 	 * high bits or x/y position value
88*4882a593Smuzhiyun 	 * bit 7 - 4: high 4 bits of x position value
89*4882a593Smuzhiyun 	 * bit 3 - 0: high 4 bits of y position value
90*4882a593Smuzhiyun 	 */
91*4882a593Smuzhiyun 	u8 xy_hi;
92*4882a593Smuzhiyun 	u8 x_lo;  /* low 8 bits of x position value. */
93*4882a593Smuzhiyun 	u8 y_lo;  /* low 8 bits of y position value. */
94*4882a593Smuzhiyun 	u8 pressure;
95*4882a593Smuzhiyun 	/* id range is 1 - 15.  It is incremented with every new touch. */
96*4882a593Smuzhiyun 	u8 id;
97*4882a593Smuzhiyun } __packed;
98*4882a593Smuzhiyun 
99*4882a593Smuzhiyun struct cyapa_reg_data {
100*4882a593Smuzhiyun 	/*
101*4882a593Smuzhiyun 	 * bit 0 - 1: device status
102*4882a593Smuzhiyun 	 * bit 3 - 2: power mode
103*4882a593Smuzhiyun 	 * bit 6 - 4: reserved
104*4882a593Smuzhiyun 	 * bit 7: interrupt valid bit
105*4882a593Smuzhiyun 	 */
106*4882a593Smuzhiyun 	u8 device_status;
107*4882a593Smuzhiyun 	/*
108*4882a593Smuzhiyun 	 * bit 7 - 4: number of fingers currently touching pad
109*4882a593Smuzhiyun 	 * bit 3: valid data check bit
110*4882a593Smuzhiyun 	 * bit 2: middle mechanism button state if exists
111*4882a593Smuzhiyun 	 * bit 1: right mechanism button state if exists
112*4882a593Smuzhiyun 	 * bit 0: left mechanism button state if exists
113*4882a593Smuzhiyun 	 */
114*4882a593Smuzhiyun 	u8 finger_btn;
115*4882a593Smuzhiyun 	/* CYAPA reports up to 5 touches per packet. */
116*4882a593Smuzhiyun 	struct cyapa_touch touches[5];
117*4882a593Smuzhiyun } __packed;
118*4882a593Smuzhiyun 
119*4882a593Smuzhiyun struct gen3_write_block_cmd {
120*4882a593Smuzhiyun 	u8 checksum_seed;  /* Always be 0xff */
121*4882a593Smuzhiyun 	u8 cmd_code;       /* command code: 0x39 */
122*4882a593Smuzhiyun 	u8 key[8];         /* 8-byte security key */
123*4882a593Smuzhiyun 	__be16 block_num;
124*4882a593Smuzhiyun 	u8 block_data[CYAPA_FW_BLOCK_SIZE];
125*4882a593Smuzhiyun 	u8 block_checksum;  /* Calculated using bytes 12 - 75 */
126*4882a593Smuzhiyun 	u8 cmd_checksum;    /* Calculated using bytes 0-76 */
127*4882a593Smuzhiyun } __packed;
128*4882a593Smuzhiyun 
129*4882a593Smuzhiyun static const u8 security_key[] = {
130*4882a593Smuzhiyun 		0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07 };
131*4882a593Smuzhiyun static const u8 bl_activate[] = { 0x00, 0xff, 0x38, 0x00, 0x01, 0x02, 0x03,
132*4882a593Smuzhiyun 		0x04, 0x05, 0x06, 0x07 };
133*4882a593Smuzhiyun static const u8 bl_deactivate[] = { 0x00, 0xff, 0x3b, 0x00, 0x01, 0x02, 0x03,
134*4882a593Smuzhiyun 		0x04, 0x05, 0x06, 0x07 };
135*4882a593Smuzhiyun static const u8 bl_exit[] = { 0x00, 0xff, 0xa5, 0x00, 0x01, 0x02, 0x03, 0x04,
136*4882a593Smuzhiyun 		0x05, 0x06, 0x07 };
137*4882a593Smuzhiyun 
138*4882a593Smuzhiyun 
139*4882a593Smuzhiyun  /* for byte read/write command */
140*4882a593Smuzhiyun #define CMD_RESET 0
141*4882a593Smuzhiyun #define CMD_POWER_MODE 1
142*4882a593Smuzhiyun #define CMD_DEV_STATUS 2
143*4882a593Smuzhiyun #define CMD_REPORT_MAX_BASELINE 3
144*4882a593Smuzhiyun #define CMD_REPORT_MIN_BASELINE 4
145*4882a593Smuzhiyun #define SMBUS_BYTE_CMD(cmd) (((cmd) & 0x3f) << 1)
146*4882a593Smuzhiyun #define CYAPA_SMBUS_RESET         SMBUS_BYTE_CMD(CMD_RESET)
147*4882a593Smuzhiyun #define CYAPA_SMBUS_POWER_MODE    SMBUS_BYTE_CMD(CMD_POWER_MODE)
148*4882a593Smuzhiyun #define CYAPA_SMBUS_DEV_STATUS    SMBUS_BYTE_CMD(CMD_DEV_STATUS)
149*4882a593Smuzhiyun #define CYAPA_SMBUS_MAX_BASELINE  SMBUS_BYTE_CMD(CMD_REPORT_MAX_BASELINE)
150*4882a593Smuzhiyun #define CYAPA_SMBUS_MIN_BASELINE  SMBUS_BYTE_CMD(CMD_REPORT_MIN_BASELINE)
151*4882a593Smuzhiyun 
152*4882a593Smuzhiyun  /* for group registers read/write command */
153*4882a593Smuzhiyun #define REG_GROUP_DATA  0
154*4882a593Smuzhiyun #define REG_GROUP_CMD   2
155*4882a593Smuzhiyun #define REG_GROUP_QUERY 3
156*4882a593Smuzhiyun #define SMBUS_GROUP_CMD(grp) (0x80 | (((grp) & 0x07) << 3))
157*4882a593Smuzhiyun #define CYAPA_SMBUS_GROUP_DATA  SMBUS_GROUP_CMD(REG_GROUP_DATA)
158*4882a593Smuzhiyun #define CYAPA_SMBUS_GROUP_CMD   SMBUS_GROUP_CMD(REG_GROUP_CMD)
159*4882a593Smuzhiyun #define CYAPA_SMBUS_GROUP_QUERY SMBUS_GROUP_CMD(REG_GROUP_QUERY)
160*4882a593Smuzhiyun 
161*4882a593Smuzhiyun  /* for register block read/write command */
162*4882a593Smuzhiyun #define CMD_BL_STATUS		0
163*4882a593Smuzhiyun #define CMD_BL_HEAD		1
164*4882a593Smuzhiyun #define CMD_BL_CMD		2
165*4882a593Smuzhiyun #define CMD_BL_DATA		3
166*4882a593Smuzhiyun #define CMD_BL_ALL		4
167*4882a593Smuzhiyun #define CMD_BLK_PRODUCT_ID	5
168*4882a593Smuzhiyun #define CMD_BLK_HEAD		6
169*4882a593Smuzhiyun #define SMBUS_BLOCK_CMD(cmd) (0xc0 | (((cmd) & 0x1f) << 1))
170*4882a593Smuzhiyun 
171*4882a593Smuzhiyun /* register block read/write command in bootloader mode */
172*4882a593Smuzhiyun #define CYAPA_SMBUS_BL_STATUS SMBUS_BLOCK_CMD(CMD_BL_STATUS)
173*4882a593Smuzhiyun #define CYAPA_SMBUS_BL_HEAD   SMBUS_BLOCK_CMD(CMD_BL_HEAD)
174*4882a593Smuzhiyun #define CYAPA_SMBUS_BL_CMD    SMBUS_BLOCK_CMD(CMD_BL_CMD)
175*4882a593Smuzhiyun #define CYAPA_SMBUS_BL_DATA   SMBUS_BLOCK_CMD(CMD_BL_DATA)
176*4882a593Smuzhiyun #define CYAPA_SMBUS_BL_ALL    SMBUS_BLOCK_CMD(CMD_BL_ALL)
177*4882a593Smuzhiyun 
178*4882a593Smuzhiyun /* register block read/write command in operational mode */
179*4882a593Smuzhiyun #define CYAPA_SMBUS_BLK_PRODUCT_ID SMBUS_BLOCK_CMD(CMD_BLK_PRODUCT_ID)
180*4882a593Smuzhiyun #define CYAPA_SMBUS_BLK_HEAD       SMBUS_BLOCK_CMD(CMD_BLK_HEAD)
181*4882a593Smuzhiyun 
182*4882a593Smuzhiyun struct cyapa_cmd_len {
183*4882a593Smuzhiyun 	u8 cmd;
184*4882a593Smuzhiyun 	u8 len;
185*4882a593Smuzhiyun };
186*4882a593Smuzhiyun 
187*4882a593Smuzhiyun /* maps generic CYAPA_CMD_* code to the I2C equivalent */
188*4882a593Smuzhiyun static const struct cyapa_cmd_len cyapa_i2c_cmds[] = {
189*4882a593Smuzhiyun 	{ CYAPA_OFFSET_SOFT_RESET, 1 },		/* CYAPA_CMD_SOFT_RESET */
190*4882a593Smuzhiyun 	{ REG_OFFSET_COMMAND_BASE + 1, 1 },	/* CYAPA_CMD_POWER_MODE */
191*4882a593Smuzhiyun 	{ REG_OFFSET_DATA_BASE, 1 },		/* CYAPA_CMD_DEV_STATUS */
192*4882a593Smuzhiyun 	{ REG_OFFSET_DATA_BASE, sizeof(struct cyapa_reg_data) },
193*4882a593Smuzhiyun 						/* CYAPA_CMD_GROUP_DATA */
194*4882a593Smuzhiyun 	{ REG_OFFSET_COMMAND_BASE, 0 },		/* CYAPA_CMD_GROUP_CMD */
195*4882a593Smuzhiyun 	{ REG_OFFSET_QUERY_BASE, QUERY_DATA_SIZE }, /* CYAPA_CMD_GROUP_QUERY */
196*4882a593Smuzhiyun 	{ BL_HEAD_OFFSET, 3 },			/* CYAPA_CMD_BL_STATUS */
197*4882a593Smuzhiyun 	{ BL_HEAD_OFFSET, 16 },			/* CYAPA_CMD_BL_HEAD */
198*4882a593Smuzhiyun 	{ BL_HEAD_OFFSET, 16 },			/* CYAPA_CMD_BL_CMD */
199*4882a593Smuzhiyun 	{ BL_DATA_OFFSET, 16 },			/* CYAPA_CMD_BL_DATA */
200*4882a593Smuzhiyun 	{ BL_HEAD_OFFSET, 32 },			/* CYAPA_CMD_BL_ALL */
201*4882a593Smuzhiyun 	{ REG_OFFSET_QUERY_BASE, PRODUCT_ID_SIZE },
202*4882a593Smuzhiyun 						/* CYAPA_CMD_BLK_PRODUCT_ID */
203*4882a593Smuzhiyun 	{ REG_OFFSET_DATA_BASE, 32 },		/* CYAPA_CMD_BLK_HEAD */
204*4882a593Smuzhiyun 	{ REG_OFFSET_MAX_BASELINE, 1 },		/* CYAPA_CMD_MAX_BASELINE */
205*4882a593Smuzhiyun 	{ REG_OFFSET_MIN_BASELINE, 1 },		/* CYAPA_CMD_MIN_BASELINE */
206*4882a593Smuzhiyun };
207*4882a593Smuzhiyun 
208*4882a593Smuzhiyun static const struct cyapa_cmd_len cyapa_smbus_cmds[] = {
209*4882a593Smuzhiyun 	{ CYAPA_SMBUS_RESET, 1 },		/* CYAPA_CMD_SOFT_RESET */
210*4882a593Smuzhiyun 	{ CYAPA_SMBUS_POWER_MODE, 1 },		/* CYAPA_CMD_POWER_MODE */
211*4882a593Smuzhiyun 	{ CYAPA_SMBUS_DEV_STATUS, 1 },		/* CYAPA_CMD_DEV_STATUS */
212*4882a593Smuzhiyun 	{ CYAPA_SMBUS_GROUP_DATA, sizeof(struct cyapa_reg_data) },
213*4882a593Smuzhiyun 						/* CYAPA_CMD_GROUP_DATA */
214*4882a593Smuzhiyun 	{ CYAPA_SMBUS_GROUP_CMD, 2 },		/* CYAPA_CMD_GROUP_CMD */
215*4882a593Smuzhiyun 	{ CYAPA_SMBUS_GROUP_QUERY, QUERY_DATA_SIZE },
216*4882a593Smuzhiyun 						/* CYAPA_CMD_GROUP_QUERY */
217*4882a593Smuzhiyun 	{ CYAPA_SMBUS_BL_STATUS, 3 },		/* CYAPA_CMD_BL_STATUS */
218*4882a593Smuzhiyun 	{ CYAPA_SMBUS_BL_HEAD, 16 },		/* CYAPA_CMD_BL_HEAD */
219*4882a593Smuzhiyun 	{ CYAPA_SMBUS_BL_CMD, 16 },		/* CYAPA_CMD_BL_CMD */
220*4882a593Smuzhiyun 	{ CYAPA_SMBUS_BL_DATA, 16 },		/* CYAPA_CMD_BL_DATA */
221*4882a593Smuzhiyun 	{ CYAPA_SMBUS_BL_ALL, 32 },		/* CYAPA_CMD_BL_ALL */
222*4882a593Smuzhiyun 	{ CYAPA_SMBUS_BLK_PRODUCT_ID, PRODUCT_ID_SIZE },
223*4882a593Smuzhiyun 						/* CYAPA_CMD_BLK_PRODUCT_ID */
224*4882a593Smuzhiyun 	{ CYAPA_SMBUS_BLK_HEAD, 16 },		/* CYAPA_CMD_BLK_HEAD */
225*4882a593Smuzhiyun 	{ CYAPA_SMBUS_MAX_BASELINE, 1 },	/* CYAPA_CMD_MAX_BASELINE */
226*4882a593Smuzhiyun 	{ CYAPA_SMBUS_MIN_BASELINE, 1 },	/* CYAPA_CMD_MIN_BASELINE */
227*4882a593Smuzhiyun };
228*4882a593Smuzhiyun 
229*4882a593Smuzhiyun static int cyapa_gen3_try_poll_handler(struct cyapa *cyapa);
230*4882a593Smuzhiyun 
231*4882a593Smuzhiyun /*
232*4882a593Smuzhiyun  * cyapa_smbus_read_block - perform smbus block read command
233*4882a593Smuzhiyun  * @cyapa  - private data structure of the driver
234*4882a593Smuzhiyun  * @cmd    - the properly encoded smbus command
235*4882a593Smuzhiyun  * @len    - expected length of smbus command result
236*4882a593Smuzhiyun  * @values - buffer to store smbus command result
237*4882a593Smuzhiyun  *
238*4882a593Smuzhiyun  * Returns negative errno, else the number of bytes written.
239*4882a593Smuzhiyun  *
240*4882a593Smuzhiyun  * Note:
241*4882a593Smuzhiyun  * In trackpad device, the memory block allocated for I2C register map
242*4882a593Smuzhiyun  * is 256 bytes, so the max read block for I2C bus is 256 bytes.
243*4882a593Smuzhiyun  */
cyapa_smbus_read_block(struct cyapa * cyapa,u8 cmd,size_t len,u8 * values)244*4882a593Smuzhiyun ssize_t cyapa_smbus_read_block(struct cyapa *cyapa, u8 cmd, size_t len,
245*4882a593Smuzhiyun 				      u8 *values)
246*4882a593Smuzhiyun {
247*4882a593Smuzhiyun 	ssize_t ret;
248*4882a593Smuzhiyun 	u8 index;
249*4882a593Smuzhiyun 	u8 smbus_cmd;
250*4882a593Smuzhiyun 	u8 *buf;
251*4882a593Smuzhiyun 	struct i2c_client *client = cyapa->client;
252*4882a593Smuzhiyun 
253*4882a593Smuzhiyun 	if (!(SMBUS_BYTE_BLOCK_CMD_MASK & cmd))
254*4882a593Smuzhiyun 		return -EINVAL;
255*4882a593Smuzhiyun 
256*4882a593Smuzhiyun 	if (SMBUS_GROUP_BLOCK_CMD_MASK & cmd) {
257*4882a593Smuzhiyun 		/* read specific block registers command. */
258*4882a593Smuzhiyun 		smbus_cmd = SMBUS_ENCODE_RW(cmd, SMBUS_READ);
259*4882a593Smuzhiyun 		ret = i2c_smbus_read_block_data(client, smbus_cmd, values);
260*4882a593Smuzhiyun 		goto out;
261*4882a593Smuzhiyun 	}
262*4882a593Smuzhiyun 
263*4882a593Smuzhiyun 	ret = 0;
264*4882a593Smuzhiyun 	for (index = 0; index * I2C_SMBUS_BLOCK_MAX < len; index++) {
265*4882a593Smuzhiyun 		smbus_cmd = SMBUS_ENCODE_IDX(cmd, index);
266*4882a593Smuzhiyun 		smbus_cmd = SMBUS_ENCODE_RW(smbus_cmd, SMBUS_READ);
267*4882a593Smuzhiyun 		buf = values + I2C_SMBUS_BLOCK_MAX * index;
268*4882a593Smuzhiyun 		ret = i2c_smbus_read_block_data(client, smbus_cmd, buf);
269*4882a593Smuzhiyun 		if (ret < 0)
270*4882a593Smuzhiyun 			goto out;
271*4882a593Smuzhiyun 	}
272*4882a593Smuzhiyun 
273*4882a593Smuzhiyun out:
274*4882a593Smuzhiyun 	return ret > 0 ? len : ret;
275*4882a593Smuzhiyun }
276*4882a593Smuzhiyun 
cyapa_read_byte(struct cyapa * cyapa,u8 cmd_idx)277*4882a593Smuzhiyun static s32 cyapa_read_byte(struct cyapa *cyapa, u8 cmd_idx)
278*4882a593Smuzhiyun {
279*4882a593Smuzhiyun 	u8 cmd;
280*4882a593Smuzhiyun 
281*4882a593Smuzhiyun 	if (cyapa->smbus) {
282*4882a593Smuzhiyun 		cmd = cyapa_smbus_cmds[cmd_idx].cmd;
283*4882a593Smuzhiyun 		cmd = SMBUS_ENCODE_RW(cmd, SMBUS_READ);
284*4882a593Smuzhiyun 	} else {
285*4882a593Smuzhiyun 		cmd = cyapa_i2c_cmds[cmd_idx].cmd;
286*4882a593Smuzhiyun 	}
287*4882a593Smuzhiyun 	return i2c_smbus_read_byte_data(cyapa->client, cmd);
288*4882a593Smuzhiyun }
289*4882a593Smuzhiyun 
cyapa_write_byte(struct cyapa * cyapa,u8 cmd_idx,u8 value)290*4882a593Smuzhiyun static s32 cyapa_write_byte(struct cyapa *cyapa, u8 cmd_idx, u8 value)
291*4882a593Smuzhiyun {
292*4882a593Smuzhiyun 	u8 cmd;
293*4882a593Smuzhiyun 
294*4882a593Smuzhiyun 	if (cyapa->smbus) {
295*4882a593Smuzhiyun 		cmd = cyapa_smbus_cmds[cmd_idx].cmd;
296*4882a593Smuzhiyun 		cmd = SMBUS_ENCODE_RW(cmd, SMBUS_WRITE);
297*4882a593Smuzhiyun 	} else {
298*4882a593Smuzhiyun 		cmd = cyapa_i2c_cmds[cmd_idx].cmd;
299*4882a593Smuzhiyun 	}
300*4882a593Smuzhiyun 	return i2c_smbus_write_byte_data(cyapa->client, cmd, value);
301*4882a593Smuzhiyun }
302*4882a593Smuzhiyun 
cyapa_i2c_reg_read_block(struct cyapa * cyapa,u8 reg,size_t len,u8 * values)303*4882a593Smuzhiyun ssize_t cyapa_i2c_reg_read_block(struct cyapa *cyapa, u8 reg, size_t len,
304*4882a593Smuzhiyun 					u8 *values)
305*4882a593Smuzhiyun {
306*4882a593Smuzhiyun 	return i2c_smbus_read_i2c_block_data(cyapa->client, reg, len, values);
307*4882a593Smuzhiyun }
308*4882a593Smuzhiyun 
cyapa_i2c_reg_write_block(struct cyapa * cyapa,u8 reg,size_t len,const u8 * values)309*4882a593Smuzhiyun static ssize_t cyapa_i2c_reg_write_block(struct cyapa *cyapa, u8 reg,
310*4882a593Smuzhiyun 					 size_t len, const u8 *values)
311*4882a593Smuzhiyun {
312*4882a593Smuzhiyun 	return i2c_smbus_write_i2c_block_data(cyapa->client, reg, len, values);
313*4882a593Smuzhiyun }
314*4882a593Smuzhiyun 
cyapa_read_block(struct cyapa * cyapa,u8 cmd_idx,u8 * values)315*4882a593Smuzhiyun ssize_t cyapa_read_block(struct cyapa *cyapa, u8 cmd_idx, u8 *values)
316*4882a593Smuzhiyun {
317*4882a593Smuzhiyun 	u8 cmd;
318*4882a593Smuzhiyun 	size_t len;
319*4882a593Smuzhiyun 
320*4882a593Smuzhiyun 	if (cyapa->smbus) {
321*4882a593Smuzhiyun 		cmd = cyapa_smbus_cmds[cmd_idx].cmd;
322*4882a593Smuzhiyun 		len = cyapa_smbus_cmds[cmd_idx].len;
323*4882a593Smuzhiyun 		return cyapa_smbus_read_block(cyapa, cmd, len, values);
324*4882a593Smuzhiyun 	}
325*4882a593Smuzhiyun 	cmd = cyapa_i2c_cmds[cmd_idx].cmd;
326*4882a593Smuzhiyun 	len = cyapa_i2c_cmds[cmd_idx].len;
327*4882a593Smuzhiyun 	return cyapa_i2c_reg_read_block(cyapa, cmd, len, values);
328*4882a593Smuzhiyun }
329*4882a593Smuzhiyun 
330*4882a593Smuzhiyun /*
331*4882a593Smuzhiyun  * Determine the Gen3 trackpad device's current operating state.
332*4882a593Smuzhiyun  *
333*4882a593Smuzhiyun  */
cyapa_gen3_state_parse(struct cyapa * cyapa,u8 * reg_data,int len)334*4882a593Smuzhiyun static int cyapa_gen3_state_parse(struct cyapa *cyapa, u8 *reg_data, int len)
335*4882a593Smuzhiyun {
336*4882a593Smuzhiyun 	cyapa->state = CYAPA_STATE_NO_DEVICE;
337*4882a593Smuzhiyun 
338*4882a593Smuzhiyun 	/* Parse based on Gen3 characteristic registers and bits */
339*4882a593Smuzhiyun 	if (reg_data[REG_BL_FILE] == BL_FILE &&
340*4882a593Smuzhiyun 		reg_data[REG_BL_ERROR] == BL_ERROR_NO_ERR_IDLE &&
341*4882a593Smuzhiyun 		(reg_data[REG_BL_STATUS] ==
342*4882a593Smuzhiyun 			(BL_STATUS_RUNNING | BL_STATUS_CSUM_VALID) ||
343*4882a593Smuzhiyun 			reg_data[REG_BL_STATUS] == BL_STATUS_RUNNING)) {
344*4882a593Smuzhiyun 		/*
345*4882a593Smuzhiyun 		 * Normal state after power on or reset,
346*4882a593Smuzhiyun 		 * REG_BL_STATUS == 0x11, firmware image checksum is valid.
347*4882a593Smuzhiyun 		 * REG_BL_STATUS == 0x10, firmware image checksum is invalid.
348*4882a593Smuzhiyun 		 */
349*4882a593Smuzhiyun 		cyapa->gen = CYAPA_GEN3;
350*4882a593Smuzhiyun 		cyapa->state = CYAPA_STATE_BL_IDLE;
351*4882a593Smuzhiyun 	} else if (reg_data[REG_BL_FILE] == BL_FILE &&
352*4882a593Smuzhiyun 		(reg_data[REG_BL_STATUS] & BL_STATUS_RUNNING) ==
353*4882a593Smuzhiyun 			BL_STATUS_RUNNING) {
354*4882a593Smuzhiyun 		cyapa->gen = CYAPA_GEN3;
355*4882a593Smuzhiyun 		if (reg_data[REG_BL_STATUS] & BL_STATUS_BUSY) {
356*4882a593Smuzhiyun 			cyapa->state = CYAPA_STATE_BL_BUSY;
357*4882a593Smuzhiyun 		} else {
358*4882a593Smuzhiyun 			if ((reg_data[REG_BL_ERROR] & BL_ERROR_BOOTLOADING) ==
359*4882a593Smuzhiyun 					BL_ERROR_BOOTLOADING)
360*4882a593Smuzhiyun 				cyapa->state = CYAPA_STATE_BL_ACTIVE;
361*4882a593Smuzhiyun 			else
362*4882a593Smuzhiyun 				cyapa->state = CYAPA_STATE_BL_IDLE;
363*4882a593Smuzhiyun 		}
364*4882a593Smuzhiyun 	} else if ((reg_data[REG_OP_STATUS] & OP_STATUS_SRC) &&
365*4882a593Smuzhiyun 			(reg_data[REG_OP_DATA1] & OP_DATA_VALID)) {
366*4882a593Smuzhiyun 		/*
367*4882a593Smuzhiyun 		 * Normal state when running in operational mode,
368*4882a593Smuzhiyun 		 * may also not in full power state or
369*4882a593Smuzhiyun 		 * busying in command process.
370*4882a593Smuzhiyun 		 */
371*4882a593Smuzhiyun 		if (GEN3_FINGER_NUM(reg_data[REG_OP_DATA1]) <=
372*4882a593Smuzhiyun 				GEN3_MAX_FINGERS) {
373*4882a593Smuzhiyun 			/* Finger number data is valid. */
374*4882a593Smuzhiyun 			cyapa->gen = CYAPA_GEN3;
375*4882a593Smuzhiyun 			cyapa->state = CYAPA_STATE_OP;
376*4882a593Smuzhiyun 		}
377*4882a593Smuzhiyun 	} else if (reg_data[REG_OP_STATUS] == 0x0C &&
378*4882a593Smuzhiyun 			reg_data[REG_OP_DATA1] == 0x08) {
379*4882a593Smuzhiyun 		/* Op state when first two registers overwritten with 0x00 */
380*4882a593Smuzhiyun 		cyapa->gen = CYAPA_GEN3;
381*4882a593Smuzhiyun 		cyapa->state = CYAPA_STATE_OP;
382*4882a593Smuzhiyun 	} else if (reg_data[REG_BL_STATUS] &
383*4882a593Smuzhiyun 			(BL_STATUS_RUNNING | BL_STATUS_BUSY)) {
384*4882a593Smuzhiyun 		cyapa->gen = CYAPA_GEN3;
385*4882a593Smuzhiyun 		cyapa->state = CYAPA_STATE_BL_BUSY;
386*4882a593Smuzhiyun 	}
387*4882a593Smuzhiyun 
388*4882a593Smuzhiyun 	if (cyapa->gen == CYAPA_GEN3 && (cyapa->state == CYAPA_STATE_OP ||
389*4882a593Smuzhiyun 		cyapa->state == CYAPA_STATE_BL_IDLE ||
390*4882a593Smuzhiyun 		cyapa->state == CYAPA_STATE_BL_ACTIVE ||
391*4882a593Smuzhiyun 		cyapa->state == CYAPA_STATE_BL_BUSY))
392*4882a593Smuzhiyun 		return 0;
393*4882a593Smuzhiyun 
394*4882a593Smuzhiyun 	return -EAGAIN;
395*4882a593Smuzhiyun }
396*4882a593Smuzhiyun 
397*4882a593Smuzhiyun /*
398*4882a593Smuzhiyun  * Enter bootloader by soft resetting the device.
399*4882a593Smuzhiyun  *
400*4882a593Smuzhiyun  * If device is already in the bootloader, the function just returns.
401*4882a593Smuzhiyun  * Otherwise, reset the device; after reset, device enters bootloader idle
402*4882a593Smuzhiyun  * state immediately.
403*4882a593Smuzhiyun  *
404*4882a593Smuzhiyun  * Returns:
405*4882a593Smuzhiyun  *   0        on success
406*4882a593Smuzhiyun  *   -EAGAIN  device was reset, but is not now in bootloader idle state
407*4882a593Smuzhiyun  *   < 0      if the device never responds within the timeout
408*4882a593Smuzhiyun  */
cyapa_gen3_bl_enter(struct cyapa * cyapa)409*4882a593Smuzhiyun static int cyapa_gen3_bl_enter(struct cyapa *cyapa)
410*4882a593Smuzhiyun {
411*4882a593Smuzhiyun 	int error;
412*4882a593Smuzhiyun 	int waiting_time;
413*4882a593Smuzhiyun 
414*4882a593Smuzhiyun 	error = cyapa_poll_state(cyapa, 500);
415*4882a593Smuzhiyun 	if (error)
416*4882a593Smuzhiyun 		return error;
417*4882a593Smuzhiyun 	if (cyapa->state == CYAPA_STATE_BL_IDLE) {
418*4882a593Smuzhiyun 		/* Already in BL_IDLE. Skipping reset. */
419*4882a593Smuzhiyun 		return 0;
420*4882a593Smuzhiyun 	}
421*4882a593Smuzhiyun 
422*4882a593Smuzhiyun 	if (cyapa->state != CYAPA_STATE_OP)
423*4882a593Smuzhiyun 		return -EAGAIN;
424*4882a593Smuzhiyun 
425*4882a593Smuzhiyun 	cyapa->operational = false;
426*4882a593Smuzhiyun 	cyapa->state = CYAPA_STATE_NO_DEVICE;
427*4882a593Smuzhiyun 	error = cyapa_write_byte(cyapa, CYAPA_CMD_SOFT_RESET, 0x01);
428*4882a593Smuzhiyun 	if (error)
429*4882a593Smuzhiyun 		return -EIO;
430*4882a593Smuzhiyun 
431*4882a593Smuzhiyun 	usleep_range(25000, 50000);
432*4882a593Smuzhiyun 	waiting_time = 2000;  /* For some shipset, max waiting time is 1~2s. */
433*4882a593Smuzhiyun 	do {
434*4882a593Smuzhiyun 		error = cyapa_poll_state(cyapa, 500);
435*4882a593Smuzhiyun 		if (error) {
436*4882a593Smuzhiyun 			if (error == -ETIMEDOUT) {
437*4882a593Smuzhiyun 				waiting_time -= 500;
438*4882a593Smuzhiyun 				continue;
439*4882a593Smuzhiyun 			}
440*4882a593Smuzhiyun 			return error;
441*4882a593Smuzhiyun 		}
442*4882a593Smuzhiyun 
443*4882a593Smuzhiyun 		if ((cyapa->state == CYAPA_STATE_BL_IDLE) &&
444*4882a593Smuzhiyun 			!(cyapa->status[REG_BL_STATUS] & BL_STATUS_WATCHDOG))
445*4882a593Smuzhiyun 			break;
446*4882a593Smuzhiyun 
447*4882a593Smuzhiyun 		msleep(100);
448*4882a593Smuzhiyun 		waiting_time -= 100;
449*4882a593Smuzhiyun 	} while (waiting_time > 0);
450*4882a593Smuzhiyun 
451*4882a593Smuzhiyun 	if ((cyapa->state != CYAPA_STATE_BL_IDLE) ||
452*4882a593Smuzhiyun 		(cyapa->status[REG_BL_STATUS] & BL_STATUS_WATCHDOG))
453*4882a593Smuzhiyun 		return -EAGAIN;
454*4882a593Smuzhiyun 
455*4882a593Smuzhiyun 	return 0;
456*4882a593Smuzhiyun }
457*4882a593Smuzhiyun 
cyapa_gen3_bl_activate(struct cyapa * cyapa)458*4882a593Smuzhiyun static int cyapa_gen3_bl_activate(struct cyapa *cyapa)
459*4882a593Smuzhiyun {
460*4882a593Smuzhiyun 	int error;
461*4882a593Smuzhiyun 
462*4882a593Smuzhiyun 	error = cyapa_i2c_reg_write_block(cyapa, 0, sizeof(bl_activate),
463*4882a593Smuzhiyun 					bl_activate);
464*4882a593Smuzhiyun 	if (error)
465*4882a593Smuzhiyun 		return error;
466*4882a593Smuzhiyun 
467*4882a593Smuzhiyun 	/* Wait for bootloader to activate; takes between 2 and 12 seconds */
468*4882a593Smuzhiyun 	msleep(2000);
469*4882a593Smuzhiyun 	error = cyapa_poll_state(cyapa, 11000);
470*4882a593Smuzhiyun 	if (error)
471*4882a593Smuzhiyun 		return error;
472*4882a593Smuzhiyun 	if (cyapa->state != CYAPA_STATE_BL_ACTIVE)
473*4882a593Smuzhiyun 		return -EAGAIN;
474*4882a593Smuzhiyun 
475*4882a593Smuzhiyun 	return 0;
476*4882a593Smuzhiyun }
477*4882a593Smuzhiyun 
cyapa_gen3_bl_deactivate(struct cyapa * cyapa)478*4882a593Smuzhiyun static int cyapa_gen3_bl_deactivate(struct cyapa *cyapa)
479*4882a593Smuzhiyun {
480*4882a593Smuzhiyun 	int error;
481*4882a593Smuzhiyun 
482*4882a593Smuzhiyun 	error = cyapa_i2c_reg_write_block(cyapa, 0, sizeof(bl_deactivate),
483*4882a593Smuzhiyun 					bl_deactivate);
484*4882a593Smuzhiyun 	if (error)
485*4882a593Smuzhiyun 		return error;
486*4882a593Smuzhiyun 
487*4882a593Smuzhiyun 	/* Wait for bootloader to switch to idle state; should take < 100ms */
488*4882a593Smuzhiyun 	msleep(100);
489*4882a593Smuzhiyun 	error = cyapa_poll_state(cyapa, 500);
490*4882a593Smuzhiyun 	if (error)
491*4882a593Smuzhiyun 		return error;
492*4882a593Smuzhiyun 	if (cyapa->state != CYAPA_STATE_BL_IDLE)
493*4882a593Smuzhiyun 		return -EAGAIN;
494*4882a593Smuzhiyun 	return 0;
495*4882a593Smuzhiyun }
496*4882a593Smuzhiyun 
497*4882a593Smuzhiyun /*
498*4882a593Smuzhiyun  * Exit bootloader
499*4882a593Smuzhiyun  *
500*4882a593Smuzhiyun  * Send bl_exit command, then wait 50 - 100 ms to let device transition to
501*4882a593Smuzhiyun  * operational mode.  If this is the first time the device's firmware is
502*4882a593Smuzhiyun  * running, it can take up to 2 seconds to calibrate its sensors.  So, poll
503*4882a593Smuzhiyun  * the device's new state for up to 2 seconds.
504*4882a593Smuzhiyun  *
505*4882a593Smuzhiyun  * Returns:
506*4882a593Smuzhiyun  *   -EIO    failure while reading from device
507*4882a593Smuzhiyun  *   -EAGAIN device is stuck in bootloader, b/c it has invalid firmware
508*4882a593Smuzhiyun  *   0       device is supported and in operational mode
509*4882a593Smuzhiyun  */
cyapa_gen3_bl_exit(struct cyapa * cyapa)510*4882a593Smuzhiyun static int cyapa_gen3_bl_exit(struct cyapa *cyapa)
511*4882a593Smuzhiyun {
512*4882a593Smuzhiyun 	int error;
513*4882a593Smuzhiyun 
514*4882a593Smuzhiyun 	error = cyapa_i2c_reg_write_block(cyapa, 0, sizeof(bl_exit), bl_exit);
515*4882a593Smuzhiyun 	if (error)
516*4882a593Smuzhiyun 		return error;
517*4882a593Smuzhiyun 
518*4882a593Smuzhiyun 	/*
519*4882a593Smuzhiyun 	 * Wait for bootloader to exit, and operation mode to start.
520*4882a593Smuzhiyun 	 * Normally, this takes at least 50 ms.
521*4882a593Smuzhiyun 	 */
522*4882a593Smuzhiyun 	msleep(50);
523*4882a593Smuzhiyun 	/*
524*4882a593Smuzhiyun 	 * In addition, when a device boots for the first time after being
525*4882a593Smuzhiyun 	 * updated to new firmware, it must first calibrate its sensors, which
526*4882a593Smuzhiyun 	 * can take up to an additional 2 seconds. If the device power is
527*4882a593Smuzhiyun 	 * running low, this may take even longer.
528*4882a593Smuzhiyun 	 */
529*4882a593Smuzhiyun 	error = cyapa_poll_state(cyapa, 4000);
530*4882a593Smuzhiyun 	if (error < 0)
531*4882a593Smuzhiyun 		return error;
532*4882a593Smuzhiyun 	if (cyapa->state != CYAPA_STATE_OP)
533*4882a593Smuzhiyun 		return -EAGAIN;
534*4882a593Smuzhiyun 
535*4882a593Smuzhiyun 	return 0;
536*4882a593Smuzhiyun }
537*4882a593Smuzhiyun 
cyapa_gen3_csum(const u8 * buf,size_t count)538*4882a593Smuzhiyun static u16 cyapa_gen3_csum(const u8 *buf, size_t count)
539*4882a593Smuzhiyun {
540*4882a593Smuzhiyun 	int i;
541*4882a593Smuzhiyun 	u16 csum = 0;
542*4882a593Smuzhiyun 
543*4882a593Smuzhiyun 	for (i = 0; i < count; i++)
544*4882a593Smuzhiyun 		csum += buf[i];
545*4882a593Smuzhiyun 
546*4882a593Smuzhiyun 	return csum;
547*4882a593Smuzhiyun }
548*4882a593Smuzhiyun 
549*4882a593Smuzhiyun /*
550*4882a593Smuzhiyun  * Verify the integrity of a CYAPA firmware image file.
551*4882a593Smuzhiyun  *
552*4882a593Smuzhiyun  * The firmware image file is 30848 bytes, composed of 482 64-byte blocks.
553*4882a593Smuzhiyun  *
554*4882a593Smuzhiyun  * The first 2 blocks are the firmware header.
555*4882a593Smuzhiyun  * The next 480 blocks are the firmware image.
556*4882a593Smuzhiyun  *
557*4882a593Smuzhiyun  * The first two bytes of the header hold the header checksum, computed by
558*4882a593Smuzhiyun  * summing the other 126 bytes of the header.
559*4882a593Smuzhiyun  * The last two bytes of the header hold the firmware image checksum, computed
560*4882a593Smuzhiyun  * by summing the 30720 bytes of the image modulo 0xffff.
561*4882a593Smuzhiyun  *
562*4882a593Smuzhiyun  * Both checksums are stored little-endian.
563*4882a593Smuzhiyun  */
cyapa_gen3_check_fw(struct cyapa * cyapa,const struct firmware * fw)564*4882a593Smuzhiyun static int cyapa_gen3_check_fw(struct cyapa *cyapa, const struct firmware *fw)
565*4882a593Smuzhiyun {
566*4882a593Smuzhiyun 	struct device *dev = &cyapa->client->dev;
567*4882a593Smuzhiyun 	u16 csum;
568*4882a593Smuzhiyun 	u16 csum_expected;
569*4882a593Smuzhiyun 
570*4882a593Smuzhiyun 	/* Firmware must match exact 30848 bytes = 482 64-byte blocks. */
571*4882a593Smuzhiyun 	if (fw->size != CYAPA_FW_SIZE) {
572*4882a593Smuzhiyun 		dev_err(dev, "invalid firmware size = %zu, expected %u.\n",
573*4882a593Smuzhiyun 			fw->size, CYAPA_FW_SIZE);
574*4882a593Smuzhiyun 		return -EINVAL;
575*4882a593Smuzhiyun 	}
576*4882a593Smuzhiyun 
577*4882a593Smuzhiyun 	/* Verify header block */
578*4882a593Smuzhiyun 	csum_expected = (fw->data[0] << 8) | fw->data[1];
579*4882a593Smuzhiyun 	csum = cyapa_gen3_csum(&fw->data[2], CYAPA_FW_HDR_SIZE - 2);
580*4882a593Smuzhiyun 	if (csum != csum_expected) {
581*4882a593Smuzhiyun 		dev_err(dev, "%s %04x, expected: %04x\n",
582*4882a593Smuzhiyun 			"invalid firmware header checksum = ",
583*4882a593Smuzhiyun 			csum, csum_expected);
584*4882a593Smuzhiyun 		return -EINVAL;
585*4882a593Smuzhiyun 	}
586*4882a593Smuzhiyun 
587*4882a593Smuzhiyun 	/* Verify firmware image */
588*4882a593Smuzhiyun 	csum_expected = (fw->data[CYAPA_FW_HDR_SIZE - 2] << 8) |
589*4882a593Smuzhiyun 			 fw->data[CYAPA_FW_HDR_SIZE - 1];
590*4882a593Smuzhiyun 	csum = cyapa_gen3_csum(&fw->data[CYAPA_FW_HDR_SIZE],
591*4882a593Smuzhiyun 			CYAPA_FW_DATA_SIZE);
592*4882a593Smuzhiyun 	if (csum != csum_expected) {
593*4882a593Smuzhiyun 		dev_err(dev, "%s %04x, expected: %04x\n",
594*4882a593Smuzhiyun 			"invalid firmware header checksum = ",
595*4882a593Smuzhiyun 			csum, csum_expected);
596*4882a593Smuzhiyun 		return -EINVAL;
597*4882a593Smuzhiyun 	}
598*4882a593Smuzhiyun 	return 0;
599*4882a593Smuzhiyun }
600*4882a593Smuzhiyun 
601*4882a593Smuzhiyun /*
602*4882a593Smuzhiyun  * Write a |len| byte long buffer |buf| to the device, by chopping it up into a
603*4882a593Smuzhiyun  * sequence of smaller |CYAPA_CMD_LEN|-length write commands.
604*4882a593Smuzhiyun  *
605*4882a593Smuzhiyun  * The data bytes for a write command are prepended with the 1-byte offset
606*4882a593Smuzhiyun  * of the data relative to the start of |buf|.
607*4882a593Smuzhiyun  */
cyapa_gen3_write_buffer(struct cyapa * cyapa,const u8 * buf,size_t len)608*4882a593Smuzhiyun static int cyapa_gen3_write_buffer(struct cyapa *cyapa,
609*4882a593Smuzhiyun 		const u8 *buf, size_t len)
610*4882a593Smuzhiyun {
611*4882a593Smuzhiyun 	int error;
612*4882a593Smuzhiyun 	size_t i;
613*4882a593Smuzhiyun 	unsigned char cmd[CYAPA_CMD_LEN + 1];
614*4882a593Smuzhiyun 	size_t cmd_len;
615*4882a593Smuzhiyun 
616*4882a593Smuzhiyun 	for (i = 0; i < len; i += CYAPA_CMD_LEN) {
617*4882a593Smuzhiyun 		const u8 *payload = &buf[i];
618*4882a593Smuzhiyun 
619*4882a593Smuzhiyun 		cmd_len = (len - i >= CYAPA_CMD_LEN) ? CYAPA_CMD_LEN : len - i;
620*4882a593Smuzhiyun 		cmd[0] = i;
621*4882a593Smuzhiyun 		memcpy(&cmd[1], payload, cmd_len);
622*4882a593Smuzhiyun 
623*4882a593Smuzhiyun 		error = cyapa_i2c_reg_write_block(cyapa, 0, cmd_len + 1, cmd);
624*4882a593Smuzhiyun 		if (error)
625*4882a593Smuzhiyun 			return error;
626*4882a593Smuzhiyun 	}
627*4882a593Smuzhiyun 	return 0;
628*4882a593Smuzhiyun }
629*4882a593Smuzhiyun 
630*4882a593Smuzhiyun /*
631*4882a593Smuzhiyun  * A firmware block write command writes 64 bytes of data to a single flash
632*4882a593Smuzhiyun  * page in the device.  The 78-byte block write command has the format:
633*4882a593Smuzhiyun  *   <0xff> <CMD> <Key> <Start> <Data> <Data-Checksum> <CMD Checksum>
634*4882a593Smuzhiyun  *
635*4882a593Smuzhiyun  *  <0xff>  - every command starts with 0xff
636*4882a593Smuzhiyun  *  <CMD>   - the write command value is 0x39
637*4882a593Smuzhiyun  *  <Key>   - write commands include an 8-byte key: { 00 01 02 03 04 05 06 07 }
638*4882a593Smuzhiyun  *  <Block> - Memory Block number (address / 64) (16-bit, big-endian)
639*4882a593Smuzhiyun  *  <Data>  - 64 bytes of firmware image data
640*4882a593Smuzhiyun  *  <Data Checksum> - sum of 64 <Data> bytes, modulo 0xff
641*4882a593Smuzhiyun  *  <CMD Checksum> - sum of 77 bytes, from 0xff to <Data Checksum>
642*4882a593Smuzhiyun  *
643*4882a593Smuzhiyun  * Each write command is split into 5 i2c write transactions of up to 16 bytes.
644*4882a593Smuzhiyun  * Each transaction starts with an i2c register offset: (00, 10, 20, 30, 40).
645*4882a593Smuzhiyun  */
cyapa_gen3_write_fw_block(struct cyapa * cyapa,u16 block,const u8 * data)646*4882a593Smuzhiyun static int cyapa_gen3_write_fw_block(struct cyapa *cyapa,
647*4882a593Smuzhiyun 		u16 block, const u8 *data)
648*4882a593Smuzhiyun {
649*4882a593Smuzhiyun 	int ret;
650*4882a593Smuzhiyun 	struct gen3_write_block_cmd write_block_cmd;
651*4882a593Smuzhiyun 	u8 status[BL_STATUS_SIZE];
652*4882a593Smuzhiyun 	int tries;
653*4882a593Smuzhiyun 	u8 bl_status, bl_error;
654*4882a593Smuzhiyun 
655*4882a593Smuzhiyun 	/* Set write command and security key bytes. */
656*4882a593Smuzhiyun 	write_block_cmd.checksum_seed = GEN3_BL_CMD_CHECKSUM_SEED;
657*4882a593Smuzhiyun 	write_block_cmd.cmd_code = GEN3_BL_CMD_WRITE_BLOCK;
658*4882a593Smuzhiyun 	memcpy(write_block_cmd.key, security_key, sizeof(security_key));
659*4882a593Smuzhiyun 	put_unaligned_be16(block, &write_block_cmd.block_num);
660*4882a593Smuzhiyun 	memcpy(write_block_cmd.block_data, data, CYAPA_FW_BLOCK_SIZE);
661*4882a593Smuzhiyun 	write_block_cmd.block_checksum = cyapa_gen3_csum(
662*4882a593Smuzhiyun 			write_block_cmd.block_data, CYAPA_FW_BLOCK_SIZE);
663*4882a593Smuzhiyun 	write_block_cmd.cmd_checksum = cyapa_gen3_csum((u8 *)&write_block_cmd,
664*4882a593Smuzhiyun 			sizeof(write_block_cmd) - 1);
665*4882a593Smuzhiyun 
666*4882a593Smuzhiyun 	ret = cyapa_gen3_write_buffer(cyapa, (u8 *)&write_block_cmd,
667*4882a593Smuzhiyun 			sizeof(write_block_cmd));
668*4882a593Smuzhiyun 	if (ret)
669*4882a593Smuzhiyun 		return ret;
670*4882a593Smuzhiyun 
671*4882a593Smuzhiyun 	/* Wait for write to finish */
672*4882a593Smuzhiyun 	tries = 11;  /* Programming for one block can take about 100ms. */
673*4882a593Smuzhiyun 	do {
674*4882a593Smuzhiyun 		usleep_range(10000, 20000);
675*4882a593Smuzhiyun 
676*4882a593Smuzhiyun 		/* Check block write command result status. */
677*4882a593Smuzhiyun 		ret = cyapa_i2c_reg_read_block(cyapa, BL_HEAD_OFFSET,
678*4882a593Smuzhiyun 					       BL_STATUS_SIZE, status);
679*4882a593Smuzhiyun 		if (ret != BL_STATUS_SIZE)
680*4882a593Smuzhiyun 			return (ret < 0) ? ret : -EIO;
681*4882a593Smuzhiyun 	} while ((status[REG_BL_STATUS] & BL_STATUS_BUSY) && --tries);
682*4882a593Smuzhiyun 
683*4882a593Smuzhiyun 	/* Ignore WATCHDOG bit and reserved bits. */
684*4882a593Smuzhiyun 	bl_status = status[REG_BL_STATUS] & ~BL_STATUS_REV_MASK;
685*4882a593Smuzhiyun 	bl_error = status[REG_BL_ERROR] & ~BL_ERROR_RESERVED;
686*4882a593Smuzhiyun 
687*4882a593Smuzhiyun 	if (bl_status & BL_STATUS_BUSY)
688*4882a593Smuzhiyun 		ret = -ETIMEDOUT;
689*4882a593Smuzhiyun 	else if (bl_status != BL_STATUS_RUNNING ||
690*4882a593Smuzhiyun 		bl_error != BL_ERROR_BOOTLOADING)
691*4882a593Smuzhiyun 		ret = -EIO;
692*4882a593Smuzhiyun 	else
693*4882a593Smuzhiyun 		ret = 0;
694*4882a593Smuzhiyun 
695*4882a593Smuzhiyun 	return ret;
696*4882a593Smuzhiyun }
697*4882a593Smuzhiyun 
cyapa_gen3_write_blocks(struct cyapa * cyapa,size_t start_block,size_t block_count,const u8 * image_data)698*4882a593Smuzhiyun static int cyapa_gen3_write_blocks(struct cyapa *cyapa,
699*4882a593Smuzhiyun 		size_t start_block, size_t block_count,
700*4882a593Smuzhiyun 		const u8 *image_data)
701*4882a593Smuzhiyun {
702*4882a593Smuzhiyun 	int error;
703*4882a593Smuzhiyun 	int i;
704*4882a593Smuzhiyun 
705*4882a593Smuzhiyun 	for (i = 0; i < block_count; i++) {
706*4882a593Smuzhiyun 		size_t block = start_block + i;
707*4882a593Smuzhiyun 		size_t addr = i * CYAPA_FW_BLOCK_SIZE;
708*4882a593Smuzhiyun 		const u8 *data = &image_data[addr];
709*4882a593Smuzhiyun 
710*4882a593Smuzhiyun 		error = cyapa_gen3_write_fw_block(cyapa, block, data);
711*4882a593Smuzhiyun 		if (error)
712*4882a593Smuzhiyun 			return error;
713*4882a593Smuzhiyun 	}
714*4882a593Smuzhiyun 	return 0;
715*4882a593Smuzhiyun }
716*4882a593Smuzhiyun 
cyapa_gen3_do_fw_update(struct cyapa * cyapa,const struct firmware * fw)717*4882a593Smuzhiyun static int cyapa_gen3_do_fw_update(struct cyapa *cyapa,
718*4882a593Smuzhiyun 		const struct firmware *fw)
719*4882a593Smuzhiyun {
720*4882a593Smuzhiyun 	struct device *dev = &cyapa->client->dev;
721*4882a593Smuzhiyun 	int error;
722*4882a593Smuzhiyun 
723*4882a593Smuzhiyun 	/* First write data, starting at byte 128 of fw->data */
724*4882a593Smuzhiyun 	error = cyapa_gen3_write_blocks(cyapa,
725*4882a593Smuzhiyun 		CYAPA_FW_DATA_BLOCK_START, CYAPA_FW_DATA_BLOCK_COUNT,
726*4882a593Smuzhiyun 		&fw->data[CYAPA_FW_HDR_BLOCK_COUNT * CYAPA_FW_BLOCK_SIZE]);
727*4882a593Smuzhiyun 	if (error) {
728*4882a593Smuzhiyun 		dev_err(dev, "FW update aborted, write image: %d\n", error);
729*4882a593Smuzhiyun 		return error;
730*4882a593Smuzhiyun 	}
731*4882a593Smuzhiyun 
732*4882a593Smuzhiyun 	/* Then write checksum */
733*4882a593Smuzhiyun 	error = cyapa_gen3_write_blocks(cyapa,
734*4882a593Smuzhiyun 		CYAPA_FW_HDR_BLOCK_START, CYAPA_FW_HDR_BLOCK_COUNT,
735*4882a593Smuzhiyun 		&fw->data[0]);
736*4882a593Smuzhiyun 	if (error) {
737*4882a593Smuzhiyun 		dev_err(dev, "FW update aborted, write checksum: %d\n", error);
738*4882a593Smuzhiyun 		return error;
739*4882a593Smuzhiyun 	}
740*4882a593Smuzhiyun 
741*4882a593Smuzhiyun 	return 0;
742*4882a593Smuzhiyun }
743*4882a593Smuzhiyun 
cyapa_gen3_do_calibrate(struct device * dev,struct device_attribute * attr,const char * buf,size_t count)744*4882a593Smuzhiyun static ssize_t cyapa_gen3_do_calibrate(struct device *dev,
745*4882a593Smuzhiyun 				     struct device_attribute *attr,
746*4882a593Smuzhiyun 				     const char *buf, size_t count)
747*4882a593Smuzhiyun {
748*4882a593Smuzhiyun 	struct cyapa *cyapa = dev_get_drvdata(dev);
749*4882a593Smuzhiyun 	unsigned long timeout;
750*4882a593Smuzhiyun 	int ret;
751*4882a593Smuzhiyun 
752*4882a593Smuzhiyun 	ret = cyapa_read_byte(cyapa, CYAPA_CMD_DEV_STATUS);
753*4882a593Smuzhiyun 	if (ret < 0) {
754*4882a593Smuzhiyun 		dev_err(dev, "Error reading dev status: %d\n", ret);
755*4882a593Smuzhiyun 		goto out;
756*4882a593Smuzhiyun 	}
757*4882a593Smuzhiyun 	if ((ret & CYAPA_DEV_NORMAL) != CYAPA_DEV_NORMAL) {
758*4882a593Smuzhiyun 		dev_warn(dev, "Trackpad device is busy, device state: 0x%02x\n",
759*4882a593Smuzhiyun 			 ret);
760*4882a593Smuzhiyun 		ret = -EAGAIN;
761*4882a593Smuzhiyun 		goto out;
762*4882a593Smuzhiyun 	}
763*4882a593Smuzhiyun 
764*4882a593Smuzhiyun 	ret = cyapa_write_byte(cyapa, CYAPA_CMD_SOFT_RESET,
765*4882a593Smuzhiyun 			       OP_RECALIBRATION_MASK);
766*4882a593Smuzhiyun 	if (ret < 0) {
767*4882a593Smuzhiyun 		dev_err(dev, "Failed to send calibrate command: %d\n",
768*4882a593Smuzhiyun 			ret);
769*4882a593Smuzhiyun 		goto out;
770*4882a593Smuzhiyun 	}
771*4882a593Smuzhiyun 
772*4882a593Smuzhiyun 	/* max recalibration timeout 2s. */
773*4882a593Smuzhiyun 	timeout = jiffies + 2 * HZ;
774*4882a593Smuzhiyun 	do {
775*4882a593Smuzhiyun 		/*
776*4882a593Smuzhiyun 		 * For this recalibration, the max time will not exceed 2s.
777*4882a593Smuzhiyun 		 * The average time is approximately 500 - 700 ms, and we
778*4882a593Smuzhiyun 		 * will check the status every 100 - 200ms.
779*4882a593Smuzhiyun 		 */
780*4882a593Smuzhiyun 		msleep(100);
781*4882a593Smuzhiyun 		ret = cyapa_read_byte(cyapa, CYAPA_CMD_DEV_STATUS);
782*4882a593Smuzhiyun 		if (ret < 0) {
783*4882a593Smuzhiyun 			dev_err(dev, "Error reading dev status: %d\n", ret);
784*4882a593Smuzhiyun 			goto out;
785*4882a593Smuzhiyun 		}
786*4882a593Smuzhiyun 		if ((ret & CYAPA_DEV_NORMAL) == CYAPA_DEV_NORMAL) {
787*4882a593Smuzhiyun 			dev_dbg(dev, "Calibration successful.\n");
788*4882a593Smuzhiyun 			goto out;
789*4882a593Smuzhiyun 		}
790*4882a593Smuzhiyun 	} while (time_is_after_jiffies(timeout));
791*4882a593Smuzhiyun 
792*4882a593Smuzhiyun 	dev_err(dev, "Failed to calibrate. Timeout.\n");
793*4882a593Smuzhiyun 	ret = -ETIMEDOUT;
794*4882a593Smuzhiyun 
795*4882a593Smuzhiyun out:
796*4882a593Smuzhiyun 	return ret < 0 ? ret : count;
797*4882a593Smuzhiyun }
798*4882a593Smuzhiyun 
cyapa_gen3_show_baseline(struct device * dev,struct device_attribute * attr,char * buf)799*4882a593Smuzhiyun static ssize_t cyapa_gen3_show_baseline(struct device *dev,
800*4882a593Smuzhiyun 				   struct device_attribute *attr, char *buf)
801*4882a593Smuzhiyun {
802*4882a593Smuzhiyun 	struct cyapa *cyapa = dev_get_drvdata(dev);
803*4882a593Smuzhiyun 	int max_baseline, min_baseline;
804*4882a593Smuzhiyun 	int tries;
805*4882a593Smuzhiyun 	int ret;
806*4882a593Smuzhiyun 
807*4882a593Smuzhiyun 	ret = cyapa_read_byte(cyapa, CYAPA_CMD_DEV_STATUS);
808*4882a593Smuzhiyun 	if (ret < 0) {
809*4882a593Smuzhiyun 		dev_err(dev, "Error reading dev status. err = %d\n", ret);
810*4882a593Smuzhiyun 		goto out;
811*4882a593Smuzhiyun 	}
812*4882a593Smuzhiyun 	if ((ret & CYAPA_DEV_NORMAL) != CYAPA_DEV_NORMAL) {
813*4882a593Smuzhiyun 		dev_warn(dev, "Trackpad device is busy. device state = 0x%x\n",
814*4882a593Smuzhiyun 			 ret);
815*4882a593Smuzhiyun 		ret = -EAGAIN;
816*4882a593Smuzhiyun 		goto out;
817*4882a593Smuzhiyun 	}
818*4882a593Smuzhiyun 
819*4882a593Smuzhiyun 	ret = cyapa_write_byte(cyapa, CYAPA_CMD_SOFT_RESET,
820*4882a593Smuzhiyun 			       OP_REPORT_BASELINE_MASK);
821*4882a593Smuzhiyun 	if (ret < 0) {
822*4882a593Smuzhiyun 		dev_err(dev, "Failed to send report baseline command. %d\n",
823*4882a593Smuzhiyun 			ret);
824*4882a593Smuzhiyun 		goto out;
825*4882a593Smuzhiyun 	}
826*4882a593Smuzhiyun 
827*4882a593Smuzhiyun 	tries = 3;  /* Try for 30 to 60 ms */
828*4882a593Smuzhiyun 	do {
829*4882a593Smuzhiyun 		usleep_range(10000, 20000);
830*4882a593Smuzhiyun 
831*4882a593Smuzhiyun 		ret = cyapa_read_byte(cyapa, CYAPA_CMD_DEV_STATUS);
832*4882a593Smuzhiyun 		if (ret < 0) {
833*4882a593Smuzhiyun 			dev_err(dev, "Error reading dev status. err = %d\n",
834*4882a593Smuzhiyun 				ret);
835*4882a593Smuzhiyun 			goto out;
836*4882a593Smuzhiyun 		}
837*4882a593Smuzhiyun 		if ((ret & CYAPA_DEV_NORMAL) == CYAPA_DEV_NORMAL)
838*4882a593Smuzhiyun 			break;
839*4882a593Smuzhiyun 	} while (--tries);
840*4882a593Smuzhiyun 
841*4882a593Smuzhiyun 	if (tries == 0) {
842*4882a593Smuzhiyun 		dev_err(dev, "Device timed out going to Normal state.\n");
843*4882a593Smuzhiyun 		ret = -ETIMEDOUT;
844*4882a593Smuzhiyun 		goto out;
845*4882a593Smuzhiyun 	}
846*4882a593Smuzhiyun 
847*4882a593Smuzhiyun 	ret = cyapa_read_byte(cyapa, CYAPA_CMD_MAX_BASELINE);
848*4882a593Smuzhiyun 	if (ret < 0) {
849*4882a593Smuzhiyun 		dev_err(dev, "Failed to read max baseline. err = %d\n", ret);
850*4882a593Smuzhiyun 		goto out;
851*4882a593Smuzhiyun 	}
852*4882a593Smuzhiyun 	max_baseline = ret;
853*4882a593Smuzhiyun 
854*4882a593Smuzhiyun 	ret = cyapa_read_byte(cyapa, CYAPA_CMD_MIN_BASELINE);
855*4882a593Smuzhiyun 	if (ret < 0) {
856*4882a593Smuzhiyun 		dev_err(dev, "Failed to read min baseline. err = %d\n", ret);
857*4882a593Smuzhiyun 		goto out;
858*4882a593Smuzhiyun 	}
859*4882a593Smuzhiyun 	min_baseline = ret;
860*4882a593Smuzhiyun 
861*4882a593Smuzhiyun 	dev_dbg(dev, "Baseline report successful. Max: %d Min: %d\n",
862*4882a593Smuzhiyun 		max_baseline, min_baseline);
863*4882a593Smuzhiyun 	ret = scnprintf(buf, PAGE_SIZE, "%d %d\n", max_baseline, min_baseline);
864*4882a593Smuzhiyun 
865*4882a593Smuzhiyun out:
866*4882a593Smuzhiyun 	return ret;
867*4882a593Smuzhiyun }
868*4882a593Smuzhiyun 
869*4882a593Smuzhiyun /*
870*4882a593Smuzhiyun  * cyapa_get_wait_time_for_pwr_cmd
871*4882a593Smuzhiyun  *
872*4882a593Smuzhiyun  * Compute the amount of time we need to wait after updating the touchpad
873*4882a593Smuzhiyun  * power mode. The touchpad needs to consume the incoming power mode set
874*4882a593Smuzhiyun  * command at the current clock rate.
875*4882a593Smuzhiyun  */
876*4882a593Smuzhiyun 
cyapa_get_wait_time_for_pwr_cmd(u8 pwr_mode)877*4882a593Smuzhiyun static u16 cyapa_get_wait_time_for_pwr_cmd(u8 pwr_mode)
878*4882a593Smuzhiyun {
879*4882a593Smuzhiyun 	switch (pwr_mode) {
880*4882a593Smuzhiyun 	case PWR_MODE_FULL_ACTIVE: return 20;
881*4882a593Smuzhiyun 	case PWR_MODE_BTN_ONLY: return 20;
882*4882a593Smuzhiyun 	case PWR_MODE_OFF: return 20;
883*4882a593Smuzhiyun 	default: return cyapa_pwr_cmd_to_sleep_time(pwr_mode) + 50;
884*4882a593Smuzhiyun 	}
885*4882a593Smuzhiyun }
886*4882a593Smuzhiyun 
887*4882a593Smuzhiyun /*
888*4882a593Smuzhiyun  * Set device power mode
889*4882a593Smuzhiyun  *
890*4882a593Smuzhiyun  * Write to the field to configure power state. Power states include :
891*4882a593Smuzhiyun  *   Full : Max scans and report rate.
892*4882a593Smuzhiyun  *   Idle : Report rate set by user specified time.
893*4882a593Smuzhiyun  *   ButtonOnly : No scans for fingers. When the button is triggered,
894*4882a593Smuzhiyun  *     a slave interrupt is asserted to notify host to wake up.
895*4882a593Smuzhiyun  *   Off : Only awake for i2c commands from host. No function for button
896*4882a593Smuzhiyun  *     or touch sensors.
897*4882a593Smuzhiyun  *
898*4882a593Smuzhiyun  * The power_mode command should conform to the following :
899*4882a593Smuzhiyun  *   Full : 0x3f
900*4882a593Smuzhiyun  *   Idle : Configurable from 20 to 1000ms. See note below for
901*4882a593Smuzhiyun  *     cyapa_sleep_time_to_pwr_cmd and cyapa_pwr_cmd_to_sleep_time
902*4882a593Smuzhiyun  *   ButtonOnly : 0x01
903*4882a593Smuzhiyun  *   Off : 0x00
904*4882a593Smuzhiyun  *
905*4882a593Smuzhiyun  * Device power mode can only be set when device is in operational mode.
906*4882a593Smuzhiyun  */
cyapa_gen3_set_power_mode(struct cyapa * cyapa,u8 power_mode,u16 always_unused,enum cyapa_pm_stage pm_stage)907*4882a593Smuzhiyun static int cyapa_gen3_set_power_mode(struct cyapa *cyapa, u8 power_mode,
908*4882a593Smuzhiyun 		u16 always_unused, enum cyapa_pm_stage pm_stage)
909*4882a593Smuzhiyun {
910*4882a593Smuzhiyun 	struct input_dev *input = cyapa->input;
911*4882a593Smuzhiyun 	u8 power;
912*4882a593Smuzhiyun 	int tries;
913*4882a593Smuzhiyun 	int sleep_time;
914*4882a593Smuzhiyun 	int interval;
915*4882a593Smuzhiyun 	int ret;
916*4882a593Smuzhiyun 
917*4882a593Smuzhiyun 	if (cyapa->state != CYAPA_STATE_OP)
918*4882a593Smuzhiyun 		return 0;
919*4882a593Smuzhiyun 
920*4882a593Smuzhiyun 	tries = SET_POWER_MODE_TRIES;
921*4882a593Smuzhiyun 	while (tries--) {
922*4882a593Smuzhiyun 		ret = cyapa_read_byte(cyapa, CYAPA_CMD_POWER_MODE);
923*4882a593Smuzhiyun 		if (ret >= 0)
924*4882a593Smuzhiyun 			break;
925*4882a593Smuzhiyun 		usleep_range(SET_POWER_MODE_DELAY, 2 * SET_POWER_MODE_DELAY);
926*4882a593Smuzhiyun 	}
927*4882a593Smuzhiyun 	if (ret < 0)
928*4882a593Smuzhiyun 		return ret;
929*4882a593Smuzhiyun 
930*4882a593Smuzhiyun 	/*
931*4882a593Smuzhiyun 	 * Return early if the power mode to set is the same as the current
932*4882a593Smuzhiyun 	 * one.
933*4882a593Smuzhiyun 	 */
934*4882a593Smuzhiyun 	if ((ret & PWR_MODE_MASK) == power_mode)
935*4882a593Smuzhiyun 		return 0;
936*4882a593Smuzhiyun 
937*4882a593Smuzhiyun 	sleep_time = (int)cyapa_get_wait_time_for_pwr_cmd(ret & PWR_MODE_MASK);
938*4882a593Smuzhiyun 	power = ret;
939*4882a593Smuzhiyun 	power &= ~PWR_MODE_MASK;
940*4882a593Smuzhiyun 	power |= power_mode & PWR_MODE_MASK;
941*4882a593Smuzhiyun 	tries = SET_POWER_MODE_TRIES;
942*4882a593Smuzhiyun 	while (tries--) {
943*4882a593Smuzhiyun 		ret = cyapa_write_byte(cyapa, CYAPA_CMD_POWER_MODE, power);
944*4882a593Smuzhiyun 		if (!ret)
945*4882a593Smuzhiyun 			break;
946*4882a593Smuzhiyun 		usleep_range(SET_POWER_MODE_DELAY, 2 * SET_POWER_MODE_DELAY);
947*4882a593Smuzhiyun 	}
948*4882a593Smuzhiyun 
949*4882a593Smuzhiyun 	/*
950*4882a593Smuzhiyun 	 * Wait for the newly set power command to go in at the previous
951*4882a593Smuzhiyun 	 * clock speed (scanrate) used by the touchpad firmware. Not
952*4882a593Smuzhiyun 	 * doing so before issuing the next command may result in errors
953*4882a593Smuzhiyun 	 * depending on the command's content.
954*4882a593Smuzhiyun 	 */
955*4882a593Smuzhiyun 	if (cyapa->operational && input && input->users &&
956*4882a593Smuzhiyun 	    (pm_stage == CYAPA_PM_RUNTIME_SUSPEND ||
957*4882a593Smuzhiyun 	     pm_stage == CYAPA_PM_RUNTIME_RESUME)) {
958*4882a593Smuzhiyun 		/* Try to polling in 120Hz, read may fail, just ignore it. */
959*4882a593Smuzhiyun 		interval = 1000 / 120;
960*4882a593Smuzhiyun 		while (sleep_time > 0) {
961*4882a593Smuzhiyun 			if (sleep_time > interval)
962*4882a593Smuzhiyun 				msleep(interval);
963*4882a593Smuzhiyun 			else
964*4882a593Smuzhiyun 				msleep(sleep_time);
965*4882a593Smuzhiyun 			sleep_time -= interval;
966*4882a593Smuzhiyun 			cyapa_gen3_try_poll_handler(cyapa);
967*4882a593Smuzhiyun 		}
968*4882a593Smuzhiyun 	} else {
969*4882a593Smuzhiyun 		msleep(sleep_time);
970*4882a593Smuzhiyun 	}
971*4882a593Smuzhiyun 
972*4882a593Smuzhiyun 	return ret;
973*4882a593Smuzhiyun }
974*4882a593Smuzhiyun 
cyapa_gen3_set_proximity(struct cyapa * cyapa,bool enable)975*4882a593Smuzhiyun static int cyapa_gen3_set_proximity(struct cyapa *cyapa, bool enable)
976*4882a593Smuzhiyun {
977*4882a593Smuzhiyun 	return -EOPNOTSUPP;
978*4882a593Smuzhiyun }
979*4882a593Smuzhiyun 
cyapa_gen3_get_query_data(struct cyapa * cyapa)980*4882a593Smuzhiyun static int cyapa_gen3_get_query_data(struct cyapa *cyapa)
981*4882a593Smuzhiyun {
982*4882a593Smuzhiyun 	u8 query_data[QUERY_DATA_SIZE];
983*4882a593Smuzhiyun 	int ret;
984*4882a593Smuzhiyun 
985*4882a593Smuzhiyun 	if (cyapa->state != CYAPA_STATE_OP)
986*4882a593Smuzhiyun 		return -EBUSY;
987*4882a593Smuzhiyun 
988*4882a593Smuzhiyun 	ret = cyapa_read_block(cyapa, CYAPA_CMD_GROUP_QUERY, query_data);
989*4882a593Smuzhiyun 	if (ret != QUERY_DATA_SIZE)
990*4882a593Smuzhiyun 		return (ret < 0) ? ret : -EIO;
991*4882a593Smuzhiyun 
992*4882a593Smuzhiyun 	memcpy(&cyapa->product_id[0], &query_data[0], 5);
993*4882a593Smuzhiyun 	cyapa->product_id[5] = '-';
994*4882a593Smuzhiyun 	memcpy(&cyapa->product_id[6], &query_data[5], 6);
995*4882a593Smuzhiyun 	cyapa->product_id[12] = '-';
996*4882a593Smuzhiyun 	memcpy(&cyapa->product_id[13], &query_data[11], 2);
997*4882a593Smuzhiyun 	cyapa->product_id[15] = '\0';
998*4882a593Smuzhiyun 
999*4882a593Smuzhiyun 	cyapa->fw_maj_ver = query_data[15];
1000*4882a593Smuzhiyun 	cyapa->fw_min_ver = query_data[16];
1001*4882a593Smuzhiyun 
1002*4882a593Smuzhiyun 	cyapa->btn_capability = query_data[19] & CAPABILITY_BTN_MASK;
1003*4882a593Smuzhiyun 
1004*4882a593Smuzhiyun 	cyapa->gen = query_data[20] & 0x0f;
1005*4882a593Smuzhiyun 
1006*4882a593Smuzhiyun 	cyapa->max_abs_x = ((query_data[21] & 0xf0) << 4) | query_data[22];
1007*4882a593Smuzhiyun 	cyapa->max_abs_y = ((query_data[21] & 0x0f) << 8) | query_data[23];
1008*4882a593Smuzhiyun 
1009*4882a593Smuzhiyun 	cyapa->physical_size_x =
1010*4882a593Smuzhiyun 		((query_data[24] & 0xf0) << 4) | query_data[25];
1011*4882a593Smuzhiyun 	cyapa->physical_size_y =
1012*4882a593Smuzhiyun 		((query_data[24] & 0x0f) << 8) | query_data[26];
1013*4882a593Smuzhiyun 
1014*4882a593Smuzhiyun 	cyapa->max_z = 255;
1015*4882a593Smuzhiyun 
1016*4882a593Smuzhiyun 	return 0;
1017*4882a593Smuzhiyun }
1018*4882a593Smuzhiyun 
cyapa_gen3_bl_query_data(struct cyapa * cyapa)1019*4882a593Smuzhiyun static int cyapa_gen3_bl_query_data(struct cyapa *cyapa)
1020*4882a593Smuzhiyun {
1021*4882a593Smuzhiyun 	u8 bl_data[CYAPA_CMD_LEN];
1022*4882a593Smuzhiyun 	int ret;
1023*4882a593Smuzhiyun 
1024*4882a593Smuzhiyun 	ret = cyapa_i2c_reg_read_block(cyapa, 0, CYAPA_CMD_LEN, bl_data);
1025*4882a593Smuzhiyun 	if (ret != CYAPA_CMD_LEN)
1026*4882a593Smuzhiyun 		return (ret < 0) ? ret : -EIO;
1027*4882a593Smuzhiyun 
1028*4882a593Smuzhiyun 	/*
1029*4882a593Smuzhiyun 	 * This value will be updated again when entered application mode.
1030*4882a593Smuzhiyun 	 * If TP failed to enter application mode, this fw version values
1031*4882a593Smuzhiyun 	 * can be used as a reference.
1032*4882a593Smuzhiyun 	 * This firmware version valid when fw image checksum is valid.
1033*4882a593Smuzhiyun 	 */
1034*4882a593Smuzhiyun 	if (bl_data[REG_BL_STATUS] ==
1035*4882a593Smuzhiyun 			(BL_STATUS_RUNNING | BL_STATUS_CSUM_VALID)) {
1036*4882a593Smuzhiyun 		cyapa->fw_maj_ver = bl_data[GEN3_BL_IDLE_FW_MAJ_VER_OFFSET];
1037*4882a593Smuzhiyun 		cyapa->fw_min_ver = bl_data[GEN3_BL_IDLE_FW_MIN_VER_OFFSET];
1038*4882a593Smuzhiyun 	}
1039*4882a593Smuzhiyun 
1040*4882a593Smuzhiyun 	return 0;
1041*4882a593Smuzhiyun }
1042*4882a593Smuzhiyun 
1043*4882a593Smuzhiyun /*
1044*4882a593Smuzhiyun  * Check if device is operational.
1045*4882a593Smuzhiyun  *
1046*4882a593Smuzhiyun  * An operational device is responding, has exited bootloader, and has
1047*4882a593Smuzhiyun  * firmware supported by this driver.
1048*4882a593Smuzhiyun  *
1049*4882a593Smuzhiyun  * Returns:
1050*4882a593Smuzhiyun  *   -EBUSY  no device or in bootloader
1051*4882a593Smuzhiyun  *   -EIO    failure while reading from device
1052*4882a593Smuzhiyun  *   -EAGAIN device is still in bootloader
1053*4882a593Smuzhiyun  *           if ->state = CYAPA_STATE_BL_IDLE, device has invalid firmware
1054*4882a593Smuzhiyun  *   -EINVAL device is in operational mode, but not supported by this driver
1055*4882a593Smuzhiyun  *   0       device is supported
1056*4882a593Smuzhiyun  */
cyapa_gen3_do_operational_check(struct cyapa * cyapa)1057*4882a593Smuzhiyun static int cyapa_gen3_do_operational_check(struct cyapa *cyapa)
1058*4882a593Smuzhiyun {
1059*4882a593Smuzhiyun 	struct device *dev = &cyapa->client->dev;
1060*4882a593Smuzhiyun 	int error;
1061*4882a593Smuzhiyun 
1062*4882a593Smuzhiyun 	switch (cyapa->state) {
1063*4882a593Smuzhiyun 	case CYAPA_STATE_BL_ACTIVE:
1064*4882a593Smuzhiyun 		error = cyapa_gen3_bl_deactivate(cyapa);
1065*4882a593Smuzhiyun 		if (error) {
1066*4882a593Smuzhiyun 			dev_err(dev, "failed to bl_deactivate: %d\n", error);
1067*4882a593Smuzhiyun 			return error;
1068*4882a593Smuzhiyun 		}
1069*4882a593Smuzhiyun 
1070*4882a593Smuzhiyun 		fallthrough;
1071*4882a593Smuzhiyun 	case CYAPA_STATE_BL_IDLE:
1072*4882a593Smuzhiyun 		/* Try to get firmware version in bootloader mode. */
1073*4882a593Smuzhiyun 		cyapa_gen3_bl_query_data(cyapa);
1074*4882a593Smuzhiyun 
1075*4882a593Smuzhiyun 		error = cyapa_gen3_bl_exit(cyapa);
1076*4882a593Smuzhiyun 		if (error) {
1077*4882a593Smuzhiyun 			dev_err(dev, "failed to bl_exit: %d\n", error);
1078*4882a593Smuzhiyun 			return error;
1079*4882a593Smuzhiyun 		}
1080*4882a593Smuzhiyun 
1081*4882a593Smuzhiyun 		fallthrough;
1082*4882a593Smuzhiyun 	case CYAPA_STATE_OP:
1083*4882a593Smuzhiyun 		/*
1084*4882a593Smuzhiyun 		 * Reading query data before going back to the full mode
1085*4882a593Smuzhiyun 		 * may cause problems, so we set the power mode first here.
1086*4882a593Smuzhiyun 		 */
1087*4882a593Smuzhiyun 		error = cyapa_gen3_set_power_mode(cyapa,
1088*4882a593Smuzhiyun 				PWR_MODE_FULL_ACTIVE, 0, CYAPA_PM_ACTIVE);
1089*4882a593Smuzhiyun 		if (error)
1090*4882a593Smuzhiyun 			dev_err(dev, "%s: set full power mode failed: %d\n",
1091*4882a593Smuzhiyun 				__func__, error);
1092*4882a593Smuzhiyun 		error = cyapa_gen3_get_query_data(cyapa);
1093*4882a593Smuzhiyun 		if (error < 0)
1094*4882a593Smuzhiyun 			return error;
1095*4882a593Smuzhiyun 
1096*4882a593Smuzhiyun 		/* Only support firmware protocol gen3 */
1097*4882a593Smuzhiyun 		if (cyapa->gen != CYAPA_GEN3) {
1098*4882a593Smuzhiyun 			dev_err(dev, "unsupported protocol version (%d)",
1099*4882a593Smuzhiyun 				cyapa->gen);
1100*4882a593Smuzhiyun 			return -EINVAL;
1101*4882a593Smuzhiyun 		}
1102*4882a593Smuzhiyun 
1103*4882a593Smuzhiyun 		/* Only support product ID starting with CYTRA */
1104*4882a593Smuzhiyun 		if (memcmp(cyapa->product_id, product_id,
1105*4882a593Smuzhiyun 				strlen(product_id)) != 0) {
1106*4882a593Smuzhiyun 			dev_err(dev, "unsupported product ID (%s)\n",
1107*4882a593Smuzhiyun 				cyapa->product_id);
1108*4882a593Smuzhiyun 			return -EINVAL;
1109*4882a593Smuzhiyun 		}
1110*4882a593Smuzhiyun 
1111*4882a593Smuzhiyun 		return 0;
1112*4882a593Smuzhiyun 
1113*4882a593Smuzhiyun 	default:
1114*4882a593Smuzhiyun 		return -EIO;
1115*4882a593Smuzhiyun 	}
1116*4882a593Smuzhiyun 	return 0;
1117*4882a593Smuzhiyun }
1118*4882a593Smuzhiyun 
1119*4882a593Smuzhiyun /*
1120*4882a593Smuzhiyun  * Return false, do not continue process
1121*4882a593Smuzhiyun  * Return true, continue process.
1122*4882a593Smuzhiyun  */
cyapa_gen3_irq_cmd_handler(struct cyapa * cyapa)1123*4882a593Smuzhiyun static bool cyapa_gen3_irq_cmd_handler(struct cyapa *cyapa)
1124*4882a593Smuzhiyun {
1125*4882a593Smuzhiyun 	/* Not gen3 irq command response, skip for continue. */
1126*4882a593Smuzhiyun 	if (cyapa->gen != CYAPA_GEN3)
1127*4882a593Smuzhiyun 		return true;
1128*4882a593Smuzhiyun 
1129*4882a593Smuzhiyun 	if (cyapa->operational)
1130*4882a593Smuzhiyun 		return true;
1131*4882a593Smuzhiyun 
1132*4882a593Smuzhiyun 	/*
1133*4882a593Smuzhiyun 	 * Driver in detecting or other interface function processing,
1134*4882a593Smuzhiyun 	 * so, stop cyapa_gen3_irq_handler to continue process to
1135*4882a593Smuzhiyun 	 * avoid unwanted to error detecting and processing.
1136*4882a593Smuzhiyun 	 *
1137*4882a593Smuzhiyun 	 * And also, avoid the periodically asserted interrupts to be processed
1138*4882a593Smuzhiyun 	 * as touch inputs when gen3 failed to launch into application mode,
1139*4882a593Smuzhiyun 	 * which will cause gen3 stays in bootloader mode.
1140*4882a593Smuzhiyun 	 */
1141*4882a593Smuzhiyun 	return false;
1142*4882a593Smuzhiyun }
1143*4882a593Smuzhiyun 
cyapa_gen3_event_process(struct cyapa * cyapa,struct cyapa_reg_data * data)1144*4882a593Smuzhiyun static int cyapa_gen3_event_process(struct cyapa *cyapa,
1145*4882a593Smuzhiyun 				    struct cyapa_reg_data *data)
1146*4882a593Smuzhiyun {
1147*4882a593Smuzhiyun 	struct input_dev *input = cyapa->input;
1148*4882a593Smuzhiyun 	int num_fingers;
1149*4882a593Smuzhiyun 	int i;
1150*4882a593Smuzhiyun 
1151*4882a593Smuzhiyun 	num_fingers = (data->finger_btn >> 4) & 0x0f;
1152*4882a593Smuzhiyun 	for (i = 0; i < num_fingers; i++) {
1153*4882a593Smuzhiyun 		const struct cyapa_touch *touch = &data->touches[i];
1154*4882a593Smuzhiyun 		/* Note: touch->id range is 1 to 15; slots are 0 to 14. */
1155*4882a593Smuzhiyun 		int slot = touch->id - 1;
1156*4882a593Smuzhiyun 
1157*4882a593Smuzhiyun 		input_mt_slot(input, slot);
1158*4882a593Smuzhiyun 		input_mt_report_slot_state(input, MT_TOOL_FINGER, true);
1159*4882a593Smuzhiyun 		input_report_abs(input, ABS_MT_POSITION_X,
1160*4882a593Smuzhiyun 				 ((touch->xy_hi & 0xf0) << 4) | touch->x_lo);
1161*4882a593Smuzhiyun 		input_report_abs(input, ABS_MT_POSITION_Y,
1162*4882a593Smuzhiyun 				 ((touch->xy_hi & 0x0f) << 8) | touch->y_lo);
1163*4882a593Smuzhiyun 		input_report_abs(input, ABS_MT_PRESSURE, touch->pressure);
1164*4882a593Smuzhiyun 	}
1165*4882a593Smuzhiyun 
1166*4882a593Smuzhiyun 	input_mt_sync_frame(input);
1167*4882a593Smuzhiyun 
1168*4882a593Smuzhiyun 	if (cyapa->btn_capability & CAPABILITY_LEFT_BTN_MASK)
1169*4882a593Smuzhiyun 		input_report_key(input, BTN_LEFT,
1170*4882a593Smuzhiyun 				 !!(data->finger_btn & OP_DATA_LEFT_BTN));
1171*4882a593Smuzhiyun 	if (cyapa->btn_capability & CAPABILITY_MIDDLE_BTN_MASK)
1172*4882a593Smuzhiyun 		input_report_key(input, BTN_MIDDLE,
1173*4882a593Smuzhiyun 				 !!(data->finger_btn & OP_DATA_MIDDLE_BTN));
1174*4882a593Smuzhiyun 	if (cyapa->btn_capability & CAPABILITY_RIGHT_BTN_MASK)
1175*4882a593Smuzhiyun 		input_report_key(input, BTN_RIGHT,
1176*4882a593Smuzhiyun 				 !!(data->finger_btn & OP_DATA_RIGHT_BTN));
1177*4882a593Smuzhiyun 	input_sync(input);
1178*4882a593Smuzhiyun 
1179*4882a593Smuzhiyun 	return 0;
1180*4882a593Smuzhiyun }
1181*4882a593Smuzhiyun 
cyapa_gen3_irq_handler(struct cyapa * cyapa)1182*4882a593Smuzhiyun static int cyapa_gen3_irq_handler(struct cyapa *cyapa)
1183*4882a593Smuzhiyun {
1184*4882a593Smuzhiyun 	struct device *dev = &cyapa->client->dev;
1185*4882a593Smuzhiyun 	struct cyapa_reg_data data;
1186*4882a593Smuzhiyun 	int ret;
1187*4882a593Smuzhiyun 
1188*4882a593Smuzhiyun 	ret = cyapa_read_block(cyapa, CYAPA_CMD_GROUP_DATA, (u8 *)&data);
1189*4882a593Smuzhiyun 	if (ret != sizeof(data)) {
1190*4882a593Smuzhiyun 		dev_err(dev, "failed to read report data, (%d)\n", ret);
1191*4882a593Smuzhiyun 		return -EINVAL;
1192*4882a593Smuzhiyun 	}
1193*4882a593Smuzhiyun 
1194*4882a593Smuzhiyun 	if ((data.device_status & OP_STATUS_SRC) != OP_STATUS_SRC ||
1195*4882a593Smuzhiyun 	    (data.device_status & OP_STATUS_DEV) != CYAPA_DEV_NORMAL ||
1196*4882a593Smuzhiyun 	    (data.finger_btn & OP_DATA_VALID) != OP_DATA_VALID) {
1197*4882a593Smuzhiyun 		dev_err(dev, "invalid device state bytes: %02x %02x\n",
1198*4882a593Smuzhiyun 			data.device_status, data.finger_btn);
1199*4882a593Smuzhiyun 		return -EINVAL;
1200*4882a593Smuzhiyun 	}
1201*4882a593Smuzhiyun 
1202*4882a593Smuzhiyun 	return cyapa_gen3_event_process(cyapa, &data);
1203*4882a593Smuzhiyun }
1204*4882a593Smuzhiyun 
1205*4882a593Smuzhiyun /*
1206*4882a593Smuzhiyun  * This function will be called in the cyapa_gen3_set_power_mode function,
1207*4882a593Smuzhiyun  * and it's known that it may failed in some situation after the set power
1208*4882a593Smuzhiyun  * mode command was sent. So this function is aimed to avoid the knwon
1209*4882a593Smuzhiyun  * and unwanted output I2C and data parse error messages.
1210*4882a593Smuzhiyun  */
cyapa_gen3_try_poll_handler(struct cyapa * cyapa)1211*4882a593Smuzhiyun static int cyapa_gen3_try_poll_handler(struct cyapa *cyapa)
1212*4882a593Smuzhiyun {
1213*4882a593Smuzhiyun 	struct cyapa_reg_data data;
1214*4882a593Smuzhiyun 	int ret;
1215*4882a593Smuzhiyun 
1216*4882a593Smuzhiyun 	ret = cyapa_read_block(cyapa, CYAPA_CMD_GROUP_DATA, (u8 *)&data);
1217*4882a593Smuzhiyun 	if (ret != sizeof(data))
1218*4882a593Smuzhiyun 		return -EINVAL;
1219*4882a593Smuzhiyun 
1220*4882a593Smuzhiyun 	if ((data.device_status & OP_STATUS_SRC) != OP_STATUS_SRC ||
1221*4882a593Smuzhiyun 	    (data.device_status & OP_STATUS_DEV) != CYAPA_DEV_NORMAL ||
1222*4882a593Smuzhiyun 	    (data.finger_btn & OP_DATA_VALID) != OP_DATA_VALID)
1223*4882a593Smuzhiyun 		return -EINVAL;
1224*4882a593Smuzhiyun 
1225*4882a593Smuzhiyun 	return cyapa_gen3_event_process(cyapa, &data);
1226*4882a593Smuzhiyun 
1227*4882a593Smuzhiyun }
1228*4882a593Smuzhiyun 
cyapa_gen3_initialize(struct cyapa * cyapa)1229*4882a593Smuzhiyun static int cyapa_gen3_initialize(struct cyapa *cyapa) { return 0; }
cyapa_gen3_bl_initiate(struct cyapa * cyapa,const struct firmware * fw)1230*4882a593Smuzhiyun static int cyapa_gen3_bl_initiate(struct cyapa *cyapa,
1231*4882a593Smuzhiyun 		const struct firmware *fw) { return 0; }
cyapa_gen3_empty_output_data(struct cyapa * cyapa,u8 * buf,int * len,cb_sort func)1232*4882a593Smuzhiyun static int cyapa_gen3_empty_output_data(struct cyapa *cyapa,
1233*4882a593Smuzhiyun 		u8 *buf, int *len, cb_sort func) { return 0; }
1234*4882a593Smuzhiyun 
1235*4882a593Smuzhiyun const struct cyapa_dev_ops cyapa_gen3_ops = {
1236*4882a593Smuzhiyun 	.check_fw = cyapa_gen3_check_fw,
1237*4882a593Smuzhiyun 	.bl_enter = cyapa_gen3_bl_enter,
1238*4882a593Smuzhiyun 	.bl_activate = cyapa_gen3_bl_activate,
1239*4882a593Smuzhiyun 	.update_fw = cyapa_gen3_do_fw_update,
1240*4882a593Smuzhiyun 	.bl_deactivate = cyapa_gen3_bl_deactivate,
1241*4882a593Smuzhiyun 	.bl_initiate = cyapa_gen3_bl_initiate,
1242*4882a593Smuzhiyun 
1243*4882a593Smuzhiyun 	.show_baseline = cyapa_gen3_show_baseline,
1244*4882a593Smuzhiyun 	.calibrate_store = cyapa_gen3_do_calibrate,
1245*4882a593Smuzhiyun 
1246*4882a593Smuzhiyun 	.initialize = cyapa_gen3_initialize,
1247*4882a593Smuzhiyun 
1248*4882a593Smuzhiyun 	.state_parse = cyapa_gen3_state_parse,
1249*4882a593Smuzhiyun 	.operational_check = cyapa_gen3_do_operational_check,
1250*4882a593Smuzhiyun 
1251*4882a593Smuzhiyun 	.irq_handler = cyapa_gen3_irq_handler,
1252*4882a593Smuzhiyun 	.irq_cmd_handler = cyapa_gen3_irq_cmd_handler,
1253*4882a593Smuzhiyun 	.sort_empty_output_data = cyapa_gen3_empty_output_data,
1254*4882a593Smuzhiyun 	.set_power_mode = cyapa_gen3_set_power_mode,
1255*4882a593Smuzhiyun 
1256*4882a593Smuzhiyun 	.set_proximity = cyapa_gen3_set_proximity,
1257*4882a593Smuzhiyun };
1258