setRandomInBuffer : clear tempByteArray content
This commit is contained in:
ylecollen
parent
2593c69a8a
commit
da2e1c5902
1 changed files with 100 additions and 95 deletions
|
|
@ -29,91 +29,96 @@ using namespace AndroidOlmSdk;
|
||||||
**/
|
**/
|
||||||
bool setRandomInBuffer(JNIEnv *env, uint8_t **aBuffer2Ptr, size_t aRandomSize)
|
bool setRandomInBuffer(JNIEnv *env, uint8_t **aBuffer2Ptr, size_t aRandomSize)
|
||||||
{
|
{
|
||||||
bool retCode = false;
|
bool retCode = false;
|
||||||
int bufferLen = aRandomSize*sizeof(uint8_t);
|
int bufferLen = aRandomSize*sizeof(uint8_t);
|
||||||
|
|
||||||
if(NULL == aBuffer2Ptr)
|
if(NULL == aBuffer2Ptr)
|
||||||
|
{
|
||||||
|
LOGE("## setRandomInBuffer(): failure - aBuffer=NULL");
|
||||||
|
}
|
||||||
|
else if(0 == aRandomSize)
|
||||||
|
{
|
||||||
|
LOGE("## setRandomInBuffer(): failure - random size=0");
|
||||||
|
}
|
||||||
|
else if(NULL == (*aBuffer2Ptr = (uint8_t*)malloc(bufferLen)))
|
||||||
|
{
|
||||||
|
LOGE("## setRandomInBuffer(): failure - alloc mem OOM");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
LOGD("## setRandomInBuffer(): randomSize=%lu",static_cast<long unsigned int>(aRandomSize));
|
||||||
|
|
||||||
|
bool secureRandomSucceeds = false;
|
||||||
|
|
||||||
|
// use the secureRandom class
|
||||||
|
jclass cls = env->FindClass("java/security/SecureRandom");
|
||||||
|
|
||||||
|
if (cls)
|
||||||
{
|
{
|
||||||
LOGE("## setRandomInBuffer(): failure - aBuffer=NULL");
|
jobject newObj = 0;
|
||||||
}
|
jmethodID constructor = env->GetMethodID(cls, "<init>", "()V");
|
||||||
else if(0 == aRandomSize)
|
jmethodID nextByteMethod = env->GetMethodID(cls, "nextBytes", "([B)V");
|
||||||
{
|
|
||||||
LOGE("## setRandomInBuffer(): failure - random size=0");
|
|
||||||
}
|
|
||||||
else if(NULL == (*aBuffer2Ptr = (uint8_t*)malloc(bufferLen)))
|
|
||||||
{
|
|
||||||
LOGE("## setRandomInBuffer(): failure - alloc mem OOM");
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
LOGD("## setRandomInBuffer(): randomSize=%lu",static_cast<long unsigned int>(aRandomSize));
|
|
||||||
|
|
||||||
bool secureRandomSucceeds = false;
|
if (constructor)
|
||||||
|
{
|
||||||
|
newObj = env->NewObject(cls, constructor);
|
||||||
|
jbyteArray tempByteArray = env->NewByteArray(bufferLen);
|
||||||
|
|
||||||
// clear the buffer
|
if (newObj && tempByteArray)
|
||||||
memset(*aBuffer2Ptr, 0, bufferLen);
|
|
||||||
|
|
||||||
// use the secureRandom class
|
|
||||||
jclass cls = env->FindClass("java/security/SecureRandom");
|
|
||||||
|
|
||||||
if (cls)
|
|
||||||
{
|
{
|
||||||
jobject newObj = 0;
|
env->CallVoidMethod(newObj, nextByteMethod, tempByteArray);
|
||||||
jmethodID constructor = env->GetMethodID(cls, "<init>", "()V");
|
|
||||||
jmethodID nextByteMethod = env->GetMethodID(cls, "nextBytes", "([B)V");
|
|
||||||
|
|
||||||
if (constructor)
|
jbyte* buffer = env->GetByteArrayElements(tempByteArray, NULL);
|
||||||
{
|
|
||||||
newObj = env->NewObject(cls, constructor);
|
|
||||||
jbyteArray tempByteArray = env->NewByteArray(bufferLen);
|
|
||||||
|
|
||||||
if (newObj && tempByteArray)
|
if (buffer)
|
||||||
{
|
{
|
||||||
env->CallVoidMethod(newObj, nextByteMethod, tempByteArray);
|
memcpy(*aBuffer2Ptr, buffer, bufferLen);
|
||||||
|
secureRandomSucceeds = true;
|
||||||
|
|
||||||
jbyte* buffer = env->GetByteArrayElements(tempByteArray,0);
|
// clear tempByteArray to hide sensitive data.
|
||||||
|
memset(buffer, 0, bufferLen);
|
||||||
|
env->SetByteArrayRegion(tempByteArray, 0, bufferLen, buffer);
|
||||||
|
|
||||||
if (buffer)
|
// ensure that the buffer is released
|
||||||
{
|
env->ReleaseByteArrayElements(tempByteArray, buffer, JNI_ABORT);
|
||||||
memcpy(*aBuffer2Ptr, buffer, bufferLen);
|
}
|
||||||
secureRandomSucceeds = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (tempByteArray)
|
|
||||||
{
|
|
||||||
env->DeleteLocalRef(tempByteArray);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (newObj)
|
|
||||||
{
|
|
||||||
env->DeleteLocalRef(newObj);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!secureRandomSucceeds)
|
if (tempByteArray)
|
||||||
{
|
{
|
||||||
LOGE("## setRandomInBuffer(): SecureRandom failed, use a fallback");
|
env->DeleteLocalRef(tempByteArray);
|
||||||
struct timeval timeValue;
|
|
||||||
gettimeofday(&timeValue, NULL);
|
|
||||||
srand(timeValue.tv_usec); // init seed
|
|
||||||
|
|
||||||
for(size_t i=0;i<aRandomSize;i++)
|
|
||||||
{
|
|
||||||
(*aBuffer2Ptr)[i] = (uint8_t)(rand()%ACCOUNT_CREATION_RANDOM_MODULO);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// debug purpose
|
if (newObj)
|
||||||
/*for(int i = 0; i < aRandomSize; i++)
|
|
||||||
{
|
{
|
||||||
LOGD("## setRandomInBuffer(): randomBuffPtr[%ld]=%d",i, (*aBuffer2Ptr)[i]);
|
env->DeleteLocalRef(newObj);
|
||||||
}*/
|
}
|
||||||
|
}
|
||||||
retCode = true;
|
|
||||||
}
|
}
|
||||||
return retCode;
|
|
||||||
|
if (!secureRandomSucceeds)
|
||||||
|
{
|
||||||
|
LOGE("## setRandomInBuffer(): SecureRandom failed, use a fallback");
|
||||||
|
struct timeval timeValue;
|
||||||
|
gettimeofday(&timeValue, NULL);
|
||||||
|
srand(timeValue.tv_usec); // init seed
|
||||||
|
|
||||||
|
for(size_t i=0;i<aRandomSize;i++)
|
||||||
|
{
|
||||||
|
(*aBuffer2Ptr)[i] = (uint8_t)(rand()%ACCOUNT_CREATION_RANDOM_MODULO);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// debug purpose
|
||||||
|
/*for(int i = 0; i < aRandomSize; i++)
|
||||||
|
{
|
||||||
|
LOGD("## setRandomInBuffer(): randomBuffPtr[%ld]=%d",i, (*aBuffer2Ptr)[i]);
|
||||||
|
}*/
|
||||||
|
|
||||||
|
retCode = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return retCode;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -242,37 +247,37 @@ jlong getUtilityInstanceId(JNIEnv* aJniEnv, jobject aJavaObject)
|
||||||
*/
|
*/
|
||||||
jstring javaCStringToUtf8(JNIEnv *env, uint8_t *aCStringMsgPtr, size_t aMsgLength)
|
jstring javaCStringToUtf8(JNIEnv *env, uint8_t *aCStringMsgPtr, size_t aMsgLength)
|
||||||
{
|
{
|
||||||
jstring convertedRetValue = 0;
|
jstring convertedRetValue = 0;
|
||||||
jbyteArray tempByteArray = NULL;
|
jbyteArray tempByteArray = NULL;
|
||||||
|
|
||||||
if((NULL == aCStringMsgPtr) || (NULL == env))
|
if((NULL == aCStringMsgPtr) || (NULL == env))
|
||||||
|
{
|
||||||
|
LOGE("## javaCStringToUtf8(): failure - invalid parameters (null)");
|
||||||
|
}
|
||||||
|
else if(NULL == (tempByteArray=env->NewByteArray(aMsgLength)))
|
||||||
|
{
|
||||||
|
LOGE("## javaCStringToUtf8(): failure - return byte array OOM");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
env->SetByteArrayRegion(tempByteArray, 0, aMsgLength, (const jbyte*)aCStringMsgPtr);
|
||||||
|
|
||||||
|
// UTF-8 conversion from JAVA
|
||||||
|
jstring strEncode = (env)->NewStringUTF("UTF-8");
|
||||||
|
jclass jClass = env->FindClass("java/lang/String");
|
||||||
|
jmethodID cstor = env->GetMethodID(jClass, "<init>", "([BLjava/lang/String;)V");
|
||||||
|
|
||||||
|
if((0!=jClass) && (0!=jClass) && (0!=strEncode))
|
||||||
{
|
{
|
||||||
LOGE("## javaCStringToUtf8(): failure - invalid parameters (null)");
|
convertedRetValue = (jstring) env->NewObject(jClass, cstor, tempByteArray, strEncode);
|
||||||
}
|
LOGD(" ## javaCStringToUtf8(): succeed");
|
||||||
else if(NULL == (tempByteArray=env->NewByteArray(aMsgLength)))
|
env->DeleteLocalRef(tempByteArray);
|
||||||
{
|
|
||||||
LOGE("## javaCStringToUtf8(): failure - return byte array OOM");
|
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
env->SetByteArrayRegion(tempByteArray, 0, aMsgLength, (const jbyte*)aCStringMsgPtr);
|
LOGE(" ## javaCStringToUtf8(): failure - invalid Java references");
|
||||||
|
|
||||||
// UTF-8 conversion from JAVA
|
|
||||||
jstring strEncode = (env)->NewStringUTF("UTF-8");
|
|
||||||
jclass jClass = env->FindClass("java/lang/String");
|
|
||||||
jmethodID cstor = env->GetMethodID(jClass, "<init>", "([BLjava/lang/String;)V");
|
|
||||||
|
|
||||||
if((0!=jClass) && (0!=jClass) && (0!=strEncode))
|
|
||||||
{
|
|
||||||
convertedRetValue = (jstring) env->NewObject(jClass, cstor, tempByteArray, strEncode);
|
|
||||||
LOGD(" ## javaCStringToUtf8(): succeed");
|
|
||||||
env->DeleteLocalRef(tempByteArray);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
LOGE(" ## javaCStringToUtf8(): failure - invalid Java references");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return convertedRetValue;
|
return convertedRetValue;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue