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
14int 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
23TEST(LlvmLibcVASPrintfTest, SimpleNoConv) {
24 char *buff = nullptr;
25 int written;
26 written = call_vasprintf(buffer: &buff, format: "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
32TEST(LlvmLibcVASPrintfTest, PercentConv) {
33 char *buff = nullptr;
34 int written;
35
36 written = call_vasprintf(buffer: &buff, format: "%%");
37 EXPECT_EQ(written, 1);
38 ASSERT_STREQ(buff, "%");
39 free(buff);
40
41 written = call_vasprintf(buffer: &buff, format: "abc %% def");
42 EXPECT_EQ(written, 9);
43 ASSERT_STREQ(buff, "abc % def");
44 free(buff);
45
46 written = call_vasprintf(buffer: &buff, format: "%%%%%%");
47 EXPECT_EQ(written, 3);
48 ASSERT_STREQ(buff, "%%%");
49 free(buff);
50}
51
52TEST(LlvmLibcVASPrintfTest, CharConv) {
53 char *buff = nullptr;
54 int written;
55
56 written = call_vasprintf(buffer: &buff, format: "%c", 'a');
57 EXPECT_EQ(written, 1);
58 ASSERT_STREQ(buff, "a");
59 free(buff);
60
61 written = call_vasprintf(buffer: &buff, format: "%3c %-3c", '1', '2');
62 EXPECT_EQ(written, 7);
63 ASSERT_STREQ(buff, " 1 2 ");
64 free(buff);
65
66 written = call_vasprintf(buffer: &buff, format: "%*c", 2, '3');
67 EXPECT_EQ(written, 2);
68 ASSERT_STREQ(buff, " 3");
69 free(buff);
70}
71
72TEST(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(buffer: &buff, format: long_str);
79 EXPECT_EQ(written, 1000);
80 ASSERT_STREQ(buff, long_str);
81 free(buff);
82}
83
84TEST(LlvmLibcVASPrintfTest, ManyReAlloc) {
85 char *buff = nullptr;
86 const int expected_num_chars = 600;
87 int written = call_vasprintf(buffer: &buff, format: "%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

source code of libc/test/src/stdio/vasprintf_test.cpp