139 lines
4.9 KiB
C++
139 lines
4.9 KiB
C++
/*
|
|
* Copyright (C) 2014 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/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "android-base/stringprintf.h"
|
|
|
|
#include "arch/instruction_set.h"
|
|
#include "base/os.h"
|
|
#include "base/utils.h"
|
|
#include "common_runtime_test.h"
|
|
#include "exec_utils.h"
|
|
#include "gc/heap.h"
|
|
#include "gc/space/image_space.h"
|
|
#include "runtime.h"
|
|
|
|
namespace art {
|
|
|
|
static const char* kImgDiagBinaryName = "imgdiag";
|
|
|
|
// from kernel <include/linux/threads.h>
|
|
#define PID_MAX_LIMIT (4*1024*1024) // Upper bound. Most kernel configs will have smaller max pid.
|
|
|
|
static const pid_t kImgDiagGuaranteedBadPid = (PID_MAX_LIMIT + 1);
|
|
|
|
class ImgDiagTest : public CommonRuntimeTest {
|
|
protected:
|
|
void SetUp() override {
|
|
CommonRuntimeTest::SetUp();
|
|
|
|
// We loaded the runtime with an explicit image. Therefore the image space must exist.
|
|
std::vector<gc::space::ImageSpace*> image_spaces =
|
|
Runtime::Current()->GetHeap()->GetBootImageSpaces();
|
|
ASSERT_TRUE(!image_spaces.empty());
|
|
boot_image_location_ = image_spaces[0]->GetImageLocation();
|
|
}
|
|
|
|
void SetUpRuntimeOptions(RuntimeOptions* options) override {
|
|
// Needs to live until CommonRuntimeTest::SetUp finishes, since we pass it a cstring.
|
|
runtime_args_image_ = android::base::StringPrintf("-Ximage:%s", GetCoreArtLocation().c_str());
|
|
options->push_back(std::make_pair(runtime_args_image_, nullptr));
|
|
}
|
|
|
|
// Path to the imgdiag(d?)[32|64] binary.
|
|
std::string GetImgDiagFilePath() {
|
|
std::string path = GetArtBinDir() + '/' + kImgDiagBinaryName;
|
|
if (kIsDebugBuild) {
|
|
path += 'd';
|
|
}
|
|
std::string path32 = path + "32";
|
|
// If we have both a 32-bit and a 64-bit build, the 32-bit file will have a 32 suffix.
|
|
if (OS::FileExists(path32.c_str()) && !Is64BitInstructionSet(kRuntimeISA)) {
|
|
return path32;
|
|
// Only a single build exists, so the filename never has an extra suffix.
|
|
} else {
|
|
return path;
|
|
}
|
|
}
|
|
|
|
// Run imgdiag with a custom boot image location.
|
|
bool Exec(pid_t image_diff_pid, const std::string& boot_image, std::string* error_msg) {
|
|
// Invoke 'img_diag' against the current process.
|
|
// This should succeed because we have a runtime and so it should
|
|
// be able to map in the boot.art and do a diff for it.
|
|
std::string file_path = GetImgDiagFilePath();
|
|
EXPECT_TRUE(OS::FileExists(file_path.c_str())) << file_path << " should be a valid file path";
|
|
|
|
// Run imgdiag --image-diff-pid=$image_diff_pid and wait until it's done with a 0 exit code.
|
|
std::vector<std::string> exec_argv = {
|
|
file_path,
|
|
"--image-diff-pid=" + std::to_string(image_diff_pid),
|
|
"--zygote-diff-pid=" + std::to_string(image_diff_pid),
|
|
"--runtime-arg",
|
|
GetClassPathOption("-Xbootclasspath:", GetLibCoreDexFileNames()),
|
|
"--runtime-arg",
|
|
GetClassPathOption("-Xbootclasspath-locations:", GetLibCoreDexLocations()),
|
|
"--boot-image=" + boot_image
|
|
};
|
|
|
|
return ::art::Exec(exec_argv, error_msg);
|
|
}
|
|
|
|
// Run imgdiag with the default boot image location.
|
|
bool ExecDefaultBootImage(pid_t image_diff_pid, std::string* error_msg) {
|
|
return Exec(image_diff_pid, boot_image_location_, error_msg);
|
|
}
|
|
|
|
private:
|
|
std::string runtime_args_image_;
|
|
std::string boot_image_location_;
|
|
};
|
|
|
|
#if defined (ART_TARGET)
|
|
TEST_F(ImgDiagTest, ImageDiffPidSelf) {
|
|
#else
|
|
// Can't run this test on the host, it will fail when trying to open /proc/kpagestats
|
|
// because it's root read-only.
|
|
TEST_F(ImgDiagTest, DISABLED_ImageDiffPidSelf) {
|
|
#endif
|
|
// Invoke 'img_diag' against the current process.
|
|
// This should succeed because we have a runtime and so it should
|
|
// be able to map in the boot.art and do a diff for it.
|
|
|
|
// Run imgdiag --image-diff-pid=$(self pid) and wait until it's done with a 0 exit code.
|
|
std::string error_msg;
|
|
ASSERT_TRUE(ExecDefaultBootImage(getpid(), &error_msg)) << "Failed to execute -- because: "
|
|
<< error_msg;
|
|
}
|
|
|
|
TEST_F(ImgDiagTest, ImageDiffBadPid) {
|
|
// Invoke 'img_diag' against a non-existing process. This should fail.
|
|
|
|
// Run imgdiag --image-diff-pid=some_bad_pid and wait until it's done with a 0 exit code.
|
|
std::string error_msg;
|
|
ASSERT_FALSE(ExecDefaultBootImage(kImgDiagGuaranteedBadPid,
|
|
&error_msg)) << "Incorrectly executed";
|
|
UNUSED(error_msg);
|
|
}
|
|
|
|
} // namespace art
|