unittests/sshbuf/test_sshbuf.c

33 	struct sshbuf *p1;
40 	p1 = sshbuf_new();
41 	ASSERT_PTR_NE(p1, NULL);
45 	ASSERT_SIZE_T_GT(sshbuf_max_size(p1), 0);
49 	ASSERT_SIZE_T_GT(sshbuf_avail(p1), 0);
53 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 0);
57 	ASSERT_INT_EQ(sshbuf_set_max_size(p1, 65536), 0);
58 	ASSERT_SIZE_T_EQ(sshbuf_max_size(p1), 65536);
62 	ASSERT_SIZE_T_EQ(sshbuf_avail(p1), 65536);
66 	sshbuf_free(p1);
70 	p1 = sshbuf_new();
71 	ASSERT_PTR_NE(p1, NULL);
72 	ASSERT_INT_EQ(sshbuf_consume(p1, 0), 0);
73 	ASSERT_INT_EQ(sshbuf_consume(p1, 1), SSH_ERR_MESSAGE_INCOMPLETE);
74 	sshbuf_free(p1);
78 	p1 = sshbuf_new();
79 	ASSERT_PTR_NE(p1, NULL);
80 	ASSERT_INT_EQ(sshbuf_consume_end(p1, 0), 0);
81 	ASSERT_INT_EQ(sshbuf_consume_end(p1, 1), SSH_ERR_MESSAGE_INCOMPLETE);
82 	sshbuf_free(p1);
86 	p1 = sshbuf_new();
87 	ASSERT_PTR_NE(p1, NULL);
88 	r = sshbuf_reserve(p1, 1, &dp);
92 	r = sshbuf_reserve(p1, 3, &dp);
101 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 4);
105 	cdp = sshbuf_ptr(p1);
114 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 4);
115 	ASSERT_INT_EQ(sshbuf_consume(p1, 0), 0);
116 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 4);
117 	r = sshbuf_consume(p1, 64);
119 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 4);
120 	ASSERT_INT_EQ(sshbuf_consume(p1, 1), 0);
121 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 3);
122 	cdp = sshbuf_ptr(p1);
123 	ASSERT_PTR_NE(p1, NULL);
125 	ASSERT_INT_EQ(sshbuf_consume(p1, 2), 0);
126 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 1);
127 	cdp = sshbuf_ptr(p1);
128 	ASSERT_PTR_NE(p1, NULL);
130 	r = sshbuf_consume(p1, 2);
132 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 1);
133 	ASSERT_INT_EQ(sshbuf_consume(p1, 1), 0);
134 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 0);
135 	r = sshbuf_consume(p1, 1);
137 	sshbuf_free(p1);
141 	p1 = sshbuf_new();
142 	ASSERT_PTR_NE(p1, NULL);
143 	r = sshbuf_reserve(p1, 4, &dp);
150 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 4);
151 	r = sshbuf_consume_end(p1, 5);
153 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 4);
154 	ASSERT_INT_EQ(sshbuf_consume_end(p1, 3), 0);
155 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 1);
156 	cdp = sshbuf_ptr(p1);
159 	r = sshbuf_consume_end(p1, 2);
161 	ASSERT_INT_EQ(sshbuf_consume_end(p1, 1), 0);
162 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 0);
163 	sshbuf_free(p1);
167 	p1 = sshbuf_new();
168 	ASSERT_PTR_NE(p1, NULL);
169 	ASSERT_INT_EQ(sshbuf_set_max_size(p1, 1223), 0);
170 	ASSERT_SIZE_T_EQ(sshbuf_max_size(p1), 1223);
171 	ASSERT_SIZE_T_EQ(sshbuf_avail(p1), 1223);
172 	r = sshbuf_reserve(p1, 1223, &dp);
176 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 1223);
177 	ASSERT_SIZE_T_EQ(sshbuf_avail(p1), 0);
178 	r = sshbuf_reserve(p1, 1, &dp);
184 	ASSERT_INT_EQ(sshbuf_consume(p1, 223), 0);
185 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 1000);
186 	ASSERT_SIZE_T_EQ(sshbuf_avail(p1), 223);
187 	r = sshbuf_reserve(p1, 224, &dp);
190 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 1000);
191 	ASSERT_SIZE_T_EQ(sshbuf_avail(p1), 223);
192 	r = sshbuf_reserve(p1, 223, &dp);
196 	cdp = sshbuf_ptr(p1);
203 	r = sshbuf_set_max_size(p1, 1000);
206 	ASSERT_INT_EQ(sshbuf_set_max_size(p1, sz), 0);
207 	ASSERT_SIZE_T_EQ(sshbuf_max_size(p1), sz);
208 	ASSERT_SIZE_T_EQ(sshbuf_avail(p1), sz - 1223);
209 	ASSERT_INT_EQ(sshbuf_len(p1), 1223);
214 	r = sshbuf_reserve(p1, 1, &dp);
218 	cdp = sshbuf_ptr(p1);
223 	ASSERT_SIZE_T_EQ(sshbuf_alloc(p1) % SSHBUF_SIZE_INC, 0);
224 	sshbuf_free(p1);
228 	p1 = sshbuf_new();
229 	ASSERT_PTR_NE(p1, NULL);
230 	ASSERT_INT_EQ(sshbuf_set_max_size(p1, 1223), 0);
231 	ASSERT_SIZE_T_EQ(sshbuf_max_size(p1), 1223);
232 	r = sshbuf_reserve(p1, 1223, &dp);
236 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 1223);
237 	sshbuf_reset(p1);
238 	ASSERT_SIZE_T_EQ(sshbuf_max_size(p1), 1223);
239 	ASSERT_SIZE_T_EQ(sshbuf_len(p1), 0);
240 	ASSERT_SIZE_T_EQ(sshbuf_avail(p1), 1223);
241 	sshbuf_free(p1);