From 2696e635e2df464ffc968278dbd8dbb4ca6faa8c Mon Sep 17 00:00:00 2001
From: Jake Charman <jake@jakecharman.co.uk>
Date: Tue, 25 Mar 2025 14:29:48 +0000
Subject: [PATCH] Sanity check error codes

---
 src/index.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/index.py b/src/index.py
index 5347255..3550988 100755
--- a/src/index.py
+++ b/src/index.py
@@ -5,7 +5,7 @@ from os import environ
 import threading
 import logging
 from requests import post
-from flask import Flask, render_template
+from flask import Flask, render_template, Response
 
 app = Flask(__name__)
 
@@ -53,6 +53,12 @@ def index() -> str:
 @app.route('/error/<code>')
 def error(code) -> str:
     ''' Render a nicer error page for a given code '''
+
+    if not code.isDigit():
+        code=400
+    elif code not in error_definitions:
+        return Response(code)
+
     error_definitions = {
         400: 'Bad Request',
         403: 'Forbidden',