Fix GetVersion

This commit is contained in:
Thomas LAY
2020-05-05 16:11:16 +02:00
parent cf02b3ef5c
commit 3ac524b657
2 changed files with 7 additions and 2 deletions

View File

@@ -18,7 +18,12 @@ func GetVersion(s string) (version TcfVersion, err error) {
} }
}() }()
b, err := base64.RawURLEncoding.DecodeString(s) segments := strings.Split(s, ".")
if len(segments) == 0 {
return TcfVersionUndefined, err
}
b, err := base64.RawURLEncoding.DecodeString(segments[0])
if err != nil { if err != nil {
return TcfVersionUndefined, err return TcfVersionUndefined, err
} }

View File

@@ -15,7 +15,7 @@ func TestGetVersion(t *testing.T) {
t.Errorf("Version should be %d", TcfVersion1) t.Errorf("Version should be %d", TcfVersion1)
} }
version, err = GetVersion("COxR03kOxR1CqBcABCENAgCMAP_AAH_AAAqIF3EXySoGY2thI2YVFxBEIYwfJxyigMgChgQIsSwNQIeFLBoGLiAAHBGYJAQAGBAEEACBAQIkHGBMCQAAgAgBiRCMQEGMCzNIBIBAggEbY0FACCVmHkHSmZCY7064O__QLuIJEFQMAkSBAIACLECIQwAQDiAAAYAlAAABAhIaAAgIWBQEeAAAACAwAAgAAABBAAACAAQAAICIAAABAAAgAiAQAAAAGgIQAACBABACRIAAAEANCAAgiCEAQg4EAo4AAA") version, err = GetVersion("COy7f9HOy7f_1BcABBENAjCoAPKAAFKAAAqIDaQCQABAAVAAyACAAFoANQAkgEdANoA2kAYAAQAFQAMgAgABaAbQAUMAQAAEABUADIAIAAWgBJgDCAMQA9ACEAEdAKuAXUAwIBhADRAG0FAEQABAAVAAyACAAFoANQAkwBhAGIAegBCACOgFXALqAYEAwgBogDaDACIAAgAKgAZABAAC0AGoASYAwgDEAPQAhABHQCrgF1AMCAYQA0QBtCABEAAQAFQAMgAgABaADUAJMAYQBiAHoAQgAjoBVwC6gGBAMIAaIA2hQAiAAIACoAGQAQAAtABqAEmAMIAxAD0AIQAR0Aq4BdQDAgGEANEAbQA.cAAACAAAAUg")
if err != nil { if err != nil {
t.Errorf("Version should be decoded without error: %s", err) t.Errorf("Version should be decoded without error: %s", err)
return return