shkong/MiraiGo
1
0
Fork 0
mirror of https://github.com/Mrs4s/MiraiGo.git synced 2025-05-04 19:17:38 +08:00
Code Issues Projects Releases Wiki Activity

test: unify tea benchmark

Browse Source
This commit is contained in:
wdvxdr 2021-11-06 14:41:47 +08:00
parent ec0075ebba
commit 7e2033fd75
No known key found for this signature in database
GPG Key ID: 703F8C071DE7A1B6
1 changed files with 30 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
binary/tea_test.go
View File

@ -75,52 +75,57 @@ func TestTEA(t *testing.T) {
} }
} }
func BenchmarkTEAen16(b *testing.B) { func benchEncrypt(b *testing.B, data []byte) {
data := make([]byte, 16)
_, err := rand.Read(data) _, err := rand.Read(data)
if err != nil { if err != nil {
panic(err) panic(err)
} }
b.SetBytes(int64(len(data)))
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
testTEA.Encrypt(data) testTEA.Encrypt(data)
} }
} }
func BenchmarkTEAde16(b *testing.B) { func benchDecrypt(b *testing.B, data []byte) {
data := make([]byte, 16)
_, err := rand.Read(data) _, err := rand.Read(data)
if err != nil { if err != nil {
panic(err) panic(err)
} }
data = testTEA.Encrypt(data) data = testTEA.Encrypt(data)
b.SetBytes(int64(len(data)))
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
testTEA.Decrypt(data) testTEA.Decrypt(data)
} }
} }
func BenchmarkTEAen256(b *testing.B) { func BenchmarkTEAen(b *testing.B) {
data := make([]byte, 256) b.Run("16", func(b *testing.B) {
_, err := rand.Read(data) data := make([]byte, 16)
if err != nil { benchEncrypt(b, data)
panic(err) })
} b.Run("256", func(b *testing.B) {
b.ResetTimer() data := make([]byte, 256)
for i := 0; i < b.N; i++ { benchEncrypt(b, data)
testTEA.Encrypt(data) })
} b.Run("4K", func(b *testing.B) {
data := make([]byte, 4096)
benchEncrypt(b, data)
})
} }
func BenchmarkTEAde256(b *testing.B) { func BenchmarkTEAde(b *testing.B) {
data := make([]byte, 256) b.Run("16", func(b *testing.B) {
_, err := rand.Read(data) data := make([]byte, 16)
if err != nil { benchDecrypt(b, data)
panic(err) })
} b.Run("256", func(b *testing.B) {
data = testTEA.Encrypt(data) data := make([]byte, 256)
b.ResetTimer() benchDecrypt(b, data)
for i := 0; i < b.N; i++ { })
testTEA.Decrypt(data) b.Run("4K", func(b *testing.B) {
} data := make([]byte, 4096)
benchDecrypt(b, data)
})
} }