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.
39 lines
883 B
39 lines
883 B
6 years ago
|
package shadowsocks_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/rand"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"v2ray.com/core/common"
|
||
|
|
"v2ray.com/core/common/buf"
|
||
|
|
"v2ray.com/core/common/compare"
|
||
|
|
"v2ray.com/core/proxy/shadowsocks"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAEADCipherUDP(t *testing.T) {
|
||
|
|
rawAccount := &shadowsocks.Account{
|
||
|
|
CipherType: shadowsocks.CipherType_AES_128_GCM,
|
||
|
|
Password: "test",
|
||
|
|
}
|
||
|
|
account, err := rawAccount.AsAccount()
|
||
|
|
common.Must(err)
|
||
|
|
|
||
|
|
cipher := account.(*shadowsocks.MemoryAccount).Cipher
|
||
|
|
|
||
|
|
key := make([]byte, cipher.KeySize())
|
||
|
|
common.Must2(rand.Read(key))
|
||
|
|
|
||
|
|
payload := make([]byte, 1024)
|
||
|
|
common.Must2(rand.Read(payload))
|
||
|
|
|
||
|
|
b1 := buf.New()
|
||
|
|
common.Must2(b1.ReadFullFrom(rand.Reader, cipher.IVSize()))
|
||
|
|
common.Must2(b1.Write(payload))
|
||
|
|
common.Must(cipher.EncodePacket(key, b1))
|
||
|
|
|
||
|
|
common.Must(cipher.DecodePacket(key, b1))
|
||
|
|
if err := compare.BytesEqualWithDetail(b1.Bytes(), payload); err != nil {
|
||
|
|
t.Error(err)
|
||
|
|
}
|
||
|
|
}
|