DKIM should never be inferred from reports

This commit is contained in:
2026-05-20 14:12:57 -03:00
parent c91c3f1023
commit c68ccc8065
4 changed files with 9 additions and 107 deletions
-41
View File
@@ -29,21 +29,11 @@ class ParsedSpfRecord:
all_mechanism: str | None = None
@dataclass
class DkimRecord:
selector: str
domain: str
query_name: str
record: str | None = None
error: str | None = None
@dataclass
class DomainDnsPolicy:
domain: str
dmarc: ParsedDmarcRecord = field(default_factory=ParsedDmarcRecord)
spf: ParsedSpfRecord = field(default_factory=ParsedSpfRecord)
dkim: list[DkimRecord] = field(default_factory=list)
mx_records: list[str] = field(default_factory=list)
errors: list[str] = field(default_factory=list)
@@ -139,7 +129,6 @@ def parse_spf_records(records: list[str]) -> tuple[ParsedSpfRecord, list[str]]:
def collect_domain_dns_policy(
domain: str,
*,
selectors: list[str | tuple[str, str]] | None = None,
txt_lookup: TxtLookup | None = None,
mx_lookup: MxLookup | None = None,
) -> DomainDnsPolicy:
@@ -165,34 +154,4 @@ def collect_domain_dns_policy(
except Exception as exc:
policy.errors.append(f"MX lookup failed: {exc}")
selector_domains: set[tuple[str, str]] = set()
for item in selectors or []:
if isinstance(item, tuple):
selector, dkim_domain = item
else:
selector, dkim_domain = item, domain
selector = (selector or "").strip().lower()
dkim_domain = (dkim_domain or domain).strip().lower().rstrip(".")
if selector and dkim_domain:
selector_domains.add((selector, dkim_domain))
for selector, dkim_domain in sorted(selector_domains):
query_name = f"{selector}._domainkey.{dkim_domain}"
try:
records = txt_lookup(query_name)
dkim_records = [record for record in records if record.strip().lower().startswith("v=dkim1")]
policy.dkim.append(
DkimRecord(
selector=selector,
domain=dkim_domain,
query_name=query_name,
record=dkim_records[0] if dkim_records else None,
)
)
if not dkim_records:
policy.errors.append(f"DKIM record not found for selector {selector} on {dkim_domain}")
except Exception as exc:
policy.dkim.append(DkimRecord(selector=selector, domain=dkim_domain, query_name=query_name, error=str(exc)))
policy.errors.append(f"DKIM lookup failed for selector {selector} on {dkim_domain}: {exc}")
return policy
+5 -33
View File
@@ -22,7 +22,7 @@ from app.dns_policy import DomainDnsPolicy, collect_domain_dns_policy
from app.homepage import domain_homepage_summary, domain_metrics, homepage_summary, latest_summary, resolve_date_range, traffic_distribution
from app.inbox_locks import InboxRunLease, inbox_run_locks
from app.jobs import import_jobs
from app.models import Alert, AuthResult, DailyStat, DomainDnsSnapshot, InboxStatus, LLMReport, Record, Report, SkippedReportPayload, utcnow
from app.models import Alert, DailyStat, DomainDnsSnapshot, InboxStatus, LLMReport, Record, Report, SkippedReportPayload, utcnow
from app.scheduler import generate_open_posture_summaries, scheduler_ok, start_scheduler
from app.schemas import BacklogRequest, ProcessNowRequest
from app.message_processor import process_inbox
@@ -400,24 +400,6 @@ def _json_list(value: str | None) -> list:
return data if isinstance(data, list) else []
def _observed_dkim_selectors(session: Session, domain: str) -> list[tuple[str, str]]:
rows = session.execute(
select(AuthResult.selector, AuthResult.domain)
.select_from(AuthResult)
.join(Record)
.join(Report)
.where(
Report.domain == domain,
AuthResult.auth_type == "dkim",
AuthResult.selector.is_not(None),
AuthResult.domain.is_not(None),
)
.distinct()
.order_by(AuthResult.domain, AuthResult.selector)
).all()
return [(selector, auth_domain) for selector, auth_domain in rows if selector and auth_domain]
def _snapshot_model(domain: str, policy: DomainDnsPolicy) -> DomainDnsSnapshot:
return DomainDnsSnapshot(
domain=domain,
@@ -433,7 +415,7 @@ def _snapshot_model(domain: str, policy: DomainDnsPolicy) -> DomainDnsSnapshot:
spf_record=policy.spf.raw,
spf_all=policy.spf.all_mechanism,
spf_includes_json=json.dumps(policy.spf.includes, sort_keys=True),
dkim_records_json=json.dumps([item.__dict__ for item in policy.dkim], sort_keys=True),
dkim_records_json="[]",
mx_records_json=json.dumps(policy.mx_records, sort_keys=True),
errors_json=json.dumps(policy.errors, sort_keys=True),
)
@@ -448,14 +430,6 @@ def _latest_dns_snapshot(session: Session, domain: str) -> SimpleNamespace | Non
)
if not snapshot:
return None
dkim_records = _json_list(snapshot.dkim_records_json)
dkim_found = [item for item in dkim_records if isinstance(item, dict) and item.get("record")]
dkim_missing = [item for item in dkim_records if isinstance(item, dict) and not item.get("record")]
dns_errors = [
item
for item in _json_list(snapshot.errors_json)
if isinstance(item, str) and not item.startswith("DKIM lookup failed") and not item.startswith("DKIM record not found")
]
return SimpleNamespace(
id=snapshot.id,
domain=snapshot.domain,
@@ -472,11 +446,9 @@ def _latest_dns_snapshot(session: Session, domain: str) -> SimpleNamespace | Non
spf_record=snapshot.spf_record,
spf_all=snapshot.spf_all,
spf_includes=_json_list(snapshot.spf_includes_json),
dkim_records=dkim_records,
dkim_found=dkim_found,
dkim_missing=dkim_missing,
dkim_records=_json_list(snapshot.dkim_records_json),
mx_records=_json_list(snapshot.mx_records_json),
errors=dns_errors,
errors=_json_list(snapshot.errors_json),
)
@@ -550,7 +522,7 @@ def domain_page(domain: str, request: Request, source_page: int = 1, alert_page:
@app.post("/domains/{domain}/dns/refresh", dependencies=dashboard_post_auth)
def refresh_domain_dns(domain: str, session: Session = Depends(get_db)):
policy = collect_domain_dns_policy(domain, selectors=_observed_dkim_selectors(session, domain))
policy = collect_domain_dns_policy(domain)
session.add(_snapshot_model(domain, policy))
session.commit()
return RedirectResponse(url=f"/domains/{domain}", status_code=303)
-26
View File
@@ -64,32 +64,6 @@
</div>
<code class="dw-dns-record">{{ dns_snapshot.spf_record or "No SPF record found." }}</code>
</article>
<article class="dw-dns-panel">
<div class="dw-list-row">
<span>DKIM</span>
<span class="dw-chip dw-chip-info">{{ dns_snapshot.dkim_found | length }} found{% if dns_snapshot.dkim_missing %}, {{ dns_snapshot.dkim_missing | length }} missing{% endif %}</span>
</div>
<div class="dw-dns-record-list">
{% for item in dns_snapshot.dkim_found %}
<div>
<strong>{{ item.selector }}._domainkey.{{ item.domain or domain }}</strong>
<code>{{ item.record }}</code>
</div>
{% else %}
<code>No currently resolvable observed DKIM selectors.</code>
{% endfor %}
</div>
{% if dns_snapshot.dkim_missing %}
<details class="dw-dns-missing">
<summary>{{ dns_snapshot.dkim_missing | length }} observed selector lookups did not resolve</summary>
<div class="dw-dns-missing-list">
{% for item in dns_snapshot.dkim_missing %}
<code>{{ item.query_name }}{% if item.error %}: {{ item.error }}{% endif %}</code>
{% endfor %}
</div>
</details>
{% endif %}
</article>
<article class="dw-dns-panel">
<div class="dw-list-row">
<span>MX</span>