From: wzt.wzt@gmail.com <wzt.wzt@gmail.com>
Date: Wed, 5 May 2010 06:56:52 +0000 (+0800)
Subject: Staging: rtl8192u: Check kmalloc return value before use the buffer in ieee80211_soft... 
X-Git-Tag: v2.6.35-rc1~441^2^2~186
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=dde27e03cc964526531fdf2bd58c1e4ee4525074;p=profile%2Fivi%2Fkernel-adaptation-intel-automotive.git

Staging: rtl8192u: Check kmalloc return value before use the buffer in ieee80211_softmac.c

Check kmalloc return value before use the buffer.

Signed-off-by: Zhitong Wang <zhitong.wangzt@alibaba-inc.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@suse.de>
---

diff --git a/drivers/staging/rtl8192u/ieee80211/ieee80211_softmac.c b/drivers/staging/rtl8192u/ieee80211/ieee80211_softmac.c
index 148424a..e5e583e 100644
--- a/drivers/staging/rtl8192u/ieee80211/ieee80211_softmac.c
+++ b/drivers/staging/rtl8192u/ieee80211/ieee80211_softmac.c
@@ -1580,6 +1580,8 @@ static inline u16 auth_parse(struct sk_buff *skb, u8** challenge, int *chlen)
 		if(*(t++) == MFIE_TYPE_CHALLENGE){
 			*chlen = *(t++);
 			*challenge = (u8*)kmalloc(*chlen, GFP_ATOMIC);
+			if (!*challenge)
+				return -ENOMEM;
 			memcpy(*challenge, t, *chlen);
 		}
 	}