Generate proper 404's in redirector

Instead of generating a 500 error, because that is both wrong and will
prevent caching.

diff --git a/redirector/redirector.py b/redirector/redirector.py
index d6dcf87bd60ae292a2936bf69bf731e83f0101a3..cb9f141294071d607e74d4d5fc64fbd20574e18a 100755 (executable)
--- a/redirector/redirector.py
+++ b/redirector/redirector.py
@@ -18,10 +18,16 @@ _urlvalmap = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', '
 connstr = ""
 
 
+class Make404(Exception):
+    pass
+
+
 def iddecode(idstr):
     idval = 0
     for c in idstr:
         idval *= 64
+        if c not in _urlvalmap:
+            raise Make404()
         idval += _urlvalmap.index(c)
     return idval
 
@@ -52,10 +58,7 @@ def application(environ, start_response):
         conn.close()
 
         if len(r) != 1:
-            start_response('404 Not Found', [
-                ('Content-type', 'text/plain'),
-            ])
-            return [b"Link not found\n"]
+            raise Make404()
 
         # We have a link, return a redirect to it
         start_response('301 Moved Permanently', [
@@ -68,6 +71,11 @@ def application(environ, start_response):
             b"<a href=\"%s\">moved here</a>\n" % r[0][0].encode('utf8'),
             b"</body>\n</html>\n"
         ]
+    except Make404:
+        start_response('404 Not Found', [
+            ('Content-type', 'text/plain'),
+        ])
+        return [b"Link not found\n"]
     except Exception as ex:
         start_response('500 Internal Server Error', [
             ('Content-type', 'text/plain')