mirror of
https://github.com/evilsocket/opensnitch.git
synced 2025-03-04 16:44:46 +01:00
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
from PyQt5 import QtWidgets, QtGui
|
|
|
|
import sys
|
|
import os
|
|
import time
|
|
import signal
|
|
import argparse
|
|
from concurrent import futures
|
|
|
|
path = os.path.abspath(os.path.dirname(__file__))
|
|
sys.path.append(path)
|
|
sys.path.append(path + "/../proto/")
|
|
|
|
import grpc
|
|
import ui_pb2
|
|
import ui_pb2_grpc
|
|
|
|
from service import UIService
|
|
from version import version
|
|
|
|
def on_exit():
|
|
app.quit()
|
|
server.stop(0)
|
|
sys.exit(0)
|
|
|
|
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")
|
|
|
|
args = parser.parse_args()
|
|
|
|
app = QtWidgets.QApplication(sys.argv)
|
|
|
|
service = UIService(app, on_exit)
|
|
server = grpc.server(futures.ThreadPoolExecutor(max_workers=4))
|
|
|
|
ui_pb2_grpc.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()
|
|
|