Home | History | Annotate | Download | only in asn1 
      1 // Copyright 2009 The Go Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style
      3 // license that can be found in the LICENSE file.
      4 
      5 package asn1
      6 
      7 import (
      8 	"bytes"
      9 	"encoding/hex"
     10 	"math/big"
     11 	"testing"
     12 	"time"
     13 )
     14 
     15 type intStruct struct {
     16 	A int
     17 }
     18 
     19 type twoIntStruct struct {
     20 	A int
     21 	B int
     22 }
     23 
     24 type bigIntStruct struct {
     25 	A *big.Int
     26 }
     27 
     28 type nestedStruct struct {
     29 	A intStruct
     30 }
     31 
     32 type rawContentsStruct struct {
     33 	Raw RawContent
     34 	A   int
     35 }
     36 
     37 type implicitTagTest struct {
     38 	A int `asn1:"implicit,tag:5"`
     39 }
     40 
     41 type explicitTagTest struct {
     42 	A int `asn1:"explicit,tag:5"`
     43 }
     44 
     45 type flagTest struct {
     46 	A Flag `asn1:"tag:0,optional"`
     47 }
     48 
     49 type generalizedTimeTest struct {
     50 	A time.Time `asn1:"generalized"`
     51 }
     52 
     53 type ia5StringTest struct {
     54 	A string `asn1:"ia5"`
     55 }
     56 
     57 type printableStringTest struct {
     58 	A string `asn1:"printable"`
     59 }
     60 
     61 type optionalRawValueTest struct {
     62 	A RawValue `asn1:"optional"`
     63 }
     64 
     65 type omitEmptyTest struct {
     66 	A []string `asn1:"omitempty"`
     67 }
     68 
     69 type defaultTest struct {
     70 	A int `asn1:"optional,default:1"`
     71 }
     72 
     73 type testSET []int
     74 
     75 var PST = time.FixedZone("PST", -8*60*60)
     76 
     77 type marshalTest struct {
     78 	in  interface{}
     79 	out string // hex encoded
     80 }
     81 
     82 func farFuture() time.Time {
     83 	t, err := time.Parse(time.RFC3339, "2100-04-05T12:01:01Z")
     84 	if err != nil {
     85 		panic(err)
     86 	}
     87 	return t
     88 }
     89 
     90 var marshalTests = []marshalTest{
     91 	{10, "02010a"},
     92 	{127, "02017f"},
     93 	{128, "02020080"},
     94 	{-128, "020180"},
     95 	{-129, "0202ff7f"},
     96 	{intStruct{64}, "3003020140"},
     97 	{bigIntStruct{big.NewInt(0x123456)}, "30050203123456"},
     98 	{twoIntStruct{64, 65}, "3006020140020141"},
     99 	{nestedStruct{intStruct{127}}, "3005300302017f"},
    100 	{[]byte{1, 2, 3}, "0403010203"},
    101 	{implicitTagTest{64}, "3003850140"},
    102 	{explicitTagTest{64}, "3005a503020140"},
    103 	{flagTest{true}, "30028000"},
    104 	{flagTest{false}, "3000"},
    105 	{time.Unix(0, 0).UTC(), "170d3730303130313030303030305a"},
    106 	{time.Unix(1258325776, 0).UTC(), "170d3039313131353232353631365a"},
    107 	{time.Unix(1258325776, 0).In(PST), "17113039313131353134353631362d30383030"},
    108 	{farFuture(), "180f32313030303430353132303130315a"},
    109 	{generalizedTimeTest{time.Unix(1258325776, 0).UTC()}, "3011180f32303039313131353232353631365a"},
    110 	{BitString{[]byte{0x80}, 1}, "03020780"},
    111 	{BitString{[]byte{0x81, 0xf0}, 12}, "03030481f0"},
    112 	{ObjectIdentifier([]int{1, 2, 3, 4}), "06032a0304"},
    113 	{ObjectIdentifier([]int{1, 2, 840, 133549, 1, 1, 5}), "06092a864888932d010105"},
    114 	{ObjectIdentifier([]int{2, 100, 3}), "0603813403"},
    115 	{"test", "130474657374"},
    116 	{
    117 		"" +
    118 			"xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
    119 			"xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
    120 			"xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
    121 			"xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx", // This is 127 times 'x'
    122 		"137f" +
    123 			"7878787878787878787878787878787878787878787878787878787878787878" +
    124 			"7878787878787878787878787878787878787878787878787878787878787878" +
    125 			"7878787878787878787878787878787878787878787878787878787878787878" +
    126 			"78787878787878787878787878787878787878787878787878787878787878",
    127 	},
    128 	{
    129 		"" +
    130 			"xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
    131 			"xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
    132 			"xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
    133 			"xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx", // This is 128 times 'x'
    134 		"138180" +
    135 			"7878787878787878787878787878787878787878787878787878787878787878" +
    136 			"7878787878787878787878787878787878787878787878787878787878787878" +
    137 			"7878787878787878787878787878787878787878787878787878787878787878" +
    138 			"7878787878787878787878787878787878787878787878787878787878787878",
    139 	},
    140 	{ia5StringTest{"test"}, "3006160474657374"},
    141 	{optionalRawValueTest{}, "3000"},
    142 	{printableStringTest{"test"}, "3006130474657374"},
    143 	{printableStringTest{"test*"}, "30071305746573742a"},
    144 	{rawContentsStruct{nil, 64}, "3003020140"},
    145 	{rawContentsStruct{[]byte{0x30, 3, 1, 2, 3}, 64}, "3003010203"},
    146 	{RawValue{Tag: 1, Class: 2, IsCompound: false, Bytes: []byte{1, 2, 3}}, "8103010203"},
    147 	{testSET([]int{10}), "310302010a"},
    148 	{omitEmptyTest{[]string{}}, "3000"},
    149 	{omitEmptyTest{[]string{"1"}}, "30053003130131"},
    150 	{"", "0c02cea3"},
    151 	{defaultTest{0}, "3003020100"},
    152 	{defaultTest{1}, "3000"},
    153 	{defaultTest{2}, "3003020102"},
    154 }
    155 
    156 func TestMarshal(t *testing.T) {
    157 	for i, test := range marshalTests {
    158 		data, err := Marshal(test.in)
    159 		if err != nil {
    160 			t.Errorf("#%d failed: %s", i, err)
    161 		}
    162 		out, _ := hex.DecodeString(test.out)
    163 		if !bytes.Equal(out, data) {
    164 			t.Errorf("#%d got: %x want %x\n\t%q\n\t%q", i, data, out, data, out)
    165 
    166 		}
    167 	}
    168 }
    169 
    170 func TestInvalidUTF8(t *testing.T) {
    171 	_, err := Marshal(string([]byte{0xff, 0xff}))
    172 	if err == nil {
    173 		t.Errorf("invalid UTF8 string was accepted")
    174 	}
    175 }
    176