Introducing D2GS Admin

import pydivert
import threading
import subprocess
import time
import psutil
import ctypes
import json

log_file = "blocked_packets.log"  # Specify the log file path
ban_duration = 300  # Ban duration in seconds
banned_ips_file = "banned_ips.json"  # File to store banned IPs

# Set the window title
ctypes.windll.kernel32.SetConsoleTitleW("Revenge Firewall")

# Load banned IPs from file
def load_banned_ips():
    try:
        with open(banned_ips_file, "r") as f:
            return json.load(f)
    except (FileNotFoundError, json.JSONDecodeError):
        return {}

# Save banned IPs to file
def save_banned_ips(banned_ips):
    with open(banned_ips_file, "w") as f:
        json.dump(banned_ips, f)

# Initialize banned IPs dictionary
banned_ips = load_banned_ips()

def block_packet(packet, w):
    payload = bytes(packet.tcp.payload)
    if packet.tcp.dst_port == 4000 and payload.startswith(b'\xff\x01'):
        source_ip = packet.src_addr
        if source_ip in banned_ips:
            ban_start_time = banned_ips[source_ip]
            ban_elapsed_time = time.time() - ban_start_time
            if ban_elapsed_time < ban_duration:
                # IP is still banned, don't send the packet
                print(f"IP {source_ip} is banned. Packet dropped.", flush=True)
                return
            else:
                # Ban duration has elapsed, remove the IP from the banned list
                del banned_ips[source_ip]
                save_banned_ips(banned_ips)

        # Log blocked packet and source IP
        log_message = f"Blocked packet: {payload}\nSource IP: {source_ip}\n"
        with open(log_file, "a") as f:
            f.write(log_message)

        # Display blocked packet and source IP
        print(log_message, flush=True)

        # Ban the source IP
        banned_ips[source_ip] = time.time()
        save_banned_ips(banned_ips)

        # Do not send the packet to effectively block it

    else:
        # Allow the packet to pass through
        w.send(packet)

def packet_capture():
    print("Starting packet capture...")
    with pydivert.WinDivert("tcp.DstPort == 4000") as w:
        for packet in w:
            block_packet(packet, w)

# Start packet capture in a separate thread
packet_capture_thread = threading.Thread(target=packet_capture)
packet_capture_thread.start()

print("Scanning packets on port 4000...")

# Continuously check if the process is running and restart if necessary
while True:
    process_name = "D2GS.exe"

    # Check if the process is running
    process_running = any(
        proc.name() == process_name for proc in psutil.process_iter()
    )

    if not process_running:
        # Restart the process
        process_path = r"C:\Users\Mantenimiento\Desktop\Server\D2GS\D2GS.exe"
        try:
            subprocess.Popen(
                process_path,
                shell=True,
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                creationflags=subprocess.CREATE_NEW_CONSOLE,
            )
            print(f"Process '{process_name}' restarted.")
        except subprocess.CalledProcessError:
            print(f"Failed to restart process '{process_name}'.")

    # Wait for a few seconds before checking again
    time.sleep(5)