-
Notifications
You must be signed in to change notification settings - Fork 16
/
basepolicies.go
52 lines (42 loc) · 1.04 KB
/
basepolicies.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
package ipset
import (
"github.com/ti-mo/netfilter"
)
type BasePolicy struct {
Protocol *UInt8Box
}
func newBasePolicy() BasePolicy {
return BasePolicy{Protocol: NewUInt8Box(Protocol)}
}
func (p BasePolicy) marshalAttributes() Attributes {
attrs := newAttributes()
attrs.append(AttrProtocol, p.Protocol)
return attrs
}
func (p *BasePolicy) unmarshalAttribute(nfa netfilter.Attribute) {
if at := AttributeType(nfa.Type); at == AttrProtocol {
p.Protocol = unmarshalUInt8Box(nfa)
}
}
type NamePolicy struct {
BasePolicy
Name *NullStringBox
}
func newNamePolicy(name string) NamePolicy {
return NamePolicy{
BasePolicy: newBasePolicy(),
Name: NewNullStringBox(name),
}
}
func (p NamePolicy) marshalAttributes() Attributes {
attrs := p.BasePolicy.marshalAttributes()
attrs.append(AttrSetName, p.Name)
return attrs
}
func (p *NamePolicy) unmarshalAttribute(nfa netfilter.Attribute) {
if at := AttributeType(nfa.Type); at == AttrSetName {
p.Name = unmarshalNullStringBox(nfa)
} else {
p.BasePolicy.unmarshalAttribute(nfa)
}
}