1 files changed, 3 insertions, 2 deletions

diff --git a/drivers/staging/rtl8192e/ieee80211/ieee80211_softmac.c b/drivers/staging/rtl8192e/ieee80211/ieee80211_softmac.c
index 88a9cd1958a..fc96676bb9c 100644
--- a/drivers/staging/rtl8192e/ieee80211/ieee80211_softmac.c
+++ b/drivers/staging/rtl8192e/ieee80211/ieee80211_softmac.c
@@ -1564,8 +1564,9 @@ static inline u16 auth_parse(struct sk_buff *skb, u8** challenge, int *chlen)
 
                 if(*(t++) == MFIE_TYPE_CHALLENGE){
                         *chlen = *(t++);
-                        *challenge = kmalloc(*chlen, GFP_ATOMIC);
-                        memcpy(*challenge, t, *chlen);
+                        *challenge = kmemdup(t, *chlen, GFP_ATOMIC);
+                        if (!*challenge)
+                                return -ENOMEM;
                 }
         }