mirror of
https://github.com/infinition/Bjorn.git
synced 2026-03-15 08:52:00 +00:00
aac77a3e76
- Implemented LokiUtils class with GET and POST endpoints for managing scripts, jobs, and payloads. - Added SentinelUtils class with GET and POST endpoints for managing events, rules, devices, and notifications. - Both classes include error handling and JSON response formatting.
117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
"""
|
|
Bifrost DB operations — networks, handshakes, epochs, activity, peers, plugin data.
|
|
"""
|
|
import logging
|
|
|
|
from logger import Logger
|
|
|
|
logger = Logger(name="db_utils.bifrost", level=logging.DEBUG)
|
|
|
|
|
|
class BifrostOps:
|
|
def __init__(self, base):
|
|
self.base = base
|
|
|
|
def create_tables(self):
|
|
"""Create all Bifrost tables."""
|
|
|
|
# WiFi networks discovered by Bifrost
|
|
self.base.execute("""
|
|
CREATE TABLE IF NOT EXISTS bifrost_networks (
|
|
bssid TEXT PRIMARY KEY,
|
|
essid TEXT DEFAULT '',
|
|
channel INTEGER DEFAULT 0,
|
|
encryption TEXT DEFAULT '',
|
|
rssi INTEGER DEFAULT 0,
|
|
vendor TEXT DEFAULT '',
|
|
num_clients INTEGER DEFAULT 0,
|
|
first_seen TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
last_seen TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
handshake INTEGER DEFAULT 0,
|
|
deauthed INTEGER DEFAULT 0,
|
|
associated INTEGER DEFAULT 0,
|
|
whitelisted INTEGER DEFAULT 0
|
|
)
|
|
""")
|
|
|
|
# Captured handshakes
|
|
self.base.execute("""
|
|
CREATE TABLE IF NOT EXISTS bifrost_handshakes (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
ap_mac TEXT NOT NULL,
|
|
sta_mac TEXT NOT NULL,
|
|
ap_essid TEXT DEFAULT '',
|
|
channel INTEGER DEFAULT 0,
|
|
rssi INTEGER DEFAULT 0,
|
|
filename TEXT DEFAULT '',
|
|
captured_at TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
uploaded INTEGER DEFAULT 0,
|
|
cracked INTEGER DEFAULT 0,
|
|
UNIQUE(ap_mac, sta_mac)
|
|
)
|
|
""")
|
|
|
|
# Epoch history
|
|
self.base.execute("""
|
|
CREATE TABLE IF NOT EXISTS bifrost_epochs (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
epoch_num INTEGER NOT NULL,
|
|
started_at TEXT NOT NULL,
|
|
duration_secs REAL DEFAULT 0,
|
|
num_deauths INTEGER DEFAULT 0,
|
|
num_assocs INTEGER DEFAULT 0,
|
|
num_handshakes INTEGER DEFAULT 0,
|
|
num_hops INTEGER DEFAULT 0,
|
|
num_missed INTEGER DEFAULT 0,
|
|
num_peers INTEGER DEFAULT 0,
|
|
mood TEXT DEFAULT 'ready',
|
|
reward REAL DEFAULT 0,
|
|
cpu_load REAL DEFAULT 0,
|
|
mem_usage REAL DEFAULT 0,
|
|
temperature REAL DEFAULT 0,
|
|
meta_json TEXT DEFAULT '{}'
|
|
)
|
|
""")
|
|
|
|
# Activity log (event feed)
|
|
self.base.execute("""
|
|
CREATE TABLE IF NOT EXISTS bifrost_activity (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
timestamp TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
event_type TEXT NOT NULL,
|
|
title TEXT NOT NULL,
|
|
details TEXT DEFAULT '',
|
|
meta_json TEXT DEFAULT '{}'
|
|
)
|
|
""")
|
|
self.base.execute(
|
|
"CREATE INDEX IF NOT EXISTS idx_bifrost_activity_ts "
|
|
"ON bifrost_activity(timestamp DESC)"
|
|
)
|
|
|
|
# Peers (mesh networking — Phase 2)
|
|
self.base.execute("""
|
|
CREATE TABLE IF NOT EXISTS bifrost_peers (
|
|
peer_id TEXT PRIMARY KEY,
|
|
name TEXT DEFAULT '',
|
|
version TEXT DEFAULT '',
|
|
face TEXT DEFAULT '',
|
|
encounters INTEGER DEFAULT 0,
|
|
last_channel INTEGER DEFAULT 0,
|
|
last_seen TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
first_seen TEXT DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
""")
|
|
|
|
# Plugin persistent state
|
|
self.base.execute("""
|
|
CREATE TABLE IF NOT EXISTS bifrost_plugin_data (
|
|
plugin_name TEXT NOT NULL,
|
|
key TEXT NOT NULL,
|
|
value TEXT DEFAULT '',
|
|
PRIMARY KEY (plugin_name, key)
|
|
)
|
|
""")
|
|
|
|
logger.debug("Bifrost tables created/verified")
|