Fix incorrect EOF when decoding some peer protocol message types

[btrtrc.git] / peer_protocol / msg.go

diff --git a/peer_protocol/msg.go b/peer_protocol/msg.go
index 23710e6339d4cd93b79071cdb4c17b0165fbf5cb..f1b1f10e836af5890f5070eaf1fb0a0c2c0ee219 100644 (file)
--- a/peer_protocol/msg.go
+++ b/peer_protocol/msg.go
@@ -66,7 +66,7 @@ func (msg Message) MarshalBinary() (data []byte, err error) {
                }
                switch msg.Type {
                case Choke, Unchoke, Interested, NotInterested, HaveAll, HaveNone:
-               case Have:
+               case Have, AllowedFast, Suggest:
                        err = binary.Write(&buf, binary.BigEndian, msg.Index)
                case Request, Cancel, Reject:
                        for _, i := range []Integer{msg.Index, msg.Begin, msg.Length} {