package crypto_test import ( "crypto/aes" "crypto/cipher" "crypto/rand" "io" "testing" "time" "v2ray.com/core/common/buf" . "v2ray.com/core/common/crypto" "v2ray.com/core/testing/assert" ) func TestAuthenticationReaderWriter(t *testing.T) { assert := assert.On(t) key := make([]byte, 16) rand.Read(key) block, err := aes.NewCipher(key) assert.Error(err).IsNil() aead, err := cipher.NewGCM(block) assert.Error(err).IsNil() payload := make([]byte, 8*1024) rand.Read(payload) cache := buf.NewLocal(16 * 1024) iv := make([]byte, 12) rand.Read(iv) writer := NewAuthenticationWriter(&AEADAuthenticator{ AEAD: aead, NonceGenerator: &StaticBytesGenerator{ Content: iv, }, AdditionalDataGenerator: &NoOpBytesGenerator{}, }, cache, NewShakeUint16Generator([]byte{'a'})) nBytes, err := writer.Write(payload) assert.Error(err).IsNil() assert.Int(nBytes).Equals(len(payload)) assert.Int(cache.Len()).GreaterThan(0) _, err = writer.Write([]byte{}) assert.Error(err).IsNil() reader := NewAuthenticationReader(&AEADAuthenticator{ AEAD: aead, NonceGenerator: &StaticBytesGenerator{ Content: iv, }, AdditionalDataGenerator: &NoOpBytesGenerator{}, }, cache, NewShakeUint16Generator([]byte{'a'})) actualPayload := make([]byte, 16*1024) nBytes, err = reader.Read(actualPayload) assert.Error(err).IsNil() assert.Int(nBytes).Equals(len(payload)) assert.Bytes(actualPayload[:nBytes]).Equals(payload) _, err = reader.Read(actualPayload) assert.Error(err).Equals(io.EOF) } func TestAuthenticationReaderWriterPartial(t *testing.T) { assert := assert.On(t) key := make([]byte, 16) rand.Read(key) block, err := aes.NewCipher(key) assert.Error(err).IsNil() aead, err := cipher.NewGCM(block) assert.Error(err).IsNil() payload := make([]byte, 8*1024) rand.Read(payload) iv := make([]byte, 12) rand.Read(iv) cache := buf.NewLocal(16 * 1024) writer := NewAuthenticationWriter(&AEADAuthenticator{ AEAD: aead, NonceGenerator: &StaticBytesGenerator{ Content: iv, }, AdditionalDataGenerator: &NoOpBytesGenerator{}, }, cache, NewShakeUint16Generator([]byte{'a', 'b'})) writer.Write([]byte{'a', 'b', 'c', 'd'}) nBytes, err := writer.Write(payload) assert.Error(err).IsNil() assert.Int(nBytes).Equals(len(payload)) assert.Int(cache.Len()).GreaterThan(0) _, err = writer.Write([]byte{}) assert.Error(err).IsNil() pr, pw := io.Pipe() go func() { pw.Write(cache.BytesTo(1024)) time.Sleep(time.Second * 2) pw.Write(cache.BytesRange(1024, 2048)) time.Sleep(time.Second * 2) pw.Write(cache.BytesRange(2048, 3072)) time.Sleep(time.Second * 2) pw.Write(cache.BytesFrom(3072)) time.Sleep(time.Second * 2) pw.Close() }() reader := NewAuthenticationReader(&AEADAuthenticator{ AEAD: aead, NonceGenerator: &StaticBytesGenerator{ Content: iv, }, AdditionalDataGenerator: &NoOpBytesGenerator{}, }, pr, NewShakeUint16Generator([]byte{'a', 'b'})) actualPayload := make([]byte, 7*1024) nBytes, err = reader.Read(actualPayload) assert.Error(err).IsNil() assert.Int(nBytes).Equals(4) assert.Bytes(actualPayload[:nBytes]).Equals([]byte{'a', 'b', 'c', 'd'}) nBytes, err = reader.Read(actualPayload) assert.Error(err).IsNil() assert.Int(nBytes).Equals(len(actualPayload)) assert.Bytes(actualPayload[:nBytes]).Equals(payload[:nBytes]) nBytes, err = reader.Read(actualPayload) assert.Error(err).IsNil() assert.Int(nBytes).Equals(len(payload) - len(actualPayload)) assert.Bytes(actualPayload[:nBytes]).Equals(payload[7*1024:]) _, err = reader.Read(actualPayload) assert.Error(err).Equals(io.EOF) }