forked from ferranbt/fastssz
/
validate.go
61 lines (56 loc) · 983 Bytes
/
validate.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
package main
func (v *Value) validate() string {
switch v.t {
case TypeBitList, TypeBytes:
cmp := "!="
if v.t == TypeBitList {
cmp = ">"
}
if v.c {
return ""
}
// fixed []byte
size := v.s
if size == 0 {
// dynamic []byte
size = v.m
cmp = ">"
}
tmpl := `if len(::.{{.name}}) {{.cmp}} {{.size}} {
err = ssz.ErrBytesLength
return
}
`
return execTmpl(tmpl, map[string]interface{}{
"cmp": cmp,
"name": v.name,
"size": size,
})
case TypeVector:
if v.c {
return ""
}
// We only have vectors for [][]byte roots
tmpl := `if len(::.{{.name}}) != {{.size}} {
err = ssz.ErrVectorLength
return
}
`
return execTmpl(tmpl, map[string]interface{}{
"name": v.name,
"size": v.s,
})
case TypeList:
tmpl := `if len(::.{{.name}}) > {{.size}} {
err = ssz.ErrListTooBig
return
}
`
return execTmpl(tmpl, map[string]interface{}{
"name": v.name,
"size": v.s,
})
default:
return ""
}
}