xref: /linux/drivers/power/supply/max77759_charger.c (revision 889600e21e3be388a6817c2a0dac0411df860751)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * max77759_charger.c - Battery charger driver for MAX77759 charger device.
4  *
5  * Copyright 2025 Google LLC.
6  */
7 
8 #include <linux/bitfield.h>
9 #include <linux/cleanup.h>
10 #include <linux/device.h>
11 #include <linux/devm-helpers.h>
12 #include <linux/interrupt.h>
13 #include <linux/irq.h>
14 #include <linux/linear_range.h>
15 #include <linux/mfd/max77759.h>
16 #include <linux/module.h>
17 #include <linux/mutex.h>
18 #include <linux/of.h>
19 #include <linux/platform_device.h>
20 #include <linux/power_supply.h>
21 #include <linux/regmap.h>
22 #include <linux/regulator/driver.h>
23 #include <linux/string_choices.h>
24 #include <linux/workqueue.h>
25 
26 /* Default values for Fast Charge Current & Float Voltage */
27 #define CHG_CC_DEFAULT_UA			2266770
28 #define CHG_FV_DEFAULT_MV			4300
29 
30 #define MAX_NUM_RETRIES				3
31 #define PSY_WORK_RETRY_DELAY_MS			10
32 
33 #define FOREACH_IRQ(S)			\
34 	S(AICL),			\
35 	S(CHGIN),			\
36 	S(CHG),				\
37 	S(INLIM),			\
38 	S(BAT_OILO),			\
39 	S(CHG_STA_CC),			\
40 	S(CHG_STA_CV),			\
41 	S(CHG_STA_TO),			\
42 	S(CHG_STA_DONE)
43 
44 #define GENERATE_ENUM(e)		e
45 #define GENERATE_STRING(s)		#s
46 
47 enum {
48 	FOREACH_IRQ(GENERATE_ENUM)
49 };
50 
51 static const char *const chgr_irqs_str[] = {
52 	FOREACH_IRQ(GENERATE_STRING)
53 };
54 
55 #define NUM_IRQS			ARRAY_SIZE(chgr_irqs_str)
56 
57 /* Fast charge current limits (in uA) */
58 static const struct linear_range chgcc_limit_ranges[] = {
59 	LINEAR_RANGE(133330, 0x0, 0x2, 0),
60 	LINEAR_RANGE(200000, 0x3, 0x3C, 66670),
61 };
62 
63 /* Charge Termination Voltage Limits (in mV) */
64 static const struct linear_range chg_cv_prm_ranges[] = {
65 	LINEAR_RANGE(3800, 0x38, 0x39, 100),
66 	LINEAR_RANGE(4000, 0x0, 0x32, 10),
67 };
68 
69 /* USB input current limits (in uA) */
70 static const struct linear_range chgin_ilim_ranges[] = {
71 	LINEAR_RANGE(100000, 0x3, 0x7F, 25000),
72 };
73 
74 struct max77759_charger {
75 	struct device *dev;
76 	struct regmap *regmap;
77 	struct power_supply *psy;
78 	struct regulator_dev *chgin_otg_rdev;
79 	struct notifier_block nb;
80 	struct power_supply *tcpm_psy;
81 	struct delayed_work psy_work;
82 	struct mutex retry_lock; /* Protects psy_work_retry_cnt */
83 	u32 psy_work_retry_cnt;
84 	int irqs[NUM_IRQS];
85 	struct mutex lock; /* protects the state below */
86 	enum max77759_chgr_mode mode;
87 };
88 
unlock_prot_regs(struct max77759_charger * chg,bool unlock)89 static inline int unlock_prot_regs(struct max77759_charger *chg, bool unlock)
90 {
91 	return regmap_update_bits(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_06,
92 				  MAX77759_CHGR_REG_CHG_CNFG_06_CHGPROT, unlock
93 				  ? MAX77759_CHGR_REG_CHG_CNFG_06_CHGPROT : 0);
94 }
95 
charger_input_valid(struct max77759_charger * chg)96 static int charger_input_valid(struct max77759_charger *chg)
97 {
98 	u32 val;
99 	int ret;
100 
101 	ret = regmap_read(chg->regmap, MAX77759_CHGR_REG_CHG_INT_OK, &val);
102 	if (ret)
103 		return ret;
104 
105 	return (val & MAX77759_CHGR_REG_CHG_INT_CHG) &&
106 		(val & MAX77759_CHGR_REG_CHG_INT_CHGIN);
107 }
108 
get_online(struct max77759_charger * chg)109 static int get_online(struct max77759_charger *chg)
110 {
111 	u32 val;
112 	int ret;
113 
114 	ret = charger_input_valid(chg);
115 	if (ret <= 0)
116 		return ret;
117 
118 	ret = regmap_read(chg->regmap, MAX77759_CHGR_REG_CHG_DETAILS_02, &val);
119 	if (ret)
120 		return ret;
121 
122 	guard(mutex)(&chg->lock);
123 
124 	return (val & MAX77759_CHGR_REG_CHG_DETAILS_02_CHGIN_STS) &&
125 		(chg->mode == MAX77759_CHGR_MODE_CHG_BUCK_ON);
126 }
127 
get_status(struct max77759_charger * chg)128 static int get_status(struct max77759_charger *chg)
129 {
130 	u32 val;
131 	int ret;
132 
133 	ret = regmap_read(chg->regmap, MAX77759_CHGR_REG_CHG_DETAILS_01, &val);
134 	if (ret)
135 		return ret;
136 
137 	switch (FIELD_GET(MAX77759_CHGR_REG_CHG_DETAILS_01_CHG_DTLS, val)) {
138 	case MAX77759_CHGR_CHG_DTLS_PREQUAL:
139 	case MAX77759_CHGR_CHG_DTLS_CC:
140 	case MAX77759_CHGR_CHG_DTLS_CV:
141 	case MAX77759_CHGR_CHG_DTLS_TO:
142 		return POWER_SUPPLY_STATUS_CHARGING;
143 	case MAX77759_CHGR_CHG_DTLS_DONE:
144 		return POWER_SUPPLY_STATUS_FULL;
145 	case MAX77759_CHGR_CHG_DTLS_TIMER_FAULT:
146 	case MAX77759_CHGR_CHG_DTLS_SUSP_BATT_THM:
147 	case MAX77759_CHGR_CHG_DTLS_OFF_WDOG_TIMER:
148 	case MAX77759_CHGR_CHG_DTLS_SUSP_JEITA:
149 		return POWER_SUPPLY_STATUS_NOT_CHARGING;
150 	case MAX77759_CHGR_CHG_DTLS_OFF:
151 		return POWER_SUPPLY_STATUS_DISCHARGING;
152 	default:
153 		break;
154 	}
155 
156 	return POWER_SUPPLY_STATUS_UNKNOWN;
157 }
158 
get_charge_type(struct max77759_charger * chg)159 static int get_charge_type(struct max77759_charger *chg)
160 {
161 	u32 val;
162 	int ret;
163 
164 	ret = regmap_read(chg->regmap, MAX77759_CHGR_REG_CHG_DETAILS_01, &val);
165 	if (ret)
166 		return ret;
167 
168 	switch (FIELD_GET(MAX77759_CHGR_REG_CHG_DETAILS_01_CHG_DTLS, val)) {
169 	case MAX77759_CHGR_CHG_DTLS_PREQUAL:
170 		return POWER_SUPPLY_CHARGE_TYPE_TRICKLE;
171 	case MAX77759_CHGR_CHG_DTLS_CC:
172 	case MAX77759_CHGR_CHG_DTLS_CV:
173 		return POWER_SUPPLY_CHARGE_TYPE_FAST;
174 	case MAX77759_CHGR_CHG_DTLS_TO:
175 		return POWER_SUPPLY_CHARGE_TYPE_STANDARD;
176 	case MAX77759_CHGR_CHG_DTLS_DONE:
177 	case MAX77759_CHGR_CHG_DTLS_TIMER_FAULT:
178 	case MAX77759_CHGR_CHG_DTLS_SUSP_BATT_THM:
179 	case MAX77759_CHGR_CHG_DTLS_OFF_WDOG_TIMER:
180 	case MAX77759_CHGR_CHG_DTLS_SUSP_JEITA:
181 	case MAX77759_CHGR_CHG_DTLS_OFF:
182 		return POWER_SUPPLY_CHARGE_TYPE_NONE;
183 	default:
184 		break;
185 	}
186 
187 	return POWER_SUPPLY_CHARGE_TYPE_UNKNOWN;
188 }
189 
get_chg_health(struct max77759_charger * chg)190 static int get_chg_health(struct max77759_charger *chg)
191 {
192 	u32 val;
193 	int ret;
194 
195 	ret = regmap_read(chg->regmap, MAX77759_CHGR_REG_CHG_DETAILS_00, &val);
196 	if (ret)
197 		return ret;
198 
199 	switch (FIELD_GET(MAX77759_CHGR_REG_CHG_DETAILS_00_CHGIN_DTLS, val)) {
200 	case MAX77759_CHGR_CHGIN_DTLS_VBUS_UNDERVOLTAGE:
201 	case MAX77759_CHGR_CHGIN_DTLS_VBUS_MARGINAL_VOLTAGE:
202 		return POWER_SUPPLY_HEALTH_UNDERVOLTAGE;
203 	case MAX77759_CHGR_CHGIN_DTLS_VBUS_OVERVOLTAGE:
204 		return POWER_SUPPLY_HEALTH_OVERVOLTAGE;
205 	case MAX77759_CHGR_CHGIN_DTLS_VBUS_VALID:
206 		return POWER_SUPPLY_HEALTH_GOOD;
207 	default:
208 		break;
209 	}
210 
211 	return POWER_SUPPLY_HEALTH_UNKNOWN;
212 }
213 
get_batt_health(struct max77759_charger * chg)214 static int get_batt_health(struct max77759_charger *chg)
215 {
216 	u32 val;
217 	int ret;
218 
219 	ret = regmap_read(chg->regmap, MAX77759_CHGR_REG_CHG_DETAILS_01, &val);
220 	if (ret)
221 		return ret;
222 
223 	switch (FIELD_GET(MAX77759_CHGR_REG_CHG_DETAILS_01_BAT_DTLS, val)) {
224 	case MAX77759_CHGR_BAT_DTLS_NO_BATT_CHG_SUSP:
225 		return POWER_SUPPLY_HEALTH_NO_BATTERY;
226 	case MAX77759_CHGR_BAT_DTLS_DEAD_BATTERY:
227 		return POWER_SUPPLY_HEALTH_DEAD;
228 	case MAX77759_CHGR_BAT_DTLS_BAT_CHG_TIMER_FAULT:
229 		return POWER_SUPPLY_HEALTH_SAFETY_TIMER_EXPIRE;
230 	case MAX77759_CHGR_BAT_DTLS_BAT_OKAY:
231 	case MAX77759_CHGR_BAT_DTLS_BAT_ONLY_MODE:
232 		return POWER_SUPPLY_HEALTH_GOOD;
233 	case MAX77759_CHGR_BAT_DTLS_BAT_UNDERVOLTAGE:
234 		return POWER_SUPPLY_HEALTH_UNDERVOLTAGE;
235 	case MAX77759_CHGR_BAT_DTLS_BAT_OVERVOLTAGE:
236 		return POWER_SUPPLY_HEALTH_OVERVOLTAGE;
237 	case MAX77759_CHGR_BAT_DTLS_BAT_OVERCURRENT:
238 		return POWER_SUPPLY_HEALTH_OVERCURRENT;
239 	default:
240 		break;
241 	}
242 
243 	return POWER_SUPPLY_HEALTH_UNKNOWN;
244 }
245 
get_health(struct max77759_charger * chg)246 static int get_health(struct max77759_charger *chg)
247 {
248 	int ret;
249 
250 	ret = get_online(chg);
251 	if (ret < 0)
252 		return ret;
253 
254 	if (ret) {
255 		ret = get_chg_health(chg);
256 		if (ret < 0 || ret != POWER_SUPPLY_HEALTH_GOOD)
257 			return ret;
258 	}
259 
260 	return get_batt_health(chg);
261 }
262 
get_fast_charge_current(struct max77759_charger * chg)263 static int get_fast_charge_current(struct max77759_charger *chg)
264 {
265 	u32 regval, val;
266 	int ret;
267 
268 	ret = regmap_read(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_02, &regval);
269 	if (ret)
270 		return ret;
271 
272 	regval = FIELD_GET(MAX77759_CHGR_REG_CHG_CNFG_02_CHGCC, regval);
273 	ret = linear_range_get_value_array(chgcc_limit_ranges,
274 					   ARRAY_SIZE(chgcc_limit_ranges),
275 					   regval, &val);
276 	return ret ? ret : val;
277 }
278 
set_fast_charge_current_limit(struct max77759_charger * chg,u32 cc_max_ua)279 static int set_fast_charge_current_limit(struct max77759_charger *chg,
280 					 u32 cc_max_ua)
281 {
282 	bool found;
283 	u32 regval;
284 
285 	linear_range_get_selector_high_array(chgcc_limit_ranges,
286 					     ARRAY_SIZE(chgcc_limit_ranges),
287 					     cc_max_ua, &regval, &found);
288 	if (!found)
289 		return -EINVAL;
290 
291 	return regmap_update_bits(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_02,
292 				  MAX77759_CHGR_REG_CHG_CNFG_02_CHGCC, regval);
293 }
294 
get_float_voltage(struct max77759_charger * chg)295 static int get_float_voltage(struct max77759_charger *chg)
296 {
297 	u32 regval, val;
298 	int ret;
299 
300 	ret = regmap_read(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_04, &regval);
301 	if (ret)
302 		return ret;
303 
304 	regval = FIELD_GET(MAX77759_CHGR_REG_CHG_CNFG_04_CHG_CV_PRM, regval);
305 	ret = linear_range_get_value_array(chg_cv_prm_ranges,
306 					   ARRAY_SIZE(chg_cv_prm_ranges),
307 					   regval, &val);
308 
309 	return ret ? ret : val;
310 }
311 
set_float_voltage_limit(struct max77759_charger * chg,u32 fv_mv)312 static int set_float_voltage_limit(struct max77759_charger *chg, u32 fv_mv)
313 {
314 	u32 regval;
315 	bool found;
316 
317 	linear_range_get_selector_high_array(chg_cv_prm_ranges,
318 					     ARRAY_SIZE(chg_cv_prm_ranges),
319 					     fv_mv, &regval, &found);
320 	if (!found)
321 		return -EINVAL;
322 
323 	return regmap_update_bits(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_04,
324 				  MAX77759_CHGR_REG_CHG_CNFG_04_CHG_CV_PRM,
325 				  regval);
326 }
327 
get_input_current_limit(struct max77759_charger * chg)328 static int get_input_current_limit(struct max77759_charger *chg)
329 {
330 	u32 regval, val;
331 	int ret;
332 
333 	ret = regmap_read(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_09, &regval);
334 	if (ret)
335 		return ret;
336 
337 	regval = FIELD_GET(MAX77759_CHGR_REG_CHG_CNFG_09_CHGIN_ILIM, regval);
338 	regval = umax(regval, chgin_ilim_ranges[0].min_sel);
339 
340 	ret = linear_range_get_value_array(chgin_ilim_ranges,
341 					   ARRAY_SIZE(chgin_ilim_ranges),
342 					   regval, &val);
343 
344 	return ret ? ret : val;
345 }
346 
set_input_current_limit(struct max77759_charger * chg,int ilim_ua)347 static int set_input_current_limit(struct max77759_charger *chg, int ilim_ua)
348 {
349 	u32 regval;
350 
351 	if (ilim_ua < 0)
352 		return -EINVAL;
353 
354 	linear_range_get_selector_within(chgin_ilim_ranges, ilim_ua, &regval);
355 
356 	return regmap_update_bits(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_09,
357 				  MAX77759_CHGR_REG_CHG_CNFG_09_CHGIN_ILIM,
358 				  regval);
359 }
360 
361 static const enum power_supply_property max77759_charger_props[] = {
362 	POWER_SUPPLY_PROP_ONLINE,
363 	POWER_SUPPLY_PROP_PRESENT,
364 	POWER_SUPPLY_PROP_STATUS,
365 	POWER_SUPPLY_PROP_CHARGE_TYPE,
366 	POWER_SUPPLY_PROP_HEALTH,
367 	POWER_SUPPLY_PROP_CONSTANT_CHARGE_CURRENT_MAX,
368 	POWER_SUPPLY_PROP_CONSTANT_CHARGE_VOLTAGE_MAX,
369 	POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT,
370 };
371 
max77759_charger_get_property(struct power_supply * psy,enum power_supply_property psp,union power_supply_propval * pval)372 static int max77759_charger_get_property(struct power_supply *psy,
373 					 enum power_supply_property psp,
374 					 union power_supply_propval *pval)
375 {
376 	struct max77759_charger *chg = power_supply_get_drvdata(psy);
377 	int ret;
378 
379 	switch (psp) {
380 	case POWER_SUPPLY_PROP_ONLINE:
381 		ret = get_online(chg);
382 		break;
383 	case POWER_SUPPLY_PROP_PRESENT:
384 		ret = charger_input_valid(chg);
385 		break;
386 	case POWER_SUPPLY_PROP_STATUS:
387 		ret = get_status(chg);
388 		break;
389 	case POWER_SUPPLY_PROP_CHARGE_TYPE:
390 		ret = get_charge_type(chg);
391 		break;
392 	case POWER_SUPPLY_PROP_HEALTH:
393 		ret = get_health(chg);
394 		break;
395 	case POWER_SUPPLY_PROP_CONSTANT_CHARGE_CURRENT_MAX:
396 		ret = get_fast_charge_current(chg);
397 		break;
398 	case POWER_SUPPLY_PROP_CONSTANT_CHARGE_VOLTAGE_MAX:
399 		ret = get_float_voltage(chg);
400 		break;
401 	case POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT:
402 		ret = get_input_current_limit(chg);
403 		break;
404 	default:
405 		ret = -EINVAL;
406 	}
407 
408 	pval->intval = ret;
409 	return ret < 0 ? ret : 0;
410 }
411 
412 static const struct power_supply_desc max77759_charger_desc = {
413 	.name = "max77759-charger",
414 	.type = POWER_SUPPLY_TYPE_USB,
415 	.properties = max77759_charger_props,
416 	.num_properties = ARRAY_SIZE(max77759_charger_props),
417 	.get_property = max77759_charger_get_property,
418 };
419 
charger_set_mode(struct max77759_charger * chg,enum max77759_chgr_mode mode)420 static int charger_set_mode(struct max77759_charger *chg,
421 			    enum max77759_chgr_mode mode)
422 {
423 	int ret;
424 
425 	guard(mutex)(&chg->lock);
426 
427 	if (chg->mode == mode)
428 		return 0;
429 
430 	if ((mode == MAX77759_CHGR_MODE_CHG_BUCK_ON ||
431 	     mode == MAX77759_CHGR_MODE_OTG_BOOST_ON) &&
432 	    chg->mode != MAX77759_CHGR_MODE_OFF) {
433 		dev_err(chg->dev, "Invalid mode transition from %d to %d\n",
434 			chg->mode, mode);
435 		return -EINVAL;
436 	}
437 
438 	ret = regmap_update_bits(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_00,
439 				 MAX77759_CHGR_REG_CHG_CNFG_00_MODE, mode);
440 	if (ret)
441 		return ret;
442 
443 	chg->mode = mode;
444 	return 0;
445 }
446 
enable_chgin_otg(struct regulator_dev * rdev)447 static int enable_chgin_otg(struct regulator_dev *rdev)
448 {
449 	struct max77759_charger *chg = rdev_get_drvdata(rdev);
450 
451 	return charger_set_mode(chg, MAX77759_CHGR_MODE_OTG_BOOST_ON);
452 }
453 
disable_chgin_otg(struct regulator_dev * rdev)454 static int disable_chgin_otg(struct regulator_dev *rdev)
455 {
456 	struct max77759_charger *chg = rdev_get_drvdata(rdev);
457 
458 	return charger_set_mode(chg, MAX77759_CHGR_MODE_OFF);
459 }
460 
chgin_otg_status(struct regulator_dev * rdev)461 static int chgin_otg_status(struct regulator_dev *rdev)
462 {
463 	struct max77759_charger *chg = rdev_get_drvdata(rdev);
464 
465 	guard(mutex)(&chg->lock);
466 
467 	return chg->mode == MAX77759_CHGR_MODE_OTG_BOOST_ON;
468 }
469 
470 static const struct regulator_ops chgin_otg_reg_ops = {
471 	.enable = enable_chgin_otg,
472 	.disable = disable_chgin_otg,
473 	.is_enabled = chgin_otg_status,
474 };
475 
476 static const struct regulator_desc chgin_otg_reg_desc = {
477 	.name = "chgin-otg",
478 	.of_match = of_match_ptr("chgin-otg-regulator"),
479 	.owner = THIS_MODULE,
480 	.ops = &chgin_otg_reg_ops,
481 	.fixed_uV = 5000000,
482 	.n_voltages = 1,
483 };
484 
irq_handler(int irq,void * data)485 static irqreturn_t irq_handler(int irq, void *data)
486 {
487 	struct max77759_charger *chg = data;
488 
489 	power_supply_changed(chg->psy);
490 
491 	return IRQ_HANDLED;
492 }
493 
bat_oilo_irq_handler(int irq,void * data)494 static irqreturn_t bat_oilo_irq_handler(int irq, void *data)
495 {
496 	struct max77759_charger *chg = data;
497 
498 	dev_warn_ratelimited(chg->dev,
499 			     "Battery over-current threshold crossed\n");
500 
501 	return irq_handler(irq, data);
502 }
503 
max77759_init_irqhandler(struct max77759_charger * chg)504 static int max77759_init_irqhandler(struct max77759_charger *chg)
505 {
506 	struct device *dev = chg->dev;
507 	irq_handler_t thread_fn;
508 	char *name;
509 	int i, ret;
510 
511 	for (i = 0; i < ARRAY_SIZE(chgr_irqs_str); i++) {
512 		ret = platform_get_irq_byname(to_platform_device(dev),
513 					      chgr_irqs_str[i]);
514 		if (ret < 0)
515 			return dev_err_probe(dev, ret,
516 					     "Failed to get irq resource for %s\n",
517 					     chgr_irqs_str[i]);
518 
519 		chg->irqs[i] = ret;
520 		name = devm_kasprintf(dev, GFP_KERNEL, "%s:%s", dev_name(dev),
521 				      chgr_irqs_str[i]);
522 		if (!name)
523 			return dev_err_probe(dev, -ENOMEM,
524 					     "Failed to allocate space for irqname: %s\n",
525 					     chgr_irqs_str[i]);
526 
527 		if (i == BAT_OILO)
528 			thread_fn = bat_oilo_irq_handler;
529 		else
530 			thread_fn = irq_handler;
531 
532 		ret = devm_request_threaded_irq(dev, chg->irqs[i], NULL,
533 						thread_fn, 0, name, chg);
534 		if (ret)
535 			return ret;
536 	}
537 
538 	return 0;
539 }
540 
max77759_charger_init(struct max77759_charger * chg)541 static int max77759_charger_init(struct max77759_charger *chg)
542 {
543 	struct power_supply_battery_info *info;
544 	u32 regval, fast_chg_curr, fv;
545 	int ret;
546 
547 	ret = regmap_read(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_00, &regval);
548 	if (ret)
549 		return ret;
550 
551 	chg->mode = FIELD_GET(MAX77759_CHGR_REG_CHG_CNFG_00_MODE, regval);
552 	ret = charger_set_mode(chg, MAX77759_CHGR_MODE_OFF);
553 	if (ret)
554 		return ret;
555 
556 	if (power_supply_get_battery_info(chg->psy, &info)) {
557 		fv = CHG_FV_DEFAULT_MV;
558 		fast_chg_curr = CHG_CC_DEFAULT_UA;
559 	} else {
560 		fv = info->constant_charge_voltage_max_uv / 1000;
561 		fast_chg_curr = info->constant_charge_current_max_ua;
562 	}
563 
564 	ret = set_fast_charge_current_limit(chg, fast_chg_curr);
565 	if (ret)
566 		return ret;
567 
568 	ret = set_float_voltage_limit(chg, fv);
569 	if (ret)
570 		return ret;
571 
572 	ret = unlock_prot_regs(chg, true);
573 	if (ret)
574 		return ret;
575 
576 	/* Disable wireless charging input */
577 	ret = regmap_update_bits(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_12,
578 				 MAX77759_CHGR_REG_CHG_CNFG_12_WCINSEL, 0);
579 	if (ret)
580 		goto relock;
581 
582 	ret = regmap_update_bits(chg->regmap, MAX77759_CHGR_REG_CHG_CNFG_18,
583 				 MAX77759_CHGR_REG_CHG_CNFG_18_WDTEN, 0);
584 	if (ret)
585 		goto relock;
586 
587 	return unlock_prot_regs(chg, false);
588 
589 relock:
590 	(void)unlock_prot_regs(chg, false);
591 	return ret;
592 }
593 
psy_work_item(struct work_struct * work)594 static void psy_work_item(struct work_struct *work)
595 {
596 	struct max77759_charger *chg =
597 		container_of(work, struct max77759_charger, psy_work.work);
598 	union power_supply_propval current_limit, online;
599 	int ret;
600 
601 	ret = power_supply_get_property(chg->tcpm_psy,
602 					POWER_SUPPLY_PROP_CURRENT_MAX,
603 					&current_limit);
604 	if (ret) {
605 		dev_err(chg->dev,
606 			"Failed to get CURRENT_MAX psy property, ret=%d\n",
607 			ret);
608 		goto err;
609 	}
610 
611 	ret = power_supply_get_property(chg->tcpm_psy, POWER_SUPPLY_PROP_ONLINE,
612 					&online);
613 	if (ret) {
614 		dev_err(chg->dev,
615 			"Failed to get ONLINE psy property, ret=%d\n",
616 			ret);
617 		goto err;
618 	}
619 
620 	if (online.intval && current_limit.intval) {
621 		ret = set_input_current_limit(chg, current_limit.intval);
622 		if (ret) {
623 			dev_err(chg->dev,
624 				"Unable to set current limit, ret=%d\n", ret);
625 			goto err;
626 		}
627 
628 		charger_set_mode(chg, MAX77759_CHGR_MODE_CHG_BUCK_ON);
629 	} else {
630 		charger_set_mode(chg, MAX77759_CHGR_MODE_OFF);
631 	}
632 
633 	scoped_guard(mutex, &chg->retry_lock) {
634 		if (chg->psy_work_retry_cnt)
635 			dev_dbg(chg->dev,
636 				"chg psy_work succeeded after %u tries\n",
637 				chg->psy_work_retry_cnt);
638 		chg->psy_work_retry_cnt = 0;
639 	}
640 
641 	return;
642 
643 err:
644 	charger_set_mode(chg, MAX77759_CHGR_MODE_OFF);
645 	scoped_guard(mutex, &chg->retry_lock) {
646 		if (chg->psy_work_retry_cnt >= MAX_NUM_RETRIES) {
647 			dev_err(chg->dev, "chg psy work failed, giving up\n");
648 			return;
649 		}
650 
651 		++chg->psy_work_retry_cnt;
652 		dev_dbg(chg->dev, "Retrying %u/%u chg psy_work\n",
653 			chg->psy_work_retry_cnt, MAX_NUM_RETRIES);
654 		schedule_delayed_work(&chg->psy_work,
655 				      msecs_to_jiffies(PSY_WORK_RETRY_DELAY_MS));
656 	}
657 }
658 
psy_changed(struct notifier_block * nb,unsigned long evt,void * data)659 static int psy_changed(struct notifier_block *nb, unsigned long evt, void *data)
660 {
661 	struct max77759_charger *chg = container_of(nb, struct max77759_charger,
662 						    nb);
663 	static const char *psy_name = "tcpm-source";
664 	struct power_supply *psy = data;
665 
666 	if (!strnstr(psy->desc->name, psy_name, strlen(psy_name)) ||
667 	    evt != PSY_EVENT_PROP_CHANGED)
668 		return NOTIFY_OK;
669 
670 	chg->tcpm_psy = psy;
671 	scoped_guard(mutex, &chg->retry_lock)
672 		chg->psy_work_retry_cnt = 0;
673 
674 	schedule_delayed_work(&chg->psy_work, 0);
675 
676 	return NOTIFY_OK;
677 }
678 
max_tcpci_unregister_psy_notifier(void * nb)679 static void max_tcpci_unregister_psy_notifier(void *nb)
680 {
681 	power_supply_unreg_notifier(nb);
682 }
683 
max77759_charger_probe(struct platform_device * pdev)684 static int max77759_charger_probe(struct platform_device *pdev)
685 {
686 	struct regulator_config chgin_otg_reg_cfg;
687 	struct power_supply_config psy_cfg;
688 	struct device *dev = &pdev->dev;
689 	struct max77759_charger *chg;
690 	int ret;
691 
692 	device_set_of_node_from_dev(dev, dev->parent);
693 	chg = devm_kzalloc(dev, sizeof(*chg), GFP_KERNEL);
694 	if (!chg)
695 		return -ENOMEM;
696 
697 	platform_set_drvdata(pdev, chg);
698 	chg->dev = dev;
699 	chg->regmap = dev_get_regmap(dev->parent, "charger");
700 	if (!chg->regmap)
701 		return dev_err_probe(dev, -ENODEV, "Missing regmap\n");
702 
703 	ret = devm_mutex_init(dev, &chg->lock);
704 	if (ret)
705 		return dev_err_probe(dev, ret, "Failed to initialize lock\n");
706 
707 	ret = devm_mutex_init(dev, &chg->retry_lock);
708 	if (ret)
709 		return dev_err_probe(dev, ret,
710 				     "Failed to initialize retry_lock\n");
711 
712 	psy_cfg.fwnode = dev_fwnode(dev);
713 	psy_cfg.drv_data = chg;
714 	chg->psy = devm_power_supply_register(dev, &max77759_charger_desc,
715 					      &psy_cfg);
716 	if (IS_ERR(chg->psy))
717 		return dev_err_probe(dev, PTR_ERR(chg->psy),
718 				     "Failed to register psy\n");
719 
720 	ret = max77759_charger_init(chg);
721 	if (ret)
722 		return dev_err_probe(dev, ret,
723 				     "Failed to initialize max77759 charger\n");
724 
725 	chgin_otg_reg_cfg.dev = dev;
726 	chgin_otg_reg_cfg.driver_data = chg;
727 	chgin_otg_reg_cfg.of_node = dev_of_node(dev);
728 	chg->chgin_otg_rdev = devm_regulator_register(dev, &chgin_otg_reg_desc,
729 						      &chgin_otg_reg_cfg);
730 	if (IS_ERR(chg->chgin_otg_rdev))
731 		return dev_err_probe(dev, PTR_ERR(chg->chgin_otg_rdev),
732 				     "Failed to register chgin otg regulator\n");
733 
734 	ret = devm_delayed_work_autocancel(dev, &chg->psy_work, psy_work_item);
735 	if (ret)
736 		return dev_err_probe(dev, ret, "Failed to initialize psy work\n");
737 
738 	chg->nb.notifier_call = psy_changed;
739 	ret = power_supply_reg_notifier(&chg->nb);
740 	if (ret)
741 		return dev_err_probe(dev, ret,
742 				     "Unable to register psy notifier\n");
743 
744 	ret = devm_add_action_or_reset(dev, max_tcpci_unregister_psy_notifier,
745 				       &chg->nb);
746 	if (ret)
747 		return dev_err_probe(dev, ret,
748 				     "Failed to add devm action to unregister psy notifier\n");
749 
750 	return max77759_init_irqhandler(chg);
751 }
752 
753 static const struct platform_device_id max77759_charger_id[] = {
754 	{ .name = "max77759-charger" },
755 	{ }
756 };
757 MODULE_DEVICE_TABLE(platform, max77759_charger_id);
758 
759 static struct platform_driver max77759_charger_driver = {
760 	.driver = {
761 		.name = "max77759-charger",
762 		.probe_type = PROBE_PREFER_ASYNCHRONOUS,
763 	},
764 	.probe = max77759_charger_probe,
765 	.id_table = max77759_charger_id,
766 };
767 module_platform_driver(max77759_charger_driver);
768 
769 MODULE_AUTHOR("Amit Sunil Dhamne <amitsd@google.com>");
770 MODULE_DESCRIPTION("Maxim MAX77759 charger driver");
771 MODULE_LICENSE("GPL");
772