xref: /openbmc/linux/drivers/usb/typec/rt1719.c (revision 51ded7cd)
1 // SPDX-License-Identifier: GPL-2.0
2 
3 #include <linux/bitfield.h>
4 #include <linux/completion.h>
5 #include <linux/i2c.h>
6 #include <linux/interrupt.h>
7 #include <linux/kernel.h>
8 #include <linux/module.h>
9 #include <linux/power_supply.h>
10 #include <linux/regmap.h>
11 #include <linux/usb/pd.h>
12 #include <linux/usb/role.h>
13 #include <linux/usb/typec.h>
14 
15 #define RT1719_REG_TXCTRL1	0x03
16 #define RT1719_REG_TXCTRL2	0x04
17 #define RT1719_REG_POLICYINFO	0x0E
18 #define RT1719_REG_SRCPDO1	0x11
19 #define RT1719_REG_MASKS	0x2D
20 #define RT1719_REG_EVENTS	0x33
21 #define RT1719_REG_STATS	0x37
22 #define RT1719_REG_PSELINFO	0x3C
23 #define RT1719_REG_USBSETINFO	0x3E
24 #define RT1719_REG_VENID	0x82
25 
26 #define RT1719_UNIQUE_PID	0x1719
27 #define RT1719_REQDRSWAP_MASK	BIT(7)
28 #define RT1719_EVALMODE_MASK	BIT(4)
29 #define RT1719_REQSRCPDO_MASK	GENMASK(2, 0)
30 #define RT1719_TXSPDOREQ_MASK	BIT(7)
31 #define RT1719_INT_DRSW_ACCEPT	BIT(23)
32 #define RT1719_INT_RX_SRCCAP	BIT(21)
33 #define RT1719_INT_VBUS_DCT	BIT(6)
34 #define RT1719_INT_VBUS_PRESENT	BIT(5)
35 #define RT1719_INT_PE_SNK_RDY	BIT(2)
36 #define RT1719_CC1_STAT		GENMASK(9, 8)
37 #define RT1719_CC2_STAT		GENMASK(11, 10)
38 #define RT1719_POLARITY_MASK	BIT(23)
39 #define RT1719_DATAROLE_MASK	BIT(22)
40 #define RT1719_PDSPECREV_MASK	GENMASK(21, 20)
41 #define RT1719_SPDOSEL_MASK	GENMASK(18, 16)
42 #define RT1719_SPDONUM_MASK	GENMASK(15, 13)
43 #define RT1719_ATTACH_VBUS	BIT(12)
44 #define RT1719_ATTACH_DBG	BIT(10)
45 #define RT1719_ATTACH_SNK	BIT(9)
46 #define RT1719_ATTACHDEV_MASK	(RT1719_ATTACH_VBUS | RT1719_ATTACH_DBG | \
47 				 RT1719_ATTACH_SNK)
48 #define RT1719_PE_EXP_CONTRACT	BIT(2)
49 #define RT1719_PSEL_SUPPORT	BIT(15)
50 #define RT1719_TBLSEL_MASK	BIT(6)
51 #define RT1719_LATPSEL_MASK	GENMASK(5, 0)
52 #define RT1719_USBINFO_MASK	GENMASK(1, 0)
53 #define RT1719_USB_DFPUFP	3
54 #define RT1719_MAX_SRCPDO	7
55 
56 enum {
57 	SNK_PWR_OPEN = 0,
58 	SNK_PWR_DEF,
59 	SNK_PWR_1P5A,
60 	SNK_PWR_3A
61 };
62 
63 enum {
64 	USBPD_SPECREV_1_0 = 0,
65 	USBPD_SPECREV_2_0,
66 	USBPD_SPECREV_3_0
67 };
68 
69 enum rt1719_snkcap {
70 	RT1719_SNKCAP_5V = 0,
71 	RT1719_SNKCAP_9V,
72 	RT1719_SNKCAP_12V,
73 	RT1719_SNKCAP_15V,
74 	RT1719_SNKCAP_20V,
75 	RT1719_MAX_SNKCAP
76 };
77 
78 struct rt1719_psel_cap {
79 	u8 lomask;
80 	u8 himask;
81 	u32 milliwatt;
82 	u32 milliamp;
83 };
84 
85 struct rt1719_data {
86 	struct device *dev;
87 	struct regmap *regmap;
88 	struct typec_port *port;
89 	struct usb_role_switch *role_sw;
90 	struct power_supply *psy;
91 	struct typec_partner *partner;
92 	struct power_supply_desc psy_desc;
93 	struct usb_pd_identity partner_ident;
94 	struct typec_partner_desc partner_desc;
95 	struct completion req_completion;
96 	enum power_supply_usb_type usb_type;
97 	bool attached;
98 	bool pd_capable;
99 	bool drswap_support;
100 	u32 voltage;
101 	u32 req_voltage;
102 	u32 max_current;
103 	u32 op_current;
104 	u32 spdos[RT1719_MAX_SRCPDO];
105 	u16 snkcaps[RT1719_MAX_SNKCAP];
106 	int spdo_num;
107 	int spdo_sel;
108 	u32 conn_info;
109 	u16 conn_stat;
110 };
111 
112 static const enum power_supply_usb_type rt1719_psy_usb_types[] = {
113 	POWER_SUPPLY_USB_TYPE_C,
114 	POWER_SUPPLY_USB_TYPE_PD,
115 	POWER_SUPPLY_USB_TYPE_PD_PPS
116 };
117 
118 static const enum power_supply_property rt1719_psy_properties[] = {
119 	POWER_SUPPLY_PROP_ONLINE,
120 	POWER_SUPPLY_PROP_USB_TYPE,
121 	POWER_SUPPLY_PROP_VOLTAGE_NOW,
122 	POWER_SUPPLY_PROP_CURRENT_MAX,
123 	POWER_SUPPLY_PROP_CURRENT_NOW
124 };
125 
126 static int rt1719_read16(struct rt1719_data *data, unsigned int reg, u16 *val)
127 {
128 	__le16 regval;
129 	int ret;
130 
131 	ret = regmap_raw_read(data->regmap, reg, &regval, sizeof(regval));
132 	if (ret)
133 		return ret;
134 
135 	*val = le16_to_cpu(regval);
136 	return 0;
137 }
138 
139 static int rt1719_read32(struct rt1719_data *data, unsigned int reg, u32 *val)
140 {
141 	__le32 regval;
142 	int ret;
143 
144 	ret = regmap_raw_read(data->regmap, reg, &regval, sizeof(regval));
145 	if (ret)
146 		return ret;
147 
148 	*val = le32_to_cpu(regval);
149 	return 0;
150 }
151 
152 static int rt1719_write32(struct rt1719_data *data, unsigned int reg, u32 val)
153 {
154 	__le32 regval = cpu_to_le32(val);
155 
156 	return regmap_raw_write(data->regmap, reg, &regval, sizeof(regval));
157 }
158 
159 static enum typec_pwr_opmode rt1719_get_pwr_opmode(u32 conn, u16 stat)
160 {
161 	u16 cc1, cc2, cc_stat;
162 
163 	cc1 = FIELD_GET(RT1719_CC1_STAT, stat);
164 	cc2 = FIELD_GET(RT1719_CC2_STAT, stat);
165 
166 	if (conn & RT1719_ATTACH_SNK) {
167 		if (conn & RT1719_POLARITY_MASK)
168 			cc_stat = cc2;
169 		else
170 			cc_stat = cc1;
171 
172 		switch (cc_stat) {
173 		case SNK_PWR_3A:
174 			return TYPEC_PWR_MODE_3_0A;
175 		case SNK_PWR_1P5A:
176 			return TYPEC_PWR_MODE_1_5A;
177 		}
178 	} else if (conn & RT1719_ATTACH_DBG) {
179 		if ((cc1 == SNK_PWR_1P5A && cc2 == SNK_PWR_DEF) ||
180 		    (cc1 == SNK_PWR_DEF && cc2 == SNK_PWR_1P5A))
181 			return TYPEC_PWR_MODE_1_5A;
182 		else if ((cc1 == SNK_PWR_3A && cc2 == SNK_PWR_DEF) ||
183 			 (cc1 == SNK_PWR_DEF && cc2 == SNK_PWR_3A))
184 			return TYPEC_PWR_MODE_3_0A;
185 	}
186 
187 	return TYPEC_PWR_MODE_USB;
188 }
189 
190 static enum typec_data_role rt1719_get_data_role(u32 conn)
191 {
192 	if (conn & RT1719_DATAROLE_MASK)
193 		return TYPEC_HOST;
194 	return TYPEC_DEVICE;
195 }
196 
197 static void rt1719_set_data_role(struct rt1719_data *data,
198 				 enum typec_data_role data_role,
199 				 bool attached)
200 {
201 	enum usb_role usb_role = USB_ROLE_NONE;
202 
203 	if (attached) {
204 		if (data_role == TYPEC_HOST)
205 			usb_role = USB_ROLE_HOST;
206 		else
207 			usb_role = USB_ROLE_DEVICE;
208 	}
209 
210 	usb_role_switch_set_role(data->role_sw, usb_role);
211 	typec_set_data_role(data->port, data_role);
212 }
213 
214 static void rt1719_update_data_role(struct rt1719_data *data)
215 {
216 	if (!data->attached)
217 		return;
218 
219 	rt1719_set_data_role(data, rt1719_get_data_role(data->conn_info), true);
220 }
221 
222 static void rt1719_register_partner(struct rt1719_data *data)
223 {
224 	u16 spec_rev = 0;
225 
226 	if (data->pd_capable) {
227 		u32 rev;
228 
229 		rev = FIELD_GET(RT1719_PDSPECREV_MASK, data->conn_info);
230 		switch (rev) {
231 		case USBPD_SPECREV_3_0:
232 			spec_rev = 0x0300;
233 			break;
234 		case USBPD_SPECREV_2_0:
235 			spec_rev = 0x0200;
236 			break;
237 		default:
238 			spec_rev = 0x0100;
239 			break;
240 		}
241 	}
242 
243 	/* Just to prevent multiple times attach */
244 	if (data->partner)
245 		typec_unregister_partner(data->partner);
246 
247 	memset(&data->partner_ident, 0, sizeof(data->partner_ident));
248 	data->partner_desc.usb_pd = data->pd_capable;
249 	data->partner_desc.pd_revision = spec_rev;
250 
251 	if (data->conn_info & RT1719_ATTACH_DBG)
252 		data->partner_desc.accessory = TYPEC_ACCESSORY_DEBUG;
253 	else
254 		data->partner_desc.accessory = TYPEC_ACCESSORY_NONE;
255 
256 	data->partner = typec_register_partner(data->port, &data->partner_desc);
257 }
258 
259 static void rt1719_attach(struct rt1719_data *data)
260 {
261 	enum typec_pwr_opmode pwr_opmode;
262 	enum typec_data_role data_role;
263 	u32 volt = 5000, curr = 500;
264 
265 	if (!(data->conn_info & RT1719_ATTACHDEV_MASK))
266 		return;
267 
268 	pwr_opmode = rt1719_get_pwr_opmode(data->conn_info, data->conn_stat);
269 	data_role = rt1719_get_data_role(data->conn_info);
270 
271 	typec_set_pwr_opmode(data->port, pwr_opmode);
272 	rt1719_set_data_role(data, data_role, true);
273 
274 	if (data->conn_info & RT1719_ATTACH_SNK)
275 		rt1719_register_partner(data);
276 
277 	if (pwr_opmode == TYPEC_PWR_MODE_3_0A)
278 		curr = 3000;
279 	else if (pwr_opmode == TYPEC_PWR_MODE_1_5A)
280 		curr = 1500;
281 
282 	data->voltage = volt * 1000;
283 	data->max_current = data->op_current = curr * 1000;
284 	data->attached = true;
285 
286 	power_supply_changed(data->psy);
287 }
288 
289 static void rt1719_detach(struct rt1719_data *data)
290 {
291 	if (!data->attached || (data->conn_info & RT1719_ATTACHDEV_MASK))
292 		return;
293 
294 	typec_unregister_partner(data->partner);
295 	data->partner = NULL;
296 
297 	typec_set_pwr_opmode(data->port, TYPEC_PWR_MODE_USB);
298 	rt1719_set_data_role(data, TYPEC_DEVICE, false);
299 
300 	memset32(data->spdos, 0, RT1719_MAX_SRCPDO);
301 	data->spdo_num = 0;
302 	data->voltage = data->max_current = data->op_current = 0;
303 	data->attached = data->pd_capable = false;
304 
305 	data->usb_type = POWER_SUPPLY_USB_TYPE_C;
306 
307 	power_supply_changed(data->psy);
308 }
309 
310 static void rt1719_update_operating_status(struct rt1719_data *data)
311 {
312 	enum power_supply_usb_type usb_type = POWER_SUPPLY_USB_TYPE_PD;
313 	u32 voltage, max_current, op_current;
314 	int i, snk_sel;
315 
316 	for (i = 0; i < data->spdo_num; i++) {
317 		u32 pdo = data->spdos[i];
318 		enum pd_pdo_type type = pdo_type(pdo);
319 
320 		if (type == PDO_TYPE_APDO) {
321 			usb_type = POWER_SUPPLY_USB_TYPE_PD_PPS;
322 			break;
323 		}
324 	}
325 
326 	data->spdo_sel = FIELD_GET(RT1719_SPDOSEL_MASK, data->conn_info);
327 	if (data->spdo_sel <= 0)
328 		return;
329 
330 	data->usb_type = usb_type;
331 
332 	voltage = pdo_fixed_voltage(data->spdos[data->spdo_sel - 1]);
333 	max_current = pdo_max_current(data->spdos[data->spdo_sel - 1]);
334 
335 	switch (voltage) {
336 	case 5000:
337 		snk_sel = RT1719_SNKCAP_5V;
338 		break;
339 	case 9000:
340 		snk_sel = RT1719_SNKCAP_9V;
341 		break;
342 	case 12000:
343 		snk_sel = RT1719_SNKCAP_12V;
344 		break;
345 	case 15000:
346 		snk_sel = RT1719_SNKCAP_15V;
347 		break;
348 	case 20000:
349 		snk_sel = RT1719_SNKCAP_20V;
350 		break;
351 	default:
352 		return;
353 	}
354 
355 	op_current = min(max_current, pdo_max_current(data->snkcaps[snk_sel]));
356 
357 	/* covert mV/mA to uV/uA */
358 	data->voltage = voltage * 1000;
359 	data->max_current = max_current * 1000;
360 	data->op_current = op_current * 1000;
361 
362 	power_supply_changed(data->psy);
363 }
364 
365 static void rt1719_update_pwr_opmode(struct rt1719_data *data)
366 {
367 	if (!data->attached)
368 		return;
369 
370 	if (!data->pd_capable) {
371 		data->pd_capable = true;
372 
373 		typec_set_pwr_opmode(data->port, TYPEC_PWR_MODE_PD);
374 		rt1719_register_partner(data);
375 	}
376 
377 	rt1719_update_operating_status(data);
378 }
379 
380 static void rt1719_update_source_pdos(struct rt1719_data *data)
381 {
382 	int spdo_num = FIELD_GET(RT1719_SPDONUM_MASK, data->conn_info);
383 	__le32 src_pdos[RT1719_MAX_SRCPDO] = { };
384 	int i, ret;
385 
386 	if (!data->attached)
387 		return;
388 
389 	ret = regmap_raw_read(data->regmap, RT1719_REG_SRCPDO1, src_pdos,
390 			      sizeof(__le32) * spdo_num);
391 	if (ret)
392 		return;
393 
394 	data->spdo_num = spdo_num;
395 	for (i = 0; i < spdo_num; i++)
396 		data->spdos[i] = le32_to_cpu(src_pdos[i]);
397 }
398 
399 static int rt1719_dr_set(struct typec_port *port, enum typec_data_role role)
400 {
401 	struct rt1719_data *data = typec_get_drvdata(port);
402 	enum typec_data_role cur_role;
403 	int ret;
404 
405 	if (!data->attached || !data->pd_capable || !data->drswap_support)
406 		return -EOPNOTSUPP;
407 
408 	if (data->spdo_num > 0 && !(data->spdos[0] & PDO_FIXED_DATA_SWAP))
409 		return -EINVAL;
410 
411 	cur_role = rt1719_get_data_role(data->conn_info);
412 	if (cur_role == role)
413 		return 0;
414 
415 	ret = regmap_update_bits(data->regmap, RT1719_REG_TXCTRL1,
416 				 RT1719_REQDRSWAP_MASK, RT1719_REQDRSWAP_MASK);
417 	if (ret)
418 		return ret;
419 
420 	reinit_completion(&data->req_completion);
421 	ret = wait_for_completion_timeout(&data->req_completion,
422 					  msecs_to_jiffies(400));
423 	if (ret == 0)
424 		return -ETIMEDOUT;
425 
426 	cur_role = rt1719_get_data_role(data->conn_info);
427 	if (cur_role != role)
428 		return -EAGAIN;
429 
430 	rt1719_set_data_role(data, role, true);
431 	return 0;
432 }
433 
434 static const struct typec_operations rt1719_port_ops = {
435 	.dr_set = rt1719_dr_set,
436 };
437 
438 static int rt1719_usbpd_request_voltage(struct rt1719_data *data)
439 {
440 	u32 src_voltage;
441 	int snk_sel, src_sel = -1;
442 	int i, ret;
443 
444 	if (!data->attached || !data->pd_capable || data->spdo_sel <= 0)
445 		return -EINVAL;
446 
447 	src_voltage = pdo_fixed_voltage(data->spdos[data->spdo_sel - 1]);
448 	if (src_voltage == data->req_voltage)
449 		return 0;
450 
451 	switch (data->req_voltage) {
452 	case 5000:
453 		snk_sel = RT1719_SNKCAP_5V;
454 		break;
455 	case 9000:
456 		snk_sel = RT1719_SNKCAP_9V;
457 		break;
458 	case 12000:
459 		snk_sel = RT1719_SNKCAP_12V;
460 		break;
461 	case 15000:
462 		snk_sel = RT1719_SNKCAP_15V;
463 		break;
464 	case 20000:
465 		snk_sel = RT1719_SNKCAP_20V;
466 		break;
467 	default:
468 		return -EINVAL;
469 	}
470 
471 	if (!(data->snkcaps[snk_sel] & RT1719_PSEL_SUPPORT))
472 		return -EINVAL;
473 
474 	for (i = 0; i < data->spdo_num; i++) {
475 		enum pd_pdo_type type = pdo_type(data->spdos[i]);
476 
477 		if (type != PDO_TYPE_FIXED)
478 			continue;
479 
480 		src_voltage = pdo_fixed_voltage(data->spdos[i]);
481 		if (src_voltage == data->req_voltage) {
482 			src_sel = i;
483 			break;
484 		}
485 	}
486 
487 	if (src_sel == -1)
488 		return -EOPNOTSUPP;
489 
490 	ret = regmap_update_bits(data->regmap, RT1719_REG_TXCTRL1,
491 				 RT1719_EVALMODE_MASK | RT1719_REQSRCPDO_MASK,
492 				 RT1719_EVALMODE_MASK | (src_sel + 1));
493 	ret |= regmap_update_bits(data->regmap, RT1719_REG_TXCTRL2,
494 				  RT1719_TXSPDOREQ_MASK, RT1719_TXSPDOREQ_MASK);
495 	if (ret)
496 		return ret;
497 
498 	reinit_completion(&data->req_completion);
499 	ret = wait_for_completion_timeout(&data->req_completion,
500 					  msecs_to_jiffies(400));
501 	if (!ret)
502 		return -ETIMEDOUT;
503 
504 	return 0;
505 }
506 
507 static int rt1719_psy_set_property(struct power_supply *psy,
508 				   enum power_supply_property psp,
509 				   const union power_supply_propval *val)
510 {
511 	struct rt1719_data *data = power_supply_get_drvdata(psy);
512 
513 	if (psp == POWER_SUPPLY_PROP_VOLTAGE_NOW) {
514 		data->req_voltage = val->intval / 1000;
515 		return rt1719_usbpd_request_voltage(data);
516 	}
517 
518 	return -EINVAL;
519 }
520 
521 static int rt1719_psy_get_property(struct power_supply *psy,
522 				   enum power_supply_property psp,
523 				   union power_supply_propval *val)
524 {
525 	struct rt1719_data *data = power_supply_get_drvdata(psy);
526 	int ret = 0;
527 
528 	switch (psp) {
529 	case POWER_SUPPLY_PROP_ONLINE:
530 		val->intval = data->attached ? 1 : 0;
531 		break;
532 	case POWER_SUPPLY_PROP_USB_TYPE:
533 		val->intval = data->usb_type;
534 		break;
535 	case POWER_SUPPLY_PROP_VOLTAGE_NOW:
536 		val->intval = data->voltage;
537 		break;
538 	case POWER_SUPPLY_PROP_CURRENT_MAX:
539 		val->intval = data->max_current;
540 		break;
541 	case POWER_SUPPLY_PROP_CURRENT_NOW:
542 		val->intval = data->op_current;
543 		break;
544 	default:
545 		ret = -EINVAL;
546 		break;
547 	}
548 
549 	return ret;
550 }
551 
552 static int rt1719_psy_property_is_writeable(struct power_supply *psy,
553 					    enum power_supply_property psp)
554 {
555 	if (psp == POWER_SUPPLY_PROP_VOLTAGE_NOW)
556 		return 1;
557 	return 0;
558 }
559 
560 static int devm_rt1719_psy_register(struct rt1719_data *data)
561 {
562 	struct power_supply_config psy_cfg = { };
563 	char *psy_name;
564 
565 	psy_cfg.fwnode = dev_fwnode(data->dev);
566 	psy_cfg.drv_data = data;
567 
568 	psy_name = devm_kasprintf(data->dev, GFP_KERNEL, "rt1719-source-psy-%s",
569 				  dev_name(data->dev));
570 	if (!psy_name)
571 		return -ENOMEM;
572 
573 	data->psy_desc.name = psy_name;
574 	data->psy_desc.type = POWER_SUPPLY_TYPE_USB;
575 	data->psy_desc.usb_types = rt1719_psy_usb_types;
576 	data->psy_desc.num_usb_types = ARRAY_SIZE(rt1719_psy_usb_types);
577 	data->psy_desc.properties = rt1719_psy_properties;
578 	data->psy_desc.num_properties = ARRAY_SIZE(rt1719_psy_properties);
579 	data->psy_desc.get_property = rt1719_psy_get_property;
580 	data->psy_desc.set_property = rt1719_psy_set_property;
581 	data->psy_desc.property_is_writeable = rt1719_psy_property_is_writeable;
582 
583 	data->usb_type = POWER_SUPPLY_USB_TYPE_C;
584 
585 	data->psy = devm_power_supply_register(data->dev, &data->psy_desc,
586 					       &psy_cfg);
587 
588 	return PTR_ERR_OR_ZERO(data->psy);
589 }
590 
591 static irqreturn_t rt1719_irq_handler(int irq, void *priv)
592 {
593 	struct rt1719_data *data = priv;
594 	u32 events, conn_info;
595 	u16 conn_stat;
596 	int ret;
597 
598 	ret = rt1719_read32(data, RT1719_REG_EVENTS, &events);
599 	ret |= rt1719_read32(data, RT1719_REG_POLICYINFO, &conn_info);
600 	ret |= rt1719_read16(data, RT1719_REG_STATS, &conn_stat);
601 	if (ret)
602 		return IRQ_NONE;
603 
604 	data->conn_info = conn_info;
605 	data->conn_stat = conn_stat;
606 
607 	events &= (RT1719_INT_DRSW_ACCEPT | RT1719_INT_RX_SRCCAP |
608 		   RT1719_INT_VBUS_PRESENT | RT1719_INT_VBUS_DCT |
609 		   RT1719_INT_PE_SNK_RDY);
610 
611 	if (events & RT1719_INT_DRSW_ACCEPT)
612 		rt1719_update_data_role(data);
613 
614 	if (events & RT1719_INT_VBUS_PRESENT)
615 		rt1719_attach(data);
616 
617 	if (events & RT1719_INT_VBUS_DCT)
618 		rt1719_detach(data);
619 
620 	if (events & RT1719_INT_RX_SRCCAP)
621 		rt1719_update_source_pdos(data);
622 
623 	if (events & RT1719_INT_PE_SNK_RDY) {
624 		complete(&data->req_completion);
625 		rt1719_update_pwr_opmode(data);
626 	}
627 
628 	/* Write 1 to clear already handled events */
629 	rt1719_write32(data, RT1719_REG_EVENTS, events);
630 
631 	return IRQ_HANDLED;
632 }
633 
634 static int rt1719_irq_init(struct rt1719_data *data)
635 {
636 	struct i2c_client *i2c = to_i2c_client(data->dev);
637 	u32 irq_enable;
638 	int ret;
639 
640 	irq_enable = RT1719_INT_DRSW_ACCEPT | RT1719_INT_RX_SRCCAP |
641 		     RT1719_INT_VBUS_DCT | RT1719_INT_VBUS_PRESENT |
642 		     RT1719_INT_PE_SNK_RDY;
643 
644 	ret = rt1719_write32(data, RT1719_REG_MASKS, irq_enable);
645 	if (ret) {
646 		dev_err(&i2c->dev, "Failed to config irq enable\n");
647 		return ret;
648 	}
649 
650 	return devm_request_threaded_irq(&i2c->dev, i2c->irq, NULL,
651 					 rt1719_irq_handler, IRQF_ONESHOT,
652 					 dev_name(&i2c->dev), data);
653 }
654 
655 static int rt1719_init_attach_state(struct rt1719_data *data)
656 {
657 	u32 conn_info, irq_clear;
658 	u16 conn_stat;
659 	int ret;
660 
661 	irq_clear = RT1719_INT_DRSW_ACCEPT | RT1719_INT_RX_SRCCAP |
662 		    RT1719_INT_VBUS_DCT | RT1719_INT_VBUS_PRESENT |
663 		    RT1719_INT_PE_SNK_RDY;
664 
665 	ret = rt1719_read32(data, RT1719_REG_POLICYINFO, &conn_info);
666 	ret |= rt1719_read16(data, RT1719_REG_STATS, &conn_stat);
667 	ret |= rt1719_write32(data, RT1719_REG_EVENTS, irq_clear);
668 	if (ret)
669 		return ret;
670 
671 	data->conn_info = conn_info;
672 	data->conn_stat = conn_stat;
673 
674 	if (conn_info & RT1719_ATTACHDEV_MASK)
675 		rt1719_attach(data);
676 
677 	if (conn_info & RT1719_PE_EXP_CONTRACT) {
678 		rt1719_update_source_pdos(data);
679 		rt1719_update_pwr_opmode(data);
680 	}
681 
682 	return 0;
683 }
684 
685 #define RT1719_PSEL_CAPINFO(_lomask, _milliwatt, _himask, _milliamp) { \
686 	.lomask		= _lomask, \
687 	.milliwatt	= _milliwatt, \
688 	.himask		= _himask, \
689 	.milliamp	= _milliamp, \
690 }
691 
692 static const struct rt1719_psel_cap rt1719_psel_caps[] = {
693 	RT1719_PSEL_CAPINFO(0x18, 75000, 0x10, 5000),
694 	RT1719_PSEL_CAPINFO(0x18, 60000, 0x10, 4500),
695 	RT1719_PSEL_CAPINFO(0x18, 45000, 0x10, 4000),
696 	RT1719_PSEL_CAPINFO(0x18, 30000, 0x10, 3500),
697 	RT1719_PSEL_CAPINFO(0x18, 25000, 0x10, 3000),
698 	RT1719_PSEL_CAPINFO(0x18, 20000, 0x10, 2500),
699 	RT1719_PSEL_CAPINFO(0x18, 15000, 0x10, 2000),
700 	RT1719_PSEL_CAPINFO(0x18, 10000, 0x10, 1000),
701 	RT1719_PSEL_CAPINFO(0x1C, 60000, 0x1F, 5000),
702 	RT1719_PSEL_CAPINFO(0x1C, 45000, 0x1F, 4500),
703 	RT1719_PSEL_CAPINFO(0x1C, 30000, 0x1F, 4000),
704 	RT1719_PSEL_CAPINFO(0x1C, 24000, 0x1F, 3500),
705 	RT1719_PSEL_CAPINFO(0x1C, 15000, 0x1F, 3000),
706 	RT1719_PSEL_CAPINFO(0x1C, 10000, 0x1F, 2500),
707 	RT1719_PSEL_CAPINFO(0x0C, 60000, 0x1F, 2000),
708 	RT1719_PSEL_CAPINFO(0x0C, 45000, 0x1F, 1000),
709 	RT1719_PSEL_CAPINFO(0x0C, 36000, 0x08, 5000),
710 	RT1719_PSEL_CAPINFO(0x0C, 30000, 0x08, 4500),
711 	RT1719_PSEL_CAPINFO(0x0C, 24000, 0x08, 4000),
712 	RT1719_PSEL_CAPINFO(0x0C, 15000, 0x08, 3500),
713 	RT1719_PSEL_CAPINFO(0x0C, 10000, 0x08, 3000),
714 	RT1719_PSEL_CAPINFO(0x1E, 45000, 0x08, 2500),
715 	RT1719_PSEL_CAPINFO(0x1E, 36000, 0x08, 2000),
716 	RT1719_PSEL_CAPINFO(0x1E, 27000, 0x08, 1500),
717 	RT1719_PSEL_CAPINFO(0x1E, 20000, 0x08, 1000),
718 	RT1719_PSEL_CAPINFO(0x1E, 15000, 0x0F, 5000),
719 	RT1719_PSEL_CAPINFO(0x1E, 9000, 0x0F, 4500),
720 	RT1719_PSEL_CAPINFO(0x0E, 45000, 0x0F, 4000),
721 	RT1719_PSEL_CAPINFO(0x0E, 36000, 0x0F, 3500),
722 	RT1719_PSEL_CAPINFO(0x0E, 27000, 0x0F, 3000),
723 	RT1719_PSEL_CAPINFO(0x0E, 20000, 0x0F, 2500),
724 	RT1719_PSEL_CAPINFO(0x0E, 15000, 0x0F, 2000),
725 	RT1719_PSEL_CAPINFO(0x0E, 9000, 0x0F, 1500),
726 	RT1719_PSEL_CAPINFO(0x06, 45000, 0x0F, 1000),
727 	RT1719_PSEL_CAPINFO(0x06, 36000, 0x0F, 500),
728 	RT1719_PSEL_CAPINFO(0x06, 27000, 0x04, 5000),
729 	RT1719_PSEL_CAPINFO(0x06, 24000, 0x04, 4500),
730 	RT1719_PSEL_CAPINFO(0x06, 18000, 0x04, 4000),
731 	RT1719_PSEL_CAPINFO(0x06, 12000, 0x04, 3500),
732 	RT1719_PSEL_CAPINFO(0x06, 9000, 0x04, 3000),
733 	RT1719_PSEL_CAPINFO(0x1F, 25000, 0x04, 2500),
734 	RT1719_PSEL_CAPINFO(0x1F, 20000, 0x04, 2000),
735 	RT1719_PSEL_CAPINFO(0x1F, 15000, 0x04, 1500),
736 	RT1719_PSEL_CAPINFO(0x1F, 10000, 0x04, 1000),
737 	RT1719_PSEL_CAPINFO(0x1F, 7500, 0x07, 5000),
738 	RT1719_PSEL_CAPINFO(0x0F, 25000, 0x07, 4500),
739 	RT1719_PSEL_CAPINFO(0x0F, 20000, 0x07, 4000),
740 	RT1719_PSEL_CAPINFO(0x0F, 15000, 0x07, 3500),
741 	RT1719_PSEL_CAPINFO(0x0F, 10000, 0x07, 3000),
742 	RT1719_PSEL_CAPINFO(0x0F, 7500, 0x07, 2500),
743 	RT1719_PSEL_CAPINFO(0x07, 25000, 0x07, 2000),
744 	RT1719_PSEL_CAPINFO(0x07, 20000, 0x07, 1500),
745 	RT1719_PSEL_CAPINFO(0x07, 15000, 0x07, 1000),
746 	RT1719_PSEL_CAPINFO(0x07, 10000, 0x07, 500),
747 	RT1719_PSEL_CAPINFO(0x07, 7500, 0x03, 5000),
748 	RT1719_PSEL_CAPINFO(0x03, 25000, 0x03, 4500),
749 	RT1719_PSEL_CAPINFO(0x03, 20000, 0x03, 4000),
750 	RT1719_PSEL_CAPINFO(0x03, 15000, 0x03, 3500),
751 	RT1719_PSEL_CAPINFO(0x03, 10000, 0x03, 3000),
752 	RT1719_PSEL_CAPINFO(0x03, 7500, 0x03, 2500),
753 	RT1719_PSEL_CAPINFO(0x01, 15000, 0x03, 2000),
754 	RT1719_PSEL_CAPINFO(0x01, 10000, 0x03, 1500),
755 	RT1719_PSEL_CAPINFO(0x01, 7500, 0x03, 1000),
756 	RT1719_PSEL_CAPINFO(0x01, 2500, 0x03, 500)
757 };
758 
759 static u16 rt1719_gen_snkcap_by_current(const struct rt1719_psel_cap *psel_cap,
760 					enum rt1719_snkcap capsel)
761 {
762 	u16 cap = RT1719_PSEL_SUPPORT;
763 
764 	if (!(psel_cap->himask & BIT(capsel)))
765 		return 0;
766 
767 	cap |= psel_cap->milliamp / 10;
768 	return cap;
769 }
770 
771 static u16 rt1719_gen_snkcap_by_watt(const struct rt1719_psel_cap *psel_cap,
772 				     enum rt1719_snkcap capsel)
773 {
774 	u32 volt_div[RT1719_MAX_SNKCAP] = { 5, 9, 12, 15, 20 };
775 	u16 cap = RT1719_PSEL_SUPPORT;
776 
777 	if (!(psel_cap->lomask & BIT(capsel)))
778 		return 0;
779 
780 	cap |= min(psel_cap->milliwatt / volt_div[capsel], (u32)5000) / 10;
781 	return cap;
782 }
783 
784 static u16 rt1719_gen_snkcap(unsigned int pselinfo, enum rt1719_snkcap capsel)
785 {
786 	int psel = FIELD_GET(RT1719_LATPSEL_MASK, pselinfo);
787 	const struct rt1719_psel_cap *psel_cap;
788 	bool by_current = false;
789 
790 	if (pselinfo & RT1719_TBLSEL_MASK)
791 		by_current = true;
792 
793 	psel_cap = rt1719_psel_caps + psel;
794 	if (by_current)
795 		return rt1719_gen_snkcap_by_current(psel_cap, capsel);
796 
797 	return rt1719_gen_snkcap_by_watt(psel_cap, capsel);
798 }
799 
800 static int rt1719_get_caps(struct rt1719_data *data)
801 {
802 	unsigned int pselinfo, usbinfo;
803 	int i, ret;
804 
805 	ret = regmap_read(data->regmap, RT1719_REG_PSELINFO, &pselinfo);
806 	ret |= regmap_read(data->regmap, RT1719_REG_USBSETINFO, &usbinfo);
807 	if (ret)
808 		return ret;
809 
810 	for (i = 0; i < RT1719_MAX_SNKCAP; i++)
811 		data->snkcaps[i] = rt1719_gen_snkcap(pselinfo, i);
812 
813 	usbinfo = FIELD_GET(RT1719_USBINFO_MASK, usbinfo);
814 	if (usbinfo == RT1719_USB_DFPUFP)
815 		data->drswap_support = true;
816 
817 	return 0;
818 }
819 
820 static int rt1719_check_exist(struct rt1719_data *data)
821 {
822 	u16 pid;
823 	int ret;
824 
825 	ret = rt1719_read16(data, RT1719_REG_VENID, &pid);
826 	if (ret)
827 		return ret;
828 
829 	if (pid != RT1719_UNIQUE_PID) {
830 		dev_err(data->dev, "Incorrect PID 0x%04x\n", pid);
831 		return -ENODEV;
832 	}
833 
834 	return 0;
835 }
836 
837 static const struct regmap_config rt1719_regmap_config = {
838 	.reg_bits = 8,
839 	.val_bits = 8,
840 	.max_register = 0xff,
841 };
842 
843 static int rt1719_probe(struct i2c_client *i2c)
844 {
845 	struct rt1719_data *data;
846 	struct fwnode_handle *fwnode;
847 	struct typec_capability typec_cap = { };
848 	int ret;
849 
850 	data = devm_kzalloc(&i2c->dev, sizeof(*data), GFP_KERNEL);
851 	if (!data)
852 		return -ENOMEM;
853 
854 	data->dev = &i2c->dev;
855 	init_completion(&data->req_completion);
856 
857 	data->regmap = devm_regmap_init_i2c(i2c, &rt1719_regmap_config);
858 	if (IS_ERR(data->regmap)) {
859 		ret = PTR_ERR(data->regmap);
860 		dev_err(&i2c->dev, "Failed to init regmap (%d)\n", ret);
861 		return ret;
862 	}
863 
864 	ret = rt1719_check_exist(data);
865 	if (ret)
866 		return ret;
867 
868 	ret = rt1719_get_caps(data);
869 	if (ret)
870 		return ret;
871 
872 	fwnode = device_get_named_child_node(&i2c->dev, "connector");
873 	if (!fwnode)
874 		return -ENODEV;
875 
876 	data->role_sw = fwnode_usb_role_switch_get(fwnode);
877 	if (IS_ERR(data->role_sw)) {
878 		ret = PTR_ERR(data->role_sw);
879 		dev_err(&i2c->dev, "Failed to get usb role switch (%d)\n", ret);
880 		goto err_fwnode_put;
881 	}
882 
883 	ret = devm_rt1719_psy_register(data);
884 	if (ret) {
885 		dev_err(&i2c->dev, "Failed to register psy (%d)\n", ret);
886 		goto err_role_put;
887 	}
888 
889 	typec_cap.revision = USB_TYPEC_REV_1_2;
890 	typec_cap.pd_revision = 0x300;	/* USB-PD spec release 3.0 */
891 	typec_cap.type = TYPEC_PORT_SNK;
892 	typec_cap.data = TYPEC_PORT_DRD;
893 	typec_cap.ops = &rt1719_port_ops;
894 	typec_cap.fwnode = fwnode;
895 	typec_cap.driver_data = data;
896 	typec_cap.accessory[0] = TYPEC_ACCESSORY_DEBUG;
897 
898 	data->partner_desc.identity = &data->partner_ident;
899 
900 	data->port = typec_register_port(&i2c->dev, &typec_cap);
901 	if (IS_ERR(data->port)) {
902 		ret = PTR_ERR(data->port);
903 		dev_err(&i2c->dev, "Failed to register typec port (%d)\n", ret);
904 		goto err_role_put;
905 	}
906 
907 	ret = rt1719_init_attach_state(data);
908 	if (ret) {
909 		dev_err(&i2c->dev, "Failed to init attach state (%d)\n", ret);
910 		goto err_role_put;
911 	}
912 
913 	ret = rt1719_irq_init(data);
914 	if (ret) {
915 		dev_err(&i2c->dev, "Failed to init irq\n");
916 		goto err_role_put;
917 	}
918 
919 	fwnode_handle_put(fwnode);
920 
921 	i2c_set_clientdata(i2c, data);
922 
923 	return 0;
924 
925 err_role_put:
926 	usb_role_switch_put(data->role_sw);
927 err_fwnode_put:
928 	fwnode_handle_put(fwnode);
929 
930 	return ret;
931 }
932 
933 static int rt1719_remove(struct i2c_client *i2c)
934 {
935 	struct rt1719_data *data = i2c_get_clientdata(i2c);
936 
937 	typec_unregister_port(data->port);
938 	usb_role_switch_put(data->role_sw);
939 
940 	return 0;
941 }
942 
943 static const struct of_device_id __maybe_unused rt1719_device_table[] = {
944 	{ .compatible = "richtek,rt1719", },
945 	{ }
946 };
947 MODULE_DEVICE_TABLE(of, rt1719_device_table);
948 
949 static struct i2c_driver rt1719_driver = {
950 	.driver = {
951 		.name = "rt1719",
952 		.of_match_table = rt1719_device_table,
953 	},
954 	.probe_new = rt1719_probe,
955 	.remove = rt1719_remove,
956 };
957 module_i2c_driver(rt1719_driver);
958 
959 MODULE_AUTHOR("ChiYuan Huang <cy_huang@richtek.com>");
960 MODULE_DESCRIPTION("Richtek RT1719 Sink Only USBPD Controller Driver");
961 MODULE_LICENSE("GPL v2");
962