generate valid tokens for remote addresses that are not UDP addresses

2017-05-13 14:51:41 +08:00
parent 8e01921495
commit eb72b494b2
6 changed files with 70 additions and 59 deletions
--- a/crypto/source_address_token.go
+++ b/crypto/source_address_token.go
@@ -37,8 +37,8 @@ func (t *sourceAddressToken) serialize() []byte {
 }

 func parseToken(data []byte) (*sourceAddressToken, error) {
-	if len(data) != 8+4 && len(data) != 8+16 {
-		return nil, fmt.Errorf("invalid STK length: %d", len(data))
+	if len(data) < 8 {
+		return nil, fmt.Errorf("STK too short: %d", len(data))
 	}
 	return &sourceAddressToken{
 		data:      data[8:],