Merge branch 'windows' of openprivacy/libricochet-go into master

Also hide it in checkTorVersion

.drone.yml

@@ -5,27 +5,35 @@ workspace:
 pipeline:
   fetch:
     image: golang
+    environment:
+      - GO111MODULE=on
     commands:
       - wget https://git.openprivacy.ca/openprivacy/buildfiles/raw/master/tor/tor
       - wget https://git.openprivacy.ca/openprivacy/buildfiles/raw/master/tor/torrc
       - chmod a+x tor
-      - go list ./... | xargs go get
-      - go get -u github.com/golang/lint/golint
+      - go get -u golang.org/x/lint/golint
+      - go mod download
   quality:
     image: golang
+    environment:
+      - GO111MODULE=on
     commands:
       - go list ./... | xargs go vet
       - go list ./... | grep -v "/wire/" | xargs golint -set_exit_status
   units-tests:
     image: golang
+    environment:
+      - GO111MODULE=on
     commands:
       - sh testing/tests.sh
   integ-test:
     image: golang
+    environment:
+      - GO111MODULE=on
     commands:
       - ./tor -f ./torrc
       - sleep 15
-      - go test -v git.openprivacy.ca/openprivacy/libricochet-go/testing
+      - go test -race -v git.openprivacy.ca/openprivacy/libricochet-go/testing
   notify-email:
       image: drillster/drone-email
       host: build.openprivacy.ca

application/application.go

@@ -144,7 +144,9 @@ func (ra *RicochetApplication) Run(ls connectivity.ListenService) {
 	if !ra.v3identity.Initialized() || ra.contactManager == nil {
 		return
 	}
+	ra.lock.Lock()
 	ra.ls = ls
+	ra.lock.Unlock()
 	var err error
 	for err == nil {
 		conn, err := ra.ls.Accept()

connection/connection.go

@@ -32,8 +32,9 @@ type Connection struct {
 
 	messageBuilder utils.MessageBuilder
 
-	closed  bool
-	closing bool
+	closed      bool
+	closingLock sync.Mutex
+	closing     bool
 	// This mutex is exclusively for preventing races during blocking
 	// interactions with Process; specifically Do and Break. Don't use
 	// it for anything else. See those functions for an explanation.
@@ -311,6 +312,8 @@ func (rc *Connection) Process(handler Handler) error {
 			go func() {
 				rc.processBlockMutex.Lock()
 				defer rc.processBlockMutex.Unlock()
+				rc.closingLock.Lock()
+				defer rc.closingLock.Unlock()
 				rc.closed = true
 				close(closedChan)
 			}()
@@ -468,5 +471,7 @@ func (rc *Connection) Close() {
 	// Kill the Ricochet Connection.
 	log.Debugf("Closing Ricochet Connection for %v", rc.RemoteHostname)
 	rc.conn.Close()
+	rc.closingLock.Lock()
 	rc.closed = true
+	rc.closingLock.Unlock()
 }

connection/connection_test.go

@@ -53,6 +53,9 @@ func TestProcessAuthAs3DHServer(t *testing.T) {
 		t.Errorf("Error while testing ProcessAuthAsServer: %v", err)
 	}
 
+	// Wait for server to finish
+	time.Sleep(time.Second * 2)
+
 	// Test Close
 	rc.Close()
 }

connectivity/acn.go

@@ -25,6 +25,11 @@ type ACN interface {
 	GetBootstrapStatus() (int, string)
 	// WaitTillBootstrapped Blocks until underlying network is bootstrapped
 	WaitTillBootstrapped()
+	// Sets the calback function to be called when ACN status changes
+	SetStatusCallback(callback func(int, string))
+
+	// Restarts the underlying connection
+	Restart()
 
 	// Open takes a hostname and returns a net.conn to the derived endpoint
 	// Open allows a client to resolve various hostnames to connections

connectivity/localProvider.go

@@ -34,6 +34,10 @@ func (lp *localProvider) GetBootstrapStatus() (int, string) {
 	return 100, "Done"
 }
 
+func (lp *localProvider) SetStatusCallback(callback func(int, string)) {
+	// nop
+}
+
 // WaitTillBootstrapped Blocks until underlying network is bootstrapped
 func (lp *localProvider) WaitTillBootstrapped() {
 }
@@ -59,6 +63,10 @@ func (lp *localProvider) Open(hostname string) (net.Conn, string, error) {
 
 }
 
+func (lp *localProvider) Restart() {
+	//noop
+}
+
 func (lp *localProvider) Close() {
 
 }

connectivity/sysProcAttr_rest.go

@@ -0,0 +1,9 @@
+// +build !windows
+
+package connectivity
+
+import (
+	"syscall"
+)
+
+var sysProcAttr = &syscall.SysProcAttr{}

connectivity/sysProcAttr_win.go

@@ -0,0 +1,9 @@
+// +build windows
+
+package connectivity
+
+import (
+	"syscall"
+)
+
+var sysProcAttr = &syscall.SysProcAttr{HideWindow: true}

connectivity/torProvider.go

@@ -1,10 +1,12 @@
 package connectivity
 
 import (
+	"context"
 	"errors"
 	"git.openprivacy.ca/openprivacy/libricochet-go/log"
 	"git.openprivacy.ca/openprivacy/libricochet-go/utils"
 	"github.com/cretz/bine/control"
+	"github.com/cretz/bine/process"
 	"github.com/cretz/bine/tor"
 	bineed255192 "github.com/cretz/bine/torutil/ed25519"
 	"golang.org/x/crypto/ed25519"
@@ -30,6 +32,11 @@ const (
 	CannotDialRicochetAddressError = utils.Error("CannotDialRicochetAddressError")
 )
 
+const (
+	minStatusIntervalMs = 200
+	maxStatusIntervalMs = 2000
+)
+
 type onionListenService struct {
 	os *tor.OnionService
 	tp *torProvider
@@ -37,11 +44,13 @@ type onionListenService struct {
 
 type torProvider struct {
 	t               *tor.Tor
+	dialer          *tor.Dialer
 	appDirectory    string
 	bundeledTorPath string
 	lock            sync.Mutex
 	breakChan       chan bool
 	childListeners  map[string]*onionListenService
+	statusCallback  func(int, string)
 }
 
 func (ols *onionListenService) AddressFull() string {
@@ -61,14 +70,14 @@ func (ols *onionListenService) Close() {
 	ols.os.Close()
 }
 
-// GetBootstrapStatus returns an int 0-100 on the percent the bootstrapping of the underlying network is at and an optional string message
+// GetBootstrapStatus returns an int -1 on error or 0-100 on the percent the bootstrapping of the underlying network is at and an optional string message
 func (tp *torProvider) GetBootstrapStatus() (int, string) {
 	if tp.t == nil {
-		return 0, "error: no tor, trying to restart..."
+		return -1, "error: no tor, trying to restart..."
 	}
 	kvs, err := tp.t.Control.GetInfo("status/bootstrap-phase")
 	if err != nil {
-		return 0, "error"
+		return -1, "error"
 	}
 	progress := 0
 	status := ""
@@ -103,6 +112,9 @@ func (tp *torProvider) Listen(identity PrivateKey, port int) (ListenService, err
 	var onion = ""
 	var privkey ed25519.PrivateKey
 
+	tp.lock.Lock()
+	defer tp.lock.Unlock()
+
 	if tp.t == nil {
 		return nil, errors.New("Tor Provider closed")
 	}
@@ -125,14 +137,8 @@ func (tp *torProvider) Listen(identity PrivateKey, port int) (ListenService, err
 		localport += 1024
 	}
 
-	if tp.t == nil {
-		return nil, errors.New("Tor is offline")
-	}
-
 	localListener, err := net.Listen("tcp", "127.0.0.1:"+strconv.Itoa(localport))
 
-	tp.lock.Lock()
-	defer tp.lock.Unlock()
 	conf := &tor.ListenConf{NoWait: true, Version3: true, Key: identity, RemotePorts: []int{port}, Detach: true, DiscardKey: true, LocalListener: localListener}
 	os, err := tp.t.Listen(nil, conf)
 	if err != nil && strings.Contains(err.Error(), "550 Unspecified Tor error: Onion address collision") {
@@ -151,24 +157,29 @@ func (tp *torProvider) Listen(identity PrivateKey, port int) (ListenService, err
 	return ols, nil
 }
 
+func (tp *torProvider) Restart() {
+	if tp.statusCallback != nil {
+		tp.statusCallback(0, "rebooting")
+	}
+	tp.restart()
+}
+
 func (tp *torProvider) Open(hostname string) (net.Conn, string, error) {
 	tp.lock.Lock()
-	defer tp.lock.Unlock()
 
 	if tp.t == nil {
+		tp.lock.Unlock()
 		return nil, hostname, errors.New("Tor is offline")
 	}
-	torDailer, err := tp.t.Dialer(nil, &tor.DialConf{})
-	if err != nil {
-		return nil, "", err
-	}
+	tp.lock.Unlock()
+
 	resolvedHostname := hostname
 	if strings.HasPrefix(hostname, "ricochet:") {
 		addrParts := strings.Split(hostname, ":")
 		resolvedHostname = addrParts[1]
 	}
 
-	conn, err := torDailer.Dial("tcp", resolvedHostname+".onion:9878")
+	conn, err := tp.dialer.Dial("tcp", resolvedHostname+".onion:9878")
 	return conn, resolvedHostname, err
 }
 
@@ -186,11 +197,39 @@ func (tp *torProvider) Close() {
 	}
 }
 
+func (tp *torProvider) SetStatusCallback(callback func(int, string)) {
+	tp.lock.Lock()
+	defer tp.lock.Unlock()
+	tp.statusCallback = callback
+}
+
 // StartTor creates/starts a Tor ACN and returns a usable ACN object
 func StartTor(appDirectory string, bundledTorPath string) (ACN, error) {
+	tp, err := startTor(appDirectory, bundledTorPath)
+	if err == nil {
+		tp.dialer, err = tp.t.Dialer(nil, &tor.DialConf{})
+		if err == nil {
+			go tp.monitorRestart()
+		}
+	}
+	return tp, err
+}
+
+// newHideCmd creates a Creator function for bine which generates a cmd that one windows will hide the dosbox
+func newHideCmd(exePath string) process.Creator {
+	return process.CmdCreatorFunc(func(ctx context.Context, args ...string) (*exec.Cmd, error) {
+		cmd := exec.CommandContext(ctx, exePath, args...)
+		cmd.Stdout = os.Stdout
+		cmd.Stderr = os.Stderr
+		cmd.SysProcAttr = sysProcAttr
+		return cmd, nil
+	})
+}
+
+func startTor(appDirectory string, bundledTorPath string) (*torProvider, error) {
 	dataDir := path.Join(appDirectory, "tor")
 	os.MkdirAll(dataDir, 0700)
-	tp := &torProvider{appDirectory: appDirectory, bundeledTorPath: bundledTorPath, childListeners: make(map[string]*onionListenService), breakChan: make(chan bool)}
+	tp := &torProvider{appDirectory: appDirectory, bundeledTorPath: bundledTorPath, childListeners: make(map[string]*onionListenService), breakChan: make(chan bool), statusCallback: nil}
 
 	// attempt connect to system tor
 	log.Debugf("dialing system tor control port\n")
@@ -205,7 +244,6 @@ func StartTor(appDirectory string, bundledTorPath string) (ACN, error) {
 			if err == nil && minTorVersionReqs(pinfo.TorVersion) {
 				log.Debugln("OK version " + pinfo.TorVersion)
 				tp.t = createFromExisting(controlport, dataDir)
-				go tp.monitorRestart()
 				return tp, nil
 			}
 			controlport.Close()
@@ -214,10 +252,9 @@ func StartTor(appDirectory string, bundledTorPath string) (ACN, error) {
 
 	// if not, try running system tor
 	if checkCmdlineTorVersion("tor") {
-		t, err := tor.Start(nil, &tor.StartConf{EnableNetwork: true, DataDir: dataDir, DebugWriter: nil})
+		t, err := tor.Start(nil, &tor.StartConf{EnableNetwork: true, DataDir: dataDir, DebugWriter: nil, ProcessCreator: newHideCmd("tor")})
 		if err == nil {
 			tp.t = t
-			go tp.monitorRestart()
 			return tp, nil
 		}
 		log.Debugf("Error connecting to self-run system tor: %v\n", err)
@@ -226,14 +263,11 @@ func StartTor(appDirectory string, bundledTorPath string) (ACN, error) {
 	// try running bundledTor
 	if bundledTorPath != "" && checkCmdlineTorVersion(bundledTorPath) {
 		log.Debugln("using bundled tor '" + bundledTorPath + "'")
-		t, err := tor.Start(nil, &tor.StartConf{EnableNetwork: true, DataDir: dataDir, ExePath: bundledTorPath, DebugWriter: nil})
+		t, err := tor.Start(nil, &tor.StartConf{EnableNetwork: true, DataDir: dataDir, ExePath: bundledTorPath, DebugWriter: nil, ProcessCreator: newHideCmd(bundledTorPath)})
 		if err != nil {
 			log.Debugf("Error running bundled tor: %v\n", err)
 		}
 		tp.t = t
-		if err == nil {
-			go tp.monitorRestart()
-		}
 		return tp, err
 	}
 	return nil, errors.New("Could not connect to or start Tor that met requirments (min Tor version 0.3.5.x)")
@@ -246,41 +280,56 @@ func (tp *torProvider) unregisterListener(id string) {
 }
 
 func (tp *torProvider) monitorRestart() {
+	lastBootstrapProgress := 0
+	interval := minStatusIntervalMs
+
 	for {
 		select {
-		case <-time.After(time.Duration(30 * time.Second)):
-			tp.lock.Lock()
-			if tp.t != nil {
-				_, err := tp.t.Control.GetInfo("version")
+		case <-time.After(time.Millisecond * time.Duration(interval)):
+			prog, status := tp.GetBootstrapStatus()
 
-				if err != nil {
-					tp.lock.Unlock()
-					for _, child := range tp.childListeners {
-						child.Close()
-					}
-					tp.lock.Lock()
-					tp.t.Close()
-					tp.t = nil
+			if prog == -1 && tp.t != nil {
+				if tp.statusCallback != nil {
+					tp.statusCallback(prog, status)
+				}
+				tp.restart()
+				interval = minStatusIntervalMs
+			} else if prog != lastBootstrapProgress {
+				if tp.statusCallback != nil {
+					tp.statusCallback(prog, status)
+				}
+				interval = minStatusIntervalMs
+			} else {
+				if interval < maxStatusIntervalMs {
+					interval *= 2
 				}
 			}
-
-			if tp.t == nil {
-				newACN, err := StartTor(tp.appDirectory, tp.bundeledTorPath)
-				if err == nil {
-					switch newTp := newACN.(type) {
-					case *torProvider:
-						tp.t = newTp.t
-						// startTor will have started a new monitorRestart thread
-						tp.lock.Unlock()
-						return
-					}
-				}
-			}
-			tp.lock.Unlock()
+			lastBootstrapProgress = prog
 		case <-tp.breakChan:
 			return
 		}
+	}
+}
 
+func (tp *torProvider) restart() {
+
+	for _, child := range tp.childListeners {
+		child.Close()
+	}
+
+	tp.lock.Lock()
+	defer tp.lock.Unlock()
+
+	tp.t.Close()
+	tp.t = nil
+
+	for {
+		newTp, err := startTor(tp.appDirectory, tp.bundeledTorPath)
+		if err == nil {
+			tp.t = newTp.t
+			tp.dialer, _ = tp.t.Dialer(nil, &tor.DialConf{})
+			return
+		}
 	}
 }
 
@@ -305,6 +354,7 @@ func createFromExisting(controlport *control.Conn, datadir string) *tor.Tor {
 
 func checkCmdlineTorVersion(torCmd string) bool {
 	cmd := exec.Command(torCmd, "--version")
+	cmd.SysProcAttr = sysProcAttr
 	out, err := cmd.CombinedOutput()
 	re := regexp.MustCompile("[0-1]\\.[0-9]\\.[0-9]\\.[0-9]")
 	sysTorVersion := re.Find(out)

connectivity/torProvider_test.go

@@ -3,21 +3,27 @@ package connectivity
 import (
 	"fmt"
 	"testing"
-	"time"
 )
 
+func getStatusCallback(progChan chan int) func(int, string) {
+	return func(prog int, status string) {
+		fmt.Printf("%v %v\n", prog, status)
+		progChan <- prog
+	}
+}
+
 func TestTorProvider(t *testing.T) {
+	progChan := make(chan int)
 	acn, err := StartTor(".", "")
 	if err != nil {
 		t.Error(err)
+		return
 	}
+	acn.SetStatusCallback(getStatusCallback(progChan))
 
 	progress := 0
-	status := ""
 	for progress < 100 {
-		progress, status = acn.GetBootstrapStatus()
-		fmt.Printf("%v %v\n", progress, status)
-		time.Sleep(100)
+		progress = <-progChan
 	}
 
 	acn.Close()

go.mod

@@ -2,10 +2,12 @@ module git.openprivacy.ca/openprivacy/libricochet-go
 
 require (
 	github.com/agl/ed25519 v0.0.0-20170116200512-5312a6153412
-	github.com/cretz/bine v0.1.0
+	github.com/cretz/bine v0.1.1-0.20200124154328-f9f678b84cca
 	github.com/golang/protobuf v1.2.0
 	github.com/stretchr/testify v1.3.0 // indirect
 	golang.org/x/crypto v0.0.0-20190128193316-c7b33c32a30b
 	golang.org/x/net v0.0.0-20190125091013-d26f9f9a57f3 // indirect
 	golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4 // indirect
 )
+
+go 1.13

go.sum

@@ -2,6 +2,8 @@ github.com/agl/ed25519 v0.0.0-20170116200512-5312a6153412 h1:w1UutsfOrms1J05zt7I
 github.com/agl/ed25519 v0.0.0-20170116200512-5312a6153412/go.mod h1:WPjqKcmVOxf0XSf3YxCJs6N6AOSrOx3obionmG7T0y0=
 github.com/cretz/bine v0.1.0 h1:1/fvhLE+fk0bPzjdO5Ci+0ComYxEMuB1JhM4X5skT3g=
 github.com/cretz/bine v0.1.0/go.mod h1:6PF6fWAvYtwjRGkAuDEJeWNOv3a2hUouSP/yRYXmvHw=
+github.com/cretz/bine v0.1.1-0.20200124154328-f9f678b84cca h1:Q2r7AxHdJwWfLtBZwvW621M3sPqxPc6ITv2j1FGsYpw=
+github.com/cretz/bine v0.1.1-0.20200124154328-f9f678b84cca/go.mod h1:6PF6fWAvYtwjRGkAuDEJeWNOv3a2hUouSP/yRYXmvHw=
 github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/golang/protobuf v1.2.0 h1:P3YflyNX/ehuJFLhxviNdFxQPkGK5cDcApsge1SqnvM=

log/log.go

@@ -31,6 +31,7 @@ type Logger struct {
 	level                  Level
 	nothingExceptPatterns  []string
 	everythingFromPatterns []string
+	excludeFromPatterns    []string
 }
 
 // New returns a new Logger with a filter set to the supplied level
@@ -38,8 +39,23 @@ func New(level Level) *Logger {
 	return &Logger{logger: golog.New(os.Stderr, "", golog.Ldate|golog.Ltime), level: level, everythingFromPatterns: make([]string, 0), nothingExceptPatterns: make([]string, 0)}
 }
 
+// NewFile returns a new Logger that logs to the supplied file with a filter set to the supplied level
+func NewFile(level Level, filename string) (*Logger, error) {
+	logfile, err := os.OpenFile(filename, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0600)
+	if err != nil {
+		return nil, err
+	}
+
+	return &Logger{logger: golog.New(logfile, "", golog.Ldate|golog.Ltime), level: level, everythingFromPatterns: make([]string, 0), nothingExceptPatterns: make([]string, 0)}, nil
+}
+
 var std = New(LevelWarn)
 
+// SetStd sets the default logger all other functions use
+func SetStd(logger *Logger) {
+	std = logger
+}
+
 // filter
 func (l *Logger) filter(level Level) bool {
 
@@ -48,6 +64,12 @@ func (l *Logger) filter(level Level) bool {
 		file = "???"
 	}
 
+	for _, pattern := range l.excludeFromPatterns {
+		if strings.Contains(file, pattern) {
+			return false
+		}
+	}
+
 	for _, pattern := range l.everythingFromPatterns {
 		if strings.Contains(file, pattern) {
 			return true
@@ -81,11 +103,16 @@ func (l *Logger) AddNothingExceptFilter(pattern string) {
 	l.nothingExceptPatterns = append(l.nothingExceptPatterns, pattern)
 }
 
-// AddEverythingFromPattern adds a pattern to skip log level filtering, guaranteeing all logs matching hte pattern are seen
+// AddEverythingFromPattern adds a pattern to skip log level filtering, guaranteeing all logs matching the pattern are seen
 func (l *Logger) AddEverythingFromPattern(pattern string) {
 	l.everythingFromPatterns = append(l.everythingFromPatterns, pattern)
 }
 
+// ExcludeFromPattern adds a pattern to exclude logs from
+func (l *Logger) ExcludeFromPattern(pattern string) {
+	l.excludeFromPatterns = append(l.excludeFromPatterns, pattern)
+}
+
 func (l *Logger) header(level Level) string {
 	_, file, _, ok := runtime.Caller(3)
 	if !ok {
@@ -131,11 +158,16 @@ func AddNothingExceptFilter(pattern string) {
 	std.AddNothingExceptFilter(pattern)
 }
 
-// AddEverythingFromPattern adds a pattern to skip log level filtering, guaranteeing all logs matching hte pattern are seen
+// AddEverythingFromPattern adds a pattern to skip log level filtering, guaranteeing all logs matching the pattern are seen
 func AddEverythingFromPattern(pattern string) {
 	std.AddEverythingFromPattern(pattern)
 }
 
+// ExcludeFromPattern adds a pattern to exclude logs from
+func ExcludeFromPattern(pattern string) {
+	std.ExcludeFromPattern(pattern)
+}
+
 // Printf outputs the format with variables assuming it passes the filter level
 func Printf(level Level, format string, v ...interface{}) {
 	std.Printf(level, format, v...)

testing/tests.sh

@@ -4,13 +4,14 @@
 
 set -e
 pwd
-go test ${1} -coverprofile=model.cover.out -v ./utils
-go test ${1} -coverprofile=channels.cover.out -v ./channels
-go test ${1} -coverprofile=channels.v3.inbound.cover.out -v ./channels/v3/inbound
-go test ${1} -coverprofile=connection.cover.out -v ./connection
-go test ${1} -coverprofile=policies.cover.out -v ./policies
-go test ${1} -coverprofile=identity.cover.out -v ./identity
-go test ${1} -coverprofile=root.cover.out -v ./
+GORACE="haltonerror=1"
+go test -race ${1} -coverprofile=utils.cover.out -v ./utils
+go test -race ${1} -coverprofile=channels.cover.out -v ./channels
+go test -race ${1} -coverprofile=channels.v3.inbound.cover.out -v ./channels/v3/inbound
+go test -race ${1} -coverprofile=connection.cover.out -v ./connection
+go test -race ${1} -coverprofile=policies.cover.out -v ./policies
+go test -race ${1} -coverprofile=identity.cover.out -v ./identity
+go test -race ${1} -coverprofile=root.cover.out -v ./
 echo "mode: set" > coverage.out && cat *.cover.out | grep -v mode: | sort -r | \
 awk '{if($1 != last) {print $0;last=$1}}' >> coverage.out
 rm -rf *.cover.out

utils/networking.go

@@ -8,6 +8,7 @@ import (
 	"git.openprivacy.ca/openprivacy/libricochet-go/log"
 	"golang.org/x/crypto/nacl/secretbox"
 	"io"
+	"sync"
 )
 
 const (
@@ -43,10 +44,13 @@ type RicochetNetwork struct {
 	// Derived ephemeral session key for connection
 	key     [32]byte
 	encrypt bool
+	lock    sync.Mutex
 }
 
 // SetEncryptionKey sets the ephemeral encryption key for this session.
 func (rn *RicochetNetwork) SetEncryptionKey(key [32]byte) {
+	rn.lock.Lock()
+	defer rn.lock.Unlock()
 	log.Debugf("turning on ephemeral session encryption for connection")
 	copy(rn.key[:], key[:])
 
@@ -67,6 +71,7 @@ func (rn *RicochetNetwork) SendRicochetPacket(dst io.Writer, channel int32, data
 	binary.BigEndian.PutUint16(packet[2:4], uint16(channel))
 	copy(packet[4:], data[:])
 
+	rn.lock.Lock()
 	if rn.encrypt {
 		var nonce [24]byte
 		if _, err := io.ReadFull(rand.Reader, nonce[:]); err != nil {
@@ -77,6 +82,7 @@ func (rn *RicochetNetwork) SendRicochetPacket(dst io.Writer, channel int32, data
 		binary.BigEndian.PutUint16(packet[0:2], uint16(len(encrypted)+2))
 		packet = append(packet[0:2], encrypted...)
 	}
+	rn.lock.Unlock()
 
 	for pos := 0; pos < len(packet); {
 		n, err := dst.Write(packet[pos:])
@@ -112,6 +118,7 @@ func (rn *RicochetNetwork) RecvRicochetPacket(reader io.Reader) (RicochetData, e
 		return packet, err
 	}
 
+	rn.lock.Lock()
 	if rn.encrypt {
 		var decryptNonce [24]byte
 		if len(packetBytes) > 24 {
@@ -127,6 +134,7 @@ func (rn *RicochetNetwork) RecvRicochetPacket(reader io.Reader) (RicochetData, e
 			return packet, errors.New("ciphertext length was too short")
 		}
 	}
+	rn.lock.Unlock()
 
 	packet.Channel = int32(binary.BigEndian.Uint16(packetBytes[0:2]))
 	packet.Data = make([]byte, len(packetBytes)-2)