From cfcf5c87c35805277b14312b6344ff6079a99e8d Mon Sep 17 00:00:00 2001
From: albexk <albexk@proton.me>
Date: Tue, 9 Apr 2024 21:45:50 +0300
Subject: [PATCH] Add resetting of message types when closing the device

---
 device/device.go | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/device/device.go b/device/device.go
index 21a6546..a8a9e2f 100644
--- a/device/device.go
+++ b/device/device.go
@@ -415,6 +415,8 @@ func (device *Device) Close() {
 
 	device.rate.limiter.Close()
 
+	device.resetProtocol()
+
 	device.log.Verbosef("Device closed")
 	close(device.closed)
 }
@@ -559,14 +561,17 @@ func (device *Device) isAdvancedSecurityOn() bool {
 	return device.isASecOn.IsSet()
 }
 
+func (device *Device) resetProtocol() {
+	// restore default message type values
+	MessageInitiationType = 1
+	MessageResponseType = 2
+	MessageCookieReplyType = 3
+	MessageTransportType = 4
+}
+
 func (device *Device) handlePostConfig(tempASecCfg *aSecCfgType) (err error) {
 
 	if !tempASecCfg.isSet {
-		// restore default values
-		MessageInitiationType = 1
-		MessageResponseType = 2
-		MessageCookieReplyType = 3
-		MessageTransportType = 4
 		return err
 	}