mirror of https://github.com/v2ray/v2ray-core
123 lines
3.5 KiB
Go
123 lines
3.5 KiB
Go
package mux_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
. "v2ray.com/core/app/proxyman/mux"
|
|
"v2ray.com/core/common/buf"
|
|
"v2ray.com/core/common/net"
|
|
"v2ray.com/core/testing/assert"
|
|
"v2ray.com/core/transport/ray"
|
|
)
|
|
|
|
func TestReaderWriter(t *testing.T) {
|
|
assert := assert.On(t)
|
|
|
|
stream := ray.NewStream(context.Background())
|
|
|
|
dest := net.TCPDestination(net.DomainAddress("v2ray.com"), 80)
|
|
writer := NewWriter(1, dest, stream)
|
|
|
|
dest2 := net.TCPDestination(net.LocalHostIP, 443)
|
|
writer2 := NewWriter(2, dest2, stream)
|
|
|
|
dest3 := net.TCPDestination(net.LocalHostIPv6, 18374)
|
|
writer3 := NewWriter(3, dest3, stream)
|
|
|
|
writePayload := func(writer *Writer, payload ...byte) error {
|
|
b := buf.New()
|
|
b.Append(payload)
|
|
return writer.Write(buf.NewMultiBufferValue(b))
|
|
}
|
|
|
|
assert.Error(writePayload(writer, 'a', 'b', 'c', 'd')).IsNil()
|
|
assert.Error(writePayload(writer2)).IsNil()
|
|
|
|
assert.Error(writePayload(writer, 'e', 'f', 'g', 'h')).IsNil()
|
|
assert.Error(writePayload(writer3, 'x')).IsNil()
|
|
|
|
writer.Close()
|
|
writer3.Close()
|
|
|
|
assert.Error(writePayload(writer2, 'y')).IsNil()
|
|
writer2.Close()
|
|
|
|
reader := NewReader(stream)
|
|
meta, err := reader.ReadMetadata()
|
|
assert.Error(err).IsNil()
|
|
assert.Uint16(meta.SessionID).Equals(1)
|
|
assert.Byte(byte(meta.SessionStatus)).Equals(byte(SessionStatusNew))
|
|
assert.Destination(meta.Target).Equals(dest)
|
|
assert.Byte(byte(meta.Option)).Equals(byte(OptionData))
|
|
|
|
data, err := reader.Read()
|
|
assert.Error(err).IsNil()
|
|
assert.Int(len(data)).Equals(1)
|
|
assert.String(data[0].String()).Equals("abcd")
|
|
|
|
meta, err = reader.ReadMetadata()
|
|
assert.Error(err).IsNil()
|
|
assert.Byte(byte(meta.SessionStatus)).Equals(byte(SessionStatusNew))
|
|
assert.Uint16(meta.SessionID).Equals(2)
|
|
assert.Byte(byte(meta.Option)).Equals(0)
|
|
assert.Destination(meta.Target).Equals(dest2)
|
|
|
|
meta, err = reader.ReadMetadata()
|
|
assert.Error(err).IsNil()
|
|
assert.Byte(byte(meta.SessionStatus)).Equals(byte(SessionStatusKeep))
|
|
assert.Uint16(meta.SessionID).Equals(1)
|
|
assert.Byte(byte(meta.Option)).Equals(1)
|
|
|
|
data, err = reader.Read()
|
|
assert.Error(err).IsNil()
|
|
assert.Int(len(data)).Equals(1)
|
|
assert.String(data[0].String()).Equals("efgh")
|
|
|
|
meta, err = reader.ReadMetadata()
|
|
assert.Error(err).IsNil()
|
|
assert.Byte(byte(meta.SessionStatus)).Equals(byte(SessionStatusNew))
|
|
assert.Uint16(meta.SessionID).Equals(3)
|
|
assert.Byte(byte(meta.Option)).Equals(1)
|
|
assert.Destination(meta.Target).Equals(dest3)
|
|
|
|
data, err = reader.Read()
|
|
assert.Error(err).IsNil()
|
|
assert.Int(len(data)).Equals(1)
|
|
assert.String(data[0].String()).Equals("x")
|
|
|
|
meta, err = reader.ReadMetadata()
|
|
assert.Error(err).IsNil()
|
|
assert.Byte(byte(meta.SessionStatus)).Equals(byte(SessionStatusEnd))
|
|
assert.Uint16(meta.SessionID).Equals(1)
|
|
assert.Byte(byte(meta.Option)).Equals(0)
|
|
|
|
meta, err = reader.ReadMetadata()
|
|
assert.Error(err).IsNil()
|
|
assert.Byte(byte(meta.SessionStatus)).Equals(byte(SessionStatusEnd))
|
|
assert.Uint16(meta.SessionID).Equals(3)
|
|
assert.Byte(byte(meta.Option)).Equals(0)
|
|
|
|
meta, err = reader.ReadMetadata()
|
|
assert.Error(err).IsNil()
|
|
assert.Byte(byte(meta.SessionStatus)).Equals(byte(SessionStatusKeep))
|
|
assert.Uint16(meta.SessionID).Equals(2)
|
|
assert.Byte(byte(meta.Option)).Equals(1)
|
|
|
|
data, err = reader.Read()
|
|
assert.Error(err).IsNil()
|
|
assert.Int(len(data)).Equals(1)
|
|
assert.String(data[0].String()).Equals("y")
|
|
|
|
meta, err = reader.ReadMetadata()
|
|
assert.Error(err).IsNil()
|
|
assert.Byte(byte(meta.SessionStatus)).Equals(byte(SessionStatusEnd))
|
|
assert.Uint16(meta.SessionID).Equals(2)
|
|
assert.Byte(byte(meta.Option)).Equals(0)
|
|
|
|
stream.Close()
|
|
|
|
meta, err = reader.ReadMetadata()
|
|
assert.Error(err).IsNotNil()
|
|
}
|