[INET_DIAG]: Introduce sk_diag_fill

To be called from inet_diag_get_exact, also rename inet_diag_fill to
inet_csk_diag_fill, for consistency with inet_twsk_diag_fill.

Signed-off-by: Arnaldo Carvalho de Melo <acme@mandriva.com>
Signed-off-by: David S. Miller <davem@davemloft.net>
This commit is contained in:
Arnaldo Carvalho de Melo 2006-01-09 14:56:56 -08:00 committed by David S. Miller
parent c7d58aabdc
commit dff2c03534
1 changed files with 24 additions and 10 deletions

View File

@ -50,7 +50,8 @@ static struct sock *idiagnl;
#define INET_DIAG_PUT(skb, attrtype, attrlen) \ #define INET_DIAG_PUT(skb, attrtype, attrlen) \
RTA_DATA(__RTA_PUT(skb, attrtype, attrlen)) RTA_DATA(__RTA_PUT(skb, attrtype, attrlen))
static int inet_diag_fill(struct sk_buff *skb, struct sock *sk, static int inet_csk_diag_fill(struct sock *sk,
struct sk_buff *skb,
int ext, u32 pid, u32 seq, u16 nlmsg_flags, int ext, u32 pid, u32 seq, u16 nlmsg_flags,
const struct nlmsghdr *unlh) const struct nlmsghdr *unlh)
{ {
@ -212,6 +213,17 @@ nlmsg_failure:
return -1; return -1;
} }
static int sk_diag_fill(struct sock *sk, struct sk_buff *skb,
int ext, u32 pid, u32 seq, u16 nlmsg_flags,
const struct nlmsghdr *unlh)
{
if (sk->sk_state == TCP_TIME_WAIT)
return inet_twsk_diag_fill((struct inet_timewait_sock *)sk,
skb, ext, pid, seq, nlmsg_flags,
unlh);
return inet_csk_diag_fill(sk, skb, ext, pid, seq, nlmsg_flags, unlh);
}
static int inet_diag_get_exact(struct sk_buff *in_skb, static int inet_diag_get_exact(struct sk_buff *in_skb,
const struct nlmsghdr *nlh) const struct nlmsghdr *nlh)
{ {
@ -263,7 +275,7 @@ static int inet_diag_get_exact(struct sk_buff *in_skb,
if (!rep) if (!rep)
goto out; goto out;
if (inet_diag_fill(rep, sk, req->idiag_ext, if (sk_diag_fill(sk, rep, req->idiag_ext,
NETLINK_CB(in_skb).pid, NETLINK_CB(in_skb).pid,
nlh->nlmsg_seq, 0, nlh) <= 0) nlh->nlmsg_seq, 0, nlh) <= 0)
BUG(); BUG();
@ -442,7 +454,8 @@ static int inet_diag_bc_audit(const void *bytecode, int bytecode_len)
return len == 0 ? 0 : -EINVAL; return len == 0 ? 0 : -EINVAL;
} }
static int inet_diag_dump_sock(struct sk_buff *skb, struct sock *sk, static int inet_csk_diag_dump(struct sock *sk,
struct sk_buff *skb,
struct netlink_callback *cb) struct netlink_callback *cb)
{ {
struct inet_diag_req *r = NLMSG_DATA(cb->nlh); struct inet_diag_req *r = NLMSG_DATA(cb->nlh);
@ -473,7 +486,8 @@ static int inet_diag_dump_sock(struct sk_buff *skb, struct sock *sk,
return 0; return 0;
} }
return inet_diag_fill(skb, sk, r->idiag_ext, NETLINK_CB(cb->skb).pid, return inet_csk_diag_fill(sk, skb, r->idiag_ext,
NETLINK_CB(cb->skb).pid,
cb->nlh->nlmsg_seq, NLM_F_MULTI, cb->nlh); cb->nlh->nlmsg_seq, NLM_F_MULTI, cb->nlh);
} }
@ -694,7 +708,7 @@ static int inet_diag_dump(struct sk_buff *skb, struct netlink_callback *cb)
cb->args[3] > 0) cb->args[3] > 0)
goto syn_recv; goto syn_recv;
if (inet_diag_dump_sock(skb, sk, cb) < 0) { if (inet_csk_diag_dump(sk, skb, cb) < 0) {
inet_listen_unlock(hashinfo); inet_listen_unlock(hashinfo);
goto done; goto done;
} }
@ -750,7 +764,7 @@ skip_listen_ht:
if (r->id.idiag_dport != inet->dport && if (r->id.idiag_dport != inet->dport &&
r->id.idiag_dport) r->id.idiag_dport)
goto next_normal; goto next_normal;
if (inet_diag_dump_sock(skb, sk, cb) < 0) { if (inet_csk_diag_dump(sk, skb, cb) < 0) {
read_unlock_bh(&head->lock); read_unlock_bh(&head->lock);
goto done; goto done;
} }