zircon/kernel/lib/userabi/userboot/bootfs.h

// Copyright 2016 The Fuchsia Authors
//
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file or at
// https://opensource.org/licenses/MIT

#ifndef ZIRCON_KERNEL_LIB_USERABI_USERBOOT_BOOTFS_H_
#define ZIRCON_KERNEL_LIB_USERABI_USERBOOT_BOOTFS_H_

#include <lib/zbi-format/internal/bootfs.h>
#include <lib/zbitl/items/bootfs.h>
#include <lib/zbitl/vmo.h>
#include <lib/zx/debuglog.h>
#include <lib/zx/resource.h>
#include <lib/zx/vmar.h>
#include <lib/zx/vmo.h>
#include <zircon/types.h>

#include <span>
#include <string_view>

#include "fidl.h"

class Bootfs {
 public:
  Bootfs(zx::unowned_vmar vmar_self, zx::vmo vmo, zx::resource vmex_resource, zx::debuglog log,
         bool booting_multiple_programs)
      : booting_multiple_programs_(booting_multiple_programs),
        vmex_resource_(std::move(vmex_resource)),
        log_(std::move(log)) {
    zbitl::MapOwnedVmo mapvmo{std::move(vmo), /*writable=*/false, std::move(vmar_self)};
    if (auto result = BootfsReader::Create(std::move(mapvmo)); result.is_error()) {
      Fail(result.error_value());
    } else {
      bootfs_reader_ = std::move(result.value());
    }
  }

  zx::vmo Open(std::string_view root, std::string_view filename, std::string_view purpose);

  std::span<BootfsFileVmo> entries() { return std::span(entries_).subspan(0, entry_count_); }

 private:
  using BootfsReader = zbitl::Bootfs<zbitl::MapOwnedVmo>;
  using BootfsView = zbitl::BootfsView<zbitl::MapOwnedVmo>;

  [[noreturn]] void Fail(const BootfsReader::Error& error);

  std::array<BootfsFileVmo, ZX_CHANNEL_MAX_MSG_HANDLES> entries_;
  size_t entry_count_ = 0;
  bool booting_multiple_programs_ = false;
  BootfsReader bootfs_reader_;
  zx::resource vmex_resource_;
  zx::debuglog log_;
};

#endif  // ZIRCON_KERNEL_LIB_USERABI_USERBOOT_BOOTFS_H_