mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
834 B
38 lines
834 B
package kcp_test
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"testing"
|
|
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
|
"github.com/v2ray/v2ray-core/testing/assert"
|
|
. "github.com/v2ray/v2ray-core/transport/internet/kcp"
|
|
)
|
|
|
|
func TestSimpleAuthenticator(t *testing.T) {
|
|
assert := assert.On(t)
|
|
|
|
buffer := alloc.NewBuffer().Clear()
|
|
buffer.AppendBytes('a', 'b', 'c', 'd', 'e', 'f', 'g')
|
|
|
|
auth := NewSimpleAuthenticator()
|
|
auth.Seal(buffer)
|
|
|
|
assert.Bool(auth.Open(buffer)).IsTrue()
|
|
assert.Bytes(buffer.Value).Equals([]byte{'a', 'b', 'c', 'd', 'e', 'f', 'g'})
|
|
}
|
|
|
|
func BenchmarkSimpleAuthenticator(b *testing.B) {
|
|
buffer := alloc.NewBuffer().Clear()
|
|
buffer.Slice(0, 1024)
|
|
rand.Read(buffer.Value)
|
|
|
|
auth := NewSimpleAuthenticator()
|
|
b.SetBytes(int64(buffer.Len()))
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
auth.Seal(buffer)
|
|
auth.Open(buffer)
|
|
}
|
|
}
|