src/dalvik_system_DexFile.cc - SHIFTPHONES/android_art - Gitiles

 /*
  * Copyright (C) 2008 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 <unistd.h>

 #include "class_loader.h"
 #include "class_linker.h"
 #include "dex_file.h"
 #include "logging.h"
 #include "os.h"
 #include "runtime.h"
 #include "zip_archive.h"
 #include "toStringArray.h"
 #include "ScopedLocalRef.h"
 #include "ScopedUtfChars.h"

 #include "JniConstants.h" // Last to avoid problems with LOG redefinition.

 namespace art {

 namespace {

 // A smart pointer that provides read-only access to a Java string's UTF chars.
 // Unlike libcore's NullableScopedUtfChars, this will *not* throw NullPointerException if
 // passed a null jstring. The correct idiom is:
 //
 //   NullableScopedUtfChars name(env, javaName);
 //   if (env->ExceptionCheck()) {
 //       return NULL;
 //   }
 //   // ... use name.c_str()
 //
 // TODO: rewrite to get rid of this, or change ScopedUtfChars to offer this option.
 class NullableScopedUtfChars {
  public:
   NullableScopedUtfChars(JNIEnv* env, jstring s) : mEnv(env), mString(s) {
     mUtfChars = (s != NULL) ? env->GetStringUTFChars(s, NULL) : NULL;
   }

   ~NullableScopedUtfChars() {
     if (mUtfChars) {
       mEnv->ReleaseStringUTFChars(mString, mUtfChars);
     }
   }

   const char* c_str() const {
     return mUtfChars;
   }

   size_t size() const {
     return strlen(mUtfChars);
   }

   // Element access.
   const char& operator[](size_t n) const {
     return mUtfChars[n];
   }

  private:
   JNIEnv* mEnv;
   jstring mString;
   const char* mUtfChars;

   // Disallow copy and assignment.
   NullableScopedUtfChars(const NullableScopedUtfChars&);
   void operator=(const NullableScopedUtfChars&);
 };

 static jint DexFile_openDexFile(JNIEnv* env, jclass, jstring javaSourceName, jstring javaOutputName, jint) {
   ScopedUtfChars sourceName(env, javaSourceName);
   if (sourceName.c_str() == NULL) {
     return 0;
   }
   NullableScopedUtfChars outputName(env, javaOutputName);
   if (env->ExceptionCheck()) {
     return 0;
   }
   const DexFile* dex_file;
   if (outputName.c_str() == NULL) {
     dex_file = Runtime::Current()->GetClassLinker()->FindDexFileFromDexLocation(sourceName.c_str());
     if (dex_file == NULL) {
       dex_file = DexFile::Open(sourceName.c_str(), "");
     }
   } else {
     // Sanity check the arguments.
     if (!IsValidZipFilename(sourceName.c_str()) || !IsValidDexFilename(outputName.c_str())) {
       LOG(ERROR) << "Bad filenames extracting dex '" << outputName.c_str()
                  << "' from zip '" << sourceName.c_str() << "'";
       return 0;
     }
     // Generate the output oat file for the source dex file
     ClassLinker* class_linker = Runtime::Current()->GetClassLinker();
     UniquePtr<File> file(OS::OpenFile(outputName.c_str(), true));
     if (file.get() == NULL) {
       LOG(WARNING) << "unable to create oat file: " << outputName.c_str();
       jniThrowExceptionFmt(env, "java/io/IOException", "unable to create oat file: %s",
                            outputName.c_str());
       return 0;
     }
     if (!class_linker->GenerateOatFile(sourceName.c_str(), file->Fd(), outputName.c_str())) {
       LOG(WARNING) << "unable to generate oat file: " << outputName.c_str();
       jniThrowExceptionFmt(env, "java/io/IOException", "unable to generate oat file: %s",
                            outputName.c_str());
       return 0;
     }
     UniquePtr<OatFile> oat_file(OatFile::Open(outputName.c_str(), "", NULL));
     if (oat_file.get() == NULL) {
       LOG(WARNING) << "unable to open oat file: " << outputName.c_str();
       jniThrowExceptionFmt(env, "java/io/IOException", "unable to open oat file: %s",
                            outputName.c_str());
       return 0;
     }
     const OatFile::OatDexFile* oat_dex_file = oat_file->GetOatDexFile(sourceName.c_str());
     if (oat_dex_file == NULL) {
       LOG(WARNING) << "unable to find dex file in oat file: " << outputName.c_str();
       jniThrowExceptionFmt(env, "java/io/IOException", "unable to find dex file in oat file: %s",
                            outputName.c_str());
       return 0;
     }
     Runtime::Current()->GetClassLinker()->RegisterOatFile(*oat_file.release());
     dex_file = oat_dex_file->OpenDexFile();
   }

   if (dex_file == NULL) {
     LOG(WARNING) << "unable to open dex file: " << sourceName.c_str();
     jniThrowExceptionFmt(env, "java/io/IOException", "unable to open dex file: %s",
                          sourceName.c_str());
     return 0;
   }
   return static_cast<jint>(reinterpret_cast<uintptr_t>(dex_file));
 }

 static const DexFile* toDexFile(JNIEnv* env, int dex_file_address) {
   const DexFile* dex_file = reinterpret_cast<const DexFile*>(static_cast<uintptr_t>(dex_file_address));
   if (dex_file == NULL) {
     jniThrowNullPointerException(env, "dex_file == null");
   }
   return dex_file;
 }

 void DexFile_closeDexFile(JNIEnv* env, jclass, jint cookie) {
   const DexFile* dex_file = toDexFile(env, cookie);
   if (dex_file == NULL) {
     return;
   }
   if (Runtime::Current()->GetClassLinker()->IsDexFileRegistered(*dex_file)) {
     return;
   }
   delete dex_file;
 }

 jclass DexFile_defineClass(JNIEnv* env, jclass, jstring javaName, jobject javaLoader, jint cookie) {
   ScopedThreadStateChange tsc(Thread::Current(), Thread::kRunnable);
   const DexFile* dex_file = toDexFile(env, cookie);
   if (dex_file == NULL) {
     return NULL;
   }
   ScopedUtfChars class_name(env, javaName);
   if (class_name.c_str() == NULL) {
     return NULL;
   }
   const std::string descriptor(DotToDescriptor(class_name.c_str()));
   const DexFile::ClassDef* dex_class_def = dex_file->FindClassDef(descriptor);
   if (dex_class_def == NULL) {
     return NULL;
   }

   Object* class_loader_object = Decode<Object*>(env, javaLoader);
   ClassLoader* class_loader = down_cast<ClassLoader*>(class_loader_object);
   ClassLinker* class_linker = Runtime::Current()->GetClassLinker();
   class_linker->RegisterDexFile(*dex_file);
   Class* result = class_linker->DefineClass(descriptor, class_loader, *dex_file, *dex_class_def);
   if (env->ExceptionCheck()) {
     // Swallow any ClassNotFoundException or NoClassDefFoundError; the contract with the caller
     // is that we return null if the class is not found.
     jthrowable exception = env->ExceptionOccurred();
     env->ExceptionClear();

     static jclass ClassNotFoundException_class = CacheClass(env, "java/lang/ClassNotFoundException");
     static jclass NoClassDefFoundError_class = CacheClass(env, "java/lang/NoClassDefFoundError");

     if (!env->IsInstanceOf(exception, ClassNotFoundException_class) && !env->IsInstanceOf(exception, NoClassDefFoundError_class)) {
       env->Throw(exception);
     }
     return NULL;
   }
   return AddLocalReference<jclass>(env, result);
 }

 jobjectArray DexFile_getClassNameList(JNIEnv* env, jclass, jint cookie) {
   const DexFile* dex_file = toDexFile(env, cookie);
   if (dex_file == NULL) {
     return NULL;
   }

   std::vector<std::string> class_names;
   for (size_t i = 0; i < dex_file->NumClassDefs(); ++i) {
     const DexFile::ClassDef& class_def = dex_file->GetClassDef(i);
     const char* descriptor = dex_file->GetClassDescriptor(class_def);
     class_names.push_back(DescriptorToDot(descriptor));
   }
   return toStringArray(env, class_names);
 }

 jboolean DexFile_isDexOptNeeded(JNIEnv* env, jclass, jstring javaFilename) {
   ScopedUtfChars filename(env, javaFilename);
   if (filename.c_str() == NULL) {
     return JNI_TRUE;
   }

   if (!OS::FileExists(filename.c_str())) {
     jniThrowExceptionFmt(env, "java/io/FileNotFoundException", "%s", filename.c_str());
     return JNI_TRUE;
   }

   // Always treat elements of the bootclasspath as up-to-date.  The
   // fact that code is running at all means that this should be true.
   ClassLinker* class_linker = Runtime::Current()->GetClassLinker();
   const std::vector<const DexFile*>& boot_class_path = class_linker->GetBootClassPath();
   for (size_t i = 0; i < boot_class_path.size(); i++) {
     if (boot_class_path[i]->GetLocation() == filename.c_str()) {
       return JNI_FALSE;
     }
   }

   const DexFile* dex_file = class_linker->FindDexFileFromDexLocation(filename.c_str());
   if (dex_file == NULL) {
     return JNI_TRUE;
   }
   return JNI_FALSE;
 }

 static JNINativeMethod gMethods[] = {
   NATIVE_METHOD(DexFile, closeDexFile, "(I)V"),
   NATIVE_METHOD(DexFile, defineClass, "(Ljava/lang/String;Ljava/lang/ClassLoader;I)Ljava/lang/Class;"),
   NATIVE_METHOD(DexFile, getClassNameList, "(I)[Ljava/lang/String;"),
   NATIVE_METHOD(DexFile, isDexOptNeeded, "(Ljava/lang/String;)Z"),
   NATIVE_METHOD(DexFile, openDexFile, "(Ljava/lang/String;Ljava/lang/String;I)I"),
 };

 }  // namespace

 void register_dalvik_system_DexFile(JNIEnv* env) {
   jniRegisterNativeMethods(env, "dalvik/system/DexFile", gMethods, NELEM(gMethods));
 }

 }  // namespace art
	/*
	* Copyright (C) 2008 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 <unistd.h>

	#include "class_loader.h"
	#include "class_linker.h"
	#include "dex_file.h"
	#include "logging.h"
	#include "os.h"
	#include "runtime.h"
	#include "zip_archive.h"
	#include "toStringArray.h"
	#include "ScopedLocalRef.h"
	#include "ScopedUtfChars.h"

	#include "JniConstants.h" // Last to avoid problems with LOG redefinition.

	namespace art {

	namespace {

	// A smart pointer that provides read-only access to a Java string's UTF chars.
	// Unlike libcore's NullableScopedUtfChars, this will not throw NullPointerException if
	// passed a null jstring. The correct idiom is:
	//
	// NullableScopedUtfChars name(env, javaName);
	// if (env->ExceptionCheck()) {
	// return NULL;
	// }
	// // ... use name.c_str()
	//
	// TODO: rewrite to get rid of this, or change ScopedUtfChars to offer this option.
	class NullableScopedUtfChars {
	public:
	NullableScopedUtfChars(JNIEnv* env, jstring s) : mEnv(env), mString(s) {
	mUtfChars = (s != NULL) ? env->GetStringUTFChars(s, NULL) : NULL;
	}

	~NullableScopedUtfChars() {
	if (mUtfChars) {
	mEnv->ReleaseStringUTFChars(mString, mUtfChars);
	}
	}

	const char* c_str() const {
	return mUtfChars;
	}

	size_t size() const {
	return strlen(mUtfChars);
	}

	// Element access.
	const char& operator[](size_t n) const {
	return mUtfChars[n];
	}

	private:
	JNIEnv* mEnv;
	jstring mString;
	const char* mUtfChars;

	// Disallow copy and assignment.
	NullableScopedUtfChars(const NullableScopedUtfChars&);
	void operator=(const NullableScopedUtfChars&);
	};

	static jint DexFile_openDexFile(JNIEnv* env, jclass, jstring javaSourceName, jstring javaOutputName, jint) {
	ScopedUtfChars sourceName(env, javaSourceName);
	if (sourceName.c_str() == NULL) {
	return 0;
	}
	NullableScopedUtfChars outputName(env, javaOutputName);
	if (env->ExceptionCheck()) {
	return 0;
	}
	const DexFile* dex_file;
	if (outputName.c_str() == NULL) {
	dex_file = Runtime::Current()->GetClassLinker()->FindDexFileFromDexLocation(sourceName.c_str());
	if (dex_file == NULL) {
	dex_file = DexFile::Open(sourceName.c_str(), "");
	}
	} else {
	// Sanity check the arguments.
	if (!IsValidZipFilename(sourceName.c_str()) \|\| !IsValidDexFilename(outputName.c_str())) {
	LOG(ERROR) << "Bad filenames extracting dex '" << outputName.c_str()
	<< "' from zip '" << sourceName.c_str() << "'";
	return 0;
	}
	// Generate the output oat file for the source dex file
	ClassLinker* class_linker = Runtime::Current()->GetClassLinker();
	UniquePtr<File> file(OS::OpenFile(outputName.c_str(), true));
	if (file.get() == NULL) {
	LOG(WARNING) << "unable to create oat file: " << outputName.c_str();
	jniThrowExceptionFmt(env, "java/io/IOException", "unable to create oat file: %s",
	outputName.c_str());
	return 0;
	}
	if (!class_linker->GenerateOatFile(sourceName.c_str(), file->Fd(), outputName.c_str())) {
	LOG(WARNING) << "unable to generate oat file: " << outputName.c_str();
	jniThrowExceptionFmt(env, "java/io/IOException", "unable to generate oat file: %s",
	outputName.c_str());
	return 0;
	}
	UniquePtr<OatFile> oat_file(OatFile::Open(outputName.c_str(), "", NULL));
	if (oat_file.get() == NULL) {
	LOG(WARNING) << "unable to open oat file: " << outputName.c_str();
	jniThrowExceptionFmt(env, "java/io/IOException", "unable to open oat file: %s",
	outputName.c_str());
	return 0;
	}
	const OatFile::OatDexFile* oat_dex_file = oat_file->GetOatDexFile(sourceName.c_str());
	if (oat_dex_file == NULL) {
	LOG(WARNING) << "unable to find dex file in oat file: " << outputName.c_str();
	jniThrowExceptionFmt(env, "java/io/IOException", "unable to find dex file in oat file: %s",
	outputName.c_str());
	return 0;
	}
	Runtime::Current()->GetClassLinker()->RegisterOatFile(*oat_file.release());
	dex_file = oat_dex_file->OpenDexFile();
	}

	if (dex_file == NULL) {
	LOG(WARNING) << "unable to open dex file: " << sourceName.c_str();
	jniThrowExceptionFmt(env, "java/io/IOException", "unable to open dex file: %s",
	sourceName.c_str());
	return 0;
	}
	return static_cast<jint>(reinterpret_cast<uintptr_t>(dex_file));
	}

	static const DexFile* toDexFile(JNIEnv* env, int dex_file_address) {
	const DexFile* dex_file = reinterpret_cast<const DexFile*>(static_cast<uintptr_t>(dex_file_address));
	if (dex_file == NULL) {
	jniThrowNullPointerException(env, "dex_file == null");
	}
	return dex_file;
	}

	void DexFile_closeDexFile(JNIEnv* env, jclass, jint cookie) {
	const DexFile* dex_file = toDexFile(env, cookie);
	if (dex_file == NULL) {
	return;
	}
	if (Runtime::Current()->GetClassLinker()->IsDexFileRegistered(*dex_file)) {
	return;
	}
	delete dex_file;
	}

	jclass DexFile_defineClass(JNIEnv* env, jclass, jstring javaName, jobject javaLoader, jint cookie) {
	ScopedThreadStateChange tsc(Thread::Current(), Thread::kRunnable);
	const DexFile* dex_file = toDexFile(env, cookie);
	if (dex_file == NULL) {
	return NULL;
	}
	ScopedUtfChars class_name(env, javaName);
	if (class_name.c_str() == NULL) {
	return NULL;
	}
	const std::string descriptor(DotToDescriptor(class_name.c_str()));
	const DexFile::ClassDef* dex_class_def = dex_file->FindClassDef(descriptor);
	if (dex_class_def == NULL) {
	return NULL;
	}

	Object* class_loader_object = Decode<Object*>(env, javaLoader);
	ClassLoader* class_loader = down_cast<ClassLoader*>(class_loader_object);
	ClassLinker* class_linker = Runtime::Current()->GetClassLinker();
	class_linker->RegisterDexFile(*dex_file);
	Class* result = class_linker->DefineClass(descriptor, class_loader, dex_file, dex_class_def);
	if (env->ExceptionCheck()) {
	// Swallow any ClassNotFoundException or NoClassDefFoundError; the contract with the caller
	// is that we return null if the class is not found.
	jthrowable exception = env->ExceptionOccurred();
	env->ExceptionClear();

	static jclass ClassNotFoundException_class = CacheClass(env, "java/lang/ClassNotFoundException");
	static jclass NoClassDefFoundError_class = CacheClass(env, "java/lang/NoClassDefFoundError");

	if (!env->IsInstanceOf(exception, ClassNotFoundException_class) && !env->IsInstanceOf(exception, NoClassDefFoundError_class)) {
	env->Throw(exception);
	}
	return NULL;
	}
	return AddLocalReference<jclass>(env, result);
	}

	jobjectArray DexFile_getClassNameList(JNIEnv* env, jclass, jint cookie) {
	const DexFile* dex_file = toDexFile(env, cookie);
	if (dex_file == NULL) {
	return NULL;
	}

	std::vector<std::string> class_names;
	for (size_t i = 0; i < dex_file->NumClassDefs(); ++i) {
	const DexFile::ClassDef& class_def = dex_file->GetClassDef(i);
	const char* descriptor = dex_file->GetClassDescriptor(class_def);
	class_names.push_back(DescriptorToDot(descriptor));
	}
	return toStringArray(env, class_names);
	}

	jboolean DexFile_isDexOptNeeded(JNIEnv* env, jclass, jstring javaFilename) {
	ScopedUtfChars filename(env, javaFilename);
	if (filename.c_str() == NULL) {
	return JNI_TRUE;
	}

	if (!OS::FileExists(filename.c_str())) {
	jniThrowExceptionFmt(env, "java/io/FileNotFoundException", "%s", filename.c_str());
	return JNI_TRUE;
	}

	// Always treat elements of the bootclasspath as up-to-date. The
	// fact that code is running at all means that this should be true.
	ClassLinker* class_linker = Runtime::Current()->GetClassLinker();
	const std::vector<const DexFile*>& boot_class_path = class_linker->GetBootClassPath();
	for (size_t i = 0; i < boot_class_path.size(); i++) {
	if (boot_class_path[i]->GetLocation() == filename.c_str()) {
	return JNI_FALSE;
	}
	}

	const DexFile* dex_file = class_linker->FindDexFileFromDexLocation(filename.c_str());
	if (dex_file == NULL) {
	return JNI_TRUE;
	}
	return JNI_FALSE;
	}

	static JNINativeMethod gMethods[] = {
	NATIVE_METHOD(DexFile, closeDexFile, "(I)V"),
	NATIVE_METHOD(DexFile, defineClass, "(Ljava/lang/String;Ljava/lang/ClassLoader;I)Ljava/lang/Class;"),
	NATIVE_METHOD(DexFile, getClassNameList, "(I)[Ljava/lang/String;"),
	NATIVE_METHOD(DexFile, isDexOptNeeded, "(Ljava/lang/String;)Z"),
	NATIVE_METHOD(DexFile, openDexFile, "(Ljava/lang/String;Ljava/lang/String;I)I"),
	};

	} // namespace

	void register_dalvik_system_DexFile(JNIEnv* env) {
	jniRegisterNativeMethods(env, "dalvik/system/DexFile", gMethods, NELEM(gMethods));
	}

	} // namespace art