From 0f7d35cdcac2e8db63b483dda180a334d3083712 Mon Sep 17 00:00:00 2001 From: Salad Dais Date: Sat, 30 Jul 2022 00:37:35 +0000 Subject: [PATCH] Handle HTTP messages with missing (not just empty) body --- hippolyzer/lib/proxy/message_logger.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hippolyzer/lib/proxy/message_logger.py b/hippolyzer/lib/proxy/message_logger.py index 75b7473..c1b944c 100644 --- a/hippolyzer/lib/proxy/message_logger.py +++ b/hippolyzer/lib/proxy/message_logger.py @@ -401,7 +401,7 @@ class AbstractMessageLogEntry(abc.ABC): beautified = minidom.parseString(content).toprettyxml(indent=" ") # kill blank lines. will break cdata sections. meh. beautified = re.sub(r'\n\s*\n', '\n', beautified, flags=re.MULTILINE) - return re.sub(r'<([\w]+)>\s*', r'<\1>', + return re.sub(r'<(\w+)>\s*', r'<\1>', beautified, flags=re.MULTILINE) @@ -522,7 +522,7 @@ class HTTPMessageLogEntry(AbstractMessageLogEntry): buf.write(bytes(headers).decode("utf8", errors="replace")) buf.write("\r\n") - buf.write(message_body) + buf.write(message_body or "") return buf.getvalue() def request(self, beautify=False, replacements=None):