-
Notifications
You must be signed in to change notification settings - Fork 25
/
sig-info.go
156 lines (140 loc) · 3.75 KB
/
sig-info.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
package ndn
import (
"encoding/hex"
"fmt"
"github.com/usnistgov/ndn-dpdk/ndn/an"
"github.com/usnistgov/ndn-dpdk/ndn/tlv"
)
// KeyLocator represents KeyLocator in SignatureInfo.
type KeyLocator struct {
Name Name
Digest []byte
}
// Empty returns true if KeyLocator has zero fields.
func (kl KeyLocator) Empty() bool {
return len(kl.Name)+len(kl.Digest) == 0
}
// MarshalTlv encodes this KeyLocator.
func (kl KeyLocator) MarshalTlv() (typ uint32, value []byte, e error) {
if len(kl.Name) > 0 && len(kl.Digest) > 0 {
return 0, nil, ErrKeyLocator
}
if len(kl.Digest) > 0 {
return tlv.EncodeTlv(an.TtKeyLocator, tlv.MakeElement(an.TtKeyDigest, kl.Digest))
}
return tlv.EncodeTlv(an.TtKeyLocator, kl.Name)
}
// UnmarshalBinary decodes from TLV-VALUE.
func (kl *KeyLocator) UnmarshalBinary(wire []byte) error {
*kl = KeyLocator{}
d := tlv.Decoder(wire)
for _, field := range d.Elements() {
switch field.Type {
case an.TtName:
if e := field.UnmarshalValue(&kl.Name); e != nil {
return e
}
case an.TtKeyDigest:
kl.Digest = field.Value
default:
if field.IsCriticalType() {
return tlv.ErrCritical
}
}
}
if len(kl.Name) > 0 && len(kl.Digest) > 0 {
return ErrKeyLocator
}
return d.ErrUnlessEOF()
}
func (kl KeyLocator) String() string {
if len(kl.Digest) > 0 {
return hex.EncodeToString(kl.Digest)
}
return kl.Name.String()
}
// SigInfo represents SignatureInfo on Interest or Data.
type SigInfo struct {
Type uint32
KeyLocator KeyLocator
Nonce []byte
Time uint64
SeqNum uint64
Extensions []tlv.Element
}
// EncodeAs creates an encodable object for either ISigInfo or DSigInfo TLV-TYPE.
// If si is nil, the encoding result contains SigType=SigNull.
func (si *SigInfo) EncodeAs(typ uint32) tlv.Marshaler {
return sigInfoMarshaler{typ, si}
}
// UnmarshalBinary decodes from TLV-VALUE.
func (si *SigInfo) UnmarshalBinary(wire []byte) error {
*si = SigInfo{}
d := tlv.Decoder(wire)
for _, field := range d.Elements() {
switch field.Type {
case an.TtSigType:
if e := field.UnmarshalNNI(&si.Type); e != nil {
return e
}
case an.TtKeyLocator:
if e := field.UnmarshalValue(&si.KeyLocator); e != nil {
return e
}
case an.TtSigNonce:
if field.Length() < 1 {
return ErrSigNonce
}
si.Nonce = field.Value
case an.TtSigTime:
if e := field.UnmarshalNNI(&si.Time); e != nil {
return e
}
case an.TtSigSeqNum:
if e := field.UnmarshalNNI(&si.SeqNum); e != nil {
return e
}
default:
if sigInfoExtensionTypes[field.Type] {
si.Extensions = append(si.Extensions, field.Element)
} else if field.IsCriticalType() {
return tlv.ErrCritical
}
}
}
return d.ErrUnlessEOF()
}
func (si SigInfo) String() string {
return fmt.Sprintf("%s:%v", an.SigTypeString(si.Type), si.KeyLocator)
}
type sigInfoMarshaler struct {
typ uint32
si *SigInfo
}
func (sim sigInfoMarshaler) MarshalTlv() (typ uint32, value []byte, e error) {
var fields []interface{}
if si := sim.si; si == nil {
fields = append(fields, tlv.MakeElementNNI(an.TtSigType, an.SigNull))
} else {
fields = append(fields, tlv.MakeElementNNI(an.TtSigType, si.Type))
if !si.KeyLocator.Empty() {
fields = append(fields, si.KeyLocator)
}
if si.Time > 0 {
fields = append(fields, tlv.MakeElementNNI(an.TtSigTime, si.Time))
}
if len(si.Nonce) > 0 {
fields = append(fields, tlv.MakeElement(an.TtSigNonce, si.Nonce))
}
if si.SeqNum > 0 {
fields = append(fields, tlv.MakeElementNNI(an.TtSigSeqNum, si.SeqNum))
}
fields = append(fields, si.Extensions)
}
return tlv.EncodeTlv(sim.typ, fields...)
}
var sigInfoExtensionTypes = make(map[uint32]bool)
// RegisterSigInfoExtension registers an extension TLV-TYPE in SigInfo.
func RegisterSigInfoExtension(typ uint32) {
sigInfoExtensionTypes[typ] = true
}