sciveo 0.1.17__tar.gz → 0.1.19__tar.gz

{sciveo-0.1.17 → sciveo-0.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sciveo
-Version: 0.1.17
+Version: 0.1.19
 Description-Content-Type: text/markdown
 Provides-Extra: mon
 Provides-Extra: net

{sciveo-0.1.17 → sciveo-0.1.19}/sciveo/common/tools/random.py

@@ -1,3 +1,14 @@
+#
+# Pavlin Georgiev, Softel Labs
+#
+# This is a proprietary file and may not be copied,
+# distributed, or modified without express permission
+# from the owner. For licensing inquiries, please
+# contact pavlin@softel.bg.
+#
+# 2023
+#
+
 import random
 import string
 

sciveo-0.1.19/sciveo/common/tools/remote.py

@@ -0,0 +1,73 @@
+#
+# Pavlin Georgiev, Softel Labs
+#
+# This is a proprietary file and may not be copied,
+# distributed, or modified without express permission
+# from the owner. For licensing inquiries, please
+# contact pavlin@softel.bg.
+#
+# 2024
+#
+
+import os
+import time
+import cv2
+import base64
+import requests
+
+from sciveo.common.tools.logger import *
+from sciveo.common.tools.configuration import GlobalConfiguration
+from sciveo.common.tools.compress import CompressJsonData
+
+
+class PredictorRemoteClient:
+  def __init__(self, url="http://localhost:8901", verify=True, auth_token=None, api_prefix=None):
+    debug(type(self).__name__, "init url", url, verify)
+    config = GlobalConfiguration.get()
+    if auth_token is None:
+      auth_token = config["api_auth_token"]
+    if api_prefix is None:
+      api_prefix = config["api_prefix"]
+    self.auth_token = auth_token
+    self.api_prefix = api_prefix
+    self.url = f"{url}/{self.api_prefix}/predict"
+    self.verify = verify
+    requests.packages.urllib3.disable_warnings()
+
+  def predict(self, params):
+    try:
+      debug(type(self).__name__, "predict", params)
+      headers = {"Authorization": f"Bearer {self.auth_token}"}
+      response = requests.post(self.url, json=params, headers=headers, verify=self.verify)
+
+      if response.status_code == 200:
+        data = response.json()
+
+        if params.get("compressed", 0) > 0 and params["predictor"] in data:
+          predicted = data[params["predictor"]]
+          predicted = CompressJsonData().decompress(predicted)
+          data[params["predictor"]] = predicted
+      else:
+        error(type(self).__name__, f"Request [{self.url}] failed with status code {response.status_code}")
+        data = {"error": response.status_code}
+    except Exception as e:
+      exception(type(self).__name__, "predict", e)
+      data = {"error": str(e)}
+    return data
+
+  def image_encoded(self, image=None, resize_to=(256, 256), local_path=None):
+    if local_path is not None:
+      image = cv2.imread(local_path)
+    if resize_to is not None:
+      image = cv2.resize(image, resize_to)
+    buffer = cv2.imencode('.jpg', image)[1].tobytes()
+    return base64.b64encode(buffer).decode('utf-8')
+
+  def predict_image_embedding(self, image=None, resize_to=(256, 256), local_path=None):
+    image_base64 = self.image_encoded(image=image, resize_to=resize_to, local_path=local_path)
+    params = {
+      'predictor': 'ImageEmbedding', 'compressed': 1,
+      'X': [image_base64]
+    }
+    r = self.predict(params=params)
+    return r[params["predictor"]][0]

{sciveo-0.1.17 → sciveo-0.1.19}/sciveo/monitoring/monitor.py

@@ -101,7 +101,7 @@ class BaseMonitor(DaemonBase):
     try:
       usage_per_core = psutil.cpu_percent(interval=None, percpu=True)
       self.data["CPU"]["usage per core"] = usage_per_core
-      self.data["CPU"]["usage"] = np.array(usage_per_core).mean()
+      self.data["CPU"]["usage"] = float(np.array(usage_per_core).mean())
     except Exception:
       pass
 

{sciveo-0.1.17 → sciveo-0.1.19}/sciveo/network/camera.py

@@ -1,6 +1,19 @@
+#
+# Pavlin Georgiev, Softel Labs
+#
+# This is a proprietary file and may not be copied,
+# distributed, or modified without express permission
+# from the owner. For licensing inquiries, please
+# contact pavlin@softel.bg.
+#
+# 2024
+#
+
+import time
 import threading
 import cv2
 import numpy as np
+from scapy.all import sniff, IP, TCP, UDP
 
 from sciveo.common.tools.logger import *
 from sciveo.common.tools.timers import FPSCounter
@@ -8,12 +21,17 @@ from sciveo.network.tools import StreamSniffer
 
 
 class RTSPStreamSniffer(StreamSniffer):
+  def __init__(self, iface=None, port=554):
+    super().__init__(iface)
+    self.port = port
+
   def on_packet(self, packet):
     if self.is_rtsp_packet(packet):
       self.append_ip_packet(packet)
 
   def is_rtsp_packet(self, packet):
-    return IP in packet and TCP in packet and (packet[TCP].dport == 554 or packet[TCP].sport == 554)
+    return IP in packet and TCP in packet and (packet[TCP].dport == self.port or packet[TCP].sport == self.port)
+    # return IP in packet and ( (TCP in packet and (packet[TCP].dport == self.port or packet[TCP].sport == self.port)) or (UDP in packet and (packet[UDP].dport == self.port or packet[UDP].sport == self.port)) )
 
   def get_rtsp_frames(self, ip_src):
     frames = []
@@ -40,24 +58,25 @@ class RTSPStreamSniffer(StreamSniffer):
         if cv2.waitKey(1) & 0xFF == ord('q'):
           return
 
+  def play_cams(self, list_cams):
+    threads = []
+    for ip in list_cams:
+      t = threading.Thread(target=self.play, args=(ip,))
+      t.start()
+      threads.append(t)
 
-if __name__ == '__main__':
-  sniffer = RTSPStreamSniffer(iface="eth0")
-  sniffer.start()
+    self.start()
 
-  camera_ips = []
+    try:
+      while True:
+        time.sleep(1)
+    except KeyboardInterrupt:
+      self.stop()
+      for t in threads:
+        t.join()
+      cv2.destroyAllWindows()
 
-  threads = []
-  for ip in camera_ips:
-    t = threading.Thread(target=sniffer.play, args=(ip,))
-    t.start()
-    threads.append(t)
 
-  try:
-    while True:
-      pass
-  except KeyboardInterrupt:
-    sniffer.stop_sniffing()
-    for t in threads:
-      t.join()
-    cv2.destroyAllWindows()
+if __name__ == '__main__':
+  sniffer = RTSPStreamSniffer(iface="en0", port=554)
+  sniffer.play_cams([])

{sciveo-0.1.17 → sciveo-0.1.19}/sciveo/network/tools.py

@@ -93,7 +93,6 @@ class NetworkTools:
       pass
 
 
-
 class StreamSniffer:
   def __init__(self, iface=None):
     self.iface = iface

sciveo-0.1.19/sciveo/network/tunnel.py

@@ -0,0 +1,131 @@
+#
+# Pavlin Georgiev, Softel Labs
+#
+# This is a proprietary file and may not be copied,
+# distributed, or modified without express permission
+# from the owner. For licensing inquiries, please
+# contact pavlin@softel.bg.
+#
+# 2024
+#
+
+
+import os
+import subprocess
+
+from sciveo.common.tools.logger import *
+
+
+class WireGuardBase:
+  def __init__(self, interface='wg0'):
+    self.interface = interface
+    self.config_path = f'/etc/wireguard/{self.interface}.conf'
+    self.private_key = None
+    self.public_key = None
+
+  def _run_command(self, command):
+    try:
+      result = subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+      return result.stdout.decode().strip()
+    except subprocess.CalledProcessError as e:
+      debug(type(self).__name__, f"Error executing command: {e.stderr.decode().strip()}")
+      return None
+
+  def generate_keys(self, private_key_path=None, public_key_path=None):
+    # Generate the private key
+    self.private_key = self._run_command("wg genkey")
+    if not self.private_key:
+      raise Exception("Failed to generate private key.")
+
+    # Generate the public key from the private key
+    self.public_key = self._run_command(f"echo {self.private_key} | wg pubkey")
+    if not self.public_key:
+      raise Exception("Failed to generate public key.")
+
+    # Optionally save the keys to files
+    if private_key_path:
+      with open(private_key_path, 'w') as f:
+        f.write(self.private_key)
+      debug(type(self).__name__, f"Private key saved to {private_key_path}")
+
+    if public_key_path:
+      with open(public_key_path, 'w') as f:
+        f.write(self.public_key)
+      debug(type(self).__name__, f"Public key saved to {public_key_path}")
+
+    return self.private_key, self.public_key
+
+  def start(self):
+    debug(type(self).__name__, f"Starting WireGuard interface {self.interface}...")
+    return self._run_command(f'sudo systemctl start [email protected]{self.interface}')
+
+  def stop(self):
+    debug(type(self).__name__, f"Stopping WireGuard interface {self.interface}...")
+    return self._run_command(f'sudo systemctl stop [email protected]{self.interface}')
+
+  def restart(self):
+    debug(type(self).__name__, f"Restarting WireGuard interface {self.interface}...")
+    return self._run_command(f'sudo systemctl restart [email protected]{self.interface}')
+
+  def status(self):
+    debug(type(self).__name__, f"Checking status of WireGuard interface {self.interface}...")
+    return self._run_command(f'sudo systemctl status [email protected]{self.interface}')
+
+class WGServer(WireGuardBase):
+  def init(self, private_key=None, listen_port=51820, address='192.168.21.1/24'):
+    if not private_key:
+      private_key = self.private_key
+    config = f"""
+    [Interface]
+    Address = {address}
+    ListenPort = {listen_port}
+    PrivateKey = {private_key}
+    SaveConfig = true
+    """
+    with open(self.config_path, 'w') as config_file:
+      config_file.write(config.strip())
+    debug(type(self).__name__, f"Server configuration written to {self.config_path}")
+
+  def add_peer(self, peer_public_key, allowed_ips):
+    peer_config = f"""
+    [Peer]
+    PublicKey = {peer_public_key}
+    AllowedIPs = {allowed_ips}
+    """
+    with open(self.config_path, 'a') as config_file:
+      config_file.write(peer_config.strip())
+    debug(type(self).__name__, f"Added peer with PublicKey: {peer_public_key}")
+
+class WGClient(WireGuardBase):
+  def init(self, private_key=None, server_public_key=None, endpoint=None, address='192.168.21.2/24', allowed_ips='0.0.0.0/0', keepalive=25):
+    if not private_key:
+      private_key = self.private_key
+    config = f"""
+    [Interface]
+    Address = {address}
+    PrivateKey = {private_key}
+
+    [Peer]
+    PublicKey = {server_public_key}
+    Endpoint = {endpoint}
+    AllowedIPs = {allowed_ips}
+    PersistentKeepalive = {keepalive}
+    """
+    with open(self.config_path, 'w') as config_file:
+      config_file.write(config.strip())
+    debug(type(self).__name__, f"Client configuration written to {self.config_path}")
+
+
+if __name__ == "__main__":
+  server = WGServer(interface='wg0')
+  server.generate_keys(private_key_path='/etc/wireguard/server_private.key', public_key_path='/etc/wireguard/server_public.key')
+  server.init(private_key=server.private_key, listen_port=51820, address='192.168.21.1/24')
+  server.add_peer(peer_public_key='<client_public_key>', allowed_ips='192.168.21.2/32')
+  server.start()
+  debug(type(self).__name__, server.status())
+
+  # client = WGClient(interface='wg0')
+  # client.generate_keys(private_key_path='/etc/wireguard/client_private.key', public_key_path='/etc/wireguard/client_public.key')
+  # client.init(private_key=client.private_key, server_public_key='<server_public_key>', endpoint='<server_ip>:51820')
+  # client.start()
+  # debug(type(self).__name__, client.status())

sciveo-0.1.19/sciveo/version.py

@@ -0,0 +1,2 @@
+
+__version__ = '0.1.19'

{sciveo-0.1.17 → sciveo-0.1.19}/sciveo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sciveo
-Version: 0.1.17
+Version: 0.1.19
 Description-Content-Type: text/markdown
 Provides-Extra: mon
 Provides-Extra: net

{sciveo-0.1.17 → sciveo-0.1.19}/sciveo.egg-info/SOURCES.txt

@@ -26,6 +26,7 @@ sciveo/common/tools/formating.py
 sciveo/common/tools/hardware.py
 sciveo/common/tools/logger.py
 sciveo/common/tools/random.py
+sciveo/common/tools/remote.py
 sciveo/common/tools/synchronized.py
 sciveo/common/tools/timers.py
 sciveo/content/__init__.py
@@ -39,6 +40,7 @@ sciveo/monitoring/start.py
 sciveo/network/__init__.py
 sciveo/network/camera.py
 sciveo/network/tools.py
+sciveo/network/tunnel.py
 test/test_compress.py
 test/test_configuration.py
 test/test_crypto.py

sciveo-0.1.17/sciveo/version.py

@@ -1,2 +0,0 @@
-
-__version__ = '0.1.17'