zircon/system/utest/fs/test-fcntl.cc

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

#include <fbl/unique_fd.h>
#include <zircon/syscalls.h>
#include <unittest/unittest.h>

#include "filesystems.h"

bool TestFcntlAppend(void) {
  BEGIN_TEST;

  fbl::unique_fd fd(open("::file", O_APPEND | O_RDWR | O_CREAT, 0644));
  ASSERT_TRUE(fd);

  // Do a quick check that O_APPEND is appending
  char buf[5];
  memset(buf, 'a', sizeof(buf));
  ASSERT_EQ(lseek(fd.get(), 0, SEEK_SET), 0);
  ASSERT_EQ(write(fd.get(), buf, sizeof(buf)), sizeof(buf));
  ASSERT_EQ(lseek(fd.get(), 0, SEEK_SET), 0);
  ASSERT_EQ(write(fd.get(), buf, sizeof(buf)), sizeof(buf));
  struct stat sb;
  ASSERT_EQ(fstat(fd.get(), &sb), 0);
  ASSERT_EQ(sb.st_size, sizeof(buf) * 2);

  // Use F_GETFL; observe O_APPEND
  int flags = fcntl(fd.get(), F_GETFL);
  ASSERT_GT(flags, -1, "Fcntl failed");
  ASSERT_EQ(flags & O_ACCMODE, O_RDWR, "Access mode flags did not match");
  ASSERT_EQ(flags & ~O_ACCMODE, O_APPEND, "Status flags did not match");

  // Use F_SETFL; turn off O_APPEND
  ASSERT_EQ(fcntl(fd.get(), F_SETFL, flags & ~O_APPEND), 0, "Fcntl failed");

  // Use F_GETFL; observe O_APPEND has been turned off
  flags = fcntl(fd.get(), F_GETFL);
  ASSERT_GT(flags, -1, "Fcntl failed");
  ASSERT_EQ(flags & O_ACCMODE, O_RDWR, "Access mode flags did not match");
  ASSERT_EQ(flags & ~O_ACCMODE, 0, "Status flags did not match");

  // Write to the file, verify it is no longer appending.
  ASSERT_EQ(lseek(fd.get(), 0, SEEK_SET), 0);
  ASSERT_EQ(write(fd.get(), buf, sizeof(buf)), sizeof(buf));
  ASSERT_EQ(fstat(fd.get(), &sb), 0);
  ASSERT_EQ(sb.st_size, sizeof(buf) * 2);

  // Clean up
  ASSERT_EQ(close(fd.release()), 0);
  ASSERT_EQ(unlink("::file"), 0);
  END_TEST;
}

bool TestFcntlAccessBits(void) {
  BEGIN_TEST;

  fbl::unique_fd fd(open("::file", O_APPEND | O_RDWR | O_CREAT, 0644));
  ASSERT_TRUE(fd);

  // Do a quick check that we can write
  char buf[5];
  memset(buf, 'a', sizeof(buf));
  ASSERT_EQ(lseek(fd.get(), 0, SEEK_SET), 0);
  ASSERT_EQ(write(fd.get(), buf, sizeof(buf)), sizeof(buf));
  struct stat sb;
  ASSERT_EQ(fstat(fd.get(), &sb), 0);
  ASSERT_EQ(sb.st_size, sizeof(buf));

  // Use F_GETFL; observe O_APPEND
  int flags = fcntl(fd.get(), F_GETFL);
  ASSERT_GT(flags, -1, "Fcntl failed");
  ASSERT_EQ(flags & O_ACCMODE, O_RDWR, "Access mode flags did not match");
  ASSERT_EQ(flags & ~O_ACCMODE, O_APPEND, "Status flags did not match");

  // Use F_SETFL; try to turn off everything except O_APPEND
  // (if fcntl paid attention to access bits, this would make the file
  // read-only).
  ASSERT_EQ(fcntl(fd.get(), F_SETFL, O_APPEND), 0, "Fcntl failed");

  // We're still appending -- AND writable, because the access bits haven't
  // changed.
  ASSERT_EQ(lseek(fd.get(), 0, SEEK_SET), 0);
  ASSERT_EQ(write(fd.get(), buf, sizeof(buf)), sizeof(buf));
  ASSERT_EQ(fstat(fd.get(), &sb), 0);
  ASSERT_EQ(sb.st_size, sizeof(buf) * 2);

  // Clean up
  ASSERT_EQ(close(fd.release()), 0);
  ASSERT_EQ(unlink("::file"), 0);
  END_TEST;
}

RUN_FOR_ALL_FILESYSTEMS(fcntl_tests,
                        RUN_TEST_MEDIUM(TestFcntlAppend) RUN_TEST_MEDIUM(TestFcntlAccessBits))