2021-11-25 06:44:12 +00:00
|
|
|
# Python imports
|
2023-02-05 05:38:44 +00:00
|
|
|
import os
|
|
|
|
import time
|
2022-11-29 04:34:13 +00:00
|
|
|
from multiprocessing.connection import Client
|
|
|
|
from multiprocessing.connection import Listener
|
2021-11-25 06:44:12 +00:00
|
|
|
|
2021-12-07 04:44:39 +00:00
|
|
|
# Lib imports
|
2023-02-05 05:38:44 +00:00
|
|
|
from gi.repository import GLib
|
2021-11-25 06:44:12 +00:00
|
|
|
|
|
|
|
# Application imports
|
2023-03-28 01:07:17 +00:00
|
|
|
from .singleton import Singleton
|
2021-11-25 06:44:12 +00:00
|
|
|
|
|
|
|
|
2021-12-07 04:44:39 +00:00
|
|
|
|
2023-03-28 01:07:17 +00:00
|
|
|
class IPCServer(Singleton):
|
2022-02-20 07:32:51 +00:00
|
|
|
""" Create a listener so that other SolarFM instances send requests back to existing instance. """
|
2022-06-14 22:12:25 +00:00
|
|
|
def __init__(self, ipc_address: str = '127.0.0.1', conn_type: str = "socket"):
|
|
|
|
self.is_ipc_alive = False
|
|
|
|
self._ipc_port = 4848
|
|
|
|
self._ipc_address = ipc_address
|
|
|
|
self._conn_type = conn_type
|
2022-09-06 02:21:04 +00:00
|
|
|
self._ipc_authkey = b'' + bytes(f'{app_name}-ipc', 'utf-8')
|
2022-06-14 22:12:25 +00:00
|
|
|
self._ipc_timeout = 15.0
|
2022-02-25 23:58:11 +00:00
|
|
|
|
2022-03-03 07:24:59 +00:00
|
|
|
if conn_type == "socket":
|
2022-09-06 02:21:04 +00:00
|
|
|
self._ipc_address = f'/tmp/{app_name}-ipc.sock'
|
2022-06-11 01:13:57 +00:00
|
|
|
elif conn_type == "full_network":
|
2022-06-14 22:12:25 +00:00
|
|
|
self._ipc_address = '0.0.0.0'
|
2022-06-11 01:13:57 +00:00
|
|
|
elif conn_type == "full_network_unsecured":
|
2022-06-14 22:12:25 +00:00
|
|
|
self._ipc_authkey = None
|
|
|
|
self._ipc_address = '0.0.0.0'
|
2022-06-11 01:13:57 +00:00
|
|
|
elif conn_type == "local_network_unsecured":
|
2022-06-14 22:12:25 +00:00
|
|
|
self._ipc_authkey = None
|
2022-03-03 07:24:59 +00:00
|
|
|
|
2022-09-29 22:22:33 +00:00
|
|
|
self._subscribe_to_events()
|
|
|
|
|
2022-10-23 04:53:33 +00:00
|
|
|
|
2022-09-29 22:22:33 +00:00
|
|
|
def _subscribe_to_events(self):
|
|
|
|
event_system.subscribe("post_file_to_ipc", self.send_ipc_message)
|
|
|
|
|
2022-06-14 22:12:25 +00:00
|
|
|
def create_ipc_listener(self) -> None:
|
2022-03-03 07:24:59 +00:00
|
|
|
if self._conn_type == "socket":
|
2023-10-19 02:51:24 +00:00
|
|
|
if os.path.exists(self._ipc_address) and settings_manager.is_dirty_start():
|
2022-10-21 03:23:14 +00:00
|
|
|
os.unlink(self._ipc_address)
|
2022-03-03 07:24:59 +00:00
|
|
|
|
2023-10-19 02:23:45 +00:00
|
|
|
listener = Listener(address = self._ipc_address, family = "AF_UNIX", authkey = self._ipc_authkey)
|
|
|
|
|
2022-06-14 22:12:25 +00:00
|
|
|
elif "unsecured" not in self._conn_type:
|
2023-10-19 02:23:45 +00:00
|
|
|
listener = Listener((self._ipc_address, self._ipc_port), authkey = self._ipc_authkey)
|
2022-06-11 01:13:57 +00:00
|
|
|
else:
|
2022-06-14 22:12:25 +00:00
|
|
|
listener = Listener((self._ipc_address, self._ipc_port))
|
2022-03-03 07:24:59 +00:00
|
|
|
|
2021-11-25 06:44:12 +00:00
|
|
|
self.is_ipc_alive = True
|
2022-10-23 04:53:33 +00:00
|
|
|
self._run_ipc_loop(listener)
|
|
|
|
|
|
|
|
@daemon_threaded
|
|
|
|
def _run_ipc_loop(self, listener) -> None:
|
2021-12-25 05:58:57 +00:00
|
|
|
while True:
|
2023-02-08 23:52:33 +00:00
|
|
|
try:
|
|
|
|
conn = listener.accept()
|
|
|
|
start_time = time.perf_counter()
|
|
|
|
GLib.idle_add(self._handle_ipc_message, *(conn, start_time,))
|
|
|
|
except Exception as e:
|
2023-10-19 02:23:45 +00:00
|
|
|
logger.debug( repr(e) )
|
2021-11-28 04:03:00 +00:00
|
|
|
|
2022-06-14 22:12:25 +00:00
|
|
|
listener.close()
|
2021-11-25 06:44:12 +00:00
|
|
|
|
2022-10-23 04:53:33 +00:00
|
|
|
def _handle_ipc_message(self, conn, start_time) -> None:
|
2022-06-14 22:12:25 +00:00
|
|
|
while True:
|
|
|
|
msg = conn.recv()
|
2023-10-19 02:23:45 +00:00
|
|
|
logger.debug(msg)
|
2021-11-25 08:21:10 +00:00
|
|
|
|
2022-06-14 22:12:25 +00:00
|
|
|
if "FILE|" in msg:
|
|
|
|
file = msg.split("FILE|")[1].strip()
|
|
|
|
if file:
|
2022-10-01 04:30:38 +00:00
|
|
|
event_system.emit("handle_file_from_ipc", file)
|
2021-11-25 06:44:12 +00:00
|
|
|
|
2022-06-14 22:12:25 +00:00
|
|
|
conn.close()
|
|
|
|
break
|
2021-11-25 06:44:12 +00:00
|
|
|
|
2021-11-28 04:03:00 +00:00
|
|
|
|
2022-06-14 22:12:25 +00:00
|
|
|
if msg in ['close connection', 'close server']:
|
|
|
|
conn.close()
|
|
|
|
break
|
2021-11-28 04:03:00 +00:00
|
|
|
|
2022-06-14 22:12:25 +00:00
|
|
|
# NOTE: Not perfect but insures we don't lock up the connection for too long.
|
|
|
|
end_time = time.perf_counter()
|
|
|
|
if (end_time - start_time) > self._ipc_timeout:
|
|
|
|
conn.close()
|
|
|
|
break
|
2021-11-25 06:44:12 +00:00
|
|
|
|
|
|
|
|
2022-03-25 03:15:08 +00:00
|
|
|
def send_ipc_message(self, message: str = "Empty Data...") -> None:
|
2021-11-25 06:44:12 +00:00
|
|
|
try:
|
2022-03-03 07:24:59 +00:00
|
|
|
if self._conn_type == "socket":
|
2022-06-14 22:12:25 +00:00
|
|
|
conn = Client(address=self._ipc_address, family="AF_UNIX", authkey=self._ipc_authkey)
|
|
|
|
elif "unsecured" not in self._conn_type:
|
|
|
|
conn = Client((self._ipc_address, self._ipc_port), authkey=self._ipc_authkey)
|
2022-06-11 01:13:57 +00:00
|
|
|
else:
|
2022-06-14 22:12:25 +00:00
|
|
|
conn = Client((self._ipc_address, self._ipc_port))
|
2022-03-03 07:24:59 +00:00
|
|
|
|
2021-11-25 06:44:12 +00:00
|
|
|
conn.send(message)
|
2022-03-25 03:15:08 +00:00
|
|
|
conn.close()
|
2022-06-14 22:12:25 +00:00
|
|
|
except ConnectionRefusedError as e:
|
2023-10-19 02:23:45 +00:00
|
|
|
logger.error("Connection refused...")
|
|
|
|
except Exception as e:
|
|
|
|
logger.error( repr(e) )
|
|
|
|
|
|
|
|
|
|
|
|
def send_test_ipc_message(self, message: str = "Empty Data...") -> None:
|
|
|
|
try:
|
|
|
|
if self._conn_type == "socket":
|
|
|
|
conn = Client(address=self._ipc_address, family="AF_UNIX", authkey=self._ipc_authkey)
|
|
|
|
elif "unsecured" not in self._conn_type:
|
|
|
|
conn = Client((self._ipc_address, self._ipc_port), authkey=self._ipc_authkey)
|
|
|
|
else:
|
|
|
|
conn = Client((self._ipc_address, self._ipc_port))
|
|
|
|
|
|
|
|
conn.send(message)
|
|
|
|
conn.close()
|
|
|
|
except ConnectionRefusedError as e:
|
|
|
|
if self._conn_type == "socket":
|
|
|
|
logger.error("IPC Socket no longer valid.... Removing.")
|
|
|
|
os.unlink(self._ipc_address)
|
2021-11-25 06:44:12 +00:00
|
|
|
except Exception as e:
|
2023-10-19 02:23:45 +00:00
|
|
|
logger.error( repr(e) )
|