v2ray-core/proxy/mtproto/auth_test.go

package mtproto_test

import (
	"bytes"
	"crypto/rand"
	"testing"

	"github.com/google/go-cmp/cmp"

	"v2ray.com/core/common"
	. "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 r := cmp.Diff(bii, b); r != "" {
		t.Fatal(r)
	}
}

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[:])
}