class MainWindow(QMainWindow):
def __init__(self, parent=None):
super().__init__(parent)
self.process_page = ProcessSelectPage(self)
self.setCentralWidget(self.process_page)
self.text = TextPrefDialog(self)
self.view = QDeclarativeView()
self.tray = TrayIcon(self)
self.tray.show()
self._connect()
def _connect(self):
self.text.textSelected.connect(self.text_display)
def open_filter(self):
self.text.set_active(True)
self.text.show()
@Slot()
def text_display(self):
# _register()
# from qmlregister import register_qml_type
# register_qml_type()
# need to test qmlRegisterType once more
self.view.engine().addImportPath('ui/lib')
self.view.engine().addImportPath('ui/imports')
self.view.setSource('ui/main.qml')
self.view.setAttribute(Qt.WA_TranslucentBackground)
self.view.setStyleSheet("background-color:transparent")
self.view.setWindowFlags(Qt.SplashScreen | Qt.FramelessWindowHint
| Qt.WindowStaysOnTopHint)
self.view.showFullScreen()
#!/usr/bin/python
import sys
from PySide.QtCore import *
from PySide.QtGui import *
from PySide.QtDeclarative import QDeclarativeView
from PySide.QtDeclarative import QDeclarativeEngine
app = QApplication(sys.argv)
view = QDeclarativeView()
url = QUrl('MemoGame.qml')
view.setSource(url)
QObject.connect(view.engine(), SIGNAL('quit()'), app, SLOT('quit()'))
view.show()
app.exec_()
class MainWindow(QObject):
################################################
# validations
################################################
def validate_email(self, string):
if len(string) > 7:
if re.match("^.+\\@(\\[?)[a-zA-Z0-9\\-\\.]+\\.([a-zA-Z]{2,3}|[0-9]{1,3})(\\]?)$", string) != None:
return True
return False
def validate_fields(self):
if self.curVars.user_name == '':
self.rootO.show_alert("Empty name!", "Please type in your name.",False)
return False
if self.validate_email(self.curVars.email) == False:
self.rootO.show_alert("Invalid email", "Please make sure the email address you typed is valid.",False)
return False
if len(self.curVars.password) < 6:
self.rootO.show_alert("Bad password", "Password should contain at least 6 chars. Please try again.",False)
return False
elif self.curVars.password != self.curVars.password_confirm:
self.rootO.show_alert("Passwords don't match", "Please make sure both passwords match!",False)
return False
return True
################################################
# setting getting
################################################
def prey_exists(self):
if not os.path.exists(PREY_PATH + '/core'):
self.rootO.show_alert("Prey not installed", "Couldn't find a Prey installation on this system. Sorry.", True)
else:
return True
def is_config_writable(self):
command = 'if [ ! -w "'+PREY_PATH+'/config" ]; then echo 1; fi'
no_access = os.popen(command).read().strip()
if no_access == '1':
self.rootO.show_alert("Unauthorized", "You don't have access to manage Prey's configuration. Sorry.", True)
else:
return True
def get_setting(self, var):
command = 'grep \''+var+'=\' '+CONFIG_FILE+' | sed "s/'+var+'=\'\(.*\)\'/\\1/"'
return os.popen(command).read().strip()
def get_current_settings(self):
#delay = os.popen("/opt/cron/bin/crontab -l | grep prey | cut -c 3-4").read()
command = 'cat '+DAEMON_FILE+' | grep preyd | cut -d \' \' -f 3'
self.curVars.delay = int(os.popen(command).read())
#if not delay:
# self.curVars.delay = 30
#else:
# self.curVars.delay = int(delay)
self.curVars.auto_connect = self.get_setting('auto_connect')
self.curVars.extended_headers = self.get_setting('extended_headers')
self.curVars.lang = self.get_setting('lang')
self.curVars.check_url = self.get_setting('check_url')
self.curVars.post_method = self.get_setting('post_method')
self.curVars.api_key = self.get_setting('api_key')
self.curVars.device_key = self.get_setting('device_key')
self.curVars.mail_to = self.get_setting('mail_to')
self.curVars.smtp_server = self.get_setting('smtp_server')
self.curVars.smtp_username = self.get_setting('smtp_username')
def check_if_configured(self):
if self.curVars.post_method == 'http' and self.curVars.api_key == '':
return False
else:
return True
################################################
# setting settings
################################################
def save(self, param, value):
if param == 'check_url': value = value.replace('/', '\/')
command = 'sed -i -e "s/'+param+'=\'.*\'/'+param+'=\''+value+'\'/" '+ CONFIG_FILE
os.system(command)
@Slot(int)
def apply_main_settings(self,delay):
# save('lang', text('lang'))
self.save('auto_connect', self.curVars.auto_connect)
self.save('extended_headers', self.curVars.extended_headers)
# check and change the crontab interval
if delay != int(self.curVars.delay):
# print 'Updating delay in crontab...'
sub1 = ' -e \'s/preyd '+str(self.curVars.delay)+'/preyd '+str(delay)+'/g\' '
sub2 = ' -e \'s/every '+str(self.curVars.delay)+'/every '+str(delay)+'/g\' '
command = 'sed -i ' + sub1 + sub2 + DAEMON_FILE
os.system('initctl stop apps/preyd')
os.system(command)
os.system('initctl start apps/preyd')
#os.system('(/opt/cron/bin/crontab -l | tail -n+4 | grep -v prey; echo "*/'+str(delay)+' * * * * aegis-exec -s /opt/prey/prey.sh > /var/log/prey.log") | /opt/cron/bin/crontab -')
if self.check_if_configured() == False:
self.rootO.show_alert("All good.", "Configuration saved. Remember you still need to set up your posting method, otherwise Prey won't work!", True)
else:
self.rootO.show_alert("All good.", "Configuration saved!", True)
@Slot()
def create_new_user(self):
if self.validate_fields():
self.create_user()
def apply_control_panel_settings(self):
if self.curVars.post_method != 'http':
self.save('post_method', 'http')
if self.curVars.check_url != CONTROL_PANEL_URL:
self.save('check_url', CONTROL_PANEL_URL)
# we could eventually use the email as a checking method to remove prey
# i.e. "under which email was this account set up?"
# self.save('mail_to', self.email)
self.save('api_key', self.curVars.api_key)
if self.curVars.device_key != "":
self.save('device_key', self.curVars.device_key)
@Slot(str)
def apply_standalone_settings(self,password):
if self.curVars.post_method != 'email':
self.save('post_method', 'email')
self.save('check_url', self.curVars.check_url)
self.save('mail_to', self.curVars.mail_to)
self.save('smtp_server', self.curVars.smtp_server)
self.save('smtp_username', self.curVars.smtp_username)
#smtp_password = self.('smtp_password')
if password != '':
encoded_pass = os.popen('echo -n "'+ password +'" | openssl enc -base64').read().strip()
self.save('smtp_password', encoded_pass)
self.exit_configurator()
def exit_configurator(self):
self.run_prey()
self.rootO.show_alert(_("Success"), _("Configuration saved! Your device is now setup and being tracked by Prey. Happy hunting!"), True)
def run_prey(self):
os.system(PREY_PATH + '/run ')
################################################
# control panel api
################################################
def report_connection_issue(self, result):
print("Connection error. Response from server: " + result)
self.rootO.show_alert("Problem connecting", "We seem to be having a problem connecting to the Prey Control Panel. This is likely a temporary issue. Please try again in a few moments.",False)
def user_has_available_slots(self, string):
matches = re.search(r"<available_slots>(\w*)</available_slots>", string)
if matches and int(matches.groups()[0]) > 0:
return True
else:
return False
def get_api_key(self, string):
matches = re.search(r"<key>(\w*)</key>", string)
if matches:
self.curVars.api_key = matches.groups()[0]
@Slot(result=int)
def create_device_list(self):
self.deviceList = self.rootO.findChild(QObject,"deviceList")
self.deviceList.clear()
for device in self.deviceNames:
self.deviceList.add(device)
return self.chosen
def get_device_keys(self, string, has_available_slots):
hostname = os.popen("hostname").read().strip()
index = -1
self.deviceKeys = []
self.deviceNames = []
self.chosen = index
matches = re.findall(r"<device>\s*<key>(\w*)</key>.*?<title>([\.\s\w]*)</title>\s*</device>", string, re.DOTALL)
for match in matches:
index += 1
key = match[0]
title = match[1]
self.deviceKeys.append(key)
self.deviceNames.append(title)
if key == self.curVars.device_key: #set the choice because we have a matching device key
self.chosen = index
elif title.lower() == hostname.lower and self.chosen < 0: #set the choice because we likely have a matching title (but device key takes precedence)
self.chosen = index
if index < 0:
self.rootO.show_alert("No devices exist", "There are no devices currently defined in your Control Panel.\n\nPlease select the option to create a new device.",False)
return False
if self.chosen < 0:
self.chosen = 0
return True
def create_user(self):
params = urllib.urlencode({'user[name]': self.curVars.user_name, 'user[email]': self.curVars.email, 'user[password]': self.curVars.password, 'user[password_confirmation]' : self.curVars.password_confirm})
result = os.popen('curl -i -s -k --connect-timeout 10 '+ CONTROL_PANEL_URL_SSL + '/users.xml -d \"'+params+'\"').read().strip()
if result.find("<key>") != -1:
self.get_api_key(result)
self.curVars.device_key = ""
elif result.find("Email has already been taken") != -1:
self.rootO.show_alert("Email has already been taken", "That email address already exists! If you signed up previously, please go back and select the Existing User option.",False)
return
else:
self.rootO.show_alert("Couldn't create user!", "There was a problem creating your account. Please make sure the email address you entered is valid, as well as your password.",False)
return
self.apply_control_panel_settings()
self.run_prey()
self.rootO.show_alert("Account created!", "Your account has been succesfully created and configured in Prey's Control Panel.\n\nPlease check your inbox now, you should have received a verification email.", True)
@Slot(bool, result=bool)
def get_existing_user(self, show_devices):
email = self.curVars.email
password = self.curVars.password
print email+' '+password
result = os.popen('curl -i -s -k --connect-timeout 10 '+ CONTROL_PANEL_URL_SSL + '/profile.xml -u '+email+":'"+password+"'").read().strip()
if result.find('401 Unauthorized') != -1:
self.rootO.show_alert("User does not exist", "Couldn't log you in. Remember you need to activate your account opening the link we emailed you.\n\nIf you forgot your password please visit preyproject.com.",False)
return False
if result.find("<user>") != -1:
self.get_api_key(result)
else:
self.report_connection_issue(result)
return False
has_available_slots = self.user_has_available_slots(result)
if not has_available_slots and not show_devices:
self.rootO.show_alert("Not allowed", "It seems you've reached your limit for devices!\n\nIf you had previously added this PC, you should select the \"Device already exists\" option to select the device from a list of the ones you have already created.\n\nIf this is a new device, you can also upgrade to a Pro Account to increase your slot count and get access to additional features. For more information, please check\nhttp://preyproject.com/plans.",False)
return False
if show_devices:
result = os.popen('curl -i -s -k --connect-timeout 10 '+ CONTROL_PANEL_URL_SSL + '/devices.xml -u '+email+":'"+password+"'").read().strip()
if result.find("</devices>") != -1:
return self.get_device_keys(result,has_available_slots)
else:
self.report_connection_issue(result)
return False
else:
self.curVars.device_key = ""
self.apply_control_panel_settings()
self.exit_configurator()
@Slot(int)
def apply_device_settings(self,index):
self.curVars.device_key = self.deviceKeys[index]
self.apply_control_panel_settings()
self.exit_configurator()
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
# Create the Qt Application
self.app = QApplication(["Prey-Config"])
self.app.setWindowIcon(QIcon(''))
self.view = QDeclarativeView()
#self.setWindowTitle("Main Window")
#get rootContext of QDeclarativeView
self.context = self.view.rootContext()
#make this class available to QML files as 'main' context
self.context.setContextProperty('main', self)
#create variables
self.curVars = Vars()
#make curVars available to QML files as 'vars' context
self.context.setContextProperty('vars', self.curVars)
# QML resizes to main window
self.view.setResizeMode(QDeclarativeView.SizeRootObjectToView)
# Renders './qml/main.qml'
self.view.setSource(QUrl.fromLocalFile('./qml/main.qml'))
#get rootObject
self.rootO = self.view.rootObject()
#connect quit signal
self.view.engine().quit.connect(self.quit_app)
#check for prey installation, write acess to config file, and configurationstatus
if self.prey_exists():
if self.is_config_writable():
self.get_current_settings()
if self.check_if_configured() == False:
self.rootO.show_alert('Welcome!',"It seems this is the first time you run this setup. Please set up your reporting method now, otherwise Prey won't work!",False)
#os.system('(/opt/cron/bin/crontab -l | tail -n+4 | grep -v prey; echo "*/30 * * * * aegis-exec -s /opt/prey/prey.sh > /var/log/prey.log") | /opt/cron/bin/crontab -')
def quit_app(self):
self.view.hide()
self.app.exit()
from PySide.QtCore import QUrl
from PySide.QtGui import QApplication
from PySide.QtDeclarative import QDeclarativeView
import sys
from device_model import DeviceListModel
from service_model import ServiceListModel
from device_manager import DeviceManager
if __name__ == '__main__':
QApplication.setGraphicsSystem("raster")
app = QApplication(sys.argv)
view = QDeclarativeView()
engine = view.engine()
engine.quit.connect(app.quit)
if len(sys.argv) > 1:
manager = DeviceManager(sys.argv[1])
else:
manager = DeviceManager()
deviceListModel = DeviceListModel(manager)
serviceListModel = ServiceListModel()
deviceListModel.deviceSelected.connect(serviceListModel.loadDevice)
view.rootContext().setContextProperty("deviceManager", manager)
view.rootContext().setContextProperty("deviceListModel", deviceListModel)
view.rootContext().setContextProperty("serviceListModel", serviceListModel)
context = view.rootContext()
