| .. | .. |
|---|
| 1 | +// SPDX-License-Identifier: GPL-2.0-only |
|---|
| 1 | 2 | #include <linux/module.h> |
|---|
| 2 | 3 | |
|---|
| 3 | 4 | #include <linux/inet_diag.h> |
|---|
| .. | .. |
|---|
| 86 | 87 | return sk ? sk : ERR_PTR(-ENOENT); |
|---|
| 87 | 88 | } |
|---|
| 88 | 89 | |
|---|
| 89 | | -static int raw_diag_dump_one(struct sk_buff *in_skb, |
|---|
| 90 | | - const struct nlmsghdr *nlh, |
|---|
| 90 | +static int raw_diag_dump_one(struct netlink_callback *cb, |
|---|
| 91 | 91 | const struct inet_diag_req_v2 *r) |
|---|
| 92 | 92 | { |
|---|
| 93 | | - struct net *net = sock_net(in_skb->sk); |
|---|
| 93 | + struct sk_buff *in_skb = cb->skb; |
|---|
| 94 | 94 | struct sk_buff *rep; |
|---|
| 95 | 95 | struct sock *sk; |
|---|
| 96 | + struct net *net; |
|---|
| 96 | 97 | int err; |
|---|
| 97 | 98 | |
|---|
| 99 | + net = sock_net(in_skb->sk); |
|---|
| 98 | 100 | sk = raw_sock_get(net, r); |
|---|
| 99 | 101 | if (IS_ERR(sk)) |
|---|
| 100 | 102 | return PTR_ERR(sk); |
|---|
| .. | .. |
|---|
| 108 | 110 | return -ENOMEM; |
|---|
| 109 | 111 | } |
|---|
| 110 | 112 | |
|---|
| 111 | | - err = inet_sk_diag_fill(sk, NULL, rep, r, |
|---|
| 112 | | - sk_user_ns(NETLINK_CB(in_skb).sk), |
|---|
| 113 | | - NETLINK_CB(in_skb).portid, |
|---|
| 114 | | - nlh->nlmsg_seq, 0, nlh, |
|---|
| 113 | + err = inet_sk_diag_fill(sk, NULL, rep, cb, r, 0, |
|---|
| 115 | 114 | netlink_net_capable(in_skb, CAP_NET_ADMIN)); |
|---|
| 116 | 115 | sock_put(sk); |
|---|
| 117 | 116 | |
|---|
| .. | .. |
|---|
| 136 | 135 | if (!inet_diag_bc_sk(bc, sk)) |
|---|
| 137 | 136 | return 0; |
|---|
| 138 | 137 | |
|---|
| 139 | | - return inet_sk_diag_fill(sk, NULL, skb, r, |
|---|
| 140 | | - sk_user_ns(NETLINK_CB(cb->skb).sk), |
|---|
| 141 | | - NETLINK_CB(cb->skb).portid, |
|---|
| 142 | | - cb->nlh->nlmsg_seq, NLM_F_MULTI, |
|---|
| 143 | | - cb->nlh, net_admin); |
|---|
| 138 | + return inet_sk_diag_fill(sk, NULL, skb, cb, r, NLM_F_MULTI, net_admin); |
|---|
| 144 | 139 | } |
|---|
| 145 | 140 | |
|---|
| 146 | 141 | static void raw_diag_dump(struct sk_buff *skb, struct netlink_callback *cb, |
|---|
| 147 | | - const struct inet_diag_req_v2 *r, struct nlattr *bc) |
|---|
| 142 | + const struct inet_diag_req_v2 *r) |
|---|
| 148 | 143 | { |
|---|
| 149 | 144 | bool net_admin = netlink_net_capable(cb->skb, CAP_NET_ADMIN); |
|---|
| 150 | 145 | struct raw_hashinfo *hashinfo = raw_get_hashinfo(r); |
|---|
| 151 | 146 | struct net *net = sock_net(skb->sk); |
|---|
| 147 | + struct inet_diag_dump_data *cb_data; |
|---|
| 152 | 148 | int num, s_num, slot, s_slot; |
|---|
| 153 | 149 | struct sock *sk = NULL; |
|---|
| 150 | + struct nlattr *bc; |
|---|
| 154 | 151 | |
|---|
| 155 | 152 | if (IS_ERR(hashinfo)) |
|---|
| 156 | 153 | return; |
|---|
| 157 | 154 | |
|---|
| 155 | + cb_data = cb->data; |
|---|
| 156 | + bc = cb_data->inet_diag_nla_bc; |
|---|
| 158 | 157 | s_slot = cb->args[0]; |
|---|
| 159 | 158 | num = s_num = cb->args[1]; |
|---|
| 160 | 159 | |
|---|