diff options
Diffstat (limited to 'regress/unittests/sshbuf/test_sshbuf_misc.c')
-rw-r--r-- | regress/unittests/sshbuf/test_sshbuf_misc.c | 138 |
1 files changed, 138 insertions, 0 deletions
diff --git a/regress/unittests/sshbuf/test_sshbuf_misc.c b/regress/unittests/sshbuf/test_sshbuf_misc.c new file mode 100644 index 000000000..f155491a0 --- /dev/null +++ b/regress/unittests/sshbuf/test_sshbuf_misc.c | |||
@@ -0,0 +1,138 @@ | |||
1 | /* $OpenBSD: test_sshbuf_misc.c,v 1.1 2014/04/30 05:32:00 djm Exp $ */ | ||
2 | /* | ||
3 | * Regress test for sshbuf.h buffer API | ||
4 | * | ||
5 | * Placed in the public domain | ||
6 | */ | ||
7 | |||
8 | #include "includes.h" | ||
9 | |||
10 | #include <sys/types.h> | ||
11 | #include <sys/param.h> | ||
12 | #include <stdio.h> | ||
13 | #ifdef HAVE_STDINT_H | ||
14 | # include <stdint.h> | ||
15 | #endif | ||
16 | #include <stdlib.h> | ||
17 | #include <string.h> | ||
18 | |||
19 | #include "../test_helper/test_helper.h" | ||
20 | |||
21 | #include "sshbuf.h" | ||
22 | |||
23 | void sshbuf_misc_tests(void); | ||
24 | |||
25 | void | ||
26 | sshbuf_misc_tests(void) | ||
27 | { | ||
28 | struct sshbuf *p1; | ||
29 | char tmp[512], *p; | ||
30 | FILE *out; | ||
31 | size_t sz; | ||
32 | |||
33 | TEST_START("sshbuf_dump"); | ||
34 | out = tmpfile(); | ||
35 | ASSERT_PTR_NE(out, NULL); | ||
36 | p1 = sshbuf_new(); | ||
37 | ASSERT_PTR_NE(p1, NULL); | ||
38 | ASSERT_INT_EQ(sshbuf_put_u32(p1, 0x12345678), 0); | ||
39 | sshbuf_dump(p1, out); | ||
40 | fflush(out); | ||
41 | rewind(out); | ||
42 | sz = fread(tmp, 1, sizeof(tmp), out); | ||
43 | ASSERT_INT_EQ(ferror(out), 0); | ||
44 | ASSERT_INT_NE(feof(out), 0); | ||
45 | ASSERT_SIZE_T_GT(sz, 0); | ||
46 | tmp[sz] = '\0'; | ||
47 | ASSERT_PTR_NE(strstr(tmp, "12 34 56 78"), NULL); | ||
48 | fclose(out); | ||
49 | sshbuf_free(p1); | ||
50 | TEST_DONE(); | ||
51 | |||
52 | TEST_START("sshbuf_dtob16"); | ||
53 | p1 = sshbuf_new(); | ||
54 | ASSERT_PTR_NE(p1, NULL); | ||
55 | ASSERT_INT_EQ(sshbuf_put_u32(p1, 0x12345678), 0); | ||
56 | p = sshbuf_dtob16(p1); | ||
57 | ASSERT_PTR_NE(p, NULL); | ||
58 | ASSERT_STRING_EQ(p, "12345678"); | ||
59 | free(p); | ||
60 | sshbuf_free(p1); | ||
61 | TEST_DONE(); | ||
62 | |||
63 | TEST_START("sshbuf_dtob64 len 1"); | ||
64 | p1 = sshbuf_new(); | ||
65 | ASSERT_PTR_NE(p1, NULL); | ||
66 | ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x11), 0); | ||
67 | p = sshbuf_dtob64(p1); | ||
68 | ASSERT_PTR_NE(p, NULL); | ||
69 | ASSERT_STRING_EQ(p, "EQ=="); | ||
70 | free(p); | ||
71 | sshbuf_free(p1); | ||
72 | TEST_DONE(); | ||
73 | |||
74 | TEST_START("sshbuf_dtob64 len 2"); | ||
75 | p1 = sshbuf_new(); | ||
76 | ASSERT_PTR_NE(p1, NULL); | ||
77 | ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x11), 0); | ||
78 | ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x22), 0); | ||
79 | p = sshbuf_dtob64(p1); | ||
80 | ASSERT_PTR_NE(p, NULL); | ||
81 | ASSERT_STRING_EQ(p, "ESI="); | ||
82 | free(p); | ||
83 | sshbuf_free(p1); | ||
84 | TEST_DONE(); | ||
85 | |||
86 | TEST_START("sshbuf_dtob64 len 3"); | ||
87 | p1 = sshbuf_new(); | ||
88 | ASSERT_PTR_NE(p1, NULL); | ||
89 | ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x11), 0); | ||
90 | ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x22), 0); | ||
91 | ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x33), 0); | ||
92 | p = sshbuf_dtob64(p1); | ||
93 | ASSERT_PTR_NE(p, NULL); | ||
94 | ASSERT_STRING_EQ(p, "ESIz"); | ||
95 | free(p); | ||
96 | sshbuf_free(p1); | ||
97 | TEST_DONE(); | ||
98 | |||
99 | TEST_START("sshbuf_dtob64 len 8191"); | ||
100 | p1 = sshbuf_new(); | ||
101 | ASSERT_PTR_NE(p1, NULL); | ||
102 | ASSERT_INT_EQ(sshbuf_reserve(p1, 8192, NULL), 0); | ||
103 | bzero(sshbuf_mutable_ptr(p1), 8192); | ||
104 | p = sshbuf_dtob64(p1); | ||
105 | ASSERT_PTR_NE(p, NULL); | ||
106 | ASSERT_SIZE_T_EQ(strlen(p), ((8191 + 2) / 3) * 4); | ||
107 | free(p); | ||
108 | sshbuf_free(p1); | ||
109 | TEST_DONE(); | ||
110 | |||
111 | TEST_START("sshbuf_b64tod len 1"); | ||
112 | p1 = sshbuf_new(); | ||
113 | ASSERT_PTR_NE(p1, NULL); | ||
114 | ASSERT_INT_EQ(sshbuf_b64tod(p1, "0A=="), 0); | ||
115 | ASSERT_SIZE_T_EQ(sshbuf_len(p1), 1); | ||
116 | ASSERT_U8_EQ(*sshbuf_ptr(p1), 0xd0); | ||
117 | sshbuf_free(p1); | ||
118 | TEST_DONE(); | ||
119 | |||
120 | TEST_START("sshbuf_b64tod len 2"); | ||
121 | p1 = sshbuf_new(); | ||
122 | ASSERT_PTR_NE(p1, NULL); | ||
123 | ASSERT_INT_EQ(sshbuf_b64tod(p1, "0A8="), 0); | ||
124 | ASSERT_SIZE_T_EQ(sshbuf_len(p1), 2); | ||
125 | ASSERT_U16_EQ(PEEK_U16(sshbuf_ptr(p1)), 0xd00f); | ||
126 | sshbuf_free(p1); | ||
127 | TEST_DONE(); | ||
128 | |||
129 | TEST_START("sshbuf_b64tod len 4"); | ||
130 | p1 = sshbuf_new(); | ||
131 | ASSERT_PTR_NE(p1, NULL); | ||
132 | ASSERT_INT_EQ(sshbuf_b64tod(p1, "0A/QDw=="), 0); | ||
133 | ASSERT_SIZE_T_EQ(sshbuf_len(p1), 4); | ||
134 | ASSERT_U32_EQ(PEEK_U32(sshbuf_ptr(p1)), 0xd00fd00f); | ||
135 | sshbuf_free(p1); | ||
136 | TEST_DONE(); | ||
137 | } | ||
138 | |||