Merge pull request #40 from xssnick/tests

Tests
xssnick · Jun 17, 2022 · 281140f · 281140f
2 parents 12a00d3 + dc03318
commit 281140f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/liteclient/connection.go b/liteclient/connection.go
@@ -14,6 +14,7 @@ import (
 	"io"
 	"math/big"
 	"net"
+	"sync"
 	"time"
 
 	"github.com/xssnick/tonutils-go/utils"
@@ -28,6 +29,7 @@ type node struct {
 	conn   net.Conn
 	rCrypt cipher.Stream
 	wCrypt cipher.Stream
+	wLock  sync.Mutex
 
 	reqs chan *LiteRequest
 
@@ -281,6 +283,9 @@ func (n *node) send(data []byte) error {
 
 	buf = append(buf, checksum...)
 
+	n.wLock.Lock()
+	defer n.wLock.Unlock()
+
 	// encrypt data
 	n.wCrypt.XORKeyStream(buf, buf)