busybox/libbb/bb_asprintf.c

/* vi: set sw=4 ts=4: */
/*
   Copyright (C) 2002,2005 Vladimir Oleynik <dzo@simtreas.ru>
*/

#include <stdlib.h>
#include <stdio.h>
#include <stdarg.h>
#include "libbb.h"

char *bb_xasprintf(const char *format, ...)
{
	va_list p;
	int r;
	char *string_ptr;

#ifdef HAVE_GNU_EXTENSIONS
	va_start(p, format);
	r = vasprintf(&string_ptr, format, p);
	va_end(p);
#else
	va_start(p, format);
	r = vsnprintf(NULL, 0, format, p);
	va_end(p);
	string_ptr = xmalloc(r+1);
	va_start(p, format);
	r = vsnprintf(string_ptr, r+1, format, p);
	va_end(p);
#endif

	if (r < 0) {
		bb_perror_msg_and_die("bb_xasprintf");
	}
	return string_ptr;
}