From cebf93cdc3cbd7fad5257f9dad49c71db09912ab Mon Sep 17 00:00:00 2001
From: Neil Booth <kyuupichan@gmail.com>
Date: Wed, 23 Nov 2016 11:52:17 +0900
Subject: [PATCH] Handle abuse better

- close connections on any encoding error
- if the connection is closing do not process buffered requests
---
 lib/jsonrpc.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/lib/jsonrpc.py b/lib/jsonrpc.py
index eff942d..e2ddda2 100644
--- a/lib/jsonrpc.py
+++ b/lib/jsonrpc.py
@@ -97,6 +97,9 @@ class JSONRPC(asyncio.Protocol, LoggedClass):
         decode_message for handling.
         '''
         self.recv_size += len(data)
+        if self.transport.is_closing():
+            return
+
         while True:
             npos = data.find(ord('\n'))
             if npos == -1:
@@ -118,6 +121,7 @@ class JSONRPC(asyncio.Protocol, LoggedClass):
         except UnicodeDecodeError as e:
             msg = 'cannot decode binary bytes: {}'.format(e)
             self.send_json_error(msg, self.PARSE_ERROR)
+            self.transport.close()
             return
 
         try:
@@ -125,6 +129,7 @@ class JSONRPC(asyncio.Protocol, LoggedClass):
         except json.JSONDecodeError as e:
             msg = 'cannot decode JSON: {}'.format(e)
             self.send_json_error(msg, self.PARSE_ERROR)
+            self.transport.close()
             return
 
         self.on_json_request(message)