Commit b4a96d1c authored by Levin Zimmermann's avatar Levin Zimmermann

go/proto/msgpack: Simplify debugging by printing function name in case of overflow

parent 012305fe
......@@ -44,7 +44,7 @@ func (e *mstructDecodeError) Error() string {
// mdecodeErr is called to normilize error when msgp.ReadXXX returns err when decoding path.
func mdecodeErr(path string, err error) error {
if err == msgp.ErrShortBytes {
return ErrDecodeOverflow
return &ErrDecodeOverflow{path}
}
return &mdecodeError{path, err}
}
......
......@@ -228,3 +228,12 @@ func (addr Address) String() string {
return net.JoinHostPort(addr.Host, fmt.Sprintf("%d", addr.Port))
}
}
// ErrDecodeOverflow is the error returned by neoMsgDecode when decoding hits buffer overflow
type ErrDecodeOverflow struct {
function string
}
func (e *ErrDecodeOverflow) Error() string {
return fmt.Sprintf("decode '%s': buffer overflow", e.function)
}
......@@ -70,7 +70,6 @@ import (
"lab.nexedi.com/kirr/neo/go/internal/packed"
"encoding/binary"
"errors"
"math"
)
......@@ -168,9 +167,6 @@ func (e Encoding) MsgDecode(msg Msg, data []byte) (nread int, err error) {
}
// ErrDecodeOverflow is the error returned by neoMsgDecode when decoding hits buffer overflow
var ErrDecodeOverflow = errors.New("decode: buffer overflow")
// ---- messages ----
//neo:proto enum
......
......@@ -22,6 +22,7 @@ package proto
import (
"encoding/binary"
"errors"
hexpkg "encoding/hex"
"fmt"
"reflect"
......@@ -430,9 +431,10 @@ func TestMsgDecodeLenOverflowN(t *testing.T) {
}()
n, err := enc.MsgDecode(tt.msg, data)
if !(n == 0 && err == ErrDecodeOverflow) {
errDecodeOverflow := new(ErrDecodeOverflow)
if !(n == 0 && errors.As(err, &errDecodeOverflow)) {
t.Errorf("%T: decode %x\nhave: %d, %v\nwant: %d, %v", tt.msg, data,
n, err, 0, ErrDecodeOverflow)
n, err, 0, errDecodeOverflow)
}
}()
}
......
......@@ -917,7 +917,7 @@ func (d *decoderCommon) generatedCode() string {
// NOTE for >0 check actual X in StdSizes{X} does not particularly matter
if ((&types.StdSizes{8, 8}).Sizeof(d.typ) > 0 || d.enc != 'N') && d.overflow.gotoEmitted {
code.emit("\noverflow:")
code.emit("return 0, ErrDecodeOverflow")
code.emit("return 0, &ErrDecodeOverflow{\"%s\"}", d.typeName)
}
code.emit("}\n")
......
This diff is collapsed.
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment