Module todo: DMARK/DKIM/SPF #575

# Checking for common misconfigurations in SPF, DKIM, and DMARC records

def check_spf_misconfigurations(record):
    issues = []
    if 'all' in record:
        if '+all' in record:
            issues.append("SPF configured to allow all hosts.")
    return issues

def check_dkim_misconfigurations(record):
    issues = []
    if 'k=rsa;' not in record:
        issues.append("DKIM is not using RSA for signing.")
    return issues

def check_dmarc_misconfigurations(record):
    issues = []
    if 'p=none' in record:
        issues.append("DMARC policy is set to 'none'.")
    return issues

def check_email_security(domain):
    all_issues = {}

    # Check SPF record
    spf_issues = []
    try:
        spf_record = dns.resolver.resolve(domain, 'TXT')
        spf_text = [r.to_text() for r in spf_record if 'v=spf1' in r.to_text()][0]
        spf_issues = check_spf_misconfigurations(spf_text)
    except dns.resolver.NoAnswer:
        spf_issues.append("No SPF record found.")
    all_issues['SPF'] = spf_issues

    # Check DKIM record
    dkim_issues = []
    try:
        dkim_record = dns.resolver.resolve(f"selector1._domainkey.{domain}", 'TXT')
        dkim_text = dkim_record[0].to_text()
        dkim_issues = check_dkim_misconfigurations(dkim_text)
    except dns.resolver.NoAnswer:
        dkim_issues.append("No DKIM record found.")
    all_issues['DKIM'] = dkim_issues

    # Check DMARC record
    dmarc_issues = []
    try:
        dmarc_record = dns.resolver.resolve(f"_dmarc.{domain}", 'TXT')
        dmarc_text = dmarc_record[0].to_text()
        dmarc_issues = check_dmarc_misconfigurations(dmarc_text)
    except dns.resolver.NoAnswer:
        dmarc_issues.append("No DMARC record found.")
    all_issues['DMARC'] = dmarc_issues

    return all_issues

domain = "example.com"
issues = check_email_security(domain)
for key, value in issues.items():
    if value:
        print(f"Potential {key} issues for {domain}:")
        for issue in value:
            print(f"- {issue}")