aboutsummaryrefslogtreecommitdiffstats
path: root/eth/protocol.go
diff options
context:
space:
mode:
authorobscuren <geffobscura@gmail.com>2015-03-06 00:21:32 +0800
committerobscuren <geffobscura@gmail.com>2015-03-06 00:21:32 +0800
commitba0c41436c8a067937aa0845963f9ec3b81e7aed (patch)
tree7bd9bbc9422353ab341f01581abeb346b5a6cacf /eth/protocol.go
parentfabaf4f1f01db773f6c0c2e9a9499736b1a40848 (diff)
parent215c763d53fc8e06e8c9807875eacaccf3ef45fa (diff)
downloadgo-tangerine-ba0c41436c8a067937aa0845963f9ec3b81e7aed.tar
go-tangerine-ba0c41436c8a067937aa0845963f9ec3b81e7aed.tar.gz
go-tangerine-ba0c41436c8a067937aa0845963f9ec3b81e7aed.tar.bz2
go-tangerine-ba0c41436c8a067937aa0845963f9ec3b81e7aed.tar.lz
go-tangerine-ba0c41436c8a067937aa0845963f9ec3b81e7aed.tar.xz
go-tangerine-ba0c41436c8a067937aa0845963f9ec3b81e7aed.tar.zst
go-tangerine-ba0c41436c8a067937aa0845963f9ec3b81e7aed.zip
Merge branch 'p2p-handshake-2' of https://github.com/fjl/go-ethereum into fjl-p2p-handshake-2
Diffstat (limited to 'eth/protocol.go')
-rw-r--r--eth/protocol.go36
1 files changed, 21 insertions, 15 deletions
diff --git a/eth/protocol.go b/eth/protocol.go
index c887af129..b52c7db42 100644
--- a/eth/protocol.go
+++ b/eth/protocol.go
@@ -3,7 +3,6 @@ package eth
import (
"bytes"
"fmt"
- "io"
"math/big"
"github.com/ethereum/go-ethereum/core/types"
@@ -188,33 +187,37 @@ func (self *ethProtocol) handle() error {
case BlockHashesMsg:
msgStream := rlp.NewStream(msg.Payload)
- var err error
- var i int
+ if _, err := msgStream.List(); err != nil {
+ return err
+ }
+ var i int
iter := func() (hash []byte, ok bool) {
- hash, err = msgStream.Bytes()
- if err == nil {
- i++
- ok = true
- } else {
- if err != io.EOF {
- self.protoError(ErrDecode, "msg %v: after %v hashes : %v", msg, i, err)
- }
+ hash, err := msgStream.Bytes()
+ if err == rlp.EOL {
+ return nil, false
+ } else if err != nil {
+ self.protoError(ErrDecode, "msg %v: after %v hashes : %v", msg, i, err)
+ return nil, false
}
- return
+ i++
+ return hash, true
}
-
self.blockPool.AddBlockHashes(iter, self.id)
case GetBlocksMsg:
msgStream := rlp.NewStream(msg.Payload)
+ if _, err := msgStream.List(); err != nil {
+ return err
+ }
+
var blocks []interface{}
var i int
for {
i++
var hash []byte
if err := msgStream.Decode(&hash); err != nil {
- if err == io.EOF {
+ if err == rlp.EOL {
break
} else {
return self.protoError(ErrDecode, "msg %v: %v", msg, err)
@@ -232,10 +235,13 @@ func (self *ethProtocol) handle() error {
case BlocksMsg:
msgStream := rlp.NewStream(msg.Payload)
+ if _, err := msgStream.List(); err != nil {
+ return err
+ }
for {
var block types.Block
if err := msgStream.Decode(&block); err != nil {
- if err == io.EOF {
+ if err == rlp.EOL {
break
} else {
return self.protoError(ErrDecode, "msg %v: %v", msg, err)