|
|
@ -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)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|