xref: /aosp_15_r20/external/llvm-libc/src/stdio/snprintf.cpp (revision 71db0c75aadcf003ffe3238005f61d7618a3fead)
1 //===-- Implementation of snprintf ------------------------------*- C++ -*-===//
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/snprintf.h"
10 
11 #include "src/__support/arg_list.h"
12 #include "src/__support/macros/config.h"
13 #include "src/stdio/printf_core/printf_main.h"
14 #include "src/stdio/printf_core/writer.h"
15 
16 #include <stdarg.h>
17 #include <stddef.h>
18 
19 namespace LIBC_NAMESPACE_DECL {
20 
21 LLVM_LIBC_FUNCTION(int, snprintf,
22                    (char *__restrict buffer, size_t buffsz,
23                     const char *__restrict format, ...)) {
24   va_list vlist;
25   va_start(vlist, format);
26   internal::ArgList args(vlist); // This holder class allows for easier copying
27                                  // and pointer semantics, as well as handling
28                                  // destruction automatically.
29   va_end(vlist);
30   printf_core::WriteBuffer wb(buffer, (buffsz > 0 ? buffsz - 1 : 0));
31   printf_core::Writer writer(&wb);
32 
33   int ret_val = printf_core::printf_main(&writer, format, args);
34   if (buffsz > 0) // if the buffsz is 0 the buffer may be a null pointer.
35     wb.buff[wb.buff_cur] = '\0';
36   return ret_val;
37 }
38 
39 } // namespace LIBC_NAMESPACE_DECL
40