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(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                 rht_for_each_entry(nlsk, htbl->buckets[i], ht, node) {
 117                         sk = (struct sock *)nlsk;
 118
 119                         if (!net_eq(sock_net(sk), net))
 120                                 continue;
 121                         if (num < s_num) {
 122                                 num++;
 123                                 continue;
 124                         }
 125
 126                         if (sk_diag_fill(sk, skb, req,
 127                                          NETLINK_CB(cb->skb).portid,
 128                                          cb->nlh->nlmsg_seq,
 129                                          NLM_F_MULTI,
 130                                          sock_i_ino(sk)) < 0) {
 131                                 ret = 1;
 132                                 goto done;
 133                         }
 134
 135                         num++;
 136                 }
 137         }
 138
 139         sk_for_each_bound(sk, &tbl->mc_list) {
 140                 if (sk_hashed(sk))
 141                         continue;
 142                 if (!net_eq(sock_net(sk), net))
 143                         continue;
 144                 if (num < s_num) {
 145                         num++;
 146                         continue;
 147                 }
 148
 149                 if (sk_diag_fill(sk, skb, req,
 150                                  NETLINK_CB(cb->skb).portid,
 151                                  cb->nlh->nlmsg_seq,
 152                                  NLM_F_MULTI,
 153                                  sock_i_ino(sk)) < 0) {
 154                         ret = 1;
 155                         goto done;
 156                 }
 157                 num++;
 158         }
 159 done:
 160         cb->args[0] = num;
 161         cb->args[1] = protocol;
 162
 163         return ret;
 164 }
 165
 166 static int netlink_diag_dump(struct sk_buff *skb, struct netlink_callback *cb)
 167 {
 168         struct netlink_diag_req *req;
 169         int s_num = cb->args[0];
 170
 171         req = nlmsg_data(cb->nlh);
 172
 173         mutex_lock(&nl_sk_hash_lock);
 174         read_lock(&nl_table_lock);
 175
 176         if (req->sdiag_protocol == NDIAG_PROTO_ALL) {
 177                 int i;
 178
 179                 for (i = cb->args[1]; i < MAX_LINKS; i++) {
 180                         if (__netlink_diag_dump(skb, cb, i, s_num))
 181                                 break;
 182                         s_num = 0;
 183                 }
 184         } else {
 185                 if (req->sdiag_protocol >= MAX_LINKS) {
 186                         read_unlock(&nl_table_lock);
 187                         mutex_unlock(&nl_sk_hash_lock);
 188                         return -ENOENT;
 189                 }
 190
 191                 __netlink_diag_dump(skb, cb, req->sdiag_protocol, s_num);
 192         }
 193
 194         read_unlock(&nl_table_lock);
 195         mutex_unlock(&nl_sk_hash_lock);
 196
 197         return skb->len;
 198 }
 199
 200 static int netlink_diag_handler_dump(struct sk_buff *skb, struct nlmsghdr *h)
 201 {
 202         int hdrlen = sizeof(struct netlink_diag_req);
 203         struct net *net = sock_net(skb->sk);
 204
 205         if (nlmsg_len(h) < hdrlen)
 206                 return -EINVAL;
 207
 208         if (h->nlmsg_flags & NLM_F_DUMP) {
 209                 struct netlink_dump_control c = {
 210                         .dump = netlink_diag_dump,
 211                 };
 212                 return netlink_dump_start(net->diag_nlsk, skb, h, &c);
 213         } else
 214                 return -EOPNOTSUPP;
 215 }
 216
 217 static const struct sock_diag_handler netlink_diag_handler = {
 218         .family = AF_NETLINK,
 219         .dump = netlink_diag_handler_dump,
 220 };
 221
 222 static int __init netlink_diag_init(void)
 223 {
 224         return sock_diag_register(&netlink_diag_handler);
 225 }
 226
 227 static void __exit netlink_diag_exit(void)
 228 {
 229         sock_diag_unregister(&netlink_diag_handler);
 230 }
 231
 232 module_init(netlink_diag_init);
 233 module_exit(netlink_diag_exit);
 234 MODULE_LICENSE("GPL");
 235 MODULE_ALIAS_NET_PF_PROTO_TYPE(PF_NETLINK, NETLINK_SOCK_DIAG, 16 /* AF_NETLINK */);