mirror of
https://github.com/XTLS/Xray-core.git
synced 2024-09-06 08:19:20 +03:00
196 lines
4.2 KiB
Go
196 lines
4.2 KiB
Go
package mux_test
|
|
|
|
import (
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/xtls/xray-core/common"
|
|
"github.com/xtls/xray-core/common/buf"
|
|
. "github.com/xtls/xray-core/common/mux"
|
|
"github.com/xtls/xray-core/common/net"
|
|
"github.com/xtls/xray-core/common/protocol"
|
|
"github.com/xtls/xray-core/transport/pipe"
|
|
)
|
|
|
|
func readAll(reader buf.Reader) (buf.MultiBuffer, error) {
|
|
var mb buf.MultiBuffer
|
|
for {
|
|
b, err := reader.ReadMultiBuffer()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
mb = append(mb, b...)
|
|
}
|
|
return mb, nil
|
|
}
|
|
|
|
func TestReaderWriter(t *testing.T) {
|
|
pReader, pWriter := pipe.New(pipe.WithSizeLimit(1024))
|
|
|
|
dest := net.TCPDestination(net.DomainAddress("example.com"), 80)
|
|
writer := NewWriter(1, dest, pWriter, protocol.TransferTypeStream, [8]byte{})
|
|
|
|
dest2 := net.TCPDestination(net.LocalHostIP, 443)
|
|
writer2 := NewWriter(2, dest2, pWriter, protocol.TransferTypeStream, [8]byte{})
|
|
|
|
dest3 := net.TCPDestination(net.LocalHostIPv6, 18374)
|
|
writer3 := NewWriter(3, dest3, pWriter, protocol.TransferTypeStream, [8]byte{})
|
|
|
|
writePayload := func(writer *Writer, payload ...byte) error {
|
|
b := buf.New()
|
|
b.Write(payload)
|
|
return writer.WriteMultiBuffer(buf.MultiBuffer{b})
|
|
}
|
|
|
|
common.Must(writePayload(writer, 'a', 'b', 'c', 'd'))
|
|
common.Must(writePayload(writer2))
|
|
|
|
common.Must(writePayload(writer, 'e', 'f', 'g', 'h'))
|
|
common.Must(writePayload(writer3, 'x'))
|
|
|
|
writer.Close()
|
|
writer3.Close()
|
|
|
|
common.Must(writePayload(writer2, 'y'))
|
|
writer2.Close()
|
|
|
|
bytesReader := &buf.BufferedReader{Reader: pReader}
|
|
|
|
{
|
|
var meta FrameMetadata
|
|
common.Must(meta.Unmarshal(bytesReader))
|
|
if r := cmp.Diff(meta, FrameMetadata{
|
|
SessionID: 1,
|
|
SessionStatus: SessionStatusNew,
|
|
Target: dest,
|
|
Option: OptionData,
|
|
}); r != "" {
|
|
t.Error("metadata: ", r)
|
|
}
|
|
|
|
data, err := readAll(NewStreamReader(bytesReader))
|
|
common.Must(err)
|
|
if s := data.String(); s != "abcd" {
|
|
t.Error("data: ", s)
|
|
}
|
|
}
|
|
|
|
{
|
|
var meta FrameMetadata
|
|
common.Must(meta.Unmarshal(bytesReader))
|
|
if r := cmp.Diff(meta, FrameMetadata{
|
|
SessionStatus: SessionStatusNew,
|
|
SessionID: 2,
|
|
Option: 0,
|
|
Target: dest2,
|
|
}); r != "" {
|
|
t.Error("meta: ", r)
|
|
}
|
|
}
|
|
|
|
{
|
|
var meta FrameMetadata
|
|
common.Must(meta.Unmarshal(bytesReader))
|
|
if r := cmp.Diff(meta, FrameMetadata{
|
|
SessionID: 1,
|
|
SessionStatus: SessionStatusKeep,
|
|
Option: 1,
|
|
}); r != "" {
|
|
t.Error("meta: ", r)
|
|
}
|
|
|
|
data, err := readAll(NewStreamReader(bytesReader))
|
|
common.Must(err)
|
|
if s := data.String(); s != "efgh" {
|
|
t.Error("data: ", s)
|
|
}
|
|
}
|
|
|
|
{
|
|
var meta FrameMetadata
|
|
common.Must(meta.Unmarshal(bytesReader))
|
|
if r := cmp.Diff(meta, FrameMetadata{
|
|
SessionID: 3,
|
|
SessionStatus: SessionStatusNew,
|
|
Option: 1,
|
|
Target: dest3,
|
|
}); r != "" {
|
|
t.Error("meta: ", r)
|
|
}
|
|
|
|
data, err := readAll(NewStreamReader(bytesReader))
|
|
common.Must(err)
|
|
if s := data.String(); s != "x" {
|
|
t.Error("data: ", s)
|
|
}
|
|
}
|
|
|
|
{
|
|
var meta FrameMetadata
|
|
common.Must(meta.Unmarshal(bytesReader))
|
|
if r := cmp.Diff(meta, FrameMetadata{
|
|
SessionID: 1,
|
|
SessionStatus: SessionStatusEnd,
|
|
Option: 0,
|
|
}); r != "" {
|
|
t.Error("meta: ", r)
|
|
}
|
|
}
|
|
|
|
{
|
|
var meta FrameMetadata
|
|
common.Must(meta.Unmarshal(bytesReader))
|
|
if r := cmp.Diff(meta, FrameMetadata{
|
|
SessionID: 3,
|
|
SessionStatus: SessionStatusEnd,
|
|
Option: 0,
|
|
}); r != "" {
|
|
t.Error("meta: ", r)
|
|
}
|
|
}
|
|
|
|
{
|
|
var meta FrameMetadata
|
|
common.Must(meta.Unmarshal(bytesReader))
|
|
if r := cmp.Diff(meta, FrameMetadata{
|
|
SessionID: 2,
|
|
SessionStatus: SessionStatusKeep,
|
|
Option: 1,
|
|
}); r != "" {
|
|
t.Error("meta: ", r)
|
|
}
|
|
|
|
data, err := readAll(NewStreamReader(bytesReader))
|
|
common.Must(err)
|
|
if s := data.String(); s != "y" {
|
|
t.Error("data: ", s)
|
|
}
|
|
}
|
|
|
|
{
|
|
var meta FrameMetadata
|
|
common.Must(meta.Unmarshal(bytesReader))
|
|
if r := cmp.Diff(meta, FrameMetadata{
|
|
SessionID: 2,
|
|
SessionStatus: SessionStatusEnd,
|
|
Option: 0,
|
|
}); r != "" {
|
|
t.Error("meta: ", r)
|
|
}
|
|
}
|
|
|
|
pWriter.Close()
|
|
|
|
{
|
|
var meta FrameMetadata
|
|
err := meta.Unmarshal(bytesReader)
|
|
if err == nil {
|
|
t.Error("nil error")
|
|
}
|
|
}
|
|
}
|