Fix panic unmarshalling hello verify request

This could cause us to panic when unmarshalling a Hello Verify request message.
pion · Feb 5, 2023 · a50d26c · a50d26c
1 parent 7a14903
commit a50d26c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/protocol/handshake/message_hello_verify_request.go b/pkg/protocol/handshake/message_hello_verify_request.go
@@ -51,8 +51,8 @@ func (m *MessageHelloVerifyRequest) Unmarshal(data []byte) error {
 	}
 	m.Version.Major = data[0]
 	m.Version.Minor = data[1]
-	cookieLength := data[2]
-	if len(data) < (int(cookieLength) + 3) {
+	cookieLength := int(data[2])
+	if len(data) < cookieLength+3 {
 		return errBufferTooSmall
 	}
 	m.Cookie = make([]byte, cookieLength)