12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097 |
- /*
- * Copyright 2024 Vsevolod Stakhov
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
- /* Copyright (C) 2002-2015 Igor Sysoev
- * Copyright (C) 2011-2015 Nginx, Inc.
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are met:
- * * Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * * Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- *
- * THIS SOFTWARE IS PROVIDED ''AS IS'' AND ANY
- * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
- * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL AUTHOR BE LIABLE FOR ANY
- * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
- * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
- * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
- * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
-
- #include "printf.h"
- #include "str_util.h"
- #include "contrib/fpconv/fpconv.h"
-
- /**
- * From FreeBSD libutil code
- */
- static const int maxscale = 6;
- static const char _hex[] = "0123456789abcdef";
- static const char _HEX[] = "0123456789ABCDEF";
-
- static char *
- rspamd_humanize_number(char *buf, char *last, int64_t num, gboolean bytes)
- {
- const char *prefixes;
- int i, r, remainder, sign;
- int64_t divisor;
- gsize len = last - buf;
-
- remainder = 0;
-
- if (!bytes) {
- divisor = 1000;
- prefixes = "\0\0\0\0k\0\0\0M\0\0\0G\0\0\0T\0\0\0P\0\0\0E";
- }
- else {
- divisor = 1024;
- prefixes = "B\0\0\0KiB\0MiB\0GiB\0TiB\0PiB\0EiB";
- }
-
- #define SCALE2PREFIX(scale) (&prefixes[(scale) *4])
-
- if (num < 0) {
- sign = -1;
- num = -num;
- }
- else {
- sign = 1;
- }
-
- /*
- * Divide the number until it fits the given column.
- * If there will be an overflow by the rounding below,
- * divide once more.
- */
- for (i = 0; i < maxscale && num > divisor; i++) {
- remainder = num % divisor;
- num /= divisor;
- }
-
- if (remainder == 0 || num > divisor / 2) {
- r = rspamd_snprintf(buf, len, "%L%s",
- sign * (num + (remainder + 50) / divisor),
- SCALE2PREFIX(i));
- }
- else {
- /* Floating point version */
- r = rspamd_snprintf(buf, len, "%.2f%s",
- sign * (num + remainder / (double) divisor),
- SCALE2PREFIX(i));
- }
-
- #undef SCALE2PREFIX
-
- return buf + r;
- }
-
-
- static inline unsigned
- rspamd_decimal_digits32(uint32_t val)
- {
- static const uint32_t powers_of_10[] = {
- 0,
- 10,
- 100,
- 1000,
- 10000,
- 100000,
- 1000000,
- 10000000,
- 100000000,
- 1000000000};
- unsigned tmp;
-
- #if defined(_MSC_VER)
- unsigned long r = 0;
- _BitScanReverse(&r, val | 1);
- tmp = (r + 1) * 1233 >> 12;
- #elif defined(__GNUC__) && (__GNUC__ >= 3)
- tmp = (32 - __builtin_clz(val | 1U)) * 1233 >> 12;
-
- #else /* Software version */
- static const unsigned debruijn_tbl[32] = {0, 9, 1, 10, 13, 21, 2, 29,
- 11, 14, 16, 18, 22, 25, 3, 30,
- 8, 12, 20, 28, 15, 17, 24, 7,
- 19, 27, 23, 6, 26, 5, 4, 31};
- uint32_t v = val | 1;
-
- v |= v >> 1;
- v |= v >> 2;
- v |= v >> 4;
- v |= v >> 8;
- v |= v >> 16;
- tmp = (1 + debruijn_tbl[(v * 0x07C4ACDDU) >> 27]) * 1233 >> 12;
- #endif
- return tmp - (val < powers_of_10[tmp]) + 1;
- }
-
- static inline unsigned
- rspamd_decimal_digits64(uint64_t val)
- {
- static const uint64_t powers_of_10[] = {
- 0,
- 10ULL,
- 100ULL,
- 1000ULL,
- 10000ULL,
- 100000ULL,
- 1000000ULL,
- 10000000ULL,
- 100000000ULL,
- 1000000000ULL,
- 10000000000ULL,
- 100000000000ULL,
- 1000000000000ULL,
- 10000000000000ULL,
- 100000000000000ULL,
- 1000000000000000ULL,
- 10000000000000000ULL,
- 100000000000000000ULL,
- 1000000000000000000ULL,
- 10000000000000000000ULL};
- unsigned tmp;
-
- #if defined(_MSC_VER)
- #if _M_IX86
- unsigned long r = 0;
- uint64_t m = val | 1;
- if (_BitScanReverse(&r, m >> 32)) {
- r += 32;
- }
- else {
- _BitScanReverse(&r, m & 0xFFFFFFFF);
- }
- tmp = (r + 1) * 1233 >> 12;
- #else
- unsigned long r = 0;
- _BitScanReverse64(&r, val | 1);
- tmp = (r + 1) * 1233 >> 12;
- #endif
- #elif defined(__GNUC__) && (__GNUC__ >= 3)
- tmp = (64 - __builtin_clzll(val | 1ULL)) * 1233 >> 12;
- #else /* Software version */
- static const unsigned debruijn_tbl[32] = {0, 9, 1, 10, 13, 21, 2, 29,
- 11, 14, 16, 18, 22, 25, 3, 30,
- 8, 12, 20, 28, 15, 17, 24, 7,
- 19, 27, 23, 6, 26, 5, 4, 31};
- uint32_t v = val >> 32;
-
- if (v) {
- v |= 1;
- v |= v >> 1;
- v |= v >> 2;
- v |= v >> 4;
- v |= v >> 8;
- v |= v >> 16;
- tmp = 32 + debruijn_tbl[(v * 0x07C4ACDDU) >> 27];
- }
- else {
- v = val & 0xFFFFFFFF;
- v |= 1;
- v |= v >> 1;
- v |= v >> 2;
- v |= v >> 4;
- v |= v >> 8;
- v |= v >> 16;
-
- tmp = debruijn_tbl[(v * 0x07C4ACDDU) >> 27];
- }
-
-
- tmp = (tmp + 1) * 1233 >> 12;
- #endif
-
- return tmp - (val < powers_of_10[tmp]) + 1;
- }
-
- /*
- * Idea from https://github.com/miloyip/itoa-benchmark:
- * Uses lookup table (LUT) of digit pairs for division/modulo of 100.
- *
- * Mentioned in:
- * https://www.slideshare.net/andreialexandrescu1/three-optimization-tips-for-c-15708507
- */
-
- static const char int_lookup_table[200] = {
- '0', '0', '0', '1', '0', '2', '0', '3', '0', '4',
- '0', '5', '0', '6', '0', '7', '0', '8', '0', '9',
- '1', '0', '1', '1', '1', '2', '1', '3', '1', '4',
- '1', '5', '1', '6', '1', '7', '1', '8', '1', '9',
- '2', '0', '2', '1', '2', '2', '2', '3', '2', '4',
- '2', '5', '2', '6', '2', '7', '2', '8', '2', '9',
- '3', '0', '3', '1', '3', '2', '3', '3', '3', '4',
- '3', '5', '3', '6', '3', '7', '3', '8', '3', '9',
- '4', '0', '4', '1', '4', '2', '4', '3', '4', '4',
- '4', '5', '4', '6', '4', '7', '4', '8', '4', '9',
- '5', '0', '5', '1', '5', '2', '5', '3', '5', '4',
- '5', '5', '5', '6', '5', '7', '5', '8', '5', '9',
- '6', '0', '6', '1', '6', '2', '6', '3', '6', '4',
- '6', '5', '6', '6', '6', '7', '6', '8', '6', '9',
- '7', '0', '7', '1', '7', '2', '7', '3', '7', '4',
- '7', '5', '7', '6', '7', '7', '7', '8', '7', '9',
- '8', '0', '8', '1', '8', '2', '8', '3', '8', '4',
- '8', '5', '8', '6', '8', '7', '8', '8', '8', '9',
- '9', '0', '9', '1', '9', '2', '9', '3', '9', '4',
- '9', '5', '9', '6', '9', '7', '9', '8', '9', '9'};
-
- static inline unsigned int
- rspamd_uint32_print(uint32_t in, char *out)
- {
- unsigned int ndigits = rspamd_decimal_digits32(in);
- char *p;
-
- p = out + ndigits - 1;
-
- while (in >= 100) {
- unsigned idx = (in % 100) * 2;
-
- /* Do two digits at once */
- *p-- = int_lookup_table[idx + 1];
- *p-- = int_lookup_table[idx];
-
- in /= 100;
- }
-
- if (in < 10) {
- *p = ((char) in) + '0';
- }
- else {
- unsigned idx = in * 2;
-
- *p-- = int_lookup_table[idx + 1];
- *p = int_lookup_table[idx];
- }
-
- return ndigits;
- }
-
- static inline unsigned int
- rspamd_uint64_print(uint64_t in, char *out)
- {
- unsigned int ndigits = rspamd_decimal_digits64(in);
- uint32_t v32;
- char *p;
-
- p = out + ndigits - 1;
-
- while (in >= 100000000) {
- v32 = (uint32_t) (in % 100000000);
- uint32_t a, b, a1, a2, b1, b2;
-
- /* Initial spill */
- a = v32 / 10000;
- b = v32 % 10000;
- a1 = (a / 100) * 2;
- a2 = (a % 100) * 2;
- b1 = (b / 100) * 2;
- b2 = (b % 100) * 2;
-
- /* Fill 8 digits at once */
- *p-- = int_lookup_table[b2 + 1];
- *p-- = int_lookup_table[b2];
- *p-- = int_lookup_table[b1 + 1];
- *p-- = int_lookup_table[b1];
- *p-- = int_lookup_table[a2 + 1];
- *p-- = int_lookup_table[a2];
- *p-- = int_lookup_table[a1 + 1];
- *p-- = int_lookup_table[a1];
-
- in /= 100000000;
- }
-
- /* Remaining 32 bit */
- v32 = (uint32_t) in;
-
- while (v32 >= 100) {
- unsigned idx = (v32 % 100) << 1;
-
- /* Do 2 digits at once */
- *p-- = int_lookup_table[idx + 1];
- *p-- = int_lookup_table[idx];
-
- v32 /= 100;
- }
-
- if (v32 < 10) {
- *p = ((char) v32) + '0';
- }
- else {
- unsigned idx = v32 * 2;
-
- *p-- = int_lookup_table[idx + 1];
- *p = int_lookup_table[idx];
- }
-
- return ndigits;
- }
-
- static inline int
- rspamd_ffsll(long long n)
- {
- #ifdef __has_builtin
- #if __has_builtin(__builtin_ffsll)
- return __builtin_ffsll(n);
- #elif __has_builtin(__builtin_ctzll)
- if (n == 0) {
- return 0;
- }
-
- return __builtin_ctzll(n) + 1;
- #endif
- #endif /* __has_builtin */
-
- #ifdef HAVE_FFSL
- return ffsl(n);
- #else
- if (n == 0) {
- return 0;
- }
-
- int bit;
- for (bit = 1; !(n & 1); bit++) {
- n = ((unsigned long long) n) >> 1;
- }
- return bit;
- #endif
- }
-
- static char *
- rspamd_sprintf_num(char *buf, char *last, uint64_t ui64, char zero,
- unsigned int hexadecimal, unsigned int binary, unsigned int width)
- {
- char *p, temp[64];
- size_t len = 0;
-
- if (G_LIKELY(hexadecimal == 0 && binary == 0)) {
- p = temp;
-
- if (ui64 < G_MAXUINT32) {
- len = rspamd_uint32_print((uint32_t) ui64, temp);
- }
- else {
- len = rspamd_uint64_print(ui64, temp);
- }
- }
- else if (hexadecimal == 1) {
- p = temp + sizeof(temp);
- do {
- *--p = _hex[(uint32_t) (ui64 & 0xf)];
- } while (ui64 >>= 4);
-
- len = (temp + sizeof(temp)) - p;
- }
- else if (hexadecimal == 2) { /* hexadecimal == 2 */
- p = temp + sizeof(temp);
- do {
- *--p = _HEX[(uint32_t) (ui64 & 0xf)];
- } while (ui64 >>= 4);
-
- len = (temp + sizeof(temp)) - p;
- }
- else if (binary > 0) {
- int first_bit = MIN(sizeof(temp), rspamd_ffsll(ui64));
-
- p = temp + sizeof(temp);
- for (int i = 0; i <= first_bit; i++, ui64 >>= 1) {
- *--p = '0' + (ui64 & 0x1);
- }
-
- len = (temp + sizeof(temp)) - p;
- }
-
- /* zero or space padding */
-
- if (len < width) {
- width -= len;
-
- while (width-- > 0 && buf < last) {
- *buf++ = zero;
- }
- }
-
- /* number safe copy */
-
- if (buf + len > last) {
- len = last - buf;
- }
-
- return ((char *) memcpy(buf, p, len)) + len;
- }
-
- struct rspamd_printf_char_buf {
- char *begin;
- char *pos;
- glong remain;
- };
-
- static glong
- rspamd_printf_append_char(const char *buf, glong buflen, gpointer ud)
- {
- struct rspamd_printf_char_buf *dst = (struct rspamd_printf_char_buf *) ud;
- glong wr;
-
- if (dst->remain <= 0) {
- return dst->remain;
- }
-
- wr = MIN(dst->remain, buflen);
- memcpy(dst->pos, buf, wr);
- dst->remain -= wr;
- dst->pos += wr;
-
- return wr;
- }
-
- static glong
- rspamd_printf_append_file(const char *buf, glong buflen, gpointer ud)
- {
- FILE *dst = (FILE *) ud;
- if (buflen > 0) {
- return fwrite(buf, 1, buflen, dst);
- }
- else {
- return 0;
- }
- }
-
- static glong
- rspamd_printf_append_gstring(const char *buf, glong buflen, gpointer ud)
- {
- GString *dst = (GString *) ud;
-
- if (buflen > 0) {
- g_string_append_len(dst, buf, buflen);
- }
-
- return buflen;
- }
-
- static glong
- rspamd_printf_append_fstring(const char *buf, glong buflen, gpointer ud)
- {
- rspamd_fstring_t **dst = ud;
-
- if (buflen > 0) {
- *dst = rspamd_fstring_append(*dst, buf, buflen);
- }
-
- return buflen;
- }
-
- glong rspamd_fprintf(FILE *f, const char *fmt, ...)
- {
- va_list args;
- glong r;
-
- va_start(args, fmt);
- r = rspamd_vprintf_common(rspamd_printf_append_file, f, fmt, args);
- va_end(args);
-
- return r;
- }
-
- glong rspamd_printf(const char *fmt, ...)
- {
- va_list args;
- glong r;
-
- va_start(args, fmt);
- r = rspamd_vprintf_common(rspamd_printf_append_file, stdout, fmt, args);
- va_end(args);
-
- return r;
- }
-
- glong rspamd_log_fprintf(FILE *f, const char *fmt, ...)
- {
- va_list args;
- glong r;
-
- va_start(args, fmt);
- r = rspamd_vprintf_common(rspamd_printf_append_file, f, fmt, args);
- va_end(args);
-
- fflush(f);
-
- return r;
- }
-
-
- glong rspamd_snprintf(char *buf, glong max, const char *fmt, ...)
- {
- char *r;
- va_list args;
-
- va_start(args, fmt);
- r = rspamd_vsnprintf(buf, max, fmt, args);
- va_end(args);
-
- return (r - buf);
- }
-
- char *
- rspamd_vsnprintf(char *buf, glong max, const char *fmt, va_list args)
- {
- struct rspamd_printf_char_buf dst;
-
- dst.begin = buf;
- dst.pos = dst.begin;
- dst.remain = max - 1;
- (void) rspamd_vprintf_common(rspamd_printf_append_char, &dst, fmt, args);
- *dst.pos = '\0';
-
- return dst.pos;
- }
-
- glong rspamd_printf_gstring(GString *s, const char *fmt, ...)
- {
- va_list args;
- glong r;
-
- va_start(args, fmt);
- r = rspamd_vprintf_gstring(s, fmt, args);
- va_end(args);
-
- return r;
- }
-
- glong rspamd_vprintf_gstring(GString *s, const char *fmt, va_list args)
- {
- return rspamd_vprintf_common(rspamd_printf_append_gstring, s, fmt, args);
- }
-
- glong rspamd_printf_fstring(rspamd_fstring_t **s, const char *fmt, ...)
- {
- va_list args;
- glong r;
-
- va_start(args, fmt);
- r = rspamd_vprintf_fstring(s, fmt, args);
- va_end(args);
-
- return r;
- }
-
- glong rspamd_vprintf_fstring(rspamd_fstring_t **s, const char *fmt, va_list args)
- {
- return rspamd_vprintf_common(rspamd_printf_append_fstring, s, fmt, args);
- }
-
- #define RSPAMD_PRINTF_APPEND(buf, len) \
- do { \
- RSPAMD_PRINTF_APPEND_BUF(buf, len); \
- fmt++; \
- buf_start = fmt; \
- } while (0)
-
- #define RSPAMD_PRINTF_APPEND_BUF(buf, len) \
- do { \
- wr = func((buf), (len), apd); \
- if (wr < (__typeof(wr)) (len)) { \
- goto oob; \
- } \
- written += wr; \
- } while (0)
-
- glong rspamd_vprintf_common(rspamd_printf_append_func func,
- gpointer apd,
- const char *fmt,
- va_list args)
- {
- char zero, numbuf[G_ASCII_DTOSTR_BUF_SIZE], dtoabuf[32], *p, *last;
- unsigned char c;
- const char *buf_start = fmt;
- int d;
- double f;
- glong written = 0, wr, slen;
- int64_t i64;
- uint64_t ui64;
- unsigned int width, sign, hex, humanize, bytes, frac_width, b32, b64;
- rspamd_fstring_t *v;
- rspamd_ftok_t *tok;
- GString *gs;
- GError *err;
-
- while (*fmt) {
-
- /*
- * "buf < last" means that we could copy at least one character:
- * the plain character, "%%", "%c", and minus without the checking
- */
-
- if (*fmt == '%') {
-
- /* Append what we have in buf */
- if (fmt > buf_start) {
- wr = func(buf_start, fmt - buf_start, apd);
- if (wr <= 0) {
- goto oob;
- }
- written += wr;
- }
-
- i64 = 0;
- ui64 = 0;
-
- zero = (char) ((*++fmt == '0') ? '0' : ' ');
- width = 0;
- sign = 1;
- hex = 0;
- b32 = 0;
- b64 = 0;
- bytes = 0;
- humanize = 0;
- frac_width = 0;
- slen = -1;
-
- while (*fmt >= '0' && *fmt <= '9') {
- width = width * 10 + *fmt++ - '0';
- }
-
-
- for (;;) {
- switch (*fmt) {
-
- case 'u':
- sign = 0;
- fmt++;
- continue;
-
- case 'm':
- fmt++;
- continue;
-
- case 'X':
- hex = 2;
- sign = 0;
- fmt++;
- continue;
-
- case 'x':
- hex = 1;
- sign = 0;
- fmt++;
- continue;
- case 'b':
- b32 = 1;
- sign = 0;
- fmt++;
- continue;
- case 'B':
- b64 = 1;
- sign = 0;
- fmt++;
- continue;
- case 'H':
- humanize = 1;
- bytes = 1;
- sign = 0;
- fmt++;
- continue;
- case 'h':
- humanize = 1;
- sign = 0;
- fmt++;
- continue;
- case '.':
- fmt++;
-
- if (*fmt == '*') {
- d = (int) va_arg(args, int);
- if (G_UNLIKELY(d < 0)) {
- return 0;
- }
- frac_width = (unsigned int) d;
- fmt++;
- }
- else {
- while (*fmt >= '0' && *fmt <= '9') {
- frac_width = frac_width * 10 + *fmt++ - '0';
- }
- }
-
- break;
-
- case '*':
- d = (int) va_arg(args, int);
- if (G_UNLIKELY(d < 0)) {
- return 0;
- }
- slen = (glong) d;
- fmt++;
- continue;
-
- default:
- break;
- }
-
- break;
- }
-
-
- switch (*fmt) {
-
- case 'V':
- v = va_arg(args, rspamd_fstring_t *);
-
- if (v) {
- slen = v->len;
-
- if (G_UNLIKELY(width != 0)) {
- slen = MIN(v->len, width);
- }
-
- RSPAMD_PRINTF_APPEND(v->str, slen);
- }
- else {
- RSPAMD_PRINTF_APPEND("(NULL)", 6);
- }
-
- continue;
-
- case 'T':
- tok = va_arg(args, rspamd_ftok_t *);
-
- if (tok) {
- slen = tok->len;
-
- if (G_UNLIKELY(width != 0)) {
- slen = MIN(tok->len, width);
- }
- RSPAMD_PRINTF_APPEND(tok->begin, slen);
- }
- else {
- RSPAMD_PRINTF_APPEND("(NULL)", 6);
- }
- continue;
-
- case 'v':
- gs = va_arg(args, GString *);
-
- if (gs) {
- slen = gs->len;
-
- if (G_UNLIKELY(width != 0)) {
- slen = MIN(gs->len, width);
- }
-
- RSPAMD_PRINTF_APPEND(gs->str, slen);
- }
- else {
- RSPAMD_PRINTF_APPEND("(NULL)", 6);
- }
-
- continue;
-
- case 'e':
- err = va_arg(args, GError *);
-
- if (err) {
- p = err->message;
-
- if (p == NULL) {
- p = "(NULL)";
- }
- }
- else {
- p = "unknown error";
- }
-
- slen = strlen(p);
- RSPAMD_PRINTF_APPEND(p, slen);
-
- continue;
-
- case 's':
- p = va_arg(args, char *);
- if (p == NULL) {
- p = "(NULL)";
- slen = sizeof("(NULL)") - 1;
- }
-
- if (G_UNLIKELY(b32)) {
- char *b32buf;
-
- if (G_UNLIKELY(slen == -1)) {
- if (G_LIKELY(width != 0)) {
- slen = width;
- }
- else {
- /* NULL terminated string */
- slen = strlen(p);
- }
- }
-
- b32buf = rspamd_encode_base32(p, slen, RSPAMD_BASE32_DEFAULT);
-
- if (b32buf) {
- RSPAMD_PRINTF_APPEND(b32buf, strlen(b32buf));
- g_free(b32buf);
- }
- else {
- RSPAMD_PRINTF_APPEND("(NULL)", sizeof("(NULL)") - 1);
- }
- }
- else if (G_UNLIKELY(hex)) {
- char hexbuf[2];
-
- if (G_UNLIKELY(slen == -1)) {
- if (G_LIKELY(width != 0)) {
- slen = width;
- }
- else {
- /* NULL terminated string */
- slen = strlen(p);
- }
- }
-
- while (slen) {
- hexbuf[0] = hex == 2 ? _HEX[(*p >> 4u) & 0xfu] : _hex[(*p >> 4u) & 0xfu];
- hexbuf[1] = hex == 2 ? _HEX[*p & 0xfu] : _hex[*p & 0xfu];
- RSPAMD_PRINTF_APPEND_BUF(hexbuf, 2);
- p++;
- slen--;
- }
-
- fmt++;
- buf_start = fmt;
- }
- else if (G_UNLIKELY(b64)) {
- char *b64buf;
- gsize olen = 0;
-
- if (G_UNLIKELY(slen == -1)) {
- if (G_LIKELY(width != 0)) {
- slen = width;
- }
- else {
- /* NULL terminated string */
- slen = strlen(p);
- }
- }
-
- b64buf = rspamd_encode_base64(p, slen, 0, &olen);
-
- if (b64buf) {
- RSPAMD_PRINTF_APPEND(b64buf, olen);
- g_free(b64buf);
- }
- else {
- RSPAMD_PRINTF_APPEND("(NULL)", sizeof("(NULL)") - 1);
- }
- }
- else {
- if (slen == -1) {
- /* NULL terminated string */
- slen = strlen(p);
- }
-
- if (G_UNLIKELY(width != 0)) {
- slen = MIN(slen, width);
- }
-
- RSPAMD_PRINTF_APPEND(p, slen);
- }
-
- continue;
-
- case 'O':
- i64 = (int64_t) va_arg(args, off_t);
- sign = 1;
- break;
-
- case 'P':
- i64 = (int64_t) va_arg(args, pid_t);
- sign = 1;
- break;
-
- case 't':
- i64 = (int64_t) va_arg(args, time_t);
- sign = 1;
- break;
-
- case 'z':
- if (sign) {
- i64 = (int64_t) va_arg(args, ssize_t);
- }
- else {
- ui64 = (uint64_t) va_arg(args, size_t);
- }
- break;
-
- case 'd':
- if (sign) {
- i64 = (int64_t) va_arg(args, int);
- }
- else {
- ui64 = (uint64_t) va_arg(args, unsigned int);
- }
- break;
-
- case 'l':
- if (sign) {
- i64 = (int64_t) va_arg(args, glong);
- }
- else {
- ui64 = (uint64_t) va_arg(args, gulong);
- }
- break;
-
- case 'D':
- if (sign) {
- i64 = (int64_t) va_arg(args, int32_t);
- }
- else {
- ui64 = (uint64_t) va_arg(args, uint32_t);
- }
- break;
-
- case 'L':
- if (sign) {
- i64 = va_arg(args, int64_t);
- }
- else {
- ui64 = va_arg(args, uint64_t);
- }
- break;
-
-
- case 'f':
- f = (double) va_arg(args, double);
- slen = fpconv_dtoa(f, dtoabuf, frac_width, false);
-
- RSPAMD_PRINTF_APPEND(dtoabuf, slen);
-
- continue;
-
- case 'g':
- f = (double) va_arg(args, double);
- slen = fpconv_dtoa(f, dtoabuf, 0, true);
- RSPAMD_PRINTF_APPEND(dtoabuf, slen);
-
- continue;
-
- case 'F':
- f = (double) va_arg(args, long double);
- slen = fpconv_dtoa(f, dtoabuf, frac_width, false);
-
- RSPAMD_PRINTF_APPEND(dtoabuf, slen);
-
- continue;
-
- case 'G':
- f = (double) va_arg(args, long double);
- slen = fpconv_dtoa(f, dtoabuf, 0, true);
- RSPAMD_PRINTF_APPEND(dtoabuf, slen);
-
- continue;
-
- case 'p':
- ui64 = (uintptr_t) va_arg(args, void *);
- hex = 2;
- sign = 0;
- zero = '0';
- width = sizeof(void *) * 2;
- break;
-
- case 'c':
- c = va_arg(args, int);
- c &= 0xffu;
- if (G_UNLIKELY(hex)) {
- char hexbuf[2];
- hexbuf[0] = hex == 2 ? _HEX[(c >> 4u) & 0xfu] : _hex[(c >> 4u) & 0xfu];
- hexbuf[1] = hex == 2 ? _HEX[c & 0xfu] : _hex[c & 0xfu];
-
- RSPAMD_PRINTF_APPEND(hexbuf, 2);
- }
- else {
- RSPAMD_PRINTF_APPEND(&c, 1);
- }
-
- continue;
-
- case 'Z':
- c = '\0';
- RSPAMD_PRINTF_APPEND(&c, 1);
-
- continue;
-
- case 'N':
- c = '\n';
- RSPAMD_PRINTF_APPEND(&c, 1);
-
- continue;
-
- case '%':
- c = '%';
- RSPAMD_PRINTF_APPEND(&c, 1);
-
- continue;
-
- default:
- c = *fmt;
- RSPAMD_PRINTF_APPEND(&c, 1);
-
- continue;
- }
-
- /* Print number */
- p = numbuf;
- last = p + sizeof(numbuf);
- if (sign) {
- if (i64 < 0) {
- *p++ = '-';
- ui64 = (uint64_t) -i64;
- }
- else {
- ui64 = (uint64_t) i64;
- }
- }
-
- if (!humanize) {
- p = rspamd_sprintf_num(p, last, ui64, zero, hex, b64 + b32, width);
- }
- else {
- p = rspamd_humanize_number(p, last, ui64, bytes);
- }
- slen = p - numbuf;
- RSPAMD_PRINTF_APPEND(numbuf, slen);
- }
- else {
- fmt++;
- }
- }
-
- /* Finish buffer */
- if (fmt > buf_start) {
- wr = func(buf_start, fmt - buf_start, apd);
- if (wr <= 0) {
- goto oob;
- }
- written += wr;
- }
-
- oob:
- return written;
- }
|