goto out_sleep;
                }
 
-               NLA_PUT(skb, WL1271_TM_ATTR_DATA, buf_len, buf);
+               if (nla_put(skb, WL1271_TM_ATTR_DATA, buf_len, buf))
+                       goto nla_put_failure;
                ret = cfg80211_testmode_reply(skb);
                if (ret < 0)
                        goto out_sleep;
                goto out_free;
        }
 
-       NLA_PUT(skb, WL1271_TM_ATTR_DATA, sizeof(*cmd), cmd);
+       if (nla_put(skb, WL1271_TM_ATTR_DATA, sizeof(*cmd), cmd))
+               goto nla_put_failure;
        ret = cfg80211_testmode_reply(skb);
        if (ret < 0)
                goto out_free;
                goto out;
        }
 
-       NLA_PUT(skb, WL1271_TM_ATTR_DATA, ETH_ALEN, mac_addr);
+       if (nla_put(skb, WL1271_TM_ATTR_DATA, ETH_ALEN, mac_addr))
+               goto nla_put_failure;
        ret = cfg80211_testmode_reply(skb);
        if (ret < 0)
                goto out;