63 lines · c
1/* SPDX-License-Identifier: GPL-2.0 */2/*3 * C++ stream style string builder used in KUnit for building messages.4 *5 * Copyright (C) 2019, Google LLC.6 * Author: Brendan Higgins <brendanhiggins@google.com>7 */8 9#ifndef _KUNIT_STRING_STREAM_H10#define _KUNIT_STRING_STREAM_H11 12#include <linux/spinlock.h>13#include <linux/types.h>14#include <linux/stdarg.h>15 16struct string_stream_fragment {17 struct list_head node;18 char *fragment;19};20 21struct string_stream {22 size_t length;23 struct list_head fragments;24 /* length and fragments are protected by this lock */25 spinlock_t lock;26 gfp_t gfp;27 bool append_newlines;28};29 30struct kunit;31 32struct string_stream *kunit_alloc_string_stream(struct kunit *test, gfp_t gfp);33void kunit_free_string_stream(struct kunit *test, struct string_stream *stream);34 35struct string_stream *alloc_string_stream(gfp_t gfp);36void free_string_stream(struct string_stream *stream);37 38int __printf(2, 3) string_stream_add(struct string_stream *stream,39 const char *fmt, ...);40 41int __printf(2, 0) string_stream_vadd(struct string_stream *stream,42 const char *fmt,43 va_list args);44 45void string_stream_clear(struct string_stream *stream);46 47char *string_stream_get_string(struct string_stream *stream);48 49int string_stream_append(struct string_stream *stream,50 struct string_stream *other);51 52bool string_stream_is_empty(struct string_stream *stream);53 54void string_stream_destroy(struct string_stream *stream);55 56static inline void string_stream_set_append_newlines(struct string_stream *stream,57 bool append_newlines)58{59 stream->append_newlines = append_newlines;60}61 62#endif /* _KUNIT_STRING_STREAM_H */63