zircon/system/ulib/runtests-utils/service-proxy-dir.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 <fuchsia/io/llcpp/fidl.h>
#include <lib/async/default.h>
#include <lib/fdio/directory.h>
#include <lib/fdio/fd.h>
#include <lib/fdio/fdio.h>
#include <lib/fdio/vfs.h>
#include <runtests-utils/service-proxy-dir.h>
#include <zircon/status.h>

#include <string>

namespace fio = ::llcpp::fuchsia::io;

namespace runtests {

ServiceProxyDir::ServiceProxyDir(zx::channel proxy_dir) : proxy_dir_(std::move(proxy_dir)) {}

void ServiceProxyDir::AddEntry(std::string name, fbl::RefPtr<fs::Vnode> node) {
  entries_[name] = node;
}

zx_status_t ServiceProxyDir::Getattr(vnattr_t* attr) {
  memset(attr, 0, sizeof(vnattr_t));
  attr->mode = V_TYPE_DIR | V_IRUSR;
  attr->inode = fuchsia_io_INO_UNKNOWN;
  attr->nlink = 1;
  return ZX_OK;
}

zx_status_t ServiceProxyDir::GetNodeInfo(uint32_t flags, fuchsia_io_NodeInfo* info) {
  info->tag = fuchsia_io_NodeInfoTag_directory;
  return ZX_OK;
}

bool ServiceProxyDir::IsDirectory() const { return true; }

zx_status_t ServiceProxyDir::Lookup(fbl::RefPtr<fs::Vnode>* out, fbl::StringPiece name) {
  std::unique_lock lock(lock_);
  auto entry_name = std::string(name.data(), name.length());

  auto entry = entries_.find(entry_name);
  if (entry != entries_.end()) {
    *out = entry->second;
    return ZX_OK;
  }

  entries_.emplace(
      entry_name, *out = fbl::MakeRefCounted<fs::Service>([this, entry_name](zx::channel request) {
        return fio::Directory::Call::Open(zx::unowned_channel(proxy_dir_),
                                          fio::OPEN_RIGHT_READABLE | fio::OPEN_RIGHT_WRITABLE, 0755,
                                          fidl::StringView(entry_name.length(), entry_name.data()),
                                          std::move(request))
            .status();
      }));

  return ZX_OK;
}

}  // namespace runtests