1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
/*
* Copyright (C) 2006 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "jni.h"
#include <JNIHelp.h>
#include <stdlib.h>
#include <stdint.h>
#include <openssl/md5.h>
namespace android
{
struct fields_t {
jfieldID context;
};
static fields_t fields;
static void native_init(JNIEnv *env, jobject clazz)
{
MD5_CTX* context = (MD5_CTX *)malloc(sizeof(MD5_CTX));
MD5_Init(context);
env->SetIntField(clazz, fields.context, (int)context);
}
static void native_reset(JNIEnv *env, jobject clazz)
{
MD5_CTX *context = (MD5_CTX *)env->GetIntField(clazz, fields.context);
if (context != NULL) {
free(context);
env->SetIntField(clazz, fields.context, 0 );
}
}
static void native_update(JNIEnv *env, jobject clazz, jbyteArray dataArray)
{
jbyte * data;
jsize dataSize;
MD5_CTX *context = (MD5_CTX *)env->GetIntField(clazz, fields.context);
if (context == NULL) {
native_init(env, clazz);
context = (MD5_CTX *)env->GetIntField(clazz, fields.context);
}
data = env->GetByteArrayElements(dataArray, NULL);
if (data == NULL) {
LOGE("Unable to get byte array elements");
jniThrowException(env, "java/lang/IllegalArgumentException",
"Invalid data array when calling MessageDigest.update()");
return;
}
dataSize = env->GetArrayLength(dataArray);
MD5_Update(context, data, dataSize);
env->ReleaseByteArrayElements(dataArray, data, 0);
}
static jbyteArray native_digest(JNIEnv *env, jobject clazz)
{
jbyteArray array;
jbyte md[MD5_DIGEST_LENGTH];
MD5_CTX *context = (MD5_CTX *)env->GetIntField(clazz, fields.context);
MD5_Final((uint8_t*)md, context);
array = env->NewByteArray(MD5_DIGEST_LENGTH);
LOG_ASSERT(array, "Native could not create new byte[]");
env->SetByteArrayRegion(array, 0, MD5_DIGEST_LENGTH, md);
native_reset(env, clazz);
return array;
}
/*
* JNI registration.
*/
static JNINativeMethod gMethods[] =
{
/* name, signature, funcPtr */
{"init", "()V", (void *)native_init},
{"update", "([B)V", (void *)native_update},
{"digest", "()[B", (void *)native_digest},
{"reset", "()V", (void *)native_reset},
};
int register_android_security_Md5MessageDigest(JNIEnv *env)
{
jclass clazz;
clazz = env->FindClass("android/security/Md5MessageDigest");
if (clazz == NULL) {
LOGE("Can't find android/security/Md5MessageDigest");
return -1;
}
fields.context = env->GetFieldID(clazz, "mNativeMd5Context", "I");
if (fields.context == NULL) {
LOGE("Can't find Md5MessageDigest.mNativeMd5Context");
return -1;
}
return jniRegisterNativeMethods(env, "android/security/Md5MessageDigest",
gMethods, NELEM(gMethods));
}
};
|
