1/*
2 * Copyright (C) 2018 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#define LOG_TAG "Scrypt"
18
19#include <nativehelper/JNIHelp.h>
20#include "jni.h"
21
22#include <android_runtime/Log.h>
23#include <utils/Timers.h>
24#include <utils/misc.h>
25#include <utils/String8.h>
26#include <utils/Log.h>
27
28extern "C" {
29#include "crypto_scrypt.h"
30}
31
32namespace android {
33
34static jbyteArray android_security_Scrypt_nativeScrypt(JNIEnv* env, jobject, jbyteArray password, jbyteArray salt, jint N, jint r, jint p, jint outLen) {
35    if (!password || !salt) {
36        return NULL;
37    }
38
39    int passwordLen = env->GetArrayLength(password);
40    int saltLen = env->GetArrayLength(salt);
41    jbyteArray ret = env->NewByteArray(outLen);
42
43    jbyte* passwordPtr = (jbyte*)env->GetByteArrayElements(password, NULL);
44    jbyte* saltPtr = (jbyte*)env->GetByteArrayElements(salt, NULL);
45    jbyte* retPtr = (jbyte*)env->GetByteArrayElements(ret, NULL);
46
47    int rc = crypto_scrypt((const uint8_t *)passwordPtr, passwordLen,
48                       (const uint8_t *)saltPtr, saltLen, N, r, p, (uint8_t *)retPtr,
49                       outLen);
50    env->ReleaseByteArrayElements(password, passwordPtr, JNI_ABORT);
51    env->ReleaseByteArrayElements(salt, saltPtr, JNI_ABORT);
52    env->ReleaseByteArrayElements(ret, retPtr, 0);
53
54    if (!rc) {
55        return ret;
56    } else {
57        SLOGE("scrypt failed");
58        return NULL;
59    }
60}
61
62static const JNINativeMethod sMethods[] = {
63     /* name, signature, funcPtr */
64    {"nativeScrypt", "([B[BIIII)[B", (void*)android_security_Scrypt_nativeScrypt},
65};
66
67int register_android_security_Scrypt(JNIEnv* env) {
68    return jniRegisterNativeMethods(env, "android/security/Scrypt",
69                                    sMethods, NELEM(sMethods));
70}
71
72} /* namespace android */
73