vti: check nla_put_* return value

Signed-off-by: Hangbin Liu <liuhangbin@gmail.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

1 files changed, 7 insertions, 6 deletions

diff --git a/net/ipv4/ip_vti.c b/net/ipv4/ip_vti.c
index 40977413fd48..4ec9affb2252 100644
--- a/net/ipv4/ip_vti.c
+++ b/net/ipv4/ip_vti.c
@@ -546,12 +546,13 @@ static int vti_fill_info(struct sk_buff *skb, const struct net_device *dev)
         struct ip_tunnel *t = netdev_priv(dev);
         struct ip_tunnel_parm *p = &t->parms;
 
-        nla_put_u32(skb, IFLA_VTI_LINK, p->link);
-        nla_put_be32(skb, IFLA_VTI_IKEY, p->i_key);
-        nla_put_be32(skb, IFLA_VTI_OKEY, p->o_key);
-        nla_put_in_addr(skb, IFLA_VTI_LOCAL, p->iph.saddr);
-        nla_put_in_addr(skb, IFLA_VTI_REMOTE, p->iph.daddr);
-        nla_put_u32(skb, IFLA_VTI_FWMARK, t->fwmark);
+        if (nla_put_u32(skb, IFLA_VTI_LINK, p->link) ||
+            nla_put_be32(skb, IFLA_VTI_IKEY, p->i_key) ||
+            nla_put_be32(skb, IFLA_VTI_OKEY, p->o_key) ||
+            nla_put_in_addr(skb, IFLA_VTI_LOCAL, p->iph.saddr) ||
+            nla_put_in_addr(skb, IFLA_VTI_REMOTE, p->iph.daddr) ||
+            nla_put_u32(skb, IFLA_VTI_FWMARK, t->fwmark))
+                return -EMSGSIZE;
 
         return 0;
 }