#!/usr/bin/env python3.11 import subprocess import sqlite3 import re from datetime import datetime def run_traceroute(host): result = subprocess.run(['traceroute', host], stdout=subprocess.PIPE) return result.stdout.decode() def parse_traceroute_output(output): hops = [] lines = output.strip().split('\n')[1:] # Skip the first line (traceroute to ...) previous_ip = None for line in lines: parts = line.split() hop_number = int(parts[0]) ip_address = parts[1] latencies = [] for part in parts[2:]: if re.match(r'^[0-9+\s+ms$]', part): latency_str = re.sub(r'[^0-9.]', '', part) if latency_str and not latency_str == '': try: print(part) print(latency_str) latencies.append(float(latency_str)) except ValueError: print(f"Could not convert '{latency_str}' to float.") avg_latency = sum(latencies) / len(latencies) if latencies else None timestamp = datetime.now().isoformat() if previous_ip: hops.append({ 'hop_number': hop_number, 'source_ip': previous_ip, 'destination_ip': ip_address, 'latency': avg_latency, 'timestamp': timestamp, }) previous_ip = ip_address return hops def create_tables(): conn = sqlite3.connect('traceroute.db') cursor = conn.cursor() # Table to store unique links cursor.execute(''' CREATE TABLE IF NOT EXISTS links ( id INTEGER PRIMARY KEY, source_ip TEXT, destination_ip TEXT, UNIQUE(source_ip, destination_ip) ) ''') # Table to store latency and timestamp for each link cursor.execute(''' CREATE TABLE IF NOT EXISTS link_latency ( id INTEGER PRIMARY KEY, link_id INTEGER, latency REAL, timestamp TEXT, FOREIGN KEY (link_id) REFERENCES links (id) ) ''') conn.commit() conn.close() def store_traceroute(hops): conn = sqlite3.connect('traceroute.db') cursor = conn.cursor() for hop in hops: # Insert or ignore link into links table cursor.execute(''' INSERT OR IGNORE INTO links (source_ip, destination_ip) VALUES (?, ?) ''', (hop['source_ip'], hop['destination_ip'])) # Retrieve the link_id cursor.execute(''' SELECT id FROM links WHERE source_ip = ? AND destination_ip = ? ''', (hop['source_ip'], hop['destination_ip'])) link_id = cursor.fetchone()[0] # Insert latency data into link_latency table cursor.execute(''' INSERT INTO link_latency (link_id, latency, timestamp) VALUES (?, ?, ?) ''', (link_id, hop['latency'], hop['timestamp'])) conn.commit() conn.close() def retrieve_traceroute(): conn = sqlite3.connect('traceroute.db') cursor = conn.cursor() cursor.execute(''' SELECT l.source_ip, l.destination_ip, ll.latency, ll.timestamp FROM link_latency ll JOIN links l ON ll.link_id = l.id ORDER BY ll.timestamp ''') rows = cursor.fetchall() conn.close() return rows # Usage if __name__ == '__main__': create_tables() traceroute_output = run_traceroute('vi.fi') hops = parse_traceroute_output(traceroute_output) # for hop in hops: # print(hop) store_traceroute(hops) stored_hops = retrieve_traceroute() for hop in stored_hops: print(f"Link: {hop[0]} -> {hop[1]}, Latency: {hop[2]} ms, Timestamp: {hop[3]}") exit(0)