zircon/system/utest/fs/test-basic.c

// 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 <lib/fdio/directory.h>
#include <lib/fdio/fd.h>
#include <lib/fdio/fdio.h>
#include <lib/fdio/limits.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <unistd.h>
#include <zircon/syscalls.h>

#include <unittest/unittest.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, "");

  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 handle = ZX_HANDLE_INVALID;
  fdio_fd_transfer(fd, &handle);
  // TODO: Should we check the status returned by fdio_fd_transfer?
  ASSERT_GE(fd, 0, "");
  if (handle != ZX_HANDLE_INVALID) {
    ASSERT_EQ(zx_handle_close(handle), 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))