mirror of https://github.com/cbeuw/Cloak
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
151 lines
3.1 KiB
Go
151 lines
3.1 KiB
Go
package multiplex
|
|
|
|
import (
|
|
"github.com/cbeuw/Cloak/internal/util"
|
|
"math/rand"
|
|
"testing"
|
|
)
|
|
|
|
func TestSwitchboard_Send(t *testing.T) {
|
|
doTest := func(seshConfig *SessionConfig) {
|
|
sesh := MakeSession(0, seshConfig)
|
|
hole0 := newBlackHole()
|
|
sesh.sb.addConn(hole0)
|
|
connId, err := sesh.sb.assignRandomConn()
|
|
if err != nil {
|
|
t.Error("failed to get a random conn", err)
|
|
return
|
|
}
|
|
data := make([]byte, 1000)
|
|
rand.Read(data)
|
|
_, err = sesh.sb.send(data, &connId)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
|
|
hole1 := newBlackHole()
|
|
sesh.sb.addConn(hole1)
|
|
connId, err = sesh.sb.assignRandomConn()
|
|
if err != nil {
|
|
t.Error("failed to get a random conn", err)
|
|
return
|
|
}
|
|
_, err = sesh.sb.send(data, &connId)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
|
|
hole0.Close()
|
|
|
|
connId, err = sesh.sb.assignRandomConn()
|
|
if err != nil {
|
|
t.Error("failed to get a random conn", err)
|
|
return
|
|
}
|
|
_, err = sesh.sb.send(data, &connId)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
}
|
|
|
|
t.Run("Ordered", func(t *testing.T) {
|
|
seshConfig := &SessionConfig{
|
|
Obfuscator: nil,
|
|
Valve: nil,
|
|
UnitRead: util.ReadTLS,
|
|
Unordered: false,
|
|
}
|
|
doTest(seshConfig)
|
|
})
|
|
t.Run("Unordered", func(t *testing.T) {
|
|
seshConfig := &SessionConfig{
|
|
Obfuscator: nil,
|
|
Valve: nil,
|
|
UnitRead: util.ReadTLS,
|
|
Unordered: true,
|
|
}
|
|
doTest(seshConfig)
|
|
})
|
|
}
|
|
|
|
func BenchmarkSwitchboard_Send(b *testing.B) {
|
|
hole := newBlackHole()
|
|
seshConfig := &SessionConfig{
|
|
Obfuscator: nil,
|
|
Valve: nil,
|
|
UnitRead: util.ReadTLS,
|
|
}
|
|
sesh := MakeSession(0, seshConfig)
|
|
sesh.sb.addConn(hole)
|
|
connId, err := sesh.sb.assignRandomConn()
|
|
if err != nil {
|
|
b.Error("failed to get a random conn", err)
|
|
return
|
|
}
|
|
data := make([]byte, 1000)
|
|
rand.Read(data)
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
n, err := sesh.sb.send(data, &connId)
|
|
if err != nil {
|
|
b.Error(err)
|
|
return
|
|
}
|
|
b.SetBytes(int64(n))
|
|
}
|
|
}
|
|
|
|
func TestSwitchboard_TxCredit(t *testing.T) {
|
|
seshConfig := &SessionConfig{
|
|
Obfuscator: nil,
|
|
Valve: MakeValve(1<<20, 1<<20),
|
|
UnitRead: util.ReadTLS,
|
|
}
|
|
sesh := MakeSession(0, seshConfig)
|
|
hole := newBlackHole()
|
|
sesh.sb.addConn(hole)
|
|
connId, err := sesh.sb.assignRandomConn()
|
|
if err != nil {
|
|
t.Error("failed to get a random conn", err)
|
|
return
|
|
}
|
|
data := make([]byte, 1000)
|
|
rand.Read(data)
|
|
|
|
t.Run("FIXED CONN MAPPING", func(t *testing.T) {
|
|
*sesh.sb.Valve.(*LimitedValve).tx = 0
|
|
sesh.sb.strategy = FIXED_CONN_MAPPING
|
|
n, err := sesh.sb.send(data[:10], &connId)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
if n != 10 {
|
|
t.Errorf("wanted to send %v, got %v", 10, n)
|
|
return
|
|
}
|
|
if *sesh.sb.Valve.(*LimitedValve).tx != 10 {
|
|
t.Error("tx credit didn't increase by 10")
|
|
}
|
|
})
|
|
t.Run("UNIFORM", func(t *testing.T) {
|
|
*sesh.sb.Valve.(*LimitedValve).tx = 0
|
|
sesh.sb.strategy = UNIFORM_SPREAD
|
|
n, err := sesh.sb.send(data[:10], &connId)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
if n != 10 {
|
|
t.Errorf("wanted to send %v, got %v", 10, n)
|
|
return
|
|
}
|
|
if *sesh.sb.Valve.(*LimitedValve).tx != 10 {
|
|
t.Error("tx credit didn't increase by 10")
|
|
}
|
|
})
|
|
}
|