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):