1#include <assert.h>
2#include <stddef.h>
3#include <stdio.h>
4#include <stdlib.h>
5#include <string.h>
6#include <limits.h>
7
8#include "debug.h"
9#include "printbuf.h"
10
11static void test_basic_printbuf_memset(void);
12static void test_printbuf_memset_length(void);
13
14static void test_basic_printbuf_memset()
15{
16	struct printbuf *pb;
17
18	printf("%s: starting test\n", __func__);
19	pb = printbuf_new();
20	sprintbuf(pb, "blue:%d", 1);
21	printbuf_memset(pb, -1, 'x', 52);
22	printf("Buffer contents:%.*s\n", printbuf_length(pb), pb->buf);
23	printbuf_free(pb);
24	printf("%s: end test\n", __func__);
25}
26
27static void test_printbuf_memset_length()
28{
29	struct printbuf *pb;
30
31	printf("%s: starting test\n", __func__);
32	pb = printbuf_new();
33	printbuf_memset(pb, -1, ' ', 0);
34	printbuf_memset(pb, -1, ' ', 0);
35	printbuf_memset(pb, -1, ' ', 0);
36	printbuf_memset(pb, -1, ' ', 0);
37	printbuf_memset(pb, -1, ' ', 0);
38	printf("Buffer length: %d\n", printbuf_length(pb));
39	printbuf_memset(pb, -1, ' ', 2);
40	printbuf_memset(pb, -1, ' ', 4);
41	printbuf_memset(pb, -1, ' ', 6);
42	printf("Buffer length: %d\n", printbuf_length(pb));
43	printbuf_memset(pb, -1, ' ', 6);
44	printf("Buffer length: %d\n", printbuf_length(pb));
45	printbuf_memset(pb, -1, ' ', 8);
46	printbuf_memset(pb, -1, ' ', 10);
47	printbuf_memset(pb, -1, ' ', 10);
48	printbuf_memset(pb, -1, ' ', 10);
49	printbuf_memset(pb, -1, ' ', 20);
50	printf("Buffer length: %d\n", printbuf_length(pb));
51
52	// No length change should occur
53	printbuf_memset(pb, 0, 'x', 30);
54	printf("Buffer length: %d\n", printbuf_length(pb));
55
56	// This should extend it by one.
57	printbuf_memset(pb, 0, 'x', printbuf_length(pb) + 1);
58	printf("Buffer length: %d\n", printbuf_length(pb));
59
60	printbuf_free(pb);
61	printf("%s: end test\n", __func__);
62}
63
64static void test_printbuf_memappend(int *before_resize);
65static void test_printbuf_memappend(int *before_resize)
66{
67	struct printbuf *pb;
68	int initial_size;
69
70	printf("%s: starting test\n", __func__);
71	pb = printbuf_new();
72	printf("Buffer length: %d\n", printbuf_length(pb));
73
74	initial_size = pb->size;
75
76	while(pb->size == initial_size)
77	{
78		printbuf_memappend_fast(pb, "x", 1);
79	}
80	*before_resize = printbuf_length(pb) - 1;
81	printf("Appended %d bytes for resize: [%s]\n", *before_resize + 1, pb->buf);
82
83	printbuf_reset(pb);
84	printbuf_memappend_fast(pb, "bluexyz123", 3);
85	printf("Partial append: %d, [%s]\n", printbuf_length(pb), pb->buf);
86
87	char with_nulls[] = { 'a', 'b', '\0', 'c' };
88	printbuf_reset(pb);
89	printbuf_memappend_fast(pb, with_nulls, (int)sizeof(with_nulls));
90	printf("With embedded \\0 character: %d, [%s]\n", printbuf_length(pb), pb->buf);
91
92	printbuf_free(pb);
93	pb = printbuf_new();
94	char *data = malloc(*before_resize);
95	memset(data, 'X', *before_resize);
96	printbuf_memappend_fast(pb, data, *before_resize);
97	printf("Append to just before resize: %d, [%s]\n", printbuf_length(pb), pb->buf);
98
99	free(data);
100	printbuf_free(pb);
101
102	pb = printbuf_new();
103	data = malloc(*before_resize + 1);
104	memset(data, 'X', *before_resize + 1);
105	printbuf_memappend_fast(pb, data, *before_resize + 1);
106	printf("Append to just after resize: %d, [%s]\n", printbuf_length(pb), pb->buf);
107
108	free(data);
109
110	printbuf_free(pb);
111	printf("%s: end test\n", __func__);
112}
113
114static void test_sprintbuf(int before_resize);
115static void test_sprintbuf(int before_resize)
116{
117	struct printbuf *pb;
118
119	printf("%s: starting test\n", __func__);
120	pb = printbuf_new();
121	printf("Buffer length: %d\n", printbuf_length(pb));
122
123	char *data = malloc(before_resize + 1 + 1);
124	memset(data, 'X', before_resize + 1 + 1);
125	data[before_resize + 1] = '\0';
126	sprintbuf(pb, "%s", data);
127	free(data);
128	printf("sprintbuf to just after resize(%d+1): %d, [%s], strlen(buf)=%d\n", before_resize, printbuf_length(pb), pb->buf, (int)strlen(pb->buf));
129
130	printbuf_reset(pb);
131	sprintbuf(pb, "plain");
132	printf("%d, [%s]\n", printbuf_length(pb), pb->buf);
133
134	sprintbuf(pb, "%d", 1);
135	printf("%d, [%s]\n", printbuf_length(pb), pb->buf);
136
137	sprintbuf(pb, "%d", INT_MAX);
138	printf("%d, [%s]\n", printbuf_length(pb), pb->buf);
139
140	sprintbuf(pb, "%d", INT_MIN);
141	printf("%d, [%s]\n", printbuf_length(pb), pb->buf);
142
143	sprintbuf(pb, "%s", "%s");
144	printf("%d, [%s]\n", printbuf_length(pb), pb->buf);
145
146	printbuf_free(pb);
147	printf("%s: end test\n", __func__);
148}
149
150int main(int argc, char **argv)
151{
152	int before_resize = 0;
153
154	mc_set_debug(1);
155
156	test_basic_printbuf_memset();
157	printf("========================================\n");
158	test_printbuf_memset_length();
159	printf("========================================\n");
160	test_printbuf_memappend(&before_resize);
161	printf("========================================\n");
162	test_sprintbuf(before_resize);
163	printf("========================================\n");
164
165	return 0;
166}
167