67 lines · cpp
1//===-- Unittests for dup -------------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "src/fcntl/open.h"10#include "src/unistd/close.h"11#include "src/unistd/dup2.h"12#include "src/unistd/read.h"13#include "src/unistd/unlink.h"14#include "src/unistd/write.h"15#include "test/UnitTest/ErrnoCheckingTest.h"16#include "test/UnitTest/ErrnoSetterMatcher.h"17#include "test/UnitTest/Test.h"18 19#include <sys/stat.h>20 21using LlvmLibcdupTest = LIBC_NAMESPACE::testing::ErrnoCheckingTest;22 23TEST_F(LlvmLibcdupTest, ReadAndWriteViaDup) {24 constexpr int DUPFD = 0xD0;25 using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds;26 constexpr const char *FILENAME = "dup2.test";27 auto TEST_FILE = libc_make_test_file_path(FILENAME);28 int fd = LIBC_NAMESPACE::open(TEST_FILE, O_WRONLY | O_CREAT, S_IRWXU);29 ASSERT_ERRNO_SUCCESS();30 ASSERT_GT(fd, 0);31 int dupfd = LIBC_NAMESPACE::dup2(fd, DUPFD);32 ASSERT_ERRNO_SUCCESS();33 ASSERT_EQ(dupfd, DUPFD);34 35 // Write something via the dup36 constexpr char WRITE_DATA[] = "Hello, dup!";37 constexpr size_t WRITE_SIZE = sizeof(WRITE_DATA);38 ASSERT_EQ(ssize_t(WRITE_SIZE),39 LIBC_NAMESPACE::write(dupfd, WRITE_DATA, WRITE_SIZE));40 ASSERT_THAT(LIBC_NAMESPACE::close(dupfd), Succeeds(0));41 42 // Reopen the file for reading and create a dup.43 fd = LIBC_NAMESPACE::open(TEST_FILE, O_RDONLY);44 ASSERT_ERRNO_SUCCESS();45 ASSERT_GT(fd, 0);46 dupfd = LIBC_NAMESPACE::dup2(fd, DUPFD);47 ASSERT_ERRNO_SUCCESS();48 ASSERT_EQ(dupfd, DUPFD);49 50 // Read the file content via the dup.51 char buf[WRITE_SIZE];52 ASSERT_THAT(LIBC_NAMESPACE::read(dupfd, buf, WRITE_SIZE),53 Succeeds(WRITE_SIZE));54 ASSERT_STREQ(buf, WRITE_DATA);55 56 // Verify that duping to the same fd value succeeds.57 ASSERT_THAT(LIBC_NAMESPACE::dup2(dupfd, dupfd), Succeeds(dupfd));58 59 ASSERT_THAT(LIBC_NAMESPACE::close(dupfd), Succeeds(0));60 ASSERT_THAT(LIBC_NAMESPACE::unlink(TEST_FILE), Succeeds(0));61}62 63TEST_F(LlvmLibcdupTest, DupBadFD) {64 using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Fails;65 ASSERT_THAT(LIBC_NAMESPACE::dup2(-1, 123), Fails(EBADF));66}67