forked from vmware-archive/godhcpv4
/
validation.go
71 lines (58 loc) · 1.33 KB
/
validation.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
package dhcp4
import "fmt"
type Validation interface {
Validate(p Packet) error
}
func Validate(p Packet, vs []Validation) error {
for _, v := range vs {
if err := v.Validate(p); err != nil {
return err
}
}
return nil
}
type ValidationError struct {
Option
MustHave bool
}
func (e *ValidationError) Error() string {
if e.MustHave {
return fmt.Sprintf("dhcp4: packet MUST have field %d", e.Option)
}
return fmt.Sprintf("dhcp4: packet MUST NOT have field %d", e.Option)
}
type validateMust struct {
o Option
have bool
}
func (v validateMust) Validate(p Packet) error {
if _, ok := p.GetOption(v.o); v.have != ok {
return &ValidationError{Option: v.o, MustHave: v.have}
}
return nil
}
func ValidateMustNot(o Option) Validation {
return validateMust{o, false}
}
func ValidateMust(o Option) Validation {
return validateMust{o, true}
}
type validateAllowedOptions struct {
allowed map[Option]bool
}
func (v validateAllowedOptions) Validate(p Packet) error {
for k := range p.OptionMap {
// If an option is not allowed, the packet MUST NOT have it.
if !v.allowed[k] {
return &ValidationError{Option: k, MustHave: false}
}
}
return nil
}
func ValidateAllowedOptions(os []Option) Validation {
allowed := make(map[Option]bool)
for _, o := range os {
allowed[o] = true
}
return validateAllowedOptions{allowed}
}