1
2
3
4
5 package asn1
6
7 import (
8 "bytes"
9 "encoding/hex"
10 "math/big"
11 "reflect"
12 "strings"
13 "testing"
14 "time"
15 )
16
17 type intStruct struct {
18 A int
19 }
20
21 type twoIntStruct struct {
22 A int
23 B int
24 }
25
26 type bigIntStruct struct {
27 A *big.Int
28 }
29
30 type nestedStruct struct {
31 A intStruct
32 }
33
34 type rawContentsStruct struct {
35 Raw RawContent
36 A int
37 }
38
39 type implicitTagTest struct {
40 A int `asn1:"implicit,tag:5"`
41 }
42
43 type explicitTagTest struct {
44 A int `asn1:"explicit,tag:5"`
45 }
46
47 type flagTest struct {
48 A Flag `asn1:"tag:0,optional"`
49 }
50
51 type generalizedTimeTest struct {
52 A time.Time `asn1:"generalized"`
53 }
54
55 type ia5StringTest struct {
56 A string `asn1:"ia5"`
57 }
58
59 type printableStringTest struct {
60 A string `asn1:"printable"`
61 }
62
63 type genericStringTest struct {
64 A string
65 }
66
67 type optionalRawValueTest struct {
68 A RawValue `asn1:"optional"`
69 }
70
71 type omitEmptyTest struct {
72 A []string `asn1:"omitempty"`
73 }
74
75 type defaultTest struct {
76 A int `asn1:"optional,default:1"`
77 }
78
79 type applicationTest struct {
80 A int `asn1:"application,tag:0"`
81 B int `asn1:"application,tag:1,explicit"`
82 }
83
84 type privateTest struct {
85 A int `asn1:"private,tag:0"`
86 B int `asn1:"private,tag:1,explicit"`
87 C int `asn1:"private,tag:31"`
88 D int `asn1:"private,tag:128"`
89 }
90
91 type numericStringTest struct {
92 A string `asn1:"numeric"`
93 }
94
95 type testSET []int
96
97 var PST = time.FixedZone("PST", -8*60*60)
98
99 type marshalTest struct {
100 in interface{}
101 out string
102 }
103
104 func farFuture() time.Time {
105 t, err := time.Parse(time.RFC3339, "2100-04-05T12:01:01Z")
106 if err != nil {
107 panic(err)
108 }
109 return t
110 }
111
112 var marshalTests = []marshalTest{
113 {10, "02010a"},
114 {127, "02017f"},
115 {128, "02020080"},
116 {-128, "020180"},
117 {-129, "0202ff7f"},
118 {intStruct{64}, "3003020140"},
119 {bigIntStruct{big.NewInt(0x123456)}, "30050203123456"},
120 {twoIntStruct{64, 65}, "3006020140020141"},
121 {nestedStruct{intStruct{127}}, "3005300302017f"},
122 {[]byte{1, 2, 3}, "0403010203"},
123 {implicitTagTest{64}, "3003850140"},
124 {explicitTagTest{64}, "3005a503020140"},
125 {flagTest{true}, "30028000"},
126 {flagTest{false}, "3000"},
127 {time.Unix(0, 0).UTC(), "170d3730303130313030303030305a"},
128 {time.Unix(1258325776, 0).UTC(), "170d3039313131353232353631365a"},
129 {time.Unix(1258325776, 0).In(PST), "17113039313131353134353631362d30383030"},
130 {farFuture(), "180f32313030303430353132303130315a"},
131 {generalizedTimeTest{time.Unix(1258325776, 0).UTC()}, "3011180f32303039313131353232353631365a"},
132 {BitString{[]byte{0x80}, 1}, "03020780"},
133 {BitString{[]byte{0x81, 0xf0}, 12}, "03030481f0"},
134 {ObjectIdentifier([]int{1, 2, 3, 4}), "06032a0304"},
135 {ObjectIdentifier([]int{1, 2, 840, 133549, 1, 1, 5}), "06092a864888932d010105"},
136 {ObjectIdentifier([]int{2, 100, 3}), "0603813403"},
137 {"test", "130474657374"},
138 {
139 "" +
140 "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
141 "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
142 "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
143 "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",
144 "137f" +
145 "7878787878787878787878787878787878787878787878787878787878787878" +
146 "7878787878787878787878787878787878787878787878787878787878787878" +
147 "7878787878787878787878787878787878787878787878787878787878787878" +
148 "78787878787878787878787878787878787878787878787878787878787878",
149 },
150 {
151 "" +
152 "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
153 "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
154 "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" +
155 "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",
156 "138180" +
157 "7878787878787878787878787878787878787878787878787878787878787878" +
158 "7878787878787878787878787878787878787878787878787878787878787878" +
159 "7878787878787878787878787878787878787878787878787878787878787878" +
160 "7878787878787878787878787878787878787878787878787878787878787878",
161 },
162 {ia5StringTest{"test"}, "3006160474657374"},
163 {optionalRawValueTest{}, "3000"},
164 {printableStringTest{"test"}, "3006130474657374"},
165 {printableStringTest{"test*"}, "30071305746573742a"},
166 {genericStringTest{"test"}, "3006130474657374"},
167 {genericStringTest{"test*"}, "30070c05746573742a"},
168 {genericStringTest{"test&"}, "30070c057465737426"},
169 {rawContentsStruct{nil, 64}, "3003020140"},
170 {rawContentsStruct{[]byte{0x30, 3, 1, 2, 3}, 64}, "3003010203"},
171 {RawValue{Tag: 1, Class: 2, IsCompound: false, Bytes: []byte{1, 2, 3}}, "8103010203"},
172 {testSET([]int{10}), "310302010a"},
173 {omitEmptyTest{[]string{}}, "3000"},
174 {omitEmptyTest{[]string{"1"}}, "30053003130131"},
175 {"Σ", "0c02cea3"},
176 {defaultTest{0}, "3003020100"},
177 {defaultTest{1}, "3000"},
178 {defaultTest{2}, "3003020102"},
179 {applicationTest{1, 2}, "30084001016103020102"},
180 {privateTest{1, 2, 3, 4}, "3011c00101e103020102df1f0103df81000104"},
181 {numericStringTest{"1 9"}, "30051203312039"},
182 }
183
184 func TestMarshal(t *testing.T) {
185 for i, test := range marshalTests {
186 data, err := Marshal(test.in)
187 if err != nil {
188 t.Errorf("#%d failed: %s", i, err)
189 }
190 out, _ := hex.DecodeString(test.out)
191 if !bytes.Equal(out, data) {
192 t.Errorf("#%d got: %x want %x\n\t%q\n\t%q", i, data, out, data, out)
193
194 }
195 }
196 }
197
198 type marshalWithParamsTest struct {
199 in interface{}
200 params string
201 out string
202 }
203
204 var marshalWithParamsTests = []marshalWithParamsTest{
205 {intStruct{10}, "set", "310302010a"},
206 {intStruct{10}, "application", "600302010a"},
207 {intStruct{10}, "private", "e00302010a"},
208 }
209
210 func TestMarshalWithParams(t *testing.T) {
211 for i, test := range marshalWithParamsTests {
212 data, err := MarshalWithParams(test.in, test.params)
213 if err != nil {
214 t.Errorf("#%d failed: %s", i, err)
215 }
216 out, _ := hex.DecodeString(test.out)
217 if !bytes.Equal(out, data) {
218 t.Errorf("#%d got: %x want %x\n\t%q\n\t%q", i, data, out, data, out)
219
220 }
221 }
222 }
223
224 type marshalErrTest struct {
225 in interface{}
226 err string
227 }
228
229 var marshalErrTests = []marshalErrTest{
230 {bigIntStruct{nil}, "empty integer"},
231 {numericStringTest{"a"}, "invalid character"},
232 {ia5StringTest{"\xb0"}, "invalid character"},
233 {printableStringTest{"!"}, "invalid character"},
234 }
235
236 func TestMarshalError(t *testing.T) {
237 for i, test := range marshalErrTests {
238 _, err := Marshal(test.in)
239 if err == nil {
240 t.Errorf("#%d should fail, but success", i)
241 continue
242 }
243
244 if !strings.Contains(err.Error(), test.err) {
245 t.Errorf("#%d got: %v want %v", i, err, test.err)
246 }
247 }
248 }
249
250 func TestInvalidUTF8(t *testing.T) {
251 _, err := Marshal(string([]byte{0xff, 0xff}))
252 if err == nil {
253 t.Errorf("invalid UTF8 string was accepted")
254 }
255 }
256
257 func TestMarshalOID(t *testing.T) {
258 var marshalTestsOID = []marshalTest{
259 {[]byte("\x06\x01\x30"), "0403060130"},
260
261 {[]byte("\x06\x010"), "0403060130"},
262 {ObjectIdentifier([]int{2, 999, 3}), "0603883703"},
263 {ObjectIdentifier([]int{0, 0}), "060100"},
264 }
265 for i, test := range marshalTestsOID {
266 data, err := Marshal(test.in)
267 if err != nil {
268 t.Errorf("#%d failed: %s", i, err)
269 }
270 out, _ := hex.DecodeString(test.out)
271 if !bytes.Equal(out, data) {
272 t.Errorf("#%d got: %x want %x\n\t%q\n\t%q", i, data, out, data, out)
273 }
274 }
275 }
276
277 func TestIssue11130(t *testing.T) {
278 data := []byte("\x06\x010")
279 var v interface{}
280
281 _, err := Unmarshal(data, &v)
282 if err != nil {
283 t.Errorf("%v", err)
284 return
285 }
286 if reflect.TypeOf(v).String() != reflect.TypeOf(ObjectIdentifier{}).String() {
287 t.Errorf("marshal OID returned an invalid type")
288 return
289 }
290
291 data1, err := Marshal(v)
292 if err != nil {
293 t.Errorf("%v", err)
294 return
295 }
296
297 if !bytes.Equal(data, data1) {
298 t.Errorf("got: %q, want: %q \n", data1, data)
299 return
300 }
301
302 var v1 interface{}
303 _, err = Unmarshal(data1, &v1)
304 if err != nil {
305 t.Errorf("%v", err)
306 return
307 }
308 if !reflect.DeepEqual(v, v1) {
309 t.Errorf("got: %#v data=%q, want : %#v data=%q\n ", v1, data1, v, data)
310 }
311 }
312
313 func BenchmarkMarshal(b *testing.B) {
314 b.ReportAllocs()
315
316 for i := 0; i < b.N; i++ {
317 for _, test := range marshalTests {
318 Marshal(test.in)
319 }
320 }
321 }
322
323 func TestSetEncoder(t *testing.T) {
324 testStruct := struct {
325 Strings []string `asn1:"set"`
326 }{
327 Strings: []string{"a", "aa", "b", "bb", "c", "cc"},
328 }
329
330
331
332
333 output, err := Marshal(testStruct)
334 if err != nil {
335 t.Errorf("%v", err)
336 }
337
338 expectedOrder := []string{"a", "b", "c", "aa", "bb", "cc"}
339 var resultStruct struct {
340 Strings []string `asn1:"set"`
341 }
342 rest, err := Unmarshal(output, &resultStruct)
343 if err != nil {
344 t.Errorf("%v", err)
345 }
346 if len(rest) != 0 {
347 t.Error("Unmarshal returned extra garbage")
348 }
349 if !reflect.DeepEqual(expectedOrder, resultStruct.Strings) {
350 t.Errorf("Unexpected SET content. got: %s, want: %s", resultStruct.Strings, expectedOrder)
351 }
352 }
353
354 func TestSetEncoderSETSliceSuffix(t *testing.T) {
355 type testSetSET []string
356 testSet := testSetSET{"a", "aa", "b", "bb", "c", "cc"}
357
358
359
360
361 output, err := Marshal(testSet)
362 if err != nil {
363 t.Errorf("%v", err)
364 }
365
366 expectedOrder := testSetSET{"a", "b", "c", "aa", "bb", "cc"}
367 var resultSet testSetSET
368 rest, err := Unmarshal(output, &resultSet)
369 if err != nil {
370 t.Errorf("%v", err)
371 }
372 if len(rest) != 0 {
373 t.Error("Unmarshal returned extra garbage")
374 }
375 if !reflect.DeepEqual(expectedOrder, resultSet) {
376 t.Errorf("Unexpected SET content. got: %s, want: %s", resultSet, expectedOrder)
377 }
378 }
379
380 func BenchmarkUnmarshal(b *testing.B) {
381 b.ReportAllocs()
382
383 type testCase struct {
384 in []byte
385 out interface{}
386 }
387 var testData []testCase
388 for _, test := range unmarshalTestData {
389 pv := reflect.New(reflect.TypeOf(test.out).Elem())
390 inCopy := make([]byte, len(test.in))
391 copy(inCopy, test.in)
392 outCopy := pv.Interface()
393
394 testData = append(testData, testCase{
395 in: inCopy,
396 out: outCopy,
397 })
398 }
399
400 b.ResetTimer()
401 for i := 0; i < b.N; i++ {
402 for _, testCase := range testData {
403 _, _ = Unmarshal(testCase.in, testCase.out)
404 }
405 }
406 }
407
View as plain text