168 lines
7.2 KiB
Python
168 lines
7.2 KiB
Python
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()
|