mirror of
https://github.com/maxDcb/C2TeamServer
synced 2026-06-06 16:14:27 +00:00
kali
90 lines
2.3 KiB
Python
90 lines
2.3 KiB
Python
import sys
|
|
import signal
|
|
from threading import Thread, Lock
|
|
from PyQt5.QtWidgets import *
|
|
from PyQt5.QtGui import *
|
|
from PyQt5.QtCore import *
|
|
|
|
from grpcClient import *
|
|
from ListenerPanel import *
|
|
from SessionPanel import *
|
|
from ConsolePanel import *
|
|
from PayloadPanel import *
|
|
|
|
import qdarktheme
|
|
|
|
|
|
signal.signal(signal.SIGINT, signal.SIG_DFL)
|
|
|
|
class App(QMainWindow):
|
|
|
|
def __init__(self, ip, port):
|
|
super().__init__()
|
|
|
|
self.ip = ip
|
|
self.port = port
|
|
|
|
self.createPayloadWindow = None
|
|
|
|
self.title = 'Exploration C2'
|
|
self.left = 0
|
|
self.top = 0
|
|
self.width = 1000
|
|
self.height = 1000
|
|
self.setWindowTitle(self.title)
|
|
self.setGeometry(self.left, self.top, self.width, self.height)
|
|
|
|
central_widget = QWidget()
|
|
self.setCentralWidget(central_widget)
|
|
|
|
self.m_w11 = QWidget()
|
|
self.m_w12 = QWidget()
|
|
self.m_w21 = QWidget()
|
|
|
|
config_button = QPushButton("Payload")
|
|
config_button.clicked.connect(self.payloadForm)
|
|
|
|
lay = QGridLayout(central_widget)
|
|
lay.setRowStretch(1, 3)
|
|
lay.setRowStretch(2, 7)
|
|
# TODO complet PayloadPanel
|
|
# row: int, column: int, rowSpan: int, columnSpan: int, alignment
|
|
#lay.addWidget(config_button, 0, 0, 1, 1)
|
|
lay.addWidget(self.m_w11, 1, 0, 1, 1)
|
|
lay.addWidget(self.m_w12, 1, 1, 1, 1)
|
|
lay.addWidget(self.m_w21, 2, 0, 1, 2)
|
|
|
|
lay = QVBoxLayout(self.m_w11)
|
|
sessionsWidget = Sessions(self, ip, port)
|
|
lay.addWidget(sessionsWidget)
|
|
|
|
lay = QVBoxLayout(self.m_w12)
|
|
listenersWidget = Listeners(self, ip, port)
|
|
lay.addWidget(listenersWidget)
|
|
|
|
lay = QVBoxLayout(self.m_w21)
|
|
consoleWidget = ConsolesTab(self, ip, port)
|
|
lay.addWidget(consoleWidget)
|
|
|
|
sessionsWidget.interactWithSession.connect(consoleWidget.addConsole)
|
|
|
|
self.show()
|
|
|
|
def __del__(self):
|
|
print("Exit")
|
|
|
|
def payloadForm(self):
|
|
if self.createPayloadWindow is None:
|
|
self.createPayloadWindow = CreatePayload()
|
|
self.createPayloadWindow.show()
|
|
|
|
if __name__ == '__main__':
|
|
|
|
ip = "localhost"
|
|
port = 50051
|
|
|
|
app = QApplication(sys.argv)
|
|
app.setStyleSheet(qdarktheme.load_stylesheet())
|
|
|
|
ex = App(ip, port)
|
|
sys.exit(app.exec_()) |