Maj Graph

Author: kali

client/GraphPanel.py

@@ -10,6 +10,20 @@
 from grpcClient import *
 
 
+#
+# Constant
+#
+BeaconNodeItemType = "Beacon"
+ListenerNodeItemType = "Listener"
+
+PrimaryListenerImage = "images/firewall.svg"
+WindowsSessionImage = "images/pc.svg"
+LinuxSessionImage = "images/linux.svg"
+
+
+#
+# Graph Tab Implementation
+#
 # needed to send the message of mouseMoveEvent because QGraphicsPixmapItem doesn't herit from QObject
 class Signaller(QObject):
     signal = pyqtSignal()
@@ -22,17 +36,23 @@ class NodeItem(QGraphicsPixmapItem):
     # Signal to notify position changes
     signaller = Signaller()
 
-    def __init__(self, type, hash, parent=None):
-        if type == "Listener":
-            self.type = "Listener"
-            pixmap = QPixmap("firewall.png").scaled(64, 64, Qt.KeepAspectRatio, Qt.SmoothTransformation)
+    def __init__(self, type, hash, os="",  privilege="", parent=None):
+        if type == ListenerNodeItemType:
+            self.type = ListenerNodeItemType
+            pixmap = QPixmap(PrimaryListenerImage).scaled(64, 64, Qt.KeepAspectRatio, Qt.SmoothTransformation)
             self.beaconHash = ""
             self.connectedListenerHash = ""
             self.listenerHash = []
             self.listenerHash.append(hash)
-        elif type == "Beacon":
-            self.type = "Beacon"
-            pixmap = QPixmap("pc.png").scaled(64, 64, Qt.KeepAspectRatio, Qt.SmoothTransformation)
+        elif type == BeaconNodeItemType:
+            self.type = BeaconNodeItemType
+            # print("NodeItem beaconHash", hash, "os", os, "privilege", privilege)
+            if "linux" in os.lower():
+                pixmap = QPixmap(LinuxSessionImage).scaled(64, 64, Qt.KeepAspectRatio, Qt.SmoothTransformation)
+            elif "windows" in os.lower():
+                pixmap = QPixmap(WindowsSessionImage).scaled(64, 64, Qt.KeepAspectRatio, Qt.SmoothTransformation)
+            else:
+                pixmap = QPixmap(LinuxSessionImage).scaled(64, 64, Qt.KeepAspectRatio, Qt.SmoothTransformation)
             self.beaconHash=hash
             self.connectedListenerHash = ""
             self.listenerHash=[]
@@ -68,14 +88,16 @@ def __init__(self, listener, beacon, pen=None):
         self.listener = listener
         self.beacon = beacon
 
-        self.pen = pen or QPen(QColor("black"), 2)
+        self.pen = pen or QPen(QColor("white"), 3)
         self.setPen(self.pen)
         self.update_line()
 
     def print(self):
         print("Connector", "beaconHash", self.beacon.beaconHash, "connectedListenerHash", self.beacon.connectedListenerHash, "listenerHash", self.listener.listenerHash)
 
     def update_line(self):
+        # print("listener", self.listener.pos())
+        # print("beacon", self.beacon.pos())
         center1 = self.listener.pos() + self.listener.boundingRect().center()
         center2 = self.beacon.pos() + self.beacon.boundingRect().center()
         self.setLine(QLineF(center1, center2))
@@ -145,7 +167,7 @@ def updateGraph(self):
             for session in sessions:
                 if session.beaconHash == nodeItem.beaconHash:
                     runing=True
-            if not runing and self.listNodeItem[ix].type == "Beacon":
+            if not runing and self.listNodeItem[ix].type == BeaconNodeItemType:
                 for ix2, connector in enumerate(self.listConnector):
                     if connector.beacon.beaconHash == nodeItem.beaconHash:
                         print("[-] delete connector")
@@ -162,7 +184,7 @@ def updateGraph(self):
                 if session.beaconHash == nodeItem.beaconHash:
                     inStore=True
             if not inStore:
-                item = NodeItem("Beacon", session.beaconHash)
+                item = NodeItem(BeaconNodeItemType, session.beaconHash, session.os, session.privilege)
                 item.connectedListenerHash = session.listenerHash
                 item.signaller.signal.connect(self.updateConnectors)
                 self.scene.addItem(item)
@@ -185,7 +207,7 @@ def updateGraph(self):
                     runing=True
             if not runing:
                 # primary listener
-                if self.listNodeItem[ix].type == "Listener":
+                if self.listNodeItem[ix].type == ListenerNodeItemType:
                     for ix2, connector in enumerate(self.listConnector):
                         if self.listNodeItem[ix2].listenerHash in connector.listener.listenerHash:
                             print("[-] delete connector")
@@ -196,7 +218,7 @@ def updateGraph(self):
                     del self.listNodeItem[ix]
 
                 # beacon listener
-                elif self.listNodeItem[ix].type == "Beacon":
+                elif self.listNodeItem[ix].type == BeaconNodeItemType:
                     if listener.listenerHash in self.listNodeItem[ix].listenerHash:
                         for ix2, connector in enumerate(self.listConnector):
                             if self.listNodeItem[ix2].listenerHash in connector.listener.listenerHash:
@@ -214,7 +236,7 @@ def updateGraph(self):
                     inStore=True
             if not inStore:
                 if not listener.beaconHash:
-                    item = NodeItem("Listener", listener.listenerHash)
+                    item = NodeItem(ListenerNodeItemType, listener.listenerHash)
                     item.signaller.signal.connect(self.updateConnectors)
                     self.scene.addItem(item)
                     self.listNodeItem.append(item)
@@ -229,7 +251,7 @@ def updateGraph(self):
         # Update connectors
         #        
         for nodeItem in self.listNodeItem:
-            if nodeItem.type == "Beacon":
+            if nodeItem.type == BeaconNodeItemType:
                 inStore=False
                 beaconHash = nodeItem.beaconHash
                 listenerHash = nodeItem.connectedListenerHash
@@ -241,6 +263,7 @@ def updateGraph(self):
                         if listener.isResponsableForListener(listenerHash)==True:
                             connector = Connector(listener, nodeItem)
                             self.scene.addItem(connector)
+                            connector.setZValue(-1)
                             self.listConnector.append(connector)
                             print("[+] add connector listener:", listenerHash, "beacon", beaconHash)