mirror of
https://github.com/torvalds/linux.git
synced 2024-12-04 01:51:34 +00:00
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>
63 lines
1.6 KiB
C
63 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 */
|