141 lines
3.9 KiB
C++
141 lines
3.9 KiB
C++
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "NativeCrypto.h"
|
|
#include "APKOpen.h"
|
|
|
|
#include <jni.h>
|
|
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <inttypes.h>
|
|
|
|
#include "mozilla/SHA1.h"
|
|
#include "pbkdf2_sha256.h"
|
|
|
|
/**
|
|
* Helper function to invoke native PBKDF2 function with JNI
|
|
* arguments.
|
|
*/
|
|
extern "C" JNIEXPORT jbyteArray MOZ_JNICALL
|
|
Java_org_mozilla_gecko_background_nativecode_NativeCrypto_pbkdf2SHA256(
|
|
JNIEnv* env, jclass jc, jbyteArray jpassword, jbyteArray jsalt, jint c,
|
|
jint dkLen) {
|
|
if (dkLen < 0) {
|
|
env->ThrowNew(env->FindClass("java/lang/IllegalArgumentException"),
|
|
"dkLen should not be less than 0");
|
|
return nullptr;
|
|
}
|
|
|
|
jbyte* password = env->GetByteArrayElements(jpassword, nullptr);
|
|
size_t passwordLen = env->GetArrayLength(jpassword);
|
|
|
|
jbyte* salt = env->GetByteArrayElements(jsalt, nullptr);
|
|
size_t saltLen = env->GetArrayLength(jsalt);
|
|
|
|
uint8_t hashResult[dkLen];
|
|
PBKDF2_SHA256((uint8_t*)password, passwordLen, (uint8_t*)salt, saltLen,
|
|
(uint64_t)c, hashResult, (size_t)dkLen);
|
|
|
|
env->ReleaseByteArrayElements(jpassword, password, JNI_ABORT);
|
|
env->ReleaseByteArrayElements(jsalt, salt, JNI_ABORT);
|
|
|
|
jbyteArray out = env->NewByteArray(dkLen);
|
|
if (out == nullptr) {
|
|
return nullptr;
|
|
}
|
|
env->SetByteArrayRegion(out, 0, dkLen, (jbyte*)hashResult);
|
|
|
|
return out;
|
|
}
|
|
|
|
using namespace mozilla;
|
|
|
|
/**
|
|
* Helper function to invoke native SHA-1 function with JNI arguments.
|
|
*/
|
|
extern "C" JNIEXPORT jbyteArray MOZ_JNICALL
|
|
Java_org_mozilla_gecko_background_nativecode_NativeCrypto_sha1(
|
|
JNIEnv* env, jclass jc, jbyteArray jstr) {
|
|
jbyte* str = env->GetByteArrayElements(jstr, nullptr);
|
|
size_t strLen = env->GetArrayLength(jstr);
|
|
|
|
SHA1Sum sha1;
|
|
SHA1Sum::Hash hashResult;
|
|
sha1.update((void*)str, (uint32_t)strLen);
|
|
sha1.finish(hashResult);
|
|
|
|
env->ReleaseByteArrayElements(jstr, str, JNI_ABORT);
|
|
|
|
jbyteArray out = env->NewByteArray(SHA1Sum::kHashSize);
|
|
if (out == nullptr) {
|
|
return nullptr;
|
|
}
|
|
env->SetByteArrayRegion(out, 0, SHA1Sum::kHashSize, (jbyte*)hashResult);
|
|
|
|
return out;
|
|
}
|
|
|
|
/**
|
|
* Helper function to invoke native SHA-256 init with JNI arguments.
|
|
*/
|
|
extern "C" JNIEXPORT jbyteArray MOZ_JNICALL
|
|
Java_org_mozilla_gecko_background_nativecode_NativeCrypto_sha256init(
|
|
JNIEnv* env, jclass jc) {
|
|
jbyteArray out = env->NewByteArray(sizeof(SHA256_CTX));
|
|
if (nullptr == out) {
|
|
return nullptr;
|
|
}
|
|
|
|
SHA256_CTX* shaContext = (SHA256_CTX*)env->GetByteArrayElements(out, nullptr);
|
|
SHA256_Init(shaContext);
|
|
|
|
env->ReleaseByteArrayElements(out, (jbyte*)shaContext, 0);
|
|
|
|
return out;
|
|
}
|
|
|
|
/**
|
|
* Helper function to invoke native SHA-256 update with JNI arguments.
|
|
*/
|
|
extern "C" JNIEXPORT void MOZ_JNICALL
|
|
Java_org_mozilla_gecko_background_nativecode_NativeCrypto_sha256update(
|
|
JNIEnv* env, jclass jc, jbyteArray jctx, jbyteArray jstr, jint len) {
|
|
jbyte* str = env->GetByteArrayElements(jstr, nullptr);
|
|
|
|
SHA256_CTX* shaContext =
|
|
(SHA256_CTX*)env->GetByteArrayElements(jctx, nullptr);
|
|
|
|
SHA256_Update(shaContext, (void*)str, (size_t)len);
|
|
|
|
env->ReleaseByteArrayElements(jstr, str, JNI_ABORT);
|
|
env->ReleaseByteArrayElements(jctx, (jbyte*)shaContext, 0);
|
|
|
|
return;
|
|
}
|
|
|
|
/**
|
|
* Helper function to invoke native SHA-256 finalize with JNI arguments.
|
|
*/
|
|
extern "C" JNIEXPORT jbyteArray MOZ_JNICALL
|
|
Java_org_mozilla_gecko_background_nativecode_NativeCrypto_sha256finalize(
|
|
JNIEnv* env, jclass jc, jbyteArray jctx) {
|
|
SHA256_CTX* shaContext =
|
|
(SHA256_CTX*)env->GetByteArrayElements(jctx, nullptr);
|
|
|
|
unsigned char* digest = new unsigned char[32];
|
|
SHA256_Final(digest, shaContext);
|
|
|
|
env->ReleaseByteArrayElements(jctx, (jbyte*)shaContext, JNI_ABORT);
|
|
|
|
jbyteArray out = env->NewByteArray(32);
|
|
if (nullptr != out) {
|
|
env->SetByteArrayRegion(out, 0, 32, (jbyte*)digest);
|
|
}
|
|
|
|
delete[] digest;
|
|
|
|
return out;
|
|
}
|