auto import from //depot/cupcake/@135843

1 files changed, 126 insertions, 0 deletions

diff --git a/core/jni/android_security_Md5MessageDigest.cpp b/core/jni/android_security_Md5MessageDigest.cpp
new file mode 100644
index 0000000..3533559
--- /dev/null
+++ b/core/jni/android_security_Md5MessageDigest.cpp
@@ -0,0 +1,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));
+}
+
+};