summaryrefslogtreecommitdiff
path: root/gpxe/src/core/asprintf.c
blob: 03cf45cfc2617eb7809fdf45fb177ee04f9d76c0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
#include <stdint.h>
#include <stddef.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>

FILE_LICENCE ( GPL2_OR_LATER );

/**
 * Write a formatted string to newly allocated memory.
 *
 * @v strp		Pointer to hold allocated string
 * @v fmt		Format string
 * @v args		Arguments corresponding to the format string
 * @ret	len		Length of formatted string
 */
int vasprintf ( char **strp, const char *fmt, va_list args ) {
	size_t len;
	va_list args_tmp;

	/* Calculate length needed for string */
	va_copy ( args_tmp, args );
	len = ( vsnprintf ( NULL, 0, fmt, args_tmp ) + 1 );
	va_end ( args_tmp );

	/* Allocate and fill string */
	*strp = malloc ( len );
	if ( ! *strp )
		return -ENOMEM;
	return vsnprintf ( *strp, len, fmt, args );
}

/**
 * Write a formatted string to newly allocated memory.
 *
 * @v strp		Pointer to hold allocated string
 * @v fmt		Format string
 * @v ...		Arguments corresponding to the format string
 * @ret	len		Length of formatted string
 */
int asprintf ( char **strp, const char *fmt, ... ) {
	va_list args;
	int len;

	va_start ( args, fmt );
	len = vasprintf ( strp, fmt, args );
	va_end ( args );
	return len;
}