v2ray-core/common/crypto/benchmark_test.go

51 lines
979 B
Go
Raw Normal View History

2016-02-23 16:33:54 +00:00
package crypto_test
import (
"crypto/cipher"
"testing"
2016-08-20 18:55:45 +00:00
. "v2ray.com/core/common/crypto"
2016-02-23 16:33:54 +00:00
)
const benchSize = 1024 * 1024
func benchmarkStream(b *testing.B, c cipher.Stream) {
b.SetBytes(benchSize)
input := make([]byte, benchSize)
output := make([]byte, benchSize)
2016-08-17 08:28:30 +00:00
b.ResetTimer()
2016-02-23 16:33:54 +00:00
for i := 0; i < b.N; i++ {
c.XORKeyStream(output, input)
}
}
func BenchmarkChaCha20(b *testing.B) {
key := make([]byte, 32)
nonce := make([]byte, 8)
c := NewChaCha20Stream(key, nonce)
benchmarkStream(b, c)
}
func BenchmarkChaCha20IETF(b *testing.B) {
key := make([]byte, 32)
nonce := make([]byte, 12)
c := NewChaCha20Stream(key, nonce)
benchmarkStream(b, c)
}
func BenchmarkAESEncryption(b *testing.B) {
key := make([]byte, 32)
iv := make([]byte, 16)
2016-02-25 20:50:10 +00:00
c := NewAesEncryptionStream(key, iv)
2016-02-23 16:33:54 +00:00
benchmarkStream(b, c)
}
func BenchmarkAESDecryption(b *testing.B) {
key := make([]byte, 32)
iv := make([]byte, 16)
2016-02-25 20:50:10 +00:00
c := NewAesDecryptionStream(key, iv)
2016-02-23 16:33:54 +00:00
benchmarkStream(b, c)
}