Erik Brakkee
f59c8368ea
1. two websocket clients connecting: one as agent and one as client 2. bidirectional communication between the two
54 lines
1008 B
Go
54 lines
1008 B
Go
package main
|
|
|
|
import (
|
|
"cidebug/pkg/iowrappers"
|
|
"github.com/gorilla/websocket"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
)
|
|
|
|
func closeConnection(conn net.Conn) {
|
|
if tcpConn, ok := conn.(*net.TCPConn); ok {
|
|
tcpConn.SetLinger(0)
|
|
}
|
|
_ = conn.Close()
|
|
}
|
|
|
|
func handleConnection(conn net.Conn, wsURL string) {
|
|
defer closeConnection(conn)
|
|
|
|
_wsConn, _, err := websocket.DefaultDialer.Dial(wsURL, nil)
|
|
if err != nil {
|
|
log.Println("WebSocket connection error:", err)
|
|
return
|
|
}
|
|
wsConn := iowrappers.NewWebSocketConn(_wsConn)
|
|
defer wsConn.Close()
|
|
|
|
iowrappers.SynchronizeStreams(wsConn, conn)
|
|
}
|
|
|
|
func main() {
|
|
tcpPort := os.Args[1]
|
|
wsURL := os.Args[2]
|
|
|
|
listener, err := net.Listen("tcp", ":"+tcpPort)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer listener.Close()
|
|
|
|
log.Printf("TCP server listening on port %s", tcpPort)
|
|
log.Printf("Forwarding connections to WebSocket server at %s", wsURL)
|
|
|
|
for {
|
|
conn, err := listener.Accept()
|
|
if err != nil {
|
|
log.Println(err)
|
|
continue
|
|
}
|
|
go handleConnection(conn, wsURL)
|
|
}
|
|
}
|