mirror of https://github.com/v2ray/v2ray-core
45 lines
957 B
Go
45 lines
957 B
Go
package mtproto_test
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/rand"
|
|
"testing"
|
|
|
|
"v2ray.com/core/common"
|
|
"v2ray.com/core/common/compare"
|
|
. "v2ray.com/core/proxy/mtproto"
|
|
. "v2ray.com/ext/assert"
|
|
)
|
|
|
|
func TestInverse(t *testing.T) {
|
|
const size = 64
|
|
b := make([]byte, 64)
|
|
for b[0] == b[size-1] {
|
|
common.Must2(rand.Read(b))
|
|
}
|
|
|
|
bi := Inverse(b)
|
|
if b[0] == bi[0] {
|
|
t.Fatal("seems bytes are not inversed: ", b[0], "vs", bi[0])
|
|
}
|
|
|
|
bii := Inverse(bi)
|
|
if err := compare.BytesEqualWithDetail(bii, b); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestAuthenticationReadWrite(t *testing.T) {
|
|
assert := With(t)
|
|
|
|
a := NewAuthentication(DefaultSessionContext())
|
|
b := bytes.NewReader(a.Header[:])
|
|
a2, err := ReadAuthentication(b)
|
|
assert(err, IsNil)
|
|
|
|
assert(a.EncodingKey[:], Equals, a2.DecodingKey[:])
|
|
assert(a.EncodingNonce[:], Equals, a2.DecodingNonce[:])
|
|
assert(a.DecodingKey[:], Equals, a2.EncodingKey[:])
|
|
assert(a.DecodingNonce[:], Equals, a2.EncodingNonce[:])
|
|
}
|