Being towards death

pip install google-generativeai

import google.generativeai as genai
genai.configure(api_key="YOUR_GEMINI_API_KEY")
model = genai.GenerativeModel('gemini-2.0-flash')

def analyze_with_gemini(recon_data, api_key):
    try:
        genai.configure(api_key=api_key)
        model = genai.GenerativeModel("gemini-2.0-flash")

        prompt = f"""
        Analyze the following reconnaissance JSON report for the domain `{recon_data.get("domain")}`:

        {json.dumps(recon_data, indent=2)}

        Please provide:
        1. Key positive findings (what looks secure or well-configured)
        2. Key concerns or risks identified
        3. Specific actionable next steps to improve security posture

        Be concise and structured.
        """

        response = model.generate_content(prompt)
        return response.text

    except Exception as e:
        return f"AI Analysis Error: {str(e)}"

import subprocess
import json
import re
import whois
import os
import tempfile
import uuid
from datetime import datetime
import google.generativeai as genai

def clean_output(output):
    # 清理输出的ANSI转义序列
    ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])')
    return ansi_escape.sub('', output)

# 用户输入目标域名
DOMAIN = input("Enter the target domain: ").strip()

# WHOIS查询功能
def get_whois_info(domain):
    try:
        w = whois.whois(domain)
        return {
            "domain_name": w.domain_name,
            "registrar": w.registrar,
            "creation_date": str(w.creation_date),
            "expiration_date": str(w.expiration_date),
            "name_servers": w.name_servers
        }
    except Exception as e:
        return {"error": str(e)}

# Nmap扫描实现
def run_nmap_scan(domain):
    try:
        print("[*] Running Nmap scan...")
        result = subprocess.run(["nmap", "--top-ports", "1000", domain], capture_output=True, text=True)
        cleaned = clean_output(result.stdout)
        return cleaned.splitlines()
    except Exception as e:
        return str(e)

# ...（其他工具函数保持原样）...

# Gemini AI分析核心
def analyze_with_gemini(recon_data, api_key):
    try:
        genai.configure(api_key=api_key)
        model = genai.GenerativeModel("gemini-2.0-flash")

        prompt = f"""
        Analyze reconnaissance data for {recon_data.get("domain")}:
        {json.dumps(recon_data, indent=2)}
        
        Provide:
        1. Security strengths
        2. Critical risks
        3. Actionable recommendations
        """

        response = model.generate_content(prompt)
        return response.text

    except Exception as e:
        return f"AI Analysis Error: {str(e)}"

# 主执行流程
print("\n[+] Starting Recon on:", DOMAIN)
recon_results = {
    "domain": DOMAIN,
    "whois": get_whois_info(DOMAIN),
    "port_scan": run_nmap_scan(DOMAIN),
    "sslscan": run_sslscan(DOMAIN),
    "web_fingerprint": run_web_fingerprint(DOMAIN),
    "directory_bruteforce": run_dirsearch(DOMAIN),
    "amass_subdomains": run_amass(DOMAIN),
    "secretfinder": run_secretfinder(DOMAIN),
    "nuclei_findings": run_nuclei(DOMAIN),
    "dnsrecon": run_dnsrecon(DOMAIN),
}

GEMINI_API_KEY = "YOUR_API_KEY"
print("[*] Running AI analysis via Gemini...")
ai_analysis = analyze_with_gemini(recon_results, GEMINI_API_KEY)
recon_results["ai_analysis"] = ai_analysis.splitlines()

# 保存结果
output_file = "recon_results.json"
with open(output_file, "w") as json_file:
    json.dump(recon_results, json_file, indent=4)

print(f"\n[] Recon results saved to {output_file}")