59 lines
950 B
C
59 lines
950 B
C
#define _PDCLIB_EXTENSIONS
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <stdarg.h>
|
|
|
|
static size_t mcb(void *p, const char *buf, size_t size)
|
|
{
|
|
*(size_t *)p += size;
|
|
return size;
|
|
}
|
|
|
|
static size_t vprintflen(const char *restrict fmt, va_list arg)
|
|
{
|
|
size_t ret = 0;
|
|
_vcbprintf(&ret, mcb, fmt, arg);
|
|
return ret + 1;
|
|
}
|
|
|
|
int asprintf(char **strp, const char *restrict fmt, ...)
|
|
{
|
|
va_list arg, arg2;
|
|
va_start(arg, fmt);
|
|
va_copy(arg2, arg);
|
|
|
|
size_t sz = vprintflen(fmt, arg);
|
|
|
|
*strp = malloc(sz);
|
|
if (!strp)
|
|
{
|
|
va_end(arg);
|
|
va_end(arg2);
|
|
return -1;
|
|
}
|
|
|
|
int ret = vsnprintf(*strp, sz, fmt, arg2);
|
|
va_end(arg);
|
|
va_end(arg2);
|
|
return ret;
|
|
}
|
|
|
|
int vasprintf(char **strp, const char *restrict fmt, va_list arg)
|
|
{
|
|
va_list arg2;
|
|
va_copy(arg2, arg);
|
|
|
|
size_t sz = vprintflen(fmt, arg);
|
|
|
|
*strp = malloc(sz);
|
|
if (!strp)
|
|
{
|
|
va_end(arg2);
|
|
return -1;
|
|
}
|
|
|
|
int ret = vsnprintf(*strp, sz, fmt, arg2);
|
|
va_end(arg2);
|
|
return ret;
|
|
}
|