Intial commit. Some stuff I made last night.

2024-05-26 13:07:49 +03:00
parent 1984e7e511
commit 93e785c2ae
3 changed files with 243 additions and 0 deletions
--- a/traceroute_collector.py
+++ b/traceroute_collector.py
@ -0,0 +1,126 @@
+#!/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)