// Copyright 2018 The gVisor Authors. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. #include <errno.h> #include <fcntl.h> #include <signal.h> #include <sys/resource.h> #include <sys/stat.h> #include <syscall.h> #include <time.h> #include <unistd.h> #include "gtest/gtest.h" #include "test/syscalls/linux/file_base.h" #include "test/util/cleanup.h" #include "test/util/file_descriptor.h" #include "test/util/temp_path.h" #include "test/util/test_util.h" namespace gvisor { namespace testing { namespace { int fallocate(int fd, int mode, off_t offset, off_t len) { return syscall(__NR_fallocate, fd, mode, offset, len); } class AllocateTest : public FileTest { void SetUp() override { FileTest::SetUp(); } }; TEST_F(AllocateTest, Fallocate) { // Check that it starts at size zero. struct stat buf; ASSERT_THAT(fstat(test_file_fd_.get(), &buf), SyscallSucceeds()); EXPECT_EQ(buf.st_size, 0); // Grow to ten bytes. EXPECT_THAT(fallocate(test_file_fd_.get(), 0, 0, 10), SyscallSucceeds()); ASSERT_THAT(fstat(test_file_fd_.get(), &buf), SyscallSucceeds()); EXPECT_EQ(buf.st_size, 10); // Allocate to a smaller size should be noop. EXPECT_THAT(fallocate(test_file_fd_.get(), 0, 0, 5), SyscallSucceeds()); ASSERT_THAT(fstat(test_file_fd_.get(), &buf), SyscallSucceeds()); EXPECT_EQ(buf.st_size, 10); // Grow again. EXPECT_THAT(fallocate(test_file_fd_.get(), 0, 0, 20), SyscallSucceeds()); ASSERT_THAT(fstat(test_file_fd_.get(), &buf), SyscallSucceeds()); EXPECT_EQ(buf.st_size, 20); // Grow with offset. EXPECT_THAT(fallocate(test_file_fd_.get(), 0, 10, 20), SyscallSucceeds()); ASSERT_THAT(fstat(test_file_fd_.get(), &buf), SyscallSucceeds()); EXPECT_EQ(buf.st_size, 30); // Grow with offset beyond EOF. EXPECT_THAT(fallocate(test_file_fd_.get(), 0, 39, 1), SyscallSucceeds()); ASSERT_THAT(fstat(test_file_fd_.get(), &buf), SyscallSucceeds()); EXPECT_EQ(buf.st_size, 40); } TEST_F(AllocateTest, FallocateInvalid) { // Invalid FD EXPECT_THAT(fallocate(-1, 0, 0, 10), SyscallFailsWithErrno(EBADF)); // Negative offset and size. EXPECT_THAT(fallocate(test_file_fd_.get(), 0, -1, 10), SyscallFailsWithErrno(EINVAL)); EXPECT_THAT(fallocate(test_file_fd_.get(), 0, 0, -1), SyscallFailsWithErrno(EINVAL)); EXPECT_THAT(fallocate(test_file_fd_.get(), 0, -1, -1), SyscallFailsWithErrno(EINVAL)); } TEST_F(AllocateTest, FallocateReadonly) { auto file = ASSERT_NO_ERRNO_AND_VALUE(TempPath::CreateFile()); FileDescriptor fd = ASSERT_NO_ERRNO_AND_VALUE(Open(file.path(), O_RDONLY)); EXPECT_THAT(fallocate(fd.get(), 0, 0, 10), SyscallFailsWithErrno(EBADF)); } TEST_F(AllocateTest, FallocatePipe) { int pipes[2]; EXPECT_THAT(pipe(pipes), SyscallSucceeds()); auto cleanup = Cleanup([&pipes] { EXPECT_THAT(close(pipes[0]), SyscallSucceeds()); EXPECT_THAT(close(pipes[1]), SyscallSucceeds()); }); EXPECT_THAT(fallocate(pipes[1], 0, 0, 10), SyscallFailsWithErrno(ESPIPE)); } TEST_F(AllocateTest, FallocateChar) { const FileDescriptor fd = ASSERT_NO_ERRNO_AND_VALUE(Open("/dev/null", O_RDWR)); EXPECT_THAT(fallocate(fd.get(), 0, 0, 10), SyscallFailsWithErrno(ENODEV)); } TEST_F(AllocateTest, FallocateRlimit) { // Get the current rlimit and restore after test run. struct rlimit initial_lim; ASSERT_THAT(getrlimit(RLIMIT_FSIZE, &initial_lim), SyscallSucceeds()); auto cleanup = Cleanup([&initial_lim] { EXPECT_THAT(setrlimit(RLIMIT_FSIZE, &initial_lim), SyscallSucceeds()); }); // Try growing past the file size limit. sigset_t new_mask; sigemptyset(&new_mask); sigaddset(&new_mask, SIGXFSZ); sigprocmask(SIG_BLOCK, &new_mask, nullptr); struct rlimit setlim = {}; setlim.rlim_cur = 1024; setlim.rlim_max = RLIM_INFINITY; ASSERT_THAT(setrlimit(RLIMIT_FSIZE, &setlim), SyscallSucceeds()); EXPECT_THAT(fallocate(test_file_fd_.get(), 0, 0, 1025), SyscallFailsWithErrno(EFBIG)); struct timespec timelimit = {}; timelimit.tv_sec = 10; EXPECT_EQ(sigtimedwait(&new_mask, nullptr, &timelimit), SIGXFSZ); ASSERT_THAT(sigprocmask(SIG_UNBLOCK, &new_mask, nullptr), SyscallSucceeds()); } } // namespace } // namespace testing } // namespace gvisor