src/basic/stdio-util.h

   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)