import os import subprocess import random import time import datetime import requests import re import sys from psql_utils import connect_to_database, update_or_insert_ip from ovpn_downloader import download_and_extract_ovpn_configs, download_ovpn_if_needed from openvpn_manager import establish_vpn_connection, is_vpn_active, disconnect_vpn, get_external_ip # Variables CREDENTIAL_FILE = "nord.creds" OVPN_DIR = "ovpn_configs/ovpn_tcp" LOG_FILE = "openvpn.log" # This is not used anymore RESULT_FILE = "vpnlist.txt" PING_URL = "https://health.ext.ben.io/ping/a1a55915-1051-48ed-bd13-ea3f1717a3ee" # Define DEBUG_MODE DEBUG_MODE = False if len(sys.argv) > 1 and sys.argv[1] == "-d": DEBUG_MODE = True def debug_print(message): if DEBUG_MODE: print(f"{datetime.datetime.now()} + [Debug Mode]:{message}") try: requests.get(PING_URL, timeout=10) print(f"{datetime.datetime.now()} [Main Script]: Ping sent to: %s" % PING_URL) except requests.RequestException as e: # Log ping failure here... print(f"{datetime.datetime.now()} [Main Script]: Ping failed: %s" % e) # Function to check if the script is already running def is_script_running(): """Checks if another instance of the script is already running.""" try: # Use pgrep to find processes with the exact command line of the script command_line = f"/usr/bin/python3 {os.path.abspath(__file__)} >> /var/log/nord-checker.log 2>&1" # Include redirection process = subprocess.run(["pgrep", "-f", command_line], capture_output=True, text=True) # If pgrep finds any matches, it means the script is already running if process.returncode == 0 and process.stdout: # Get the PIDs of the running instances pids = [int(pid) for pid in process.stdout.strip().split("\n")] # Exclude the current process's PID pids = [pid for pid in pids if pid != os.getpid()] # Only quit if there are two or more other instances running if len(pids) >= 2: for pid in pids: try: # Get the CMD value using ps process_info = subprocess.check_output(["ps", "-o", "cmd=", "-p", str(pid)], text=True).strip() print(f"{datetime.datetime.now()} [Main Script]: Another instance of the script is already running (PID: {pid}): {process_info}") except subprocess.CalledProcessError: print(f"{datetime.datetime.now()} [Main Script]: Another instance of the script is already running (PID: {pid}), but unable to retrieve process information.") return True except Exception as e: print(f"{datetime.datetime.now()} [Main Script]: Error checking for running instances: {e}") return False # Get current IP before starting (using get_external_ip) CURRENT_IP = get_external_ip() if CURRENT_IP is None: print(f"{datetime.datetime.now()} [Main Script]: Unable to get current IP. Exiting.") sys.exit(1) # Exit if unable to get current IP else: print(f"{datetime.datetime.now()} [Main Script]: Current IP before VPN connections: {CURRENT_IP}") if is_script_running(): sys.exit(1) # Terminate any existing OpenVPN connections try: result = subprocess.run(["pgrep", "openvpn"], check=True, capture_output=True) pids = result.stdout.decode().strip().split('\n') # Get PIDs as a list for pid in pids: subprocess.run(["kill", pid]) print(f"{datetime.datetime.now()} [Main Script]: Existing OpenVPN connections terminated.") except subprocess.CalledProcessError: pass # No openvpn process found # Connect to PostgreSQL database conn, cursor = connect_to_database() # Download and extract OVPN configurations, and get the list of files to check debug_print(f"{datetime.datetime.now()} [Main Script]: Downloading and extracting OVPN configurations...") ovpn_files_to_check = download_and_extract_ovpn_configs(cursor, conn, debug_print=debug_print) total_files = len(ovpn_files_to_check) debug_print(f"{datetime.datetime.now()} [Main Script]: Found {total_files} OVPN files to check.") print(f"{datetime.datetime.now()} [Main Script]: Found {total_files} OVPN files to check.") # Process each OVPN file FILE_NUM = 1 for OVPN_FILE in ovpn_files_to_check: OVPN_FILENAME = os.path.basename(OVPN_FILE) debug_print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: Processing OVPN file...") # Check if the OVPN file exists, download if needed file_exists = download_ovpn_if_needed(OVPN_FILENAME, cursor, conn, debug_print=debug_print) if not file_exists: print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: Skipping, file not found and removed from database.") FILE_NUM += 1 continue # Establish VPN connection connection_successful, connection_message = establish_vpn_connection(f"{OVPN_DIR}/{OVPN_FILE}", CREDENTIAL_FILE, LOG_FILE, debug_print=debug_print) if connection_successful: print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: {connection_message}") if not connection_successful: print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: Error starting OpenVPN connection:{connection_message}") # Wait for a random time between 1 and 4 seconds sleep_time = random.randint(8, 15) print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: Waiting for {sleep_time} seconds before the next connection...") time.sleep(sleep_time) print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: Resuming...") continue # Skip to the next OVPN file # Check if the VPN connection is active (using the function from openvpn_manager.py) try: subprocess.run(["pgrep", "-f", OVPN_FILENAME], check=True, capture_output=True) # Check external IP (with validation and retries) external_ip = get_external_ip() if external_ip is None: print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: Unable to get external IP. Skipping this OVPN file.") continue print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: External IP via VPN: {external_ip}") # Update or insert into database update_or_insert_ip(cursor, conn, OVPN_FILENAME, external_ip) except subprocess.CalledProcessError: print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: VPN connection failed to establish.") # Disconnect VPN disconnect_vpn() print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: VPN disconnected.") # Wait for a random time between 5 and 8 seconds sleep_time = random.randint(5, 8) print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: Waiting for {sleep_time} seconds before the next connection...") time.sleep(sleep_time) print(f"{datetime.datetime.now()} [{FILE_NUM}/{total_files}] [{OVPN_FILENAME}]: Resuming...") # Increment file number FILE_NUM += 1 # Close the database connection cursor.close() conn.close()