1 files changed, 72 insertions, 29 deletions

diff --git a/net/nfc/netlink.c b/net/nfc/netlink.c
index 6404052d6c07..f1829f6ae9c5 100644
--- a/net/nfc/netlink.c
+++ b/net/nfc/netlink.c
@@ -63,19 +63,23 @@ static int nfc_genl_send_target(struct sk_buff *msg, struct nfc_target *target,
 
         genl_dump_check_consistent(cb, hdr, &nfc_genl_family);
 
-        NLA_PUT_U32(msg, NFC_ATTR_TARGET_INDEX, target->idx);
-        NLA_PUT_U32(msg, NFC_ATTR_PROTOCOLS, target->supported_protocols);
-        NLA_PUT_U16(msg, NFC_ATTR_TARGET_SENS_RES, target->sens_res);
-        NLA_PUT_U8(msg, NFC_ATTR_TARGET_SEL_RES, target->sel_res);
-        if (target->nfcid1_len > 0)
-                NLA_PUT(msg, NFC_ATTR_TARGET_NFCID1, target->nfcid1_len,
-                        target->nfcid1);
-        if (target->sensb_res_len > 0)
-                NLA_PUT(msg, NFC_ATTR_TARGET_SENSB_RES, target->sensb_res_len,
-                        target->sensb_res);
-        if (target->sensf_res_len > 0)
-                NLA_PUT(msg, NFC_ATTR_TARGET_SENSF_RES, target->sensf_res_len,
-                        target->sensf_res);
+        if (nla_put_u32(msg, NFC_ATTR_TARGET_INDEX, target->idx) ||
+            nla_put_u32(msg, NFC_ATTR_PROTOCOLS, target->supported_protocols) ||
+            nla_put_u16(msg, NFC_ATTR_TARGET_SENS_RES, target->sens_res) ||
+            nla_put_u8(msg, NFC_ATTR_TARGET_SEL_RES, target->sel_res))
+                goto nla_put_failure;
+        if (target->nfcid1_len > 0 &&
+            nla_put(msg, NFC_ATTR_TARGET_NFCID1, target->nfcid1_len,
+                    target->nfcid1))
+                goto nla_put_failure;
+        if (target->sensb_res_len > 0 &&
+            nla_put(msg, NFC_ATTR_TARGET_SENSB_RES, target->sensb_res_len,
+                    target->sensb_res))
+                goto nla_put_failure;
+        if (target->sensf_res_len > 0 &&
+            nla_put(msg, NFC_ATTR_TARGET_SENSF_RES, target->sensf_res_len,
+                    target->sensf_res))
+                goto nla_put_failure;
 
         return genlmsg_end(msg, hdr);
 
@@ -170,7 +174,8 @@ int nfc_genl_targets_found(struct nfc_dev *dev)
         if (!hdr)
                 goto free_msg;
 
-        NLA_PUT_U32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx);
+        if (nla_put_u32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx))
+                goto nla_put_failure;
 
         genlmsg_end(msg, hdr);
 
@@ -183,6 +188,37 @@ free_msg:
         return -EMSGSIZE;
 }
 
+int nfc_genl_target_lost(struct nfc_dev *dev, u32 target_idx)
+{
+        struct sk_buff *msg;
+        void *hdr;
+
+        msg = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
+        if (!msg)
+                return -ENOMEM;
+
+        hdr = genlmsg_put(msg, 0, 0, &nfc_genl_family, 0,
+                          NFC_EVENT_TARGET_LOST);
+        if (!hdr)
+                goto free_msg;
+
+        if (nla_put_string(msg, NFC_ATTR_DEVICE_NAME, nfc_device_name(dev)) ||
+            nla_put_u32(msg, NFC_ATTR_TARGET_INDEX, target_idx))
+                goto nla_put_failure;
+
+        genlmsg_end(msg, hdr);
+
+        genlmsg_multicast(msg, 0, nfc_genl_event_mcgrp.id, GFP_KERNEL);
+
+        return 0;
+
+nla_put_failure:
+        genlmsg_cancel(msg, hdr);
+free_msg:
+        nlmsg_free(msg);
+        return -EMSGSIZE;
+}
+
 int nfc_genl_device_added(struct nfc_dev *dev)
 {
         struct sk_buff *msg;
@@ -197,10 +233,11 @@ int nfc_genl_device_added(struct nfc_dev *dev)
         if (!hdr)
                 goto free_msg;
 
-        NLA_PUT_STRING(msg, NFC_ATTR_DEVICE_NAME, nfc_device_name(dev));
-        NLA_PUT_U32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx);
-        NLA_PUT_U32(msg, NFC_ATTR_PROTOCOLS, dev->supported_protocols);
-        NLA_PUT_U8(msg, NFC_ATTR_DEVICE_POWERED, dev->dev_up);
+        if (nla_put_string(msg, NFC_ATTR_DEVICE_NAME, nfc_device_name(dev)) ||
+            nla_put_u32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx) ||
+            nla_put_u32(msg, NFC_ATTR_PROTOCOLS, dev->supported_protocols) ||
+            nla_put_u8(msg, NFC_ATTR_DEVICE_POWERED, dev->dev_up))
+                goto nla_put_failure;
 
         genlmsg_end(msg, hdr);
 
@@ -229,7 +266,8 @@ int nfc_genl_device_removed(struct nfc_dev *dev)
         if (!hdr)
                 goto free_msg;
 
-        NLA_PUT_U32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx);
+        if (nla_put_u32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx))
+                goto nla_put_failure;
 
         genlmsg_end(msg, hdr);
 
@@ -259,10 +297,11 @@ static int nfc_genl_send_device(struct sk_buff *msg, struct nfc_dev *dev,
         if (cb)
                 genl_dump_check_consistent(cb, hdr, &nfc_genl_family);
 
-        NLA_PUT_STRING(msg, NFC_ATTR_DEVICE_NAME, nfc_device_name(dev));
-        NLA_PUT_U32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx);
-        NLA_PUT_U32(msg, NFC_ATTR_PROTOCOLS, dev->supported_protocols);
-        NLA_PUT_U8(msg, NFC_ATTR_DEVICE_POWERED, dev->dev_up);
+        if (nla_put_string(msg, NFC_ATTR_DEVICE_NAME, nfc_device_name(dev)) ||
+            nla_put_u32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx) ||
+            nla_put_u32(msg, NFC_ATTR_PROTOCOLS, dev->supported_protocols) ||
+            nla_put_u8(msg, NFC_ATTR_DEVICE_POWERED, dev->dev_up))
+                goto nla_put_failure;
 
         return genlmsg_end(msg, hdr);
 
@@ -339,11 +378,14 @@ int nfc_genl_dep_link_up_event(struct nfc_dev *dev, u32 target_idx,
         if (!hdr)
                 goto free_msg;
 
-        NLA_PUT_U32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx);
-        if (rf_mode == NFC_RF_INITIATOR)
-                NLA_PUT_U32(msg, NFC_ATTR_TARGET_INDEX, target_idx);
-        NLA_PUT_U8(msg, NFC_ATTR_COMM_MODE, comm_mode);
-        NLA_PUT_U8(msg, NFC_ATTR_RF_MODE, rf_mode);
+        if (nla_put_u32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx))
+                goto nla_put_failure;
+        if (rf_mode == NFC_RF_INITIATOR &&
+            nla_put_u32(msg, NFC_ATTR_TARGET_INDEX, target_idx))
+                goto nla_put_failure;
+        if (nla_put_u8(msg, NFC_ATTR_COMM_MODE, comm_mode) ||
+            nla_put_u8(msg, NFC_ATTR_RF_MODE, rf_mode))
+                goto nla_put_failure;
 
         genlmsg_end(msg, hdr);
 
@@ -376,7 +418,8 @@ int nfc_genl_dep_link_down_event(struct nfc_dev *dev)
         if (!hdr)
                 goto free_msg;
 
-        NLA_PUT_U32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx);
+        if (nla_put_u32(msg, NFC_ATTR_DEVICE_INDEX, dev->idx))
+                goto nla_put_failure;
 
         genlmsg_end(msg, hdr);