Handle missing-Host-header condition.

diff --git a/wrw/dispatch.py b/wrw/dispatch.py
index 15ea99e..fa1f669 100644 (file)
--- a/wrw/dispatch.py
+++ b/wrw/dispatch.py
@@ -1,5 +1,5 @@
 import sys, traceback
-import env
+import env, req, proto
 
 __all__ = ["restart"]
 
@@ -73,3 +73,9 @@ def handle(req, startreq, handler):
         return resp
     finally:
         req.cleanup()
+
+def handleenv(env, startreq, handler):
+    if not "HTTP_HOST" in env:
+        return proto.simpleerror(env, startreq, 400, "Bad Request", "Request must include Host header.")
+    r = req.origrequest(env)
+    return handle(r, startreq, handler)

diff --git a/wrw/util.py b/wrw/util.py
index 5ee002b..6368b18 100644 (file)
--- a/wrw/util.py
+++ b/wrw/util.py
@@ -3,7 +3,7 @@ import req, dispatch, session, form
 
 def wsgiwrap(callable):
     def wrapper(env, startreq):
-        return dispatch.handle(req.origrequest(env), startreq, callable)
+        return dispatch.handleenv(env, startreq, callable)
     return wrapper
 
 def formparams(callable):