From 28f483350be92654edec24d035d07cdeced6acec Mon Sep 17 00:00:00 2001 From: Radoslaw Bartosiak Date: Mon, 9 Mar 2015 19:13:22 +0100 Subject: [PATCH] Fix calloc return check Handle null returned by calloc. Change-Id: If2f8085893d283b663f5928f93b90728b175b679 Signed-off-by: Radoslaw Bartosiak --- src/dukgen.c | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/dukgen.c b/src/dukgen.c index 86a2e4d..6a76c68 100644 --- a/src/dukgen.c +++ b/src/dukgen.c @@ -30,6 +30,8 @@ char* GetDeviceUniqueKey(char* pAppId, int idLen, int keyLen) bool result = true; pUniqueKey = (unsigned char*)calloc(keyLen,1); + if (pUniqueKey == NULL) + return NULL; result = SecFrameGeneratePlatformUniqueKey((unsigned int)keyLen , pUniqueKey); if(result == false) { @@ -38,6 +40,11 @@ char* GetDeviceUniqueKey(char* pAppId, int idLen, int keyLen) } pDuk = (char*)calloc(keyLen, 1); + if (pDuk == NULL) + { + free(pUniqueKey); + return NULL; + } PKCS5_PBKDF2_HMAC_SHA1(pAppId, idLen, (unsigned char*)pUniqueKey, keyLen, 1, keyLen, (unsigned char*)pDuk); free(pUniqueKey); -- 2.7.4