| // Copyright 2018 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "src/developer/debug/debug_agent/system_interface.h" |
| |
| #include <gtest/gtest.h> |
| |
| #include "src/developer/debug/debug_agent/zircon_job_handle.h" |
| #include "src/developer/debug/debug_agent/zircon_system_interface.h" |
| #include "src/developer/debug/debug_agent/zircon_utils.h" |
| |
| namespace debug_agent { |
| |
| namespace { |
| |
| // Recursively walks the process tree and returns true if there is a process |
| // with the given name and koid. |
| bool FindProcess(const debug_ipc::ProcessTreeRecord& record, const std::string& name_to_find, |
| zx_koid_t koid_to_find) { |
| if (record.name == name_to_find && record.koid == koid_to_find) |
| return true; |
| for (const auto& child : record.children) { |
| if (FindProcess(child, name_to_find, koid_to_find)) |
| return true; |
| } |
| return false; |
| } |
| |
| } // namespace |
| |
| TEST(SystemInfo, GetProcessTree) { |
| ZirconSystemInterface system_interface; |
| |
| debug_ipc::ProcessTreeRecord root = system_interface.GetProcessTree(); |
| |
| // The root node should be a job with some children. |
| EXPECT_EQ(debug_ipc::ProcessTreeRecord::Type::kJob, root.type); |
| EXPECT_FALSE(root.children.empty()); |
| |
| // Compute our own process name and koid. |
| auto self = zx::process::self(); |
| std::string self_name = zircon::NameForObject(*self); |
| EXPECT_FALSE(self_name.empty()); |
| zx_koid_t self_koid = zircon::KoidForObject(*self); |
| ASSERT_NE(0u, self_koid); |
| |
| // Our name and koid should be somewhere in the tree. |
| EXPECT_TRUE(FindProcess(root, self_name, self_koid)); |
| } |
| |
| } // namespace debug_agent |