kirillius
/
Cloak
mirror of https://github.com/cbeuw/Cloak
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Logging for silent connections

This commit is contained in:
Andy Wang 2020-04-05 01:15:45 +01:00
parent 8d84ebd145
commit 402cfc9e25
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cmd/ck-server/ck-server.go
View File

@ -32,7 +32,9 @@ func dispatchConnection(conn net.Conn, sta *server.State) {
conn.SetReadDeadline(time.Now().Add(3 * time.Second)) conn.SetReadDeadline(time.Now().Add(3 * time.Second))
i, err := io.ReadAtLeast(conn, buf, 1) i, err := io.ReadAtLeast(conn, buf, 1)
if err != nil { if err != nil {
go conn.Close() log.WithField("remoteAddr", remoteAddr).
Infof("failed to read anything after connection is established: %v", err)
conn.Close()
return return
} }
conn.SetReadDeadline(time.Time{}) conn.SetReadDeadline(time.Time{})

2
internal/multiplex/switchboard.go
View File

@ -152,6 +152,7 @@ func (sb *switchboard) closeAll() {
// deplex function costantly reads from a TCP connection // deplex function costantly reads from a TCP connection
func (sb *switchboard) deplex(connId uint32, conn net.Conn) { func (sb *switchboard) deplex(connId uint32, conn net.Conn) {
defer conn.Close()
buf := make([]byte, 20480) buf := make([]byte, 20480)
for { for {
n, err := sb.session.UnitRead(conn, buf) n, err := sb.session.UnitRead(conn, buf)
@ -159,7 +160,6 @@ func (sb *switchboard) deplex(connId uint32, conn net.Conn) {
sb.Valve.AddRx(int64(n)) sb.Valve.AddRx(int64(n))
if err != nil { if err != nil {
log.Debugf("a connection for session %v has closed: %v", sb.session.id, err) log.Debugf("a connection for session %v has closed: %v", sb.session.id, err)
go conn.Close()
sb.close("a connection has dropped unexpectedly") sb.close("a connection has dropped unexpectedly")
return return
} }