sdk/lib/fdio/tests/fdio_unsafe.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 <fcntl.h>
#include <fidl/fuchsia.io/cpp/wire.h>
#include <lib/fdio/directory.h>
#include <lib/fdio/namespace.h>
#include <lib/fdio/unsafe.h>
#include <unistd.h>

#include <fbl/unique_fd.h>
#include <zxtest/zxtest.h>

namespace {

TEST(UnsafeTest, BorrowChannel) {
  fbl::unique_fd fd(open("/pkg", O_DIRECTORY | O_RDONLY));
  ASSERT_LE(0, fd.get());

  fdio_t* io = fdio_unsafe_fd_to_io(fd.get());
  ASSERT_TRUE(io);

  auto dir = fidl::UnownedClientEnd<fuchsia_io::Node>(fdio_unsafe_borrow_channel(io));
  ASSERT_TRUE(dir.is_valid());

  auto endpoints = fidl::CreateEndpoints<fuchsia_io::Node>();
  ASSERT_OK(endpoints.status_value());

  auto result = fidl::WireCall(dir)->Clone(fuchsia_io::wire::OpenFlags::kCloneSameRights,
                                           std::move(endpoints->server));
  ASSERT_OK(result.status());

  fdio_unsafe_release(io);
  fd.reset();
}

TEST(UnsafeTest, BorrowChannelFromUnsupportedObject) {
  // Local namespaces do not have a backing channel, so
  // |fdio_unsafe_borrow_channel| and |fdio_get_service_handle| should fail.

  fdio_ns_t* ns;
  ASSERT_OK(fdio_ns_create(&ns));
  fbl::unique_fd fd(open("/pkg", O_RDONLY | O_DIRECTORY));
  ASSERT_LE(0, fd.get());
  ASSERT_OK(fdio_ns_bind_fd(ns, "/test-ns-item", fd.get()));
  ASSERT_EQ(0, close(fd.release()));

  fbl::unique_fd ns_fd(fdio_ns_opendir(ns));
  ASSERT_LE(0, ns_fd.get());
  fdio_t* io = fdio_unsafe_fd_to_io(ns_fd.get());
  ASSERT_TRUE(io);

  EXPECT_EQ(ZX_HANDLE_INVALID, fdio_unsafe_borrow_channel(io));
  fdio_unsafe_release(io);

  zx_handle_t handle = ZX_HANDLE_INVALID;
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wdeprecated-declarations"
  EXPECT_STATUS(ZX_ERR_NOT_SUPPORTED, fdio_get_service_handle(ns_fd.release(), &handle));
#pragma clang diagnostic pop
  EXPECT_EQ(ZX_HANDLE_INVALID, handle);

  ASSERT_OK(fdio_ns_destroy(ns));
}

}  // namespace