lin
2025-08-14 dae8bad597b6607a449b32bf76c523423f7720ed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
/*
 * This ought to be provided by libnl
 */
 
#include <asm/errno.h>
#include <netlink/genl/genl.h>
#include <netlink/genl/family.h>
#include <netlink/genl/ctrl.h>
#include <netlink/msg.h>
#include <netlink/attr.h>
#include <linux/genetlink.h>
 
#include "iw.h"
 
static int error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err,
            void *arg)
{
   int *ret = arg;
   *ret = err->error;
   return NL_STOP;
}
 
static int ack_handler(struct nl_msg *msg, void *arg)
{
   int *ret = arg;
   *ret = 0;
   return NL_STOP;
}
 
struct handler_args {
   const char *group;
   int id;
};
 
static int family_handler(struct nl_msg *msg, void *arg)
{
   struct handler_args *grp = arg;
   struct nlattr *tb[CTRL_ATTR_MAX + 1];
   struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
   struct nlattr *mcgrp;
   int rem_mcgrp;
 
   nla_parse(tb, CTRL_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
         genlmsg_attrlen(gnlh, 0), NULL);
 
   if (!tb[CTRL_ATTR_MCAST_GROUPS])
       return NL_SKIP;
 
   nla_for_each_nested(mcgrp, tb[CTRL_ATTR_MCAST_GROUPS], rem_mcgrp) {
       struct nlattr *tb_mcgrp[CTRL_ATTR_MCAST_GRP_MAX + 1];
 
       nla_parse(tb_mcgrp, CTRL_ATTR_MCAST_GRP_MAX,
             nla_data(mcgrp), nla_len(mcgrp), NULL);
 
       if (!tb_mcgrp[CTRL_ATTR_MCAST_GRP_NAME] ||
           !tb_mcgrp[CTRL_ATTR_MCAST_GRP_ID])
           continue;
       if (strncmp(nla_data(tb_mcgrp[CTRL_ATTR_MCAST_GRP_NAME]),
               grp->group, nla_len(tb_mcgrp[CTRL_ATTR_MCAST_GRP_NAME])))
           continue;
       grp->id = nla_get_u32(tb_mcgrp[CTRL_ATTR_MCAST_GRP_ID]);
       break;
   }
 
   return NL_SKIP;
}
 
int nl_get_multicast_id(struct nl_sock *sock, const char *family, const char *group)
{
   struct nl_msg *msg;
   struct nl_cb *cb;
   int ret, ctrlid;
   struct handler_args grp = {
       .group = group,
       .id = -ENOENT,
   };
 
   msg = nlmsg_alloc();
   if (!msg)
       return -ENOMEM;
 
   cb = nl_cb_alloc(NL_CB_DEFAULT);
   if (!cb) {
       ret = -ENOMEM;
       goto out_fail_cb;
   }
 
   ctrlid = genl_ctrl_resolve(sock, "nlctrl");
 
   genlmsg_put(msg, 0, 0, ctrlid, 0,
           0, CTRL_CMD_GETFAMILY, 0);
 
   ret = -ENOBUFS;
   NLA_PUT_STRING(msg, CTRL_ATTR_FAMILY_NAME, family);
 
   ret = nl_send_auto_complete(sock, msg);
   if (ret < 0)
       goto out;
 
   ret = 1;
 
   nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &ret);
   nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &ret);
   nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, family_handler, &grp);
 
   while (ret > 0)
       nl_recvmsgs(sock, cb);
 
   if (ret == 0)
       ret = grp.id;
 nla_put_failure:
 out:
   nl_cb_put(cb);
 out_fail_cb:
   nlmsg_free(msg);
   return ret;
}