zircon/system/ulib/fs/test_support/test_support.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 "fs/test_support/test_support.h"

#include <fcntl.h>
#include <fuchsia/device/llcpp/fidl.h>
#include <lib/fdio/cpp/caller.h>
#include <limits.h>
#include <zircon/status.h>

#include <fbl/unique_fd.h>

namespace fs {

std::string GetTopologicalPath(const std::string& path) {
  fbl::unique_fd fd(open(path.c_str(), O_RDWR));
  if (!fd) {
    printf("Could not open block device\n");
    return std::string();
  }
  fdio_cpp::FdioCaller caller(std::move(fd));
  return GetTopologicalPath(caller.borrow_channel());
}

std::string GetTopologicalPath(zx_handle_t channel) {
  zx_status_t status;
  size_t path_len;
  char disk_path[PATH_MAX];
  auto resp =
      ::llcpp::fuchsia::device::Controller::Call::GetTopologicalPath(zx::unowned_channel(channel));
  zx_status_t io_status = resp.status();
  if (io_status != ZX_OK) {
    status = io_status;
  }

  if (resp->result.is_err()) {
    status = resp->result.err();
  } else {
    status = ZX_OK;
    auto& r = resp->result.response();
    path_len = r.path.size();
    if (path_len > PATH_MAX) {
      return std::string("");
    }
    memcpy(disk_path, r.path.data(), r.path.size());
  }

  if (status != ZX_OK || path_len > sizeof(disk_path) - 1) {
    printf("Could not acquire topological path of block device: %s\n",
           zx_status_get_string(status));
    return std::string();
  }

  disk_path[path_len] = 0;
  return std::string(disk_path);
}

}  // namespace fs