1 //===-- Unittests for vasprintf--------------------------------------------===// 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/stdio/sprintf.h" 10 #include "src/stdio/vasprintf.h" 11 #include "src/string/memset.h" 12 #include "test/UnitTest/Test.h" 13 14 int call_vasprintf(char **__restrict buffer, const char *__restrict format, 15 ...) { 16 va_list vlist; 17 va_start(vlist, format); 18 int ret = LIBC_NAMESPACE::vasprintf(buffer, format, vlist); 19 va_end(vlist); 20 return ret; 21 } 22 23 TEST(LlvmLibcVASPrintfTest, SimpleNoConv) { 24 char *buff = nullptr; 25 int written; 26 written = call_vasprintf(&buff, "A simple string with no conversions."); 27 EXPECT_EQ(written, 36); 28 ASSERT_STREQ(buff, "A simple string with no conversions."); 29 free(buff); 30 } 31 32 TEST(LlvmLibcVASPrintfTest, PercentConv) { 33 char *buff = nullptr; 34 int written; 35 36 written = call_vasprintf(&buff, "%%"); 37 EXPECT_EQ(written, 1); 38 ASSERT_STREQ(buff, "%"); 39 free(buff); 40 41 written = call_vasprintf(&buff, "abc %% def"); 42 EXPECT_EQ(written, 9); 43 ASSERT_STREQ(buff, "abc % def"); 44 free(buff); 45 46 written = call_vasprintf(&buff, "%%%%%%"); 47 EXPECT_EQ(written, 3); 48 ASSERT_STREQ(buff, "%%%"); 49 free(buff); 50 } 51 52 TEST(LlvmLibcVASPrintfTest, CharConv) { 53 char *buff = nullptr; 54 int written; 55 56 written = call_vasprintf(&buff, "%c", 'a'); 57 EXPECT_EQ(written, 1); 58 ASSERT_STREQ(buff, "a"); 59 free(buff); 60 61 written = call_vasprintf(&buff, "%3c %-3c", '1', '2'); 62 EXPECT_EQ(written, 7); 63 ASSERT_STREQ(buff, " 1 2 "); 64 free(buff); 65 66 written = call_vasprintf(&buff, "%*c", 2, '3'); 67 EXPECT_EQ(written, 2); 68 ASSERT_STREQ(buff, " 3"); 69 free(buff); 70 } 71 72 TEST(LlvmLibcVASPrintfTest, LargeStringNoConv) { 73 char *buff = nullptr; 74 char long_str[1001]; 75 LIBC_NAMESPACE::memset(long_str, 'a', 1000); 76 long_str[1000] = '\0'; 77 int written; 78 written = call_vasprintf(&buff, long_str); 79 EXPECT_EQ(written, 1000); 80 ASSERT_STREQ(buff, long_str); 81 free(buff); 82 } 83 84 TEST(LlvmLibcVASPrintfTest, ManyReAlloc) { 85 char *buff = nullptr; 86 const int expected_num_chars = 600; 87 int written = call_vasprintf(&buff, "%200s%200s%200s", "", "", ""); 88 EXPECT_EQ(written, expected_num_chars); 89 90 bool isPadding = true; 91 for (int i = 0; i < expected_num_chars; i++) { 92 if (buff[i] != ' ') { 93 isPadding = false; 94 break; 95 } 96 } 97 EXPECT_TRUE(isPadding); 98 free(buff); 99 } 100