From 731c9bb990f15e87df907d9c87ac50f177386849 Mon Sep 17 00:00:00 2001
From: Rob Austein <sra@hactrn.net>
Date: Thu, 28 Aug 2014 18:23:52 +0000
Subject: Don't return HTML error messages on POST failures.

svn path=/branches/tk705/; revision=5938
---
 rpki/http_simple.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'rpki')

diff --git a/rpki/http_simple.py b/rpki/http_simple.py
index 16a1d792..8f609e46 100644
--- a/rpki/http_simple.py
+++ b/rpki/http_simple.py
@@ -62,6 +62,14 @@ class HTTPRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):
   def log_message(self, *args):
     logger.info(*args, extra = dict(context = "%s:%s" % self.client_address))
 
+  def send_error(self, code, message = None):
+    # BaseHTTPRequestHandler.send_error() generates HTML error messages,
+    # which we don't want, so we override the method to suppress this.
+    self.send_response(code, message)
+    self.send_header("Content-Type", rpki_content_type)
+    self.send_header("Connection", "close")
+    self.end_headers()
+
 
 def server(handlers, port, host = ""):
   """
-- 
cgit v1.2.3