disable the default tcp keepalive mechanism

Author: VHSgunzo

Makefile

@@ -1,7 +1,10 @@
 VERSION=$(shell git describe --abbrev=0 --tags)
-DIRBASE=./build
-DIR=${DIRBASE}/${VERSION}/bin
+BUILD=$(shell git rev-parse HEAD)
+DIRBUILD=./build
+# DIR=${DIRBUILD}/${VERSION}/${BUILD}/bin
+DIR=${DIRBUILD}
 
+# LDFLAGS=-ldflags "-s -w ${XBUILD} -buildid=${BUILD} -X github.com/jpillora/chisel/share.BuildVersion=${VERSION}"
 LDFLAGS=-ldflags "-s -w ${XBUILD} -buildid= -X github.com/jpillora/chisel/share.BuildVersion=${VERSION}"
 
 GOFILES=`go list ./...`
@@ -48,6 +51,6 @@ release: lint test
 	goreleaser release --config .github/goreleaser.yml
 
 clean:
-	rm -rf ${DIRBASE}/*
+	rm -rf ${DIRBUILD}/*
 
 .PHONY: all freebsd linux windows docker dep lint test release clean

client/client.go

@@ -70,9 +70,7 @@ func NewClient(c *Config) (*Client, error) {
 		return nil, err
 	}
 	//swap to websockets scheme
-	// if !strings.HasPrefix(c.Server, "ws") {
 	u.Scheme = strings.Replace(u.Scheme, "http", "ws", 1)
-	// }
 	//apply default port
 	if !regexp.MustCompile(`:\d+$`).MatchString(u.Host) {
 		u.Host = u.Host + ":2871"

client/client_connect.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"io"
+	"net"
 	"strings"
 	"time"
 
@@ -93,6 +94,11 @@ func (c *Client) connectionOnce(ctx context.Context) (connected bool, err error)
 	if err != nil {
 		return false, err
 	}
+	// disable the default tcp keepalive mechanism.
+	uwsConn := wsConn.UnderlyingConn()
+	if tcpConn, ok := uwsConn.(*net.TCPConn); ok {
+		tcpConn.SetKeepAlive(false)
+	}
 	conn := cnet.NewWebSocketConn(wsConn)
 	// perform SSH handshake on net.Conn
 	c.Debugf("Handshaking...")

server/server.go

@@ -19,7 +19,6 @@ import (
 	"github.com/jpillora/chisel/share/settings"
 	"github.com/jpillora/requestlog"
 	"golang.org/x/crypto/ssh"
-	// websocket "github.com/lxzan/gws"
 )
 
 // Config is the configuration for the chisel service

server/server_handler.go

@@ -1,6 +1,7 @@
 package chserver
 
 import (
+	"net"
 	"net/http"
 	"strings"
 	"sync/atomic"
@@ -56,6 +57,11 @@ func (s *Server) handleWebsocket(w http.ResponseWriter, req *http.Request) {
 		l.Debugf("Failed to upgrade (%s)", err)
 		return
 	}
+	// disable the default tcp keepalive mechanism.
+	uwsConn := wsConn.UnderlyingConn()
+	if tcpConn, ok := uwsConn.(*net.TCPConn); ok {
+		tcpConn.SetKeepAlive(false)
+	}
 	conn := cnet.NewWebSocketConn(wsConn)
 	// perform SSH handshake on net.Conn
 	l.Debugf("Handshaking with %s...", req.RemoteAddr)