// SPDX-License-Identifier: GPL-2.0-only #include <linux/ethtool.h> #include "netlink.h"#include "common.h"#include "bitset.h" struct module_req_info {	struct ethnl_req_info base;}; struct module_reply_data {	struct ethnl_reply_data	base;	struct ethtool_module_power_mode_params power;}; #define MODULE_REPDATA(__reply_base) \	container_of(__reply_base, struct module_reply_data, base) /* MODULE_GET */ const struct nla_policy ethnl_module_get_policy[ETHTOOL_A_MODULE_HEADER + 1] = {	[ETHTOOL_A_MODULE_HEADER] = NLA_POLICY_NESTED(ethnl_header_policy),}; static int module_get_power_mode(struct net_device *dev,				 struct module_reply_data *data,				 struct netlink_ext_ack *extack){	const struct ethtool_ops *ops = dev->ethtool_ops; 	if (!ops->get_module_power_mode)		return 0; 	return ops->get_module_power_mode(dev, &data->power, extack);} static int module_prepare_data(const struct ethnl_req_info *req_base,			       struct ethnl_reply_data *reply_base,			       struct genl_info *info){	struct module_reply_data *data = MODULE_REPDATA(reply_base);	struct netlink_ext_ack *extack = info ? info->extack : NULL;	struct net_device *dev = reply_base->dev;	int ret; 	ret = ethnl_ops_begin(dev);	if (ret < 0)		return ret; 	ret = module_get_power_mode(dev, data, extack);	if (ret < 0)		goto out_complete; out_complete:	ethnl_ops_complete(dev);	return ret;} static int module_reply_size(const struct ethnl_req_info *req_base,			     const struct ethnl_reply_data *reply_base){	struct module_reply_data *data = MODULE_REPDATA(reply_base);	int len = 0; 	if (data->power.policy)		len += nla_total_size(sizeof(u8));	/* _MODULE_POWER_MODE_POLICY */ 	if (data->power.mode)		len += nla_total_size(sizeof(u8));	/* _MODULE_POWER_MODE */ 	return len;} static int module_fill_reply(struct sk_buff *skb,			     const struct ethnl_req_info *req_base,			     const struct ethnl_reply_data *reply_base){	const struct module_reply_data *data = MODULE_REPDATA(reply_base); 	if (data->power.policy &&	    nla_put_u8(skb, ETHTOOL_A_MODULE_POWER_MODE_POLICY,		       data->power.policy))		return -EMSGSIZE; 	if (data->power.mode &&	    nla_put_u8(skb, ETHTOOL_A_MODULE_POWER_MODE, data->power.mode))		return -EMSGSIZE; 	return 0;} const struct ethnl_request_ops ethnl_module_request_ops = {	.request_cmd		= ETHTOOL_MSG_MODULE_GET,	.reply_cmd		= ETHTOOL_MSG_MODULE_GET_REPLY,	.hdr_attr		= ETHTOOL_A_MODULE_HEADER,	.req_info_size		= sizeof(struct module_req_info),	.reply_data_size	= sizeof(struct module_reply_data), 	.prepare_data		= module_prepare_data,	.reply_size		= module_reply_size,	.fill_reply		= module_fill_reply,}; /* MODULE_SET */ const struct nla_policy ethnl_module_set_policy[ETHTOOL_A_MODULE_POWER_MODE_POLICY + 1] = {	[ETHTOOL_A_MODULE_HEADER] = NLA_POLICY_NESTED(ethnl_header_policy),	[ETHTOOL_A_MODULE_POWER_MODE_POLICY] =		NLA_POLICY_RANGE(NLA_U8, ETHTOOL_MODULE_POWER_MODE_POLICY_HIGH,				 ETHTOOL_MODULE_POWER_MODE_POLICY_AUTO),}; static int module_set_power_mode(struct net_device *dev, struct nlattr **tb,				 bool *p_mod, struct netlink_ext_ack *extack){	struct ethtool_module_power_mode_params power = {};	struct ethtool_module_power_mode_params power_new;	const struct ethtool_ops *ops = dev->ethtool_ops;	int ret; 	if (!tb[ETHTOOL_A_MODULE_POWER_MODE_POLICY])		return 0; 	if (!ops->get_module_power_mode || !ops->set_module_power_mode) {		NL_SET_ERR_MSG_ATTR(extack,				    tb[ETHTOOL_A_MODULE_POWER_MODE_POLICY],				    "Setting power mode policy is not supported by this device");		return -EOPNOTSUPP;	} 	power_new.policy = nla_get_u8(tb[ETHTOOL_A_MODULE_POWER_MODE_POLICY]);	ret = ops->get_module_power_mode(dev, &power, extack);	if (ret < 0)		return ret; 	if (power_new.policy == power.policy)		return 0;	*p_mod = true; 	return ops->set_module_power_mode(dev, &power_new, extack);} int ethnl_set_module(struct sk_buff *skb, struct genl_info *info){	struct ethnl_req_info req_info = {};	struct nlattr **tb = info->attrs;	struct net_device *dev;	bool mod = false;	int ret; 	ret = ethnl_parse_header_dev_get(&req_info, tb[ETHTOOL_A_MODULE_HEADER],					 genl_info_net(info), info->extack,					 true);	if (ret < 0)		return ret;	dev = req_info.dev; 	rtnl_lock();	ret = ethnl_ops_begin(dev);	if (ret < 0)		goto out_rtnl; 	ret = module_set_power_mode(dev, tb, &mod, info->extack);	if (ret < 0)		goto out_ops; 	if (!mod)		goto out_ops; 	ethtool_notify(dev, ETHTOOL_MSG_MODULE_NTF, NULL); out_ops:	ethnl_ops_complete(dev);out_rtnl:	rtnl_unlock();	ethnl_parse_header_dev_put(&req_info);	return ret;}