src/sys/early_boot_instrumentation/early_boot_instrumentation_test.cc

// 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)));
}