summaryrefslogtreecommitdiffstats
path: root/src/crypto/cipher/fuzz_test.go
blob: ffceeef5f59ce007747d8fac9861cf13006c93ad (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
// Copyright 2021 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

//go:build ppc64le

package cipher_test

import (
	"bytes"
	"crypto/aes"
	"crypto/cipher"
	"crypto/rand"
	"testing"
	"time"
)

var cbcAESFuzzTests = []struct {
	name string
	key  []byte
}{
	{
		"CBC-AES128",
		commonKey128,
	},
	{
		"CBC-AES192",
		commonKey192,
	},
	{
		"CBC-AES256",
		commonKey256,
	},
}

var timeout *time.Timer

const datalen = 1024

func TestFuzz(t *testing.T) {

	for _, ft := range cbcAESFuzzTests {
		c, _ := aes.NewCipher(ft.key)

		cbcAsm := cipher.NewCBCEncrypter(c, commonIV)
		cbcGeneric := cipher.NewCBCGenericEncrypter(c, commonIV)

		if testing.Short() {
			timeout = time.NewTimer(10 * time.Millisecond)
		} else {
			timeout = time.NewTimer(2 * time.Second)
		}

		indata := make([]byte, datalen)
		outgeneric := make([]byte, datalen)
		outdata := make([]byte, datalen)

	fuzzencrypt:
		for {
			select {
			case <-timeout.C:
				break fuzzencrypt
			default:
			}

			rand.Read(indata[:])

			cbcGeneric.CryptBlocks(indata, outgeneric)
			cbcAsm.CryptBlocks(indata, outdata)

			if !bytes.Equal(outdata, outgeneric) {
				t.Fatalf("AES-CBC encryption does not match reference result: %x and %x, please report this error to security@golang.org", outdata, outgeneric)
			}
		}

		cbcAsm = cipher.NewCBCDecrypter(c, commonIV)
		cbcGeneric = cipher.NewCBCGenericDecrypter(c, commonIV)

		if testing.Short() {
			timeout = time.NewTimer(10 * time.Millisecond)
		} else {
			timeout = time.NewTimer(2 * time.Second)
		}

	fuzzdecrypt:
		for {
			select {
			case <-timeout.C:
				break fuzzdecrypt
			default:
			}

			rand.Read(indata[:])

			cbcGeneric.CryptBlocks(indata, outgeneric)
			cbcAsm.CryptBlocks(indata, outdata)

			if !bytes.Equal(outdata, outgeneric) {
				t.Fatalf("AES-CBC decryption does not match reference result: %x and %x, please report this error to security@golang.org", outdata, outgeneric)
			}
		}
	}
}