[PATCH v2 2/7] vsprintf: Add rasprintf(): the reallocing string printf family

[Tobias Waldekranz]

Generates the formatted version of the input and appends it to an
existing dynamically allocated string, growing it as necessary using
realloc().

Signed-off-by: Tobias Waldekranz <tobias at waldekranz.com>
---
 include/stdio.h  | 10 ++++++++++
 include/xfuncs.h |  4 ++++
 lib/vsprintf.c   | 35 +++++++++++++++++++++++++++++++++++
 lib/xfuncs.c     | 24 ++++++++++++++++++++++++
 4 files changed, 73 insertions(+)

diff --git a/include/stdio.h b/include/stdio.h
index cecb97ea1a..f3e430b3b3 100644
--- a/include/stdio.h
+++ b/include/stdio.h
@@ -21,6 +21,8 @@ int vscnprintf(char *buf, size_t size, const char *fmt, va_list args);
 int asprintf(char **strp, const char *fmt, ...) __printf(2, 3);
 char *bvasprintf(const char *fmt, va_list ap);
 int vasprintf(char **strp, const char *fmt, va_list ap);
+int vrasprintf(char **strp, const char *fmt, va_list ap);
+int rasprintf(char **strp, const char *fmt, ...) __printf(2, 3);
 #else
 static inline __printf(2, 3) int asprintf(char **strp, const char *fmt, ...)
 {
@@ -34,6 +36,14 @@ static inline int vasprintf(char **strp, const char *fmt, va_list ap)
 {
 	return -1;
 }
+static inline int vrasprintf(char **strp, const char *fmt, va_list ap)
+{
+	return -1;
+}
+static inline __printf(2, 3) int rasprintf(char **strp, const char *fmt, ...)
+{
+	return -1;
+}
 #endif
 
 #define basprintf xasprintf
diff --git a/include/xfuncs.h b/include/xfuncs.h
index eb4050f489..2cc2048bd5 100644
--- a/include/xfuncs.h
+++ b/include/xfuncs.h
@@ -19,6 +19,8 @@ void* xmemalign(size_t alignment, size_t bytes) __xalloc_size(2);
 void* xmemdup(const void *orig, size_t size) __returns_nonnull;
 char *xasprintf(const char *fmt, ...) __printf(1, 2) __returns_nonnull;
 char *xvasprintf(const char *fmt, va_list ap) __returns_nonnull;
+char *xvrasprintf(char *str, const char *fmt, va_list ap) __returns_nonnull;
+char *xrasprintf(char *str, const char *fmt, ...) __printf(2, 3) __returns_nonnull;
 
 wchar_t *xstrdup_wchar(const wchar_t *src);
 wchar_t *xstrdup_char_to_wchar(const char *src);
@@ -35,6 +37,8 @@ static inline void* xmemalign(size_t alignment, size_t bytes) { BUG(); }
 static inline void* xmemdup(const void *orig, size_t size) { BUG(); }
 static inline __printf(1, 2) char *xasprintf(const char *fmt, ...) { BUG(); }
 static inline char *xvasprintf(const char *fmt, va_list ap) { BUG(); }
+static inline char *xvrasprintf(char *str, const char *fmt, va_list ap) { BUG(); }
+static inline __printf(2, 3) char *xrasprintf(char *str, const char *fmt, ...) { BUG(); }
 
 static inline wchar_t *xstrdup_wchar(const wchar_t *src) { BUG(); }
 static inline wchar_t *xstrdup_char_to_wchar(const char *src) { BUG(); }
diff --git a/lib/vsprintf.c b/lib/vsprintf.c
index 409a8f02de..c7295c1067 100644
--- a/lib/vsprintf.c
+++ b/lib/vsprintf.c
@@ -1005,3 +1005,38 @@ int asprintf(char **strp, const char *fmt, ...)
 	return len;
 }
 EXPORT_SYMBOL(asprintf);
+
+int vrasprintf(char **strp, const char *fmt, va_list ap)
+{
+	int fmtlen, inlen = *strp ? strlen(*strp) : 0;
+	va_list aq;
+	char *p;
+
+	va_copy(aq, ap);
+	fmtlen = vsnprintf(NULL, 0, fmt, aq);
+	va_end(aq);
+
+	p = realloc(*strp, inlen + fmtlen + 1);
+	if (!p)
+		return -1;
+
+	vsnprintf(&p[inlen], fmtlen + 1, fmt, ap);
+
+	*strp = p;
+
+	return inlen + fmtlen;
+}
+EXPORT_SYMBOL(vrasprintf);
+
+int rasprintf(char **strp, const char *fmt, ...)
+{
+	va_list ap;
+	int len;
+
+	va_start(ap, fmt);
+	len = vrasprintf(strp, fmt, ap);
+	va_end(ap);
+
+	return len;
+}
+EXPORT_SYMBOL(rasprintf);
diff --git a/lib/xfuncs.c b/lib/xfuncs.c
index d4beecddf5..4c69451305 100644
--- a/lib/xfuncs.c
+++ b/lib/xfuncs.c
@@ -144,6 +144,30 @@ char *xasprintf(const char *fmt, ...)
 }
 EXPORT_SYMBOL(xasprintf);
 
+char *xvrasprintf(char *str, const char *fmt, va_list ap)
+{
+	int len;
+
+	len = vrasprintf(&str, fmt, ap);
+	if (len < 0)
+		enomem_panic(0);
+
+	return str;
+}
+EXPORT_SYMBOL(xvrasprintf);
+
+char *xrasprintf(char *str, const char *fmt, ...)
+{
+	va_list ap;
+
+	va_start(ap, fmt);
+	str = xvrasprintf(str, fmt, ap);
+	va_end(ap);
+
+	return str;
+}
+EXPORT_SYMBOL(xrasprintf);
+
 wchar_t *xstrdup_wchar(const wchar_t *s)
 {
 	wchar_t *p;
-- 
2.43.0