Skip to content
Snippets Groups Projects
oid_test.go 3.7 KiB
Newer Older
  • Learn to ignore specific revisions
  • // Copyright 2023 The Go Authors. All rights reserved.
    // Use of this source code is governed by a BSD-style
    // license that can be found in the LICENSE file.
    
    package x509
    
    import (
    	"encoding/asn1"
    	"math"
    	"testing"
    )
    
    func TestOID(t *testing.T) {
    	var tests = []struct {
    		raw   []byte
    		valid bool
    		str   string
    		ints  []uint64
    	}{
    		{[]byte{}, false, "", nil},
    		{[]byte{0x80, 0x01}, false, "", nil},
    		{[]byte{0x01, 0x80, 0x01}, false, "", nil},
    
    		{[]byte{1, 2, 3}, true, "0.1.2.3", []uint64{0, 1, 2, 3}},
    		{[]byte{41, 2, 3}, true, "1.1.2.3", []uint64{1, 1, 2, 3}},
    		{[]byte{86, 2, 3}, true, "2.6.2.3", []uint64{2, 6, 2, 3}},
    
    		{[]byte{41, 255, 255, 255, 127}, true, "1.1.268435455", []uint64{1, 1, 268435455}},
    		{[]byte{41, 0x87, 255, 255, 255, 127}, true, "1.1.2147483647", []uint64{1, 1, 2147483647}},
    		{[]byte{41, 255, 255, 255, 255, 127}, true, "1.1.34359738367", []uint64{1, 1, 34359738367}},
    		{[]byte{42, 255, 255, 255, 255, 255, 255, 255, 255, 127}, true, "1.2.9223372036854775807", []uint64{1, 2, 9223372036854775807}},
    		{[]byte{43, 0x81, 255, 255, 255, 255, 255, 255, 255, 255, 127}, true, "1.3.18446744073709551615", []uint64{1, 3, 18446744073709551615}},
    		{[]byte{44, 0x83, 255, 255, 255, 255, 255, 255, 255, 255, 127}, true, "1.4.36893488147419103231", nil},
    		{[]byte{85, 255, 255, 255, 255, 255, 255, 255, 255, 255, 127}, true, "2.5.1180591620717411303423", nil},
    		{[]byte{85, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 127}, true, "2.5.19342813113834066795298815", nil},
    
    		{[]byte{255, 255, 255, 127}, true, "2.268435375", []uint64{2, 268435375}},
    		{[]byte{0x87, 255, 255, 255, 127}, true, "2.2147483567", []uint64{2, 2147483567}},
    		{[]byte{255, 127}, true, "2.16303", []uint64{2, 16303}},
    		{[]byte{255, 255, 255, 255, 127}, true, "2.34359738287", []uint64{2, 34359738287}},
    		{[]byte{255, 255, 255, 255, 255, 255, 255, 255, 127}, true, "2.9223372036854775727", []uint64{2, 9223372036854775727}},
    		{[]byte{0x81, 255, 255, 255, 255, 255, 255, 255, 255, 127}, true, "2.18446744073709551535", []uint64{2, 18446744073709551535}},
    		{[]byte{0x83, 255, 255, 255, 255, 255, 255, 255, 255, 127}, true, "2.36893488147419103151", nil},
    		{[]byte{255, 255, 255, 255, 255, 255, 255, 255, 255, 127}, true, "2.1180591620717411303343", nil},
    		{[]byte{255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 127}, true, "2.19342813113834066795298735", nil},
    	}
    
    	for _, v := range tests {
    		oid, ok := newOIDFromDER(v.raw)
    		if ok != v.valid {
    			if ok {
    				t.Errorf("%v: unexpected success while parsing: %v", v.raw, oid)
    			} else {
    				t.Errorf("%v: unexpected failure while parsing", v.raw)
    			}
    			continue
    		}
    
    		if !ok {
    			continue
    		}
    
    		if str := oid.String(); str != v.str {
    			t.Errorf("%v: oid.String() = %v, want; %v", v.raw, str, v.str)
    		}
    
    		var asn1OID asn1.ObjectIdentifier
    		for _, v := range v.ints {
    			if v > math.MaxInt32 {
    				asn1OID = nil
    				break
    			}
    			asn1OID = append(asn1OID, int(v))
    		}
    
    		o, ok := oid.toASN1OID()
    		if shouldOk := asn1OID != nil; shouldOk != ok {
    			if ok {
    				t.Errorf("%v: oid.toASN1OID() unexpected success", v.raw)
    			} else {
    
    Jes Cok's avatar
    Jes Cok committed
    				t.Errorf("%v: oid.toASN1OID() unexpected failure", v.raw)
    
    			}
    			continue
    		}
    
    		if asn1OID != nil {
    			if !o.Equal(asn1OID) {
    				t.Errorf("%v: oid.toASN1OID(asn1OID).Equal(oid) = false, want: true", v.raw)
    			}
    		}
    
    		if v.ints != nil {
    			oid2, err := OIDFromInts(v.ints)
    			if err != nil {
    				t.Errorf("%v: OIDFromInts() unexpected error: %v", v.raw, err)
    			}
    			if !oid2.Equal(oid) {
    				t.Errorf("%v: %#v.Equal(%#v) = false, want: true", v.raw, oid2, oid)
    			}
    		}
    	}
    }
    
    func mustNewOIDFromInts(t *testing.T, ints []uint64) OID {
    	oid, err := OIDFromInts(ints)
    	if err != nil {
    		t.Fatalf("OIDFromInts(%v) unexpected error: %v", ints, err)
    	}
    	return oid
    }