RPRX 7 days ago committed by GitHub
parent 2d7b0e8cd4
commit 1f570d9cef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -424,8 +424,8 @@ func Test_maxUpload(t *testing.T) {
ProtocolSettings: &Config{ ProtocolSettings: &Config{
Path: "/sh", Path: "/sh",
ScMaxEachPostBytes: &RandRangeConfig{ ScMaxEachPostBytes: &RandRangeConfig{
From: 100, From: 10000,
To: 100, To: 10000,
}, },
}, },
} }
@ -434,7 +434,7 @@ func Test_maxUpload(t *testing.T) {
listen, err := ListenSH(context.Background(), net.LocalHostIP, listenPort, streamSettings, func(conn stat.Connection) { listen, err := ListenSH(context.Background(), net.LocalHostIP, listenPort, streamSettings, func(conn stat.Connection) {
go func(c stat.Connection) { go func(c stat.Connection) {
defer c.Close() defer c.Close()
var b [1024]byte var b [10240]byte
c.SetReadDeadline(time.Now().Add(2 * time.Second)) c.SetReadDeadline(time.Now().Add(2 * time.Second))
n, err := c.Read(b[:]) n, err := c.Read(b[:])
if err != nil { if err != nil {
@ -452,11 +452,11 @@ func Test_maxUpload(t *testing.T) {
conn, err := Dial(ctx, net.TCPDestination(net.DomainAddress("localhost"), listenPort), streamSettings) conn, err := Dial(ctx, net.TCPDestination(net.DomainAddress("localhost"), listenPort), streamSettings)
// send a slightly too large upload // send a slightly too large upload
var upload [101]byte var upload [10001]byte
_, err = conn.Write(upload[:]) _, err = conn.Write(upload[:])
common.Must(err) common.Must(err)
var b [1024]byte var b [10240]byte
n, _ := io.ReadFull(conn, b[:]) n, _ := io.ReadFull(conn, b[:])
fmt.Println("string is", n) fmt.Println("string is", n)
if string(b[:n]) != "Response" { if string(b[:n]) != "Response" {
@ -464,7 +464,7 @@ func Test_maxUpload(t *testing.T) {
} }
common.Must(conn.Close()) common.Must(conn.Close())
if uploadSize > 100 || uploadSize == 0 { if uploadSize > 10000 || uploadSize == 0 {
t.Error("incorrect upload size: ", uploadSize) t.Error("incorrect upload size: ", uploadSize)
} }

Loading…
Cancel
Save