// Copyright 2016 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 <assert.h>
#include <errno.h>
#include <fcntl.h>
#include <lib/fdio/directory.h>
#include <lib/fdio/fd.h>
#include <lib/fdio/fdio.h>
#include <lib/fdio/io.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <threads.h>
#include <unistd.h>
#include <zircon/processargs.h>
#include <zircon/syscalls.h>

#include <zxtest/zxtest.h>

TEST(HandleFDTest, Close) {
  zx_handle_t h = ZX_HANDLE_INVALID;
  ASSERT_OK(zx_event_create(0u, &h), "zx_event_create() failed");
  ASSERT_NE(h, ZX_HANDLE_INVALID, "");

  // fdio_handle_fd() with shared_handle = true
  int fd = fdio_handle_fd(h, ZX_USER_SIGNAL_0, ZX_USER_SIGNAL_1, true);
  ASSERT_GT(fd, 0, "fdio_handle_fd() failed");

  close(fd);

  // close(fd) has not closed the wrapped handle
  EXPECT_OK(zx_object_signal(h, 0, ZX_USER_SIGNAL_0), "zx_object_signal() should succeed");

  // fdio_handle_fd() with shared_handle = false
  fd = fdio_handle_fd(h, ZX_USER_SIGNAL_0, ZX_USER_SIGNAL_1, false);
  ASSERT_GT(fd, 0, "fdio_handle_fd() failed");

  close(fd);

  // close(fd) has closed the wrapped handle
  EXPECT_EQ(ZX_ERR_BAD_HANDLE, zx_object_signal(h, 0, ZX_USER_SIGNAL_0),
            "zx_object_signal() should fail");
}

TEST(HandleFDTest, Pipe) {
  int fds[2];
  int status = pipe(fds);
  ASSERT_EQ(status, 0, "pipe() failed");

  struct stat st;
  ASSERT_EQ(fstat(fds[0], &st), 0, "fstat() on pipe failed");
  ASSERT_EQ(st.st_mode & S_IFMT, (mode_t)S_IFIFO, "Unexpected mode");
  ASSERT_EQ(fstat(fds[1], &st), 0, "fstat() on pipe failed");
  ASSERT_EQ(st.st_mode & S_IFMT, (mode_t)S_IFIFO, "Unexpected mode");

  status = fcntl(fds[0], F_GETFL);
  ASSERT_EQ(status, 0, "fcntl(F_GETFL) failed");

  status |= O_NONBLOCK;
  status = fcntl(fds[0], F_SETFL, status);
  ASSERT_EQ(status, 0, "fcntl(FSETFL, O_NONBLOCK) failed");

  status = fcntl(fds[0], F_GETFL);
  ASSERT_EQ(status, O_NONBLOCK, "fcntl(F_GETFL) failed");

  int message[2] = {-6, 1};
  ssize_t written = write(fds[1], message, sizeof(message));
  ASSERT_GE(written, 0, "write() failed");
  ASSERT_EQ((uint32_t)written, sizeof(message), "write() should have written the whole message.");

  int available = 0;
  status = ioctl(fds[0], FIONREAD, &available);
  ASSERT_GE(status, 0, "ioctl(FIONREAD) failed");
  EXPECT_EQ((uint32_t)available, sizeof(message), "ioctl(FIONREAD) queried wrong number of bytes");

  int read_message[2];
  ssize_t bytes_read = read(fds[0], read_message, sizeof(read_message));
  ASSERT_GE(bytes_read, 0, "read() failed");
  ASSERT_EQ((uint32_t)bytes_read, sizeof(read_message), "read() read wrong number of bytes");

  EXPECT_EQ(read_message[0], message[0], "read() read wrong value");
  EXPECT_EQ(read_message[1], message[1], "read() read wrong value");
}

int write_thread(void* arg) {
  // Sleep to try to ensure the write happens after the poll.
  zx_nanosleep(ZX_MSEC(5));
  int message[2] = {-6, 1};
  ssize_t written = write(*(int*)arg, message, sizeof(message));
  EXPECT_GE(written, 0, "write() failed");
  EXPECT_EQ((uint32_t)written, sizeof(message), "write() should have written the whole message.");
  return 0;
}

void ppoll_test_handler(struct timespec* timeout) {
  int fds[2];
  int status = pipe(fds);
  ASSERT_EQ(status, 0, "pipe() failed");

  thrd_t t;
  int thrd_create_result = thrd_create(&t, write_thread, &fds[1]);
  ASSERT_EQ(thrd_create_result, thrd_success, "create blocking send thread");

  struct pollfd poll_fds[1] = {{fds[0], POLLIN, 0}};
  int ppoll_result = ppoll(poll_fds, 1, timeout, NULL);

  EXPECT_EQ(1, ppoll_result, "didn't read anything");

  ASSERT_EQ(thrd_join(t, NULL), thrd_success, "join blocking send thread");
}

TEST(HandleFDTest, PPollNegative) {
  struct timespec timeout_ts = {-1, -1};
  ppoll_test_handler(&timeout_ts);
}

TEST(HandleFDTest, PPollNull) { ppoll_test_handler(NULL); }

TEST(HandleFDTest, Overflow) {
  unsigned int nanoseconds_in_seconds = 1000000000;
  struct timespec timeout_ts = {UINT64_MAX / nanoseconds_in_seconds,
                                UINT64_MAX % nanoseconds_in_seconds};
  ppoll_test_handler(&timeout_ts);
}

TEST(HandleFDTest, PPollImmediateTimeout) {
  int fds[2];
  int status = pipe(fds);
  ASSERT_EQ(status, 0, "pipe() failed");

  struct timespec timeout = {0, 0};
  struct pollfd poll_fds[1] = {{fds[0], POLLIN, 0}};
  int ppoll_result = ppoll(poll_fds, 1, &timeout, NULL);

  EXPECT_EQ(0, ppoll_result, "no fds should be readable");
}

TEST(HandleFDTest, TransferFD) {
  int fds[2];
  int status = pipe(fds);
  ASSERT_EQ(status, 0, "pipe() failed");

  // Make pipe nonblocking, write message
  status |= O_NONBLOCK;
  status = fcntl(fds[0], F_SETFL, status);
  ASSERT_EQ(status, 0, "fcntl(FSETFL, O_NONBLOCK) failed");
  int message[2] = {-6, 1};
  ssize_t written = write(fds[1], message, sizeof(message));
  ASSERT_GE(written, 0, "write() failed");
  ASSERT_EQ((uint32_t)written, sizeof(message), "write() should have written the whole message.");

  // fd --> handle
  zx_handle_t handle = ZX_HANDLE_INVALID;
  status = fdio_fd_transfer(fds[0], &handle);
  ASSERT_OK(status, "failed to transfer fds to handle");

  // handle --> fd
  ASSERT_EQ(fdio_fd_create(handle, &fds[0]), ZX_OK, "failed to transfer handles to fds");

  // Read message
  int read_message[2];
  ssize_t bytes_read = read(fds[0], read_message, sizeof(read_message));
  ASSERT_GE(bytes_read, 0, "read() failed");
  ASSERT_EQ((uint32_t)bytes_read, sizeof(read_message), "read() read wrong number of bytes");

  EXPECT_EQ(read_message[0], message[0], "read() read wrong value");
  EXPECT_EQ(read_message[1], message[1], "read() read wrong value");
}

TEST(HandleFDTest, TransferDevice) {
  int fd = open("/dev/zero", O_RDONLY);
  ASSERT_GE(fd, 0, "Failed to open /dev/zero");

  // fd --> handle
  zx_handle_t handle = ZX_HANDLE_INVALID;
  zx_status_t status = fdio_fd_transfer(fd, &handle);
  ASSERT_OK(status, "failed to transfer fds to handles");

  // handle --> fd
  ASSERT_EQ(fdio_fd_create(handle, &fd), ZX_OK, "failed to transfer handles to fds");

  ASSERT_EQ(close(fd), 0, "Failed to close fd");
}

TEST(HandleFDTest, CreateFDFromConnectedSocket) {
  int fd;
  zx_handle_t h1, h2;
  ASSERT_OK(zx_socket_create(ZX_SOCKET_STREAM, &h1, &h2), "failed to create socket pair");
  ASSERT_OK(fdio_fd_create(h1, &fd), "failed to create FD for socket handle");

  int message[2] = {0xab, 0x1234};
  size_t written;
  ASSERT_OK(zx_socket_write(h2, 0, message, sizeof(message), &written),
            "failed to write to socket handle");
  ASSERT_EQ(sizeof(message), written, "failed to write full message to socket handle");

  int read_message[2] = {};
  ssize_t bytes_read = read(fd, read_message, sizeof(read_message));
  ASSERT_EQ(sizeof(message), (uint32_t)bytes_read, "failed to read from socket fd");
  ASSERT_EQ(0, memcmp(message, read_message, sizeof(message)),
            "incorrect bytes read from socket fd");

  // Set O_NONBLOCK
  int flags = fcntl(fd, F_GETFL);
  ASSERT_EQ(flags, 0, "fcntl(F_GETFL) failed");
  flags |= O_NONBLOCK;
  int ret = fcntl(fd, F_SETFL, flags);
  ASSERT_EQ(ret, 0, "fcntl(FSETFL, O_NONBLOCK) failed");
  ASSERT_EQ(-1, read(fd, read_message, sizeof(read_message)),
            "failed to read from socket with O_NONBLOCK");
  ASSERT_EQ(EAGAIN, errno, "errno incorrect");
}

TEST(HandleFDTest, BindToFDInvalid) {
  fdio_t* fdio = fdio_null_create();
  EXPECT_NOT_NULL(fdio, "");

  // When binding and not providing a specific |fd|, the
  // |starting_fd| must be nonnegative.
  int fd = fdio_bind_to_fd(fdio, -1, -1);
  int err = errno;
  EXPECT_LT(fd, 0, "");
  EXPECT_EQ(err, EINVAL, "");

  // Starting with a huge |starting_fd| will fail since the table
  // does not hold so many.
  fd = fdio_bind_to_fd(fdio, -1, INT_MAX);
  err = errno;
  EXPECT_LT(fd, 0, "");
  EXPECT_EQ(err, EMFILE, "");

  // Do a successful one just to clean up the fdio and not leak it.
  fd = fdio_bind_to_fd(fdio, -1, 0);
  EXPECT_GE(fd, 0, "");
  EXPECT_EQ(0, close(fd), "");
}