1 files changed, 165 insertions, 0 deletions

diff --git a/net/tipc/node.c b/net/tipc/node.c
index 95cc78b51532..21974191e425 100644
--- a/net/tipc/node.c
+++ b/net/tipc/node.c
@@ -1928,3 +1928,168 @@ out:
 
         return skb->len;
 }
+
+int tipc_nl_node_set_monitor(struct sk_buff *skb, struct genl_info *info)
+{
+        struct nlattr *attrs[TIPC_NLA_MON_MAX + 1];
+        struct net *net = sock_net(skb->sk);
+        int err;
+
+        if (!info->attrs[TIPC_NLA_MON])
+                return -EINVAL;
+
+        err = nla_parse_nested(attrs, TIPC_NLA_MON_MAX,
+                               info->attrs[TIPC_NLA_MON],
+                               tipc_nl_monitor_policy);
+        if (err)
+                return err;
+
+        if (attrs[TIPC_NLA_MON_ACTIVATION_THRESHOLD]) {
+                u32 val;
+
+                val = nla_get_u32(attrs[TIPC_NLA_MON_ACTIVATION_THRESHOLD]);
+                err = tipc_nl_monitor_set_threshold(net, val);
+                if (err)
+                        return err;
+        }
+
+        return 0;
+}
+
+static int __tipc_nl_add_monitor_prop(struct net *net, struct tipc_nl_msg *msg)
+{
+        struct nlattr *attrs;
+        void *hdr;
+        u32 val;
+
+        hdr = genlmsg_put(msg->skb, msg->portid, msg->seq, &tipc_genl_family,
+                          0, TIPC_NL_MON_GET);
+        if (!hdr)
+                return -EMSGSIZE;
+
+        attrs = nla_nest_start(msg->skb, TIPC_NLA_MON);
+        if (!attrs)
+                goto msg_full;
+
+        val = tipc_nl_monitor_get_threshold(net);
+
+        if (nla_put_u32(msg->skb, TIPC_NLA_MON_ACTIVATION_THRESHOLD, val))
+                goto attr_msg_full;
+
+        nla_nest_end(msg->skb, attrs);
+        genlmsg_end(msg->skb, hdr);
+
+        return 0;
+
+attr_msg_full:
+        nla_nest_cancel(msg->skb, attrs);
+msg_full:
+        genlmsg_cancel(msg->skb, hdr);
+
+        return -EMSGSIZE;
+}
+
+int tipc_nl_node_get_monitor(struct sk_buff *skb, struct genl_info *info)
+{
+        struct net *net = sock_net(skb->sk);
+        struct tipc_nl_msg msg;
+        int err;
+
+        msg.skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
+        msg.portid = info->snd_portid;
+        msg.seq = info->snd_seq;
+
+        err = __tipc_nl_add_monitor_prop(net, &msg);
+        if (err) {
+                nlmsg_free(msg.skb);
+                return err;
+        }
+
+        return genlmsg_reply(msg.skb, info);
+}
+
+int tipc_nl_node_dump_monitor(struct sk_buff *skb, struct netlink_callback *cb)
+{
+        struct net *net = sock_net(skb->sk);
+        u32 prev_bearer = cb->args[0];
+        struct tipc_nl_msg msg;
+        int err;
+        int i;
+
+        if (prev_bearer == MAX_BEARERS)
+                return 0;
+
+        msg.skb = skb;
+        msg.portid = NETLINK_CB(cb->skb).portid;
+        msg.seq = cb->nlh->nlmsg_seq;
+
+        rtnl_lock();
+        for (i = prev_bearer; i < MAX_BEARERS; i++) {
+                prev_bearer = i;
+                err = __tipc_nl_add_monitor(net, &msg, prev_bearer);
+                if (err)
+                        goto out;
+        }
+
+out:
+        rtnl_unlock();
+        cb->args[0] = prev_bearer;
+
+        return skb->len;
+}
+
+int tipc_nl_node_dump_monitor_peer(struct sk_buff *skb,
+                                   struct netlink_callback *cb)
+{
+        struct net *net = sock_net(skb->sk);
+        u32 prev_node = cb->args[1];
+        u32 bearer_id = cb->args[2];
+        int done = cb->args[0];
+        struct tipc_nl_msg msg;
+        int err;
+
+        if (!prev_node) {
+                struct nlattr **attrs;
+                struct nlattr *mon[TIPC_NLA_MON_MAX + 1];
+
+                err = tipc_nlmsg_parse(cb->nlh, &attrs);
+                if (err)
+                        return err;
+
+                if (!attrs[TIPC_NLA_MON])
+                        return -EINVAL;
+
+                err = nla_parse_nested(mon, TIPC_NLA_MON_MAX,
+                                       attrs[TIPC_NLA_MON],
+                                       tipc_nl_monitor_policy);
+                if (err)
+                        return err;
+
+                if (!mon[TIPC_NLA_MON_REF])
+                        return -EINVAL;
+
+                bearer_id = nla_get_u32(mon[TIPC_NLA_MON_REF]);
+
+                if (bearer_id >= MAX_BEARERS)
+                        return -EINVAL;
+        }
+
+        if (done)
+                return 0;
+
+        msg.skb = skb;
+        msg.portid = NETLINK_CB(cb->skb).portid;
+        msg.seq = cb->nlh->nlmsg_seq;
+
+        rtnl_lock();
+        err = tipc_nl_add_monitor_peer(net, &msg, bearer_id, &prev_node);
+        if (!err)
+                done = 1;
+
+        rtnl_unlock();
+        cb->args[0] = done;
+        cb->args[1] = prev_node;
+        cb->args[2] = bearer_id;
+
+        return skb->len;
+}