76 lines · c
1/* SPDX-License-Identifier: GPL-2.0 */2#ifndef __SUBCMD_UTIL_H3#define __SUBCMD_UTIL_H4 5#include <stdarg.h>6#include <stdlib.h>7#include <stdio.h>8#include <linux/compiler.h>9 10static inline void report(const char *prefix, const char *err, va_list params)11{12 char msg[1024];13 vsnprintf(msg, sizeof(msg), err, params);14 fprintf(stderr, " %s%s\n", prefix, msg);15}16 17static __noreturn inline void die(const char *err, ...)18{19 va_list params;20 21 va_start(params, err);22 report(" Fatal: ", err, params);23 exit(128);24 va_end(params);25}26 27#define zfree(ptr) ({ free(*ptr); *ptr = NULL; })28 29#define alloc_nr(x) (((x)+16)*3/2)30 31/*32 * Realloc the buffer pointed at by variable 'x' so that it can hold33 * at least 'nr' entries; the number of entries currently allocated34 * is 'alloc', using the standard growing factor alloc_nr() macro.35 *36 * DO NOT USE any expression with side-effect for 'x' or 'alloc'.37 */38#define ALLOC_GROW(x, nr, alloc) \39 do { \40 if ((nr) > alloc) { \41 if (alloc_nr(alloc) < (nr)) \42 alloc = (nr); \43 else \44 alloc = alloc_nr(alloc); \45 x = xrealloc((x), alloc * sizeof(*(x))); \46 } \47 } while(0)48 49static inline void *xrealloc(void *ptr, size_t size)50{51 void *ret = realloc(ptr, size);52 if (!ret)53 die("Out of memory, realloc failed");54 return ret;55}56 57#define astrcatf(out, fmt, ...) \58({ \59 char *tmp = *(out); \60 if (asprintf((out), "%s" fmt, tmp ?: "", ## __VA_ARGS__) == -1) \61 die("asprintf failed"); \62 free(tmp); \63})64 65static inline void astrcat(char **out, const char *add)66{67 char *tmp = *out;68 69 if (asprintf(out, "%s%s", tmp ?: "", add) == -1)70 die("asprintf failed");71 72 free(tmp);73}74 75#endif /* __SUBCMD_UTIL_H */76