From 73c1bb817c7be238566481878ac0bb0c8dfd8427 Mon Sep 17 00:00:00 2001
From: Zahary Karadjov <zahary@gmail.com>
Date: Wed, 19 Aug 2020 14:14:59 +0300
Subject: [PATCH] Adapt to latest snappy; Avoid decompression bombs

---
 eth/p2p/rlpx.nim | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/eth/p2p/rlpx.nim b/eth/p2p/rlpx.nim
index 2de0e9c..cfa18e0 100644
--- a/eth/p2p/rlpx.nim
+++ b/eth/p2p/rlpx.nim
@@ -258,7 +258,7 @@ proc invokeThunk*(peer: Peer, msgId: int, msgData: var Rlp): Future[void] =
 template compressMsg(peer: Peer, data: seq[byte]): seq[byte] =
   when useSnappy:
     if peer.snappyEnabled:
-      snappy.compress(data)
+      snappy.encode(data)
     else: data
   else:
     data
@@ -422,7 +422,7 @@ proc recvMsg*(peer: Peer): Future[tuple[msgId: int, msgData: Rlp]] {.async.} =
 
   when useSnappy:
     if peer.snappyEnabled:
-      decryptedBytes = snappy.uncompress(decryptedBytes)
+      decryptedBytes = snappy.decode(decryptedBytes, maxMsgSize)
       if decryptedBytes.len == 0:
         await peer.disconnectAndRaise(BreachOfProtocol,
                                       "Snappy uncompress encountered malformed data")