1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
/*
* Copyright (C) 2022 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 <sys/wait.h>
#include "android-base/result-gmock.h"
#include "android-base/result.h"
#include "android-base/strings.h"
#include "base/file_utils.h"
#include "dex2oat_environment_test.h"
namespace art {
using ::android::base::Result;
using ::android::base::testing::HasValue;
// Test the binary with the same bitness as the test. This is also done to avoid
// the symlink /apex/com.android.art/bin/dex2oat, which we don't have selinux
// permission to read on S.
#if defined(__LP64__)
constexpr const char* kDex2oatBinary = "dex2oat64";
#else
constexpr const char* kDex2oatBinary = "dex2oat32";
#endif
class Dex2oatCtsTest : public CommonArtTest, public Dex2oatScratchDirs {
public:
void SetUp() override {
CommonArtTest::SetUp();
Dex2oatScratchDirs::SetUp(android_data_);
}
void TearDown() override {
Dex2oatScratchDirs::TearDown();
CommonArtTest::TearDown();
}
protected:
// Stripped down counterpart to Dex2oatEnvironmentTest::Dex2Oat that only adds
// enough arguments for our purposes.
Result<int> Dex2Oat(const std::vector<std::string>& dex2oat_args, std::string* output) {
std::vector<std::string> argv = {std::string(kAndroidArtApexDefaultPath) + "/bin/" +
kDex2oatBinary};
argv.insert(argv.end(), dex2oat_args.begin(), dex2oat_args.end());
// We must set --android-root.
const char* android_root = getenv("ANDROID_ROOT");
CHECK(android_root != nullptr);
argv.push_back("--android-root=" + std::string(android_root));
// We need dex2oat to actually log things.
auto post_fork_fn = []() { return setenv("ANDROID_LOG_TAGS", "*:d", 1) == 0; };
ForkAndExecResult res = ForkAndExec(argv, post_fork_fn, output);
if (res.stage != ForkAndExecResult::kFinished) {
return ErrnoErrorf("Failed to finish dex2oat invocation '{}'",
android::base::Join(argv, ' '));
}
if (!WIFEXITED(res.status_code)) {
return Errorf("dex2oat didn't terminate normally (status_code={:#x}): {}",
res.status_code,
android::base::Join(argv, ' '));
}
return WEXITSTATUS(res.status_code);
}
};
// Run dex2oat with --force-palette-compilation-hooks to force calls to
// PaletteNotify{Start,End}Dex2oatCompilation.
TEST_F(Dex2oatCtsTest, CompilationHooks) {
const std::string dex_location = GetTestDexFileName("Main");
const std::string oat_location = GetScratchDir() + "/base.oat";
const std::string vdex_location = GetScratchDir() + "/base.vdex";
std::vector<std::string> args;
args.emplace_back("--dex-file=" + dex_location);
std::unique_ptr<File> oat_file(OS::CreateEmptyFile(oat_location.c_str()));
ASSERT_NE(oat_file, nullptr) << oat_location;
args.emplace_back("--oat-fd=" + std::to_string(oat_file->Fd()));
args.emplace_back("--oat-location=" + oat_location);
std::unique_ptr<File> vdex_file(OS::CreateEmptyFile(vdex_location.c_str()));
ASSERT_NE(vdex_file, nullptr) << vdex_location;
args.emplace_back("--output-vdex-fd=" + std::to_string(vdex_file->Fd()));
args.emplace_back("--force-palette-compilation-hooks");
std::string output = "";
EXPECT_THAT(Dex2Oat(args, &output), HasValue(0));
EXPECT_EQ(oat_file->FlushCloseOrErase(), 0);
EXPECT_EQ(vdex_file->FlushCloseOrErase(), 0);
}
} // namespace art
|