xref: /linux/drivers/hwmon/corsair-psu.c (revision bdd1a21b52557ea8f61d0a5dc2f77151b576eb70)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * corsair-psu.c - Linux driver for Corsair power supplies with HID sensors interface
4  * Copyright (C) 2020 Wilken Gottwalt <wilken.gottwalt@posteo.net>
5  */
6 
7 #include <linux/completion.h>
8 #include <linux/debugfs.h>
9 #include <linux/errno.h>
10 #include <linux/hid.h>
11 #include <linux/hwmon.h>
12 #include <linux/hwmon-sysfs.h>
13 #include <linux/jiffies.h>
14 #include <linux/kernel.h>
15 #include <linux/module.h>
16 #include <linux/mutex.h>
17 #include <linux/slab.h>
18 #include <linux/types.h>
19 
20 /*
21  * Corsair protocol for PSUs
22  *
23  * message size = 64 bytes (request and response, little endian)
24  * request:
25  *	[length][command][param0][param1][paramX]...
26  * reply:
27  *	[echo of length][echo of command][data0][data1][dataX]...
28  *
29  *	- commands are byte sized opcodes
30  *	- length is the sum of all bytes of the commands/params
31  *	- the micro-controller of most of these PSUs support concatenation in the request and reply,
32  *	  but it is better to not rely on this (it is also hard to parse)
33  *	- the driver uses raw events to be accessible from userspace (though this is not really
34  *	  supported, it is just there for convenience, may be removed in the future)
35  *	- a reply always start with the length and command in the same order the request used it
36  *	- length of the reply data is specific to the command used
37  *	- some of the commands work on a rail and can be switched to a specific rail (0 = 12v,
38  *	  1 = 5v, 2 = 3.3v)
39  *	- the format of the init command 0xFE is swapped length/command bytes
40  *	- parameter bytes amount and values are specific to the command (rail setting is the only
41  *	  for now that uses non-zero values)
42  *	- there are much more commands, especially for configuring the device, but they are not
43  *	  supported because a wrong command/length can lockup the micro-controller
44  *	- the driver supports debugfs for values not fitting into the hwmon class
45  *	- not every device class (HXi, RMi or AXi) supports all commands
46  *	- it is a pure sensors reading driver (will not support configuring)
47  */
48 
49 #define DRIVER_NAME		"corsair-psu"
50 
51 #define REPLY_SIZE		16 /* max length of a reply to a single command */
52 #define CMD_BUFFER_SIZE		64
53 #define CMD_TIMEOUT_MS		250
54 #define SECONDS_PER_HOUR	(60 * 60)
55 #define SECONDS_PER_DAY		(SECONDS_PER_HOUR * 24)
56 #define RAIL_COUNT		3 /* 3v3 + 5v + 12v */
57 #define TEMP_COUNT		2
58 
59 #define PSU_CMD_SELECT_RAIL	0x00 /* expects length 2 */
60 #define PSU_CMD_RAIL_VOLTS_HCRIT 0x40 /* the rest of the commands expect length 3 */
61 #define PSU_CMD_RAIL_VOLTS_LCRIT 0x44
62 #define PSU_CMD_RAIL_AMPS_HCRIT	0x46
63 #define PSU_CMD_TEMP_HCRIT	0x4F
64 #define PSU_CMD_IN_VOLTS	0x88
65 #define PSU_CMD_IN_AMPS		0x89
66 #define PSU_CMD_RAIL_VOLTS	0x8B
67 #define PSU_CMD_RAIL_AMPS	0x8C
68 #define PSU_CMD_TEMP0		0x8D
69 #define PSU_CMD_TEMP1		0x8E
70 #define PSU_CMD_FAN		0x90
71 #define PSU_CMD_RAIL_WATTS	0x96
72 #define PSU_CMD_VEND_STR	0x99
73 #define PSU_CMD_PROD_STR	0x9A
74 #define PSU_CMD_TOTAL_WATTS	0xEE
75 #define PSU_CMD_TOTAL_UPTIME	0xD1
76 #define PSU_CMD_UPTIME		0xD2
77 #define PSU_CMD_INIT		0xFE
78 
79 #define L_IN_VOLTS		"v_in"
80 #define L_OUT_VOLTS_12V		"v_out +12v"
81 #define L_OUT_VOLTS_5V		"v_out +5v"
82 #define L_OUT_VOLTS_3_3V	"v_out +3.3v"
83 #define L_IN_AMPS		"curr in"
84 #define L_AMPS_12V		"curr +12v"
85 #define L_AMPS_5V		"curr +5v"
86 #define L_AMPS_3_3V		"curr +3.3v"
87 #define L_FAN			"psu fan"
88 #define L_TEMP0			"vrm temp"
89 #define L_TEMP1			"case temp"
90 #define L_WATTS			"power total"
91 #define L_WATTS_12V		"power +12v"
92 #define L_WATTS_5V		"power +5v"
93 #define L_WATTS_3_3V		"power +3.3v"
94 
95 static const char *const label_watts[] = {
96 	L_WATTS,
97 	L_WATTS_12V,
98 	L_WATTS_5V,
99 	L_WATTS_3_3V
100 };
101 
102 static const char *const label_volts[] = {
103 	L_IN_VOLTS,
104 	L_OUT_VOLTS_12V,
105 	L_OUT_VOLTS_5V,
106 	L_OUT_VOLTS_3_3V
107 };
108 
109 static const char *const label_amps[] = {
110 	L_IN_AMPS,
111 	L_AMPS_12V,
112 	L_AMPS_5V,
113 	L_AMPS_3_3V
114 };
115 
116 struct corsairpsu_data {
117 	struct hid_device *hdev;
118 	struct device *hwmon_dev;
119 	struct dentry *debugfs;
120 	struct completion wait_completion;
121 	struct mutex lock; /* for locking access to cmd_buffer */
122 	u8 *cmd_buffer;
123 	char vendor[REPLY_SIZE];
124 	char product[REPLY_SIZE];
125 	long temp_crit[TEMP_COUNT];
126 	long in_crit[RAIL_COUNT];
127 	long in_lcrit[RAIL_COUNT];
128 	long curr_crit[RAIL_COUNT];
129 	u8 temp_crit_support;
130 	u8 in_crit_support;
131 	u8 in_lcrit_support;
132 	u8 curr_crit_support;
133 	bool in_curr_cmd_support; /* not all commands are supported on every PSU */
134 };
135 
136 /* some values are SMBus LINEAR11 data which need a conversion */
137 static int corsairpsu_linear11_to_int(const u16 val, const int scale)
138 {
139 	const int exp = ((s16)val) >> 11;
140 	const int mant = (((s16)(val & 0x7ff)) << 5) >> 5;
141 	const int result = mant * scale;
142 
143 	return (exp >= 0) ? (result << exp) : (result >> -exp);
144 }
145 
146 static int corsairpsu_usb_cmd(struct corsairpsu_data *priv, u8 p0, u8 p1, u8 p2, void *data)
147 {
148 	unsigned long time;
149 	int ret;
150 
151 	memset(priv->cmd_buffer, 0, CMD_BUFFER_SIZE);
152 	priv->cmd_buffer[0] = p0;
153 	priv->cmd_buffer[1] = p1;
154 	priv->cmd_buffer[2] = p2;
155 
156 	reinit_completion(&priv->wait_completion);
157 
158 	ret = hid_hw_output_report(priv->hdev, priv->cmd_buffer, CMD_BUFFER_SIZE);
159 	if (ret < 0)
160 		return ret;
161 
162 	time = wait_for_completion_timeout(&priv->wait_completion,
163 					   msecs_to_jiffies(CMD_TIMEOUT_MS));
164 	if (!time)
165 		return -ETIMEDOUT;
166 
167 	/*
168 	 * at the start of the reply is an echo of the send command/length in the same order it
169 	 * was send, not every command is supported on every device class, if a command is not
170 	 * supported, the length value in the reply is okay, but the command value is set to 0
171 	 */
172 	if (p0 != priv->cmd_buffer[0] || p1 != priv->cmd_buffer[1])
173 		return -EOPNOTSUPP;
174 
175 	if (data)
176 		memcpy(data, priv->cmd_buffer + 2, REPLY_SIZE);
177 
178 	return 0;
179 }
180 
181 static int corsairpsu_init(struct corsairpsu_data *priv)
182 {
183 	/*
184 	 * PSU_CMD_INIT uses swapped length/command and expects 2 parameter bytes, this command
185 	 * actually generates a reply, but we don't need it
186 	 */
187 	return corsairpsu_usb_cmd(priv, PSU_CMD_INIT, 3, 0, NULL);
188 }
189 
190 static int corsairpsu_fwinfo(struct corsairpsu_data *priv)
191 {
192 	int ret;
193 
194 	ret = corsairpsu_usb_cmd(priv, 3, PSU_CMD_VEND_STR, 0, priv->vendor);
195 	if (ret < 0)
196 		return ret;
197 
198 	ret = corsairpsu_usb_cmd(priv, 3, PSU_CMD_PROD_STR, 0, priv->product);
199 	if (ret < 0)
200 		return ret;
201 
202 	return 0;
203 }
204 
205 static int corsairpsu_request(struct corsairpsu_data *priv, u8 cmd, u8 rail, void *data)
206 {
207 	int ret;
208 
209 	mutex_lock(&priv->lock);
210 	switch (cmd) {
211 	case PSU_CMD_RAIL_VOLTS_HCRIT:
212 	case PSU_CMD_RAIL_VOLTS_LCRIT:
213 	case PSU_CMD_RAIL_AMPS_HCRIT:
214 	case PSU_CMD_RAIL_VOLTS:
215 	case PSU_CMD_RAIL_AMPS:
216 	case PSU_CMD_RAIL_WATTS:
217 		ret = corsairpsu_usb_cmd(priv, 2, PSU_CMD_SELECT_RAIL, rail, NULL);
218 		if (ret < 0)
219 			goto cmd_fail;
220 		break;
221 	default:
222 		break;
223 	}
224 
225 	ret = corsairpsu_usb_cmd(priv, 3, cmd, 0, data);
226 
227 cmd_fail:
228 	mutex_unlock(&priv->lock);
229 	return ret;
230 }
231 
232 static int corsairpsu_get_value(struct corsairpsu_data *priv, u8 cmd, u8 rail, long *val)
233 {
234 	u8 data[REPLY_SIZE];
235 	long tmp;
236 	int ret;
237 
238 	ret = corsairpsu_request(priv, cmd, rail, data);
239 	if (ret < 0)
240 		return ret;
241 
242 	/*
243 	 * the biggest value here comes from the uptime command and to exceed MAXINT total uptime
244 	 * needs to be about 68 years, the rest are u16 values and the biggest value coming out of
245 	 * the LINEAR11 conversion are the watts values which are about 1200 for the strongest psu
246 	 * supported (HX1200i)
247 	 */
248 	tmp = ((long)data[3] << 24) + (data[2] << 16) + (data[1] << 8) + data[0];
249 	switch (cmd) {
250 	case PSU_CMD_RAIL_VOLTS_HCRIT:
251 	case PSU_CMD_RAIL_VOLTS_LCRIT:
252 	case PSU_CMD_RAIL_AMPS_HCRIT:
253 	case PSU_CMD_TEMP_HCRIT:
254 	case PSU_CMD_IN_VOLTS:
255 	case PSU_CMD_IN_AMPS:
256 	case PSU_CMD_RAIL_VOLTS:
257 	case PSU_CMD_RAIL_AMPS:
258 	case PSU_CMD_TEMP0:
259 	case PSU_CMD_TEMP1:
260 		*val = corsairpsu_linear11_to_int(tmp & 0xFFFF, 1000);
261 		break;
262 	case PSU_CMD_FAN:
263 		*val = corsairpsu_linear11_to_int(tmp & 0xFFFF, 1);
264 		break;
265 	case PSU_CMD_RAIL_WATTS:
266 	case PSU_CMD_TOTAL_WATTS:
267 		*val = corsairpsu_linear11_to_int(tmp & 0xFFFF, 1000000);
268 		break;
269 	case PSU_CMD_TOTAL_UPTIME:
270 	case PSU_CMD_UPTIME:
271 		*val = tmp;
272 		break;
273 	default:
274 		ret = -EOPNOTSUPP;
275 		break;
276 	}
277 
278 	return ret;
279 }
280 
281 static void corsairpsu_get_criticals(struct corsairpsu_data *priv)
282 {
283 	long tmp;
284 	int rail;
285 
286 	for (rail = 0; rail < TEMP_COUNT; ++rail) {
287 		if (!corsairpsu_get_value(priv, PSU_CMD_TEMP_HCRIT, rail, &tmp)) {
288 			priv->temp_crit_support |= BIT(rail);
289 			priv->temp_crit[rail] = tmp;
290 		}
291 	}
292 
293 	for (rail = 0; rail < RAIL_COUNT; ++rail) {
294 		if (!corsairpsu_get_value(priv, PSU_CMD_RAIL_VOLTS_HCRIT, rail, &tmp)) {
295 			priv->in_crit_support |= BIT(rail);
296 			priv->in_crit[rail] = tmp;
297 		}
298 
299 		if (!corsairpsu_get_value(priv, PSU_CMD_RAIL_VOLTS_LCRIT, rail, &tmp)) {
300 			priv->in_lcrit_support |= BIT(rail);
301 			priv->in_lcrit[rail] = tmp;
302 		}
303 
304 		if (!corsairpsu_get_value(priv, PSU_CMD_RAIL_AMPS_HCRIT, rail, &tmp)) {
305 			priv->curr_crit_support |= BIT(rail);
306 			priv->curr_crit[rail] = tmp;
307 		}
308 	}
309 }
310 
311 static void corsairpsu_check_cmd_support(struct corsairpsu_data *priv)
312 {
313 	long tmp;
314 
315 	priv->in_curr_cmd_support = !corsairpsu_get_value(priv, PSU_CMD_IN_AMPS, 0, &tmp);
316 }
317 
318 static umode_t corsairpsu_hwmon_temp_is_visible(const struct corsairpsu_data *priv, u32 attr,
319 						int channel)
320 {
321 	umode_t res = 0444;
322 
323 	switch (attr) {
324 	case hwmon_temp_input:
325 	case hwmon_temp_label:
326 	case hwmon_temp_crit:
327 		if (channel > 0 && !(priv->temp_crit_support & BIT(channel - 1)))
328 			res = 0;
329 		break;
330 	default:
331 		break;
332 	}
333 
334 	return res;
335 }
336 
337 static umode_t corsairpsu_hwmon_fan_is_visible(const struct corsairpsu_data *priv, u32 attr,
338 					       int channel)
339 {
340 	switch (attr) {
341 	case hwmon_fan_input:
342 	case hwmon_fan_label:
343 		return 0444;
344 	default:
345 		return 0;
346 	}
347 }
348 
349 static umode_t corsairpsu_hwmon_power_is_visible(const struct corsairpsu_data *priv, u32 attr,
350 						 int channel)
351 {
352 	switch (attr) {
353 	case hwmon_power_input:
354 	case hwmon_power_label:
355 		return 0444;
356 	default:
357 		return 0;
358 	}
359 }
360 
361 static umode_t corsairpsu_hwmon_in_is_visible(const struct corsairpsu_data *priv, u32 attr,
362 					      int channel)
363 {
364 	umode_t res = 0444;
365 
366 	switch (attr) {
367 	case hwmon_in_input:
368 	case hwmon_in_label:
369 	case hwmon_in_crit:
370 		if (channel > 0 && !(priv->in_crit_support & BIT(channel - 1)))
371 			res = 0;
372 		break;
373 	case hwmon_in_lcrit:
374 		if (channel > 0 && !(priv->in_lcrit_support & BIT(channel - 1)))
375 			res = 0;
376 		break;
377 	default:
378 		break;
379 	}
380 
381 	return res;
382 }
383 
384 static umode_t corsairpsu_hwmon_curr_is_visible(const struct corsairpsu_data *priv, u32 attr,
385 						int channel)
386 {
387 	umode_t res = 0444;
388 
389 	switch (attr) {
390 	case hwmon_curr_input:
391 		if (channel == 0 && !priv->in_curr_cmd_support)
392 			res = 0;
393 		break;
394 	case hwmon_curr_label:
395 	case hwmon_curr_crit:
396 		if (channel > 0 && !(priv->curr_crit_support & BIT(channel - 1)))
397 			res = 0;
398 		break;
399 	default:
400 		break;
401 	}
402 
403 	return res;
404 }
405 
406 static umode_t corsairpsu_hwmon_ops_is_visible(const void *data, enum hwmon_sensor_types type,
407 					       u32 attr, int channel)
408 {
409 	const struct corsairpsu_data *priv = data;
410 
411 	switch (type) {
412 	case hwmon_temp:
413 		return corsairpsu_hwmon_temp_is_visible(priv, attr, channel);
414 	case hwmon_fan:
415 		return corsairpsu_hwmon_fan_is_visible(priv, attr, channel);
416 	case hwmon_power:
417 		return corsairpsu_hwmon_power_is_visible(priv, attr, channel);
418 	case hwmon_in:
419 		return corsairpsu_hwmon_in_is_visible(priv, attr, channel);
420 	case hwmon_curr:
421 		return corsairpsu_hwmon_curr_is_visible(priv, attr, channel);
422 	default:
423 		return 0;
424 	}
425 }
426 
427 static int corsairpsu_hwmon_temp_read(struct corsairpsu_data *priv, u32 attr, int channel,
428 				      long *val)
429 {
430 	int err = -EOPNOTSUPP;
431 
432 	switch (attr) {
433 	case hwmon_temp_input:
434 		return corsairpsu_get_value(priv, channel ? PSU_CMD_TEMP1 : PSU_CMD_TEMP0,
435 					    channel, val);
436 	case hwmon_temp_crit:
437 		*val = priv->temp_crit[channel];
438 		err = 0;
439 		break;
440 	default:
441 		break;
442 	}
443 
444 	return err;
445 }
446 
447 static int corsairpsu_hwmon_power_read(struct corsairpsu_data *priv, u32 attr, int channel,
448 				       long *val)
449 {
450 	if (attr == hwmon_power_input) {
451 		switch (channel) {
452 		case 0:
453 			return corsairpsu_get_value(priv, PSU_CMD_TOTAL_WATTS, 0, val);
454 		case 1 ... 3:
455 			return corsairpsu_get_value(priv, PSU_CMD_RAIL_WATTS, channel - 1, val);
456 		default:
457 			break;
458 		}
459 	}
460 
461 	return -EOPNOTSUPP;
462 }
463 
464 static int corsairpsu_hwmon_in_read(struct corsairpsu_data *priv, u32 attr, int channel, long *val)
465 {
466 	int err = -EOPNOTSUPP;
467 
468 	switch (attr) {
469 	case hwmon_in_input:
470 		switch (channel) {
471 		case 0:
472 			return corsairpsu_get_value(priv, PSU_CMD_IN_VOLTS, 0, val);
473 		case 1 ... 3:
474 			return corsairpsu_get_value(priv, PSU_CMD_RAIL_VOLTS, channel - 1, val);
475 		default:
476 			break;
477 		}
478 		break;
479 	case hwmon_in_crit:
480 		*val = priv->in_crit[channel - 1];
481 		err = 0;
482 		break;
483 	case hwmon_in_lcrit:
484 		*val = priv->in_lcrit[channel - 1];
485 		err = 0;
486 		break;
487 	}
488 
489 	return err;
490 }
491 
492 static int corsairpsu_hwmon_curr_read(struct corsairpsu_data *priv, u32 attr, int channel,
493 				      long *val)
494 {
495 	int err = -EOPNOTSUPP;
496 
497 	switch (attr) {
498 	case hwmon_curr_input:
499 		switch (channel) {
500 		case 0:
501 			return corsairpsu_get_value(priv, PSU_CMD_IN_AMPS, 0, val);
502 		case 1 ... 3:
503 			return corsairpsu_get_value(priv, PSU_CMD_RAIL_AMPS, channel - 1, val);
504 		default:
505 			break;
506 		}
507 		break;
508 	case hwmon_curr_crit:
509 		*val = priv->curr_crit[channel - 1];
510 		err = 0;
511 		break;
512 	default:
513 		break;
514 	}
515 
516 	return err;
517 }
518 
519 static int corsairpsu_hwmon_ops_read(struct device *dev, enum hwmon_sensor_types type, u32 attr,
520 				     int channel, long *val)
521 {
522 	struct corsairpsu_data *priv = dev_get_drvdata(dev);
523 
524 	switch (type) {
525 	case hwmon_temp:
526 		return corsairpsu_hwmon_temp_read(priv, attr, channel, val);
527 	case hwmon_fan:
528 		if (attr == hwmon_fan_input)
529 			return corsairpsu_get_value(priv, PSU_CMD_FAN, 0, val);
530 		return -EOPNOTSUPP;
531 	case hwmon_power:
532 		return corsairpsu_hwmon_power_read(priv, attr, channel, val);
533 	case hwmon_in:
534 		return corsairpsu_hwmon_in_read(priv, attr, channel, val);
535 	case hwmon_curr:
536 		return corsairpsu_hwmon_curr_read(priv, attr, channel, val);
537 	default:
538 		return -EOPNOTSUPP;
539 	}
540 }
541 
542 static int corsairpsu_hwmon_ops_read_string(struct device *dev, enum hwmon_sensor_types type,
543 					    u32 attr, int channel, const char **str)
544 {
545 	if (type == hwmon_temp && attr == hwmon_temp_label) {
546 		*str = channel ? L_TEMP1 : L_TEMP0;
547 		return 0;
548 	} else if (type == hwmon_fan && attr == hwmon_fan_label) {
549 		*str = L_FAN;
550 		return 0;
551 	} else if (type == hwmon_power && attr == hwmon_power_label && channel < 4) {
552 		*str = label_watts[channel];
553 		return 0;
554 	} else if (type == hwmon_in && attr == hwmon_in_label && channel < 4) {
555 		*str = label_volts[channel];
556 		return 0;
557 	} else if (type == hwmon_curr && attr == hwmon_curr_label && channel < 4) {
558 		*str = label_amps[channel];
559 		return 0;
560 	}
561 
562 	return -EOPNOTSUPP;
563 }
564 
565 static const struct hwmon_ops corsairpsu_hwmon_ops = {
566 	.is_visible	= corsairpsu_hwmon_ops_is_visible,
567 	.read		= corsairpsu_hwmon_ops_read,
568 	.read_string	= corsairpsu_hwmon_ops_read_string,
569 };
570 
571 static const struct hwmon_channel_info *corsairpsu_info[] = {
572 	HWMON_CHANNEL_INFO(chip,
573 			   HWMON_C_REGISTER_TZ),
574 	HWMON_CHANNEL_INFO(temp,
575 			   HWMON_T_INPUT | HWMON_T_LABEL | HWMON_T_CRIT,
576 			   HWMON_T_INPUT | HWMON_T_LABEL | HWMON_T_CRIT),
577 	HWMON_CHANNEL_INFO(fan,
578 			   HWMON_F_INPUT | HWMON_F_LABEL),
579 	HWMON_CHANNEL_INFO(power,
580 			   HWMON_P_INPUT | HWMON_P_LABEL,
581 			   HWMON_P_INPUT | HWMON_P_LABEL,
582 			   HWMON_P_INPUT | HWMON_P_LABEL,
583 			   HWMON_P_INPUT | HWMON_P_LABEL),
584 	HWMON_CHANNEL_INFO(in,
585 			   HWMON_I_INPUT | HWMON_I_LABEL,
586 			   HWMON_I_INPUT | HWMON_I_LABEL | HWMON_I_LCRIT | HWMON_I_CRIT,
587 			   HWMON_I_INPUT | HWMON_I_LABEL | HWMON_I_LCRIT | HWMON_I_CRIT,
588 			   HWMON_I_INPUT | HWMON_I_LABEL | HWMON_I_LCRIT | HWMON_I_CRIT),
589 	HWMON_CHANNEL_INFO(curr,
590 			   HWMON_C_INPUT | HWMON_C_LABEL,
591 			   HWMON_C_INPUT | HWMON_C_LABEL | HWMON_C_CRIT,
592 			   HWMON_C_INPUT | HWMON_C_LABEL | HWMON_C_CRIT,
593 			   HWMON_C_INPUT | HWMON_C_LABEL | HWMON_C_CRIT),
594 	NULL
595 };
596 
597 static const struct hwmon_chip_info corsairpsu_chip_info = {
598 	.ops	= &corsairpsu_hwmon_ops,
599 	.info	= corsairpsu_info,
600 };
601 
602 #ifdef CONFIG_DEBUG_FS
603 
604 static void print_uptime(struct seq_file *seqf, u8 cmd)
605 {
606 	struct corsairpsu_data *priv = seqf->private;
607 	long val;
608 	int ret;
609 
610 	ret = corsairpsu_get_value(priv, cmd, 0, &val);
611 	if (ret < 0) {
612 		seq_puts(seqf, "N/A\n");
613 		return;
614 	}
615 
616 	if (val > SECONDS_PER_DAY) {
617 		seq_printf(seqf, "%ld day(s), %02ld:%02ld:%02ld\n", val / SECONDS_PER_DAY,
618 			   val % SECONDS_PER_DAY / SECONDS_PER_HOUR, val % SECONDS_PER_HOUR / 60,
619 			   val % 60);
620 		return;
621 	}
622 
623 	seq_printf(seqf, "%02ld:%02ld:%02ld\n", val % SECONDS_PER_DAY / SECONDS_PER_HOUR,
624 		   val % SECONDS_PER_HOUR / 60, val % 60);
625 }
626 
627 static int uptime_show(struct seq_file *seqf, void *unused)
628 {
629 	print_uptime(seqf, PSU_CMD_UPTIME);
630 
631 	return 0;
632 }
633 DEFINE_SHOW_ATTRIBUTE(uptime);
634 
635 static int uptime_total_show(struct seq_file *seqf, void *unused)
636 {
637 	print_uptime(seqf, PSU_CMD_TOTAL_UPTIME);
638 
639 	return 0;
640 }
641 DEFINE_SHOW_ATTRIBUTE(uptime_total);
642 
643 static int vendor_show(struct seq_file *seqf, void *unused)
644 {
645 	struct corsairpsu_data *priv = seqf->private;
646 
647 	seq_printf(seqf, "%s\n", priv->vendor);
648 
649 	return 0;
650 }
651 DEFINE_SHOW_ATTRIBUTE(vendor);
652 
653 static int product_show(struct seq_file *seqf, void *unused)
654 {
655 	struct corsairpsu_data *priv = seqf->private;
656 
657 	seq_printf(seqf, "%s\n", priv->product);
658 
659 	return 0;
660 }
661 DEFINE_SHOW_ATTRIBUTE(product);
662 
663 static void corsairpsu_debugfs_init(struct corsairpsu_data *priv)
664 {
665 	char name[32];
666 
667 	scnprintf(name, sizeof(name), "%s-%s", DRIVER_NAME, dev_name(&priv->hdev->dev));
668 
669 	priv->debugfs = debugfs_create_dir(name, NULL);
670 	debugfs_create_file("uptime", 0444, priv->debugfs, priv, &uptime_fops);
671 	debugfs_create_file("uptime_total", 0444, priv->debugfs, priv, &uptime_total_fops);
672 	debugfs_create_file("vendor", 0444, priv->debugfs, priv, &vendor_fops);
673 	debugfs_create_file("product", 0444, priv->debugfs, priv, &product_fops);
674 }
675 
676 #else
677 
678 static void corsairpsu_debugfs_init(struct corsairpsu_data *priv)
679 {
680 }
681 
682 #endif
683 
684 static int corsairpsu_probe(struct hid_device *hdev, const struct hid_device_id *id)
685 {
686 	struct corsairpsu_data *priv;
687 	int ret;
688 
689 	priv = devm_kzalloc(&hdev->dev, sizeof(struct corsairpsu_data), GFP_KERNEL);
690 	if (!priv)
691 		return -ENOMEM;
692 
693 	priv->cmd_buffer = devm_kmalloc(&hdev->dev, CMD_BUFFER_SIZE, GFP_KERNEL);
694 	if (!priv->cmd_buffer)
695 		return -ENOMEM;
696 
697 	ret = hid_parse(hdev);
698 	if (ret)
699 		return ret;
700 
701 	ret = hid_hw_start(hdev, HID_CONNECT_HIDRAW);
702 	if (ret)
703 		return ret;
704 
705 	ret = hid_hw_open(hdev);
706 	if (ret)
707 		goto fail_and_stop;
708 
709 	priv->hdev = hdev;
710 	hid_set_drvdata(hdev, priv);
711 	mutex_init(&priv->lock);
712 	init_completion(&priv->wait_completion);
713 
714 	hid_device_io_start(hdev);
715 
716 	ret = corsairpsu_init(priv);
717 	if (ret < 0) {
718 		dev_err(&hdev->dev, "unable to initialize device (%d)\n", ret);
719 		goto fail_and_stop;
720 	}
721 
722 	ret = corsairpsu_fwinfo(priv);
723 	if (ret < 0) {
724 		dev_err(&hdev->dev, "unable to query firmware (%d)\n", ret);
725 		goto fail_and_stop;
726 	}
727 
728 	corsairpsu_get_criticals(priv);
729 	corsairpsu_check_cmd_support(priv);
730 
731 	priv->hwmon_dev = hwmon_device_register_with_info(&hdev->dev, "corsairpsu", priv,
732 							  &corsairpsu_chip_info, 0);
733 
734 	if (IS_ERR(priv->hwmon_dev)) {
735 		ret = PTR_ERR(priv->hwmon_dev);
736 		goto fail_and_close;
737 	}
738 
739 	corsairpsu_debugfs_init(priv);
740 
741 	return 0;
742 
743 fail_and_close:
744 	hid_hw_close(hdev);
745 fail_and_stop:
746 	hid_hw_stop(hdev);
747 	return ret;
748 }
749 
750 static void corsairpsu_remove(struct hid_device *hdev)
751 {
752 	struct corsairpsu_data *priv = hid_get_drvdata(hdev);
753 
754 	debugfs_remove_recursive(priv->debugfs);
755 	hwmon_device_unregister(priv->hwmon_dev);
756 	hid_hw_close(hdev);
757 	hid_hw_stop(hdev);
758 }
759 
760 static int corsairpsu_raw_event(struct hid_device *hdev, struct hid_report *report, u8 *data,
761 				int size)
762 {
763 	struct corsairpsu_data *priv = hid_get_drvdata(hdev);
764 
765 	if (completion_done(&priv->wait_completion))
766 		return 0;
767 
768 	memcpy(priv->cmd_buffer, data, min(CMD_BUFFER_SIZE, size));
769 	complete(&priv->wait_completion);
770 
771 	return 0;
772 }
773 
774 #ifdef CONFIG_PM
775 static int corsairpsu_resume(struct hid_device *hdev)
776 {
777 	struct corsairpsu_data *priv = hid_get_drvdata(hdev);
778 
779 	/* some PSUs turn off the microcontroller during standby, so a reinit is required */
780 	return corsairpsu_init(priv);
781 }
782 #endif
783 
784 static const struct hid_device_id corsairpsu_idtable[] = {
785 	{ HID_USB_DEVICE(0x1b1c, 0x1c03) }, /* Corsair HX550i */
786 	{ HID_USB_DEVICE(0x1b1c, 0x1c04) }, /* Corsair HX650i */
787 	{ HID_USB_DEVICE(0x1b1c, 0x1c05) }, /* Corsair HX750i */
788 	{ HID_USB_DEVICE(0x1b1c, 0x1c06) }, /* Corsair HX850i */
789 	{ HID_USB_DEVICE(0x1b1c, 0x1c07) }, /* Corsair HX1000i */
790 	{ HID_USB_DEVICE(0x1b1c, 0x1c08) }, /* Corsair HX1200i */
791 	{ HID_USB_DEVICE(0x1b1c, 0x1c09) }, /* Corsair RM550i */
792 	{ HID_USB_DEVICE(0x1b1c, 0x1c0a) }, /* Corsair RM650i */
793 	{ HID_USB_DEVICE(0x1b1c, 0x1c0b) }, /* Corsair RM750i */
794 	{ HID_USB_DEVICE(0x1b1c, 0x1c0c) }, /* Corsair RM850i */
795 	{ HID_USB_DEVICE(0x1b1c, 0x1c0d) }, /* Corsair RM1000i */
796 	{ },
797 };
798 MODULE_DEVICE_TABLE(hid, corsairpsu_idtable);
799 
800 static struct hid_driver corsairpsu_driver = {
801 	.name		= DRIVER_NAME,
802 	.id_table	= corsairpsu_idtable,
803 	.probe		= corsairpsu_probe,
804 	.remove		= corsairpsu_remove,
805 	.raw_event	= corsairpsu_raw_event,
806 #ifdef CONFIG_PM
807 	.resume		= corsairpsu_resume,
808 	.reset_resume	= corsairpsu_resume,
809 #endif
810 };
811 module_hid_driver(corsairpsu_driver);
812 
813 MODULE_LICENSE("GPL");
814 MODULE_AUTHOR("Wilken Gottwalt <wilken.gottwalt@posteo.net>");
815 MODULE_DESCRIPTION("Linux driver for Corsair power supplies with HID sensors interface");
816