1 // SPDX-License-Identifier: GPL-2.0-only
2 //
3 // Copyright (C) 2021-2022 Samuel Holland <samuel@sholland.org>
4 
5 #include <linux/crc8.h>
6 #include <linux/delay.h>
7 #include <linux/err.h>
8 #include <linux/i2c.h>
9 #include <linux/input.h>
10 #include <linux/input/matrix_keypad.h>
11 #include <linux/interrupt.h>
12 #include <linux/module.h>
13 #include <linux/mod_devicetable.h>
14 #include <linux/of.h>
15 #include <linux/regulator/consumer.h>
16 #include <linux/types.h>
17 
18 #define DRV_NAME			"pinephone-keyboard"
19 
20 #define PPKB_CRC8_POLYNOMIAL		0x07
21 
22 #define PPKB_DEVICE_ID_HI		0x00
23 #define PPKB_DEVICE_ID_HI_VALUE			'K'
24 #define PPKB_DEVICE_ID_LO		0x01
25 #define PPKB_DEVICE_ID_LO_VALUE			'B'
26 #define PPKB_FW_REVISION		0x02
27 #define PPKB_FW_FEATURES		0x03
28 #define PPKB_MATRIX_SIZE		0x06
29 #define PPKB_SCAN_CRC			0x07
30 #define PPKB_SCAN_DATA			0x08
31 #define PPKB_SYS_CONFIG			0x20
32 #define PPKB_SYS_CONFIG_DISABLE_SCAN		BIT(0)
33 #define PPKB_SYS_SMBUS_COMMAND		0x21
34 #define PPKB_SYS_SMBUS_DATA		0x22
35 #define PPKB_SYS_COMMAND		0x23
36 #define PPKB_SYS_COMMAND_SMBUS_READ		0x91
37 #define PPKB_SYS_COMMAND_SMBUS_WRITE		0xa1
38 
39 #define PPKB_ROWS			6
40 #define PPKB_COLS			12
41 
42 /* Size of the scan buffer, including the CRC byte at the beginning. */
43 #define PPKB_BUF_LEN			(1 + PPKB_COLS)
44 
45 static const uint32_t ppkb_keymap[] = {
46 	KEY(0,  0, KEY_ESC),
47 	KEY(0,  1, KEY_1),
48 	KEY(0,  2, KEY_2),
49 	KEY(0,  3, KEY_3),
50 	KEY(0,  4, KEY_4),
51 	KEY(0,  5, KEY_5),
52 	KEY(0,  6, KEY_6),
53 	KEY(0,  7, KEY_7),
54 	KEY(0,  8, KEY_8),
55 	KEY(0,  9, KEY_9),
56 	KEY(0, 10, KEY_0),
57 	KEY(0, 11, KEY_BACKSPACE),
58 
59 	KEY(1,  0, KEY_TAB),
60 	KEY(1,  1, KEY_Q),
61 	KEY(1,  2, KEY_W),
62 	KEY(1,  3, KEY_E),
63 	KEY(1,  4, KEY_R),
64 	KEY(1,  5, KEY_T),
65 	KEY(1,  6, KEY_Y),
66 	KEY(1,  7, KEY_U),
67 	KEY(1,  8, KEY_I),
68 	KEY(1,  9, KEY_O),
69 	KEY(1, 10, KEY_P),
70 	KEY(1, 11, KEY_ENTER),
71 
72 	KEY(2,  0, KEY_LEFTMETA),
73 	KEY(2,  1, KEY_A),
74 	KEY(2,  2, KEY_S),
75 	KEY(2,  3, KEY_D),
76 	KEY(2,  4, KEY_F),
77 	KEY(2,  5, KEY_G),
78 	KEY(2,  6, KEY_H),
79 	KEY(2,  7, KEY_J),
80 	KEY(2,  8, KEY_K),
81 	KEY(2,  9, KEY_L),
82 	KEY(2, 10, KEY_SEMICOLON),
83 
84 	KEY(3,  0, KEY_LEFTSHIFT),
85 	KEY(3,  1, KEY_Z),
86 	KEY(3,  2, KEY_X),
87 	KEY(3,  3, KEY_C),
88 	KEY(3,  4, KEY_V),
89 	KEY(3,  5, KEY_B),
90 	KEY(3,  6, KEY_N),
91 	KEY(3,  7, KEY_M),
92 	KEY(3,  8, KEY_COMMA),
93 	KEY(3,  9, KEY_DOT),
94 	KEY(3, 10, KEY_SLASH),
95 
96 	KEY(4,  1, KEY_LEFTCTRL),
97 	KEY(4,  4, KEY_SPACE),
98 	KEY(4,  6, KEY_APOSTROPHE),
99 	KEY(4,  8, KEY_RIGHTBRACE),
100 	KEY(4,  9, KEY_LEFTBRACE),
101 
102 	KEY(5,  2, KEY_FN),
103 	KEY(5,  3, KEY_LEFTALT),
104 	KEY(5,  5, KEY_RIGHTALT),
105 
106 	/* FN layer */
107 	KEY(PPKB_ROWS + 0,  0, KEY_FN_ESC),
108 	KEY(PPKB_ROWS + 0,  1, KEY_F1),
109 	KEY(PPKB_ROWS + 0,  2, KEY_F2),
110 	KEY(PPKB_ROWS + 0,  3, KEY_F3),
111 	KEY(PPKB_ROWS + 0,  4, KEY_F4),
112 	KEY(PPKB_ROWS + 0,  5, KEY_F5),
113 	KEY(PPKB_ROWS + 0,  6, KEY_F6),
114 	KEY(PPKB_ROWS + 0,  7, KEY_F7),
115 	KEY(PPKB_ROWS + 0,  8, KEY_F8),
116 	KEY(PPKB_ROWS + 0,  9, KEY_F9),
117 	KEY(PPKB_ROWS + 0, 10, KEY_F10),
118 	KEY(PPKB_ROWS + 0, 11, KEY_DELETE),
119 
120 	KEY(PPKB_ROWS + 1, 10, KEY_PAGEUP),
121 
122 	KEY(PPKB_ROWS + 2,  0, KEY_SYSRQ),
123 	KEY(PPKB_ROWS + 2,  9, KEY_PAGEDOWN),
124 	KEY(PPKB_ROWS + 2, 10, KEY_INSERT),
125 
126 	KEY(PPKB_ROWS + 3,  0, KEY_LEFTSHIFT),
127 	KEY(PPKB_ROWS + 3,  8, KEY_HOME),
128 	KEY(PPKB_ROWS + 3,  9, KEY_UP),
129 	KEY(PPKB_ROWS + 3, 10, KEY_END),
130 
131 	KEY(PPKB_ROWS + 4, 1, KEY_LEFTCTRL),
132 	KEY(PPKB_ROWS + 4, 6, KEY_LEFT),
133 	KEY(PPKB_ROWS + 4, 8, KEY_RIGHT),
134 	KEY(PPKB_ROWS + 4, 9, KEY_DOWN),
135 
136 	KEY(PPKB_ROWS + 5, 3, KEY_LEFTALT),
137 	KEY(PPKB_ROWS + 5, 5, KEY_RIGHTALT),
138 };
139 
140 static const struct matrix_keymap_data ppkb_keymap_data = {
141 	.keymap		= ppkb_keymap,
142 	.keymap_size	= ARRAY_SIZE(ppkb_keymap),
143 };
144 
145 struct pinephone_keyboard {
146 	struct i2c_adapter adapter;
147 	struct input_dev *input;
148 	u8 buf[2][PPKB_BUF_LEN];
149 	u8 crc_table[CRC8_TABLE_SIZE];
150 	u8 fn_state[PPKB_COLS];
151 	bool buf_swap;
152 	bool fn_pressed;
153 };
154 
155 static int ppkb_adap_smbus_xfer(struct i2c_adapter *adap, u16 addr,
156 				unsigned short flags, char read_write,
157 				u8 command, int size,
158 				union i2c_smbus_data *data)
159 {
160 	struct i2c_client *client = adap->algo_data;
161 	u8 buf[3];
162 	int ret;
163 
164 	buf[0] = command;
165 	buf[1] = data->byte;
166 	buf[2] = read_write == I2C_SMBUS_READ ? PPKB_SYS_COMMAND_SMBUS_READ
167 					      : PPKB_SYS_COMMAND_SMBUS_WRITE;
168 
169 	ret = i2c_smbus_write_i2c_block_data(client, PPKB_SYS_SMBUS_COMMAND,
170 					     sizeof(buf), buf);
171 	if (ret)
172 		return ret;
173 
174 	/* Read back the command status until it passes or fails. */
175 	do {
176 		usleep_range(300, 500);
177 		ret = i2c_smbus_read_byte_data(client, PPKB_SYS_COMMAND);
178 	} while (ret == buf[2]);
179 	if (ret < 0)
180 		return ret;
181 	/* Commands return 0x00 on success and 0xff on failure. */
182 	if (ret)
183 		return -EIO;
184 
185 	if (read_write == I2C_SMBUS_READ) {
186 		ret = i2c_smbus_read_byte_data(client, PPKB_SYS_SMBUS_DATA);
187 		if (ret < 0)
188 			return ret;
189 
190 		data->byte = ret;
191 	}
192 
193 	return 0;
194 }
195 
196 static u32 ppkg_adap_functionality(struct i2c_adapter *adap)
197 {
198 	return I2C_FUNC_SMBUS_BYTE_DATA;
199 }
200 
201 static const struct i2c_algorithm ppkb_adap_algo = {
202 	.smbus_xfer		= ppkb_adap_smbus_xfer,
203 	.functionality		= ppkg_adap_functionality,
204 };
205 
206 static void ppkb_update(struct i2c_client *client)
207 {
208 	struct pinephone_keyboard *ppkb = i2c_get_clientdata(client);
209 	unsigned short *keymap = ppkb->input->keycode;
210 	int row_shift = get_count_order(PPKB_COLS);
211 	u8 *old_buf = ppkb->buf[!ppkb->buf_swap];
212 	u8 *new_buf = ppkb->buf[ppkb->buf_swap];
213 	int col, crc, ret, row;
214 	struct device *dev = &client->dev;
215 
216 	ret = i2c_smbus_read_i2c_block_data(client, PPKB_SCAN_CRC,
217 					    PPKB_BUF_LEN, new_buf);
218 	if (ret != PPKB_BUF_LEN) {
219 		dev_err(dev, "Failed to read scan data: %d\n", ret);
220 		return;
221 	}
222 
223 	crc = crc8(ppkb->crc_table, &new_buf[1], PPKB_COLS, CRC8_INIT_VALUE);
224 	if (crc != new_buf[0]) {
225 		dev_err(dev, "Bad scan data (%02x != %02x)\n", crc, new_buf[0]);
226 		return;
227 	}
228 
229 	ppkb->buf_swap = !ppkb->buf_swap;
230 
231 	for (col = 0; col < PPKB_COLS; ++col) {
232 		u8 old = old_buf[1 + col];
233 		u8 new = new_buf[1 + col];
234 		u8 changed = old ^ new;
235 
236 		if (!changed)
237 			continue;
238 
239 		for (row = 0; row < PPKB_ROWS; ++row) {
240 			u8 mask = BIT(row);
241 			u8 value = new & mask;
242 			unsigned short code;
243 			bool fn_state;
244 
245 			if (!(changed & mask))
246 				continue;
247 
248 			/*
249 			 * Save off the FN key state when the key was pressed,
250 			 * and use that to determine the code during a release.
251 			 */
252 			fn_state = value ? ppkb->fn_pressed : ppkb->fn_state[col] & mask;
253 			if (fn_state)
254 				ppkb->fn_state[col] ^= mask;
255 
256 			/* The FN layer is a second set of rows. */
257 			code = MATRIX_SCAN_CODE(fn_state ? PPKB_ROWS + row : row,
258 						col, row_shift);
259 			input_event(ppkb->input, EV_MSC, MSC_SCAN, code);
260 			input_report_key(ppkb->input, keymap[code], value);
261 			if (keymap[code] == KEY_FN)
262 				ppkb->fn_pressed = value;
263 		}
264 	}
265 	input_sync(ppkb->input);
266 }
267 
268 static irqreturn_t ppkb_irq_thread(int irq, void *data)
269 {
270 	struct i2c_client *client = data;
271 
272 	ppkb_update(client);
273 
274 	return IRQ_HANDLED;
275 }
276 
277 static int ppkb_set_scan(struct i2c_client *client, bool enable)
278 {
279 	struct device *dev = &client->dev;
280 	int ret, val;
281 
282 	ret = i2c_smbus_read_byte_data(client, PPKB_SYS_CONFIG);
283 	if (ret < 0) {
284 		dev_err(dev, "Failed to read config: %d\n", ret);
285 		return ret;
286 	}
287 
288 	if (enable)
289 		val = ret & ~PPKB_SYS_CONFIG_DISABLE_SCAN;
290 	else
291 		val = ret | PPKB_SYS_CONFIG_DISABLE_SCAN;
292 
293 	ret = i2c_smbus_write_byte_data(client, PPKB_SYS_CONFIG, val);
294 	if (ret) {
295 		dev_err(dev, "Failed to write config: %d\n", ret);
296 		return ret;
297 	}
298 
299 	return 0;
300 }
301 
302 static int ppkb_open(struct input_dev *input)
303 {
304 	struct i2c_client *client = input_get_drvdata(input);
305 	int error;
306 
307 	error = ppkb_set_scan(client, true);
308 	if (error)
309 		return error;
310 
311 	return 0;
312 }
313 
314 static void ppkb_close(struct input_dev *input)
315 {
316 	struct i2c_client *client = input_get_drvdata(input);
317 
318 	ppkb_set_scan(client, false);
319 }
320 
321 static void ppkb_regulator_disable(void *regulator)
322 {
323 	regulator_disable(regulator);
324 }
325 
326 static int ppkb_probe(struct i2c_client *client)
327 {
328 	struct device *dev = &client->dev;
329 	unsigned int phys_rows, phys_cols;
330 	struct pinephone_keyboard *ppkb;
331 	struct regulator *vbat_supply;
332 	u8 info[PPKB_MATRIX_SIZE + 1];
333 	struct device_node *i2c_bus;
334 	int ret;
335 	int error;
336 
337 	vbat_supply = devm_regulator_get(dev, "vbat");
338 	error = PTR_ERR_OR_ZERO(vbat_supply);
339 	if (error) {
340 		dev_err(dev, "Failed to get VBAT supply: %d\n", error);
341 		return error;
342 	}
343 
344 	error = regulator_enable(vbat_supply);
345 	if (error) {
346 		dev_err(dev, "Failed to enable VBAT: %d\n", error);
347 		return error;
348 	}
349 
350 	error = devm_add_action_or_reset(dev, ppkb_regulator_disable,
351 					 vbat_supply);
352 	if (error)
353 		return error;
354 
355 	ret = i2c_smbus_read_i2c_block_data(client, 0, sizeof(info), info);
356 	if (ret != sizeof(info)) {
357 		error = ret < 0 ? ret : -EIO;
358 		dev_err(dev, "Failed to read device ID: %d\n", error);
359 		return error;
360 	}
361 
362 	if (info[PPKB_DEVICE_ID_HI] != PPKB_DEVICE_ID_HI_VALUE ||
363 	    info[PPKB_DEVICE_ID_LO] != PPKB_DEVICE_ID_LO_VALUE) {
364 		dev_warn(dev, "Unexpected device ID: %#02x %#02x\n",
365 			 info[PPKB_DEVICE_ID_HI], info[PPKB_DEVICE_ID_LO]);
366 		return -ENODEV;
367 	}
368 
369 	dev_info(dev, "Found firmware version %d.%d features %#x\n",
370 		 info[PPKB_FW_REVISION] >> 4,
371 		 info[PPKB_FW_REVISION] & 0xf,
372 		 info[PPKB_FW_FEATURES]);
373 
374 	phys_rows = info[PPKB_MATRIX_SIZE] & 0xf;
375 	phys_cols = info[PPKB_MATRIX_SIZE] >> 4;
376 	if (phys_rows != PPKB_ROWS || phys_cols != PPKB_COLS) {
377 		dev_err(dev, "Unexpected keyboard size %ux%u\n",
378 			phys_rows, phys_cols);
379 		return -EINVAL;
380 	}
381 
382 	/* Disable scan by default to save power. */
383 	error = ppkb_set_scan(client, false);
384 	if (error)
385 		return error;
386 
387 	ppkb = devm_kzalloc(dev, sizeof(*ppkb), GFP_KERNEL);
388 	if (!ppkb)
389 		return -ENOMEM;
390 
391 	i2c_set_clientdata(client, ppkb);
392 
393 	i2c_bus = of_get_child_by_name(dev->of_node, "i2c");
394 	if (i2c_bus) {
395 		ppkb->adapter.owner = THIS_MODULE;
396 		ppkb->adapter.algo = &ppkb_adap_algo;
397 		ppkb->adapter.algo_data = client;
398 		ppkb->adapter.dev.parent = dev;
399 		ppkb->adapter.dev.of_node = i2c_bus;
400 		strscpy(ppkb->adapter.name, DRV_NAME, sizeof(ppkb->adapter.name));
401 
402 		error = devm_i2c_add_adapter(dev, &ppkb->adapter);
403 		if (error) {
404 			dev_err(dev, "Failed to add I2C adapter: %d\n", error);
405 			return error;
406 		}
407 	}
408 
409 	crc8_populate_msb(ppkb->crc_table, PPKB_CRC8_POLYNOMIAL);
410 
411 	ppkb->input = devm_input_allocate_device(dev);
412 	if (!ppkb->input)
413 		return -ENOMEM;
414 
415 	input_set_drvdata(ppkb->input, client);
416 
417 	ppkb->input->name = "PinePhone Keyboard";
418 	ppkb->input->phys = DRV_NAME "/input0";
419 	ppkb->input->id.bustype = BUS_I2C;
420 	ppkb->input->open = ppkb_open;
421 	ppkb->input->close = ppkb_close;
422 
423 	input_set_capability(ppkb->input, EV_MSC, MSC_SCAN);
424 	__set_bit(EV_REP, ppkb->input->evbit);
425 
426 	error = matrix_keypad_build_keymap(&ppkb_keymap_data, NULL,
427 					   2 * PPKB_ROWS, PPKB_COLS, NULL,
428 					   ppkb->input);
429 	if (error) {
430 		dev_err(dev, "Failed to build keymap: %d\n", error);
431 		return error;
432 	}
433 
434 	error = input_register_device(ppkb->input);
435 	if (error) {
436 		dev_err(dev, "Failed to register input: %d\n", error);
437 		return error;
438 	}
439 
440 	error = devm_request_threaded_irq(dev, client->irq,
441 					  NULL, ppkb_irq_thread,
442 					  IRQF_ONESHOT, client->name, client);
443 	if (error) {
444 		dev_err(dev, "Failed to request IRQ: %d\n", error);
445 		return error;
446 	}
447 
448 	return 0;
449 }
450 
451 static const struct of_device_id ppkb_of_match[] = {
452 	{ .compatible = "pine64,pinephone-keyboard" },
453 	{ }
454 };
455 MODULE_DEVICE_TABLE(of, ppkb_of_match);
456 
457 static struct i2c_driver ppkb_driver = {
458 	.probe_new	= ppkb_probe,
459 	.driver		= {
460 		.name		= DRV_NAME,
461 		.of_match_table = ppkb_of_match,
462 	},
463 };
464 module_i2c_driver(ppkb_driver);
465 
466 MODULE_AUTHOR("Samuel Holland <samuel@sholland.org>");
467 MODULE_DESCRIPTION("Pine64 PinePhone keyboard driver");
468 MODULE_LICENSE("GPL");
469