forked from mirrors/linux
a3fdf78478
Re-work string_stream so that it is not tied to a struct kunit. This is
to allow using it for the log of struct kunit_suite.
Instead of resource-managing individual allocations the whole string_stream
can be resource-managed, if required.
alloc_string_stream() now allocates a string stream that is
not resource-managed.
string_stream_destroy() now works on an unmanaged string_stream
allocated by alloc_string_stream() and frees the entire
string_stream (previously it only freed the fragments).
string_stream_clear() has been made public for callers that
want to free the fragments without destroying the string_stream.
For resource-managed allocations use kunit_alloc_string_stream()
and kunit_free_string_stream().
In addition to this, string_stream_get_string() now returns an
unmanaged buffer that the caller must kfree().
Signed-off-by: Richard Fitzgerald <rf@opensource.cirrus.com>
Reviewed-by: David Gow <davidgow@google.com>
Signed-off-by: Shuah Khan <skhan@linuxfoundation.org>
62 lines
1.6 KiB
C
62 lines
1.6 KiB
C
/* SPDX-License-Identifier: GPL-2.0 */
|
|
/*
|
|
* C++ stream style string builder used in KUnit for building messages.
|
|
*
|
|
* Copyright (C) 2019, Google LLC.
|
|
* Author: Brendan Higgins <brendanhiggins@google.com>
|
|
*/
|
|
|
|
#ifndef _KUNIT_STRING_STREAM_H
|
|
#define _KUNIT_STRING_STREAM_H
|
|
|
|
#include <linux/spinlock.h>
|
|
#include <linux/types.h>
|
|
#include <linux/stdarg.h>
|
|
|
|
struct string_stream_fragment {
|
|
struct list_head node;
|
|
char *fragment;
|
|
};
|
|
|
|
struct string_stream {
|
|
size_t length;
|
|
struct list_head fragments;
|
|
/* length and fragments are protected by this lock */
|
|
spinlock_t lock;
|
|
gfp_t gfp;
|
|
bool append_newlines;
|
|
};
|
|
|
|
struct kunit;
|
|
|
|
struct string_stream *kunit_alloc_string_stream(struct kunit *test, gfp_t gfp);
|
|
void kunit_free_string_stream(struct kunit *test, struct string_stream *stream);
|
|
|
|
struct string_stream *alloc_string_stream(gfp_t gfp);
|
|
void free_string_stream(struct string_stream *stream);
|
|
|
|
int __printf(2, 3) string_stream_add(struct string_stream *stream,
|
|
const char *fmt, ...);
|
|
|
|
int __printf(2, 0) string_stream_vadd(struct string_stream *stream,
|
|
const char *fmt,
|
|
va_list args);
|
|
|
|
void string_stream_clear(struct string_stream *stream);
|
|
|
|
char *string_stream_get_string(struct string_stream *stream);
|
|
|
|
int string_stream_append(struct string_stream *stream,
|
|
struct string_stream *other);
|
|
|
|
bool string_stream_is_empty(struct string_stream *stream);
|
|
|
|
void string_stream_destroy(struct string_stream *stream);
|
|
|
|
static inline void string_stream_set_append_newlines(struct string_stream *stream,
|
|
bool append_newlines)
|
|
{
|
|
stream->append_newlines = append_newlines;
|
|
}
|
|
|
|
#endif /* _KUNIT_STRING_STREAM_H */
|