| // Copyright 2022 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 <fcntl.h> |
| #include <lib/stdcompat/array.h> |
| |
| #include <array> |
| |
| #include <fbl/unique_fd.h> |
| #include <gtest/gtest.h> |
| |
| namespace { |
| template <size_t N> |
| bool CheckContents(const char (&str)[N], fbl::unique_fd file) { |
| auto expected = cpp20::to_array(str); |
| decltype(expected) actual = {}; |
| size_t acc_offset = 0; |
| while (auto read_bytes = |
| read(file.get(), actual.data() + acc_offset, actual.size() - acc_offset)) { |
| acc_offset += read_bytes; |
| } |
| return acc_offset == expected.size() && |
| memcmp(actual.data(), expected.data(), actual.size()) == 0; |
| } |
| |
| } // namespace |
| |
| TEST(EarlyBootInstrumentationTest, HasKernelInDynamic) { |
| fbl::unique_fd kernel_file(open("/debugdata/llvm-profile/dynamic/zircon.profraw", O_RDONLY)); |
| ASSERT_TRUE(kernel_file); |
| ASSERT_TRUE(CheckContents("kernel", std::move(kernel_file))); |
| } |
| |
| TEST(EarlyBootInstrumentationTest, HasPhysbootInStatic) { |
| fbl::unique_fd physboot_file(open("/debugdata/llvm-profile/static/physboot.profraw", O_RDONLY)); |
| ASSERT_TRUE(physboot_file); |
| ASSERT_TRUE(CheckContents("physboot", std::move(physboot_file))); |
| } |
| |
| TEST(EarlyBootInstrumentationTest, HasSvcStashDataWithLlvmProfile) { |
| fbl::unique_fd static_file(open("/debugdata/llvm-profile/static/0-0.profraw", O_RDONLY)); |
| ASSERT_TRUE(static_file); |
| ASSERT_TRUE(CheckContents("1234", std::move(static_file))); |
| |
| fbl::unique_fd dynamic_file(open("/debugdata/llvm-profile/dynamic/0-1.profraw", O_RDONLY)); |
| ASSERT_TRUE(dynamic_file); |
| ASSERT_TRUE(CheckContents("567890123", std::move(dynamic_file))); |
| } |
| |
| TEST(EarlyBootInstrumentationTest, HasSvcStashDataWithCustomSink) { |
| fbl::unique_fd static_file(open("/debugdata/my-custom-sink/static/0-2.custom", O_RDONLY)); |
| ASSERT_TRUE(static_file); |
| ASSERT_TRUE(CheckContents("789", std::move(static_file))); |
| |
| fbl::unique_fd dynamic_file(open("/debugdata/my-custom-sink/dynamic/0-3", O_RDONLY)); |
| ASSERT_TRUE(dynamic_file); |
| ASSERT_TRUE(CheckContents("43218765", std::move(dynamic_file))); |
| } |