extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2025-06-22 02:32:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

Protect ws writing in Gorilla ws

Browse Source
This commit is contained in:
Zoltan Papp 2024-06-07 16:07:35 +02:00
parent 1e115e3893
commit ed8def4d9b
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
relay/client/dialer/ws/client_conn.go
View File

@ -3,6 +3,7 @@ package ws
import ( import (
"fmt" "fmt"
"net" "net"
"sync"
"time" "time"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
@ -10,11 +11,12 @@ import (
type Conn struct { type Conn struct {
*websocket.Conn *websocket.Conn
mu sync.Mutex
} }
func NewConn(wsConn *websocket.Conn) net.Conn { func NewConn(wsConn *websocket.Conn) net.Conn {
return &Conn{ return &Conn{
wsConn, Conn: wsConn,
} }
} }
@ -32,7 +34,9 @@ func (c *Conn) Read(b []byte) (n int, err error) {
} }
func (c *Conn) Write(b []byte) (int, error) { func (c *Conn) Write(b []byte) (int, error) {
c.mu.Lock()
err := c.WriteMessage(websocket.BinaryMessage, b) err := c.WriteMessage(websocket.BinaryMessage, b)
c.mu.Unlock()
return len(b), err return len(b), err
} }