upload android base code part3
This commit is contained in:
parent
71b83c22f1
commit
b9e30e05b1
15122 changed files with 2089659 additions and 0 deletions
435
android/art/runtime/runtime_callbacks_test.cc
Normal file
435
android/art/runtime/runtime_callbacks_test.cc
Normal file
|
@ -0,0 +1,435 @@
|
|||
/*
|
||||
* 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 "runtime_callbacks.h"
|
||||
|
||||
#include "jni.h"
|
||||
#include <signal.h>
|
||||
#include <sys/types.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include <initializer_list>
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
||||
#include "art_method-inl.h"
|
||||
#include "base/mutex.h"
|
||||
#include "class_linker.h"
|
||||
#include "common_runtime_test.h"
|
||||
#include "handle.h"
|
||||
#include "handle_scope-inl.h"
|
||||
#include "mem_map.h"
|
||||
#include "mirror/class-inl.h"
|
||||
#include "mirror/class_loader.h"
|
||||
#include "nativehelper/ScopedLocalRef.h"
|
||||
#include "obj_ptr.h"
|
||||
#include "runtime.h"
|
||||
#include "scoped_thread_state_change-inl.h"
|
||||
#include "thread-inl.h"
|
||||
#include "thread_list.h"
|
||||
#include "well_known_classes.h"
|
||||
|
||||
namespace art {
|
||||
|
||||
class RuntimeCallbacksTest : public CommonRuntimeTest {
|
||||
protected:
|
||||
void SetUp() OVERRIDE {
|
||||
CommonRuntimeTest::SetUp();
|
||||
|
||||
Thread* self = Thread::Current();
|
||||
ScopedObjectAccess soa(self);
|
||||
ScopedThreadSuspension sts(self, kWaitingForDebuggerToAttach);
|
||||
ScopedSuspendAll ssa("RuntimeCallbacksTest SetUp");
|
||||
AddListener();
|
||||
}
|
||||
|
||||
void TearDown() OVERRIDE {
|
||||
{
|
||||
Thread* self = Thread::Current();
|
||||
ScopedObjectAccess soa(self);
|
||||
ScopedThreadSuspension sts(self, kWaitingForDebuggerToAttach);
|
||||
ScopedSuspendAll ssa("RuntimeCallbacksTest TearDown");
|
||||
RemoveListener();
|
||||
}
|
||||
|
||||
CommonRuntimeTest::TearDown();
|
||||
}
|
||||
|
||||
virtual void AddListener() REQUIRES(Locks::mutator_lock_) = 0;
|
||||
virtual void RemoveListener() REQUIRES(Locks::mutator_lock_) = 0;
|
||||
|
||||
void MakeExecutable(ObjPtr<mirror::Class> klass) REQUIRES_SHARED(Locks::mutator_lock_) {
|
||||
CHECK(klass != nullptr);
|
||||
PointerSize pointer_size = class_linker_->GetImagePointerSize();
|
||||
for (auto& m : klass->GetMethods(pointer_size)) {
|
||||
if (!m.IsAbstract()) {
|
||||
class_linker_->SetEntryPointsToInterpreter(&m);
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
class ThreadLifecycleCallbackRuntimeCallbacksTest : public RuntimeCallbacksTest {
|
||||
public:
|
||||
static void* PthreadsCallback(void* arg ATTRIBUTE_UNUSED) {
|
||||
// Attach.
|
||||
Runtime* runtime = Runtime::Current();
|
||||
CHECK(runtime->AttachCurrentThread("ThreadLifecycle test thread", true, nullptr, false));
|
||||
|
||||
// Detach.
|
||||
runtime->DetachCurrentThread();
|
||||
|
||||
// Die...
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
protected:
|
||||
void AddListener() OVERRIDE REQUIRES(Locks::mutator_lock_) {
|
||||
Runtime::Current()->GetRuntimeCallbacks()->AddThreadLifecycleCallback(&cb_);
|
||||
}
|
||||
void RemoveListener() OVERRIDE REQUIRES(Locks::mutator_lock_) {
|
||||
Runtime::Current()->GetRuntimeCallbacks()->RemoveThreadLifecycleCallback(&cb_);
|
||||
}
|
||||
|
||||
enum CallbackState {
|
||||
kBase,
|
||||
kStarted,
|
||||
kDied,
|
||||
kWrongStart,
|
||||
kWrongDeath,
|
||||
};
|
||||
|
||||
struct Callback : public ThreadLifecycleCallback {
|
||||
void ThreadStart(Thread* self) OVERRIDE {
|
||||
if (state == CallbackState::kBase) {
|
||||
state = CallbackState::kStarted;
|
||||
stored_self = self;
|
||||
} else {
|
||||
state = CallbackState::kWrongStart;
|
||||
}
|
||||
}
|
||||
|
||||
void ThreadDeath(Thread* self) OVERRIDE {
|
||||
if (state == CallbackState::kStarted && self == stored_self) {
|
||||
state = CallbackState::kDied;
|
||||
} else {
|
||||
state = CallbackState::kWrongDeath;
|
||||
}
|
||||
}
|
||||
|
||||
Thread* stored_self;
|
||||
CallbackState state = CallbackState::kBase;
|
||||
};
|
||||
|
||||
Callback cb_;
|
||||
};
|
||||
|
||||
TEST_F(ThreadLifecycleCallbackRuntimeCallbacksTest, ThreadLifecycleCallbackJava) {
|
||||
Thread* self = Thread::Current();
|
||||
|
||||
self->TransitionFromSuspendedToRunnable();
|
||||
bool started = runtime_->Start();
|
||||
ASSERT_TRUE(started);
|
||||
|
||||
cb_.state = CallbackState::kBase; // Ignore main thread attach.
|
||||
|
||||
{
|
||||
ScopedObjectAccess soa(self);
|
||||
MakeExecutable(soa.Decode<mirror::Class>(WellKnownClasses::java_lang_Thread));
|
||||
}
|
||||
|
||||
JNIEnv* env = self->GetJniEnv();
|
||||
|
||||
ScopedLocalRef<jobject> thread_name(env,
|
||||
env->NewStringUTF("ThreadLifecycleCallback test thread"));
|
||||
ASSERT_TRUE(thread_name.get() != nullptr);
|
||||
|
||||
ScopedLocalRef<jobject> thread(env, env->AllocObject(WellKnownClasses::java_lang_Thread));
|
||||
ASSERT_TRUE(thread.get() != nullptr);
|
||||
|
||||
env->CallNonvirtualVoidMethod(thread.get(),
|
||||
WellKnownClasses::java_lang_Thread,
|
||||
WellKnownClasses::java_lang_Thread_init,
|
||||
runtime_->GetMainThreadGroup(),
|
||||
thread_name.get(),
|
||||
kMinThreadPriority,
|
||||
JNI_FALSE);
|
||||
ASSERT_FALSE(env->ExceptionCheck());
|
||||
|
||||
jmethodID start_id = env->GetMethodID(WellKnownClasses::java_lang_Thread, "start", "()V");
|
||||
ASSERT_TRUE(start_id != nullptr);
|
||||
|
||||
env->CallVoidMethod(thread.get(), start_id);
|
||||
ASSERT_FALSE(env->ExceptionCheck());
|
||||
|
||||
jmethodID join_id = env->GetMethodID(WellKnownClasses::java_lang_Thread, "join", "()V");
|
||||
ASSERT_TRUE(join_id != nullptr);
|
||||
|
||||
env->CallVoidMethod(thread.get(), join_id);
|
||||
ASSERT_FALSE(env->ExceptionCheck());
|
||||
|
||||
EXPECT_TRUE(cb_.state == CallbackState::kDied) << static_cast<int>(cb_.state);
|
||||
}
|
||||
|
||||
TEST_F(ThreadLifecycleCallbackRuntimeCallbacksTest, ThreadLifecycleCallbackAttach) {
|
||||
std::string error_msg;
|
||||
std::unique_ptr<MemMap> stack(MemMap::MapAnonymous("ThreadLifecycleCallback Thread",
|
||||
nullptr,
|
||||
128 * kPageSize, // Just some small stack.
|
||||
PROT_READ | PROT_WRITE,
|
||||
false,
|
||||
false,
|
||||
&error_msg));
|
||||
ASSERT_FALSE(stack == nullptr) << error_msg;
|
||||
|
||||
const char* reason = "ThreadLifecycleCallback test thread";
|
||||
pthread_attr_t attr;
|
||||
CHECK_PTHREAD_CALL(pthread_attr_init, (&attr), reason);
|
||||
CHECK_PTHREAD_CALL(pthread_attr_setstack, (&attr, stack->Begin(), stack->Size()), reason);
|
||||
pthread_t pthread;
|
||||
CHECK_PTHREAD_CALL(pthread_create,
|
||||
(&pthread,
|
||||
&attr,
|
||||
&ThreadLifecycleCallbackRuntimeCallbacksTest::PthreadsCallback,
|
||||
this),
|
||||
reason);
|
||||
CHECK_PTHREAD_CALL(pthread_attr_destroy, (&attr), reason);
|
||||
|
||||
CHECK_PTHREAD_CALL(pthread_join, (pthread, nullptr), "ThreadLifecycleCallback test shutdown");
|
||||
|
||||
// Detach is not a ThreadDeath event, so we expect to be in state Started.
|
||||
EXPECT_TRUE(cb_.state == CallbackState::kStarted) << static_cast<int>(cb_.state);
|
||||
}
|
||||
|
||||
class ClassLoadCallbackRuntimeCallbacksTest : public RuntimeCallbacksTest {
|
||||
protected:
|
||||
void AddListener() OVERRIDE REQUIRES(Locks::mutator_lock_) {
|
||||
Runtime::Current()->GetRuntimeCallbacks()->AddClassLoadCallback(&cb_);
|
||||
}
|
||||
void RemoveListener() OVERRIDE REQUIRES(Locks::mutator_lock_) {
|
||||
Runtime::Current()->GetRuntimeCallbacks()->RemoveClassLoadCallback(&cb_);
|
||||
}
|
||||
|
||||
bool Expect(std::initializer_list<const char*> list) {
|
||||
if (cb_.data.size() != list.size()) {
|
||||
PrintError(list);
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!std::equal(cb_.data.begin(), cb_.data.end(), list.begin())) {
|
||||
PrintError(list);
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void PrintError(std::initializer_list<const char*> list) {
|
||||
LOG(ERROR) << "Expected:";
|
||||
for (const char* expected : list) {
|
||||
LOG(ERROR) << " " << expected;
|
||||
}
|
||||
LOG(ERROR) << "Found:";
|
||||
for (const auto& s : cb_.data) {
|
||||
LOG(ERROR) << " " << s;
|
||||
}
|
||||
}
|
||||
|
||||
struct Callback : public ClassLoadCallback {
|
||||
virtual void ClassPreDefine(const char* descriptor,
|
||||
Handle<mirror::Class> klass ATTRIBUTE_UNUSED,
|
||||
Handle<mirror::ClassLoader> class_loader ATTRIBUTE_UNUSED,
|
||||
const DexFile& initial_dex_file,
|
||||
const DexFile::ClassDef& initial_class_def ATTRIBUTE_UNUSED,
|
||||
/*out*/DexFile const** final_dex_file ATTRIBUTE_UNUSED,
|
||||
/*out*/DexFile::ClassDef const** final_class_def ATTRIBUTE_UNUSED)
|
||||
OVERRIDE REQUIRES_SHARED(Locks::mutator_lock_) {
|
||||
std::string location(initial_dex_file.GetLocation());
|
||||
std::string event =
|
||||
std::string("PreDefine:") + descriptor + " <" +
|
||||
location.substr(location.rfind("/") + 1, location.size()) + ">";
|
||||
data.push_back(event);
|
||||
}
|
||||
|
||||
void ClassLoad(Handle<mirror::Class> klass) OVERRIDE REQUIRES_SHARED(Locks::mutator_lock_) {
|
||||
std::string tmp;
|
||||
std::string event = std::string("Load:") + klass->GetDescriptor(&tmp);
|
||||
data.push_back(event);
|
||||
}
|
||||
|
||||
void ClassPrepare(Handle<mirror::Class> temp_klass,
|
||||
Handle<mirror::Class> klass) OVERRIDE REQUIRES_SHARED(Locks::mutator_lock_) {
|
||||
std::string tmp, tmp2;
|
||||
std::string event = std::string("Prepare:") + klass->GetDescriptor(&tmp)
|
||||
+ "[" + temp_klass->GetDescriptor(&tmp2) + "]";
|
||||
data.push_back(event);
|
||||
}
|
||||
|
||||
std::vector<std::string> data;
|
||||
};
|
||||
|
||||
Callback cb_;
|
||||
};
|
||||
|
||||
TEST_F(ClassLoadCallbackRuntimeCallbacksTest, ClassLoadCallback) {
|
||||
ScopedObjectAccess soa(Thread::Current());
|
||||
jobject jclass_loader = LoadDex("XandY");
|
||||
VariableSizedHandleScope hs(soa.Self());
|
||||
Handle<mirror::ClassLoader> class_loader(hs.NewHandle(
|
||||
soa.Decode<mirror::ClassLoader>(jclass_loader)));
|
||||
|
||||
const char* descriptor_y = "LY;";
|
||||
Handle<mirror::Class> h_Y(
|
||||
hs.NewHandle(class_linker_->FindClass(soa.Self(), descriptor_y, class_loader)));
|
||||
ASSERT_TRUE(h_Y != nullptr);
|
||||
|
||||
bool expect1 = Expect({ "PreDefine:LY; <art-gtest-XandY.jar>",
|
||||
"PreDefine:LX; <art-gtest-XandY.jar>",
|
||||
"Load:LX;",
|
||||
"Prepare:LX;[LX;]",
|
||||
"Load:LY;",
|
||||
"Prepare:LY;[LY;]" });
|
||||
EXPECT_TRUE(expect1);
|
||||
|
||||
cb_.data.clear();
|
||||
|
||||
ASSERT_TRUE(class_linker_->EnsureInitialized(Thread::Current(), h_Y, true, true));
|
||||
|
||||
bool expect2 = Expect({ "PreDefine:LY$Z; <art-gtest-XandY.jar>",
|
||||
"Load:LY$Z;",
|
||||
"Prepare:LY$Z;[LY$Z;]" });
|
||||
EXPECT_TRUE(expect2);
|
||||
}
|
||||
|
||||
class RuntimeSigQuitCallbackRuntimeCallbacksTest : public RuntimeCallbacksTest {
|
||||
protected:
|
||||
void AddListener() OVERRIDE REQUIRES(Locks::mutator_lock_) {
|
||||
Runtime::Current()->GetRuntimeCallbacks()->AddRuntimeSigQuitCallback(&cb_);
|
||||
}
|
||||
void RemoveListener() OVERRIDE REQUIRES(Locks::mutator_lock_) {
|
||||
Runtime::Current()->GetRuntimeCallbacks()->RemoveRuntimeSigQuitCallback(&cb_);
|
||||
}
|
||||
|
||||
struct Callback : public RuntimeSigQuitCallback {
|
||||
void SigQuit() OVERRIDE {
|
||||
++sigquit_count;
|
||||
}
|
||||
|
||||
size_t sigquit_count = 0;
|
||||
};
|
||||
|
||||
Callback cb_;
|
||||
};
|
||||
|
||||
TEST_F(RuntimeSigQuitCallbackRuntimeCallbacksTest, SigQuit) {
|
||||
// SigQuit induces a dump. ASAN isn't happy with libunwind reading memory.
|
||||
TEST_DISABLED_FOR_MEMORY_TOOL_ASAN();
|
||||
|
||||
// The runtime needs to be started for the signal handler.
|
||||
Thread* self = Thread::Current();
|
||||
|
||||
self->TransitionFromSuspendedToRunnable();
|
||||
bool started = runtime_->Start();
|
||||
ASSERT_TRUE(started);
|
||||
|
||||
EXPECT_EQ(0u, cb_.sigquit_count);
|
||||
|
||||
kill(getpid(), SIGQUIT);
|
||||
|
||||
// Try a few times.
|
||||
for (size_t i = 0; i != 30; ++i) {
|
||||
if (cb_.sigquit_count == 0) {
|
||||
sleep(1);
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
EXPECT_EQ(1u, cb_.sigquit_count);
|
||||
}
|
||||
|
||||
class RuntimePhaseCallbackRuntimeCallbacksTest : public RuntimeCallbacksTest {
|
||||
protected:
|
||||
void AddListener() OVERRIDE REQUIRES(Locks::mutator_lock_) {
|
||||
Runtime::Current()->GetRuntimeCallbacks()->AddRuntimePhaseCallback(&cb_);
|
||||
}
|
||||
void RemoveListener() OVERRIDE REQUIRES(Locks::mutator_lock_) {
|
||||
Runtime::Current()->GetRuntimeCallbacks()->RemoveRuntimePhaseCallback(&cb_);
|
||||
}
|
||||
|
||||
void TearDown() OVERRIDE {
|
||||
// Bypass RuntimeCallbacksTest::TearDown, as the runtime is already gone.
|
||||
CommonRuntimeTest::TearDown();
|
||||
}
|
||||
|
||||
struct Callback : public RuntimePhaseCallback {
|
||||
void NextRuntimePhase(RuntimePhaseCallback::RuntimePhase p) OVERRIDE {
|
||||
if (p == RuntimePhaseCallback::RuntimePhase::kInitialAgents) {
|
||||
if (start_seen > 0 || init_seen > 0 || death_seen > 0) {
|
||||
LOG(FATAL) << "Unexpected order";
|
||||
}
|
||||
++initial_agents_seen;
|
||||
} else if (p == RuntimePhaseCallback::RuntimePhase::kStart) {
|
||||
if (init_seen > 0 || death_seen > 0) {
|
||||
LOG(FATAL) << "Init seen before start.";
|
||||
}
|
||||
++start_seen;
|
||||
} else if (p == RuntimePhaseCallback::RuntimePhase::kInit) {
|
||||
++init_seen;
|
||||
} else if (p == RuntimePhaseCallback::RuntimePhase::kDeath) {
|
||||
++death_seen;
|
||||
} else {
|
||||
LOG(FATAL) << "Unknown phase " << static_cast<uint32_t>(p);
|
||||
}
|
||||
}
|
||||
|
||||
size_t initial_agents_seen = 0;
|
||||
size_t start_seen = 0;
|
||||
size_t init_seen = 0;
|
||||
size_t death_seen = 0;
|
||||
};
|
||||
|
||||
Callback cb_;
|
||||
};
|
||||
|
||||
TEST_F(RuntimePhaseCallbackRuntimeCallbacksTest, Phases) {
|
||||
ASSERT_EQ(0u, cb_.initial_agents_seen);
|
||||
ASSERT_EQ(0u, cb_.start_seen);
|
||||
ASSERT_EQ(0u, cb_.init_seen);
|
||||
ASSERT_EQ(0u, cb_.death_seen);
|
||||
|
||||
// Start the runtime.
|
||||
{
|
||||
Thread* self = Thread::Current();
|
||||
self->TransitionFromSuspendedToRunnable();
|
||||
bool started = runtime_->Start();
|
||||
ASSERT_TRUE(started);
|
||||
}
|
||||
|
||||
ASSERT_EQ(0u, cb_.initial_agents_seen);
|
||||
ASSERT_EQ(1u, cb_.start_seen);
|
||||
ASSERT_EQ(1u, cb_.init_seen);
|
||||
ASSERT_EQ(0u, cb_.death_seen);
|
||||
|
||||
// Delete the runtime.
|
||||
runtime_.reset();
|
||||
|
||||
ASSERT_EQ(0u, cb_.initial_agents_seen);
|
||||
ASSERT_EQ(1u, cb_.start_seen);
|
||||
ASSERT_EQ(1u, cb_.init_seen);
|
||||
ASSERT_EQ(1u, cb_.death_seen);
|
||||
}
|
||||
|
||||
} // namespace art
|
Loading…
Add table
Add a link
Reference in a new issue