From 9ded7e7e872940d2fa60206af9f7f07aa214eb80 Mon Sep 17 00:00:00 2001 From: Matt Joiner Date: Wed, 5 May 2021 15:48:57 +1000 Subject: [PATCH] Use bytes.Reader instead of bytes.Buffer in a few places --- bencode/api.go | 2 +- peer_protocol/pex_test.go | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/bencode/api.go b/bencode/api.go index 3e3c1633..10375c43 100644 --- a/bencode/api.go +++ b/bencode/api.go @@ -130,7 +130,7 @@ func MustMarshal(v interface{}) []byte { // Unmarshal the bencode value in the 'data' to a value pointed by the 'v' // pointer, return a non-nil error if any. func Unmarshal(data []byte, v interface{}) (err error) { - buf := bytes.NewBuffer(data) + buf := bytes.NewReader(data) e := Decoder{r: buf} err = e.Decode(v) if err == nil && buf.Len() != 0 { diff --git a/peer_protocol/pex_test.go b/peer_protocol/pex_test.go index adfab295..cc45059e 100644 --- a/peer_protocol/pex_test.go +++ b/peer_protocol/pex_test.go @@ -48,7 +48,7 @@ func TestMarshalPexMessage(t *testing.T) { msg = Message{} dec := Decoder{ - R: bufio.NewReader(bytes.NewBuffer(b)), + R: bufio.NewReader(bytes.NewReader(b)), MaxLength: 128, } pmOut := PexMsg{} -- 2.48.1