Input: synaptics-rmi4 - add support for F03

Commit Message

margeyang Aug. 1, 2022, 6:47 a.m. UTC

From: Marge Yang <[e-mail]>

RMI4 F03 supports the Stick function,
it's designed to support relative packet.
This patch adds to support complete relative packet.
When timeout error or parity error happens,
sends 0xFE command to ask Re-send stick packet again.
Supports some conditions to filter wrong relative packet.

Signed-off-by: Marge Yang<[e-mail]>
---
 drivers/input/rmi4/rmi_f03.c | 82 ++++++++++++++++++++++++++++++++++--
 1 file changed, 78 insertions(+), 4 deletions(-)

Patch

diff --git a/drivers/input/rmi4/rmi_f03.c b/drivers/input/rmi4/rmi_f03.c
index c194b1664b10..57f03dfcb4ff 100644
--- a/drivers/input/rmi4/rmi_f03.c
+++ b/drivers/input/rmi4/rmi_f03.c
@@ -23,8 +23,12 @@ 
 #define RMI_F03_BYTES_PER_DEVICE_SHIFT	4
 #define RMI_F03_QUEUE_LENGTH		0x0F
 
+#define RMI_F03_RESET_STYK 0xFE
+
 #define PSMOUSE_OOB_EXTRA_BTNS		0x01
 
+#define RELATIVE_PACKET_SIZE		0x03
+
 struct f03_data {
 	struct rmi_function *fn;
 
@@ -36,7 +40,8 @@  struct f03_data {
 	u8 device_count;
 	u8 rx_queue_length;
 };
-
+int iwritecommandcounter;
+unsigned int ipacketindex;
 int rmi_f03_overwrite_button(struct rmi_function *fn, unsigned int button,
 			     int value)
 {
@@ -87,7 +92,7 @@  static int rmi_f03_pt_write(struct serio *id, unsigned char val)
 			__func__, error);
 		return error;
 	}
-
+	iwritecommandcounter++;
 	return 0;
 }
 
@@ -197,10 +202,12 @@  static int rmi_f03_register_pt(struct f03_data *f03)
 
 static int rmi_f03_probe(struct rmi_function *fn)
 {
+
 	struct device *dev = &fn->dev;
 	struct f03_data *f03;
 	int error;
-
+	iwritecommandcounter = 0;
+	ipacketindex = 0;
 	f03 = devm_kzalloc(dev, sizeof(struct f03_data), GFP_KERNEL);
 	if (!f03)
 		return -ENOMEM;
@@ -251,9 +258,12 @@  static irqreturn_t rmi_f03_attention(int irq, void *ctx)
 	const u8 ob_len = f03->rx_queue_length * RMI_F03_OB_SIZE;
 	u8 obs[RMI_F03_QUEUE_LENGTH * RMI_F03_OB_SIZE];
 	u8 ob_status;
+	static u8 ob_dataArry[RELATIVE_PACKET_SIZE];
 	u8 ob_data;
 	unsigned int serio_flags;
+	static unsigned int serio_flagsArry[RELATIVE_PACKET_SIZE];
 	int i;
+
 	int error;
 
 	if (drvdata->attn_data.data) {
@@ -284,6 +294,22 @@  static irqreturn_t rmi_f03_attention(int irq, void *ctx)
 		ob_data = obs[i + RMI_F03_OB_DATA_OFFSET];
 		serio_flags = 0;
 
+		if (ob_status & (RMI_F03_OB_FLAG_TIMEOUT | RMI_F03_OB_FLAG_PARITY)) {
+			//  Send resend command to stick when timeout or parity error.
+			//  Driver can receive the last stick packet.
+
+			error = rmi_write(f03->fn->rmi_dev, f03->fn->fd.data_base_addr,
+			 RMI_F03_RESET_STYK);
+			if (error) {
+				dev_err(&f03->fn->dev,
+					"%s: Failed to rmi_write to F03 TX register (%d).\n",
+					__func__, error);
+				return error;
+			}
+			ipacketindex = 0;
+			break;
+		}
+
 		if (!(ob_status & RMI_F03_RX_DATA_OFB))
 			continue;
 
@@ -298,9 +324,57 @@  static irqreturn_t rmi_f03_attention(int irq, void *ctx)
 			serio_flags & SERIO_TIMEOUT ?  'Y' : 'N',
 			serio_flags & SERIO_PARITY ? 'Y' : 'N');
 
-		serio_interrupt(f03->serio, ob_data, serio_flags);
+		if (iwritecommandcounter > 0) {
+			// Read Acknowledge Byte after writing the PS2 command.
+			// It is not trackpoint data.
+			serio_interrupt(f03->serio, ob_data, serio_flags);
+
+		} else {
+			//   The relative-mode PS/2 packet format is as follows:
+			//
+			//              bit position            position (as array of bytes)
+			//     7   6   5   4   3   2   1   0
+			//   =================================+
+			//    Yov Xov DY8 DX8  1   M   R   L  | DATA[0]
+			//                DX[7:0]             | DATA[1]
+			//                DY[7:0]             | DATA[2]
+			//   =================================+
+			//		Yov: Y overflow
+			//    Xov: X overflow
+			if ((ipacketindex == 0) && (ob_data & ((BIT(7)|BIT(6))))) {
+				dev_err(&f03->fn->dev,
+				"%s: X or Y is overflow. (%x)\n",
+				__func__, ob_data);
+				break;
+			} else if ((ipacketindex == 0) && !(ob_data & BIT(3))) {
+				dev_err(&f03->fn->dev,
+				"%s: New BIT 3 is not 1 for the first byte\n",
+				__func__);
+			} else {
+				if (ipacketindex >= RELATIVE_PACKET_SIZE) {
+					ipacketindex = 0;
+				} else {
+					ob_dataArry[ipacketindex] = ob_data;
+					serio_flagsArry[ipacketindex] = serio_flags;
+					ipacketindex++;
+				}
+				if (ipacketindex == RELATIVE_PACKET_SIZE)	{
+					serio_interrupt(f03->serio, ob_dataArry[0],
+					 serio_flagsArry[0]);
+					serio_interrupt(f03->serio, ob_dataArry[1],
+					 serio_flagsArry[1]);
+					serio_interrupt(f03->serio, ob_dataArry[2],
+					 serio_flagsArry[2]);
+					ipacketindex = 0;
+				}
+			}
+		}
 	}
 
+	if (iwritecommandcounter > 0) {
+		ipacketindex = 0;
+		iwritecommandcounter = iwritecommandcounter - 1;
+	}
 	return IRQ_HANDLED;
 }