tools/lib/thermal/sampling.c

   1 // SPDX-License-Identifier: LGPL-2.1+
   2 // Copyright (C) 2022, Linaro Ltd - Daniel Lezcano <[email protected]>
   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
  11 static 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         arg = thp->arg;
  20
  21         genlmsg_parse(nlh, 0, attrs, THERMAL_GENL_ATTR_MAX, NULL);
  22
  23         switch (genlhdr->cmd) {
  24
  25         case THERMAL_GENL_SAMPLING_TEMP:
  26                 return th->ops->sampling.tz_temp(
  27                         nla_get_u32(attrs[THERMAL_GENL_ATTR_TZ_ID]),
  28                         nla_get_u32(attrs[THERMAL_GENL_ATTR_TZ_TEMP]), arg);
  29         default:
  30                 return THERMAL_ERROR;
  31         }
  32 }
  33
  34 thermal_error_t thermal_sampling_handle(struct thermal_handler *th, void *arg)
  35 {
  36         struct thermal_handler_param thp = { .th = th, .arg = arg };
  37
  38         if (!th)
  39                 return THERMAL_ERROR;
  40
  41         if (nl_cb_set(th->cb_sampling, NL_CB_VALID, NL_CB_CUSTOM,
  42                       handle_thermal_sample, &thp))
  43                 return THERMAL_ERROR;
  44
  45         return nl_recvmsgs(th->sk_sampling, th->cb_sampling);
  46 }
  47
  48 int thermal_sampling_fd(struct thermal_handler *th)
  49 {
  50         if (!th)
  51                 return -1;
  52
  53         return nl_socket_get_fd(th->sk_sampling);
  54 }
  55
  56 thermal_error_t thermal_sampling_exit(struct thermal_handler *th)
  57 {
  58         if (nl_unsubscribe_thermal(th->sk_sampling, th->cb_sampling,
  59                                    THERMAL_GENL_SAMPLING_GROUP_NAME))
  60                 return THERMAL_ERROR;
  61
  62         nl_thermal_disconnect(th->sk_sampling, th->cb_sampling);
  63
  64         return THERMAL_SUCCESS;
  65 }
  66
  67 thermal_error_t thermal_sampling_init(struct thermal_handler *th)
  68 {
  69         if (nl_thermal_connect(&th->sk_sampling, &th->cb_sampling))
  70                 return THERMAL_ERROR;
  71
  72         if (nl_subscribe_thermal(th->sk_sampling, th->cb_sampling,
  73                                  THERMAL_GENL_SAMPLING_GROUP_NAME))
  74                 return THERMAL_ERROR;
  75
  76         return THERMAL_SUCCESS;
  77 }