moul/sshportal
1
1
Fork 0
mirror of https://github.com/moul/sshportal.git synced 2025-01-15 11:59:54 +08:00
Code Issues Projects Releases Packages Wiki Activity

fixup! Fix early closure of data stream.

Browse source
This commit is contained in:
jerard@alfa-safety.fr 2020-06-30 18:28:47 +02:00
parent d2f3f460b2
commit bcb5d3b7ef
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
pkg/bastion/session.go
View file

@ -201,7 +201,7 @@ func pipe(lreqs, rreqs <-chan *gossh.Request, lch, rch gossh.Channel, logsLocati
quit <- "rreqs"
}(quit)
lchEof, rchEof, lchClosed, rchClosed := false, false, false, false
lchEOF, rchEOF, lchClosed, rchClosed := false, false, false, false
for {
select {
case err := <-errch:
@ -209,10 +209,10 @@ func pipe(lreqs, rreqs <-chan *gossh.Request, lch, rch gossh.Channel, logsLocati
case q := <-quit:
switch q {
case "lch":
lchEof = true
lchEOF = true
_ = rch.CloseWrite()
case "rch":
rchEof = true
rchEOF = true
_ = lch.CloseWrite()
case "lreqs":
lchClosed = true
@ -220,15 +220,15 @@ func pipe(lreqs, rreqs <-chan *gossh.Request, lch, rch gossh.Channel, logsLocati
rchClosed = true
}
if lchEof && lchClosed && !rchClosed {
if lchEOF && lchClosed && !rchClosed {
rch.Close()
}
if rchEof && rchClosed && !lchClosed {
if rchEOF && rchClosed && !lchClosed {
lch.Close()
}
if lchEof && rchEof && lchClosed && rchClosed {
if lchEOF && rchEOF && lchClosed && rchClosed {
return nil
}
}