diff options
Diffstat (limited to 'tools/lib/thermal/commands.c')
| -rw-r--r-- | tools/lib/thermal/commands.c | 188 | 
1 files changed, 175 insertions, 13 deletions
diff --git a/tools/lib/thermal/commands.c b/tools/lib/thermal/commands.c index 73d4d4e8d6ec..4998cec793ed 100644 --- a/tools/lib/thermal/commands.c +++ b/tools/lib/thermal/commands.c @@ -5,6 +5,7 @@  #include <stdio.h>  #include <stdlib.h>  #include <unistd.h> +#include <limits.h>  #include <thermal.h>  #include "thermal_nl.h" @@ -33,6 +34,11 @@ static struct nla_policy thermal_genl_policy[THERMAL_GENL_ATTR_MAX + 1] = {  	[THERMAL_GENL_ATTR_CDEV_CUR_STATE]      = { .type = NLA_U32 },  	[THERMAL_GENL_ATTR_CDEV_MAX_STATE]      = { .type = NLA_U32 },  	[THERMAL_GENL_ATTR_CDEV_NAME]           = { .type = NLA_STRING }, + +        /* Thresholds */ +        [THERMAL_GENL_ATTR_THRESHOLD]      	= { .type = NLA_NESTED }, +        [THERMAL_GENL_ATTR_THRESHOLD_TEMP]      = { .type = NLA_U32 }, +        [THERMAL_GENL_ATTR_THRESHOLD_DIRECTION] = { .type = NLA_U32 },  };  static int parse_tz_get(struct genl_info *info, struct thermal_zone **tz) @@ -182,6 +188,48 @@ static int parse_tz_get_gov(struct genl_info *info, struct thermal_zone *tz)  	return THERMAL_SUCCESS;  } +static int parse_threshold_get(struct genl_info *info, struct thermal_zone *tz) +{ +	struct nlattr *attr; +	struct thermal_threshold *__tt = NULL; +	size_t size = 0; +	int rem; + +	/* +	 * The size contains the size of the array and we want to +	 * access the last element, size - 1. +	 * +	 * The variable size is initialized to zero but it will be +	 * then incremented by the first if() statement. The message +	 * attributes are ordered, so the first if() statement will be +	 * always called before the second one. If it happens that is +	 * not the case, then it is a kernel bug. +	 */ +	nla_for_each_nested(attr, info->attrs[THERMAL_GENL_ATTR_THRESHOLD], rem) { + +		if (nla_type(attr) == THERMAL_GENL_ATTR_THRESHOLD_TEMP) { + +			size++; + +			__tt = realloc(__tt, sizeof(*__tt) * (size + 2)); +			if (!__tt) +				return THERMAL_ERROR; + +			__tt[size - 1].temperature = nla_get_u32(attr); +		} + +		if (nla_type(attr) == THERMAL_GENL_ATTR_THRESHOLD_DIRECTION) +			__tt[size - 1].direction = nla_get_u32(attr); +	} + +	if (__tt) +		__tt[size].temperature = INT_MAX; + +	tz->thresholds = __tt; + +	return THERMAL_SUCCESS; +} +  static int handle_netlink(struct nl_cache_ops *unused,  			  struct genl_cmd *cmd,  			  struct genl_info *info, void *arg) @@ -210,6 +258,10 @@ static int handle_netlink(struct nl_cache_ops *unused,  		ret = parse_tz_get_gov(info, arg);  		break; +	case THERMAL_GENL_CMD_THRESHOLD_GET: +		ret = parse_threshold_get(info, arg); +		break; +  	default:  		return THERMAL_ERROR;  	} @@ -253,6 +305,34 @@ static struct genl_cmd thermal_cmds[] = {  		.c_maxattr	= THERMAL_GENL_ATTR_MAX,  		.c_attr_policy	= thermal_genl_policy,  	}, +        { +                .c_id           = THERMAL_GENL_CMD_THRESHOLD_GET, +                .c_name         = (char *)"Get thresholds list", +                .c_msg_parser   = handle_netlink, +                .c_maxattr      = THERMAL_GENL_ATTR_MAX, +                .c_attr_policy  = thermal_genl_policy, +        }, +        { +                .c_id           = THERMAL_GENL_CMD_THRESHOLD_ADD, +                .c_name         = (char *)"Add a threshold", +                .c_msg_parser   = handle_netlink, +                .c_maxattr      = THERMAL_GENL_ATTR_MAX, +                .c_attr_policy  = thermal_genl_policy, +        }, +        { +                .c_id           = THERMAL_GENL_CMD_THRESHOLD_DELETE, +                .c_name         = (char *)"Delete a threshold", +                .c_msg_parser   = handle_netlink, +                .c_maxattr      = THERMAL_GENL_ATTR_MAX, +                .c_attr_policy  = thermal_genl_policy, +        }, +        { +                .c_id           = THERMAL_GENL_CMD_THRESHOLD_FLUSH, +                .c_name         = (char *)"Flush the thresholds", +                .c_msg_parser   = handle_netlink, +                .c_maxattr      = THERMAL_GENL_ATTR_MAX, +                .c_attr_policy  = thermal_genl_policy, +        },  };  static struct genl_ops thermal_cmd_ops = { @@ -261,9 +341,41 @@ static struct genl_ops thermal_cmd_ops = {  	.o_ncmds	= ARRAY_SIZE(thermal_cmds),  }; -static thermal_error_t thermal_genl_auto(struct thermal_handler *th, int id, int cmd, -					 int flags, void *arg) +struct cmd_param { +	int tz_id; +	int temp; +	int direction; +}; + +typedef int (*cmd_cb_t)(struct nl_msg *, struct cmd_param *); + +static int thermal_genl_tz_id_encode(struct nl_msg *msg, struct cmd_param *p)  { +	if (nla_put_u32(msg, THERMAL_GENL_ATTR_TZ_ID, p->tz_id)) +		return -1; + +	return 0; +} + +static int thermal_genl_threshold_encode(struct nl_msg *msg, struct cmd_param *p) +{ +	if (thermal_genl_tz_id_encode(msg, p)) +		return -1; + +	if (nla_put_u32(msg, THERMAL_GENL_ATTR_THRESHOLD_TEMP, p->temp)) +		return -1; + +	if (nla_put_u32(msg, THERMAL_GENL_ATTR_THRESHOLD_DIRECTION, p->direction)) +		return -1; + +	return 0; +} + +static thermal_error_t thermal_genl_auto(struct thermal_handler *th, cmd_cb_t cmd_cb, +					 struct cmd_param *param, +					 int cmd, int flags, void *arg) +{ +	thermal_error_t ret = THERMAL_ERROR;  	struct nl_msg *msg;  	void *hdr; @@ -274,45 +386,95 @@ static thermal_error_t thermal_genl_auto(struct thermal_handler *th, int id, int  	hdr = genlmsg_put(msg, NL_AUTO_PORT, NL_AUTO_SEQ, thermal_cmd_ops.o_id,  			  0, flags, cmd, THERMAL_GENL_VERSION);  	if (!hdr) -		return THERMAL_ERROR; +		goto out; -	if (id >= 0 && nla_put_u32(msg, THERMAL_GENL_ATTR_TZ_ID, id)) -		return THERMAL_ERROR; +	if (cmd_cb && cmd_cb(msg, param)) +		goto out;  	if (nl_send_msg(th->sk_cmd, th->cb_cmd, msg, genl_handle_msg, arg)) -		return THERMAL_ERROR; +		goto out; +	ret = THERMAL_SUCCESS; +out:  	nlmsg_free(msg); -	return THERMAL_SUCCESS; +	return ret;  }  thermal_error_t thermal_cmd_get_tz(struct thermal_handler *th, struct thermal_zone **tz)  { -	return thermal_genl_auto(th, -1, THERMAL_GENL_CMD_TZ_GET_ID, +	return thermal_genl_auto(th, NULL, NULL, THERMAL_GENL_CMD_TZ_GET_ID,  				 NLM_F_DUMP | NLM_F_ACK, tz);  }  thermal_error_t thermal_cmd_get_cdev(struct thermal_handler *th, struct thermal_cdev **tc)  { -	return thermal_genl_auto(th, -1, THERMAL_GENL_CMD_CDEV_GET, +	return thermal_genl_auto(th, NULL, NULL, THERMAL_GENL_CMD_CDEV_GET,  				 NLM_F_DUMP | NLM_F_ACK, tc);  }  thermal_error_t thermal_cmd_get_trip(struct thermal_handler *th, struct thermal_zone *tz)  { -	return thermal_genl_auto(th, tz->id, THERMAL_GENL_CMD_TZ_GET_TRIP, -				 0, tz); +	struct cmd_param p = { .tz_id = tz->id }; + +	return thermal_genl_auto(th, thermal_genl_tz_id_encode, &p, +				 THERMAL_GENL_CMD_TZ_GET_TRIP, 0, tz);  }  thermal_error_t thermal_cmd_get_governor(struct thermal_handler *th, struct thermal_zone *tz)  { -	return thermal_genl_auto(th, tz->id, THERMAL_GENL_CMD_TZ_GET_GOV, 0, tz); +	struct cmd_param p = { .tz_id = tz->id }; + +	return thermal_genl_auto(th, thermal_genl_tz_id_encode, &p, +				 THERMAL_GENL_CMD_TZ_GET_GOV, 0, tz);  }  thermal_error_t thermal_cmd_get_temp(struct thermal_handler *th, struct thermal_zone *tz)  { -	return thermal_genl_auto(th, tz->id, THERMAL_GENL_CMD_TZ_GET_TEMP, 0, tz); +	struct cmd_param p = { .tz_id = tz->id }; + +	return thermal_genl_auto(th, thermal_genl_tz_id_encode, &p, +				 THERMAL_GENL_CMD_TZ_GET_TEMP, 0, tz); +} + +thermal_error_t thermal_cmd_threshold_get(struct thermal_handler *th, +                                          struct thermal_zone *tz) +{ +	struct cmd_param p = { .tz_id = tz->id }; + +        return thermal_genl_auto(th, thermal_genl_tz_id_encode, &p, +				 THERMAL_GENL_CMD_THRESHOLD_GET, 0, tz); +} + +thermal_error_t thermal_cmd_threshold_add(struct thermal_handler *th, +                                          struct thermal_zone *tz, +                                          int temperature, +                                          int direction) +{ +	struct cmd_param p = { .tz_id = tz->id, .temp = temperature, .direction = direction }; + +        return thermal_genl_auto(th, thermal_genl_threshold_encode, &p, +				 THERMAL_GENL_CMD_THRESHOLD_ADD, 0, tz); +} + +thermal_error_t thermal_cmd_threshold_delete(struct thermal_handler *th, +                                             struct thermal_zone *tz, +                                             int temperature, +                                             int direction) +{ +	struct cmd_param p = { .tz_id = tz->id, .temp = temperature, .direction = direction }; + +        return thermal_genl_auto(th, thermal_genl_threshold_encode, &p, +				 THERMAL_GENL_CMD_THRESHOLD_DELETE, 0, tz); +} + +thermal_error_t thermal_cmd_threshold_flush(struct thermal_handler *th, +                                            struct thermal_zone *tz) +{ +	struct cmd_param p = { .tz_id = tz->id }; + +        return thermal_genl_auto(th, thermal_genl_tz_id_encode, &p, +				 THERMAL_GENL_CMD_THRESHOLD_FLUSH, 0, tz);  }  thermal_error_t thermal_cmd_exit(struct thermal_handler *th)  | 
