TheArchive/tailscale
1
0
Fork 0
mirror of https://github.com/tailscale/tailscale.git synced 2025-03-29 04:22:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

types/key: add missing ChallengePublic.UnmarshalText

Browse Source 
Forgot it when adding the Challenge types earlier.

Change-Id: Ie0872c4e6dc25e5d832aa58c7b3f66d450bf6b71
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
This commit is contained in:
Brad Fitzpatrick 2022-10-27 10:34:07 -07:00 committed by Brad Fitzpatrick
parent 8c09ae9032
commit 4021ae6b9d
2 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
types/key/chal.go
View File

@ -7,6 +7,7 @@ package key
import ( import (
"errors" "errors"
"go4.org/mem"
"tailscale.com/types/structs" "tailscale.com/types/structs"
) )
@ -76,3 +77,8 @@ func (k ChallengePublic) String() string {
func (k ChallengePublic) MarshalText() ([]byte, error) { func (k ChallengePublic) MarshalText() ([]byte, error) {
return toHex(k.k[:], chalPublicHexPrefix), nil return toHex(k.k[:], chalPublicHexPrefix), nil
} }
// UnmarshalText implements encoding.TextUnmarshaler.
func (k *ChallengePublic) UnmarshalText(b []byte) error {
return parseHex(k.k[:], mem.B(b), mem.S(chalPublicHexPrefix))
}

16
types/key/node_test.go
View File

@ -142,3 +142,19 @@ func TestNodeWriteRawWithoutAllocating(t *testing.T) {
t.Fatalf("WriteRawWithoutAllocating got %f allocs, want %f", got, want) t.Fatalf("WriteRawWithoutAllocating got %f allocs, want %f", got, want)
} }
} }
func TestChallenge(t *testing.T) {
priv := NewChallenge()
pub := priv.Public()
txt, err := pub.MarshalText()
if err != nil {
t.Fatal(err)
}
var back ChallengePublic
if err := back.UnmarshalText(txt); err != nil {
t.Fatal(err)
}
if back != pub {
t.Errorf("didn't round trip: %v != %v", back, pub)
}
}