Improved timer state machine

1 files changed, 5 insertions, 8 deletions

diff --git a/src/device.go b/src/device.go
index 0564068..12d1ed9 100644
--- a/src/device.go
+++ b/src/device.go
@@ -31,16 +31,11 @@ type Device struct {
 	signal struct {
 		stop chan struct{}
 	}
-	congestionState int32 // used as an atomic bool
-	peers           map[NoisePublicKey]*Peer
-	mac             MACStateDevice
+	underLoad int32 // used as an atomic bool
+	peers     map[NoisePublicKey]*Peer
+	mac       MACStateDevice
 }
 
-const (
-	CongestionStateUnderLoad = iota
-	CongestionStateOkay
-)
-
 func (device *Device) SetPrivateKey(sk NoisePrivateKey) {
 	device.mutex.Lock()
 	defer device.mutex.Unlock()
@@ -99,10 +94,12 @@ func NewDevice(tun TUNDevice, logLevel int) *Device {
 		go device.RoutineDecryption()
 		go device.RoutineHandshake()
 	}
+
 	go device.RoutineBusyMonitor()
 	go device.RoutineReadFromTUN(tun)
 	go device.RoutineReceiveIncomming()
 	go device.RoutineWriteToTUN(tun)
+
 	return device
 }