| /* |
| * Copyright (C) 2017 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 <inttypes.h> |
| #include <stdio.h> |
| #include <string.h> |
| |
| #include <iostream> |
| #include <vector> |
| |
| #include "android-base/stringprintf.h" |
| |
| #include "base/logging.h" |
| #include "base/macros.h" |
| #include "bytecode_utils.h" |
| #include "dex_file.h" |
| #include "dex_instruction.h" |
| #include "jit/jit.h" |
| #include "jni.h" |
| #include "native_stack_dump.h" |
| #include "jvmti.h" |
| #include "runtime.h" |
| #include "scoped_thread_state_change-inl.h" |
| #include "thread-current-inl.h" |
| #include "thread_list.h" |
| |
| // Test infrastructure |
| #include "jvmti_helper.h" |
| #include "test_env.h" |
| |
| namespace art { |
| namespace Test983SourceTransformVerify { |
| |
| constexpr bool kSkipInitialLoad = true; |
| |
| // The hook we are using. |
| void JNICALL CheckDexFileHook(jvmtiEnv* jvmti_env ATTRIBUTE_UNUSED, |
| JNIEnv* jni_env ATTRIBUTE_UNUSED, |
| jclass class_being_redefined, |
| jobject loader ATTRIBUTE_UNUSED, |
| const char* name, |
| jobject protection_domain ATTRIBUTE_UNUSED, |
| jint class_data_len, |
| const unsigned char* class_data, |
| jint* new_class_data_len ATTRIBUTE_UNUSED, |
| unsigned char** new_class_data ATTRIBUTE_UNUSED) { |
| if (kSkipInitialLoad && class_being_redefined == nullptr) { |
| // Something got loaded concurrently. Just ignore it for now. |
| return; |
| } |
| std::cout << "Dex file hook for " << name << std::endl; |
| if (IsJVM()) { |
| return; |
| } |
| std::string error; |
| std::unique_ptr<const DexFile> dex(DexFile::Open(class_data, |
| class_data_len, |
| "fake_location.dex", |
| /*location_checksum*/ 0, |
| /*oat_dex_file*/ nullptr, |
| /*verify*/ true, |
| /*verify_checksum*/ true, |
| &error)); |
| if (dex.get() == nullptr) { |
| std::cout << "Failed to verify dex file for " << name << " because " << error << std::endl; |
| return; |
| } |
| for (uint32_t i = 0; i < dex->NumClassDefs(); i++) { |
| const DexFile::ClassDef& def = dex->GetClassDef(i); |
| const uint8_t* data_item = dex->GetClassData(def); |
| if (data_item == nullptr) { |
| continue; |
| } |
| for (ClassDataItemIterator it(*dex, data_item); it.HasNext(); it.Next()) { |
| if (!it.IsAtMethod() || it.GetMethodCodeItem() == nullptr) { |
| continue; |
| } |
| for (CodeItemIterator code_it(*it.GetMethodCodeItem()); !code_it.Done(); code_it.Advance()) { |
| const Instruction& inst = code_it.CurrentInstruction(); |
| int forbiden_flags = (Instruction::kVerifyError | Instruction::kVerifyRuntimeOnly); |
| if (inst.Opcode() == Instruction::RETURN_VOID_NO_BARRIER || |
| (inst.GetVerifyExtraFlags() & forbiden_flags) != 0) { |
| std::cout << "Unexpected instruction found in " << dex->PrettyMethod(it.GetMemberIndex()) |
| << " [Dex PC: 0x" << std::hex << code_it.CurrentDexPc() << std::dec << "] : " |
| << inst.DumpString(dex.get()) << std::endl; |
| continue; |
| } |
| } |
| } |
| } |
| } |
| |
| // Get all capabilities except those related to retransformation. |
| jint OnLoad(JavaVM* vm, |
| char* options ATTRIBUTE_UNUSED, |
| void* reserved ATTRIBUTE_UNUSED) { |
| if (vm->GetEnv(reinterpret_cast<void**>(&jvmti_env), JVMTI_VERSION_1_0)) { |
| printf("Unable to get jvmti env!\n"); |
| return 1; |
| } |
| SetStandardCapabilities(jvmti_env); |
| jvmtiEventCallbacks cb; |
| memset(&cb, 0, sizeof(cb)); |
| cb.ClassFileLoadHook = CheckDexFileHook; |
| if (jvmti_env->SetEventCallbacks(&cb, sizeof(cb)) != JVMTI_ERROR_NONE) { |
| printf("Unable to set class file load hook cb!\n"); |
| return 1; |
| } |
| return 0; |
| } |
| |
| } // namespace Test983SourceTransformVerify |
| } // namespace art |