1 // SPDX-License-Identifier: GPL-2.0-only
2 
3 #include "netlink.h"
4 #include "common.h"
5 #include "bitset.h"
6 
7 struct privflags_req_info {
8 	struct ethnl_req_info		base;
9 };
10 
11 struct privflags_reply_data {
12 	struct ethnl_reply_data		base;
13 	const char			(*priv_flag_names)[ETH_GSTRING_LEN];
14 	unsigned int			n_priv_flags;
15 	u32				priv_flags;
16 };
17 
18 #define PRIVFLAGS_REPDATA(__reply_base) \
19 	container_of(__reply_base, struct privflags_reply_data, base)
20 
21 const struct nla_policy ethnl_privflags_get_policy[] = {
22 	[ETHTOOL_A_PRIVFLAGS_HEADER]		=
23 		NLA_POLICY_NESTED(ethnl_header_policy),
24 };
25 
ethnl_get_priv_flags_info(struct net_device * dev,unsigned int * count,const char (** names)[ETH_GSTRING_LEN])26 static int ethnl_get_priv_flags_info(struct net_device *dev,
27 				     unsigned int *count,
28 				     const char (**names)[ETH_GSTRING_LEN])
29 {
30 	const struct ethtool_ops *ops = dev->ethtool_ops;
31 	int nflags;
32 
33 	nflags = ops->get_sset_count(dev, ETH_SS_PRIV_FLAGS);
34 	if (nflags < 0)
35 		return nflags;
36 
37 	if (names) {
38 		*names = kcalloc(nflags, ETH_GSTRING_LEN, GFP_KERNEL);
39 		if (!*names)
40 			return -ENOMEM;
41 		ops->get_strings(dev, ETH_SS_PRIV_FLAGS, (u8 *)*names);
42 	}
43 
44 	/* We can pass more than 32 private flags to userspace via netlink but
45 	 * we cannot get more with ethtool_ops::get_priv_flags(). Note that we
46 	 * must not adjust nflags before allocating the space for flag names
47 	 * as the buffer must be large enough for all flags.
48 	 */
49 	if (WARN_ONCE(nflags > 32,
50 		      "device %s reports more than 32 private flags (%d)\n",
51 		      netdev_name(dev), nflags))
52 		nflags = 32;
53 	*count = nflags;
54 
55 	return 0;
56 }
57 
privflags_prepare_data(const struct ethnl_req_info * req_base,struct ethnl_reply_data * reply_base,struct genl_info * info)58 static int privflags_prepare_data(const struct ethnl_req_info *req_base,
59 				  struct ethnl_reply_data *reply_base,
60 				  struct genl_info *info)
61 {
62 	struct privflags_reply_data *data = PRIVFLAGS_REPDATA(reply_base);
63 	struct net_device *dev = reply_base->dev;
64 	const char (*names)[ETH_GSTRING_LEN];
65 	const struct ethtool_ops *ops;
66 	unsigned int nflags;
67 	int ret;
68 
69 	ops = dev->ethtool_ops;
70 	if (!ops->get_priv_flags || !ops->get_sset_count || !ops->get_strings)
71 		return -EOPNOTSUPP;
72 	ret = ethnl_ops_begin(dev);
73 	if (ret < 0)
74 		return ret;
75 
76 	ret = ethnl_get_priv_flags_info(dev, &nflags, &names);
77 	if (ret < 0)
78 		goto out_ops;
79 	data->priv_flags = ops->get_priv_flags(dev);
80 	data->priv_flag_names = names;
81 	data->n_priv_flags = nflags;
82 
83 out_ops:
84 	ethnl_ops_complete(dev);
85 	return ret;
86 }
87 
privflags_reply_size(const struct ethnl_req_info * req_base,const struct ethnl_reply_data * reply_base)88 static int privflags_reply_size(const struct ethnl_req_info *req_base,
89 				const struct ethnl_reply_data *reply_base)
90 {
91 	const struct privflags_reply_data *data = PRIVFLAGS_REPDATA(reply_base);
92 	bool compact = req_base->flags & ETHTOOL_FLAG_COMPACT_BITSETS;
93 	const u32 all_flags = ~(u32)0 >> (32 - data->n_priv_flags);
94 
95 	return ethnl_bitset32_size(&data->priv_flags, &all_flags,
96 				   data->n_priv_flags,
97 				   data->priv_flag_names, compact);
98 }
99 
privflags_fill_reply(struct sk_buff * skb,const struct ethnl_req_info * req_base,const struct ethnl_reply_data * reply_base)100 static int privflags_fill_reply(struct sk_buff *skb,
101 				const struct ethnl_req_info *req_base,
102 				const struct ethnl_reply_data *reply_base)
103 {
104 	const struct privflags_reply_data *data = PRIVFLAGS_REPDATA(reply_base);
105 	bool compact = req_base->flags & ETHTOOL_FLAG_COMPACT_BITSETS;
106 	const u32 all_flags = ~(u32)0 >> (32 - data->n_priv_flags);
107 
108 	return ethnl_put_bitset32(skb, ETHTOOL_A_PRIVFLAGS_FLAGS,
109 				  &data->priv_flags, &all_flags,
110 				  data->n_priv_flags, data->priv_flag_names,
111 				  compact);
112 }
113 
privflags_cleanup_data(struct ethnl_reply_data * reply_data)114 static void privflags_cleanup_data(struct ethnl_reply_data *reply_data)
115 {
116 	struct privflags_reply_data *data = PRIVFLAGS_REPDATA(reply_data);
117 
118 	kfree(data->priv_flag_names);
119 }
120 
121 const struct ethnl_request_ops ethnl_privflags_request_ops = {
122 	.request_cmd		= ETHTOOL_MSG_PRIVFLAGS_GET,
123 	.reply_cmd		= ETHTOOL_MSG_PRIVFLAGS_GET_REPLY,
124 	.hdr_attr		= ETHTOOL_A_PRIVFLAGS_HEADER,
125 	.req_info_size		= sizeof(struct privflags_req_info),
126 	.reply_data_size	= sizeof(struct privflags_reply_data),
127 
128 	.prepare_data		= privflags_prepare_data,
129 	.reply_size		= privflags_reply_size,
130 	.fill_reply		= privflags_fill_reply,
131 	.cleanup_data		= privflags_cleanup_data,
132 };
133 
134 /* PRIVFLAGS_SET */
135 
136 const struct nla_policy ethnl_privflags_set_policy[] = {
137 	[ETHTOOL_A_PRIVFLAGS_HEADER]		=
138 		NLA_POLICY_NESTED(ethnl_header_policy),
139 	[ETHTOOL_A_PRIVFLAGS_FLAGS]		= { .type = NLA_NESTED },
140 };
141 
ethnl_set_privflags(struct sk_buff * skb,struct genl_info * info)142 int ethnl_set_privflags(struct sk_buff *skb, struct genl_info *info)
143 {
144 	const char (*names)[ETH_GSTRING_LEN] = NULL;
145 	struct ethnl_req_info req_info = {};
146 	struct nlattr **tb = info->attrs;
147 	const struct ethtool_ops *ops;
148 	struct net_device *dev;
149 	unsigned int nflags;
150 	bool mod = false;
151 	bool compact;
152 	u32 flags;
153 	int ret;
154 
155 	if (!tb[ETHTOOL_A_PRIVFLAGS_FLAGS])
156 		return -EINVAL;
157 	ret = ethnl_bitset_is_compact(tb[ETHTOOL_A_PRIVFLAGS_FLAGS], &compact);
158 	if (ret < 0)
159 		return ret;
160 	ret = ethnl_parse_header_dev_get(&req_info,
161 					 tb[ETHTOOL_A_PRIVFLAGS_HEADER],
162 					 genl_info_net(info), info->extack,
163 					 true);
164 	if (ret < 0)
165 		return ret;
166 	dev = req_info.dev;
167 	ops = dev->ethtool_ops;
168 	ret = -EOPNOTSUPP;
169 	if (!ops->get_priv_flags || !ops->set_priv_flags ||
170 	    !ops->get_sset_count || !ops->get_strings)
171 		goto out_dev;
172 
173 	rtnl_lock();
174 	ret = ethnl_ops_begin(dev);
175 	if (ret < 0)
176 		goto out_rtnl;
177 	ret = ethnl_get_priv_flags_info(dev, &nflags, compact ? NULL : &names);
178 	if (ret < 0)
179 		goto out_ops;
180 	flags = ops->get_priv_flags(dev);
181 
182 	ret = ethnl_update_bitset32(&flags, nflags,
183 				    tb[ETHTOOL_A_PRIVFLAGS_FLAGS], names,
184 				    info->extack, &mod);
185 	if (ret < 0 || !mod)
186 		goto out_free;
187 	ret = ops->set_priv_flags(dev, flags);
188 	if (ret < 0)
189 		goto out_free;
190 	ethtool_notify(dev, ETHTOOL_MSG_PRIVFLAGS_NTF, NULL);
191 
192 out_free:
193 	kfree(names);
194 out_ops:
195 	ethnl_ops_complete(dev);
196 out_rtnl:
197 	rtnl_unlock();
198 out_dev:
199 	dev_put(dev);
200 	return ret;
201 }
202