Tue, 16 Jan 2024 23:13:01 +0100
add cx_sprintf() variants - fixes #353
1 /*
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
3 *
4 * Copyright 2021 Mike Becker, Olaf Wintermann All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions are met:
8 *
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 *
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
29 #include "cx/printf.h"
31 #include <stdio.h>
32 #include <string.h>
34 #ifndef CX_PRINTF_SBO_SIZE
35 #define CX_PRINTF_SBO_SIZE 512
36 #endif
37 unsigned const cx_printf_sbo_size = CX_PRINTF_SBO_SIZE;
39 int cx_fprintf(
40 void *stream,
41 cx_write_func wfc,
42 char const *fmt,
43 ...
44 ) {
45 int ret;
46 va_list ap;
47 va_start(ap, fmt);
48 ret = cx_vfprintf(stream, wfc, fmt, ap);
49 va_end(ap);
50 return ret;
51 }
53 int cx_vfprintf(
54 void *stream,
55 cx_write_func wfc,
56 char const *fmt,
57 va_list ap
58 ) {
59 char buf[CX_PRINTF_SBO_SIZE];
60 va_list ap2;
61 va_copy(ap2, ap);
62 int ret = vsnprintf(buf, CX_PRINTF_SBO_SIZE, fmt, ap);
63 if (ret < 0) {
64 return ret;
65 } else if (ret < CX_PRINTF_SBO_SIZE) {
66 return (int) wfc(buf, 1, ret, stream);
67 } else {
68 int len = ret + 1;
69 char *newbuf = malloc(len);
70 if (!newbuf) {
71 return -1;
72 }
74 ret = vsnprintf(newbuf, len, fmt, ap2);
75 if (ret > 0) {
76 ret = (int) wfc(newbuf, 1, ret, stream);
77 }
78 free(newbuf);
79 }
80 return ret;
81 }
83 cxmutstr cx_asprintf_a(
84 CxAllocator const *allocator,
85 char const *fmt,
86 ...
87 ) {
88 va_list ap;
89 va_start(ap, fmt);
90 cxmutstr ret = cx_vasprintf_a(allocator, fmt, ap);
91 va_end(ap);
92 return ret;
93 }
95 cxmutstr cx_vasprintf_a(
96 CxAllocator const *a,
97 char const *fmt,
98 va_list ap
99 ) {
100 cxmutstr s;
101 s.ptr = NULL;
102 s.length = 0;
103 char buf[CX_PRINTF_SBO_SIZE];
104 va_list ap2;
105 va_copy(ap2, ap);
106 int ret = vsnprintf(buf, CX_PRINTF_SBO_SIZE, fmt, ap);
107 if (ret > 0 && ret < CX_PRINTF_SBO_SIZE) {
108 s.ptr = cxMalloc(a, ret + 1);
109 if (s.ptr) {
110 s.length = (size_t) ret;
111 memcpy(s.ptr, buf, ret);
112 s.ptr[s.length] = '\0';
113 }
114 } else {
115 int len = ret + 1;
116 s.ptr = cxMalloc(a, len);
117 if (s.ptr) {
118 ret = vsnprintf(s.ptr, len, fmt, ap2);
119 if (ret < 0) {
120 free(s.ptr);
121 s.ptr = NULL;
122 } else {
123 s.length = (size_t) ret;
124 }
125 }
126 }
127 return s;
128 }
130 int cx_sprintf_a(CxAllocator *alloc, char **str, size_t len, const char *fmt, ... ) {
131 va_list ap;
132 va_start(ap, fmt);
133 int ret = cx_vsprintf_a(alloc, str, len, fmt, ap);
134 va_end(ap);
135 return ret;
136 }
138 int cx_vsprintf_a(CxAllocator *alloc, char **str, size_t len, const char *fmt, va_list ap) {
139 va_list ap2;
140 va_copy(ap2, ap);
141 int ret = vsnprintf(*str, len, fmt, ap);
142 if (ret < 0 || ((unsigned)ret) < len) {
143 va_end(ap2);
144 return ret;
145 } else {
146 unsigned newlen = ret + 1;
147 char *ptr = cxRealloc(alloc, *str, newlen);
148 if (ptr) {
149 int newret = vsnprintf(ptr, newlen, fmt, ap2);
150 va_end(ap2);
151 if (newret < 0) {
152 cxFree(alloc, ptr);
153 return ret;
154 } else {
155 *str = ptr;
156 return newret;
157 }
158 } else {
159 va_end(ap2);
160 return ret;
161 }
162 }
163 }
165 int cx_sprintf_sa(CxAllocator *alloc, char *buf, size_t len, char **str, const char *fmt, ... ) {
166 va_list ap;
167 va_start(ap, fmt);
168 int ret = cx_vsprintf_sa(alloc, buf, len, str, fmt, ap);
169 va_end(ap);
170 return ret;
171 }
173 int cx_vsprintf_sa(CxAllocator *alloc, char *buf, size_t len, char **str, const char *fmt, va_list ap) {
174 va_list ap2;
175 va_copy(ap2, ap);
176 int ret = vsnprintf(buf, len, fmt, ap);
177 *str = buf;
178 if (ret < 0 || ((unsigned)ret) < len) {
179 va_end(ap2);
180 return ret;
181 } else {
182 unsigned newlen = ret + 1;
183 char *ptr = cxMalloc(alloc, newlen);
184 if (ptr) {
185 int newret = vsnprintf(ptr, newlen, fmt, ap2);
186 va_end(ap2);
187 if (newret < 0) {
188 cxFree(alloc, ptr);
189 return ret;
190 } else {
191 *str = ptr;
192 return newret;
193 }
194 } else {
195 va_end(ap2);
196 return ret;
197 }
198 }
199 }