net/netlink/diag.c

   1 #include <linux/module.h>
   2
   3 #include <net/sock.h>
   4 #include <linux/netlink.h>
   5 #include <linux/sock_diag.h>
   6 #include <linux/netlink_diag.h>
   7 #include <linux/rhashtable.h>
   8
   9 #include "af_netlink.h"
  10
  11 #ifdef CONFIG_NETLINK_MMAP
  12 static int sk_diag_put_ring(struct netlink_ring *ring, int nl_type,
  13                             struct sk_buff *nlskb)
  14 {
  15         struct netlink_diag_ring ndr;
  16
  17         ndr.ndr_block_size = ring->pg_vec_pages << PAGE_SHIFT;
  18         ndr.ndr_block_nr   = ring->pg_vec_len;
  19         ndr.ndr_frame_size = ring->frame_size;
  20         ndr.ndr_frame_nr   = ring->frame_max + 1;
  21
  22         return nla_put(nlskb, nl_type, sizeof(ndr), &ndr);
  23 }
  24
  25 static int sk_diag_put_rings_cfg(struct sock *sk, struct sk_buff *nlskb)
  26 {
  27         struct netlink_sock *nlk = nlk_sk(sk);
  28         int ret;
  29
  30         mutex_lock(&nlk->pg_vec_lock);
  31         ret = sk_diag_put_ring(&nlk->rx_ring, NETLINK_DIAG_RX_RING, nlskb);
  32         if (!ret)
  33                 ret = sk_diag_put_ring(&nlk->tx_ring, NETLINK_DIAG_TX_RING,
  34                                        nlskb);
  35         mutex_unlock(&nlk->pg_vec_lock);
  36
  37         return ret;
  38 }
  39 #else
  40 static int sk_diag_put_rings_cfg(struct sock *sk, struct sk_buff *nlskb)
  41 {
  42         return 0;
  43 }
  44 #endif
  45
  46 static int sk_diag_dump_groups(struct sock *sk, struct sk_buff *nlskb)
  47 {
  48         struct netlink_sock *nlk = nlk_sk(sk);
  49
  50         if (nlk->groups == NULL)
  51                 return 0;
  52
  53         return nla_put(nlskb, NETLINK_DIAG_GROUPS, NLGRPSZ(nlk->ngroups),
  54                        nlk->groups);
  55 }
  56
  57 static int sk_diag_fill(struct sock *sk, struct sk_buff *skb,
  58                         struct netlink_diag_req *req,
  59                         u32 portid, u32 seq, u32 flags, int sk_ino)
  60 {
  61         struct nlmsghdr *nlh;
  62         struct netlink_diag_msg *rep;
  63         struct netlink_sock *nlk = nlk_sk(sk);
  64
  65         nlh = nlmsg_put(skb, portid, seq, SOCK_DIAG_BY_FAMILY, sizeof(*rep),
  66                         flags);
  67         if (!nlh)
  68                 return -EMSGSIZE;
  69
  70         rep = nlmsg_data(nlh);
  71         rep->ndiag_family       = AF_NETLINK;
  72         rep->ndiag_type         = sk->sk_type;
  73         rep->ndiag_protocol     = sk->sk_protocol;
  74         rep->ndiag_state        = sk->sk_state;
  75
  76         rep->ndiag_ino          = sk_ino;
  77         rep->ndiag_portid       = nlk->portid;
  78         rep->ndiag_dst_portid   = nlk->dst_portid;
  79         rep->ndiag_dst_group    = nlk->dst_group;
  80         sock_diag_save_cookie(sk, rep->ndiag_cookie);
  81
  82         if ((req->ndiag_show & NDIAG_SHOW_GROUPS) &&
  83             sk_diag_dump_groups(sk, skb))
  84                 goto out_nlmsg_trim;
  85
  86         if ((req->ndiag_show & NDIAG_SHOW_MEMINFO) &&
  87             sock_diag_put_meminfo(sk, skb, NETLINK_DIAG_MEMINFO))
  88                 goto out_nlmsg_trim;
  89
  90         if ((req->ndiag_show & NDIAG_SHOW_RING_CFG) &&
  91             sk_diag_put_rings_cfg(sk, skb))
  92                 goto out_nlmsg_trim;
  93
  94         return nlmsg_end(skb, nlh);
  95
  96 out_nlmsg_trim:
  97         nlmsg_cancel(skb, nlh);
  98         return -EMSGSIZE;
  99 }
 100
 101 static int __netlink_diag_dump(struct sk_buff *skb, struct netlink_callback *cb,
 102                                 int protocol, int s_num)
 103 {
 104         struct netlink_table *tbl = &nl_table[protocol];
 105         struct rhashtable *ht = &tbl->hash;
 106         const struct bucket_table *htbl = rht_dereference_rcu(ht->tbl, ht);
 107         struct net *net = sock_net(skb->sk);
 108         struct netlink_diag_req *req;
 109         struct netlink_sock *nlsk;
 110         struct sock *sk;
 111         int ret = 0, num = 0, i;
 112
 113         req = nlmsg_data(cb->nlh);
 114
 115         for (i = 0; i < htbl->size; i++) {
 116                 struct rhash_head *pos;
 117
 118                 rht_for_each_entry_rcu(nlsk, pos, htbl, i, node) {
 119                         sk = (struct sock *)nlsk;
 120
 121                         if (!net_eq(sock_net(sk), net))
 122                                 continue;
 123                         if (num < s_num) {
 124                                 num++;
 125                                 continue;
 126                         }
 127
 128                         if (sk_diag_fill(sk, skb, req,
 129                                          NETLINK_CB(cb->skb).portid,
 130                                          cb->nlh->nlmsg_seq,
 131                                          NLM_F_MULTI,
 132                                          sock_i_ino(sk)) < 0) {
 133                                 ret = 1;
 134                                 goto done;
 135                         }
 136
 137                         num++;
 138                 }
 139         }
 140
 141         sk_for_each_bound(sk, &tbl->mc_list) {
 142                 if (sk_hashed(sk))
 143                         continue;
 144                 if (!net_eq(sock_net(sk), net))
 145                         continue;
 146                 if (num < s_num) {
 147                         num++;
 148                         continue;
 149                 }
 150
 151                 if (sk_diag_fill(sk, skb, req,
 152                                  NETLINK_CB(cb->skb).portid,
 153                                  cb->nlh->nlmsg_seq,
 154                                  NLM_F_MULTI,
 155                                  sock_i_ino(sk)) < 0) {
 156                         ret = 1;
 157                         goto done;
 158                 }
 159                 num++;
 160         }
 161 done:
 162         cb->args[0] = num;
 163         cb->args[1] = protocol;
 164
 165         return ret;
 166 }
 167
 168 static int netlink_diag_dump(struct sk_buff *skb, struct netlink_callback *cb)
 169 {
 170         struct netlink_diag_req *req;
 171         int s_num = cb->args[0];
 172
 173         req = nlmsg_data(cb->nlh);
 174
 175         rcu_read_lock();
 176         read_lock(&nl_table_lock);
 177
 178         if (req->sdiag_protocol == NDIAG_PROTO_ALL) {
 179                 int i;
 180
 181                 for (i = cb->args[1]; i < MAX_LINKS; i++) {
 182                         if (__netlink_diag_dump(skb, cb, i, s_num))
 183                                 break;
 184                         s_num = 0;
 185                 }
 186         } else {
 187                 if (req->sdiag_protocol >= MAX_LINKS) {
 188                         read_unlock(&nl_table_lock);
 189                         rcu_read_unlock();
 190                         return -ENOENT;
 191                 }
 192
 193                 __netlink_diag_dump(skb, cb, req->sdiag_protocol, s_num);
 194         }
 195
 196         read_unlock(&nl_table_lock);
 197         rcu_read_unlock();
 198
 199         return skb->len;
 200 }
 201
 202 static int netlink_diag_handler_dump(struct sk_buff *skb, struct nlmsghdr *h)
 203 {
 204         int hdrlen = sizeof(struct netlink_diag_req);
 205         struct net *net = sock_net(skb->sk);
 206
 207         if (nlmsg_len(h) < hdrlen)
 208                 return -EINVAL;
 209
 210         if (h->nlmsg_flags & NLM_F_DUMP) {
 211                 struct netlink_dump_control c = {
 212                         .dump = netlink_diag_dump,
 213                 };
 214                 return netlink_dump_start(net->diag_nlsk, skb, h, &c);
 215         } else
 216                 return -EOPNOTSUPP;
 217 }
 218
 219 static const struct sock_diag_handler netlink_diag_handler = {
 220         .family = AF_NETLINK,
 221         .dump = netlink_diag_handler_dump,
 222 };
 223
 224 static int __init netlink_diag_init(void)
 225 {
 226         return sock_diag_register(&netlink_diag_handler);
 227 }
 228
 229 static void __exit netlink_diag_exit(void)
 230 {
 231         sock_diag_unregister(&netlink_diag_handler);
 232 }
 233
 234 module_init(netlink_diag_init);
 235 module_exit(netlink_diag_exit);
 236 MODULE_LICENSE("GPL");
 237 MODULE_ALIAS_NET_PF_PROTO_TYPE(PF_NETLINK, NETLINK_SOCK_DIAG, 16 /* AF_NETLINK */);