Commit 7c99c909 authored by Stephen Hemminger's avatar Stephen Hemminger Committed by David S. Miller
Browse files

[TCP]: Change tcp_diag to use the existing __RTA_PUT() macro.

parent 317a76f9
Loading
Loading
Loading
Loading
+2 −7
Original line number Original line Diff line number Diff line
@@ -43,13 +43,7 @@ struct tcpdiag_entry
static struct sock *tcpnl;
static struct sock *tcpnl;


#define TCPDIAG_PUT(skb, attrtype, attrlen) \
#define TCPDIAG_PUT(skb, attrtype, attrlen) \
({ int rtalen = RTA_LENGTH(attrlen);        \
	RTA_DATA(__RTA_PUT(skb, attrtype, attrlen))
   struct rtattr *rta;                      \
   if (skb_tailroom(skb) < RTA_ALIGN(rtalen)) goto nlmsg_failure; \
   rta = (void*)__skb_put(skb, RTA_ALIGN(rtalen)); \
   rta->rta_type = attrtype;                \
   rta->rta_len = rtalen;                   \
   RTA_DATA(rta); })


static int tcpdiag_fill(struct sk_buff *skb, struct sock *sk,
static int tcpdiag_fill(struct sk_buff *skb, struct sock *sk,
			int ext, u32 pid, u32 seq, u16 nlmsg_flags)
			int ext, u32 pid, u32 seq, u16 nlmsg_flags)
@@ -167,6 +161,7 @@ static int tcpdiag_fill(struct sk_buff *skb, struct sock *sk,
	nlh->nlmsg_len = skb->tail - b;
	nlh->nlmsg_len = skb->tail - b;
	return skb->len;
	return skb->len;


rtattr_failure:
nlmsg_failure:
nlmsg_failure:
	skb_trim(skb, b - skb->data);
	skb_trim(skb, b - skb->data);
	return -1;
	return -1;