[thirdparty/systemd.git] / src / basic / stdio-util.h

/* SPDX-License-Identifier: LGPL-2.1-or-later */
#pragma once

#include <printf.h>
#include <stdio.h>

#include "forward.h"

_printf_(3, 4)
static inline char* snprintf_ok(char *buf, size_t len, const char *format, ...) {
        va_list ap;
        int r;

        va_start(ap, format);
        r = vsnprintf(buf, len, format, ap);
        va_end(ap);

        return r >= 0 && (size_t) r < len ? buf : NULL;
}

#define xsprintf(buf, fmt, ...) \
        assert_message_se(snprintf_ok(buf, ELEMENTSOF(buf), fmt, ##__VA_ARGS__), "xsprintf: " #buf "[] must be big enough")

#define VA_FORMAT_ADVANCE(format, ap)                                   \
do {                                                                    \
        int _argtypes[128];                                             \
        size_t _i, _k;                                                  \
        /* See https://github.com/google/sanitizers/issues/992 */       \
        if (HAS_FEATURE_MEMORY_SANITIZER)                               \
                memset(_argtypes, 0, sizeof(_argtypes));                \
        _k = parse_printf_format((format), ELEMENTSOF(_argtypes), _argtypes); \
        assert(_k < ELEMENTSOF(_argtypes));                             \
        for (_i = 0; _i < _k; _i++) {                                   \
                if (_argtypes[_i] & PA_FLAG_PTR)  {                     \
                        (void) va_arg(ap, void*);                       \
                        continue;                                       \
                }                                                       \
                                                                        \
                switch (_argtypes[_i]) {                                \
                case PA_INT:                                            \
                case PA_INT|PA_FLAG_SHORT:                              \
                case PA_CHAR:                                           \
                        (void) va_arg(ap, int);                         \
                        break;                                          \
                case PA_INT|PA_FLAG_LONG:                               \
                        (void) va_arg(ap, long int);                    \
                        break;                                          \
                case PA_INT|PA_FLAG_LONG_LONG:                          \
                        (void) va_arg(ap, long long int);               \
                        break;                                          \
                case PA_WCHAR:                                          \
                        (void) va_arg(ap, wchar_t);                     \
                        break;                                          \
                case PA_WSTRING:                                        \
                case PA_STRING:                                         \
                case PA_POINTER:                                        \
                        (void) va_arg(ap, void*);                       \
                        break;                                          \
                case PA_FLOAT:                                          \
                case PA_DOUBLE:                                         \
                        (void) va_arg(ap, double);                      \
                        break;                                          \
                case PA_DOUBLE|PA_FLAG_LONG_DOUBLE:                     \
                        (void) va_arg(ap, long double);                 \
                        break;                                          \
                default:                                                \
                        assert_not_reached();                           \
                }                                                       \
        }                                                               \
} while (false)
Commit	Line	Data
	1	/* SPDX-License-Identifier: LGPL-2.1-or-later */
	2	#pragma once
	3
	4	#include <printf.h>
	5	#include <stdio.h>
	6
	7	#include "forward.h"
	8
	9	_printf_(3, 4)
	10	static inline char* snprintf_ok(char buf, size_t len, const char format, ...) {
	11	va_list ap;
	12	int r;
	13
	14	va_start(ap, format);
	15	r = vsnprintf(buf, len, format, ap);
	16	va_end(ap);
	17
	18	return r >= 0 && (size_t) r < len ? buf : NULL;
	19	}
	20
	21	#define xsprintf(buf, fmt, ...) \
	22	assert_message_se(snprintf_ok(buf, ELEMENTSOF(buf), fmt, ##__VA_ARGS__), "xsprintf: " #buf "[] must be big enough")
	23
	24	#define VA_FORMAT_ADVANCE(format, ap) \
	25	do { \
	26	int _argtypes[128]; \
	27	size_t _i, _k; \
	28	/* See https://github.com/google/sanitizers/issues/992 */ \
	29	if (HAS_FEATURE_MEMORY_SANITIZER) \
	30	memset(_argtypes, 0, sizeof(_argtypes)); \
	31	_k = parse_printf_format((format), ELEMENTSOF(_argtypes), _argtypes); \
	32	assert(_k < ELEMENTSOF(_argtypes)); \
	33	for (_i = 0; _i < _k; _i++) { \
	34	if (_argtypes[_i] & PA_FLAG_PTR) { \
	35	(void) va_arg(ap, void*); \
	36	continue; \
	37	} \
	38	\
	39	switch (_argtypes[_i]) { \
	40	case PA_INT: \
	41	case PA_INT\|PA_FLAG_SHORT: \
	42	case PA_CHAR: \
	43	(void) va_arg(ap, int); \
	44	break; \
	45	case PA_INT\|PA_FLAG_LONG: \
	46	(void) va_arg(ap, long int); \
	47	break; \
	48	case PA_INT\|PA_FLAG_LONG_LONG: \
	49	(void) va_arg(ap, long long int); \
	50	break; \
	51	case PA_WCHAR: \
	52	(void) va_arg(ap, wchar_t); \
	53	break; \
	54	case PA_WSTRING: \
	55	case PA_STRING: \
	56	case PA_POINTER: \
	57	(void) va_arg(ap, void*); \
	58	break; \
	59	case PA_FLOAT: \
	60	case PA_DOUBLE: \
	61	(void) va_arg(ap, double); \
	62	break; \
	63	case PA_DOUBLE\|PA_FLAG_LONG_DOUBLE: \
	64	(void) va_arg(ap, long double); \
	65	break; \
	66	default: \
	67	assert_not_reached(); \
	68	} \
	69	} \
	70	} while (false)