1 //===-- Unittests for vfscanf ---------------------------------------------===// 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-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "src/__support/CPP/string_view.h" 10 11 #ifndef LIBC_COPT_STDIO_USE_SYSTEM_FILE 12 #include "src/stdio/fclose.h" 13 #include "src/stdio/ferror.h" 14 #include "src/stdio/fopen.h" 15 #include "src/stdio/fwrite.h" 16 #endif // LIBC_COPT_STDIO_USE_SYSTEM_FILE 17 18 #include "src/stdio/vfscanf.h" 19 20 #include "test/UnitTest/Test.h" 21 22 namespace scanf_test { 23 #ifndef LIBC_COPT_STDIO_USE_SYSTEM_FILE 24 using LIBC_NAMESPACE::fclose; 25 using LIBC_NAMESPACE::ferror; 26 using LIBC_NAMESPACE::fopen; 27 using LIBC_NAMESPACE::fwrite; 28 #else // defined(LIBC_COPT_STDIO_USE_SYSTEM_FILE) 29 using ::fclose; 30 using ::ferror; 31 using ::fopen; 32 using ::fwrite; 33 #endif // LIBC_COPT_STDIO_USE_SYSTEM_FILE 34 } // namespace scanf_test 35 36 static int call_vfscanf(::FILE *stream, const char *__restrict format, ...) { 37 va_list vlist; 38 va_start(vlist, format); 39 int ret = LIBC_NAMESPACE::vfscanf(stream, format, vlist); 40 va_end(vlist); 41 return ret; 42 } 43 44 TEST(LlvmLibcVFScanfTest, WriteToFile) { 45 const char *FILENAME = "vfscanf_output.test"; 46 auto FILE_PATH = libc_make_test_file_path(FILENAME); 47 ::FILE *file = scanf_test::fopen(FILE_PATH, "w"); 48 ASSERT_FALSE(file == nullptr); 49 50 int read; 51 52 constexpr char simple[] = "A simple string with no conversions.\n"; 53 54 ASSERT_EQ(sizeof(simple) - 1, 55 scanf_test::fwrite(simple, 1, sizeof(simple) - 1, file)); 56 57 constexpr char numbers[] = "1234567890\n"; 58 59 ASSERT_EQ(sizeof(numbers) - 1, 60 scanf_test::fwrite(numbers, 1, sizeof(numbers) - 1, file)); 61 62 constexpr char numbers_and_more[] = "1234 and more\n"; 63 64 ASSERT_EQ(sizeof(numbers_and_more) - 1, 65 scanf_test::fwrite(numbers_and_more, 1, 66 sizeof(numbers_and_more) - 1, file)); 67 68 read = call_vfscanf(file, "Reading from a write-only file should fail."); 69 EXPECT_LT(read, 0); 70 71 ASSERT_EQ(0, scanf_test::fclose(file)); 72 73 file = scanf_test::fopen(FILE_PATH, "r"); 74 ASSERT_FALSE(file == nullptr); 75 76 char data[50]; 77 read = call_vfscanf(file, "%[A-Za-z .\n]", data); 78 ASSERT_EQ(read, 1); 79 ASSERT_STREQ(simple, data); 80 81 read = call_vfscanf(file, "%s", data); 82 ASSERT_EQ(read, 1); 83 ASSERT_EQ(LIBC_NAMESPACE::cpp::string_view(numbers, 10), 84 LIBC_NAMESPACE::cpp::string_view(data)); 85 86 // The format string starts with a space to handle the fact that the %s leaves 87 // a trailing \n and %c doesn't strip leading whitespace. 88 read = call_vfscanf(file, " %50c", data); 89 ASSERT_EQ(read, 1); 90 ASSERT_EQ( 91 LIBC_NAMESPACE::cpp::string_view(numbers_and_more), 92 LIBC_NAMESPACE::cpp::string_view(data, sizeof(numbers_and_more) - 1)); 93 94 ASSERT_EQ(scanf_test::ferror(file), 0); 95 ASSERT_EQ(scanf_test::fclose(file), 0); 96 } 97