fix: overflow

The ID should not be allowed to overflow an int64

snowflake.go

@@ -230,6 +230,9 @@ func ParseBase32(b []byte) (ID, error) {
 			return -1, ErrInvalidBase32
 		}
 		id = id*32 + int64(decodeBase32Map[b[i]])
+		if id <= 0 {
+			return -1, ErrInvalidBase32
+		}
 	}
 
 	return ID(id), nil
@@ -277,6 +280,10 @@ func ParseBase58(b []byte) (ID, error) {
 			return -1, ErrInvalidBase58
 		}
 		id = id*58 + int64(decodeBase58Map[b[i]])
+		if id <= 0 {
+			// overflow!
+			return -1, ErrInvalidBase58
+		}
 	}
 
 	return ID(id), nil

snowflake_test.go

@@ -2,6 +2,7 @@ package snowflake
 
 import (
 	"bytes"
+	"fmt"
 	"reflect"
 	"testing"
 )
@@ -512,6 +513,12 @@ func TestParseBase32(t *testing.T) {
 			want:    -1,
 			wantErr: true,
 		},
+		{
+			name:    "overflow is invalid",
+			arg:     "byyyyyyyyyyyyy",
+			want:    -1,
+			wantErr: true,
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
@@ -525,6 +532,8 @@ func TestParseBase32(t *testing.T) {
 			}
 		})
 	}
+
+	fmt.Println(ID(int64(0xfffffffffffffff) + 1).Base32())
 }
 
 func TestParseBase58(t *testing.T) {
@@ -564,7 +573,14 @@ func TestParseBase58(t *testing.T) {
 			want:    -1,
 			wantErr: true,
 		},
+		{
+			name:    "overflow is invalid",
+			arg:     "JPwcyDCgEuq",
+			want:    -1,
+			wantErr: true,
+		},
 	}
+
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			got, err := ParseBase58([]byte(tt.arg))