// Copyright 2018 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 <unistd.h>

#include <fbl/unique_fd.h>
#include <fuchsia/io/c/fidl.h>
#include <lib/fdio/limits.h>
#include <lib/fdio/util.h>
#include <unittest/unittest.h>
#include <zircon/processargs.h>
#include <zircon/syscalls.h>

namespace {

bool test_device_clone() {
    BEGIN_TEST;

    fbl::unique_fd fd(open("/dev/zero", O_RDONLY));

    zx_handle_t handles[FDIO_MAX_HANDLES] = {};
    uint32_t types[FDIO_MAX_HANDLES] = {};
    zx_status_t r = fdio_clone_fd(fd.get(), 0, handles, types);
    ASSERT_EQ(r, 1);
    ASSERT_NE(handles[0], ZX_HANDLE_INVALID);
    ASSERT_EQ(types[0], PA_FDIO_REMOTE);
    zx_handle_close(handles[0]);

    END_TEST;
}

bool test_device_transfer() {
    BEGIN_TEST;

    fbl::unique_fd fd(open("/dev/zero", O_RDONLY));

    zx_handle_t handles[FDIO_MAX_HANDLES] = {};
    uint32_t types[FDIO_MAX_HANDLES] = {};
    zx_status_t r = fdio_transfer_fd(fd.release(), 0, handles, types);
    ASSERT_EQ(r, 1);
    ASSERT_NE(handles[0], ZX_HANDLE_INVALID);
    ASSERT_EQ(types[0], PA_FDIO_REMOTE);
    zx_handle_close(handles[0]);

    END_TEST;
}

} // namespace

BEGIN_TEST_CASE(fdio_tests)
RUN_TEST(test_device_clone)
RUN_TEST(test_device_transfer)
END_TEST_CASE(fdio_tests)