126 lines
4.6 KiB
C++
126 lines
4.6 KiB
C++
/*
|
|
* 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;
|
|
}
|
|
SetAllCapabilities(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
|