1 files changed, 4 insertions, 2 deletions

diff --git a/drivers/staging/wilc1000/host_interface.c b/drivers/staging/wilc1000/host_interface.c
index af98c43fef0e..28fef2525086 100644
--- a/drivers/staging/wilc1000/host_interface.c
+++ b/drivers/staging/wilc1000/host_interface.c
@@ -3124,8 +3124,10 @@ int host_int_add_wep_key_bss_sta(struct host_if_drv *hif_drv,
         msg.body.key_info.type = WEP;
         msg.body.key_info.action = ADDKEY;
         msg.drv = hif_drv;
-        msg.body.key_info.attr.wep.key = kmalloc(len, GFP_KERNEL);
-        memcpy(msg.body.key_info.attr.wep.key, key, len);
+        msg.body.key_info.attr.wep.key = kmemdup(key, len, GFP_KERNEL);
+        if (!msg.body.key_info.attr.wep.key)
+                return -ENOMEM;
+
         msg.body.key_info.attr.wep.key_len = len;
         msg.body.key_info.attr.wep.index = index;