1// SPDX-License-Identifier: LGPL-2.1+
2// Copyright (C) 2022, Linaro Ltd - Daniel Lezcano <daniel.lezcano@linaro.org>
3#include <errno.h>
4#include <stdio.h>
5#include <stdlib.h>
6#include <unistd.h>
7
8#include <thermal.h>
9#include "thermal_nl.h"
10
11static int handle_thermal_sample(struct nl_msg *n, void *arg)
12{
13	struct nlmsghdr *nlh = nlmsg_hdr(n);
14	struct genlmsghdr *genlhdr = genlmsg_hdr(nlh);
15	struct nlattr *attrs[THERMAL_GENL_ATTR_MAX + 1];
16	struct thermal_handler_param *thp = arg;
17	struct thermal_handler *th = thp->th;
18
19	genlmsg_parse(nlh, 0, attrs, THERMAL_GENL_ATTR_MAX, NULL);
20
21	switch (genlhdr->cmd) {
22
23	case THERMAL_GENL_SAMPLING_TEMP:
24		return th->ops->sampling.tz_temp(
25			nla_get_u32(attrs[THERMAL_GENL_ATTR_TZ_ID]),
26			nla_get_u32(attrs[THERMAL_GENL_ATTR_TZ_TEMP]), arg);
27	default:
28		return THERMAL_ERROR;
29	}
30}
31
32thermal_error_t thermal_sampling_handle(struct thermal_handler *th, void *arg)
33{
34	struct thermal_handler_param thp = { .th = th, .arg = arg };
35
36	if (!th)
37		return THERMAL_ERROR;
38
39	if (nl_cb_set(th->cb_sampling, NL_CB_VALID, NL_CB_CUSTOM,
40		      handle_thermal_sample, &thp))
41		return THERMAL_ERROR;
42
43	return nl_recvmsgs(th->sk_sampling, th->cb_sampling);
44}
45
46int thermal_sampling_fd(struct thermal_handler *th)
47{
48	if (!th)
49		return -1;
50
51	return nl_socket_get_fd(th->sk_sampling);
52}
53
54thermal_error_t thermal_sampling_exit(struct thermal_handler *th)
55{
56	if (nl_unsubscribe_thermal(th->sk_sampling, th->cb_sampling,
57				   THERMAL_GENL_SAMPLING_GROUP_NAME))
58		return THERMAL_ERROR;
59
60	nl_thermal_disconnect(th->sk_sampling, th->cb_sampling);
61
62	return THERMAL_SUCCESS;
63}
64
65thermal_error_t thermal_sampling_init(struct thermal_handler *th)
66{
67	if (nl_thermal_connect(&th->sk_sampling, &th->cb_sampling))
68		return THERMAL_ERROR;
69
70	if (nl_subscribe_thermal(th->sk_sampling, th->cb_sampling,
71				 THERMAL_GENL_SAMPLING_GROUP_NAME))
72		return THERMAL_ERROR;
73
74	return THERMAL_SUCCESS;
75}
76