mozglue/android/NativeCrypto.cpp
author Jeff Walden <jwalden@mit.edu>
Tue, 19 Nov 2019 04:55:39 +0000
changeset 502538 b5c5ba07d3dbd0d07b66fa42a103f4df2c27d3a2
parent 472094 e1993a1f09ac53cd1a04fdf6a87f8cad8e44f73e
permissions -rw-r--r--
Bug 1596544 - intl_ValidateAndCanonicalizeUnicodeExtensionType should ignore the second |option| argument until it's needed to report an error. r=anba Differential Revision: https://phabricator.services.mozilla.com/D53145

/* 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;
}