X dumb way to check for decode overflow

2cd55b6f · Kirill Smelkov · 47999e52 · 2cd55b6f · 2cd55b6f
Commit 2cd55b6f authored Jan 20, 2017 by Kirill Smelkov
Expand all Show whitespace changes
Inline Side-by-side

Showing with 988 additions and 0 deletions

t/neo/marshal.go t/neo/marshal.go +987 -0

t/neo/protogen.go t/neo/protogen.go +1 -0

No files found.
--- a/t/neo/marshal.go
+++ b/t/neo/marshal.go
--- a/t/neo/protogen.go
+++ b/t/neo/protogen.go
@@ -297,6 +297,7 @@ func (e *encoder) genBasic(path string, typ *types.Basic, userType types.Type, o
 func (d *decoder) genBasic(assignto string, typ *types.Basic, userType types.Type, obj types.Object) {
 	basic := basicTypes[typ.Kind()]
+	d.emit("if len(data) < %v { goto overflow }", d.n + basic.wireSize)
 	dataptr := fmt.Sprintf("data[%v:]", d.n)
 	decoded := fmt.Sprintf(basic.decode, dataptr)
 	d.n += basic.wireSize