summaryrefslogtreecommitdiff
path: root/lib/libutil++/stringf.cc
blob: 8c24167d70ac8648e130a00e7cad4f405538aa52 (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
50
51
52
53
54
55
56
57
/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2025 Chelsio Communications, Inc.
 * Written by: John Baldwin <jhb@FreeBSD.org>
 */

#include <cstdarg>
#include <cstdio>
#include <string>

#include "libutil++.hh"

static int
stringf_write(void *cookie, const char *buf, int len)
{
	std::string *str = reinterpret_cast<std::string *>(cookie);
	try {
		str->append(buf, len);
	} catch (std::bad_alloc) {
		errno = ENOMEM;
		return (-1);
	} catch (std::length_error) {
		errno = EFBIG;
		return (-1);
	}
	return (len);
}

std::string
freebsd::stringf(const char *fmt, va_list ap)
{
	std::string str;
	freebsd::FILE_up fp(fwopen(reinterpret_cast<void *>(&str),
	    stringf_write));

	vfprintf(fp.get(), fmt, ap);

	if (ferror(fp.get()))
		throw std::bad_alloc();
	fp.reset(nullptr);

	return str;
}

std::string
freebsd::stringf(const char *fmt, ...)
{
	std::va_list ap;
	std::string str;

	va_start(ap, fmt);
	str = freebsd::stringf(fmt, ap);
	va_end(ap);

	return str;
}