mirror of
https://github.com/apernet/hysteria.git
synced 2025-04-03 04:27:39 +03:00
45 lines
1 KiB
Go
45 lines
1 KiB
Go
package obfs
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func BenchmarkSalamanderObfuscator_Obfuscate(b *testing.B) {
|
|
o, _ := NewSalamanderObfuscator([]byte("average_password"))
|
|
in := make([]byte, 1200)
|
|
_, _ = rand.Read(in)
|
|
out := make([]byte, 2048)
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
o.Obfuscate(in, out)
|
|
}
|
|
}
|
|
|
|
func BenchmarkSalamanderObfuscator_Deobfuscate(b *testing.B) {
|
|
o, _ := NewSalamanderObfuscator([]byte("average_password"))
|
|
in := make([]byte, 1200)
|
|
_, _ = rand.Read(in)
|
|
out := make([]byte, 2048)
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
o.Deobfuscate(in, out)
|
|
}
|
|
}
|
|
|
|
func TestSalamanderObfuscator(t *testing.T) {
|
|
o, _ := NewSalamanderObfuscator([]byte("average_password"))
|
|
in := make([]byte, 1200)
|
|
oOut := make([]byte, 2048)
|
|
dOut := make([]byte, 2048)
|
|
for i := 0; i < 1000; i++ {
|
|
_, _ = rand.Read(in)
|
|
n := o.Obfuscate(in, oOut)
|
|
assert.Equal(t, len(in)+smSaltLen, n)
|
|
n = o.Deobfuscate(oOut[:n], dOut)
|
|
assert.Equal(t, len(in), n)
|
|
assert.Equal(t, in, dOut[:n])
|
|
}
|
|
}
|