Updated protocol spec to new V2 version

1 files changed, 18 insertions, 4 deletions

diff --git a/internal/packet/packet.go b/internal/packet/packet.go
index 2285065..8bc9669 100644
--- a/internal/packet/packet.go
+++ b/internal/packet/packet.go
@@ -60,6 +60,9 @@ const (
 
 	PacketAuthenticate
 
+	PacketUsersInfo
+	PacketGetUsers
+
 	PacketSetUserData
 	PacketGetUserData
 
@@ -94,9 +97,6 @@ const (
 	PacketBlockUser
 	PacketBlockInfo
 
-	PacketGetUsers
-	PacketUsersInfo
-
 	PacketMax
 )
 
@@ -269,6 +269,19 @@ func (p Packet) DecodedPayload() (Payload, error) {
 	case PacketError:
 		payload = &Error{}
 
+	case PacketTosInfo:
+		payload = &TosInfo{}
+	case PacketAcceptTos:
+		payload = &AcceptTos{}
+
+	case PacketGetNonce:
+		payload = &GetNonce{}
+	case PacketNonceInfo:
+		payload = &NonceInfo{}
+
+	case PacketAuthenticate:
+		payload = &Authenticate{}
+
 	case PacketSetUserData:
 		payload = &SetUserData{}
 	case PacketGetUserData:
@@ -335,7 +348,8 @@ func (p Packet) DecodedPayload() (Payload, error) {
 		payload = &UsersInfo{}
 
 	default:
-		assert.Never("unexpected packet.PacketType", "type", p.Type())
+		assert.Assert(!p.Type().IsSupported(), "supported PackeType wasn't handled", "type", p.Type())
+		return nil, fmt.Errorf("unsupported PackeType: %v", p.Type().String())
 	}
 	err := p.DecodePayloadInto(payload)
 	return payload, err