Branch data Line data Source code
1 : : // SPDX-License-Identifier: GPL-2.0
2 : : /*
3 : : * NETLINK Generic Netlink Family
4 : : *
5 : : * Authors: Jamal Hadi Salim
6 : : * Thomas Graf <tgraf@suug.ch>
7 : : * Johannes Berg <johannes@sipsolutions.net>
8 : : */
9 : :
10 : : #include <linux/module.h>
11 : : #include <linux/kernel.h>
12 : : #include <linux/slab.h>
13 : : #include <linux/errno.h>
14 : : #include <linux/types.h>
15 : : #include <linux/socket.h>
16 : : #include <linux/string.h>
17 : : #include <linux/skbuff.h>
18 : : #include <linux/mutex.h>
19 : : #include <linux/bitmap.h>
20 : : #include <linux/rwsem.h>
21 : : #include <linux/idr.h>
22 : : #include <net/sock.h>
23 : : #include <net/genetlink.h>
24 : :
25 : : static DEFINE_MUTEX(genl_mutex); /* serialization of message processing */
26 : : static DECLARE_RWSEM(cb_lock);
27 : :
28 : : atomic_t genl_sk_destructing_cnt = ATOMIC_INIT(0);
29 : : DECLARE_WAIT_QUEUE_HEAD(genl_sk_destructing_waitq);
30 : :
31 : 0 : void genl_lock(void)
32 : : {
33 : 3 : mutex_lock(&genl_mutex);
34 : 0 : }
35 : : EXPORT_SYMBOL(genl_lock);
36 : :
37 : 0 : void genl_unlock(void)
38 : : {
39 : 3 : mutex_unlock(&genl_mutex);
40 : 0 : }
41 : : EXPORT_SYMBOL(genl_unlock);
42 : :
43 : : #ifdef CONFIG_LOCKDEP
44 : : bool lockdep_genl_is_held(void)
45 : : {
46 : : return lockdep_is_held(&genl_mutex);
47 : : }
48 : : EXPORT_SYMBOL(lockdep_genl_is_held);
49 : : #endif
50 : :
51 : : static void genl_lock_all(void)
52 : : {
53 : 3 : down_write(&cb_lock);
54 : : genl_lock();
55 : : }
56 : :
57 : : static void genl_unlock_all(void)
58 : : {
59 : : genl_unlock();
60 : 3 : up_write(&cb_lock);
61 : : }
62 : :
63 : : static DEFINE_IDR(genl_fam_idr);
64 : :
65 : : /*
66 : : * Bitmap of multicast groups that are currently in use.
67 : : *
68 : : * To avoid an allocation at boot of just one unsigned long,
69 : : * declare it global instead.
70 : : * Bit 0 is marked as already used since group 0 is invalid.
71 : : * Bit 1 is marked as already used since the drop-monitor code
72 : : * abuses the API and thinks it can statically use group 1.
73 : : * That group will typically conflict with other groups that
74 : : * any proper users use.
75 : : * Bit 16 is marked as used since it's used for generic netlink
76 : : * and the code no longer marks pre-reserved IDs as used.
77 : : * Bit 17 is marked as already used since the VFS quota code
78 : : * also abused this API and relied on family == group ID, we
79 : : * cater to that by giving it a static family and group ID.
80 : : * Bit 18 is marked as already used since the PMCRAID driver
81 : : * did the same thing as the VFS quota code (maybe copied?)
82 : : */
83 : : static unsigned long mc_group_start = 0x3 | BIT(GENL_ID_CTRL) |
84 : : BIT(GENL_ID_VFS_DQUOT) |
85 : : BIT(GENL_ID_PMCRAID);
86 : : static unsigned long *mc_groups = &mc_group_start;
87 : : static unsigned long mc_groups_longs = 1;
88 : :
89 : : static int genl_ctrl_event(int event, const struct genl_family *family,
90 : : const struct genl_multicast_group *grp,
91 : : int grp_id);
92 : :
93 : : static const struct genl_family *genl_family_find_byid(unsigned int id)
94 : : {
95 : 3 : return idr_find(&genl_fam_idr, id);
96 : : }
97 : :
98 : 3 : static const struct genl_family *genl_family_find_byname(char *name)
99 : : {
100 : : const struct genl_family *family;
101 : : unsigned int id;
102 : :
103 : 3 : idr_for_each_entry(&genl_fam_idr, family, id)
104 : 3 : if (strcmp(family->name, name) == 0)
105 : 3 : return family;
106 : :
107 : : return NULL;
108 : : }
109 : :
110 : : static const struct genl_ops *genl_get_cmd(u8 cmd,
111 : : const struct genl_family *family)
112 : : {
113 : : int i;
114 : :
115 : 0 : for (i = 0; i < family->n_ops; i++)
116 : 3 : if (family->ops[i].cmd == cmd)
117 : 3 : return &family->ops[i];
118 : :
119 : : return NULL;
120 : : }
121 : :
122 : 3 : static int genl_allocate_reserve_groups(int n_groups, int *first_id)
123 : : {
124 : : unsigned long *new_groups;
125 : : int start = 0;
126 : : int i;
127 : : int id;
128 : : bool fits;
129 : :
130 : : do {
131 : 3 : if (start == 0)
132 : 3 : id = find_first_zero_bit(mc_groups,
133 : : mc_groups_longs *
134 : : BITS_PER_LONG);
135 : : else
136 : 0 : id = find_next_zero_bit(mc_groups,
137 : : mc_groups_longs * BITS_PER_LONG,
138 : : start);
139 : :
140 : : fits = true;
141 : 3 : for (i = id;
142 : 3 : i < min_t(int, id + n_groups,
143 : : mc_groups_longs * BITS_PER_LONG);
144 : 3 : i++) {
145 : 3 : if (test_bit(i, mc_groups)) {
146 : 0 : start = i;
147 : : fits = false;
148 : 0 : break;
149 : : }
150 : : }
151 : :
152 : 3 : if (id + n_groups > mc_groups_longs * BITS_PER_LONG) {
153 : 0 : unsigned long new_longs = mc_groups_longs +
154 : 0 : BITS_TO_LONGS(n_groups);
155 : 0 : size_t nlen = new_longs * sizeof(unsigned long);
156 : :
157 : 0 : if (mc_groups == &mc_group_start) {
158 : 0 : new_groups = kzalloc(nlen, GFP_KERNEL);
159 : 0 : if (!new_groups)
160 : : return -ENOMEM;
161 : 0 : mc_groups = new_groups;
162 : 0 : *mc_groups = mc_group_start;
163 : : } else {
164 : 0 : new_groups = krealloc(mc_groups, nlen,
165 : : GFP_KERNEL);
166 : 0 : if (!new_groups)
167 : : return -ENOMEM;
168 : 0 : mc_groups = new_groups;
169 : 0 : for (i = 0; i < BITS_TO_LONGS(n_groups); i++)
170 : 0 : mc_groups[mc_groups_longs + i] = 0;
171 : : }
172 : 0 : mc_groups_longs = new_longs;
173 : : }
174 : 3 : } while (!fits);
175 : :
176 : 3 : for (i = id; i < id + n_groups; i++)
177 : 3 : set_bit(i, mc_groups);
178 : 3 : *first_id = id;
179 : 3 : return 0;
180 : : }
181 : :
182 : : static struct genl_family genl_ctrl;
183 : :
184 : 3 : static int genl_validate_assign_mc_groups(struct genl_family *family)
185 : : {
186 : : int first_id;
187 : 3 : int n_groups = family->n_mcgrps;
188 : : int err = 0, i;
189 : : bool groups_allocated = false;
190 : :
191 : 3 : if (!n_groups)
192 : : return 0;
193 : :
194 : 3 : for (i = 0; i < n_groups; i++) {
195 : 3 : const struct genl_multicast_group *grp = &family->mcgrps[i];
196 : :
197 : 3 : if (WARN_ON(grp->name[0] == '\0'))
198 : : return -EINVAL;
199 : 3 : if (WARN_ON(memchr(grp->name, '\0', GENL_NAMSIZ) == NULL))
200 : : return -EINVAL;
201 : : }
202 : :
203 : : /* special-case our own group and hacks */
204 : 3 : if (family == &genl_ctrl) {
205 : 3 : first_id = GENL_ID_CTRL;
206 : 3 : BUG_ON(n_groups != 1);
207 : 3 : } else if (strcmp(family->name, "NET_DM") == 0) {
208 : 0 : first_id = 1;
209 : 0 : BUG_ON(n_groups != 1);
210 : 3 : } else if (family->id == GENL_ID_VFS_DQUOT) {
211 : 0 : first_id = GENL_ID_VFS_DQUOT;
212 : 0 : BUG_ON(n_groups != 1);
213 : 3 : } else if (family->id == GENL_ID_PMCRAID) {
214 : 0 : first_id = GENL_ID_PMCRAID;
215 : 0 : BUG_ON(n_groups != 1);
216 : : } else {
217 : : groups_allocated = true;
218 : 3 : err = genl_allocate_reserve_groups(n_groups, &first_id);
219 : 3 : if (err)
220 : : return err;
221 : : }
222 : :
223 : 3 : family->mcgrp_offset = first_id;
224 : :
225 : : /* if still initializing, can't and don't need to to realloc bitmaps */
226 : 3 : if (!init_net.genl_sock)
227 : : return 0;
228 : :
229 : 3 : if (family->netnsok) {
230 : : struct net *net;
231 : :
232 : 3 : netlink_table_grab();
233 : : rcu_read_lock();
234 : 3 : for_each_net_rcu(net) {
235 : 3 : err = __netlink_change_ngroups(net->genl_sock,
236 : 3 : mc_groups_longs * BITS_PER_LONG);
237 : 3 : if (err) {
238 : : /*
239 : : * No need to roll back, can only fail if
240 : : * memory allocation fails and then the
241 : : * number of _possible_ groups has been
242 : : * increased on some sockets which is ok.
243 : : */
244 : : break;
245 : : }
246 : : }
247 : : rcu_read_unlock();
248 : 3 : netlink_table_ungrab();
249 : : } else {
250 : 3 : err = netlink_change_ngroups(init_net.genl_sock,
251 : 3 : mc_groups_longs * BITS_PER_LONG);
252 : : }
253 : :
254 : 3 : if (groups_allocated && err) {
255 : 0 : for (i = 0; i < family->n_mcgrps; i++)
256 : 0 : clear_bit(family->mcgrp_offset + i, mc_groups);
257 : : }
258 : :
259 : 3 : return err;
260 : : }
261 : :
262 : 0 : static void genl_unregister_mc_groups(const struct genl_family *family)
263 : : {
264 : : struct net *net;
265 : : int i;
266 : :
267 : 0 : netlink_table_grab();
268 : : rcu_read_lock();
269 : 0 : for_each_net_rcu(net) {
270 : 0 : for (i = 0; i < family->n_mcgrps; i++)
271 : 0 : __netlink_clear_multicast_users(
272 : 0 : net->genl_sock, family->mcgrp_offset + i);
273 : : }
274 : : rcu_read_unlock();
275 : 0 : netlink_table_ungrab();
276 : :
277 : 0 : for (i = 0; i < family->n_mcgrps; i++) {
278 : 0 : int grp_id = family->mcgrp_offset + i;
279 : :
280 : 0 : if (grp_id != 1)
281 : 0 : clear_bit(grp_id, mc_groups);
282 : 0 : genl_ctrl_event(CTRL_CMD_DELMCAST_GRP, family,
283 : 0 : &family->mcgrps[i], grp_id);
284 : : }
285 : 0 : }
286 : :
287 : 3 : static int genl_validate_ops(const struct genl_family *family)
288 : : {
289 : 3 : const struct genl_ops *ops = family->ops;
290 : 3 : unsigned int n_ops = family->n_ops;
291 : : int i, j;
292 : :
293 : 3 : if (WARN_ON(n_ops && !ops))
294 : : return -EINVAL;
295 : :
296 : 3 : if (!n_ops)
297 : : return 0;
298 : :
299 : 3 : for (i = 0; i < n_ops; i++) {
300 : 3 : if (ops[i].dumpit == NULL && ops[i].doit == NULL)
301 : : return -EINVAL;
302 : 3 : for (j = i + 1; j < n_ops; j++)
303 : 3 : if (ops[i].cmd == ops[j].cmd)
304 : : return -EINVAL;
305 : : }
306 : :
307 : : return 0;
308 : : }
309 : :
310 : : /**
311 : : * genl_register_family - register a generic netlink family
312 : : * @family: generic netlink family
313 : : *
314 : : * Registers the specified family after validating it first. Only one
315 : : * family may be registered with the same family name or identifier.
316 : : *
317 : : * The family's ops, multicast groups and module pointer must already
318 : : * be assigned.
319 : : *
320 : : * Return 0 on success or a negative error code.
321 : : */
322 : 3 : int genl_register_family(struct genl_family *family)
323 : : {
324 : : int err, i;
325 : : int start = GENL_START_ALLOC, end = GENL_MAX_ID;
326 : :
327 : 3 : err = genl_validate_ops(family);
328 : 3 : if (err)
329 : : return err;
330 : :
331 : : genl_lock_all();
332 : :
333 : 3 : if (genl_family_find_byname(family->name)) {
334 : : err = -EEXIST;
335 : : goto errout_locked;
336 : : }
337 : :
338 : : /*
339 : : * Sadly, a few cases need to be special-cased
340 : : * due to them having previously abused the API
341 : : * and having used their family ID also as their
342 : : * multicast group ID, so we use reserved IDs
343 : : * for both to be sure we can do that mapping.
344 : : */
345 : 3 : if (family == &genl_ctrl) {
346 : : /* and this needs to be special for initial family lookups */
347 : : start = end = GENL_ID_CTRL;
348 : 3 : } else if (strcmp(family->name, "pmcraid") == 0) {
349 : : start = end = GENL_ID_PMCRAID;
350 : 3 : } else if (strcmp(family->name, "VFS_DQUOT") == 0) {
351 : : start = end = GENL_ID_VFS_DQUOT;
352 : : }
353 : :
354 : 3 : if (family->maxattr && !family->parallel_ops) {
355 : 3 : family->attrbuf = kmalloc_array(family->maxattr + 1,
356 : : sizeof(struct nlattr *),
357 : : GFP_KERNEL);
358 : 3 : if (family->attrbuf == NULL) {
359 : : err = -ENOMEM;
360 : : goto errout_locked;
361 : : }
362 : : } else
363 : 3 : family->attrbuf = NULL;
364 : :
365 : 3 : family->id = idr_alloc_cyclic(&genl_fam_idr, family,
366 : : start, end + 1, GFP_KERNEL);
367 : 3 : if (family->id < 0) {
368 : : err = family->id;
369 : : goto errout_free;
370 : : }
371 : :
372 : 3 : err = genl_validate_assign_mc_groups(family);
373 : 3 : if (err)
374 : : goto errout_remove;
375 : :
376 : : genl_unlock_all();
377 : :
378 : : /* send all events */
379 : 3 : genl_ctrl_event(CTRL_CMD_NEWFAMILY, family, NULL, 0);
380 : 3 : for (i = 0; i < family->n_mcgrps; i++)
381 : 3 : genl_ctrl_event(CTRL_CMD_NEWMCAST_GRP, family,
382 : 3 : &family->mcgrps[i], family->mcgrp_offset + i);
383 : :
384 : : return 0;
385 : :
386 : : errout_remove:
387 : 0 : idr_remove(&genl_fam_idr, family->id);
388 : : errout_free:
389 : 0 : kfree(family->attrbuf);
390 : : errout_locked:
391 : : genl_unlock_all();
392 : 0 : return err;
393 : : }
394 : : EXPORT_SYMBOL(genl_register_family);
395 : :
396 : : /**
397 : : * genl_unregister_family - unregister generic netlink family
398 : : * @family: generic netlink family
399 : : *
400 : : * Unregisters the specified family.
401 : : *
402 : : * Returns 0 on success or a negative error code.
403 : : */
404 : 0 : int genl_unregister_family(const struct genl_family *family)
405 : : {
406 : : genl_lock_all();
407 : :
408 : 0 : if (!genl_family_find_byid(family->id)) {
409 : : genl_unlock_all();
410 : 0 : return -ENOENT;
411 : : }
412 : :
413 : 0 : genl_unregister_mc_groups(family);
414 : :
415 : 0 : idr_remove(&genl_fam_idr, family->id);
416 : :
417 : 0 : up_write(&cb_lock);
418 : 0 : wait_event(genl_sk_destructing_waitq,
419 : : atomic_read(&genl_sk_destructing_cnt) == 0);
420 : : genl_unlock();
421 : :
422 : 0 : kfree(family->attrbuf);
423 : :
424 : 0 : genl_ctrl_event(CTRL_CMD_DELFAMILY, family, NULL, 0);
425 : :
426 : 0 : return 0;
427 : : }
428 : : EXPORT_SYMBOL(genl_unregister_family);
429 : :
430 : : /**
431 : : * genlmsg_put - Add generic netlink header to netlink message
432 : : * @skb: socket buffer holding the message
433 : : * @portid: netlink portid the message is addressed to
434 : : * @seq: sequence number (usually the one of the sender)
435 : : * @family: generic netlink family
436 : : * @flags: netlink message flags
437 : : * @cmd: generic netlink command
438 : : *
439 : : * Returns pointer to user specific header
440 : : */
441 : 3 : void *genlmsg_put(struct sk_buff *skb, u32 portid, u32 seq,
442 : : const struct genl_family *family, int flags, u8 cmd)
443 : : {
444 : : struct nlmsghdr *nlh;
445 : : struct genlmsghdr *hdr;
446 : :
447 : 3 : nlh = nlmsg_put(skb, portid, seq, family->id, GENL_HDRLEN +
448 : 3 : family->hdrsize, flags);
449 : 3 : if (nlh == NULL)
450 : : return NULL;
451 : :
452 : : hdr = nlmsg_data(nlh);
453 : 3 : hdr->cmd = cmd;
454 : 3 : hdr->version = family->version;
455 : 3 : hdr->reserved = 0;
456 : :
457 : 3 : return (char *) hdr + GENL_HDRLEN;
458 : : }
459 : : EXPORT_SYMBOL(genlmsg_put);
460 : :
461 : 0 : static int genl_lock_start(struct netlink_callback *cb)
462 : : {
463 : : /* our ops are always const - netlink API doesn't propagate that */
464 : 0 : const struct genl_ops *ops = cb->data;
465 : : int rc = 0;
466 : :
467 : 0 : if (ops->start) {
468 : : genl_lock();
469 : 0 : rc = ops->start(cb);
470 : : genl_unlock();
471 : : }
472 : 0 : return rc;
473 : : }
474 : :
475 : 0 : static int genl_lock_dumpit(struct sk_buff *skb, struct netlink_callback *cb)
476 : : {
477 : : /* our ops are always const - netlink API doesn't propagate that */
478 : 0 : const struct genl_ops *ops = cb->data;
479 : : int rc;
480 : :
481 : : genl_lock();
482 : 0 : rc = ops->dumpit(skb, cb);
483 : : genl_unlock();
484 : 0 : return rc;
485 : : }
486 : :
487 : 0 : static int genl_lock_done(struct netlink_callback *cb)
488 : : {
489 : : /* our ops are always const - netlink API doesn't propagate that */
490 : 0 : const struct genl_ops *ops = cb->data;
491 : : int rc = 0;
492 : :
493 : 0 : if (ops->done) {
494 : : genl_lock();
495 : 0 : rc = ops->done(cb);
496 : : genl_unlock();
497 : : }
498 : 0 : return rc;
499 : : }
500 : :
501 : 3 : static int genl_family_rcv_msg(const struct genl_family *family,
502 : : struct sk_buff *skb,
503 : : struct nlmsghdr *nlh,
504 : : struct netlink_ext_ack *extack)
505 : : {
506 : : const struct genl_ops *ops;
507 : 3 : struct net *net = sock_net(skb->sk);
508 : : struct genl_info info;
509 : : struct genlmsghdr *hdr = nlmsg_data(nlh);
510 : : struct nlattr **attrbuf;
511 : : int hdrlen, err;
512 : :
513 : : /* this family doesn't exist in this netns */
514 : 3 : if (!family->netnsok && !net_eq(net, &init_net))
515 : : return -ENOENT;
516 : :
517 : 3 : hdrlen = GENL_HDRLEN + family->hdrsize;
518 : 3 : if (nlh->nlmsg_len < nlmsg_msg_size(hdrlen))
519 : : return -EINVAL;
520 : :
521 : 3 : ops = genl_get_cmd(hdr->cmd, family);
522 : 3 : if (ops == NULL)
523 : : return -EOPNOTSUPP;
524 : :
525 : 3 : if ((ops->flags & GENL_ADMIN_PERM) &&
526 : 0 : !netlink_capable(skb, CAP_NET_ADMIN))
527 : : return -EPERM;
528 : :
529 : 3 : if ((ops->flags & GENL_UNS_ADMIN_PERM) &&
530 : 0 : !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
531 : : return -EPERM;
532 : :
533 : 3 : if ((nlh->nlmsg_flags & NLM_F_DUMP) == NLM_F_DUMP) {
534 : : int rc;
535 : :
536 : 0 : if (ops->dumpit == NULL)
537 : : return -EOPNOTSUPP;
538 : :
539 : 0 : if (!(ops->validate & GENL_DONT_VALIDATE_DUMP)) {
540 : 0 : int hdrlen = GENL_HDRLEN + family->hdrsize;
541 : :
542 : 0 : if (nlh->nlmsg_len < nlmsg_msg_size(hdrlen))
543 : : return -EINVAL;
544 : :
545 : 0 : if (family->maxattr) {
546 : : unsigned int validate = NL_VALIDATE_STRICT;
547 : :
548 : 0 : if (ops->validate &
549 : : GENL_DONT_VALIDATE_DUMP_STRICT)
550 : : validate = NL_VALIDATE_LIBERAL;
551 : 0 : rc = __nla_validate(nlmsg_attrdata(nlh, hdrlen),
552 : : nlmsg_attrlen(nlh, hdrlen),
553 : : family->maxattr,
554 : : family->policy,
555 : : validate, extack);
556 : 0 : if (rc)
557 : : return rc;
558 : : }
559 : : }
560 : :
561 : 0 : if (!family->parallel_ops) {
562 : 0 : struct netlink_dump_control c = {
563 : 0 : .module = family->module,
564 : : /* we have const, but the netlink API doesn't */
565 : : .data = (void *)ops,
566 : : .start = genl_lock_start,
567 : : .dump = genl_lock_dumpit,
568 : : .done = genl_lock_done,
569 : : };
570 : :
571 : : genl_unlock();
572 : 0 : rc = __netlink_dump_start(net->genl_sock, skb, nlh, &c);
573 : : genl_lock();
574 : :
575 : : } else {
576 : 0 : struct netlink_dump_control c = {
577 : 0 : .module = family->module,
578 : 0 : .start = ops->start,
579 : 0 : .dump = ops->dumpit,
580 : 0 : .done = ops->done,
581 : : };
582 : :
583 : 0 : rc = __netlink_dump_start(net->genl_sock, skb, nlh, &c);
584 : : }
585 : :
586 : 0 : return rc;
587 : : }
588 : :
589 : 3 : if (ops->doit == NULL)
590 : : return -EOPNOTSUPP;
591 : :
592 : 3 : if (family->maxattr && family->parallel_ops) {
593 : 3 : attrbuf = kmalloc_array(family->maxattr + 1,
594 : : sizeof(struct nlattr *),
595 : : GFP_KERNEL);
596 : 3 : if (attrbuf == NULL)
597 : : return -ENOMEM;
598 : : } else
599 : 3 : attrbuf = family->attrbuf;
600 : :
601 : 3 : if (attrbuf) {
602 : : enum netlink_validation validate = NL_VALIDATE_STRICT;
603 : :
604 : 3 : if (ops->validate & GENL_DONT_VALIDATE_STRICT)
605 : : validate = NL_VALIDATE_LIBERAL;
606 : :
607 : 3 : err = __nlmsg_parse(nlh, hdrlen, attrbuf, family->maxattr,
608 : : family->policy, validate, extack);
609 : 3 : if (err < 0)
610 : : goto out;
611 : : }
612 : :
613 : 3 : info.snd_seq = nlh->nlmsg_seq;
614 : 3 : info.snd_portid = NETLINK_CB(skb).portid;
615 : 3 : info.nlhdr = nlh;
616 : 3 : info.genlhdr = nlmsg_data(nlh);
617 : 3 : info.userhdr = nlmsg_data(nlh) + GENL_HDRLEN;
618 : 3 : info.attrs = attrbuf;
619 : 3 : info.extack = extack;
620 : : genl_info_net_set(&info, net);
621 : 3 : memset(&info.user_ptr, 0, sizeof(info.user_ptr));
622 : :
623 : 3 : if (family->pre_doit) {
624 : 3 : err = family->pre_doit(ops, skb, &info);
625 : 3 : if (err)
626 : : goto out;
627 : : }
628 : :
629 : 3 : err = ops->doit(skb, &info);
630 : :
631 : 3 : if (family->post_doit)
632 : 0 : family->post_doit(ops, skb, &info);
633 : :
634 : : out:
635 : 3 : if (family->parallel_ops)
636 : 3 : kfree(attrbuf);
637 : :
638 : 3 : return err;
639 : : }
640 : :
641 : 3 : static int genl_rcv_msg(struct sk_buff *skb, struct nlmsghdr *nlh,
642 : : struct netlink_ext_ack *extack)
643 : : {
644 : : const struct genl_family *family;
645 : : int err;
646 : :
647 : 3 : family = genl_family_find_byid(nlh->nlmsg_type);
648 : 3 : if (family == NULL)
649 : : return -ENOENT;
650 : :
651 : 3 : if (!family->parallel_ops)
652 : : genl_lock();
653 : :
654 : 3 : err = genl_family_rcv_msg(family, skb, nlh, extack);
655 : :
656 : 3 : if (!family->parallel_ops)
657 : : genl_unlock();
658 : :
659 : 3 : return err;
660 : : }
661 : :
662 : 3 : static void genl_rcv(struct sk_buff *skb)
663 : : {
664 : 3 : down_read(&cb_lock);
665 : 3 : netlink_rcv_skb(skb, &genl_rcv_msg);
666 : 3 : up_read(&cb_lock);
667 : 3 : }
668 : :
669 : : /**************************************************************************
670 : : * Controller
671 : : **************************************************************************/
672 : :
673 : : static struct genl_family genl_ctrl;
674 : :
675 : 3 : static int ctrl_fill_info(const struct genl_family *family, u32 portid, u32 seq,
676 : : u32 flags, struct sk_buff *skb, u8 cmd)
677 : : {
678 : : void *hdr;
679 : :
680 : 3 : hdr = genlmsg_put(skb, portid, seq, &genl_ctrl, flags, cmd);
681 : 3 : if (hdr == NULL)
682 : : return -1;
683 : :
684 : 3 : if (nla_put_string(skb, CTRL_ATTR_FAMILY_NAME, family->name) ||
685 : 3 : nla_put_u16(skb, CTRL_ATTR_FAMILY_ID, family->id) ||
686 : 3 : nla_put_u32(skb, CTRL_ATTR_VERSION, family->version) ||
687 : 3 : nla_put_u32(skb, CTRL_ATTR_HDRSIZE, family->hdrsize) ||
688 : 3 : nla_put_u32(skb, CTRL_ATTR_MAXATTR, family->maxattr))
689 : : goto nla_put_failure;
690 : :
691 : 3 : if (family->n_ops) {
692 : : struct nlattr *nla_ops;
693 : : int i;
694 : :
695 : : nla_ops = nla_nest_start_noflag(skb, CTRL_ATTR_OPS);
696 : 3 : if (nla_ops == NULL)
697 : : goto nla_put_failure;
698 : :
699 : 3 : for (i = 0; i < family->n_ops; i++) {
700 : : struct nlattr *nest;
701 : 3 : const struct genl_ops *ops = &family->ops[i];
702 : 3 : u32 op_flags = ops->flags;
703 : :
704 : 3 : if (ops->dumpit)
705 : 3 : op_flags |= GENL_CMD_CAP_DUMP;
706 : 3 : if (ops->doit)
707 : 3 : op_flags |= GENL_CMD_CAP_DO;
708 : 3 : if (family->policy)
709 : 3 : op_flags |= GENL_CMD_CAP_HASPOL;
710 : :
711 : 3 : nest = nla_nest_start_noflag(skb, i + 1);
712 : 3 : if (nest == NULL)
713 : : goto nla_put_failure;
714 : :
715 : 3 : if (nla_put_u32(skb, CTRL_ATTR_OP_ID, ops->cmd) ||
716 : : nla_put_u32(skb, CTRL_ATTR_OP_FLAGS, op_flags))
717 : : goto nla_put_failure;
718 : :
719 : : nla_nest_end(skb, nest);
720 : : }
721 : :
722 : : nla_nest_end(skb, nla_ops);
723 : : }
724 : :
725 : 3 : if (family->n_mcgrps) {
726 : : struct nlattr *nla_grps;
727 : : int i;
728 : :
729 : : nla_grps = nla_nest_start_noflag(skb, CTRL_ATTR_MCAST_GROUPS);
730 : 3 : if (nla_grps == NULL)
731 : : goto nla_put_failure;
732 : :
733 : 3 : for (i = 0; i < family->n_mcgrps; i++) {
734 : : struct nlattr *nest;
735 : : const struct genl_multicast_group *grp;
736 : :
737 : 3 : grp = &family->mcgrps[i];
738 : :
739 : 3 : nest = nla_nest_start_noflag(skb, i + 1);
740 : 3 : if (nest == NULL)
741 : : goto nla_put_failure;
742 : :
743 : 3 : if (nla_put_u32(skb, CTRL_ATTR_MCAST_GRP_ID,
744 : 3 : family->mcgrp_offset + i) ||
745 : 3 : nla_put_string(skb, CTRL_ATTR_MCAST_GRP_NAME,
746 : 3 : grp->name))
747 : : goto nla_put_failure;
748 : :
749 : : nla_nest_end(skb, nest);
750 : : }
751 : : nla_nest_end(skb, nla_grps);
752 : : }
753 : :
754 : : genlmsg_end(skb, hdr);
755 : 3 : return 0;
756 : :
757 : : nla_put_failure:
758 : : genlmsg_cancel(skb, hdr);
759 : : return -EMSGSIZE;
760 : : }
761 : :
762 : 3 : static int ctrl_fill_mcgrp_info(const struct genl_family *family,
763 : : const struct genl_multicast_group *grp,
764 : : int grp_id, u32 portid, u32 seq, u32 flags,
765 : : struct sk_buff *skb, u8 cmd)
766 : : {
767 : : void *hdr;
768 : : struct nlattr *nla_grps;
769 : : struct nlattr *nest;
770 : :
771 : 3 : hdr = genlmsg_put(skb, portid, seq, &genl_ctrl, flags, cmd);
772 : 3 : if (hdr == NULL)
773 : : return -1;
774 : :
775 : 3 : if (nla_put_string(skb, CTRL_ATTR_FAMILY_NAME, family->name) ||
776 : 3 : nla_put_u16(skb, CTRL_ATTR_FAMILY_ID, family->id))
777 : : goto nla_put_failure;
778 : :
779 : : nla_grps = nla_nest_start_noflag(skb, CTRL_ATTR_MCAST_GROUPS);
780 : 3 : if (nla_grps == NULL)
781 : : goto nla_put_failure;
782 : :
783 : : nest = nla_nest_start_noflag(skb, 1);
784 : 3 : if (nest == NULL)
785 : : goto nla_put_failure;
786 : :
787 : 3 : if (nla_put_u32(skb, CTRL_ATTR_MCAST_GRP_ID, grp_id) ||
788 : 3 : nla_put_string(skb, CTRL_ATTR_MCAST_GRP_NAME,
789 : 3 : grp->name))
790 : : goto nla_put_failure;
791 : :
792 : : nla_nest_end(skb, nest);
793 : : nla_nest_end(skb, nla_grps);
794 : :
795 : : genlmsg_end(skb, hdr);
796 : 3 : return 0;
797 : :
798 : : nla_put_failure:
799 : : genlmsg_cancel(skb, hdr);
800 : : return -EMSGSIZE;
801 : : }
802 : :
803 : 0 : static int ctrl_dumpfamily(struct sk_buff *skb, struct netlink_callback *cb)
804 : : {
805 : : int n = 0;
806 : : struct genl_family *rt;
807 : 0 : struct net *net = sock_net(skb->sk);
808 : 0 : int fams_to_skip = cb->args[0];
809 : : unsigned int id;
810 : :
811 : 0 : idr_for_each_entry(&genl_fam_idr, rt, id) {
812 : 0 : if (!rt->netnsok && !net_eq(net, &init_net))
813 : 0 : continue;
814 : :
815 : 0 : if (n++ < fams_to_skip)
816 : 0 : continue;
817 : :
818 : 0 : if (ctrl_fill_info(rt, NETLINK_CB(cb->skb).portid,
819 : 0 : cb->nlh->nlmsg_seq, NLM_F_MULTI,
820 : : skb, CTRL_CMD_NEWFAMILY) < 0) {
821 : : n--;
822 : : break;
823 : : }
824 : : }
825 : :
826 : 0 : cb->args[0] = n;
827 : 0 : return skb->len;
828 : : }
829 : :
830 : 3 : static struct sk_buff *ctrl_build_family_msg(const struct genl_family *family,
831 : : u32 portid, int seq, u8 cmd)
832 : : {
833 : : struct sk_buff *skb;
834 : : int err;
835 : :
836 : : skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
837 : 3 : if (skb == NULL)
838 : : return ERR_PTR(-ENOBUFS);
839 : :
840 : 3 : err = ctrl_fill_info(family, portid, seq, 0, skb, cmd);
841 : 3 : if (err < 0) {
842 : : nlmsg_free(skb);
843 : 0 : return ERR_PTR(err);
844 : : }
845 : :
846 : : return skb;
847 : : }
848 : :
849 : : static struct sk_buff *
850 : 3 : ctrl_build_mcgrp_msg(const struct genl_family *family,
851 : : const struct genl_multicast_group *grp,
852 : : int grp_id, u32 portid, int seq, u8 cmd)
853 : : {
854 : : struct sk_buff *skb;
855 : : int err;
856 : :
857 : : skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
858 : 3 : if (skb == NULL)
859 : : return ERR_PTR(-ENOBUFS);
860 : :
861 : 3 : err = ctrl_fill_mcgrp_info(family, grp, grp_id, portid,
862 : : seq, 0, skb, cmd);
863 : 3 : if (err < 0) {
864 : : nlmsg_free(skb);
865 : 0 : return ERR_PTR(err);
866 : : }
867 : :
868 : : return skb;
869 : : }
870 : :
871 : : static const struct nla_policy ctrl_policy[CTRL_ATTR_MAX+1] = {
872 : : [CTRL_ATTR_FAMILY_ID] = { .type = NLA_U16 },
873 : : [CTRL_ATTR_FAMILY_NAME] = { .type = NLA_NUL_STRING,
874 : : .len = GENL_NAMSIZ - 1 },
875 : : };
876 : :
877 : 3 : static int ctrl_getfamily(struct sk_buff *skb, struct genl_info *info)
878 : : {
879 : : struct sk_buff *msg;
880 : : const struct genl_family *res = NULL;
881 : : int err = -EINVAL;
882 : :
883 : 3 : if (info->attrs[CTRL_ATTR_FAMILY_ID]) {
884 : : u16 id = nla_get_u16(info->attrs[CTRL_ATTR_FAMILY_ID]);
885 : 0 : res = genl_family_find_byid(id);
886 : : err = -ENOENT;
887 : : }
888 : :
889 : 3 : if (info->attrs[CTRL_ATTR_FAMILY_NAME]) {
890 : : char *name;
891 : :
892 : : name = nla_data(info->attrs[CTRL_ATTR_FAMILY_NAME]);
893 : 3 : res = genl_family_find_byname(name);
894 : : #ifdef CONFIG_MODULES
895 : 3 : if (res == NULL) {
896 : : genl_unlock();
897 : 3 : up_read(&cb_lock);
898 : 3 : request_module("net-pf-%d-proto-%d-family-%s",
899 : : PF_NETLINK, NETLINK_GENERIC, name);
900 : 3 : down_read(&cb_lock);
901 : : genl_lock();
902 : 3 : res = genl_family_find_byname(name);
903 : : }
904 : : #endif
905 : : err = -ENOENT;
906 : : }
907 : :
908 : 3 : if (res == NULL)
909 : : return err;
910 : :
911 : 3 : if (!res->netnsok && !net_eq(genl_info_net(info), &init_net)) {
912 : : /* family doesn't exist here */
913 : : return -ENOENT;
914 : : }
915 : :
916 : 3 : msg = ctrl_build_family_msg(res, info->snd_portid, info->snd_seq,
917 : : CTRL_CMD_NEWFAMILY);
918 : 3 : if (IS_ERR(msg))
919 : 0 : return PTR_ERR(msg);
920 : :
921 : 3 : return genlmsg_reply(msg, info);
922 : : }
923 : :
924 : 3 : static int genl_ctrl_event(int event, const struct genl_family *family,
925 : : const struct genl_multicast_group *grp,
926 : : int grp_id)
927 : : {
928 : : struct sk_buff *msg;
929 : :
930 : : /* genl is still initialising */
931 : 3 : if (!init_net.genl_sock)
932 : : return 0;
933 : :
934 : 3 : switch (event) {
935 : : case CTRL_CMD_NEWFAMILY:
936 : : case CTRL_CMD_DELFAMILY:
937 : 3 : WARN_ON(grp);
938 : 3 : msg = ctrl_build_family_msg(family, 0, 0, event);
939 : 3 : break;
940 : : case CTRL_CMD_NEWMCAST_GRP:
941 : : case CTRL_CMD_DELMCAST_GRP:
942 : 3 : BUG_ON(!grp);
943 : 3 : msg = ctrl_build_mcgrp_msg(family, grp, grp_id, 0, 0, event);
944 : 3 : break;
945 : : default:
946 : : return -EINVAL;
947 : : }
948 : :
949 : 3 : if (IS_ERR(msg))
950 : 0 : return PTR_ERR(msg);
951 : :
952 : 3 : if (!family->netnsok) {
953 : 3 : genlmsg_multicast_netns(&genl_ctrl, &init_net, msg, 0,
954 : : 0, GFP_KERNEL);
955 : : } else {
956 : : rcu_read_lock();
957 : 3 : genlmsg_multicast_allns(&genl_ctrl, msg, 0,
958 : : 0, GFP_ATOMIC);
959 : : rcu_read_unlock();
960 : : }
961 : :
962 : : return 0;
963 : : }
964 : :
965 : : static const struct genl_ops genl_ctrl_ops[] = {
966 : : {
967 : : .cmd = CTRL_CMD_GETFAMILY,
968 : : .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
969 : : .doit = ctrl_getfamily,
970 : : .dumpit = ctrl_dumpfamily,
971 : : },
972 : : };
973 : :
974 : : static const struct genl_multicast_group genl_ctrl_groups[] = {
975 : : { .name = "notify", },
976 : : };
977 : :
978 : : static struct genl_family genl_ctrl __ro_after_init = {
979 : : .module = THIS_MODULE,
980 : : .ops = genl_ctrl_ops,
981 : : .n_ops = ARRAY_SIZE(genl_ctrl_ops),
982 : : .mcgrps = genl_ctrl_groups,
983 : : .n_mcgrps = ARRAY_SIZE(genl_ctrl_groups),
984 : : .id = GENL_ID_CTRL,
985 : : .name = "nlctrl",
986 : : .version = 0x2,
987 : : .maxattr = CTRL_ATTR_MAX,
988 : : .policy = ctrl_policy,
989 : : .netnsok = true,
990 : : };
991 : :
992 : 3 : static int __net_init genl_pernet_init(struct net *net)
993 : : {
994 : 3 : struct netlink_kernel_cfg cfg = {
995 : : .input = genl_rcv,
996 : : .flags = NL_CFG_F_NONROOT_RECV,
997 : : };
998 : :
999 : : /* we'll bump the group number right afterwards */
1000 : 3 : net->genl_sock = netlink_kernel_create(net, NETLINK_GENERIC, &cfg);
1001 : :
1002 : 3 : if (!net->genl_sock && net_eq(net, &init_net))
1003 : 0 : panic("GENL: Cannot initialize generic netlink\n");
1004 : :
1005 : 3 : if (!net->genl_sock)
1006 : : return -ENOMEM;
1007 : :
1008 : 3 : return 0;
1009 : : }
1010 : :
1011 : 1 : static void __net_exit genl_pernet_exit(struct net *net)
1012 : : {
1013 : 1 : netlink_kernel_release(net->genl_sock);
1014 : 1 : net->genl_sock = NULL;
1015 : 1 : }
1016 : :
1017 : : static struct pernet_operations genl_pernet_ops = {
1018 : : .init = genl_pernet_init,
1019 : : .exit = genl_pernet_exit,
1020 : : };
1021 : :
1022 : 3 : static int __init genl_init(void)
1023 : : {
1024 : : int err;
1025 : :
1026 : 3 : err = genl_register_family(&genl_ctrl);
1027 : 3 : if (err < 0)
1028 : : goto problem;
1029 : :
1030 : 3 : err = register_pernet_subsys(&genl_pernet_ops);
1031 : 3 : if (err)
1032 : : goto problem;
1033 : :
1034 : 3 : return 0;
1035 : :
1036 : : problem:
1037 : 0 : panic("GENL: Cannot register controller: %d\n", err);
1038 : : }
1039 : :
1040 : : subsys_initcall(genl_init);
1041 : :
1042 : : /**
1043 : : * genl_family_attrbuf - return family's attrbuf
1044 : : * @family: the family
1045 : : *
1046 : : * Return the family's attrbuf, while validating that it's
1047 : : * actually valid to access it.
1048 : : *
1049 : : * You cannot use this function with a family that has parallel_ops
1050 : : * and you can only use it within (pre/post) doit/dumpit callbacks.
1051 : : */
1052 : 0 : struct nlattr **genl_family_attrbuf(const struct genl_family *family)
1053 : : {
1054 : 0 : if (!WARN_ON(family->parallel_ops))
1055 : : lockdep_assert_held(&genl_mutex);
1056 : :
1057 : 0 : return family->attrbuf;
1058 : : }
1059 : : EXPORT_SYMBOL(genl_family_attrbuf);
1060 : :
1061 : 3 : static int genlmsg_mcast(struct sk_buff *skb, u32 portid, unsigned long group,
1062 : : gfp_t flags)
1063 : : {
1064 : : struct sk_buff *tmp;
1065 : : struct net *net, *prev = NULL;
1066 : : bool delivered = false;
1067 : : int err;
1068 : :
1069 : 3 : for_each_net_rcu(net) {
1070 : 3 : if (prev) {
1071 : 0 : tmp = skb_clone(skb, flags);
1072 : 0 : if (!tmp) {
1073 : : err = -ENOMEM;
1074 : : goto error;
1075 : : }
1076 : 0 : err = nlmsg_multicast(prev->genl_sock, tmp,
1077 : : portid, group, flags);
1078 : 0 : if (!err)
1079 : : delivered = true;
1080 : 0 : else if (err != -ESRCH)
1081 : : goto error;
1082 : : }
1083 : :
1084 : : prev = net;
1085 : : }
1086 : :
1087 : 3 : err = nlmsg_multicast(prev->genl_sock, skb, portid, group, flags);
1088 : 3 : if (!err)
1089 : : delivered = true;
1090 : 3 : else if (err != -ESRCH)
1091 : : return err;
1092 : 3 : return delivered ? 0 : -ESRCH;
1093 : : error:
1094 : 0 : kfree_skb(skb);
1095 : 0 : return err;
1096 : : }
1097 : :
1098 : 3 : int genlmsg_multicast_allns(const struct genl_family *family,
1099 : : struct sk_buff *skb, u32 portid,
1100 : : unsigned int group, gfp_t flags)
1101 : : {
1102 : 3 : if (WARN_ON_ONCE(group >= family->n_mcgrps))
1103 : : return -EINVAL;
1104 : 3 : group = family->mcgrp_offset + group;
1105 : 3 : return genlmsg_mcast(skb, portid, group, flags);
1106 : : }
1107 : : EXPORT_SYMBOL(genlmsg_multicast_allns);
1108 : :
1109 : 0 : void genl_notify(const struct genl_family *family, struct sk_buff *skb,
1110 : : struct genl_info *info, u32 group, gfp_t flags)
1111 : : {
1112 : : struct net *net = genl_info_net(info);
1113 : 0 : struct sock *sk = net->genl_sock;
1114 : : int report = 0;
1115 : :
1116 : 0 : if (info->nlhdr)
1117 : : report = nlmsg_report(info->nlhdr);
1118 : :
1119 : 0 : if (WARN_ON_ONCE(group >= family->n_mcgrps))
1120 : 0 : return;
1121 : 0 : group = family->mcgrp_offset + group;
1122 : 0 : nlmsg_notify(sk, skb, info->snd_portid, group, report, flags);
1123 : : }
1124 : : EXPORT_SYMBOL(genl_notify);
|
