cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

sampling.c (1895B)

      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_EVENT_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}