Secure client writes using mutex

Signed-off-by: Chaitanya Munukutla <[email protected]>
c16a · Apr 7, 2024 · 20a5337 · 20a5337
1 parent 9882dbd
commit 20a5337
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/broker/client.go b/broker/client.go
@@ -25,6 +25,9 @@ func NewConnectedClient(conn WebSocketConnection, id string) *ConnectedClient {
 }
 
 func (client *ConnectedClient) WriteDataMessage(data []byte) error {
+	client.mutex.Lock()
+	defer client.mutex.Unlock()
+
 	return client.conn.WriteMessage(websocket.TextMessage, data)
 }
 

diff --git a/main.go b/main.go
@@ -29,6 +29,7 @@ func echo(upgrader websocket.Upgrader, b *broker.Broker, sp storage.Provider) fu
 		defer c.Close()
 
 		clientId := request.Header.Get("Client-Id")
+
 		client := broker.NewConnectedClient(c, clientId)
 		b.Connect(clientId, client)