1 /* 	$OpenBSD: test_sshbuf_getput_fuzz.c,v 1.5 2021/12/14 21:25:27 deraadt Exp $ */
2 /*
3  * Regress test for sshbuf.h buffer API
4  *
5  * Placed in the public domain
6  */
7 
8 #include <sys/types.h>
9 #include <stdio.h>
10 #include <stdint.h>
11 #include <stdlib.h>
12 #include <string.h>
13 
14 #include <openssl/bn.h>
15 #include <openssl/ec.h>
16 #include <openssl/objects.h>
17 
18 #include "test_helper.h"
19 #include "ssherr.h"
20 #include "sshbuf.h"
21 
22 void sshbuf_getput_fuzz_tests(void);
23 
24 static void
25 attempt_parse_blob(u_char *blob, size_t len)
26 {
27 	struct sshbuf *p1;
28 	BIGNUM *bn;
29 	EC_KEY *eck;
30 	u_char *s;
31 	size_t l;
32 	u_int8_t u8;
33 	u_int16_t u16;
34 	u_int32_t u32;
35 	u_int64_t u64;
36 
37 	p1 = sshbuf_new();
38 	ASSERT_PTR_NE(p1, NULL);
39 	ASSERT_INT_EQ(sshbuf_put(p1, blob, len), 0);
40 	sshbuf_get_u8(p1, &u8);
41 	sshbuf_get_u16(p1, &u16);
42 	sshbuf_get_u32(p1, &u32);
43 	sshbuf_get_u64(p1, &u64);
44 	if (sshbuf_get_string(p1, &s, &l) == 0) {
45 		bzero(s, l);
46 		free(s);
47 	}
48 	bn = NULL;
49 	sshbuf_get_bignum2(p1, &bn);
50 	BN_clear_free(bn);
51 	eck = EC_KEY_new_by_curve_name(NID_X9_62_prime256v1);
52 	ASSERT_PTR_NE(eck, NULL);
53 	sshbuf_get_eckey(p1, eck);
54 	EC_KEY_free(eck);
55 	sshbuf_free(p1);
56 }
57 
58 
59 static void
60 onerror(void *fuzz)
61 {
62 	fprintf(stderr, "Failed during fuzz:\n");
63 	fuzz_dump((struct fuzz *)fuzz);
64 }
65 
66 void
67 sshbuf_getput_fuzz_tests(void)
68 {
69 	u_char blob[] = {
70 		/* u8 */
71 		0xd0,
72 		/* u16 */
73 		0xc0, 0xde,
74 		/* u32 */
75 		0xfa, 0xce, 0xde, 0xad,
76 		/* u64 */
77 		0xfe, 0xed, 0xac, 0x1d, 0x1f, 0x1c, 0xbe, 0xef,
78 		/* string */
79 		0x00, 0x00, 0x00, 0x09,
80 		'O', ' ', 'G', 'o', 'r', 'g', 'o', 'n', '!',
81 		/* bignum2 */
82 		0x00, 0x00, 0x00, 0x14,
83 		0x00,
84 		0xf0, 0xe0, 0xd0, 0xc0, 0xb0, 0xa0, 0x90, 0x80,
85 		0x70, 0x60, 0x50, 0x40, 0x30, 0x20, 0x10, 0x00,
86 		0x7f, 0xff, 0x11,
87 		/* EC point (NIST-256 curve) */
88 		0x00, 0x00, 0x00, 0x41,
89 		0x04,
90 		0x0c, 0x82, 0x80, 0x04, 0x83, 0x9d, 0x01, 0x06,
91 		0xaa, 0x59, 0x57, 0x52, 0x16, 0x19, 0x13, 0x57,
92 		0x34, 0xb4, 0x51, 0x45, 0x9d, 0xad, 0xb5, 0x86,
93 		0x67, 0x7e, 0xf9, 0xdf, 0x55, 0x78, 0x49, 0x99,
94 		0x4d, 0x19, 0x6b, 0x50, 0xf0, 0xb4, 0xe9, 0x4b,
95 		0x3c, 0x73, 0xe3, 0xa9, 0xd4, 0xcd, 0x9d, 0xf2,
96 		0xc8, 0xf9, 0xa3, 0x5e, 0x42, 0xbd, 0xd0, 0x47,
97 		0x55, 0x0f, 0x69, 0xd8, 0x0e, 0xc2, 0x3c, 0xd4,
98 	};
99 	struct fuzz *fuzz;
100 	u_int fuzzers = FUZZ_1_BIT_FLIP | FUZZ_2_BIT_FLIP |
101 	    FUZZ_1_BYTE_FLIP | FUZZ_2_BYTE_FLIP |
102 	    FUZZ_TRUNCATE_START | FUZZ_TRUNCATE_END;
103 
104 	if (test_is_fast())
105 		fuzzers &= ~(FUZZ_2_BYTE_FLIP|FUZZ_2_BIT_FLIP);
106 
107 	TEST_START("fuzz blob parsing");
108 	fuzz = fuzz_begin(fuzzers, blob, sizeof(blob));
109 	TEST_ONERROR(onerror, fuzz);
110 	for(; !fuzz_done(fuzz); fuzz_next(fuzz))
111 		attempt_parse_blob(blob, sizeof(blob));
112 	fuzz_cleanup(fuzz);
113 	TEST_DONE();
114 	TEST_ONERROR(NULL, NULL);
115 }
116 
117