optimize_network.py

import joblib
import numpy as np
import os
import subprocess

# -------------------------------
# CONFIG
# -------------------------------
MODEL_PATH = "models/random_forest_regressor.pkl"
THROUGHPUT_THRESHOLD = 10.0  # Mbps
MININET_HOST = "h1"
INTERFACE = "h1-eth0"

# -------------------------------
# LOAD MODEL
# -------------------------------
try:
    model = joblib.load(MODEL_PATH)
except Exception as e:
    print("❌ Failed to load model:", e)
    exit(1)

# -------------------------------
# USER INPUT
# -------------------------------
try:
    rtt = float(input("Enter current RTT (ms): "))
    jitter = float(input("Enter current Jitter (ms): "))
except ValueError:
    print("❌ Invalid input. Please enter numeric values.")
    exit(1)

X = np.array([[rtt, jitter]])

# -------------------------------
# PREDICT THROUGHPUT
# -------------------------------
predicted_throughput = model.predict(X)[0]

print(f"\n📡 Predicted Throughput: {predicted_throughput:.2f} Mbps")

# -------------------------------
# CONGESTION LOGIC
# -------------------------------
if predicted_throughput < THROUGHPUT_THRESHOLD:
    status = "SEVERE CONGESTION"
    action = "Reduce bandwidth, increase queue, reroute traffic"
else:
    status = "NORMAL"
    action = "No action needed"

print(f"📊 Network Status: {status}")
print(f"🛠 Recommended Action: {action}")

# -------------------------------
# APPLY TC (ONLY IF MININET RUNNING)
# -------------------------------
def is_mininet_running():
    try:
        output = subprocess.check_output(["pgrep", "-f", f"mininet:{MININET_HOST}"])
        return True
    except subprocess.CalledProcessError:
        return False

if status == "SEVERE CONGESTION":
    print("\n⚠️ Attempting to apply bandwidth rule (only works if Mininet is running)...")

    if is_mininet_running():
        try:
            # Clear existing qdisc
            os.system(
                f"sudo mnexec -a $(pgrep -f 'mininet:{MININET_HOST}') "
                f"tc qdisc del dev {INTERFACE} root 2>/dev/null"
            )

            # Apply delay (example action)
            os.system(
                f"sudo mnexec -a $(pgrep -f 'mininet:{MININET_HOST}') "
                f"tc qdisc add dev {INTERFACE} root netem delay 50ms"
            )

            print("✅ Traffic control rule applied successfully on h1-eth0")

        except Exception as e:
            print("❌ Failed to apply tc rule:", e)
    else:
        print("ℹ️ Mininet not running → skipping tc command")