add icmp tunnel

pkg/internal/util/ws/ws.go

@@ -2,17 +2,25 @@ package ws
 
 import (
 	"net"
+	"sync"
 	"time"
 
 	"github.com/gorilla/websocket"
 )
 
-type websocketConn struct {
-	*websocket.Conn
-	rb []byte
+type WebsocketConn interface {
+	net.Conn
+	WriteMessage(int, []byte) error
+	ReadMessage() (int, []byte, error)
 }
 
-func Conn(conn *websocket.Conn) net.Conn {
+type websocketConn struct {
+	*websocket.Conn
+	rb  []byte
+	mux sync.Mutex
+}
+
+func Conn(conn *websocket.Conn) WebsocketConn {
 	return &websocketConn{
 		Conn: conn,
 	}
@@ -20,7 +28,7 @@ func Conn(conn *websocket.Conn) net.Conn {
 
 func (c *websocketConn) Read(b []byte) (n int, err error) {
 	if len(c.rb) == 0 {
-		_, c.rb, err = c.ReadMessage()
+		_, c.rb, err = c.Conn.ReadMessage()
 	}
 	n = copy(b, c.rb)
 	c.rb = c.rb[n:]
@@ -33,6 +41,13 @@ func (c *websocketConn) Write(b []byte) (n int, err error) {
 	return
 }
 
+func (c *websocketConn) WriteMessage(messageType int, data []byte) error {
+	c.mux.Lock()
+	defer c.mux.Unlock()
+
+	return c.Conn.WriteMessage(messageType, data)
+}
+
 func (c *websocketConn) SetDeadline(t time.Time) error {
 	if err := c.SetReadDeadline(t); err != nil {
 		return err