sciveo 0.1.16__tar.gz → 0.1.17__tar.gz

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

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

{sciveo-0.1.16 → sciveo-0.1.17}/sciveo/__init__.py

@@ -21,7 +21,7 @@ try:
   from sciveo.content.runner import ProjectRunner
   from sciveo.content.dataset import Dataset
   from sciveo.monitoring.start import MonitorStart
-  from sciveo.monitoring.network import NetworkTools
+  from sciveo.network.tools import NetworkTools
   from sciveo.version import __version__
 
 

{sciveo-0.1.16 → sciveo-0.1.17}/sciveo/cli.py

@@ -16,7 +16,7 @@ import argparse
 from sciveo.common.tools.logger import *
 from sciveo.common.tools.timers import Timer
 from sciveo.monitoring.start import MonitorStart
-from sciveo.monitoring.network import NetworkTools
+from sciveo.network.tools import NetworkTools
 from sciveo.common.tools.configuration import GlobalConfiguration
 
 

sciveo-0.1.17/sciveo/network/camera.py

@@ -0,0 +1,63 @@
+import threading
+import cv2
+import numpy as np
+
+from sciveo.common.tools.logger import *
+from sciveo.common.tools.timers import FPSCounter
+from sciveo.network.tools import StreamSniffer
+
+
+class RTSPStreamSniffer(StreamSniffer):
+  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)
+
+  def get_rtsp_frames(self, ip_src):
+    frames = []
+    current_packets = self.get_ip_stream(ip_src)
+    for packet in current_packets:
+      frame = self.packet_to_frame(packet)
+      if frame is not None:
+        frames.append(frame)
+    return frames
+
+  def packet_to_frame(self, packet):
+    payload = bytes(packet[TCP].payload)
+    nparr = np.frombuffer(payload, np.uint8)
+    frame = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+    return frame
+
+  def play(self, ip_src):
+    fps = FPSCounter(tag=f"play {ip_src}")
+    while True:
+      frames = self.get_rtsp_frames(ip_src)
+      for frame in frames:
+        fps.update()
+        cv2.imshow(f'RTSP Stream from {ip_src}', frame)
+        if cv2.waitKey(1) & 0xFF == ord('q'):
+          return
+
+
+if __name__ == '__main__':
+  sniffer = RTSPStreamSniffer(iface="eth0")
+  sniffer.start()
+
+  camera_ips = []
+
+  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()

sciveo-0.1.16/sciveo/monitoring/network.py → sciveo-0.1.17/sciveo/network/tools.py

@@ -11,6 +11,7 @@
 
 import socket
 import threading
+from scapy.all import sniff, IP, TCP
 
 from sciveo.common.tools.logger import *
 from sciveo.common.tools.timers import Timer
@@ -92,5 +93,60 @@ class NetworkTools:
       pass
 
 
+
+class StreamSniffer:
+  def __init__(self, iface=None):
+    self.iface = iface
+    self.running = False
+    self.lock = threading.Lock()
+    self.streams = {}
+
+  def start(self):
+    self.running = True
+    self.sniff_thread = threading.Thread(target=self.sniff_packets)
+    self.sniff_thread.start()
+
+  def stop(self):
+    debug(type(self).__name__, "stopping...")
+    self.running = False
+    self.sniff_thread.join()
+
+  def sniff_packets(self):
+    debug(type(self).__name__, "start sniffing on", self.iface)
+    sniff(iface=self.iface, prn=self.on_packet, stop_filter=self.should_stop)
+
+  def on_packet(self, packet):
+    if IP in packet:
+      self.append_ip_packet(packet)
+
+  def should_stop(self, packet):
+    return not self.running
+
+  def append_ip_packet(self, packet):
+    ip_src = packet[IP].src
+    with self.lock:
+      self.streams.setdefault(ip_src, [])
+      self.streams[ip_src].append(packet)
+
+  def get_ip_stream(self, ip):
+    current_packets = []
+    with self.lock:
+      if ip in self.streams:
+        current_packets = self.streams[ip][:]
+        self.streams[ip] = []
+    return current_packets
+
+  def keys(self):
+    with self.lock:
+      return list(self.streams.keys())
+
+
 if __name__ == "__main__":
-  debug(NetworkTools(timeout=1.0, localhost=False).scan_port(port=9901))
+  # debug(NetworkTools(timeout=1.0, localhost=False).scan_port(port=9901))
+
+  import time
+  sniffer = StreamSniffer(iface="en0")
+  sniffer.start()
+  time.sleep(5)
+  sniffer.stop()
+  debug(sniffer.keys())

sciveo-0.1.17/sciveo/version.py

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

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

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

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

@@ -35,8 +35,10 @@ sciveo/content/project.py
 sciveo/content/runner.py
 sciveo/monitoring/__init__.py
 sciveo/monitoring/monitor.py
-sciveo/monitoring/network.py
 sciveo/monitoring/start.py
+sciveo/network/__init__.py
+sciveo/network/camera.py
+sciveo/network/tools.py
 test/test_compress.py
 test/test_configuration.py
 test/test_crypto.py

{sciveo-0.1.16 → sciveo-0.1.17}/sciveo.egg-info/requires.txt

@@ -4,6 +4,7 @@ requests>=0.0.0
 [all]
 psutil>=0.0.0
 netifaces>=0.0.0
+scapy>=0.0.0
 pycryptodome>=0.0.0
 
 [mon]
@@ -11,3 +12,4 @@ psutil>=0.0.0
 
 [net]
 netifaces>=0.0.0
+scapy>=0.0.0

{sciveo-0.1.16 → sciveo-0.1.17}/setup.py

@@ -28,10 +28,12 @@ setup(
       ],
       'net': [
         'netifaces>=0.0.0',
+        'scapy>=0.0.0',
       ],
       'all': [
         'psutil>=0.0.0',
         'netifaces>=0.0.0',
+        'scapy>=0.0.0',
         'pycryptodome>=0.0.0'
       ]
     },

sciveo-0.1.16/sciveo/version.py

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