diff --git a/libc/config/linux/aarch64/entrypoints.txt b/libc/config/linux/aarch64/entrypoints.txt index 1ac6bd9300008..5606b06967d62 100644 --- a/libc/config/linux/aarch64/entrypoints.txt +++ b/libc/config/linux/aarch64/entrypoints.txt @@ -22,6 +22,7 @@ set(TARGET_LIBC_ENTRYPOINTS # fcntl.h entrypoints libc.src.fcntl.creat + libc.src.fcntl.fcntl libc.src.fcntl.open libc.src.fcntl.openat diff --git a/libc/config/linux/riscv/entrypoints.txt b/libc/config/linux/riscv/entrypoints.txt index 87e82e5eb9a06..1caba997742b8 100644 --- a/libc/config/linux/riscv/entrypoints.txt +++ b/libc/config/linux/riscv/entrypoints.txt @@ -22,6 +22,7 @@ set(TARGET_LIBC_ENTRYPOINTS # fcntl.h entrypoints libc.src.fcntl.creat + libc.src.fcntl.fcntl libc.src.fcntl.open libc.src.fcntl.openat diff --git a/libc/config/linux/x86_64/entrypoints.txt b/libc/config/linux/x86_64/entrypoints.txt index a8e2899276671..2b7f27d1511f3 100644 --- a/libc/config/linux/x86_64/entrypoints.txt +++ b/libc/config/linux/x86_64/entrypoints.txt @@ -22,6 +22,7 @@ set(TARGET_LIBC_ENTRYPOINTS # fcntl.h entrypoints libc.src.fcntl.creat + libc.src.fcntl.fcntl libc.src.fcntl.open libc.src.fcntl.openat diff --git a/libc/hdr/CMakeLists.txt b/libc/hdr/CMakeLists.txt index fb7c342f92b78..179b05e6ee966 100644 --- a/libc/hdr/CMakeLists.txt +++ b/libc/hdr/CMakeLists.txt @@ -32,6 +32,15 @@ add_proxy_header_library( libc.include.math ) +add_proxy_header_library( + fcntl_macros + HDRS + fcntl_macros.h + FULL_BUILD_DEPENDS + libc.include.llvm-libc-macros.fcntl_macros + libc.include.fcntl +) + add_proxy_header_library( fenv_macros HDRS diff --git a/libc/hdr/fcntl_macros.h b/libc/hdr/fcntl_macros.h new file mode 100644 index 0000000000000..828cb984c0cb1 --- /dev/null +++ b/libc/hdr/fcntl_macros.h @@ -0,0 +1,22 @@ +//===-- Definition of macros from fcntl/fcntl.h ---------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_LIBC_HDR_FCNTL_MACROS_H +#define LLVM_LIBC_HDR_FCNTL_MACROS_H + +#ifdef LIBC_FULL_BUILD + +#include "include/llvm-libc-macros/fcntl-macros.h" + +#else // Overlay mode + +#include + +#endif // LLVM_LIBC_FULL_BUILD + +#endif // LLVM_LIBC_HDR_FCNTL_MACROS_H diff --git a/libc/hdr/types/CMakeLists.txt b/libc/hdr/types/CMakeLists.txt index f53766777e753..46a66ec590202 100644 --- a/libc/hdr/types/CMakeLists.txt +++ b/libc/hdr/types/CMakeLists.txt @@ -14,6 +14,30 @@ add_proxy_header_library( libc.include.llvm-libc-types.struct_epoll_event ) +add_proxy_header_library( + struct_flock + HDRS + struct_flock.h + FULL_BUILD_DEPENDS + libc.include.llvm-libc-types.struct_flock +) + +add_proxy_header_library( + struct_flock64 + HDRS + struct_flock64.h + FULL_BUILD_DEPENDS + libc.include.llvm-libc-types.struct_flock64 +) + +add_proxy_header_library( + struct_f_owner_ex + HDRS + struct_f_owner_ex.h + FULL_BUILD_DEPENDS + libc.include.llvm-libc-types.struct_f_owner_ex +) + add_proxy_header_library( struct_timespec HDRS diff --git a/libc/hdr/types/struct_f_owner_ex.h b/libc/hdr/types/struct_f_owner_ex.h new file mode 100644 index 0000000000000..49985115ae4bb --- /dev/null +++ b/libc/hdr/types/struct_f_owner_ex.h @@ -0,0 +1,21 @@ +//===-- Proxy for struct f_owner_ex --------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +#ifndef LLVM_LIBC_HDR_TYPES_STRUCT_F_OWNER_EX_H +#define LLVM_LIBC_HDR_TYPES_STRUCT_F_OWNER_EX_H + +#ifdef LIBC_FULL_BUILD + +#include "include/llvm-libc-types/struct_f_owner_ex.h" + +#else + +#include + +#endif // LIBC_FULL_BUILD + +#endif // LLVM_LIBC_HDR_TYPES_STRUCT_F_OWNER_EX_H diff --git a/libc/hdr/types/struct_flock.h b/libc/hdr/types/struct_flock.h new file mode 100644 index 0000000000000..a552b91c432b3 --- /dev/null +++ b/libc/hdr/types/struct_flock.h @@ -0,0 +1,21 @@ +//===-- Proxy for struct flock -------------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +#ifndef LLVM_LIBC_HDR_TYPES_STRUCT_FLOCK_H +#define LLVM_LIBC_HDR_TYPES_STRUCT_FLOCK_H + +#ifdef LIBC_FULL_BUILD + +#include "include/llvm-libc-types/struct_flock.h" + +#else + +#include + +#endif // LIBC_FULL_BUILD + +#endif // LLVM_LIBC_HDR_TYPES_STRUCT_FLOCK_H diff --git a/libc/hdr/types/struct_flock64.h b/libc/hdr/types/struct_flock64.h new file mode 100644 index 0000000000000..84fe67816c337 --- /dev/null +++ b/libc/hdr/types/struct_flock64.h @@ -0,0 +1,21 @@ +//===-- Proxy for struct flock64 -----------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +#ifndef LLVM_LIBC_HDR_TYPES_STRUCT_FLOCK64_H +#define LLVM_LIBC_HDR_TYPES_STRUCT_FLOCK64_H + +#ifdef LIBC_FULL_BUILD + +#include "include/llvm-libc-types/struct_flock64.h" + +#else + +#include + +#endif // LIBC_FULL_BUILD + +#endif // LLVM_LIBC_HDR_TYPES_STRUCT_FLOCK64_H diff --git a/libc/include/CMakeLists.txt b/libc/include/CMakeLists.txt index aeef46aabfce5..7e087ccec59cd 100644 --- a/libc/include/CMakeLists.txt +++ b/libc/include/CMakeLists.txt @@ -43,6 +43,10 @@ add_gen_header( DEPENDS .llvm-libc-macros.fcntl_macros .llvm-libc-types.mode_t + .llvm-libc-types.struct_flock + .llvm-libc-types.struct_flock64 + .llvm-libc-types.off64_t + .llvm-libc-types.pid_t .llvm-libc-types.off_t .llvm_libc_common_h ) diff --git a/libc/include/llvm-libc-macros/linux/fcntl-macros.h b/libc/include/llvm-libc-macros/linux/fcntl-macros.h index 1d4e5bbbdc770..8ee95863728e1 100644 --- a/libc/include/llvm-libc-macros/linux/fcntl-macros.h +++ b/libc/include/llvm-libc-macros/linux/fcntl-macros.h @@ -67,5 +67,36 @@ #define F_SETFD 2 #define F_GETFL 3 #define F_SETFL 4 +#define F_GETLK 5 +#define F_SETLK 6 +#define F_SETLKW 7 +#define F_SETOWN 8 +#define F_GETOWN 9 +#define F_SETSIG 10 +#define F_GETSIG 11 +#define F_GETLK64 12 +#define F_SETLK64 13 +#define F_SETLKW64 14 +#define F_SETOWN_EX 15 +#define F_GETOWN_EX 16 + +// Open File Description Locks. +#define F_OFD_GETLK 36 +#define F_OFD_SETLK 37 +#define F_OFD_SETLKW 38 + +// Close on succesful +#define F_CLOEXEC 1 + +#define F_RDLCK 0 +#define F_WRLCK 1 +#define F_UNLCK 2 + +// For Large File Support +#if defined(_LARGEFILE64_SOURCE) +#define F_GETLK F_GETLK64 +#define F_SETLK F_SETLK64 +#define F_SETLKW F_SETLKW64 +#endif #endif // LLVM_LIBC_MACROS_LINUX_FCNTL_MACROS_H diff --git a/libc/include/llvm-libc-types/CMakeLists.txt b/libc/include/llvm-libc-types/CMakeLists.txt index 310374fb62ffe..a5687d5bbc853 100644 --- a/libc/include/llvm-libc-types/CMakeLists.txt +++ b/libc/include/llvm-libc-types/CMakeLists.txt @@ -60,6 +60,9 @@ add_header(rlim_t HDR rlim_t.h) add_header(time_t HDR time_t.h) add_header(stack_t HDR stack_t.h) add_header(suseconds_t HDR suseconds_t.h) +add_header(struct_flock HDR struct_flock.h DEPENDS .off_t .pid_t) +add_header(struct_flock64 HDR struct_flock64.h DEPENDS .off64_t .pid_t) +add_header(struct_f_owner_ex HDR struct_f_owner_ex.h DEPENDS .pid_t) add_header(struct_timeval HDR struct_timeval.h DEPENDS .suseconds_t .time_t) add_header(struct_rlimit HDR struct_rlimit.h DEPENDS .rlim_t) add_header(struct_rusage HDR struct_rusage.h DEPENDS .struct_timeval) diff --git a/libc/include/llvm-libc-types/struct_f_owner_ex.h b/libc/include/llvm-libc-types/struct_f_owner_ex.h new file mode 100644 index 0000000000000..c9cc85f69d2b4 --- /dev/null +++ b/libc/include/llvm-libc-types/struct_f_owner_ex.h @@ -0,0 +1,25 @@ +//===-- Definition of type struct f_owner_ex ------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_LIBC_TYPES_STRUCT_F_OWNER_EX_H +#define LLVM_LIBC_TYPES_STRUCT_F_OWNER_EX_H + +#include "llvm-libc-types/pid_t.h" + +enum pid_type { + F_OWNER_TID = 0, + F_OWNER_PID, + F_OWNER_PGRP, +}; + +struct f_owner_ex { + enum pid_type type; + pid_t pid; +}; + +#endif // LLVM_LIBC_TYPES_STRUCT_F_OWNER_EX_H diff --git a/libc/include/llvm-libc-types/struct_flock.h b/libc/include/llvm-libc-types/struct_flock.h new file mode 100644 index 0000000000000..51c9d27640ea1 --- /dev/null +++ b/libc/include/llvm-libc-types/struct_flock.h @@ -0,0 +1,25 @@ +//===-- Definition of type struct flock64 ---------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_LIBC_TYPES_STRUCT_FLOCK_H +#define LLVM_LIBC_TYPES_STRUCT_FLOCK_H + +#include "llvm-libc-types/off_t.h" +#include "llvm-libc-types/pid_t.h" + +#include + +struct flock { + int16_t l_type; + int16_t l_whence; + off_t l_start; + off_t l_len; + pid_t l_pid; +}; + +#endif // LLVM_LIBC_TYPES_STRUCT_FLOCK_H diff --git a/libc/include/llvm-libc-types/struct_flock64.h b/libc/include/llvm-libc-types/struct_flock64.h new file mode 100644 index 0000000000000..ac50003ca62f6 --- /dev/null +++ b/libc/include/llvm-libc-types/struct_flock64.h @@ -0,0 +1,25 @@ +//===-- Definition of type struct flock64 ---------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_LIBC_TYPES_STRUCT_FLOCK64_H +#define LLVM_LIBC_TYPES_STRUCT_FLOCK64_H + +#include "llvm-libc-types/off64_t.h" +#include "llvm-libc-types/pid_t.h" + +#include + +struct flock64 { + int16_t l_type; + int16_t l_whence; + off64_t l_start; + off64_t l_len; + pid_t l_pid; +}; + +#endif // LLVM_LIBC_TYPES_STRUCT_FLOCK64_H diff --git a/libc/spec/posix.td b/libc/spec/posix.td index d428d54e32a33..e7a0cf883c607 100644 --- a/libc/spec/posix.td +++ b/libc/spec/posix.td @@ -230,6 +230,11 @@ def POSIX : StandardSpec<"POSIX"> { RetValSpec, [ArgSpec, ArgSpec] >, + FunctionSpec< + "fcntl", + RetValSpec, + [ArgSpec, ArgSpec, ArgSpec] + >, FunctionSpec< "open", RetValSpec, diff --git a/libc/src/fcntl/CMakeLists.txt b/libc/src/fcntl/CMakeLists.txt index 0b9ee47c4f7c1..77400e9050d08 100644 --- a/libc/src/fcntl/CMakeLists.txt +++ b/libc/src/fcntl/CMakeLists.txt @@ -9,6 +9,13 @@ add_entrypoint_object( .${LIBC_TARGET_OS}.creat ) +add_entrypoint_object( + fcntl + ALIAS + DEPENDS + .${LIBC_TARGET_OS}.fcntl +) + add_entrypoint_object( open ALIAS diff --git a/libc/src/fcntl/fcntl.h b/libc/src/fcntl/fcntl.h new file mode 100644 index 0000000000000..8fe3fb3146b91 --- /dev/null +++ b/libc/src/fcntl/fcntl.h @@ -0,0 +1,18 @@ +//===-- Implementation header of fcntl --------------------------*- C++ -*-===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_LIBC_SRC_FCNTL_FCNTL_H +#define LLVM_LIBC_SRC_FCNTL_FCNTL_H + +namespace LIBC_NAMESPACE { + +int fcntl(int fd, int cmd, ...); + +} // namespace LIBC_NAMESPACE + +#endif // LLVM_LIBC_SRC_FCNTL_FCNTL_H diff --git a/libc/src/fcntl/linux/CMakeLists.txt b/libc/src/fcntl/linux/CMakeLists.txt index 87b8d4695c4fc..732b7beac41bf 100644 --- a/libc/src/fcntl/linux/CMakeLists.txt +++ b/libc/src/fcntl/linux/CMakeLists.txt @@ -10,6 +10,22 @@ add_entrypoint_object( libc.src.errno.errno ) +add_entrypoint_object( + fcntl + SRCS + fcntl.cpp + HDRS + ../fcntl.h + DEPENDS + libc.include.fcntl + libc.hdr.types.struct_flock + libc.hdr.types.struct_flock64 + libc.hdr.types.struct_f_owner_ex + libc.hdr.fcntl_macros + libc.src.__support.OSUtil.osutil + libc.src.errno.errno +) + add_entrypoint_object( open SRCS diff --git a/libc/src/fcntl/linux/fcntl.cpp b/libc/src/fcntl/linux/fcntl.cpp new file mode 100644 index 0000000000000..24a20fb364109 --- /dev/null +++ b/libc/src/fcntl/linux/fcntl.cpp @@ -0,0 +1,93 @@ +//===-- Implementation of fcntl -------------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#include "src/fcntl/fcntl.h" + +#include "hdr/fcntl_macros.h" +#include "hdr/types/struct_f_owner_ex.h" +#include "hdr/types/struct_flock.h" +#include "hdr/types/struct_flock64.h" +#include "src/__support/OSUtil/syscall.h" // For internal syscall function. +#include "src/__support/common.h" +#include "src/errno/libc_errno.h" + +#include +#include // For syscall numbers. + +// The OFD file locks require special handling for LARGEFILES +namespace LIBC_NAMESPACE { +LLVM_LIBC_FUNCTION(int, fcntl, (int fd, int cmd, ...)) { + void *arg; + va_list varargs; + va_start(varargs, cmd); + arg = va_arg(varargs, void *); + va_end(varargs); + + switch (cmd) { + case F_SETLKW: + return syscall_impl(SYS_fcntl, fd, cmd, arg); + case F_OFD_SETLKW: { + struct flock *flk = reinterpret_cast(arg); + // convert the struct to a flock64 + struct flock64 flk64; + flk64.l_type = flk->l_type; + flk64.l_whence = flk->l_whence; + flk64.l_start = flk->l_start; + flk64.l_len = flk->l_len; + flk64.l_pid = flk->l_pid; + // create a syscall + return syscall_impl(SYS_fcntl, fd, cmd, &flk64); + } + case F_OFD_GETLK: + case F_OFD_SETLK: { + struct flock *flk = reinterpret_cast(arg); + // convert the struct to a flock64 + struct flock64 flk64; + flk64.l_type = flk->l_type; + flk64.l_whence = flk->l_whence; + flk64.l_start = flk->l_start; + flk64.l_len = flk->l_len; + flk64.l_pid = flk->l_pid; + // create a syscall + int retVal = syscall_impl(SYS_fcntl, fd, cmd, &flk64); + // On failure, return + if (retVal == -1) + return -1; + // Check for overflow, i.e. the offsets are not the same when cast + // to off_t from off64_t. + if (static_cast(flk64.l_len) != flk64.l_len || + static_cast(flk64.l_start) != flk64.l_start) { + libc_errno = EOVERFLOW; + return -1; + } + // Now copy back into flk, in case flk64 got modified + flk->l_type = flk64.l_type; + flk->l_whence = flk64.l_whence; + flk->l_start = flk64.l_start; + flk->l_len = flk64.l_len; + flk->l_pid = flk64.l_pid; + return retVal; + } + case F_GETOWN: { + struct f_owner_ex fex; + int retVal = syscall_impl(SYS_fcntl, fd, F_GETOWN_EX, &fex); + if (retVal == -EINVAL) + return syscall_impl(SYS_fcntl, fd, cmd, + reinterpret_cast(arg)); + if (static_cast(retVal) <= -4096UL) + return fex.type == F_OWNER_PGRP ? -fex.pid : fex.pid; + + libc_errno = -retVal; + return -1; + } + // The general case + default: + return syscall_impl(SYS_fcntl, fd, cmd, reinterpret_cast(arg)); + } +} +} // namespace LIBC_NAMESPACE diff --git a/libc/test/src/fcntl/CMakeLists.txt b/libc/test/src/fcntl/CMakeLists.txt index ae39d8d5f878c..aae296f074bea 100644 --- a/libc/test/src/fcntl/CMakeLists.txt +++ b/libc/test/src/fcntl/CMakeLists.txt @@ -17,6 +17,23 @@ add_libc_unittest( libc.test.UnitTest.ErrnoSetterMatcher ) +add_libc_unittest( + fcntl_test + SUITE + libc_fcntl_unittests + SRCS + fcntl_test.cpp + DEPENDS + libc.include.fcntl + libc.src.errno.errno + libc.src.fcntl.fcntl + libc.src.fcntl.open + libc.src.unistd.close + libc.hdr.types.struct_flock + libc.hdr.fcntl_macros + libc.test.UnitTest.ErrnoSetterMatcher +) + add_libc_unittest( openat_test SUITE diff --git a/libc/test/src/fcntl/fcntl_test.cpp b/libc/test/src/fcntl/fcntl_test.cpp new file mode 100644 index 0000000000000..c5cbb61b4ed8a --- /dev/null +++ b/libc/test/src/fcntl/fcntl_test.cpp @@ -0,0 +1,155 @@ +//===-- Unittest for fcntl ------------------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#include "hdr/fcntl_macros.h" +#include "hdr/types/struct_flock.h" +#include "src/errno/libc_errno.h" +#include "src/fcntl/fcntl.h" +#include "src/fcntl/open.h" +#include "src/unistd/close.h" +#include "test/UnitTest/ErrnoSetterMatcher.h" +#include "test/UnitTest/Test.h" + +#include +#include // For S_IRWXU + +TEST(LlvmLibcFcntlTest, FcntlDupfd) { + using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds; + constexpr const char *TEST_FILE_NAME = "testdata/fcntl_dup.test"; + auto TEST_FILE = libc_make_test_file_path(TEST_FILE_NAME); + int fd2, fd3; + int fd = LIBC_NAMESPACE::open(TEST_FILE, O_CREAT | O_TRUNC, S_IRWXU); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(fd, 0); + + fd2 = LIBC_NAMESPACE::fcntl(fd, F_DUPFD, 0); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(fd2, 0); + + fd3 = LIBC_NAMESPACE::fcntl(fd, F_DUPFD, 10); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(fd3, 0); + + ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0)); + ASSERT_THAT(LIBC_NAMESPACE::close(fd2), Succeeds(0)); + ASSERT_THAT(LIBC_NAMESPACE::close(fd3), Succeeds(0)); +} + +TEST(LlvmLibcFcntlTest, FcntlGetFl) { + using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds; + constexpr const char *TEST_FILE_NAME = "testdata/fcntl_getfl.test"; + auto TEST_FILE = libc_make_test_file_path(TEST_FILE_NAME); + int retVal; + int fd = LIBC_NAMESPACE::open(TEST_FILE, O_CREAT | O_TRUNC, S_IRWXU); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(fd, 0); + + retVal = LIBC_NAMESPACE::fcntl(fd, F_GETFL); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(retVal, -1); + + ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0)); +} + +TEST(LlvmLibcFcntlTest, FcntlSetFl) { + using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds; + constexpr const char *TEST_FILE_NAME = "testdata/fcntl_setfl.test"; + auto TEST_FILE = libc_make_test_file_path(TEST_FILE_NAME); + + int retVal; + int fd = LIBC_NAMESPACE::open(TEST_FILE, O_CREAT | O_TRUNC | O_RDWR, S_IRWXU); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(fd, 0); + + retVal = LIBC_NAMESPACE::fcntl(fd, F_GETFL); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(retVal, -1); + + int oldFlags = LIBC_NAMESPACE::fcntl(fd, F_GETFL, 0); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(oldFlags, 0); + + // Add the APPEND flag; + oldFlags |= O_APPEND; + + retVal = LIBC_NAMESPACE::fcntl(fd, F_SETFL, oldFlags); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(retVal, -1); + + // Remove the APPEND flag; + oldFlags = -oldFlags & O_APPEND; + + retVal = LIBC_NAMESPACE::fcntl(fd, F_SETFL, oldFlags); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(retVal, -1); + + ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0)); +} + +TEST(LlvmLibcFcntlTest, FcntlGetLkRead) { + using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds; + constexpr const char *TEST_FILE_NAME = "testdata/fcntl_getlkread.test"; + auto TEST_FILE = libc_make_test_file_path(TEST_FILE_NAME); + + struct flock flk, svflk; + int retVal; + int fd = + LIBC_NAMESPACE::open(TEST_FILE, O_CREAT | O_TRUNC | O_RDONLY, S_IRWXU); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(fd, 0); + + flk.l_type = F_RDLCK; + flk.l_start = 0; + flk.l_whence = SEEK_SET; + flk.l_len = 50; + + // copy flk into svflk + svflk = flk; + + retVal = LIBC_NAMESPACE::fcntl(fd, F_GETLK, &svflk); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(retVal, -1); + ASSERT_NE((int)flk.l_type, F_WRLCK); // File should not be write locked. + + retVal = LIBC_NAMESPACE::fcntl(fd, F_SETLK, &svflk); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(retVal, -1); + + ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0)); +} + +TEST(LlvmLibcFcntlTest, FcntlGetLkWrite) { + using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds; + constexpr const char *TEST_FILE_NAME = "testdata/fcntl_getlkwrite.test"; + auto TEST_FILE = libc_make_test_file_path(TEST_FILE_NAME); + + struct flock flk, svflk; + int retVal; + int fd = LIBC_NAMESPACE::open(TEST_FILE, O_CREAT | O_TRUNC | O_RDWR, S_IRWXU); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(fd, 0); + + flk.l_type = F_WRLCK; + flk.l_start = 0; + flk.l_whence = SEEK_SET; + flk.l_len = 0; + + // copy flk into svflk + svflk = flk; + + retVal = LIBC_NAMESPACE::fcntl(fd, F_GETLK, &svflk); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(retVal, -1); + ASSERT_NE((int)flk.l_type, F_RDLCK); // File should not be read locked. + + retVal = LIBC_NAMESPACE::fcntl(fd, F_SETLK, &svflk); + ASSERT_ERRNO_SUCCESS(); + ASSERT_GT(retVal, -1); + + ASSERT_THAT(LIBC_NAMESPACE::close(fd), Succeeds(0)); +}