diff options
author | Jeffrey Wilcke <geffobscura@gmail.com> | 2016-05-11 19:21:25 +0800 |
---|---|---|
committer | Jeffrey Wilcke <geffobscura@gmail.com> | 2016-05-11 19:36:27 +0800 |
commit | 91a7a4a7867718ccb6c9620120a1be5680ad0abd (patch) | |
tree | f8b6b278efe61a68c6a6c0bc88d0b7ca792ccd9d /accounts/abi/abi.go | |
parent | 5782164a35ea8acdb09507a604c45941051fd5f3 (diff) | |
download | dexon-91a7a4a7867718ccb6c9620120a1be5680ad0abd.tar.gz dexon-91a7a4a7867718ccb6c9620120a1be5680ad0abd.tar.zst dexon-91a7a4a7867718ccb6c9620120a1be5680ad0abd.zip |
accounts/abi: fixed unpacking in to already slice interfaces
Previously it was assumed that wheneven type `[]interface{}` was given
that the interface was empty. The abigen rightfully assumed that
interface slices which already have pre-allocated variable sets to be
assigned.
This PR fixes that by checking that the given `[]interface{}` is larger
than zero and assigns each value using the generic `set` function (this
function has also been moved to abi/reflect.go) and checks whether the
assignment was possible.
The generic assignment function `set` now also deals with pointers
(useful for interface slice mentioned above) by dereferencing the
pointer until it finds a setable type.
Diffstat (limited to 'accounts/abi/abi.go')
-rw-r--r-- | accounts/abi/abi.go | 59 |
1 files changed, 29 insertions, 30 deletions
diff --git a/accounts/abi/abi.go b/accounts/abi/abi.go index 32df6f19d..1b07b2f68 100644 --- a/accounts/abi/abi.go +++ b/accounts/abi/abi.go @@ -238,8 +238,16 @@ func (abi ABI) Unpack(v interface{}, name string, output []byte) error { return fmt.Errorf("abi: unmarshalling empty output") } - value := reflect.ValueOf(v).Elem() - typ := value.Type() + // make sure the passed value is a pointer + valueOf := reflect.ValueOf(v) + if reflect.Ptr != valueOf.Kind() { + return fmt.Errorf("abi: Unpack(non-pointer %T)", v) + } + + var ( + value = valueOf.Elem() + typ = value.Type() + ) if len(method.Outputs) > 1 { switch value.Kind() { @@ -268,6 +276,25 @@ func (abi ABI) Unpack(v interface{}, name string, output []byte) error { return fmt.Errorf("abi: cannot marshal tuple in to slice %T (only []interface{} is supported)", v) } + // if the slice already contains values, set those instead of the interface slice itself. + if value.Len() > 0 { + if len(method.Outputs) > value.Len() { + return fmt.Errorf("abi: cannot marshal in to slices of unequal size (require: %v, got: %v)", len(method.Outputs), value.Len()) + } + + for i := 0; i < len(method.Outputs); i++ { + marshalledValue, err := toGoType(i, method.Outputs[i], output) + if err != nil { + return err + } + reflectValue := reflect.ValueOf(marshalledValue) + if err := set(value.Index(i).Elem(), reflectValue, method.Outputs[i]); err != nil { + return err + } + } + return nil + } + // create a new slice and start appending the unmarshalled // values to the new interface slice. z := reflect.MakeSlice(typ, 0, len(method.Outputs)) @@ -296,34 +323,6 @@ func (abi ABI) Unpack(v interface{}, name string, output []byte) error { return nil } -// set attempts to assign src to dst by either setting, copying or otherwise. -// -// set is a bit more lenient when it comes to assignment and doesn't force an as -// strict ruleset as bare `reflect` does. -func set(dst, src reflect.Value, output Argument) error { - dstType := dst.Type() - srcType := src.Type() - - switch { - case dstType.AssignableTo(src.Type()): - dst.Set(src) - case dstType.Kind() == reflect.Array && srcType.Kind() == reflect.Slice: - if !dstType.Elem().AssignableTo(r_byte) { - return fmt.Errorf("abi: cannot unmarshal %v in to array of elem %v", src.Type(), dstType.Elem()) - } - - if dst.Len() < output.Type.SliceSize { - return fmt.Errorf("abi: cannot unmarshal src (len=%d) in to dst (len=%d)", output.Type.SliceSize, dst.Len()) - } - reflect.Copy(dst, src) - case dstType.Kind() == reflect.Interface: - dst.Set(src) - default: - return fmt.Errorf("abi: cannot unmarshal %v in to %v", src.Type(), dst.Type()) - } - return nil -} - func (abi *ABI) UnmarshalJSON(data []byte) error { var fields []struct { Type string |