Fix Coverity CID 557.

svn path=/trunk/; revision=36526
This commit is contained in:
Gerald Combs 2011-04-08 21:21:32 +00:00
parent 1667de16c5
commit a5ddd187ca
1 changed files with 5 additions and 2 deletions

View File

@ -1718,12 +1718,15 @@ AirPDcapRsnaPwd2PskStep(
const INT count,
UCHAR *output)
{
UCHAR digest[36], digest1[AIRPDCAP_SHA_DIGEST_LEN];
UCHAR digest[64], digest1[64];
INT i, j;
if (ssidLength > 36 - 4)
if (ssidLength+4 > 36)
return AIRPDCAP_RET_UNSUCCESS;
memset(digest, 0, 64);
memset(digest1, 0, 64);
/* U1 = PRF(P, S || INT(i)) */
memcpy(digest, ssid, ssidLength);
digest[ssidLength] = (UCHAR)((count>>24) & 0xff);