Merge simtrace-analyser-server into simtrace2_pysniff.server
Add simtrace2-pysniff-server CLI — HTTP API for simtrace-analyser PWA. Two capture modes: GSMTAP UDP listener (compatible with simtrace2-sniff) or direct SIMtrace2 hardware capture. SQLite storage with sessions + messages tables. REST API with CORS. server/ sub-package: database.py, capture.py, server.py, __main__.py README updated with server usage section. pyproject.toml: simtrace2-pysniff-server script entry added.
This commit is contained in:
1 parent
a130b97cce
commit
17eeda27f3
7 files changed
+593
-2
No files matched your search
@@ -0,0 +1,111 @@
|
||||
"""Capture backends: GSMTAP listener and direct SIMtrace2 sniffer."""
|
||||
|
||||
import time
|
||||
import threading
|
||||
|
||||
from ..gsmtap import GsmtapReceiver, GSMTAP_SIM_ATR, GSMTAP_SIM_APDU
|
||||
|
||||
|
||||
class GsmtapListener:
|
||||
def __init__(self, bind_port=4729):
|
||||
self._receiver = GsmtapReceiver(bind_port=bind_port)
|
||||
self._running = False
|
||||
|
||||
def start(self):
|
||||
self._running = True
|
||||
|
||||
def stop(self):
|
||||
self._running = False
|
||||
self._receiver.close()
|
||||
|
||||
def iter_messages(self):
|
||||
while self._running:
|
||||
sub_type, data = self._receiver.read_packet()
|
||||
if sub_type is None:
|
||||
continue
|
||||
msg_type = 'atr' if sub_type == GSMTAP_SIM_ATR else 'tpdu'
|
||||
yield msg_type, data
|
||||
|
||||
|
||||
class DirectSniffer:
|
||||
def __init__(self):
|
||||
from ..device import SniffSession
|
||||
self._session = SniffSession(
|
||||
reconnect=True,
|
||||
reconnect_delay_min=1.0,
|
||||
reconnect_delay_max=10.0,
|
||||
inactivity_timeout=0.0,
|
||||
)
|
||||
self._running = False
|
||||
|
||||
def start(self):
|
||||
self._running = True
|
||||
|
||||
def stop(self):
|
||||
self._running = False
|
||||
self._session.close()
|
||||
|
||||
def iter_messages(self):
|
||||
for msg in self._session.iter_messages():
|
||||
if not self._running:
|
||||
break
|
||||
yield msg.type, msg.data
|
||||
|
||||
|
||||
class CaptureManager:
|
||||
def __init__(self, backend, db):
|
||||
self._backend = backend
|
||||
self._db = db
|
||||
self._session_id = None
|
||||
self._thread = None
|
||||
self._start_time = 0.0
|
||||
self._latest_msg_id = 0
|
||||
|
||||
@property
|
||||
def active(self):
|
||||
return self._session_id is not None
|
||||
|
||||
@property
|
||||
def session_id(self):
|
||||
return self._session_id
|
||||
|
||||
@property
|
||||
def start_time(self):
|
||||
return self._start_time
|
||||
|
||||
@property
|
||||
def latest_msg_id(self):
|
||||
return self._latest_msg_id
|
||||
|
||||
def start_session(self):
|
||||
if self._session_id is not None:
|
||||
self.stop_session()
|
||||
|
||||
mode = 'gsmtap' if isinstance(self._backend, GsmtapListener) else 'direct'
|
||||
self._session_id = self._db.create_session(mode)
|
||||
self._start_time = time.monotonic()
|
||||
self._latest_msg_id = 0
|
||||
self._backend.start()
|
||||
|
||||
self._thread = threading.Thread(target=self._capture_loop, daemon=True)
|
||||
self._thread.start()
|
||||
|
||||
return self._session_id
|
||||
|
||||
def stop_session(self):
|
||||
if self._session_id is None:
|
||||
return
|
||||
self._backend.stop()
|
||||
self._db.close_session(self._session_id)
|
||||
sid = self._session_id
|
||||
self._session_id = None
|
||||
self._start_time = 0.0
|
||||
return sid
|
||||
|
||||
def _capture_loop(self):
|
||||
for msg_type, data in self._backend.iter_messages():
|
||||
if self._session_id is None:
|
||||
break
|
||||
elapsed = round(time.monotonic() - self._start_time, 3)
|
||||
mid = self._db.insert_message(self._session_id, elapsed, msg_type, data)
|
||||
self._latest_msg_id = mid
|
||||
Reference in new issue
Block a user