src/sys/component_manager/tests/rights/expose_dir_rights.cc

// Copyright 2019 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 <fidl/fuchsia.io/cpp/wire.h>
#include <lib/async-loop/cpp/loop.h>
#include <lib/async-loop/default.h>
#include <lib/svc/outgoing.h>
#include <lib/zx/channel.h>

#include "src/storage/lib/vfs/cpp/remote_dir.h"
#include "src/storage/memfs/memfs.h"
#include "src/storage/memfs/vnode_dir.h"

int main(int argc, char* argv[]) {
  async::Loop loop(&kAsyncLoopConfigAttachToCurrentThread);

  zx::result result = memfs::Memfs::Create(loop.dispatcher(), "<tmp>");
  if (result.is_error()) {
    fprintf(stderr, "Failed to create memfs: %s\n", result.status_string());
    return -1;
  }
  auto& [memfs, root] = result.value();

  auto [memfs_dir, server] = fidl::Endpoints<fuchsia_io::Directory>::Create();

  if (zx_status_t status = memfs->ServeDirectory(std::move(root), std::move(server));
      status != ZX_OK) {
    fprintf(stderr, "Failed to server memfs directory: %s\n", zx_status_get_string(status));
    return -1;
  }

  svc::Outgoing outgoing(loop.dispatcher());

  using fuchsia_io::wire::OpenFlags;

  std::pair<fbl::String, OpenFlags> dirs[] = {
      {"read_only", OpenFlags::kRightReadable},
      {"read_write", OpenFlags::kRightReadable | OpenFlags::kRightWritable},
      {"read_exec", OpenFlags::kRightReadable | OpenFlags::kRightExecutable},
      {"read_only_after_scoped", OpenFlags::kRightReadable | OpenFlags::kRightWritable},
  };

  for (const auto& [name, rights] : dirs) {
    auto [client, node_server] = fidl::Endpoints<fuchsia_io::Directory>::Create();
    fidl::ServerEnd<fuchsia_io::Node> server(node_server.TakeChannel());
    if (fidl::Status result = fidl::WireCall(memfs_dir)->Clone(rights, std::move(server));
        !result.ok()) {
      fprintf(stderr, "Failed to clone memfs dir: %s\n", result.FormatDescription().c_str());
      return -1;
    }
    if (zx_status_t status = outgoing.root_dir()->AddEntry(
            name, fbl::MakeRefCounted<fs::RemoteDir>(std::move(client)));
        status != ZX_OK) {
      fprintf(stderr, "Failed to add outgoing entry: %s\n", zx_status_get_string(status));
      return -1;
    }
  }

  if (zx_status_t status = outgoing.ServeFromStartupInfo(); status != ZX_OK) {
    fprintf(stderr, "Failed to serve outgoing dir: %s\n", zx_status_get_string(status));
    return -1;
  }

  if (zx_status_t status = loop.Run(); status != ZX_OK) {
    fprintf(stderr, "Failed to run loop: %s\n", zx_status_get_string(status));
    return -1;
  }

  return 0;
}