From b77b329dd64f0d114ec5ef02c922036d5d4f5726 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jannes=20H=C3=B6ke?= Date: Tue, 17 Apr 2018 06:40:02 +0200 Subject: [PATCH] Handle utf8 decoding errors (#1076) --- telegram/utils/request.py | 7 ++++++- tests/test_request.py | 34 ++++++++++++++++++++++++++++++++++ 2 files changed, 40 insertions(+), 1 deletion(-) create mode 100644 tests/test_request.py diff --git a/telegram/utils/request.py b/telegram/utils/request.py index 3d65b7357..6fda64397 100644 --- a/telegram/utils/request.py +++ b/telegram/utils/request.py @@ -145,9 +145,14 @@ class Request(object): dict: A JSON parsed as Python dict with results - on error this dict will be empty. """ - decoded_s = json_data.decode('utf-8') + try: + decoded_s = json_data.decode('utf-8') data = json.loads(decoded_s) + except UnicodeDecodeError: + logging.getLogger(__name__).debug( + 'Logging raw invalid UTF-8 response:\n%r', json_data) + raise TelegramError('Server response could not be decoded using UTF-8') except ValueError: raise TelegramError('Invalid server response') diff --git a/tests/test_request.py b/tests/test_request.py new file mode 100644 index 000000000..734ac78c6 --- /dev/null +++ b/tests/test_request.py @@ -0,0 +1,34 @@ +#!/usr/bin/env python +# +# A library that provides a Python interface to the Telegram Bot API +# Copyright (C) 2015-2018 +# Leandro Toledo de Souza +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Lesser Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Lesser Public License for more details. +# +# You should have received a copy of the GNU Lesser Public License +# along with this program. If not, see [http://www.gnu.org/licenses/]. + +import pytest + +from telegram import TelegramError +from telegram.utils.request import Request + + +def test_parse_illegal_callback_data(): + """ + Clients can send arbitrary bytes in callback data. + Make sure the correct error is raised in this case. + """ + server_response = b'{"invalid utf-8": "\x80"}' + + with pytest.raises(TelegramError, match='Server response could not be decoded using UTF-8'): + Request._parse(server_response)