2015-07-07 03:54:22 +03:00
|
|
|
// Copyright 2014 The go-ethereum Authors
|
2015-07-22 19:48:40 +03:00
|
|
|
// This file is part of the go-ethereum library.
|
2015-07-07 03:54:22 +03:00
|
|
|
//
|
2015-07-23 19:35:11 +03:00
|
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
2015-07-07 03:54:22 +03:00
|
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
|
|
// (at your option) any later version.
|
|
|
|
//
|
2015-07-22 19:48:40 +03:00
|
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
2015-07-07 03:54:22 +03:00
|
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
2015-07-22 19:48:40 +03:00
|
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
2015-07-07 03:54:22 +03:00
|
|
|
// GNU Lesser General Public License for more details.
|
|
|
|
//
|
|
|
|
// You should have received a copy of the GNU Lesser General Public License
|
2015-07-22 19:48:40 +03:00
|
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
2015-07-07 03:54:22 +03:00
|
|
|
|
2014-11-13 21:31:48 +02:00
|
|
|
package rlp
|
|
|
|
|
|
|
|
import (
|
2014-11-24 20:01:25 +02:00
|
|
|
"bufio"
|
2015-03-18 14:17:39 +02:00
|
|
|
"bytes"
|
2014-11-13 21:31:48 +02:00
|
|
|
"encoding/binary"
|
|
|
|
"errors"
|
|
|
|
"fmt"
|
|
|
|
"io"
|
|
|
|
"math/big"
|
|
|
|
"reflect"
|
2015-03-23 16:02:55 +02:00
|
|
|
"strings"
|
2019-02-25 13:34:08 +03:00
|
|
|
"sync"
|
2014-11-13 21:31:48 +02:00
|
|
|
)
|
|
|
|
|
2019-11-25 15:41:53 +03:00
|
|
|
//lint:ignore ST1012 EOL is not an error.
|
|
|
|
|
|
|
|
// EOL is returned when the end of the current list
|
|
|
|
// has been reached during streaming.
|
|
|
|
var EOL = errors.New("rlp: end of list")
|
2018-05-08 12:48:07 +03:00
|
|
|
|
2019-11-25 15:41:53 +03:00
|
|
|
var (
|
2018-05-08 12:48:07 +03:00
|
|
|
ErrExpectedString = errors.New("rlp: expected String or Byte")
|
|
|
|
ErrExpectedList = errors.New("rlp: expected List")
|
|
|
|
ErrCanonInt = errors.New("rlp: non-canonical integer format")
|
|
|
|
ErrCanonSize = errors.New("rlp: non-canonical size information")
|
|
|
|
ErrElemTooLarge = errors.New("rlp: element is larger than containing list")
|
|
|
|
ErrValueTooLarge = errors.New("rlp: value size exceeds available input length")
|
|
|
|
ErrMoreThanOneValue = errors.New("rlp: input contains more than one value")
|
|
|
|
|
|
|
|
// internal errors
|
|
|
|
errNotInList = errors.New("rlp: call of ListEnd outside of any list")
|
|
|
|
errNotAtEOL = errors.New("rlp: call of ListEnd not positioned at EOL")
|
|
|
|
errUintOverflow = errors.New("rlp: uint overflow")
|
2014-11-13 21:31:48 +02:00
|
|
|
errNoPointer = errors.New("rlp: interface given to Decode must be a pointer")
|
|
|
|
errDecodeIntoNil = errors.New("rlp: pointer given to Decode must not be nil")
|
2019-02-25 13:34:08 +03:00
|
|
|
|
|
|
|
streamPool = sync.Pool{
|
|
|
|
New: func() interface{} { return new(Stream) },
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
)
|
|
|
|
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
// Decoder is implemented by types that require custom RLP decoding rules or need to decode
|
|
|
|
// into private fields.
|
2014-11-13 21:31:48 +02:00
|
|
|
//
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
// The DecodeRLP method should read one value from the given Stream. It is not forbidden to
|
|
|
|
// read less or more, but it might be confusing.
|
2014-11-13 21:31:48 +02:00
|
|
|
type Decoder interface {
|
|
|
|
DecodeRLP(*Stream) error
|
|
|
|
}
|
|
|
|
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
// Decode parses RLP-encoded data from r and stores the result in the value pointed to by
|
|
|
|
// val. Please see package-level documentation for the decoding rules. Val must be a
|
|
|
|
// non-nil pointer.
|
2014-11-13 21:31:48 +02:00
|
|
|
//
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
// If r does not implement ByteReader, Decode will do its own buffering.
|
2014-11-13 21:31:48 +02:00
|
|
|
//
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
// Note that Decode does not set an input limit for all readers and may be vulnerable to
|
|
|
|
// panics cause by huge value sizes. If you need an input limit, use
|
2015-03-23 16:02:55 +02:00
|
|
|
//
|
|
|
|
// NewStream(r, limit).Decode(val)
|
2014-11-24 20:01:25 +02:00
|
|
|
func Decode(r io.Reader, val interface{}) error {
|
2019-02-25 13:34:08 +03:00
|
|
|
stream := streamPool.Get().(*Stream)
|
|
|
|
defer streamPool.Put(stream)
|
|
|
|
|
|
|
|
stream.Reset(r, 0)
|
|
|
|
return stream.Decode(val)
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
// DecodeBytes parses RLP data from b into val. Please see package-level documentation for
|
|
|
|
// the decoding rules. The input must contain exactly one value and no trailing data.
|
2015-03-18 14:17:39 +02:00
|
|
|
func DecodeBytes(b []byte, val interface{}) error {
|
2015-07-18 02:47:17 +03:00
|
|
|
r := bytes.NewReader(b)
|
2019-02-25 13:34:08 +03:00
|
|
|
|
|
|
|
stream := streamPool.Get().(*Stream)
|
|
|
|
defer streamPool.Put(stream)
|
|
|
|
|
|
|
|
stream.Reset(r, uint64(len(b)))
|
|
|
|
if err := stream.Decode(val); err != nil {
|
2015-07-18 02:47:17 +03:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
if r.Len() > 0 {
|
|
|
|
return ErrMoreThanOneValue
|
|
|
|
}
|
|
|
|
return nil
|
2015-03-18 14:17:39 +02:00
|
|
|
}
|
|
|
|
|
2014-11-25 17:00:48 +02:00
|
|
|
type decodeError struct {
|
|
|
|
msg string
|
|
|
|
typ reflect.Type
|
2015-01-05 18:10:42 +02:00
|
|
|
ctx []string
|
2014-11-25 17:00:48 +02:00
|
|
|
}
|
|
|
|
|
2015-01-05 18:10:42 +02:00
|
|
|
func (err *decodeError) Error() string {
|
|
|
|
ctx := ""
|
|
|
|
if len(err.ctx) > 0 {
|
|
|
|
ctx = ", decoding into "
|
|
|
|
for i := len(err.ctx) - 1; i >= 0; i-- {
|
|
|
|
ctx += err.ctx[i]
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return fmt.Sprintf("rlp: %s for %v%s", err.msg, err.typ, ctx)
|
2014-11-25 17:00:48 +02:00
|
|
|
}
|
|
|
|
|
2014-12-09 14:55:47 +02:00
|
|
|
func wrapStreamError(err error, typ reflect.Type) error {
|
|
|
|
switch err {
|
2015-04-04 22:29:23 +03:00
|
|
|
case ErrCanonInt:
|
2015-04-14 17:09:06 +03:00
|
|
|
return &decodeError{msg: "non-canonical integer (leading zero bytes)", typ: typ}
|
|
|
|
case ErrCanonSize:
|
|
|
|
return &decodeError{msg: "non-canonical size information", typ: typ}
|
2014-12-09 14:55:47 +02:00
|
|
|
case ErrExpectedList:
|
2015-01-05 18:10:42 +02:00
|
|
|
return &decodeError{msg: "expected input list", typ: typ}
|
2014-12-09 14:55:47 +02:00
|
|
|
case ErrExpectedString:
|
2015-01-05 18:10:42 +02:00
|
|
|
return &decodeError{msg: "expected input string or byte", typ: typ}
|
2014-12-09 14:55:47 +02:00
|
|
|
case errUintOverflow:
|
2015-01-05 18:10:42 +02:00
|
|
|
return &decodeError{msg: "input string too long", typ: typ}
|
2014-12-09 14:55:47 +02:00
|
|
|
case errNotAtEOL:
|
2015-01-05 18:10:42 +02:00
|
|
|
return &decodeError{msg: "input list has too many elements", typ: typ}
|
|
|
|
}
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
func addErrorContext(err error, ctx string) error {
|
|
|
|
if decErr, ok := err.(*decodeError); ok {
|
|
|
|
decErr.ctx = append(decErr.ctx, ctx)
|
2014-12-09 14:55:47 +02:00
|
|
|
}
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2014-12-09 11:39:39 +02:00
|
|
|
var (
|
|
|
|
decoderInterface = reflect.TypeOf(new(Decoder)).Elem()
|
|
|
|
bigInt = reflect.TypeOf(big.Int{})
|
|
|
|
)
|
|
|
|
|
2015-04-17 02:16:46 +03:00
|
|
|
func makeDecoder(typ reflect.Type, tags tags) (dec decoder, err error) {
|
2014-12-09 11:39:39 +02:00
|
|
|
kind := typ.Kind()
|
|
|
|
switch {
|
2015-09-09 04:34:15 +03:00
|
|
|
case typ == rawValueType:
|
|
|
|
return decodeRawValue, nil
|
2014-12-09 11:39:39 +02:00
|
|
|
case typ.AssignableTo(reflect.PtrTo(bigInt)):
|
|
|
|
return decodeBigInt, nil
|
|
|
|
case typ.AssignableTo(bigInt):
|
|
|
|
return decodeBigIntNoPtr, nil
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
case kind == reflect.Ptr:
|
|
|
|
return makePtrDecoder(typ, tags)
|
|
|
|
case reflect.PtrTo(typ).Implements(decoderInterface):
|
|
|
|
return decodeDecoder, nil
|
2014-12-09 11:41:47 +02:00
|
|
|
case isUint(kind):
|
|
|
|
return decodeUint, nil
|
2015-08-13 11:12:38 +03:00
|
|
|
case kind == reflect.Bool:
|
|
|
|
return decodeBool, nil
|
2014-12-09 11:39:39 +02:00
|
|
|
case kind == reflect.String:
|
|
|
|
return decodeString, nil
|
|
|
|
case kind == reflect.Slice || kind == reflect.Array:
|
2015-12-21 22:05:20 +02:00
|
|
|
return makeListDecoder(typ, tags)
|
2014-12-09 11:39:39 +02:00
|
|
|
case kind == reflect.Struct:
|
|
|
|
return makeStructDecoder(typ)
|
2015-01-16 00:21:41 +02:00
|
|
|
case kind == reflect.Interface:
|
2014-12-09 11:39:39 +02:00
|
|
|
return decodeInterface, nil
|
|
|
|
default:
|
|
|
|
return nil, fmt.Errorf("rlp: type %v is not RLP-serializable", typ)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-09-09 04:34:15 +03:00
|
|
|
func decodeRawValue(s *Stream, val reflect.Value) error {
|
|
|
|
r, err := s.Raw()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
val.SetBytes(r)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2014-11-13 21:31:48 +02:00
|
|
|
func decodeUint(s *Stream, val reflect.Value) error {
|
2014-11-25 17:00:48 +02:00
|
|
|
typ := val.Type()
|
|
|
|
num, err := s.uint(typ.Bits())
|
2014-12-09 14:55:47 +02:00
|
|
|
if err != nil {
|
|
|
|
return wrapStreamError(err, val.Type())
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
val.SetUint(num)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-08-13 11:12:38 +03:00
|
|
|
func decodeBool(s *Stream, val reflect.Value) error {
|
|
|
|
b, err := s.Bool()
|
|
|
|
if err != nil {
|
|
|
|
return wrapStreamError(err, val.Type())
|
|
|
|
}
|
|
|
|
val.SetBool(b)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2014-11-13 21:31:48 +02:00
|
|
|
func decodeString(s *Stream, val reflect.Value) error {
|
|
|
|
b, err := s.Bytes()
|
|
|
|
if err != nil {
|
2014-12-09 14:55:47 +02:00
|
|
|
return wrapStreamError(err, val.Type())
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
val.SetString(string(b))
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func decodeBigIntNoPtr(s *Stream, val reflect.Value) error {
|
|
|
|
return decodeBigInt(s, val.Addr())
|
|
|
|
}
|
|
|
|
|
|
|
|
func decodeBigInt(s *Stream, val reflect.Value) error {
|
|
|
|
b, err := s.Bytes()
|
|
|
|
if err != nil {
|
2014-12-09 14:55:47 +02:00
|
|
|
return wrapStreamError(err, val.Type())
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
i := val.Interface().(*big.Int)
|
|
|
|
if i == nil {
|
|
|
|
i = new(big.Int)
|
|
|
|
val.Set(reflect.ValueOf(i))
|
|
|
|
}
|
2015-04-14 17:09:06 +03:00
|
|
|
// Reject leading zero bytes
|
2015-04-04 22:29:23 +03:00
|
|
|
if len(b) > 0 && b[0] == 0 {
|
|
|
|
return wrapStreamError(ErrCanonInt, val.Type())
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
i.SetBytes(b)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-12-21 22:05:20 +02:00
|
|
|
func makeListDecoder(typ reflect.Type, tag tags) (decoder, error) {
|
2014-11-13 21:31:48 +02:00
|
|
|
etype := typ.Elem()
|
|
|
|
if etype.Kind() == reflect.Uint8 && !reflect.PtrTo(etype).Implements(decoderInterface) {
|
|
|
|
if typ.Kind() == reflect.Array {
|
|
|
|
return decodeByteArray, nil
|
|
|
|
}
|
2018-05-08 12:48:07 +03:00
|
|
|
return decodeByteSlice, nil
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
2019-05-14 16:09:56 +03:00
|
|
|
etypeinfo := cachedTypeInfo1(etype, tags{})
|
|
|
|
if etypeinfo.decoderErr != nil {
|
|
|
|
return nil, etypeinfo.decoderErr
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
2015-12-21 22:05:20 +02:00
|
|
|
var dec decoder
|
|
|
|
switch {
|
|
|
|
case typ.Kind() == reflect.Array:
|
|
|
|
dec = func(s *Stream, val reflect.Value) error {
|
2015-01-05 18:10:42 +02:00
|
|
|
return decodeListArray(s, val, etypeinfo.decoder)
|
2015-12-21 22:05:20 +02:00
|
|
|
}
|
|
|
|
case tag.tail:
|
|
|
|
// A slice with "tail" tag can occur as the last field
|
2017-06-12 15:45:17 +03:00
|
|
|
// of a struct and is supposed to swallow all remaining
|
2015-12-21 22:05:20 +02:00
|
|
|
// list elements. The struct decoder already called s.List,
|
|
|
|
// proceed directly to decoding the elements.
|
|
|
|
dec = func(s *Stream, val reflect.Value) error {
|
|
|
|
return decodeSliceElems(s, val, etypeinfo.decoder)
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
dec = func(s *Stream, val reflect.Value) error {
|
2015-01-05 18:10:42 +02:00
|
|
|
return decodeListSlice(s, val, etypeinfo.decoder)
|
|
|
|
}
|
2015-12-21 22:05:20 +02:00
|
|
|
}
|
|
|
|
return dec, nil
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
|
2014-12-09 11:44:48 +02:00
|
|
|
func decodeListSlice(s *Stream, val reflect.Value, elemdec decoder) error {
|
2014-11-13 21:31:48 +02:00
|
|
|
size, err := s.List()
|
|
|
|
if err != nil {
|
2014-12-09 14:55:47 +02:00
|
|
|
return wrapStreamError(err, val.Type())
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
if size == 0 {
|
2014-12-09 11:44:48 +02:00
|
|
|
val.Set(reflect.MakeSlice(val.Type(), 0, 0))
|
2014-11-13 21:31:48 +02:00
|
|
|
return s.ListEnd()
|
|
|
|
}
|
2015-12-21 22:05:20 +02:00
|
|
|
if err := decodeSliceElems(s, val, elemdec); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return s.ListEnd()
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
|
2015-12-21 22:05:20 +02:00
|
|
|
func decodeSliceElems(s *Stream, val reflect.Value, elemdec decoder) error {
|
2014-11-13 21:31:48 +02:00
|
|
|
i := 0
|
2014-12-09 11:44:48 +02:00
|
|
|
for ; ; i++ {
|
|
|
|
// grow slice if necessary
|
|
|
|
if i >= val.Cap() {
|
|
|
|
newcap := val.Cap() + val.Cap()/2
|
|
|
|
if newcap < 4 {
|
|
|
|
newcap = 4
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
2014-12-09 11:44:48 +02:00
|
|
|
newv := reflect.MakeSlice(val.Type(), val.Len(), newcap)
|
|
|
|
reflect.Copy(newv, val)
|
|
|
|
val.Set(newv)
|
|
|
|
}
|
|
|
|
if i >= val.Len() {
|
|
|
|
val.SetLen(i + 1)
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
// decode into element
|
|
|
|
if err := elemdec(s, val.Index(i)); err == EOL {
|
|
|
|
break
|
|
|
|
} else if err != nil {
|
2015-01-05 18:10:42 +02:00
|
|
|
return addErrorContext(err, fmt.Sprint("[", i, "]"))
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if i < val.Len() {
|
2014-12-09 11:44:48 +02:00
|
|
|
val.SetLen(i)
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
2015-12-21 22:05:20 +02:00
|
|
|
return nil
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
|
2014-12-09 11:44:48 +02:00
|
|
|
func decodeListArray(s *Stream, val reflect.Value, elemdec decoder) error {
|
2015-12-21 22:05:20 +02:00
|
|
|
if _, err := s.List(); err != nil {
|
2015-04-14 17:09:06 +03:00
|
|
|
return wrapStreamError(err, val.Type())
|
2014-12-09 11:44:48 +02:00
|
|
|
}
|
|
|
|
vlen := val.Len()
|
|
|
|
i := 0
|
|
|
|
for ; i < vlen; i++ {
|
|
|
|
if err := elemdec(s, val.Index(i)); err == EOL {
|
|
|
|
break
|
|
|
|
} else if err != nil {
|
2015-01-05 18:10:42 +02:00
|
|
|
return addErrorContext(err, fmt.Sprint("[", i, "]"))
|
2014-12-09 11:44:48 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if i < vlen {
|
2015-04-17 04:11:24 +03:00
|
|
|
return &decodeError{msg: "input list has too few elements", typ: val.Type()}
|
2014-12-09 11:44:48 +02:00
|
|
|
}
|
2014-12-09 14:55:47 +02:00
|
|
|
return wrapStreamError(s.ListEnd(), val.Type())
|
2014-12-09 11:44:48 +02:00
|
|
|
}
|
|
|
|
|
2014-11-13 21:31:48 +02:00
|
|
|
func decodeByteSlice(s *Stream, val reflect.Value) error {
|
|
|
|
b, err := s.Bytes()
|
2015-04-16 11:15:14 +03:00
|
|
|
if err != nil {
|
|
|
|
return wrapStreamError(err, val.Type())
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
2015-04-16 11:15:14 +03:00
|
|
|
val.SetBytes(b)
|
|
|
|
return nil
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func decodeByteArray(s *Stream, val reflect.Value) error {
|
|
|
|
kind, size, err := s.Kind()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2015-04-17 04:11:24 +03:00
|
|
|
vlen := val.Len()
|
2014-11-13 21:31:48 +02:00
|
|
|
switch kind {
|
|
|
|
case Byte:
|
2015-04-17 04:11:24 +03:00
|
|
|
if vlen == 0 {
|
2015-01-05 18:10:42 +02:00
|
|
|
return &decodeError{msg: "input string too long", typ: val.Type()}
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
2015-04-17 04:11:24 +03:00
|
|
|
if vlen > 1 {
|
|
|
|
return &decodeError{msg: "input string too short", typ: val.Type()}
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
bv, _ := s.Uint()
|
|
|
|
val.Index(0).SetUint(bv)
|
|
|
|
case String:
|
2015-04-17 04:11:24 +03:00
|
|
|
if uint64(vlen) < size {
|
2015-01-05 18:10:42 +02:00
|
|
|
return &decodeError{msg: "input string too long", typ: val.Type()}
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
2015-04-17 04:11:24 +03:00
|
|
|
if uint64(vlen) > size {
|
|
|
|
return &decodeError{msg: "input string too short", typ: val.Type()}
|
|
|
|
}
|
|
|
|
slice := val.Slice(0, vlen).Interface().([]byte)
|
2014-11-13 21:31:48 +02:00
|
|
|
if err := s.readFull(slice); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2015-04-16 11:15:14 +03:00
|
|
|
// Reject cases where single byte encoding should have been used.
|
2015-07-18 17:13:20 +03:00
|
|
|
if size == 1 && slice[0] < 128 {
|
2015-04-16 11:15:14 +03:00
|
|
|
return wrapStreamError(ErrCanonSize, val.Type())
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
case List:
|
2015-04-17 03:01:38 +03:00
|
|
|
return wrapStreamError(ErrExpectedString, val.Type())
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func makeStructDecoder(typ reflect.Type) (decoder, error) {
|
2014-12-30 12:40:32 +02:00
|
|
|
fields, err := structFields(typ)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
for _, f := range fields {
|
|
|
|
if f.info.decoderErr != nil {
|
|
|
|
return nil, structFieldError{typ, f.index, f.info.decoderErr}
|
|
|
|
}
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
dec := func(s *Stream, val reflect.Value) (err error) {
|
2015-12-21 22:05:20 +02:00
|
|
|
if _, err := s.List(); err != nil {
|
2014-12-09 14:55:47 +02:00
|
|
|
return wrapStreamError(err, typ)
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
for _, f := range fields {
|
2015-12-21 22:05:20 +02:00
|
|
|
err := f.info.decoder(s, val.Field(f.index))
|
2014-11-13 21:31:48 +02:00
|
|
|
if err == EOL {
|
2015-04-17 02:16:46 +03:00
|
|
|
return &decodeError{msg: "too few elements", typ: typ}
|
2014-11-13 21:31:48 +02:00
|
|
|
} else if err != nil {
|
2015-01-05 18:10:42 +02:00
|
|
|
return addErrorContext(err, "."+typ.Field(f.index).Name)
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
}
|
2014-12-09 14:55:47 +02:00
|
|
|
return wrapStreamError(s.ListEnd(), typ)
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
return dec, nil
|
|
|
|
}
|
|
|
|
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
// makePtrDecoder creates a decoder that decodes into the pointer's element type.
|
|
|
|
func makePtrDecoder(typ reflect.Type, tag tags) (decoder, error) {
|
2014-11-13 21:31:48 +02:00
|
|
|
etype := typ.Elem()
|
2019-05-14 16:09:56 +03:00
|
|
|
etypeinfo := cachedTypeInfo1(etype, tags{})
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
switch {
|
|
|
|
case etypeinfo.decoderErr != nil:
|
2019-05-14 16:09:56 +03:00
|
|
|
return nil, etypeinfo.decoderErr
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
case !tag.nilOK:
|
|
|
|
return makeSimplePtrDecoder(etype, etypeinfo), nil
|
|
|
|
default:
|
|
|
|
return makeNilPtrDecoder(etype, etypeinfo, tag.nilKind), nil
|
2015-04-17 02:16:46 +03:00
|
|
|
}
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
func makeSimplePtrDecoder(etype reflect.Type, etypeinfo *typeinfo) decoder {
|
|
|
|
return func(s *Stream, val reflect.Value) (err error) {
|
2015-04-17 02:16:46 +03:00
|
|
|
newval := val
|
|
|
|
if val.IsNil() {
|
|
|
|
newval = reflect.New(etype)
|
|
|
|
}
|
|
|
|
if err = etypeinfo.decoder(s, newval.Elem()); err == nil {
|
|
|
|
val.Set(newval)
|
|
|
|
}
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
// makeNilPtrDecoder creates a decoder that decodes empty values as nil. Non-empty
|
|
|
|
// values are decoded into a value of the element type, just like makePtrDecoder does.
|
2015-04-17 02:16:46 +03:00
|
|
|
//
|
|
|
|
// This decoder is used for pointer-typed struct fields with struct tag "nil".
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
func makeNilPtrDecoder(etype reflect.Type, etypeinfo *typeinfo, nilKind Kind) decoder {
|
|
|
|
typ := reflect.PtrTo(etype)
|
|
|
|
nilPtr := reflect.Zero(typ)
|
|
|
|
return func(s *Stream, val reflect.Value) (err error) {
|
2015-04-14 13:28:19 +03:00
|
|
|
kind, size, err := s.Kind()
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
if err != nil {
|
|
|
|
val.Set(nilPtr)
|
|
|
|
return wrapStreamError(err, typ)
|
|
|
|
}
|
|
|
|
// Handle empty values as a nil pointer.
|
|
|
|
if kind != Byte && size == 0 {
|
|
|
|
if kind != nilKind {
|
|
|
|
return &decodeError{
|
|
|
|
msg: fmt.Sprintf("wrong kind of empty value (got %v, want %v)", kind, nilKind),
|
|
|
|
typ: typ,
|
|
|
|
}
|
|
|
|
}
|
2015-03-20 23:33:40 +02:00
|
|
|
// rearm s.Kind. This is important because the input
|
|
|
|
// position must advance to the next value even though
|
|
|
|
// we don't read anything.
|
|
|
|
s.kind = -1
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
val.Set(nilPtr)
|
|
|
|
return nil
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
newval := val
|
|
|
|
if val.IsNil() {
|
|
|
|
newval = reflect.New(etype)
|
|
|
|
}
|
|
|
|
if err = etypeinfo.decoder(s, newval.Elem()); err == nil {
|
|
|
|
val.Set(newval)
|
|
|
|
}
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
var ifsliceType = reflect.TypeOf([]interface{}{})
|
|
|
|
|
|
|
|
func decodeInterface(s *Stream, val reflect.Value) error {
|
2015-01-16 00:21:41 +02:00
|
|
|
if val.Type().NumMethod() != 0 {
|
|
|
|
return fmt.Errorf("rlp: type %v is not RLP-serializable", val.Type())
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
kind, _, err := s.Kind()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if kind == List {
|
|
|
|
slice := reflect.New(ifsliceType).Elem()
|
2014-12-09 11:44:48 +02:00
|
|
|
if err := decodeListSlice(s, slice, decodeInterface); err != nil {
|
2014-11-13 21:31:48 +02:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
val.Set(slice)
|
|
|
|
} else {
|
|
|
|
b, err := s.Bytes()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
val.Set(reflect.ValueOf(b))
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func decodeDecoder(s *Stream, val reflect.Value) error {
|
rlp: improve nil pointer handling (#20064)
* rlp: improve nil pointer handling
In both encoder and decoder, the rules for encoding nil pointers were a
bit hard to understand, and didn't leave much choice. Since RLP allows
two empty values (empty list, empty string), any protocol built on RLP
must choose either of these values to represent the null value in a
certain context.
This change adds choice in the form of two new struct tags, "nilString"
and "nilList". These can be used to specify how a nil pointer value is
encoded. The "nil" tag still exists, but its implementation is now
explicit and defines exactly how nil pointers are handled in a single
place.
Another important change in this commit is how nil pointers and the
Encoder interface interact. The EncodeRLP method was previously called
even on nil values, which was supposed to give users a choice of how
their value would be handled when nil. It turns out this is a stupid
idea. If you create a network protocol containing an object defined in
another package, it's better to be able to say that the object should be
a list or string when nil in the definition of the protocol message
rather than defining the encoding of nil on the object itself.
As of this commit, the encoding rules for pointers now take precedence
over the Encoder interface rule. I think the "nil" tag will work fine
for most cases. For special kinds of objects which are a struct in Go
but strings in RLP, code using the object can specify the desired
encoding of nil using the "nilString" and "nilList" tags.
* rlp: propagate struct field type errors
If a struct contained fields of undecodable type, the encoder and
decoder would panic instead of returning an error. Fix this by
propagating type errors in makeStruct{Writer,Decoder} and add a test.
2019-09-13 12:10:57 +03:00
|
|
|
return val.Addr().Interface().(Decoder).DecodeRLP(s)
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
// Kind represents the kind of value contained in an RLP stream.
|
|
|
|
type Kind int
|
|
|
|
|
|
|
|
const (
|
|
|
|
Byte Kind = iota
|
|
|
|
String
|
|
|
|
List
|
|
|
|
)
|
|
|
|
|
|
|
|
func (k Kind) String() string {
|
|
|
|
switch k {
|
|
|
|
case Byte:
|
|
|
|
return "Byte"
|
|
|
|
case String:
|
|
|
|
return "String"
|
|
|
|
case List:
|
|
|
|
return "List"
|
|
|
|
default:
|
|
|
|
return fmt.Sprintf("Unknown(%d)", k)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// ByteReader must be implemented by any input reader for a Stream. It
|
|
|
|
// is implemented by e.g. bufio.Reader and bytes.Reader.
|
|
|
|
type ByteReader interface {
|
|
|
|
io.Reader
|
|
|
|
io.ByteReader
|
|
|
|
}
|
|
|
|
|
|
|
|
// Stream can be used for piecemeal decoding of an input stream. This
|
|
|
|
// is useful if the input is very large or if the decoding rules for a
|
|
|
|
// type depend on the input structure. Stream does not keep an
|
|
|
|
// internal buffer. After decoding a value, the input reader will be
|
|
|
|
// positioned just before the type information for the next value.
|
|
|
|
//
|
|
|
|
// When decoding a list and the input position reaches the declared
|
|
|
|
// length of the list, all operations will return error EOL.
|
|
|
|
// The end of the list must be acknowledged using ListEnd to continue
|
|
|
|
// reading the enclosing list.
|
|
|
|
//
|
|
|
|
// Stream is not safe for concurrent use.
|
|
|
|
type Stream struct {
|
2015-03-23 16:02:55 +02:00
|
|
|
r ByteReader
|
|
|
|
|
|
|
|
// number of bytes remaining to be read from r.
|
|
|
|
remaining uint64
|
|
|
|
limited bool
|
|
|
|
|
|
|
|
// auxiliary buffer for integer decoding
|
2014-11-13 21:31:48 +02:00
|
|
|
uintbuf []byte
|
|
|
|
|
|
|
|
kind Kind // kind of value ahead
|
|
|
|
size uint64 // size of value ahead
|
|
|
|
byteval byte // value of single byte in type tag
|
2015-04-14 17:09:06 +03:00
|
|
|
kinderr error // error from last readKind
|
2014-11-13 21:31:48 +02:00
|
|
|
stack []listpos
|
|
|
|
}
|
|
|
|
|
|
|
|
type listpos struct{ pos, size uint64 }
|
|
|
|
|
2015-03-23 16:02:55 +02:00
|
|
|
// NewStream creates a new decoding stream reading from r.
|
|
|
|
//
|
|
|
|
// If r implements the ByteReader interface, Stream will
|
|
|
|
// not introduce any buffering.
|
|
|
|
//
|
|
|
|
// For non-toplevel values, Stream returns ErrElemTooLarge
|
|
|
|
// for values that do not fit into the enclosing list.
|
|
|
|
//
|
|
|
|
// Stream supports an optional input limit. If a limit is set, the
|
|
|
|
// size of any toplevel value will be checked against the remaining
|
|
|
|
// input length. Stream operations that encounter a value exceeding
|
|
|
|
// the remaining input length will return ErrValueTooLarge. The limit
|
|
|
|
// can be set by passing a non-zero value for inputLimit.
|
|
|
|
//
|
|
|
|
// If r is a bytes.Reader or strings.Reader, the input limit is set to
|
|
|
|
// the length of r's underlying data unless an explicit limit is
|
|
|
|
// provided.
|
|
|
|
func NewStream(r io.Reader, inputLimit uint64) *Stream {
|
2014-11-24 20:01:25 +02:00
|
|
|
s := new(Stream)
|
2015-03-23 16:02:55 +02:00
|
|
|
s.Reset(r, inputLimit)
|
2014-11-24 20:01:25 +02:00
|
|
|
return s
|
|
|
|
}
|
2014-11-24 20:02:04 +02:00
|
|
|
|
|
|
|
// NewListStream creates a new stream that pretends to be positioned
|
|
|
|
// at an encoded list of the given length.
|
|
|
|
func NewListStream(r io.Reader, len uint64) *Stream {
|
|
|
|
s := new(Stream)
|
2015-03-23 16:02:55 +02:00
|
|
|
s.Reset(r, len)
|
2014-11-24 20:02:04 +02:00
|
|
|
s.kind = List
|
|
|
|
s.size = len
|
|
|
|
return s
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
// Bytes reads an RLP string and returns its contents as a byte slice.
|
|
|
|
// If the input does not contain an RLP string, the returned
|
|
|
|
// error will be ErrExpectedString.
|
|
|
|
func (s *Stream) Bytes() ([]byte, error) {
|
|
|
|
kind, size, err := s.Kind()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
switch kind {
|
|
|
|
case Byte:
|
|
|
|
s.kind = -1 // rearm Kind
|
|
|
|
return []byte{s.byteval}, nil
|
|
|
|
case String:
|
|
|
|
b := make([]byte, size)
|
|
|
|
if err = s.readFull(b); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2015-07-18 17:13:20 +03:00
|
|
|
if size == 1 && b[0] < 128 {
|
2015-04-16 11:15:14 +03:00
|
|
|
return nil, ErrCanonSize
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
return b, nil
|
|
|
|
default:
|
|
|
|
return nil, ErrExpectedString
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-03-21 01:49:31 +02:00
|
|
|
// Raw reads a raw encoded value including RLP type information.
|
|
|
|
func (s *Stream) Raw() ([]byte, error) {
|
|
|
|
kind, size, err := s.Kind()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if kind == Byte {
|
|
|
|
s.kind = -1 // rearm Kind
|
|
|
|
return []byte{s.byteval}, nil
|
|
|
|
}
|
|
|
|
// the original header has already been read and is no longer
|
|
|
|
// available. read content and put a new header in front of it.
|
|
|
|
start := headsize(size)
|
|
|
|
buf := make([]byte, uint64(start)+size)
|
|
|
|
if err := s.readFull(buf[start:]); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if kind == String {
|
2017-08-24 13:10:50 +03:00
|
|
|
puthead(buf, 0x80, 0xB7, size)
|
2015-03-21 01:49:31 +02:00
|
|
|
} else {
|
|
|
|
puthead(buf, 0xC0, 0xF7, size)
|
|
|
|
}
|
|
|
|
return buf, nil
|
|
|
|
}
|
|
|
|
|
2014-11-13 21:31:48 +02:00
|
|
|
// Uint reads an RLP string of up to 8 bytes and returns its contents
|
|
|
|
// as an unsigned integer. If the input does not contain an RLP string, the
|
|
|
|
// returned error will be ErrExpectedString.
|
|
|
|
func (s *Stream) Uint() (uint64, error) {
|
|
|
|
return s.uint(64)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Stream) uint(maxbits int) (uint64, error) {
|
|
|
|
kind, size, err := s.Kind()
|
|
|
|
if err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
switch kind {
|
|
|
|
case Byte:
|
2015-04-14 17:09:06 +03:00
|
|
|
if s.byteval == 0 {
|
|
|
|
return 0, ErrCanonInt
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
s.kind = -1 // rearm Kind
|
|
|
|
return uint64(s.byteval), nil
|
|
|
|
case String:
|
|
|
|
if size > uint64(maxbits/8) {
|
2014-11-25 17:00:48 +02:00
|
|
|
return 0, errUintOverflow
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
2015-04-14 17:09:06 +03:00
|
|
|
v, err := s.readUint(byte(size))
|
2015-04-16 11:15:14 +03:00
|
|
|
switch {
|
|
|
|
case err == ErrCanonSize:
|
2015-04-14 17:09:06 +03:00
|
|
|
// Adjust error because we're not reading a size right now.
|
2015-04-16 11:15:14 +03:00
|
|
|
return 0, ErrCanonInt
|
|
|
|
case err != nil:
|
|
|
|
return 0, err
|
2015-07-18 17:13:20 +03:00
|
|
|
case size > 0 && v < 128:
|
2015-04-16 11:15:14 +03:00
|
|
|
return 0, ErrCanonSize
|
|
|
|
default:
|
|
|
|
return v, nil
|
2015-04-14 17:09:06 +03:00
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
default:
|
|
|
|
return 0, ErrExpectedString
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-08-13 11:12:38 +03:00
|
|
|
// Bool reads an RLP string of up to 1 byte and returns its contents
|
2017-06-12 15:45:17 +03:00
|
|
|
// as a boolean. If the input does not contain an RLP string, the
|
2015-08-13 11:12:38 +03:00
|
|
|
// returned error will be ErrExpectedString.
|
|
|
|
func (s *Stream) Bool() (bool, error) {
|
|
|
|
num, err := s.uint(8)
|
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
switch num {
|
|
|
|
case 0:
|
|
|
|
return false, nil
|
|
|
|
case 1:
|
|
|
|
return true, nil
|
|
|
|
default:
|
|
|
|
return false, fmt.Errorf("rlp: invalid boolean value: %d", num)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-13 21:31:48 +02:00
|
|
|
// List starts decoding an RLP list. If the input does not contain a
|
|
|
|
// list, the returned error will be ErrExpectedList. When the list's
|
|
|
|
// end has been reached, any Stream operation will return EOL.
|
|
|
|
func (s *Stream) List() (size uint64, err error) {
|
|
|
|
kind, size, err := s.Kind()
|
|
|
|
if err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
if kind != List {
|
|
|
|
return 0, ErrExpectedList
|
|
|
|
}
|
|
|
|
s.stack = append(s.stack, listpos{0, size})
|
|
|
|
s.kind = -1
|
|
|
|
s.size = 0
|
|
|
|
return size, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// ListEnd returns to the enclosing list.
|
|
|
|
// The input reader must be positioned at the end of a list.
|
|
|
|
func (s *Stream) ListEnd() error {
|
|
|
|
if len(s.stack) == 0 {
|
|
|
|
return errNotInList
|
|
|
|
}
|
|
|
|
tos := s.stack[len(s.stack)-1]
|
|
|
|
if tos.pos != tos.size {
|
|
|
|
return errNotAtEOL
|
|
|
|
}
|
|
|
|
s.stack = s.stack[:len(s.stack)-1] // pop
|
|
|
|
if len(s.stack) > 0 {
|
|
|
|
s.stack[len(s.stack)-1].pos += tos.size
|
|
|
|
}
|
|
|
|
s.kind = -1
|
|
|
|
s.size = 0
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Decode decodes a value and stores the result in the value pointed
|
|
|
|
// to by val. Please see the documentation for the Decode function
|
|
|
|
// to learn about the decoding rules.
|
|
|
|
func (s *Stream) Decode(val interface{}) error {
|
|
|
|
if val == nil {
|
|
|
|
return errDecodeIntoNil
|
|
|
|
}
|
|
|
|
rval := reflect.ValueOf(val)
|
|
|
|
rtyp := rval.Type()
|
|
|
|
if rtyp.Kind() != reflect.Ptr {
|
|
|
|
return errNoPointer
|
|
|
|
}
|
|
|
|
if rval.IsNil() {
|
|
|
|
return errDecodeIntoNil
|
|
|
|
}
|
2019-05-14 16:09:56 +03:00
|
|
|
decoder, err := cachedDecoder(rtyp.Elem())
|
2014-11-13 21:31:48 +02:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2015-01-05 18:10:42 +02:00
|
|
|
|
2019-05-14 16:09:56 +03:00
|
|
|
err = decoder(s, rval.Elem())
|
2015-01-05 18:10:42 +02:00
|
|
|
if decErr, ok := err.(*decodeError); ok && len(decErr.ctx) > 0 {
|
|
|
|
// add decode target type to error so context has more meaning
|
|
|
|
decErr.ctx = append(decErr.ctx, fmt.Sprint("(", rtyp.Elem(), ")"))
|
|
|
|
}
|
|
|
|
return err
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
|
2014-11-24 20:01:25 +02:00
|
|
|
// Reset discards any information about the current decoding context
|
2015-03-23 16:02:55 +02:00
|
|
|
// and starts reading from r. This method is meant to facilitate reuse
|
|
|
|
// of a preallocated Stream across many decoding operations.
|
|
|
|
//
|
|
|
|
// If r does not also implement ByteReader, Stream will do its own
|
|
|
|
// buffering.
|
|
|
|
func (s *Stream) Reset(r io.Reader, inputLimit uint64) {
|
|
|
|
if inputLimit > 0 {
|
|
|
|
s.remaining = inputLimit
|
|
|
|
s.limited = true
|
|
|
|
} else {
|
|
|
|
// Attempt to automatically discover
|
|
|
|
// the limit when reading from a byte slice.
|
|
|
|
switch br := r.(type) {
|
|
|
|
case *bytes.Reader:
|
|
|
|
s.remaining = uint64(br.Len())
|
|
|
|
s.limited = true
|
|
|
|
case *strings.Reader:
|
|
|
|
s.remaining = uint64(br.Len())
|
|
|
|
s.limited = true
|
|
|
|
default:
|
|
|
|
s.limited = false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Wrap r with a buffer if it doesn't have one.
|
2014-11-24 20:01:25 +02:00
|
|
|
bufr, ok := r.(ByteReader)
|
|
|
|
if !ok {
|
|
|
|
bufr = bufio.NewReader(r)
|
|
|
|
}
|
|
|
|
s.r = bufr
|
2015-03-23 16:02:55 +02:00
|
|
|
// Reset the decoding context.
|
2014-11-24 20:01:25 +02:00
|
|
|
s.stack = s.stack[:0]
|
|
|
|
s.size = 0
|
|
|
|
s.kind = -1
|
2015-04-14 17:09:06 +03:00
|
|
|
s.kinderr = nil
|
2014-11-24 20:01:25 +02:00
|
|
|
if s.uintbuf == nil {
|
|
|
|
s.uintbuf = make([]byte, 8)
|
|
|
|
}
|
2019-02-25 13:34:08 +03:00
|
|
|
s.byteval = 0
|
2014-11-24 20:01:25 +02:00
|
|
|
}
|
|
|
|
|
2014-11-13 21:31:48 +02:00
|
|
|
// Kind returns the kind and size of the next value in the
|
|
|
|
// input stream.
|
|
|
|
//
|
|
|
|
// The returned size is the number of bytes that make up the value.
|
|
|
|
// For kind == Byte, the size is zero because the value is
|
|
|
|
// contained in the type tag.
|
|
|
|
//
|
|
|
|
// The first call to Kind will read size information from the input
|
|
|
|
// reader and leave it positioned at the start of the actual bytes of
|
|
|
|
// the value. Subsequent calls to Kind (until the value is decoded)
|
|
|
|
// will not advance the input reader and return cached information.
|
|
|
|
func (s *Stream) Kind() (kind Kind, size uint64, err error) {
|
|
|
|
var tos *listpos
|
|
|
|
if len(s.stack) > 0 {
|
|
|
|
tos = &s.stack[len(s.stack)-1]
|
|
|
|
}
|
|
|
|
if s.kind < 0 {
|
2015-04-14 17:09:06 +03:00
|
|
|
s.kinderr = nil
|
2015-04-14 01:54:12 +03:00
|
|
|
// Don't read further if we're at the end of the
|
2015-03-23 16:02:55 +02:00
|
|
|
// innermost list.
|
2014-11-13 21:31:48 +02:00
|
|
|
if tos != nil && tos.pos == tos.size {
|
|
|
|
return 0, 0, EOL
|
|
|
|
}
|
2015-04-14 17:09:06 +03:00
|
|
|
s.kind, s.size, s.kinderr = s.readKind()
|
|
|
|
if s.kinderr == nil {
|
|
|
|
if tos == nil {
|
|
|
|
// At toplevel, check that the value is smaller
|
|
|
|
// than the remaining input length.
|
|
|
|
if s.limited && s.size > s.remaining {
|
|
|
|
s.kinderr = ErrValueTooLarge
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Inside a list, check that the value doesn't overflow the list.
|
|
|
|
if s.size > tos.size-tos.pos {
|
|
|
|
s.kinderr = ErrElemTooLarge
|
|
|
|
}
|
|
|
|
}
|
2015-03-23 16:02:55 +02:00
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
2015-04-14 17:09:06 +03:00
|
|
|
// Note: this might return a sticky error generated
|
|
|
|
// by an earlier call to readKind.
|
|
|
|
return s.kind, s.size, s.kinderr
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Stream) readKind() (kind Kind, size uint64, err error) {
|
|
|
|
b, err := s.readByte()
|
|
|
|
if err != nil {
|
2015-04-28 11:28:15 +03:00
|
|
|
if len(s.stack) == 0 {
|
|
|
|
// At toplevel, Adjust the error to actual EOF. io.EOF is
|
|
|
|
// used by callers to determine when to stop decoding.
|
|
|
|
switch err {
|
|
|
|
case io.ErrUnexpectedEOF:
|
|
|
|
err = io.EOF
|
|
|
|
case ErrValueTooLarge:
|
|
|
|
err = io.EOF
|
|
|
|
}
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
return 0, 0, err
|
|
|
|
}
|
|
|
|
s.byteval = 0
|
|
|
|
switch {
|
|
|
|
case b < 0x80:
|
|
|
|
// For a single byte whose value is in the [0x00, 0x7F] range, that byte
|
|
|
|
// is its own RLP encoding.
|
|
|
|
s.byteval = b
|
|
|
|
return Byte, 0, nil
|
|
|
|
case b < 0xB8:
|
|
|
|
// Otherwise, if a string is 0-55 bytes long,
|
|
|
|
// the RLP encoding consists of a single byte with value 0x80 plus the
|
|
|
|
// length of the string followed by the string. The range of the first
|
|
|
|
// byte is thus [0x80, 0xB7].
|
|
|
|
return String, uint64(b - 0x80), nil
|
|
|
|
case b < 0xC0:
|
|
|
|
// If a string is more than 55 bytes long, the
|
|
|
|
// RLP encoding consists of a single byte with value 0xB7 plus the length
|
|
|
|
// of the length of the string in binary form, followed by the length of
|
|
|
|
// the string, followed by the string. For example, a length-1024 string
|
|
|
|
// would be encoded as 0xB90400 followed by the string. The range of
|
|
|
|
// the first byte is thus [0xB8, 0xBF].
|
|
|
|
size, err = s.readUint(b - 0xB7)
|
2015-04-14 17:09:06 +03:00
|
|
|
if err == nil && size < 56 {
|
|
|
|
err = ErrCanonSize
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
return String, size, err
|
|
|
|
case b < 0xF8:
|
|
|
|
// If the total payload of a list
|
|
|
|
// (i.e. the combined length of all its items) is 0-55 bytes long, the
|
|
|
|
// RLP encoding consists of a single byte with value 0xC0 plus the length
|
|
|
|
// of the list followed by the concatenation of the RLP encodings of the
|
|
|
|
// items. The range of the first byte is thus [0xC0, 0xF7].
|
|
|
|
return List, uint64(b - 0xC0), nil
|
|
|
|
default:
|
|
|
|
// If the total payload of a list is more than 55 bytes long,
|
|
|
|
// the RLP encoding consists of a single byte with value 0xF7
|
|
|
|
// plus the length of the length of the payload in binary
|
|
|
|
// form, followed by the length of the payload, followed by
|
|
|
|
// the concatenation of the RLP encodings of the items. The
|
|
|
|
// range of the first byte is thus [0xF8, 0xFF].
|
|
|
|
size, err = s.readUint(b - 0xF7)
|
2015-04-14 17:09:06 +03:00
|
|
|
if err == nil && size < 56 {
|
|
|
|
err = ErrCanonSize
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
return List, size, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Stream) readUint(size byte) (uint64, error) {
|
2015-04-14 17:09:06 +03:00
|
|
|
switch size {
|
|
|
|
case 0:
|
|
|
|
s.kind = -1 // rearm Kind
|
|
|
|
return 0, nil
|
|
|
|
case 1:
|
2014-11-13 21:31:48 +02:00
|
|
|
b, err := s.readByte()
|
|
|
|
return uint64(b), err
|
2015-04-14 17:09:06 +03:00
|
|
|
default:
|
|
|
|
start := int(8 - size)
|
|
|
|
for i := 0; i < start; i++ {
|
|
|
|
s.uintbuf[i] = 0
|
|
|
|
}
|
|
|
|
if err := s.readFull(s.uintbuf[start:]); err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
if s.uintbuf[start] == 0 {
|
|
|
|
// Note: readUint is also used to decode integer
|
|
|
|
// values. The error needs to be adjusted to become
|
|
|
|
// ErrCanonInt in this case.
|
|
|
|
return 0, ErrCanonSize
|
|
|
|
}
|
|
|
|
return binary.BigEndian.Uint64(s.uintbuf), nil
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Stream) readFull(buf []byte) (err error) {
|
2015-04-28 11:28:15 +03:00
|
|
|
if err := s.willRead(uint64(len(buf))); err != nil {
|
|
|
|
return err
|
2015-03-23 16:02:55 +02:00
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
var nn, n int
|
|
|
|
for n < len(buf) && err == nil {
|
|
|
|
nn, err = s.r.Read(buf[n:])
|
|
|
|
n += nn
|
|
|
|
}
|
|
|
|
if err == io.EOF {
|
2021-02-18 12:19:49 +03:00
|
|
|
if n < len(buf) {
|
|
|
|
err = io.ErrUnexpectedEOF
|
|
|
|
} else {
|
|
|
|
// Readers are allowed to give EOF even though the read succeeded.
|
|
|
|
// In such cases, we discard the EOF, like io.ReadFull() does.
|
|
|
|
err = nil
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Stream) readByte() (byte, error) {
|
2015-04-28 11:28:15 +03:00
|
|
|
if err := s.willRead(1); err != nil {
|
|
|
|
return 0, err
|
2015-03-23 16:02:55 +02:00
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
b, err := s.r.ReadByte()
|
2015-04-28 11:28:15 +03:00
|
|
|
if err == io.EOF {
|
2014-11-13 21:31:48 +02:00
|
|
|
err = io.ErrUnexpectedEOF
|
|
|
|
}
|
|
|
|
return b, err
|
|
|
|
}
|
|
|
|
|
2015-04-28 11:28:15 +03:00
|
|
|
func (s *Stream) willRead(n uint64) error {
|
2014-11-13 21:31:48 +02:00
|
|
|
s.kind = -1 // rearm Kind
|
2015-04-28 11:28:15 +03:00
|
|
|
|
2014-11-13 21:31:48 +02:00
|
|
|
if len(s.stack) > 0 {
|
2015-04-28 11:28:15 +03:00
|
|
|
// check list overflow
|
|
|
|
tos := s.stack[len(s.stack)-1]
|
|
|
|
if n > tos.size-tos.pos {
|
|
|
|
return ErrElemTooLarge
|
|
|
|
}
|
2014-11-13 21:31:48 +02:00
|
|
|
s.stack[len(s.stack)-1].pos += n
|
|
|
|
}
|
2015-04-28 11:28:15 +03:00
|
|
|
if s.limited {
|
|
|
|
if n > s.remaining {
|
|
|
|
return ErrValueTooLarge
|
|
|
|
}
|
|
|
|
s.remaining -= n
|
|
|
|
}
|
|
|
|
return nil
|
2014-11-13 21:31:48 +02:00
|
|
|
}
|