def __init__(self, app, on_exit):
super(UIService, self).__init__()
self.MENU_ENTRY_STATS = QtCore.QCoreApplication.translate("contextual_menu", "Statistics")
self.MENU_ENTRY_FW_ENABLE = QtCore.QCoreApplication.translate("contextual_menu", "Enable")
self.MENU_ENTRY_FW_DISABLE = QtCore.QCoreApplication.translate("contextual_menu", "Disable")
self.MENU_ENTRY_HELP = QtCore.QCoreApplication.translate("contextual_menu", "Help")
self.MENU_ENTRY_CLOSE = QtCore.QCoreApplication.translate("contextual_menu", "Close")
self._cfg = Config.init()
self._db = Database.instance()
db_file=self._cfg.getSettings(self._cfg.DEFAULT_DB_FILE_KEY)
db_status, db_error = self._db.initialize(
dbtype=self._cfg.getInt(self._cfg.DEFAULT_DB_TYPE_KEY),
dbfile=db_file
)
if db_status is False:
Message.ok(
QtCore.QCoreApplication.translate("preferences", "Warning"),
QtCore.QCoreApplication.translate("preferences",
"The DB is corrupted and it's not safe to continue.<br>\
Remove, backup or recover the file before continuing.<br><br>\
Corrupted database file: {0}".format(db_file)),
QtWidgets.QMessageBox.Warning)
sys.exit(-1)
self._db_sqlite = self._db.get_db()
self._last_ping = None
self._version_warning_shown = False
self._asking = False
self._connected = False
self._fw_enabled = False
self._path = os.path.abspath(os.path.dirname(__file__))
self._app = app
self._on_exit = on_exit
self._exit = False
self._msg = QtWidgets.QMessageBox()
self._prompt_dialog = PromptDialog()
self._remote_lock = Lock()
self._remote_stats = {}
self._setup_interfaces()
self._setup_icons()
self._stats_dialog = StatsDialog(dbname="general", db=self._db)
self._setup_tray()
self._setup_slots()
self._nodes = Nodes.instance()
self._last_stats = {}
self._last_items = {
'hosts':{},
'procs':{},
'addrs':{},
'ports':{},
'users':{}
}
def __init__(self, app, on_exit):
super(UIService, self).__init__()
self.MENU_ENTRY_STATS = QtCore.QCoreApplication.translate(
"contextual_menu", "Statistics")
self.MENU_ENTRY_FW_ENABLE = QtCore.QCoreApplication.translate(
"contextual_menu", "Enable")
self.MENU_ENTRY_FW_DISABLE = QtCore.QCoreApplication.translate(
"contextual_menu", "Disable")
self.MENU_ENTRY_HELP = QtCore.QCoreApplication.translate(
"contextual_menu", "Help")
self.MENU_ENTRY_CLOSE = QtCore.QCoreApplication.translate(
"contextual_menu", "Close")
self._cfg = Config.init()
self._db = Database.instance()
self._db.initialize(
dbtype=self._cfg.getInt(self._cfg.DEFAULT_DB_TYPE_KEY),
dbfile=self._cfg.getSettings(self._cfg.DEFAULT_DB_FILE_KEY))
self._db_sqlite = self._db.get_db()
self._last_ping = None
self._version_warning_shown = False
self._asking = False
self._connected = False
self._fw_enabled = False
self._path = os.path.abspath(os.path.dirname(__file__))
self._app = app
self._on_exit = on_exit
self._exit = False
self._msg = QtWidgets.QMessageBox()
self._prompt_dialog = PromptDialog()
self._remote_lock = Lock()
self._remote_stats = {}
self._setup_interfaces()
self._setup_icons()
self._stats_dialog = StatsDialog(dbname="general", db=self._db)
self._setup_tray()
self._setup_slots()
self._nodes = Nodes.instance()
self._last_stats = {}
self._last_items = {
'hosts': {},
'procs': {},
'addrs': {},
'ports': {},
'users': {}
}
def __init__(self, _interval, _callback):
Thread.__init__(self, name="cleaner_db_thread")
self.interval = _interval * 60
self.stop_flag = Event()
self.callback = _callback
self._cfg = Config.init()
# We need to instantiate a new QsqlDatabase object with a unique name,
# because it's not thread safe:
# "A connection can only be used from within the thread that created it."
# https://doc.qt.io/qt-5/threads-modules.html#threads-and-the-sql-module
# The filename and type is the same, the one chosen by the user.
self.db = Database("db-cleaner-connection")
self.db_status, db_error = self.db.initialize(
dbtype=self._cfg.getInt(self._cfg.DEFAULT_DB_TYPE_KEY),
dbfile=self._cfg.getSettings(self._cfg.DEFAULT_DB_FILE_KEY))
class DesktopNotifications():
"""DesktopNotifications display informative pop-ups using the system D-Bus.
The notifications are handled and configured by the system.
The notification daemon also decides where to show the notifications, as well
as how to group them.
The body of a notification supports markup (if the implementation supports it):
https://people.gnome.org/~mccann/docs/notification-spec/notification-spec-latest.html#markup
Basically: <a>, <u>, <b>, <i> and <img>. New lines can be added with the regular \n.
It also support actions (buttons).
https://notify2.readthedocs.io/en/latest/
"""
_cfg = Config.init()
# list of hints:
# https://people.gnome.org/~mccann/docs/notification-spec/notification-spec-latest.html#hints
HINT_DESKTOP_ENTRY = "desktop-entry"
CATEGORY_NETWORK = "network"
EXPIRES_DEFAULT = 0
NEVER_EXPIRES = -1
def __init__(self):
self.ACTION_ALLOW = QC.translate("popups", "Allow")
self.ACTION_DENY = QC.translate("popups", "Deny")
self.IS_LIBNOTIFY_AVAILABLE = True
self.DOES_SUPPORT_ACTIONS = True
try:
import notify2
self.ntf2 = notify2
mloop = 'glib'
# First try to initialise the D-Bus connection with the given
# mainloop.
# If it fails, we'll try to initialise it without it.
try:
self.ntf2.init("opensnitch", mainloop=mloop)
except Exception:
self.DOES_SUPPORT_ACTIONS = False
# usually because dbus mainloop is not initiated, specially
# with 'qt'
# FIXME: figure out how to init it, or how to connect to an
# existing session.
print(
"DesktopNotifications(): system doesn't support actions. Available capabilities:"
)
print(self.ntf2.get_server_caps())
self.ntf2.init("opensnitch")
# Example: ['actions', 'action-icons', 'body', 'body-markup', 'icon-static', 'persistence', 'sound']
if ('actions' not in self.ntf2.get_server_caps()):
self.DOES_SUPPORT_ACTIONS = False
except Exception as e:
print(
"DesktopNotifications not available (install python3-notify2):",
e)
self.IS_LIBNOTIFY_AVAILABLE = False
def is_available(self):
return self.IS_LIBNOTIFY_AVAILABLE
def are_enabled(self):
return self._cfg.getBool(Config.NOTIFICATIONS_ENABLED, True)
def support_actions(self):
"""Returns true if the notifications daemon support actions(buttons).
This depends on 2 factors:
- If the notification server actually supports it (get_server_caps()).
- If there's a dbus instance running.
"""
return self.DOES_SUPPORT_ACTIONS
def show(self, title, body, icon="dialog-information"):
try:
ntf = self.ntf2.Notification(title, body, icon)
# timeouts seems to be ignored (on Cinnamon at least)
timeout = self._cfg.getInt(Config.DEFAULT_TIMEOUT_KEY, 15)
# -1 and 0 are special values
if timeout > 0:
timeout = timeout * 1000
ntf.set_timeout(timeout * 1000)
ntf.timeout = timeout * 1000
ntf.set_category(self.CATEGORY_NETWORK)
# used to display our app icon an name.
ntf.set_hint(self.HINT_DESKTOP_ENTRY, "opensnitch_ui")
ntf.show()
except Exception as e:
print("[notifications] show() exception:", e)
# TODO:
# - construct a rule with the default configured parameters.
# - create a common dialogs/prompt.py:_send_rule(), maybe in utils.py
def ask(self, connection, timeout, callback):
c = connection
title = QC.translate("popups", "New outgoing connection")
body = c.process_path + "\n"
body = body + QC.translate("popups", "is connecting to <b>%s</b> on %s port %d") % ( \
c.dst_host or c.dst_ip,
c.protocol.upper(),
c.dst_port )
ntf = self.ntf2.Notification(title, body, "dialog-warning")
timeout = self._cfg.getInt(Config.DEFAULT_TIMEOUT_KEY, 15)
ntf.set_timeout(timeout * 1000)
ntf.timeout = timeout * 1000
if self.DOES_SUPPORT_ACTIONS:
ntf.set_urgency(self.ntf2.URGENCY_CRITICAL)
ntf.add_action("allow", self.ACTION_ALLOW, callback, connection)
ntf.add_action("deny", self.ACTION_DENY, callback, connection)
#ntf.add_action("open-gui", QC.translate("popups", "View"), callback, connection)
ntf.set_category(self.CATEGORY_NETWORK)
ntf.set_hint(self.HINT_DESKTOP_ENTRY, "opensnitch_ui")
ntf.show()
"Firewall": "iptables",
"Stats": {
"MaxEvents": 150,
"MaxStats": 50
}
}
'''
class Connection:
protocol = "tcp"
src_ip = "127.0.0.1"
src_port = "12345"
dst_ip = "127.0.0.1"
dst_host = "localhost"
dst_port = "54321"
user_id = 1000
process_id = 9876
process_path = "/bin/cmd"
process_cwd = "/tmp"
process_args = "/bin/cmd --parm1 test"
process_env = []
db = Database.instance()
db.initialize()
Config.init()
nodes = Nodes.instance()
nodes._nodes["unix:/tmp/osui.sock"] = {'data': ClientConfig}
