blob: b62fc015225c4ab1f864c80bd7ea22b555a6bcab [file] [log] [blame]
/*
* Copyright (C) 2019 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 "common_compiler_test.h"
#include "aot_class_linker.h"
#include "base/casts.h"
#include "class_linker-inl.h"
#include "handle.h"
#include "handle_scope-inl.h"
#include "mirror/class_loader-inl.h"
#include "mirror/dex_cache.h"
#include "mirror/object-inl.h"
#include "runtime.h"
#include "scoped_thread_state_change-inl.h"
#include "thread-current-inl.h"
#include "well_known_classes.h"
namespace art {
class ModuleExclusionTest : public CommonCompilerTest {
public:
explicit ModuleExclusionTest(const std::string& module)
: CommonCompilerTest(),
module_(module) {}
std::vector<std::string> GetLibCoreModuleNames() const override {
std::vector<std::string> modules = CommonCompilerTest::GetLibCoreModuleNames();
// Exclude `module_` from boot class path.
auto it = std::find(modules.begin(), modules.end(), module_);
if (it != modules.end()) {
modules.erase(it);
}
return modules;
}
void DoTest() {
Thread* self = Thread::Current();
ScopedObjectAccess soa(self);
StackHandleScope<2u> hs(self);
Runtime* runtime = Runtime::Current();
ASSERT_TRUE(runtime->IsAotCompiler());
ClassLinker* class_linker = runtime->GetClassLinker();
CHECK(loaded_dex_files_.empty());
Handle<mirror::ClassLoader> class_loader = hs.NewHandle(LoadModule(soa, class_linker));
MutableHandle<mirror::DexCache> dex_cache = hs.NewHandle<mirror::DexCache>(nullptr);
CHECK(!loaded_dex_files_.empty());
// Verify that classes defined in the loaded dex files cannot be resolved.
for (const std::unique_ptr<const DexFile>& dex_file : loaded_dex_files_) {
dex_cache.Assign(class_linker->RegisterDexFile(*dex_file, class_loader.Get()));
for (size_t i = 0u, size = dex_file->NumClassDefs(); i != size; ++i) {
const dex::ClassDef& class_def = dex_file->GetClassDef(i);
ObjPtr<mirror::Class> resolved_type =
class_linker->ResolveType(class_def.class_idx_, dex_cache, class_loader);
ASSERT_TRUE(resolved_type == nullptr) << resolved_type->PrettyDescriptor();
ASSERT_TRUE(self->IsExceptionPending());
self->ClearException();
}
}
}
protected:
void SetUpRuntimeOptions(RuntimeOptions* options) override {
CommonCompilerTest::SetUpRuntimeOptions(options);
// Set up the image location to be used by StartDex2OatCommandLine().
// Using a prebuilt image also makes the test run faster.
options->push_back(std::make_pair("-Ximage:" + GetImageLocation(), nullptr));
}
std::string GetModuleFileName() const {
std::vector<std::string> filename = GetLibCoreDexFileNames({ module_ });
CHECK_EQ(filename.size(), 1u);
return filename[0];
}
// Load the module as an app, i.e. in a class loader other than the boot class loader.
ObjPtr<mirror::ClassLoader> LoadModule(ScopedObjectAccess& soa, ClassLinker* class_linker)
REQUIRES_SHARED(Locks::mutator_lock_) {
std::string filename = GetModuleFileName();
std::vector<std::unique_ptr<const DexFile>> dex_files = OpenDexFiles(filename.c_str());
std::vector<const DexFile*> class_path;
CHECK_NE(0U, dex_files.size());
for (auto& dex_file : dex_files) {
class_path.push_back(dex_file.get());
loaded_dex_files_.push_back(std::move(dex_file));
}
StackHandleScope<1u> hs(soa.Self());
Handle<mirror::Class> loader_class(hs.NewHandle(
soa.Decode<mirror::Class>(WellKnownClasses::dalvik_system_PathClassLoader)));
ScopedNullHandle<mirror::ClassLoader> parent_loader;
ScopedNullHandle<mirror::ObjectArray<mirror::ClassLoader>> shared_libraries;
ObjPtr<mirror::ClassLoader> result = class_linker->CreateWellKnownClassLoader(
soa.Self(),
class_path,
loader_class,
parent_loader,
shared_libraries);
// Verify that the result has the correct class.
CHECK_EQ(loader_class.Get(), result->GetClass());
// Verify that the parent is not null. The boot class loader will be set up as a
// proper BootClassLoader object.
ObjPtr<mirror::ClassLoader> actual_parent(result->GetParent());
CHECK(actual_parent != nullptr);
CHECK(class_linker->IsBootClassLoader(soa, actual_parent));
return result;
}
const std::string module_;
};
class ConscryptExclusionTest : public ModuleExclusionTest {
public:
ConscryptExclusionTest() : ModuleExclusionTest("conscrypt") {}
};
TEST_F(ConscryptExclusionTest, Test) {
Runtime* runtime = Runtime::Current();
ASSERT_TRUE(runtime->IsAotCompiler());
AotClassLinker* aot_class_linker = down_cast<AotClassLinker*>(runtime->GetClassLinker());
const std::vector<std::string> package_list = {
// Reserved conscrypt packages (includes sub-packages under these paths).
"android.net.ssl",
"com.android.org.conscrypt",
};
bool list_applied = aot_class_linker->SetUpdatableBootClassPackages(package_list);
ASSERT_TRUE(list_applied);
DoTest();
// Also test passing the list to dex2oat.
ScratchFile package_list_file;
for (const std::string& package : package_list) {
std::string data = package + '\n';
ASSERT_TRUE(package_list_file.GetFile()->WriteFully(data.data(), data.size()));
}
ASSERT_EQ(0, package_list_file.GetFile()->Flush());
ScratchDir scratch_dir;
std::string jar_name = GetModuleFileName();
std::string odex_name = scratch_dir.GetPath() + module_ + ".odex";
std::vector<std::string> argv;
std::string error_msg;
bool success = StartDex2OatCommandLine(&argv, &error_msg);
ASSERT_TRUE(success) << error_msg;
argv.insert(argv.end(), {
"--dex-file=" + jar_name,
"--dex-location=" + jar_name,
"--oat-file=" + odex_name,
"--compiler-filter=speed",
"--updatable-bcp-packages-file=" + package_list_file.GetFilename()
});
success = RunDex2Oat(argv, &error_msg);
ASSERT_TRUE(success) << error_msg;
// Load the odex file.
std::unique_ptr<OatFile> odex_file(OatFile::Open(/*zip_fd=*/ -1,
odex_name.c_str(),
odex_name.c_str(),
/*executable=*/ false,
/*low_4gb=*/ false,
jar_name,
&error_msg));
ASSERT_TRUE(odex_file != nullptr) << error_msg;
// Check that no classes have been initialized.
for (const OatDexFile* oat_dex_file : odex_file->GetOatDexFiles()) {
std::unique_ptr<const DexFile> dex_file = oat_dex_file->OpenDexFile(&error_msg);
ASSERT_TRUE(dex_file != nullptr);
for (size_t i = 0, num_class_defs = dex_file->NumClassDefs(); i != num_class_defs; ++i) {
ClassStatus status = oat_dex_file->GetOatClass(i).GetStatus();
ASSERT_FALSE(mirror::Class::IsErroneous(status));
ASSERT_LE(status, ClassStatus::kVerified);
}
}
}
} // namespace art