mirror of
https://github.com/evilsocket/opensnitch.git
synced 2025-03-04 16:44:46 +01:00
76 lines
2.3 KiB
Python
Executable file
76 lines
2.3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
from PyQt5 import QtWidgets, QtGui, QtCore
|
|
|
|
import sys
|
|
import os
|
|
import time
|
|
import signal
|
|
import argparse
|
|
import logging
|
|
|
|
logging.getLogger().disabled = True
|
|
|
|
from concurrent import futures
|
|
|
|
import grpc
|
|
|
|
dist_path = '/usr/lib/python3/dist-packages/'
|
|
if dist_path not in sys.path:
|
|
sys.path.append(dist_path)
|
|
|
|
from opensnitch.service import UIService
|
|
from opensnitch.config import Config
|
|
import opensnitch.version
|
|
import opensnitch.ui_pb2
|
|
from opensnitch.ui_pb2_grpc import add_UIServicer_to_server
|
|
|
|
def on_exit():
|
|
app.quit()
|
|
server.stop(0)
|
|
sys.exit(0)
|
|
|
|
def supported_qt_version(major, medium, minor):
|
|
q = QtCore.QT_VERSION_STR.split(".")
|
|
return int(q[0]) >= major and int(q[1]) >= medium and int(q[2]) >= minor
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='OpenSnitch UI service.')
|
|
parser.add_argument("--socket", dest="socket", default="unix:///tmp/osui.sock", help="Path of the unix socket for the gRPC service (https://github.com/grpc/grpc/blob/master/doc/naming.md).", metavar="FILE")
|
|
parser.add_argument("--max-clients", dest="serverWorkers", default=10, help="Max number of allowed clients (incoming connections).")
|
|
|
|
args = parser.parse_args()
|
|
|
|
os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "1"
|
|
if supported_qt_version(5,6,0):
|
|
try:
|
|
# NOTE: maybe we also need Qt::AA_UseHighDpiPixmaps
|
|
QtCore.QApplication.setAttribute(QtCore.Qt.AA_EnableHighDpiScaling, True)
|
|
except Exception:
|
|
pass
|
|
|
|
app = QtWidgets.QApplication(sys.argv)
|
|
|
|
service = UIService(app, on_exit)
|
|
# @doc: https://grpc.github.io/grpc/python/grpc.html#server-object
|
|
server = grpc.server(futures.ThreadPoolExecutor(max_workers=int(args.serverWorkers)))
|
|
|
|
add_UIServicer_to_server(service, server)
|
|
|
|
if args.socket.startswith("unix://"):
|
|
socket = args.socket[7:]
|
|
socket = os.path.abspath(socket)
|
|
server.add_insecure_port("unix:%s" % socket)
|
|
else:
|
|
server.add_insecure_port(args.socket)
|
|
|
|
# https://stackoverflow.com/questions/5160577/ctrl-c-doesnt-work-with-pyqt
|
|
signal.signal(signal.SIGINT, signal.SIG_DFL)
|
|
|
|
try:
|
|
# print "OpenSnitch UI service running on %s ..." % socket
|
|
server.start()
|
|
app.exec_()
|
|
except KeyboardInterrupt:
|
|
on_exit()
|
|
|