68 changes: 68 additions & 0 deletions libc/test/src/stdio/setbuf_test.cpp
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
//===-- Unittests for setbuf ----------------------------------------------===//
//
// 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/stdio/fclose.h"
#include "src/stdio/fopen.h"
#include "src/stdio/fread.h"
#include "src/stdio/fwrite.h"
#include "src/stdio/setbuf.h"
#include "src/stdio/ungetc.h"
#include "utils/UnitTest/Test.h"

#include <stdio.h>

TEST(LlvmLibcSetbufTest, DefaultBufsize) {
// The idea in this test is to change the buffer after opening a file and
// ensure that read and write work as expected.
constexpr char FILENAME[] = "testdata/setbuf_test_default_bufsize.test";
::FILE *file = __llvm_libc::fopen(FILENAME, "w");
ASSERT_FALSE(file == nullptr);
char buffer[BUFSIZ];
__llvm_libc::setbuf(file, buffer);
constexpr char CONTENT[] = "abcdef";
constexpr size_t CONTENT_SIZE = sizeof(CONTENT);
ASSERT_EQ(CONTENT_SIZE, __llvm_libc::fwrite(CONTENT, 1, CONTENT_SIZE, file));
ASSERT_EQ(0, __llvm_libc::fclose(file));

file = __llvm_libc::fopen(FILENAME, "r");
__llvm_libc::setbuf(file, buffer);
ASSERT_FALSE(file == nullptr);
char data[CONTENT_SIZE];
ASSERT_EQ(__llvm_libc::fread(&data, 1, CONTENT_SIZE, file), CONTENT_SIZE);
ASSERT_STREQ(CONTENT, data);
ASSERT_EQ(0, __llvm_libc::fclose(file));
}

TEST(LlvmLibcSetbufTest, NullBuffer) {
// The idea in this test is that we set a null buffer and ensure that
// everything works correctly.
constexpr char FILENAME[] = "testdata/setbuf_test_null_buffer.test";
::FILE *file = __llvm_libc::fopen(FILENAME, "w");
ASSERT_FALSE(file == nullptr);
__llvm_libc::setbuf(file, nullptr);
constexpr char CONTENT[] = "abcdef";
constexpr size_t CONTENT_SIZE = sizeof(CONTENT);
ASSERT_EQ(CONTENT_SIZE, __llvm_libc::fwrite(CONTENT, 1, CONTENT_SIZE, file));
ASSERT_EQ(0, __llvm_libc::fclose(file));

file = __llvm_libc::fopen(FILENAME, "r");
__llvm_libc::setbuf(file, nullptr);
ASSERT_FALSE(file == nullptr);
char data[CONTENT_SIZE];
ASSERT_EQ(__llvm_libc::fread(&data, 1, CONTENT_SIZE, file), CONTENT_SIZE);
ASSERT_STREQ(CONTENT, data);

// Ensure that ungetc also works.
char unget_char = 'z';
ASSERT_EQ(int(unget_char), __llvm_libc::ungetc(unget_char, file));
char c;
ASSERT_EQ(__llvm_libc::fread(&c, 1, 1, file), size_t(1));
ASSERT_EQ(c, unget_char);

ASSERT_EQ(0, __llvm_libc::fclose(file));
}
106 changes: 106 additions & 0 deletions libc/test/src/stdio/setvbuf_test.cpp
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,106 @@
//===-- Unittests for setvbuf ---------------------------------------------===//
//
// 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/stdio/fclose.h"
#include "src/stdio/fopen.h"
#include "src/stdio/fread.h"
#include "src/stdio/fwrite.h"
#include "src/stdio/setvbuf.h"
#include "utils/UnitTest/Test.h"

#include <errno.h>
#include <stdio.h>

TEST(LlvmLibcSetvbufTest, SetNBFBuffer) {
// The idea in this test is that we open a file for writing and reading, and
// then set a NBF buffer to the write handle. Since it is NBF, the data
// written using the write handle should be immediately readable by the read
// handle.
constexpr char FILENAME[] = "testdata/setvbuf_nbf.test";

::FILE *fw = __llvm_libc::fopen(FILENAME, "w");
ASSERT_FALSE(fw == nullptr);
char buffer[BUFSIZ];
ASSERT_EQ(__llvm_libc::setvbuf(fw, buffer, _IONBF, BUFSIZ), 0);

::FILE *fr = __llvm_libc::fopen(FILENAME, "r");
ASSERT_FALSE(fr == nullptr);

constexpr char CONTENT[] = "abcdef";
constexpr size_t CONTENT_SIZE = sizeof(CONTENT);
for (size_t i = 0; i < CONTENT_SIZE; ++i) {
ASSERT_EQ(size_t(1), __llvm_libc::fwrite(CONTENT + i, 1, 1, fw));
char c;
ASSERT_EQ(size_t(1), __llvm_libc::fread(&c, 1, 1, fr));
ASSERT_EQ(c, CONTENT[i]);
}

ASSERT_EQ(0, __llvm_libc::fclose(fw));
ASSERT_EQ(0, __llvm_libc::fclose(fr));

// Make sure NBF buffer has no effect for reading.
fr = __llvm_libc::fopen(FILENAME, "r");
char data[CONTENT_SIZE];
ASSERT_EQ(__llvm_libc::setvbuf(fr, buffer, _IONBF, BUFSIZ), 0);
ASSERT_EQ(CONTENT_SIZE, __llvm_libc::fread(data, 1, CONTENT_SIZE, fr));
ASSERT_STREQ(CONTENT, data);
ASSERT_EQ(0, __llvm_libc::fclose(fr));
}

TEST(LlvmLibcSetvbufTest, SetLBFBuffer) {
// The idea in this test is that we open a file for writing and reading, and
// then set a LBF buffer to the write handle. Since it is LBF, the data
// written using the write handle should be available right after a '\n' is
// written.
constexpr char FILENAME[] = "testdata/setvbuf_lbf.test";

::FILE *fw = __llvm_libc::fopen(FILENAME, "w");
ASSERT_FALSE(fw == nullptr);
char buffer[BUFSIZ];
ASSERT_EQ(__llvm_libc::setvbuf(fw, buffer, _IOLBF, BUFSIZ), 0);

::FILE *fr = __llvm_libc::fopen(FILENAME, "r");
ASSERT_FALSE(fr == nullptr);

constexpr char CONTENT[] = "abcdef\n";
constexpr size_t CONTENT_SIZE = sizeof(CONTENT);
ASSERT_EQ(CONTENT_SIZE, __llvm_libc::fwrite(CONTENT, 1, CONTENT_SIZE, fw));

// Note that CONTENT_SIZE worth of data written also includes the
// null-terminator '\0'. But, since it is after the new line character,
// it should not be availabe for reading.
char data[CONTENT_SIZE];
ASSERT_EQ(CONTENT_SIZE - 1, __llvm_libc::fread(data, 1, CONTENT_SIZE, fr));
char c;
ASSERT_EQ(size_t(0), __llvm_libc::fread(&c, 1, 1, fr));

data[CONTENT_SIZE - 1] = '\0';
ASSERT_STREQ(CONTENT, data);

ASSERT_EQ(0, __llvm_libc::fclose(fw));
ASSERT_EQ(0, __llvm_libc::fclose(fr));

// Make sure LBF buffer has no effect for reading.
fr = __llvm_libc::fopen(FILENAME, "r");
ASSERT_EQ(__llvm_libc::setvbuf(fr, buffer, _IOLBF, BUFSIZ), 0);
ASSERT_EQ(CONTENT_SIZE, __llvm_libc::fread(data, 1, CONTENT_SIZE, fr));
ASSERT_STREQ(CONTENT, data);
ASSERT_EQ(0, __llvm_libc::fclose(fr));
}

TEST(LlvmLibcSetbufTest, InvalidBufferMode) {
constexpr char FILENAME[] = "testdata/setvbuf_invalid_bufmode.test";
::FILE *f = __llvm_libc::fopen(FILENAME, "w");
ASSERT_FALSE(f == nullptr);
char buf[BUFSIZ];
ASSERT_NE(__llvm_libc::setvbuf(f, buf, _IOFBF + _IOLBF + _IONBF, BUFSIZ), 0);
ASSERT_EQ(errno, EINVAL);

errno = 0;
ASSERT_EQ(0, __llvm_libc::fclose(f));
}
1 change: 1 addition & 0 deletions libc/utils/HdrGen/PrototypeTestGen/PrototypeTestGen.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -94,6 +94,7 @@ bool TestGeneratorMain(llvm::raw_ostream &OS, llvm::RecordKeeper &records) {
// We provide dummy malloc and free implementations to support the case
// when LLVM libc does to include them.
OS << "void *malloc(size_t) { return nullptr; }\n";
OS << "void *realloc(void *, size_t) { return nullptr; }\n";
OS << "void free(void *) {}\n";

return false;
Expand Down