远程端连不上的时候把连接关了

tcp2ws.go

@@ -1,7 +1,7 @@
 // Tcp over WebSocket (tcp2ws)
 // 基于ws的内网穿透工具
 // Sparkle 20210430
-// 11.0
+// 11.1
 
 package main
 
@@ -382,6 +382,8 @@ func runServer(wsConn *websocket.Conn) {
 		udpConn, err = net.DialUDP("udp", nil, udpAddr)
 		if err != nil {
 			log.Print("connect to udp err: ", err)
+			wsConn.WriteMessage(websocket.TextMessage, []byte("tcp2wsSparkleClose"))
+			wsConn.Close()
 			return
 		}
 
@@ -395,6 +397,8 @@ func runServer(wsConn *websocket.Conn) {
 		tcpConn, err = net.Dial("tcp", tcpAddr)
 		if err != nil {
 			log.Print("connect to tcp err: ", err)
+			wsConn.WriteMessage(websocket.TextMessage, []byte("tcp2wsSparkleClose"))
+			wsConn.Close()
 			return
 		}
 
@@ -646,7 +650,7 @@ func dnsPreferIpWithTtl(hostname string, ttl uint32) {
 func main() {
 	arg_num := len(os.Args)
 	if arg_num < 3 {
-		fmt.Println("TCP over WebSocket (tcp2ws) with UDP support 11.0\nhttps://github.com/zanjie1999/tcp-over-websocket")
+		fmt.Println("TCP over WebSocket (tcp2ws) with UDP support 11.1\nhttps://github.com/zanjie1999/tcp-over-websocket")
 		fmt.Println("Client: ws://tcp2wsUrl localPort\nServer: ip:port tcp2wsPort\nUse wss: ip:port tcp2wsPort server.crt server.key")
 		fmt.Println("Make ssl cert:\nopenssl genrsa -out server.key 2048\nopenssl ecparam -genkey -name secp384r1 -out server.key\nopenssl req -new -x509 -sha256 -key server.key -out server.crt -days 36500")
 		os.Exit(0)