1 // SPDX-License-Identifier: GPL-2.0-or-later
3 * Copyright (c) 2016 Mellanox Technologies. All rights reserved.
7 #include <net/genetlink.h>
10 #include "devl_internal.h"
12 #define DEVLINK_NL_FLAG_NEED_PORT BIT(0)
13 #define DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT BIT(1)
14 #define DEVLINK_NL_FLAG_NEED_DEV_LOCK BIT(2)
16 static const struct genl_multicast_group devlink_nl_mcgrps[] = {
17 [DEVLINK_MCGRP_CONFIG] = { .name = DEVLINK_GENL_MCGRP_CONFIG_NAME },
20 struct devlink_nl_sock_priv {
21 struct devlink_obj_desc __rcu *flt;
22 spinlock_t flt_lock; /* Protects flt. */
25 static void devlink_nl_sock_priv_init(void *priv)
27 struct devlink_nl_sock_priv *sk_priv = priv;
29 spin_lock_init(&sk_priv->flt_lock);
32 static void devlink_nl_sock_priv_destroy(void *priv)
34 struct devlink_nl_sock_priv *sk_priv = priv;
35 struct devlink_obj_desc *flt;
37 flt = rcu_dereference_protected(sk_priv->flt, true);
41 int devlink_nl_notify_filter_set_doit(struct sk_buff *skb,
42 struct genl_info *info)
44 struct devlink_nl_sock_priv *sk_priv;
45 struct nlattr **attrs = info->attrs;
46 struct devlink_obj_desc *flt;
47 size_t data_offset = 0;
51 if (attrs[DEVLINK_ATTR_BUS_NAME])
52 data_size = size_add(data_size,
53 nla_len(attrs[DEVLINK_ATTR_BUS_NAME]) + 1);
54 if (attrs[DEVLINK_ATTR_DEV_NAME])
55 data_size = size_add(data_size,
56 nla_len(attrs[DEVLINK_ATTR_DEV_NAME]) + 1);
58 flt = kzalloc(size_add(sizeof(*flt), data_size), GFP_KERNEL);
62 pos = (char *) flt->data;
63 if (attrs[DEVLINK_ATTR_BUS_NAME]) {
64 data_offset += nla_strscpy(pos,
65 attrs[DEVLINK_ATTR_BUS_NAME],
70 if (attrs[DEVLINK_ATTR_DEV_NAME]) {
71 nla_strscpy(pos, attrs[DEVLINK_ATTR_DEV_NAME],
72 data_size - data_offset);
76 if (attrs[DEVLINK_ATTR_PORT_INDEX]) {
77 flt->port_index = nla_get_u32(attrs[DEVLINK_ATTR_PORT_INDEX]);
78 flt->port_index_valid = true;
81 /* Don't attach empty filter. */
82 if (!flt->bus_name && !flt->dev_name && !flt->port_index_valid) {
87 sk_priv = genl_sk_priv_get(&devlink_nl_family, NETLINK_CB(skb).sk);
88 if (IS_ERR(sk_priv)) {
90 return PTR_ERR(sk_priv);
92 spin_lock(&sk_priv->flt_lock);
93 flt = rcu_replace_pointer(sk_priv->flt, flt,
94 lockdep_is_held(&sk_priv->flt_lock));
95 spin_unlock(&sk_priv->flt_lock);
100 static bool devlink_obj_desc_match(const struct devlink_obj_desc *desc,
101 const struct devlink_obj_desc *flt)
103 if (desc->bus_name && flt->bus_name &&
104 strcmp(desc->bus_name, flt->bus_name))
106 if (desc->dev_name && flt->dev_name &&
107 strcmp(desc->dev_name, flt->dev_name))
109 if (desc->port_index_valid && flt->port_index_valid &&
110 desc->port_index != flt->port_index)
115 int devlink_nl_notify_filter(struct sock *dsk, struct sk_buff *skb, void *data)
117 struct devlink_obj_desc *desc = data;
118 struct devlink_nl_sock_priv *sk_priv;
119 struct devlink_obj_desc *flt;
123 sk_priv = __genl_sk_priv_get(&devlink_nl_family, dsk);
124 if (!IS_ERR_OR_NULL(sk_priv)) {
125 flt = rcu_dereference(sk_priv->flt);
127 ret = !devlink_obj_desc_match(desc, flt);
133 int devlink_nl_put_nested_handle(struct sk_buff *msg, struct net *net,
134 struct devlink *devlink, int attrtype)
136 struct nlattr *nested_attr;
137 struct net *devl_net;
139 nested_attr = nla_nest_start(msg, attrtype);
142 if (devlink_nl_put_handle(msg, devlink))
143 goto nla_put_failure;
146 devl_net = read_pnet_rcu(&devlink->_net);
147 if (!net_eq(net, devl_net)) {
148 int id = peernet2id_alloc(net, devl_net, GFP_ATOMIC);
151 if (nla_put_s32(msg, DEVLINK_ATTR_NETNS_ID, id))
157 nla_nest_end(msg, nested_attr);
161 nla_nest_cancel(msg, nested_attr);
165 int devlink_nl_msg_reply_and_new(struct sk_buff **msg, struct genl_info *info)
170 err = genlmsg_reply(*msg, info);
174 *msg = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
181 devlink_get_from_attrs_lock(struct net *net, struct nlattr **attrs,
184 struct devlink *devlink;
189 if (!attrs[DEVLINK_ATTR_BUS_NAME] || !attrs[DEVLINK_ATTR_DEV_NAME])
190 return ERR_PTR(-EINVAL);
192 busname = nla_data(attrs[DEVLINK_ATTR_BUS_NAME]);
193 devname = nla_data(attrs[DEVLINK_ATTR_DEV_NAME]);
195 devlinks_xa_for_each_registered_get(net, index, devlink) {
196 devl_dev_lock(devlink, dev_lock);
197 if (devl_is_registered(devlink) &&
198 strcmp(devlink->dev->bus->name, busname) == 0 &&
199 strcmp(dev_name(devlink->dev), devname) == 0)
201 devl_dev_unlock(devlink, dev_lock);
202 devlink_put(devlink);
205 return ERR_PTR(-ENODEV);
208 static int __devlink_nl_pre_doit(struct sk_buff *skb, struct genl_info *info,
211 bool dev_lock = flags & DEVLINK_NL_FLAG_NEED_DEV_LOCK;
212 struct devlink_port *devlink_port;
213 struct devlink *devlink;
216 devlink = devlink_get_from_attrs_lock(genl_info_net(info), info->attrs,
219 return PTR_ERR(devlink);
221 info->user_ptr[0] = devlink;
222 if (flags & DEVLINK_NL_FLAG_NEED_PORT) {
223 devlink_port = devlink_port_get_from_info(devlink, info);
224 if (IS_ERR(devlink_port)) {
225 err = PTR_ERR(devlink_port);
228 info->user_ptr[1] = devlink_port;
229 } else if (flags & DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT) {
230 devlink_port = devlink_port_get_from_info(devlink, info);
231 if (!IS_ERR(devlink_port))
232 info->user_ptr[1] = devlink_port;
237 devl_dev_unlock(devlink, dev_lock);
238 devlink_put(devlink);
242 int devlink_nl_pre_doit(const struct genl_split_ops *ops,
243 struct sk_buff *skb, struct genl_info *info)
245 return __devlink_nl_pre_doit(skb, info, 0);
248 int devlink_nl_pre_doit_port(const struct genl_split_ops *ops,
249 struct sk_buff *skb, struct genl_info *info)
251 return __devlink_nl_pre_doit(skb, info, DEVLINK_NL_FLAG_NEED_PORT);
254 int devlink_nl_pre_doit_dev_lock(const struct genl_split_ops *ops,
255 struct sk_buff *skb, struct genl_info *info)
257 return __devlink_nl_pre_doit(skb, info, DEVLINK_NL_FLAG_NEED_DEV_LOCK);
260 int devlink_nl_pre_doit_port_optional(const struct genl_split_ops *ops,
262 struct genl_info *info)
264 return __devlink_nl_pre_doit(skb, info, DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT);
267 static void __devlink_nl_post_doit(struct sk_buff *skb, struct genl_info *info,
270 bool dev_lock = flags & DEVLINK_NL_FLAG_NEED_DEV_LOCK;
271 struct devlink *devlink;
273 devlink = info->user_ptr[0];
274 devl_dev_unlock(devlink, dev_lock);
275 devlink_put(devlink);
278 void devlink_nl_post_doit(const struct genl_split_ops *ops,
279 struct sk_buff *skb, struct genl_info *info)
281 __devlink_nl_post_doit(skb, info, 0);
285 devlink_nl_post_doit_dev_lock(const struct genl_split_ops *ops,
286 struct sk_buff *skb, struct genl_info *info)
288 __devlink_nl_post_doit(skb, info, DEVLINK_NL_FLAG_NEED_DEV_LOCK);
291 static int devlink_nl_inst_single_dumpit(struct sk_buff *msg,
292 struct netlink_callback *cb, int flags,
293 devlink_nl_dump_one_func_t *dump_one,
294 struct nlattr **attrs)
296 struct devlink *devlink;
299 devlink = devlink_get_from_attrs_lock(sock_net(msg->sk), attrs, false);
301 return PTR_ERR(devlink);
302 err = dump_one(msg, devlink, cb, flags | NLM_F_DUMP_FILTERED);
304 devl_unlock(devlink);
305 devlink_put(devlink);
307 if (err != -EMSGSIZE)
312 static int devlink_nl_inst_iter_dumpit(struct sk_buff *msg,
313 struct netlink_callback *cb, int flags,
314 devlink_nl_dump_one_func_t *dump_one)
316 struct devlink_nl_dump_state *state = devlink_dump_state(cb);
317 struct devlink *devlink;
320 while ((devlink = devlinks_xa_find_get(sock_net(msg->sk),
321 &state->instance))) {
324 if (devl_is_registered(devlink))
325 err = dump_one(msg, devlink, cb, flags);
329 devl_unlock(devlink);
330 devlink_put(devlink);
337 /* restart sub-object walk for the next instance */
341 if (err != -EMSGSIZE)
346 int devlink_nl_dumpit(struct sk_buff *msg, struct netlink_callback *cb,
347 devlink_nl_dump_one_func_t *dump_one)
349 const struct genl_info *info = genl_info_dump(cb);
350 struct nlattr **attrs = info->attrs;
351 int flags = NLM_F_MULTI;
354 (attrs[DEVLINK_ATTR_BUS_NAME] || attrs[DEVLINK_ATTR_DEV_NAME]))
355 return devlink_nl_inst_single_dumpit(msg, cb, flags, dump_one,
358 return devlink_nl_inst_iter_dumpit(msg, cb, flags, dump_one);
361 struct genl_family devlink_nl_family __ro_after_init = {
362 .name = DEVLINK_GENL_NAME,
363 .version = DEVLINK_GENL_VERSION,
365 .parallel_ops = true,
366 .module = THIS_MODULE,
367 .split_ops = devlink_nl_ops,
368 .n_split_ops = ARRAY_SIZE(devlink_nl_ops),
369 .resv_start_op = DEVLINK_CMD_SELFTESTS_RUN + 1,
370 .mcgrps = devlink_nl_mcgrps,
371 .n_mcgrps = ARRAY_SIZE(devlink_nl_mcgrps),
372 .sock_priv_size = sizeof(struct devlink_nl_sock_priv),
373 .sock_priv_init = devlink_nl_sock_priv_init,
374 .sock_priv_destroy = devlink_nl_sock_priv_destroy,