From f9f9ced5320a593bb59836086d0d5983b4df58df Mon Sep 17 00:00:00 2001 From: Den4200 Date: Sat, 27 Mar 2021 13:17:12 -0400 Subject: Restrict redirects for paste uploads with an extension that is not `.py`. --- bot/utils/services.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/bot/utils/services.py b/bot/utils/services.py index 5949c9e48..68cbd896e 100644 --- a/bot/utils/services.py +++ b/bot/utils/services.py @@ -47,7 +47,13 @@ async def send_to_paste_service(contents: str, *, extension: str = "") -> Option continue elif "key" in response_json: log.info(f"Successfully uploaded contents to paste service behind key {response_json['key']}.") - return URLs.paste_service.format(key=response_json['key']) + extension + + paste_link = URLs.paste_service.format(key=response_json['key']) + extension + + if extension == '.py': + return paste_link + return paste_link + "?noredirect" + log.warning( f"Got unexpected JSON response from paste service: {response_json}\n" f"trying again ({attempt}/{FAILED_REQUEST_ATTEMPTS})." -- cgit v1.2.3