1// SPDX-License-Identifier: GPL-2.0-only
2/*
3 * Copyright (c) 2019-2020, The Linux Foundation. All rights reserved.
4 * Copyright (c) 2022, Linaro Ltd
5 */
6#include <linux/auxiliary_bus.h>
7#include <linux/module.h>
8#include <linux/mutex.h>
9#include <linux/of_device.h>
10#include <linux/power_supply.h>
11#include <linux/soc/qcom/pdr.h>
12#include <linux/soc/qcom/pmic_glink.h>
13#include <linux/math.h>
14#include <linux/units.h>
15
16#define BATTMGR_CHEMISTRY_LEN	4
17#define BATTMGR_STRING_LEN	128
18
19enum qcom_battmgr_variant {
20	QCOM_BATTMGR_SM8350,
21	QCOM_BATTMGR_SC8280XP,
22};
23
24#define BATTMGR_BAT_STATUS		0x1
25
26#define BATTMGR_REQUEST_NOTIFICATION	0x4
27
28#define BATTMGR_NOTIFICATION		0x7
29#define NOTIF_BAT_PROPERTY		0x30
30#define NOTIF_USB_PROPERTY		0x32
31#define NOTIF_WLS_PROPERTY		0x34
32#define NOTIF_BAT_INFO			0x81
33#define NOTIF_BAT_STATUS		0x80
34
35#define BATTMGR_BAT_INFO		0x9
36
37#define BATTMGR_BAT_DISCHARGE_TIME	0xc
38
39#define BATTMGR_BAT_CHARGE_TIME		0xd
40
41#define BATTMGR_BAT_PROPERTY_GET	0x30
42#define BATTMGR_BAT_PROPERTY_SET	0x31
43#define BATT_STATUS			0
44#define BATT_HEALTH			1
45#define BATT_PRESENT			2
46#define BATT_CHG_TYPE			3
47#define BATT_CAPACITY			4
48#define BATT_SOH			5
49#define BATT_VOLT_OCV			6
50#define BATT_VOLT_NOW			7
51#define BATT_VOLT_MAX			8
52#define BATT_CURR_NOW			9
53#define BATT_CHG_CTRL_LIM		10
54#define BATT_CHG_CTRL_LIM_MAX		11
55#define BATT_TEMP			12
56#define BATT_TECHNOLOGY			13
57#define BATT_CHG_COUNTER		14
58#define BATT_CYCLE_COUNT		15
59#define BATT_CHG_FULL_DESIGN		16
60#define BATT_CHG_FULL			17
61#define BATT_MODEL_NAME			18
62#define BATT_TTF_AVG			19
63#define BATT_TTE_AVG			20
64#define BATT_RESISTANCE			21
65#define BATT_POWER_NOW			22
66#define BATT_POWER_AVG			23
67
68#define BATTMGR_USB_PROPERTY_GET	0x32
69#define BATTMGR_USB_PROPERTY_SET	0x33
70#define USB_ONLINE			0
71#define USB_VOLT_NOW			1
72#define USB_VOLT_MAX			2
73#define USB_CURR_NOW			3
74#define USB_CURR_MAX			4
75#define USB_INPUT_CURR_LIMIT		5
76#define USB_TYPE			6
77#define USB_ADAP_TYPE			7
78#define USB_MOISTURE_DET_EN		8
79#define USB_MOISTURE_DET_STS		9
80
81#define BATTMGR_WLS_PROPERTY_GET	0x34
82#define BATTMGR_WLS_PROPERTY_SET	0x35
83#define WLS_ONLINE			0
84#define WLS_VOLT_NOW			1
85#define WLS_VOLT_MAX			2
86#define WLS_CURR_NOW			3
87#define WLS_CURR_MAX			4
88#define WLS_TYPE			5
89#define WLS_BOOST_EN			6
90
91struct qcom_battmgr_enable_request {
92	struct pmic_glink_hdr hdr;
93	__le32 battery_id;
94	__le32 power_state;
95	__le32 low_capacity;
96	__le32 high_capacity;
97};
98
99struct qcom_battmgr_property_request {
100	struct pmic_glink_hdr hdr;
101	__le32 battery;
102	__le32 property;
103	__le32 value;
104};
105
106struct qcom_battmgr_update_request {
107	struct pmic_glink_hdr hdr;
108	__le32 battery_id;
109};
110
111struct qcom_battmgr_charge_time_request {
112	struct pmic_glink_hdr hdr;
113	__le32 battery_id;
114	__le32 percent;
115	__le32 reserved;
116};
117
118struct qcom_battmgr_discharge_time_request {
119	struct pmic_glink_hdr hdr;
120	__le32 battery_id;
121	__le32 rate; /* 0 for current rate */
122	__le32 reserved;
123};
124
125struct qcom_battmgr_message {
126	struct pmic_glink_hdr hdr;
127	union {
128		struct {
129			__le32 property;
130			__le32 value;
131			__le32 result;
132		} intval;
133		struct {
134			__le32 property;
135			char model[BATTMGR_STRING_LEN];
136		} strval;
137		struct {
138			/*
139			 * 0: mWh
140			 * 1: mAh
141			 */
142			__le32 power_unit;
143			__le32 design_capacity;
144			__le32 last_full_capacity;
145			/*
146			 * 0 nonrechargable
147			 * 1 rechargable
148			 */
149			__le32 battery_tech;
150			__le32 design_voltage; /* mV */
151			__le32 capacity_low;
152			__le32 capacity_warning;
153			__le32 cycle_count;
154			/* thousandth of persent */
155			__le32 accuracy;
156			__le32 max_sample_time_ms;
157			__le32 min_sample_time_ms;
158			__le32 max_average_interval_ms;
159			__le32 min_average_interval_ms;
160			/* granularity between low and warning */
161			__le32 capacity_granularity1;
162			/* granularity between warning and full */
163			__le32 capacity_granularity2;
164			/*
165			 * 0: no
166			 * 1: cold
167			 * 2: hot
168			 */
169			__le32 swappable;
170			__le32 capabilities;
171			char model_number[BATTMGR_STRING_LEN];
172			char serial_number[BATTMGR_STRING_LEN];
173			char battery_type[BATTMGR_STRING_LEN];
174			char oem_info[BATTMGR_STRING_LEN];
175			char battery_chemistry[BATTMGR_CHEMISTRY_LEN];
176			char uid[BATTMGR_STRING_LEN];
177			__le32 critical_bias;
178			u8 day;
179			u8 month;
180			__le16 year;
181			__le32 battery_id;
182		} info;
183		struct {
184			/*
185			 * BIT(0) discharging
186			 * BIT(1) charging
187			 * BIT(2) critical low
188			 */
189			__le32 battery_state;
190			/* mWh or mAh, based on info->power_unit */
191			__le32 capacity;
192			__le32 rate;
193			/* mv */
194			__le32 battery_voltage;
195			/*
196			 * BIT(0) power online
197			 * BIT(1) discharging
198			 * BIT(2) charging
199			 * BIT(3) battery critical
200			 */
201			__le32 power_state;
202			/*
203			 * 1: AC
204			 * 2: USB
205			 * 3: Wireless
206			 */
207			__le32 charging_source;
208			__le32 temperature;
209		} status;
210		__le32 time;
211		__le32 notification;
212	};
213};
214
215#define BATTMGR_CHARGING_SOURCE_AC	1
216#define BATTMGR_CHARGING_SOURCE_USB	2
217#define BATTMGR_CHARGING_SOURCE_WIRELESS 3
218
219enum qcom_battmgr_unit {
220	QCOM_BATTMGR_UNIT_mWh = 0,
221	QCOM_BATTMGR_UNIT_mAh = 1
222};
223
224struct qcom_battmgr_info {
225	bool valid;
226
227	bool present;
228	unsigned int charge_type;
229	unsigned int design_capacity;
230	unsigned int last_full_capacity;
231	unsigned int voltage_max_design;
232	unsigned int voltage_max;
233	unsigned int capacity_low;
234	unsigned int capacity_warning;
235	unsigned int cycle_count;
236	unsigned int charge_count;
237	char model_number[BATTMGR_STRING_LEN];
238	char serial_number[BATTMGR_STRING_LEN];
239	char oem_info[BATTMGR_STRING_LEN];
240	unsigned char technology;
241	unsigned char day;
242	unsigned char month;
243	unsigned short year;
244};
245
246struct qcom_battmgr_status {
247	unsigned int status;
248	unsigned int health;
249	unsigned int capacity;
250	unsigned int percent;
251	int current_now;
252	int power_now;
253	unsigned int voltage_now;
254	unsigned int voltage_ocv;
255	unsigned int temperature;
256
257	unsigned int discharge_time;
258	unsigned int charge_time;
259};
260
261struct qcom_battmgr_ac {
262	bool online;
263};
264
265struct qcom_battmgr_usb {
266	bool online;
267	unsigned int voltage_now;
268	unsigned int voltage_max;
269	unsigned int current_now;
270	unsigned int current_max;
271	unsigned int current_limit;
272	unsigned int usb_type;
273};
274
275struct qcom_battmgr_wireless {
276	bool online;
277	unsigned int voltage_now;
278	unsigned int voltage_max;
279	unsigned int current_now;
280	unsigned int current_max;
281};
282
283struct qcom_battmgr {
284	struct device *dev;
285	struct pmic_glink_client *client;
286
287	enum qcom_battmgr_variant variant;
288
289	struct power_supply *ac_psy;
290	struct power_supply *bat_psy;
291	struct power_supply *usb_psy;
292	struct power_supply *wls_psy;
293
294	enum qcom_battmgr_unit unit;
295
296	int error;
297	struct completion ack;
298
299	bool service_up;
300
301	struct qcom_battmgr_info info;
302	struct qcom_battmgr_status status;
303	struct qcom_battmgr_ac ac;
304	struct qcom_battmgr_usb usb;
305	struct qcom_battmgr_wireless wireless;
306
307	struct work_struct enable_work;
308
309	/*
310	 * @lock is used to prevent concurrent power supply requests to the
311	 * firmware, as it then stops responding.
312	 */
313	struct mutex lock;
314};
315
316static int qcom_battmgr_request(struct qcom_battmgr *battmgr, void *data, size_t len)
317{
318	unsigned long left;
319	int ret;
320
321	reinit_completion(&battmgr->ack);
322
323	battmgr->error = 0;
324
325	ret = pmic_glink_send(battmgr->client, data, len);
326	if (ret < 0)
327		return ret;
328
329	left = wait_for_completion_timeout(&battmgr->ack, HZ);
330	if (!left)
331		return -ETIMEDOUT;
332
333	return battmgr->error;
334}
335
336static int qcom_battmgr_request_property(struct qcom_battmgr *battmgr, int opcode,
337					 int property, u32 value)
338{
339	struct qcom_battmgr_property_request request = {
340		.hdr.owner = cpu_to_le32(PMIC_GLINK_OWNER_BATTMGR),
341		.hdr.type = cpu_to_le32(PMIC_GLINK_REQ_RESP),
342		.hdr.opcode = cpu_to_le32(opcode),
343		.battery = cpu_to_le32(0),
344		.property = cpu_to_le32(property),
345		.value = cpu_to_le32(value),
346	};
347
348	return qcom_battmgr_request(battmgr, &request, sizeof(request));
349}
350
351static int qcom_battmgr_update_status(struct qcom_battmgr *battmgr)
352{
353	struct qcom_battmgr_update_request request = {
354		.hdr.owner = cpu_to_le32(PMIC_GLINK_OWNER_BATTMGR),
355		.hdr.type = cpu_to_le32(PMIC_GLINK_REQ_RESP),
356		.hdr.opcode = cpu_to_le32(BATTMGR_BAT_STATUS),
357		.battery_id = cpu_to_le32(0),
358	};
359
360	return qcom_battmgr_request(battmgr, &request, sizeof(request));
361}
362
363static int qcom_battmgr_update_info(struct qcom_battmgr *battmgr)
364{
365	struct qcom_battmgr_update_request request = {
366		.hdr.owner = cpu_to_le32(PMIC_GLINK_OWNER_BATTMGR),
367		.hdr.type = cpu_to_le32(PMIC_GLINK_REQ_RESP),
368		.hdr.opcode = cpu_to_le32(BATTMGR_BAT_INFO),
369		.battery_id = cpu_to_le32(0),
370	};
371
372	return qcom_battmgr_request(battmgr, &request, sizeof(request));
373}
374
375static int qcom_battmgr_update_charge_time(struct qcom_battmgr *battmgr)
376{
377	struct qcom_battmgr_charge_time_request request = {
378		.hdr.owner = cpu_to_le32(PMIC_GLINK_OWNER_BATTMGR),
379		.hdr.type = cpu_to_le32(PMIC_GLINK_REQ_RESP),
380		.hdr.opcode = cpu_to_le32(BATTMGR_BAT_CHARGE_TIME),
381		.battery_id = cpu_to_le32(0),
382		.percent = cpu_to_le32(100),
383	};
384
385	return qcom_battmgr_request(battmgr, &request, sizeof(request));
386}
387
388static int qcom_battmgr_update_discharge_time(struct qcom_battmgr *battmgr)
389{
390	struct qcom_battmgr_discharge_time_request request = {
391		.hdr.owner = cpu_to_le32(PMIC_GLINK_OWNER_BATTMGR),
392		.hdr.type = cpu_to_le32(PMIC_GLINK_REQ_RESP),
393		.hdr.opcode = cpu_to_le32(BATTMGR_BAT_DISCHARGE_TIME),
394		.battery_id = cpu_to_le32(0),
395		.rate = cpu_to_le32(0),
396	};
397
398	return qcom_battmgr_request(battmgr, &request, sizeof(request));
399}
400
401static const u8 sm8350_bat_prop_map[] = {
402	[POWER_SUPPLY_PROP_STATUS] = BATT_STATUS,
403	[POWER_SUPPLY_PROP_HEALTH] = BATT_HEALTH,
404	[POWER_SUPPLY_PROP_PRESENT] = BATT_PRESENT,
405	[POWER_SUPPLY_PROP_CHARGE_TYPE] = BATT_CHG_TYPE,
406	[POWER_SUPPLY_PROP_CAPACITY] = BATT_CAPACITY,
407	[POWER_SUPPLY_PROP_VOLTAGE_OCV] = BATT_VOLT_OCV,
408	[POWER_SUPPLY_PROP_VOLTAGE_NOW] = BATT_VOLT_NOW,
409	[POWER_SUPPLY_PROP_VOLTAGE_MAX] = BATT_VOLT_MAX,
410	[POWER_SUPPLY_PROP_CURRENT_NOW] = BATT_CURR_NOW,
411	[POWER_SUPPLY_PROP_TEMP] = BATT_TEMP,
412	[POWER_SUPPLY_PROP_TECHNOLOGY] = BATT_TECHNOLOGY,
413	[POWER_SUPPLY_PROP_CHARGE_COUNTER] =  BATT_CHG_COUNTER,
414	[POWER_SUPPLY_PROP_CYCLE_COUNT] = BATT_CYCLE_COUNT,
415	[POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN] =  BATT_CHG_FULL_DESIGN,
416	[POWER_SUPPLY_PROP_CHARGE_FULL] = BATT_CHG_FULL,
417	[POWER_SUPPLY_PROP_MODEL_NAME] = BATT_MODEL_NAME,
418	[POWER_SUPPLY_PROP_TIME_TO_FULL_AVG] = BATT_TTF_AVG,
419	[POWER_SUPPLY_PROP_TIME_TO_EMPTY_AVG] = BATT_TTE_AVG,
420	[POWER_SUPPLY_PROP_POWER_NOW] = BATT_POWER_NOW,
421};
422
423static int qcom_battmgr_bat_sm8350_update(struct qcom_battmgr *battmgr,
424					  enum power_supply_property psp)
425{
426	unsigned int prop;
427	int ret;
428
429	if (psp >= ARRAY_SIZE(sm8350_bat_prop_map))
430		return -EINVAL;
431
432	prop = sm8350_bat_prop_map[psp];
433
434	mutex_lock(&battmgr->lock);
435	ret = qcom_battmgr_request_property(battmgr, BATTMGR_BAT_PROPERTY_GET, prop, 0);
436	mutex_unlock(&battmgr->lock);
437
438	return ret;
439}
440
441static int qcom_battmgr_bat_sc8280xp_update(struct qcom_battmgr *battmgr,
442					    enum power_supply_property psp)
443{
444	int ret;
445
446	mutex_lock(&battmgr->lock);
447
448	if (!battmgr->info.valid) {
449		ret = qcom_battmgr_update_info(battmgr);
450		if (ret < 0)
451			goto out_unlock;
452		battmgr->info.valid = true;
453	}
454
455	ret = qcom_battmgr_update_status(battmgr);
456	if (ret < 0)
457		goto out_unlock;
458
459	if (psp == POWER_SUPPLY_PROP_TIME_TO_FULL_AVG) {
460		ret = qcom_battmgr_update_charge_time(battmgr);
461		if (ret < 0) {
462			ret = -ENODATA;
463			goto out_unlock;
464		}
465	}
466
467	if (psp == POWER_SUPPLY_PROP_TIME_TO_EMPTY_AVG) {
468		ret = qcom_battmgr_update_discharge_time(battmgr);
469		if (ret < 0) {
470			ret = -ENODATA;
471			goto out_unlock;
472		}
473	}
474
475out_unlock:
476	mutex_unlock(&battmgr->lock);
477	return ret;
478}
479
480static int qcom_battmgr_bat_get_property(struct power_supply *psy,
481					 enum power_supply_property psp,
482					 union power_supply_propval *val)
483{
484	struct qcom_battmgr *battmgr = power_supply_get_drvdata(psy);
485	enum qcom_battmgr_unit unit = battmgr->unit;
486	int ret;
487
488	if (!battmgr->service_up)
489		return -ENODEV;
490
491	if (battmgr->variant == QCOM_BATTMGR_SC8280XP)
492		ret = qcom_battmgr_bat_sc8280xp_update(battmgr, psp);
493	else
494		ret = qcom_battmgr_bat_sm8350_update(battmgr, psp);
495	if (ret < 0)
496		return ret;
497
498	switch (psp) {
499	case POWER_SUPPLY_PROP_STATUS:
500		val->intval = battmgr->status.status;
501		break;
502	case POWER_SUPPLY_PROP_CHARGE_TYPE:
503		val->intval = battmgr->info.charge_type;
504		break;
505	case POWER_SUPPLY_PROP_HEALTH:
506		val->intval = battmgr->status.health;
507		break;
508	case POWER_SUPPLY_PROP_PRESENT:
509		val->intval = battmgr->info.present;
510		break;
511	case POWER_SUPPLY_PROP_TECHNOLOGY:
512		val->intval = battmgr->info.technology;
513		break;
514	case POWER_SUPPLY_PROP_CYCLE_COUNT:
515		val->intval = battmgr->info.cycle_count;
516		break;
517	case POWER_SUPPLY_PROP_VOLTAGE_MAX_DESIGN:
518		val->intval = battmgr->info.voltage_max_design;
519		break;
520	case POWER_SUPPLY_PROP_VOLTAGE_MAX:
521		val->intval = battmgr->info.voltage_max;
522		break;
523	case POWER_SUPPLY_PROP_VOLTAGE_NOW:
524		val->intval = battmgr->status.voltage_now;
525		break;
526	case POWER_SUPPLY_PROP_VOLTAGE_OCV:
527		val->intval = battmgr->status.voltage_ocv;
528		break;
529	case POWER_SUPPLY_PROP_CURRENT_NOW:
530		val->intval = battmgr->status.current_now;
531		break;
532	case POWER_SUPPLY_PROP_POWER_NOW:
533		val->intval = battmgr->status.power_now;
534		break;
535	case POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN:
536		if (unit != QCOM_BATTMGR_UNIT_mAh)
537			return -ENODATA;
538		val->intval = battmgr->info.design_capacity;
539		break;
540	case POWER_SUPPLY_PROP_CHARGE_FULL:
541		if (unit != QCOM_BATTMGR_UNIT_mAh)
542			return -ENODATA;
543		val->intval = battmgr->info.last_full_capacity;
544		break;
545	case POWER_SUPPLY_PROP_CHARGE_EMPTY:
546		if (unit != QCOM_BATTMGR_UNIT_mAh)
547			return -ENODATA;
548		val->intval = battmgr->info.capacity_low;
549		break;
550	case POWER_SUPPLY_PROP_CHARGE_NOW:
551		if (unit != QCOM_BATTMGR_UNIT_mAh)
552			return -ENODATA;
553		val->intval = battmgr->status.capacity;
554		break;
555	case POWER_SUPPLY_PROP_CHARGE_COUNTER:
556		val->intval = battmgr->info.charge_count;
557		break;
558	case POWER_SUPPLY_PROP_ENERGY_FULL_DESIGN:
559		if (unit != QCOM_BATTMGR_UNIT_mWh)
560			return -ENODATA;
561		val->intval = battmgr->info.design_capacity;
562		break;
563	case POWER_SUPPLY_PROP_ENERGY_FULL:
564		if (unit != QCOM_BATTMGR_UNIT_mWh)
565			return -ENODATA;
566		val->intval = battmgr->info.last_full_capacity;
567		break;
568	case POWER_SUPPLY_PROP_ENERGY_EMPTY:
569		if (unit != QCOM_BATTMGR_UNIT_mWh)
570			return -ENODATA;
571		val->intval = battmgr->info.capacity_low;
572		break;
573	case POWER_SUPPLY_PROP_ENERGY_NOW:
574		if (unit != QCOM_BATTMGR_UNIT_mWh)
575			return -ENODATA;
576		val->intval = battmgr->status.capacity;
577		break;
578	case POWER_SUPPLY_PROP_CAPACITY:
579		val->intval = battmgr->status.percent;
580		break;
581	case POWER_SUPPLY_PROP_TEMP:
582		val->intval = battmgr->status.temperature;
583		break;
584	case POWER_SUPPLY_PROP_TIME_TO_EMPTY_AVG:
585		val->intval = battmgr->status.discharge_time;
586		break;
587	case POWER_SUPPLY_PROP_TIME_TO_FULL_AVG:
588		val->intval = battmgr->status.charge_time;
589		break;
590	case POWER_SUPPLY_PROP_MANUFACTURE_YEAR:
591		val->intval = battmgr->info.year;
592		break;
593	case POWER_SUPPLY_PROP_MANUFACTURE_MONTH:
594		val->intval = battmgr->info.month;
595		break;
596	case POWER_SUPPLY_PROP_MANUFACTURE_DAY:
597		val->intval = battmgr->info.day;
598		break;
599	case POWER_SUPPLY_PROP_MODEL_NAME:
600		val->strval = battmgr->info.model_number;
601		break;
602	case POWER_SUPPLY_PROP_MANUFACTURER:
603		val->strval = battmgr->info.oem_info;
604		break;
605	case POWER_SUPPLY_PROP_SERIAL_NUMBER:
606		val->strval = battmgr->info.serial_number;
607		break;
608	default:
609		return -EINVAL;
610	}
611
612	return 0;
613}
614
615static const enum power_supply_property sc8280xp_bat_props[] = {
616	POWER_SUPPLY_PROP_STATUS,
617	POWER_SUPPLY_PROP_PRESENT,
618	POWER_SUPPLY_PROP_TECHNOLOGY,
619	POWER_SUPPLY_PROP_CYCLE_COUNT,
620	POWER_SUPPLY_PROP_VOLTAGE_MAX_DESIGN,
621	POWER_SUPPLY_PROP_VOLTAGE_NOW,
622	POWER_SUPPLY_PROP_POWER_NOW,
623	POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN,
624	POWER_SUPPLY_PROP_CHARGE_FULL,
625	POWER_SUPPLY_PROP_CHARGE_EMPTY,
626	POWER_SUPPLY_PROP_CHARGE_NOW,
627	POWER_SUPPLY_PROP_ENERGY_FULL_DESIGN,
628	POWER_SUPPLY_PROP_ENERGY_FULL,
629	POWER_SUPPLY_PROP_ENERGY_EMPTY,
630	POWER_SUPPLY_PROP_ENERGY_NOW,
631	POWER_SUPPLY_PROP_TEMP,
632	POWER_SUPPLY_PROP_MANUFACTURE_YEAR,
633	POWER_SUPPLY_PROP_MANUFACTURE_MONTH,
634	POWER_SUPPLY_PROP_MANUFACTURE_DAY,
635	POWER_SUPPLY_PROP_MODEL_NAME,
636	POWER_SUPPLY_PROP_MANUFACTURER,
637	POWER_SUPPLY_PROP_SERIAL_NUMBER,
638};
639
640static const struct power_supply_desc sc8280xp_bat_psy_desc = {
641	.name = "qcom-battmgr-bat",
642	.type = POWER_SUPPLY_TYPE_BATTERY,
643	.properties = sc8280xp_bat_props,
644	.num_properties = ARRAY_SIZE(sc8280xp_bat_props),
645	.get_property = qcom_battmgr_bat_get_property,
646};
647
648static const enum power_supply_property sm8350_bat_props[] = {
649	POWER_SUPPLY_PROP_STATUS,
650	POWER_SUPPLY_PROP_HEALTH,
651	POWER_SUPPLY_PROP_PRESENT,
652	POWER_SUPPLY_PROP_CHARGE_TYPE,
653	POWER_SUPPLY_PROP_CAPACITY,
654	POWER_SUPPLY_PROP_VOLTAGE_OCV,
655	POWER_SUPPLY_PROP_VOLTAGE_NOW,
656	POWER_SUPPLY_PROP_VOLTAGE_MAX,
657	POWER_SUPPLY_PROP_CURRENT_NOW,
658	POWER_SUPPLY_PROP_TEMP,
659	POWER_SUPPLY_PROP_TECHNOLOGY,
660	POWER_SUPPLY_PROP_CHARGE_COUNTER,
661	POWER_SUPPLY_PROP_CYCLE_COUNT,
662	POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN,
663	POWER_SUPPLY_PROP_CHARGE_FULL,
664	POWER_SUPPLY_PROP_MODEL_NAME,
665	POWER_SUPPLY_PROP_TIME_TO_FULL_AVG,
666	POWER_SUPPLY_PROP_TIME_TO_EMPTY_AVG,
667	POWER_SUPPLY_PROP_POWER_NOW,
668};
669
670static const struct power_supply_desc sm8350_bat_psy_desc = {
671	.name = "qcom-battmgr-bat",
672	.type = POWER_SUPPLY_TYPE_BATTERY,
673	.properties = sm8350_bat_props,
674	.num_properties = ARRAY_SIZE(sm8350_bat_props),
675	.get_property = qcom_battmgr_bat_get_property,
676};
677
678static int qcom_battmgr_ac_get_property(struct power_supply *psy,
679					enum power_supply_property psp,
680					union power_supply_propval *val)
681{
682	struct qcom_battmgr *battmgr = power_supply_get_drvdata(psy);
683	int ret;
684
685	if (!battmgr->service_up)
686		return -ENODEV;
687
688	ret = qcom_battmgr_bat_sc8280xp_update(battmgr, psp);
689	if (ret)
690		return ret;
691
692	switch (psp) {
693	case POWER_SUPPLY_PROP_ONLINE:
694		val->intval = battmgr->ac.online;
695		break;
696	default:
697		return -EINVAL;
698	}
699
700	return 0;
701}
702
703static const enum power_supply_property sc8280xp_ac_props[] = {
704	POWER_SUPPLY_PROP_ONLINE,
705};
706
707static const struct power_supply_desc sc8280xp_ac_psy_desc = {
708	.name = "qcom-battmgr-ac",
709	.type = POWER_SUPPLY_TYPE_MAINS,
710	.properties = sc8280xp_ac_props,
711	.num_properties = ARRAY_SIZE(sc8280xp_ac_props),
712	.get_property = qcom_battmgr_ac_get_property,
713};
714
715static const u8 sm8350_usb_prop_map[] = {
716	[POWER_SUPPLY_PROP_ONLINE] = USB_ONLINE,
717	[POWER_SUPPLY_PROP_VOLTAGE_NOW] = USB_VOLT_NOW,
718	[POWER_SUPPLY_PROP_VOLTAGE_MAX] = USB_VOLT_MAX,
719	[POWER_SUPPLY_PROP_CURRENT_NOW] = USB_CURR_NOW,
720	[POWER_SUPPLY_PROP_CURRENT_MAX] = USB_CURR_MAX,
721	[POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT] = USB_INPUT_CURR_LIMIT,
722	[POWER_SUPPLY_PROP_USB_TYPE] = USB_TYPE,
723};
724
725static int qcom_battmgr_usb_sm8350_update(struct qcom_battmgr *battmgr,
726					  enum power_supply_property psp)
727{
728	unsigned int prop;
729	int ret;
730
731	if (psp >= ARRAY_SIZE(sm8350_usb_prop_map))
732		return -EINVAL;
733
734	prop = sm8350_usb_prop_map[psp];
735
736	mutex_lock(&battmgr->lock);
737	ret = qcom_battmgr_request_property(battmgr, BATTMGR_USB_PROPERTY_GET, prop, 0);
738	mutex_unlock(&battmgr->lock);
739
740	return ret;
741}
742
743static int qcom_battmgr_usb_get_property(struct power_supply *psy,
744					 enum power_supply_property psp,
745					 union power_supply_propval *val)
746{
747	struct qcom_battmgr *battmgr = power_supply_get_drvdata(psy);
748	int ret;
749
750	if (!battmgr->service_up)
751		return -ENODEV;
752
753	if (battmgr->variant == QCOM_BATTMGR_SC8280XP)
754		ret = qcom_battmgr_bat_sc8280xp_update(battmgr, psp);
755	else
756		ret = qcom_battmgr_usb_sm8350_update(battmgr, psp);
757	if (ret)
758		return ret;
759
760	switch (psp) {
761	case POWER_SUPPLY_PROP_ONLINE:
762		val->intval = battmgr->usb.online;
763		break;
764	case POWER_SUPPLY_PROP_VOLTAGE_NOW:
765		val->intval = battmgr->usb.voltage_now;
766		break;
767	case POWER_SUPPLY_PROP_VOLTAGE_MAX:
768		val->intval = battmgr->usb.voltage_max;
769		break;
770	case POWER_SUPPLY_PROP_CURRENT_NOW:
771		val->intval = battmgr->usb.current_now;
772		break;
773	case POWER_SUPPLY_PROP_CURRENT_MAX:
774		val->intval = battmgr->usb.current_max;
775		break;
776	case POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT:
777		val->intval = battmgr->usb.current_limit;
778		break;
779	case POWER_SUPPLY_PROP_USB_TYPE:
780		val->intval = battmgr->usb.usb_type;
781		break;
782	default:
783		return -EINVAL;
784	}
785
786	return 0;
787}
788
789static const enum power_supply_usb_type usb_psy_supported_types[] = {
790	POWER_SUPPLY_USB_TYPE_UNKNOWN,
791	POWER_SUPPLY_USB_TYPE_SDP,
792	POWER_SUPPLY_USB_TYPE_DCP,
793	POWER_SUPPLY_USB_TYPE_CDP,
794	POWER_SUPPLY_USB_TYPE_ACA,
795	POWER_SUPPLY_USB_TYPE_C,
796	POWER_SUPPLY_USB_TYPE_PD,
797	POWER_SUPPLY_USB_TYPE_PD_DRP,
798	POWER_SUPPLY_USB_TYPE_PD_PPS,
799	POWER_SUPPLY_USB_TYPE_APPLE_BRICK_ID,
800};
801
802static const enum power_supply_property sc8280xp_usb_props[] = {
803	POWER_SUPPLY_PROP_ONLINE,
804};
805
806static const struct power_supply_desc sc8280xp_usb_psy_desc = {
807	.name = "qcom-battmgr-usb",
808	.type = POWER_SUPPLY_TYPE_USB,
809	.properties = sc8280xp_usb_props,
810	.num_properties = ARRAY_SIZE(sc8280xp_usb_props),
811	.get_property = qcom_battmgr_usb_get_property,
812	.usb_types = usb_psy_supported_types,
813	.num_usb_types = ARRAY_SIZE(usb_psy_supported_types),
814};
815
816static const enum power_supply_property sm8350_usb_props[] = {
817	POWER_SUPPLY_PROP_ONLINE,
818	POWER_SUPPLY_PROP_VOLTAGE_NOW,
819	POWER_SUPPLY_PROP_VOLTAGE_MAX,
820	POWER_SUPPLY_PROP_CURRENT_NOW,
821	POWER_SUPPLY_PROP_CURRENT_MAX,
822	POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT,
823	POWER_SUPPLY_PROP_USB_TYPE,
824};
825
826static const struct power_supply_desc sm8350_usb_psy_desc = {
827	.name = "qcom-battmgr-usb",
828	.type = POWER_SUPPLY_TYPE_USB,
829	.properties = sm8350_usb_props,
830	.num_properties = ARRAY_SIZE(sm8350_usb_props),
831	.get_property = qcom_battmgr_usb_get_property,
832	.usb_types = usb_psy_supported_types,
833	.num_usb_types = ARRAY_SIZE(usb_psy_supported_types),
834};
835
836static const u8 sm8350_wls_prop_map[] = {
837	[POWER_SUPPLY_PROP_ONLINE] = WLS_ONLINE,
838	[POWER_SUPPLY_PROP_VOLTAGE_NOW] = WLS_VOLT_NOW,
839	[POWER_SUPPLY_PROP_VOLTAGE_MAX] = WLS_VOLT_MAX,
840	[POWER_SUPPLY_PROP_CURRENT_NOW] = WLS_CURR_NOW,
841	[POWER_SUPPLY_PROP_CURRENT_MAX] = WLS_CURR_MAX,
842};
843
844static int qcom_battmgr_wls_sm8350_update(struct qcom_battmgr *battmgr,
845					  enum power_supply_property psp)
846{
847	unsigned int prop;
848	int ret;
849
850	if (psp >= ARRAY_SIZE(sm8350_wls_prop_map))
851		return -EINVAL;
852
853	prop = sm8350_wls_prop_map[psp];
854
855	mutex_lock(&battmgr->lock);
856	ret = qcom_battmgr_request_property(battmgr, BATTMGR_WLS_PROPERTY_GET, prop, 0);
857	mutex_unlock(&battmgr->lock);
858
859	return ret;
860}
861
862static int qcom_battmgr_wls_get_property(struct power_supply *psy,
863					 enum power_supply_property psp,
864					 union power_supply_propval *val)
865{
866	struct qcom_battmgr *battmgr = power_supply_get_drvdata(psy);
867	int ret;
868
869	if (!battmgr->service_up)
870		return -ENODEV;
871
872	if (battmgr->variant == QCOM_BATTMGR_SC8280XP)
873		ret = qcom_battmgr_bat_sc8280xp_update(battmgr, psp);
874	else
875		ret = qcom_battmgr_wls_sm8350_update(battmgr, psp);
876	if (ret < 0)
877		return ret;
878
879	switch (psp) {
880	case POWER_SUPPLY_PROP_ONLINE:
881		val->intval = battmgr->wireless.online;
882		break;
883	case POWER_SUPPLY_PROP_VOLTAGE_NOW:
884		val->intval = battmgr->wireless.voltage_now;
885		break;
886	case POWER_SUPPLY_PROP_VOLTAGE_MAX:
887		val->intval = battmgr->wireless.voltage_max;
888		break;
889	case POWER_SUPPLY_PROP_CURRENT_NOW:
890		val->intval = battmgr->wireless.current_now;
891		break;
892	case POWER_SUPPLY_PROP_CURRENT_MAX:
893		val->intval = battmgr->wireless.current_max;
894		break;
895	default:
896		return -EINVAL;
897	}
898
899	return 0;
900}
901
902static const enum power_supply_property sc8280xp_wls_props[] = {
903	POWER_SUPPLY_PROP_ONLINE,
904};
905
906static const struct power_supply_desc sc8280xp_wls_psy_desc = {
907	.name = "qcom-battmgr-wls",
908	.type = POWER_SUPPLY_TYPE_WIRELESS,
909	.properties = sc8280xp_wls_props,
910	.num_properties = ARRAY_SIZE(sc8280xp_wls_props),
911	.get_property = qcom_battmgr_wls_get_property,
912};
913
914static const enum power_supply_property sm8350_wls_props[] = {
915	POWER_SUPPLY_PROP_ONLINE,
916	POWER_SUPPLY_PROP_VOLTAGE_NOW,
917	POWER_SUPPLY_PROP_VOLTAGE_MAX,
918	POWER_SUPPLY_PROP_CURRENT_NOW,
919	POWER_SUPPLY_PROP_CURRENT_MAX,
920};
921
922static const struct power_supply_desc sm8350_wls_psy_desc = {
923	.name = "qcom-battmgr-wls",
924	.type = POWER_SUPPLY_TYPE_WIRELESS,
925	.properties = sm8350_wls_props,
926	.num_properties = ARRAY_SIZE(sm8350_wls_props),
927	.get_property = qcom_battmgr_wls_get_property,
928};
929
930static void qcom_battmgr_notification(struct qcom_battmgr *battmgr,
931				      const struct qcom_battmgr_message *msg,
932				      int len)
933{
934	size_t payload_len = len - sizeof(struct pmic_glink_hdr);
935	unsigned int notification;
936
937	if (payload_len != sizeof(msg->notification)) {
938		dev_warn(battmgr->dev, "ignoring notification with invalid length\n");
939		return;
940	}
941
942	notification = le32_to_cpu(msg->notification);
943	switch (notification) {
944	case NOTIF_BAT_INFO:
945		battmgr->info.valid = false;
946		fallthrough;
947	case NOTIF_BAT_STATUS:
948	case NOTIF_BAT_PROPERTY:
949		power_supply_changed(battmgr->bat_psy);
950		break;
951	case NOTIF_USB_PROPERTY:
952		power_supply_changed(battmgr->usb_psy);
953		break;
954	case NOTIF_WLS_PROPERTY:
955		power_supply_changed(battmgr->wls_psy);
956		break;
957	default:
958		dev_err(battmgr->dev, "unknown notification: %#x\n", notification);
959		break;
960	}
961}
962
963static void qcom_battmgr_sc8280xp_strcpy(char *dest, const char *src)
964{
965	size_t len = src[0];
966
967	/* Some firmware versions return Pascal-style strings */
968	if (len < BATTMGR_STRING_LEN && len == strnlen(src + 1, BATTMGR_STRING_LEN - 1)) {
969		memcpy(dest, src + 1, len);
970		dest[len] = '\0';
971	} else {
972		memcpy(dest, src, BATTMGR_STRING_LEN);
973	}
974}
975
976static unsigned int qcom_battmgr_sc8280xp_parse_technology(const char *chemistry)
977{
978	if (!strncmp(chemistry, "LIO", BATTMGR_CHEMISTRY_LEN))
979		return POWER_SUPPLY_TECHNOLOGY_LION;
980
981	pr_err("Unknown battery technology '%s'\n", chemistry);
982	return POWER_SUPPLY_TECHNOLOGY_UNKNOWN;
983}
984
985static unsigned int qcom_battmgr_sc8280xp_convert_temp(unsigned int temperature)
986{
987	return DIV_ROUND_CLOSEST(temperature, 10);
988}
989
990static void qcom_battmgr_sc8280xp_callback(struct qcom_battmgr *battmgr,
991					   const struct qcom_battmgr_message *resp,
992					   size_t len)
993{
994	unsigned int opcode = le32_to_cpu(resp->hdr.opcode);
995	unsigned int source;
996	unsigned int state;
997	size_t payload_len = len - sizeof(struct pmic_glink_hdr);
998
999	if (payload_len < sizeof(__le32)) {
1000		dev_warn(battmgr->dev, "invalid payload length for %#x: %zd\n",
1001			 opcode, len);
1002		return;
1003	}
1004
1005	switch (opcode) {
1006	case BATTMGR_REQUEST_NOTIFICATION:
1007		battmgr->error = 0;
1008		break;
1009	case BATTMGR_BAT_INFO:
1010		if (payload_len != sizeof(resp->info)) {
1011			dev_warn(battmgr->dev,
1012				 "invalid payload length for battery information request: %zd\n",
1013				 payload_len);
1014			battmgr->error = -ENODATA;
1015			return;
1016		}
1017
1018		battmgr->unit = le32_to_cpu(resp->info.power_unit);
1019
1020		battmgr->info.present = true;
1021		battmgr->info.design_capacity = le32_to_cpu(resp->info.design_capacity) * 1000;
1022		battmgr->info.last_full_capacity = le32_to_cpu(resp->info.last_full_capacity) * 1000;
1023		battmgr->info.voltage_max_design = le32_to_cpu(resp->info.design_voltage) * 1000;
1024		battmgr->info.capacity_low = le32_to_cpu(resp->info.capacity_low) * 1000;
1025		battmgr->info.cycle_count = le32_to_cpu(resp->info.cycle_count);
1026		qcom_battmgr_sc8280xp_strcpy(battmgr->info.model_number, resp->info.model_number);
1027		qcom_battmgr_sc8280xp_strcpy(battmgr->info.serial_number, resp->info.serial_number);
1028		battmgr->info.technology = qcom_battmgr_sc8280xp_parse_technology(resp->info.battery_chemistry);
1029		qcom_battmgr_sc8280xp_strcpy(battmgr->info.oem_info, resp->info.oem_info);
1030		battmgr->info.day = resp->info.day;
1031		battmgr->info.month = resp->info.month;
1032		battmgr->info.year = le16_to_cpu(resp->info.year);
1033		break;
1034	case BATTMGR_BAT_STATUS:
1035		if (payload_len != sizeof(resp->status)) {
1036			dev_warn(battmgr->dev,
1037				 "invalid payload length for battery status request: %zd\n",
1038				 payload_len);
1039			battmgr->error = -ENODATA;
1040			return;
1041		}
1042
1043		state = le32_to_cpu(resp->status.battery_state);
1044		if (state & BIT(0))
1045			battmgr->status.status = POWER_SUPPLY_STATUS_DISCHARGING;
1046		else if (state & BIT(1))
1047			battmgr->status.status = POWER_SUPPLY_STATUS_CHARGING;
1048		else
1049			battmgr->status.status = POWER_SUPPLY_STATUS_NOT_CHARGING;
1050
1051		battmgr->status.capacity = le32_to_cpu(resp->status.capacity) * 1000;
1052		battmgr->status.power_now = le32_to_cpu(resp->status.rate) * 1000;
1053		battmgr->status.voltage_now = le32_to_cpu(resp->status.battery_voltage) * 1000;
1054		battmgr->status.temperature = qcom_battmgr_sc8280xp_convert_temp(le32_to_cpu(resp->status.temperature));
1055
1056		source = le32_to_cpu(resp->status.charging_source);
1057		battmgr->ac.online = source == BATTMGR_CHARGING_SOURCE_AC;
1058		battmgr->usb.online = source == BATTMGR_CHARGING_SOURCE_USB;
1059		battmgr->wireless.online = source == BATTMGR_CHARGING_SOURCE_WIRELESS;
1060		break;
1061	case BATTMGR_BAT_DISCHARGE_TIME:
1062		battmgr->status.discharge_time = le32_to_cpu(resp->time);
1063		break;
1064	case BATTMGR_BAT_CHARGE_TIME:
1065		battmgr->status.charge_time = le32_to_cpu(resp->time);
1066		break;
1067	default:
1068		dev_warn(battmgr->dev, "unknown message %#x\n", opcode);
1069		break;
1070	}
1071
1072	complete(&battmgr->ack);
1073}
1074
1075static void qcom_battmgr_sm8350_callback(struct qcom_battmgr *battmgr,
1076					 const struct qcom_battmgr_message *resp,
1077					 size_t len)
1078{
1079	unsigned int property;
1080	unsigned int opcode = le32_to_cpu(resp->hdr.opcode);
1081	size_t payload_len = len - sizeof(struct pmic_glink_hdr);
1082	unsigned int val;
1083
1084	if (payload_len < sizeof(__le32)) {
1085		dev_warn(battmgr->dev, "invalid payload length for %#x: %zd\n",
1086			 opcode, len);
1087		return;
1088	}
1089
1090	switch (opcode) {
1091	case BATTMGR_BAT_PROPERTY_GET:
1092		property = le32_to_cpu(resp->intval.property);
1093		if (property == BATT_MODEL_NAME) {
1094			if (payload_len != sizeof(resp->strval)) {
1095				dev_warn(battmgr->dev,
1096					 "invalid payload length for BATT_MODEL_NAME request: %zd\n",
1097					 payload_len);
1098				battmgr->error = -ENODATA;
1099				return;
1100			}
1101		} else {
1102			if (payload_len != sizeof(resp->intval)) {
1103				dev_warn(battmgr->dev,
1104					 "invalid payload length for %#x request: %zd\n",
1105					 property, payload_len);
1106				battmgr->error = -ENODATA;
1107				return;
1108			}
1109
1110			battmgr->error = le32_to_cpu(resp->intval.result);
1111			if (battmgr->error)
1112				goto out_complete;
1113		}
1114
1115		switch (property) {
1116		case BATT_STATUS:
1117			battmgr->status.status = le32_to_cpu(resp->intval.value);
1118			break;
1119		case BATT_HEALTH:
1120			battmgr->status.health = le32_to_cpu(resp->intval.value);
1121			break;
1122		case BATT_PRESENT:
1123			battmgr->info.present = le32_to_cpu(resp->intval.value);
1124			break;
1125		case BATT_CHG_TYPE:
1126			battmgr->info.charge_type = le32_to_cpu(resp->intval.value);
1127			break;
1128		case BATT_CAPACITY:
1129			battmgr->status.percent = le32_to_cpu(resp->intval.value) / 100;
1130			break;
1131		case BATT_VOLT_OCV:
1132			battmgr->status.voltage_ocv = le32_to_cpu(resp->intval.value);
1133			break;
1134		case BATT_VOLT_NOW:
1135			battmgr->status.voltage_now = le32_to_cpu(resp->intval.value);
1136			break;
1137		case BATT_VOLT_MAX:
1138			battmgr->info.voltage_max = le32_to_cpu(resp->intval.value);
1139			break;
1140		case BATT_CURR_NOW:
1141			battmgr->status.current_now = le32_to_cpu(resp->intval.value);
1142			break;
1143		case BATT_TEMP:
1144			val = le32_to_cpu(resp->intval.value);
1145			battmgr->status.temperature = DIV_ROUND_CLOSEST(val, 10);
1146			break;
1147		case BATT_TECHNOLOGY:
1148			battmgr->info.technology = le32_to_cpu(resp->intval.value);
1149			break;
1150		case BATT_CHG_COUNTER:
1151			battmgr->info.charge_count = le32_to_cpu(resp->intval.value);
1152			break;
1153		case BATT_CYCLE_COUNT:
1154			battmgr->info.cycle_count = le32_to_cpu(resp->intval.value);
1155			break;
1156		case BATT_CHG_FULL_DESIGN:
1157			battmgr->info.design_capacity = le32_to_cpu(resp->intval.value);
1158			break;
1159		case BATT_CHG_FULL:
1160			battmgr->info.last_full_capacity = le32_to_cpu(resp->intval.value);
1161			break;
1162		case BATT_MODEL_NAME:
1163			strscpy(battmgr->info.model_number, resp->strval.model, BATTMGR_STRING_LEN);
1164			break;
1165		case BATT_TTF_AVG:
1166			battmgr->status.charge_time = le32_to_cpu(resp->intval.value);
1167			break;
1168		case BATT_TTE_AVG:
1169			battmgr->status.discharge_time = le32_to_cpu(resp->intval.value);
1170			break;
1171		case BATT_POWER_NOW:
1172			battmgr->status.power_now = le32_to_cpu(resp->intval.value);
1173			break;
1174		default:
1175			dev_warn(battmgr->dev, "unknown property %#x\n", property);
1176			break;
1177		}
1178		break;
1179	case BATTMGR_USB_PROPERTY_GET:
1180		property = le32_to_cpu(resp->intval.property);
1181		if (payload_len != sizeof(resp->intval)) {
1182			dev_warn(battmgr->dev,
1183				 "invalid payload length for %#x request: %zd\n",
1184				 property, payload_len);
1185			battmgr->error = -ENODATA;
1186			return;
1187		}
1188
1189		battmgr->error = le32_to_cpu(resp->intval.result);
1190		if (battmgr->error)
1191			goto out_complete;
1192
1193		switch (property) {
1194		case USB_ONLINE:
1195			battmgr->usb.online = le32_to_cpu(resp->intval.value);
1196			break;
1197		case USB_VOLT_NOW:
1198			battmgr->usb.voltage_now = le32_to_cpu(resp->intval.value);
1199			break;
1200		case USB_VOLT_MAX:
1201			battmgr->usb.voltage_max = le32_to_cpu(resp->intval.value);
1202			break;
1203		case USB_CURR_NOW:
1204			battmgr->usb.current_now = le32_to_cpu(resp->intval.value);
1205			break;
1206		case USB_CURR_MAX:
1207			battmgr->usb.current_max = le32_to_cpu(resp->intval.value);
1208			break;
1209		case USB_INPUT_CURR_LIMIT:
1210			battmgr->usb.current_limit = le32_to_cpu(resp->intval.value);
1211			break;
1212		case USB_TYPE:
1213			battmgr->usb.usb_type = le32_to_cpu(resp->intval.value);
1214			break;
1215		default:
1216			dev_warn(battmgr->dev, "unknown property %#x\n", property);
1217			break;
1218		}
1219		break;
1220	case BATTMGR_WLS_PROPERTY_GET:
1221		property = le32_to_cpu(resp->intval.property);
1222		if (payload_len != sizeof(resp->intval)) {
1223			dev_warn(battmgr->dev,
1224				 "invalid payload length for %#x request: %zd\n",
1225				 property, payload_len);
1226			battmgr->error = -ENODATA;
1227			return;
1228		}
1229
1230		battmgr->error = le32_to_cpu(resp->intval.result);
1231		if (battmgr->error)
1232			goto out_complete;
1233
1234		switch (property) {
1235		case WLS_ONLINE:
1236			battmgr->wireless.online = le32_to_cpu(resp->intval.value);
1237			break;
1238		case WLS_VOLT_NOW:
1239			battmgr->wireless.voltage_now = le32_to_cpu(resp->intval.value);
1240			break;
1241		case WLS_VOLT_MAX:
1242			battmgr->wireless.voltage_max = le32_to_cpu(resp->intval.value);
1243			break;
1244		case WLS_CURR_NOW:
1245			battmgr->wireless.current_now = le32_to_cpu(resp->intval.value);
1246			break;
1247		case WLS_CURR_MAX:
1248			battmgr->wireless.current_max = le32_to_cpu(resp->intval.value);
1249			break;
1250		default:
1251			dev_warn(battmgr->dev, "unknown property %#x\n", property);
1252			break;
1253		}
1254		break;
1255	case BATTMGR_REQUEST_NOTIFICATION:
1256		battmgr->error = 0;
1257		break;
1258	default:
1259		dev_warn(battmgr->dev, "unknown message %#x\n", opcode);
1260		break;
1261	}
1262
1263out_complete:
1264	complete(&battmgr->ack);
1265}
1266
1267static void qcom_battmgr_callback(const void *data, size_t len, void *priv)
1268{
1269	const struct pmic_glink_hdr *hdr = data;
1270	struct qcom_battmgr *battmgr = priv;
1271	unsigned int opcode = le32_to_cpu(hdr->opcode);
1272
1273	if (opcode == BATTMGR_NOTIFICATION)
1274		qcom_battmgr_notification(battmgr, data, len);
1275	else if (battmgr->variant == QCOM_BATTMGR_SC8280XP)
1276		qcom_battmgr_sc8280xp_callback(battmgr, data, len);
1277	else
1278		qcom_battmgr_sm8350_callback(battmgr, data, len);
1279}
1280
1281static void qcom_battmgr_enable_worker(struct work_struct *work)
1282{
1283	struct qcom_battmgr *battmgr = container_of(work, struct qcom_battmgr, enable_work);
1284	struct qcom_battmgr_enable_request req = {
1285		.hdr.owner = cpu_to_le32(PMIC_GLINK_OWNER_BATTMGR),
1286		.hdr.type = cpu_to_le32(PMIC_GLINK_NOTIFY),
1287		.hdr.opcode = cpu_to_le32(BATTMGR_REQUEST_NOTIFICATION),
1288	};
1289	int ret;
1290
1291	ret = qcom_battmgr_request(battmgr, &req, sizeof(req));
1292	if (ret)
1293		dev_err(battmgr->dev, "failed to request power notifications\n");
1294}
1295
1296static void qcom_battmgr_pdr_notify(void *priv, int state)
1297{
1298	struct qcom_battmgr *battmgr = priv;
1299
1300	if (state == SERVREG_SERVICE_STATE_UP) {
1301		battmgr->service_up = true;
1302		schedule_work(&battmgr->enable_work);
1303	} else {
1304		battmgr->service_up = false;
1305	}
1306}
1307
1308static const struct of_device_id qcom_battmgr_of_variants[] = {
1309	{ .compatible = "qcom,sc8180x-pmic-glink", .data = (void *)QCOM_BATTMGR_SC8280XP },
1310	{ .compatible = "qcom,sc8280xp-pmic-glink", .data = (void *)QCOM_BATTMGR_SC8280XP },
1311	/* Unmatched devices falls back to QCOM_BATTMGR_SM8350 */
1312	{}
1313};
1314
1315static char *qcom_battmgr_battery[] = { "battery" };
1316
1317static int qcom_battmgr_probe(struct auxiliary_device *adev,
1318			      const struct auxiliary_device_id *id)
1319{
1320	struct power_supply_config psy_cfg_supply = {};
1321	struct power_supply_config psy_cfg = {};
1322	const struct of_device_id *match;
1323	struct qcom_battmgr *battmgr;
1324	struct device *dev = &adev->dev;
1325
1326	battmgr = devm_kzalloc(dev, sizeof(*battmgr), GFP_KERNEL);
1327	if (!battmgr)
1328		return -ENOMEM;
1329
1330	battmgr->dev = dev;
1331
1332	psy_cfg.drv_data = battmgr;
1333	psy_cfg.of_node = adev->dev.of_node;
1334
1335	psy_cfg_supply.drv_data = battmgr;
1336	psy_cfg_supply.of_node = adev->dev.of_node;
1337	psy_cfg_supply.supplied_to = qcom_battmgr_battery;
1338	psy_cfg_supply.num_supplicants = 1;
1339
1340	INIT_WORK(&battmgr->enable_work, qcom_battmgr_enable_worker);
1341	mutex_init(&battmgr->lock);
1342	init_completion(&battmgr->ack);
1343
1344	match = of_match_device(qcom_battmgr_of_variants, dev->parent);
1345	if (match)
1346		battmgr->variant = (unsigned long)match->data;
1347	else
1348		battmgr->variant = QCOM_BATTMGR_SM8350;
1349
1350	if (battmgr->variant == QCOM_BATTMGR_SC8280XP) {
1351		battmgr->bat_psy = devm_power_supply_register(dev, &sc8280xp_bat_psy_desc, &psy_cfg);
1352		if (IS_ERR(battmgr->bat_psy))
1353			return dev_err_probe(dev, PTR_ERR(battmgr->bat_psy),
1354					     "failed to register battery power supply\n");
1355
1356		battmgr->ac_psy = devm_power_supply_register(dev, &sc8280xp_ac_psy_desc, &psy_cfg_supply);
1357		if (IS_ERR(battmgr->ac_psy))
1358			return dev_err_probe(dev, PTR_ERR(battmgr->ac_psy),
1359					     "failed to register AC power supply\n");
1360
1361		battmgr->usb_psy = devm_power_supply_register(dev, &sc8280xp_usb_psy_desc, &psy_cfg_supply);
1362		if (IS_ERR(battmgr->usb_psy))
1363			return dev_err_probe(dev, PTR_ERR(battmgr->usb_psy),
1364					     "failed to register USB power supply\n");
1365
1366		battmgr->wls_psy = devm_power_supply_register(dev, &sc8280xp_wls_psy_desc, &psy_cfg_supply);
1367		if (IS_ERR(battmgr->wls_psy))
1368			return dev_err_probe(dev, PTR_ERR(battmgr->wls_psy),
1369					     "failed to register wireless charing power supply\n");
1370	} else {
1371		battmgr->bat_psy = devm_power_supply_register(dev, &sm8350_bat_psy_desc, &psy_cfg);
1372		if (IS_ERR(battmgr->bat_psy))
1373			return dev_err_probe(dev, PTR_ERR(battmgr->bat_psy),
1374					     "failed to register battery power supply\n");
1375
1376		battmgr->usb_psy = devm_power_supply_register(dev, &sm8350_usb_psy_desc, &psy_cfg_supply);
1377		if (IS_ERR(battmgr->usb_psy))
1378			return dev_err_probe(dev, PTR_ERR(battmgr->usb_psy),
1379					     "failed to register USB power supply\n");
1380
1381		battmgr->wls_psy = devm_power_supply_register(dev, &sm8350_wls_psy_desc, &psy_cfg_supply);
1382		if (IS_ERR(battmgr->wls_psy))
1383			return dev_err_probe(dev, PTR_ERR(battmgr->wls_psy),
1384					     "failed to register wireless charing power supply\n");
1385	}
1386
1387	battmgr->client = devm_pmic_glink_register_client(dev,
1388							  PMIC_GLINK_OWNER_BATTMGR,
1389							  qcom_battmgr_callback,
1390							  qcom_battmgr_pdr_notify,
1391							  battmgr);
1392	return PTR_ERR_OR_ZERO(battmgr->client);
1393}
1394
1395static const struct auxiliary_device_id qcom_battmgr_id_table[] = {
1396	{ .name = "pmic_glink.power-supply", },
1397	{},
1398};
1399MODULE_DEVICE_TABLE(auxiliary, qcom_battmgr_id_table);
1400
1401static struct auxiliary_driver qcom_battmgr_driver = {
1402	.name = "pmic_glink_power_supply",
1403	.probe = qcom_battmgr_probe,
1404	.id_table = qcom_battmgr_id_table,
1405};
1406
1407module_auxiliary_driver(qcom_battmgr_driver);
1408
1409MODULE_DESCRIPTION("Qualcomm PMIC GLINK battery manager driver");
1410MODULE_LICENSE("GPL");
1411