extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2024-12-24 15:48:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

improved code for token checksum calc

Browse Source
This commit is contained in:
Pascal Fischer 2023-03-03 14:51:33 +01:00
parent e0fc779f58
commit b3f339c753
2 changed files with 2 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
management/server/personal_access_token.go
View File

@ -22,20 +22,6 @@ type PersonalAccessToken struct {
LastUsed time.Time
}
const (
// IEEE is by far and away the most common CRC-32 polynomial.
// Used by ethernet (IEEE 802.3), v.42, fddi, gzip, zip, png, ...
IEEE = 0xedb88320
// Castagnoli polynomial, used in iSCSI.
// Has better error detection characteristics than IEEE.
// https://dx.doi.org/10.1109/26.231911
Castagnoli = 0x82f63b78
// Koopman polynomial.
// Also has better error detection characteristics than IEEE.
// https://dx.doi.org/10.1109/DSN.2002.1028931
Koopman = 0xeb31d82e
)
func CreateNewPAT(description string, expirationInDays int, createdBy User) (*PersonalAccessToken, string) {
hashedToken, plainToken := generateNewToken()
currentTime := time.Now().UTC()
@ -53,8 +39,7 @@ func CreateNewPAT(description string, expirationInDays int, createdBy User) (*Pe
func generateNewToken() ([32]byte, string) {
token := randStringRunes(30)
crc32q := crc32.MakeTable(IEEE)
checksum := crc32.Checksum([]byte(token), crc32q)
checksum := crc32.ChecksumIEEE([]byte(token))
encodedChecksum := base62.Encode(checksum)
paddedChecksum := fmt.Sprintf("%06s", encodedChecksum)
plainToken := "nbp_" + token + paddedChecksum

3
management/server/personal_access_token_test.go
View File

@ -31,8 +31,7 @@ func TestPAT_GenerateToken_Checksum(t *testing.T) {
token := tokenWithoutPrefix[:len(tokenWithoutPrefix)-6]
tokenCheckSum := tokenWithoutPrefix[len(tokenWithoutPrefix)-6:]
crc32q := crc32.MakeTable(IEEE)
expectedChecksum := crc32.Checksum([]byte(token), crc32q)
expectedChecksum := crc32.ChecksumIEEE([]byte(token))
actualChecksum, err := base62.Decode(tokenCheckSum)
if err != nil {
t.Fatal(err)