// 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 <fcntl.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <unistd.h>

#include <lib/fdio/limits.h>
#include <lib/fdio/util.h>
#include <unittest/unittest.h>
#include <zircon/syscalls.h>

#include "filesystems.h"

bool test_basic(void) {
    BEGIN_TEST;

    ASSERT_EQ(mkdir("::alpha", 0755), 0, "");
    ASSERT_EQ(mkdir("::alpha/bravo", 0755), 0, "");
    ASSERT_EQ(mkdir("::alpha/bravo/charlie", 0755), 0, "");
    ASSERT_EQ(mkdir("::alpha/bravo/charlie/delta", 0755), 0, "");
    ASSERT_EQ(mkdir("::alpha/bravo/charlie/delta/echo", 0755), 0, "");
    int fd1 = open("::alpha/bravo/charlie/delta/echo/foxtrot", O_RDWR | O_CREAT, 0644);
    ASSERT_GT(fd1, 0, "");
    int fd2 = open("::alpha/bravo/charlie/delta/echo/foxtrot", O_RDWR, 0644);
    ASSERT_GT(fd2, 0, "");
    ASSERT_EQ(write(fd1, "Hello, World!\n", 14), 14, "");
    ASSERT_EQ(close(fd1), 0, "");
    ASSERT_EQ(close(fd2), 0, "");

    // test pipelined opens
    // the open itself will always succeed if the remote side exists,
    // but we'll get an error when we try to do an operation on the file
    fd1 = open("::alpha/bravo/charlie/delta/echo/foxtrot", O_RDONLY | O_PIPELINE, 0644);
    ASSERT_GT(fd1, 0, "");
    char tmp[14];
    ASSERT_EQ(read(fd1, tmp, 14), 14, "");
    ASSERT_EQ(close(fd1), 0, "");
    ASSERT_EQ(memcmp(tmp, "Hello, World!\n", 14), 0, "");

    fd1 = open("::alpha/banana", O_RDONLY | O_PIPELINE, 0644);
    ASSERT_GT(fd1, 0, "");
    ASSERT_EQ(read(fd1, tmp, 14), -1, "");
    ASSERT_EQ(close(fd1), -1, "");

    fd1 = open("::file.txt", O_CREAT | O_RDWR, 0644);
    ASSERT_GT(fd1, 0, "");
    ASSERT_EQ(close(fd1), 0, "");

    ASSERT_EQ(unlink("::file.txt"), 0, "");
    ASSERT_EQ(mkdir("::emptydir", 0755), 0, "");
    fd1 = open("::emptydir", O_RDONLY, 0644);
    ASSERT_GT(fd1, 0, "");

    // Zero-sized reads should always succeed
    ASSERT_EQ(read(fd1, NULL, 0), 0, "");
    // But nonzero reads to directories should always fail
    char buf;
    ASSERT_EQ(read(fd1, &buf, 1), -1, "");
    ASSERT_EQ(write(fd1, "Don't write to directories", 26), -1, "");
    ASSERT_EQ(ftruncate(fd1, 0), -1, "");
    ASSERT_EQ(rmdir("::emptydir"), 0, "");
    ASSERT_EQ(rmdir("::emptydir"), -1, "");
    ASSERT_EQ(close(fd1), 0, "");
    ASSERT_EQ(rmdir("::emptydir"), -1, "");

    END_TEST;
}

bool test_unclean_close(void) {
    BEGIN_TEST;

    int fd = open("::foobar", O_CREAT | O_RDWR);
    ASSERT_GT(fd, 0, "");

    // Try closing a connection to a file with an "unclean" shutdown,
    // noticed by the filesystem server as a closed handle rather than
    // an explicit "Close" call.
    zx_handle_t handles[FDIO_MAX_HANDLES];
    uint32_t types[FDIO_MAX_HANDLES];
    zx_status_t r = fdio_transfer_fd(fd, 0, handles, types);
    ASSERT_GE(fd, 0, "");
    for (size_t i = 0; i < (size_t) r; i++) {
        ASSERT_EQ(zx_handle_close(handles[i]), ZX_OK, "");
    }

    ASSERT_EQ(unlink("::foobar"), 0, "");

    END_TEST;
}

RUN_FOR_ALL_FILESYSTEMS(basic_tests,
    RUN_TEST_MEDIUM(test_basic)
    RUN_TEST_MEDIUM(test_unclean_close)
)