From 77f15a197acc7bc4aa475bae9b04e11ecaebdd83 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Magnus=20=C3=85hall?= <magnus.ahall@gibon.se>
Date: Thu, 15 Feb 2024 10:35:57 +0100
Subject: [PATCH] Added mutex locking around modifications to websocket
 connections

---
 ws_conn_manager/pkg.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/ws_conn_manager/pkg.go b/ws_conn_manager/pkg.go
index 386482e..5a4af9e 100644
--- a/ws_conn_manager/pkg.go
+++ b/ws_conn_manager/pkg.go
@@ -10,6 +10,7 @@ import (
 	"net/http"
 	"slices"
 	"strings"
+	"sync"
 )
 
 type ReadHandler func(*ConnectionManager, *WsConnection, []byte)
@@ -28,6 +29,7 @@ type ConnectionManager struct {
 	logger         *slog.Logger
 	domains        []string
 	readHandlers   []ReadHandler
+	connSync       sync.Mutex
 }
 type SendRequest struct {
 	WsConn *WsConnection
@@ -67,7 +69,9 @@ func (cm *ConnectionManager) NewConnection(w http.ResponseWriter, r *http.Reques
 	}
 
 	// Keep track of all connections.
+	cm.connSync.Lock()
 	cm.connections[wsConn.UUID] = &wsConn
+	cm.connSync.Unlock()
 
 	// Successfully upgraded to a websocket connection.
 	cm.logger.Info("websocket", "uuid", wsConn.UUID, "remote_addr", r.RemoteAddr)
@@ -97,7 +101,9 @@ func (cm *ConnectionManager) Prune(wsConn *WsConnection, err error) { // {{{
 	cm.logger.Info("websocket", "op", "prune", "uuid", wsConn.UUID)
 	wsConn.Conn.Close()
 	wsConn.Pruned = true
+	cm.connSync.Lock()
 	delete(cm.connections, wsConn.UUID)
+	cm.connSync.Unlock()
 } // }}}
 func (cm *ConnectionManager) ReadLoop(wsConn *WsConnection) { // {{{
 	var data []byte