From e96e17bd013378dd81c72d59059a6dcf71bf3a7f Mon Sep 17 00:00:00 2001
From: Oliver Falk <oliver@linux-kernel.at>
Date: Fri, 23 Nov 2018 15:52:41 +0100
Subject: [PATCH] Add error view if digest is to long/short

---
 ivatar/urls.py       | 7 +++++++
 templates/error.html | 9 ++++++++-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/ivatar/urls.py b/ivatar/urls.py
index f875a5c..d86fd04 100644
--- a/ivatar/urls.py
+++ b/ivatar/urls.py
@@ -20,6 +20,13 @@ urlpatterns = [  # pylint: disable=invalid-name
     url(
         r'avatar/(?P<digest>\w{32})',
         AvatarImageView.as_view(), name='avatar_view'),
+    url(
+        r'avatar/(?P<digest>\w)',
+        TemplateView.as_view(
+            template_name='error.html',
+            extra_context={
+                'errormessage': 'Incorrect digest length',
+            })),
     url('description/', TemplateView.as_view(template_name='description.html'), name='description'),
     # The following two are TODO TODO TODO TODO TODO
     url('run_your_own/', TemplateView.as_view(template_name='run_your_own.html'), name='run_your_own'),
diff --git a/templates/error.html b/templates/error.html
index 420b9b0..292fff8 100644
--- a/templates/error.html
+++ b/templates/error.html
@@ -8,7 +8,14 @@
 {% block content %}
 <h1 class="error">{% trans 'Error!' %}</h1>
 
-<p>{% block errormessage %}{% trans 'Libravatar has encountered an error.' %}{% endblock errormessage %}</p>
+<p>{% block errormessage %}
+{% trans 'Libravatar has encountered an error.' %}
+{% if errormessage %}
+<br/>
+<br/>
+{% blocktrans %}{{ errormessage }}{% endblocktrans %}
+{% endif %}
+{% endblock errormessage %}</p>
 
 <div style="height:40px"></div>