Close pipe properly

This commit is contained in:
Qian Wang 2019-07-25 22:06:33 +01:00
parent eb68185f3c
commit 592175f73d
2 changed files with 9 additions and 1 deletions

View File

@ -9,6 +9,7 @@ import (
"log"
"net"
"net/http"
_ "net/http/pprof"
"os"
"runtime"
"strings"

View File

@ -73,7 +73,12 @@ func (stream *Stream) feed() {
}
_, err := stream.feederW.Write(data)
if err != nil {
log.Println(err)
if err == io.ErrClosedPipe {
stream.Close()
return
} else {
log.Println(err)
}
}
}
}
@ -164,6 +169,8 @@ func (stream *Stream) Close() error {
stream.session.sb.send(tlsRecord)
stream.session.delStream(stream.id)
stream.feederW.Close()
stream.feederR.Close()
//log.Printf("%v actively closed\n", stream.id)
stream.writingM.Unlock()
return nil