Added fields in mapping.xml

code/ID2TLib/Botnet/MessageMapping.py

@@ -30,6 +30,11 @@ class MessageMapping:
             mapping.setAttribute(self.ATTR_ID, str(message.msg_id))
             mapping.setAttribute(self.ATTR_LINENO, str(message.line_no))
 
+            mapping.setAttribute("Src", str(message.src["ID"]))
+            mapping.setAttribute("Dst", str(message.dst["ID"]))
+            mapping.setAttribute("Type", str(message.type.value))
+            mapping.setAttribute("Time", str(message.time))
+
             packet = self.id_to_packet.get(message.msg_id)
             mapping.setAttribute(self.ATTR_HAS_PACKET, "true" if packet is not None else "false")
             if packet: