Architecture

August 1, 2026 · View on GitHub

This is the map: where everything lives and how a byte becomes a finding. It links to the authoritative in-code docs rather than restating them, so there is one source of truth per fact. Read this first; then jump to the cited module.


Repository layout

Every top-level directory, one line each. Code is Rust under crates/; everything else is data, tooling, docs, or eval harness.

DirRole
crates/Rust workspace: runtime code only (six crates; see below).
detectors/Embedded detector TOMLs (data, not code). One file = one secret type; drop a file to add a detector without rewriting detection logic. The generated catalog owns the current count. See the detector reference.
rules/Tier-B data (e.g. aws-canary-accounts.toml); same drop-in model as detectors/.
ml/Python pipeline for embedded weights.bin: harvest → blend → train → gate (retrain_loop.sh). Trains; crates/scanner serves.
benchmarks/Eval harness (bench/): corpora, scanner adapters, scorer, regression/differential gate, README leaderboard.
tests/Repo-level integration tests (Docker, install, cross-OS). Per-crate tests live under each crate's tests/.
fuzz/cargo-fuzz targets (structure-aware, one sink per target).
tools/Build-time generators (gen_contracts.py, gen_companion_contracts.py). Large gitignored SecretBench corpus.
scripts/Maintained dev/release entrypoints and organization/product-truth gates. One-off corpus rewrite scripts do not ship.
docs/src/The single canonical documentation set, built and deployed as mdBook.
demo/Self-contained demo deployment (app + infra + scripts).
metrics/Star and project-health metrics.

Internal execution planning lives in the private Santh monorepo, not in this public repository.


The crates and their layering

Dependencies point one way: core and profile are foundations and depend on no other KeyHog crate; cli sits on top and wires the rest together. This DAG is enforced by Cargo and must stay acyclic (domain logic never imports CLI/transport/UI).

                          cli
            orchestration · transport · process exits
             ┌─────────────┼──────────────┬──────────┐
             ▼             ▼              ▼          ▼
          scanner        sources       verifier   profile
          detection       inputs       live checks  timing · run state
             │  ╲          │  ╲           │          ▲
             │   ╲         │   ╲ optional │          │
             │    └────────┼────┼─────────┼──────────┘
             └─────────────┼────▼─────────┘

                          core
        types · detector registry · reports · dedup · caches

scanner and verifier depend on core. scanner also records fixed stages through profile. sources depends on core and, for network-enabled source features, reuses verifier for shared SSRF and request-signing policy. cli selects features, composes all five libraries, and owns the operator-run profile session.

CrateOwnsStart reading at
coreEmbedded detector loading, detector specs, the Finding/Credential types, reporters, dedup, allowlists, the Merkle incremental-scan cache, and confidence-calibration data.crates/core/src/lib.rs, spec.rs, finding.rs, report/
profileAllocation-free fixed-stage timing, causal run identity, state transitions, process resource sampling, and portable JSON and text profile records.crates/profile/src/lib.rs
scannerThe detection engine: hardware probing and backend dispatch, prefilters, compile, scan, decode-through, entropy, ML confidence, multiline handling, and suppression. Persisted CLI route selection is intentionally not owned here.crates/scanner/src/compiled_scanner/ (construction and lifecycle), engine/mod.rs (execution flow), adjudicate/, pipeline/, lib.rs
sourcesWhere bytes come from: filesystem, Git (staged/diff/history), stdin, Docker, S3, GCS, Azure Blob, GitHub, GitLab, Bitbucket, web, HAR, strings, and optional binary/decompiler inputs.crates/sources/src/lib.rs
verifierTurning a candidate into a verified-live credential: per-detector verify endpoints, SSRF/bogon guards, OOB, rate limiting.crates/verifier/src/lib.rs, verify/, ssrf.rs
cliThe user-facing binary: argument parsing, the scan orchestrator, daemon/watch, baselines, calibrate, hook installer, output formatting.crates/cli/src/lib.rs, args/, orchestrator/; main.rs owns process/signal startup only

The crate graph does not imply that every build exposes every source or backend. The official and default CLI builds enable the full documented network-source set. The portable, ci-lean, and ci profiles deliberately remove different accelerator or source features. Library callers select their own feature set.

Load-bearing boundary owner map

The crate DAG is not the whole shipping boundary. The Action entrypoint, automatic crates.io publication, and each load-bearing library or CLI handoff have one definitional owner. Wrappers may compose these owners; they must not restate their policy.

BoundaryDefinitional owner
Marketplace metadata, documented inputs/outputs, and top-level composite stepsaction.yml
Repository-local Action metadata consumed by GitHub workflows.github/actions/keyhog/action.yml
Action input validation, authenticated binary acquisition, scan invocation, exit mapping, and output publication.github/actions/keyhog/run-scan.sh
Automatic version, changelog, and crates.io publication.github/workflows/release.yml
CLI argument dispatch and setup-error exit routingcrates/cli/src/lib.rs::cli_main
Completed-scan exit precedencecrates/cli/src/orchestrator/run.rs::resolve_scan_exit
Curated source-crate export surfacecrates/sources/src/api.rs
Live-verification construction and executioncrates/verifier/src/lib.rs::VerificationEngine
Deduplicated match to report-safe finding conversioncrates/core/src/finding.rs::VerifiedFinding::from_deduped
Scanner execution flowcrates/scanner/src/engine/mod.rs

This table is enforced by scripts/org_audit.py: every required boundary must remain paired with its exact owner row, every file must exist, and every named symbol must resolve. Merely retaining the same unordered set of paths is not enough. A move therefore updates implementation and architecture in the same change instead of leaving a plausible but stale owner behind.


The pipeline: bytes → finding

The end-to-end flow, stage by stage, each pointing at the crate/module that owns it. The scan engine's own header doc (engine/mod.rs) is the authoritative, method-level version of steps 2-4.

  1. Acquire bytes: a source yields file-path + content chunks. crates/sources/src/ (filesystem/, git/, stdin.rs, docker/, s3/, gcs.rs, cloud/azure_blob.rs, github_org.rs, github_collaboration.rs, gitlab_group.rs, bitbucket_workspace.rs, hosted_git/, web/, har.rs, strings.rs, binary/).
  2. Phase 1: trigger production (which detectors could fire, and where). Swappable backend: scalar CPU literal/regex, SIMD Hyperscan (engine/backend_triggered.rs, engine/scan_coalesced.rs), or the GPU fused resident literal-evidence route (engine/gpu_region_dispatch.rs). It produces one "which detectors may match here" bitmap plus optional confirmed-anchor and generic-keyword positions per chunk. The fast prefilters (simdsieve, bigram_bloom, alphabet_filter, prefix_trie) live at crates/scanner/src/; detector-to-matcher construction lives in compiled_scanner/compile.rs, compiler.rs, and compiler/.
  3. Phase 2: extraction (the shared tail, identical for CPU and GPU): per-chunk confirmed → phase2 capture → generic → entropy → ML (engine/extract.rs, engine/phase2*.rs, engine/backend_triggered.rs, engine/scan.rs). Decode-through (base64/hex/url/unicode/json) runs here and recurses: decode/.
  4. Finish raw matches: scanner-owned suppression, confidence, and cross-chunk seam reassembly run in engine/scan_postprocess/, engine/process.rs, and engine/boundary.rs. Confidence + ML scoring live in confidence/, ml_scorer.rs, and ml_scorer/; context inference lives in context/. The per-match policy here (suppression gates · example/placeholder · checksum · confidence penalties) is governed by one invariant; see Match adjudication: one policy, one chokepoint below.
  5. Verify (optional and networked): for detectors with a [detector.verify] plan, the verifier sends a credential-derived request to the declared service behind SSRF, bogon, and rate-limit guards.
  6. Resolve and report: the CLI orchestrator applies scan-level policy and allowlists; core deduplication and reporters emit text/JSON/SARIF and support baseline comparison. crates/cli/src/orchestrator/postprocess.rs, crates/cli/src/orchestrator/reporting.rs, crates/core/src/dedup.rs, and crates/core/src/report/ own these steps.

keyhog_core::VerifiedFinding::from_deduped is the conversion boundary from a deduplicated match to a report-safe finding. It initializes the complete finding shape, including measured entropy and redacted companions, so verifier, skipped, and diff paths cannot silently drift when the report contract grows.

The accelerated batch path is two-phase and coalesced. A file with no phase-one hit stops only when the shared no-hit admission proof also rules out phase-two patterns, generic assignments, and enabled entropy analysis. This proof uses the active corpus's compiled generic-keyword stems and the owning detector's keyword_free_min_len plus effective Shannon floor; it does not substitute the embedded corpus or a scanner-wide run length for a focused custom corpus. Chunk size never disables an active detector path; overlapping source and scanner windows bound work instead. Portable CPU, Hyperscan, CUDA, and WGPU share this proof. Large filesystem scans may instead use the fused reader/scanner pipeline so I/O and scanning overlap; crates/cli/src/orchestrator/dispatch.rs and dispatch/fused.rs own that execution choice. Both paths feed the same scanner and report contracts. Backend choice must change performance only, never finding semantics.

Within the shared SIMD/GPU coalesced tail, detector, generic, and entropy candidates retain their per-chunk state while their precomputed ML feature rows are submitted as one CPU or GPU MoE batch. Final scores return to the originating chunk before its cap, decode postprocess, seam handling, and report adjudication run. Portable CPU-fallback, single-file, and oversized windowed scans keep the same scoring contract with smaller local batches.

Execution surfaces

The CLI owns process-level routing. The scanner crate exposes explicit backend execution; it does not read the autoroute cache or silently choose from local hardware. This keeps library calls deterministic and makes CLI routing inspectable.

WorkloadExecution surfaceRouting and ownership
One in-process scankeyhog scan ... --daemon=offFull orchestrator; persisted one-shot autoroute evidence or an explicit diagnostic --backend.
Large tree, multiple inputs, Git, cloud, container, binary, or live verificationIn-process orchestratorFused or coalesced batches; the daemon is not eligible even when it is running.
Repeated eligible stdin or single-file scans on Unixkeyhog daemon start, then keyhog scan ...Client checks request eligibility and peer identity; a calibrated daemon uses warm-runtime autoroute evidence, invalid startup state is labeled autoroute-recovery, and persisted quarantine is labeled autoroute-degraded. Every affected request reports scalar recovery.
Continuous local directory monitoringkeyhog watchForeground watcher with its own compiled scanner and warm-runtime autoroute policy; not the daemon and not reported by daemon status.

Persisted backend selection lives under crates/cli/src/orchestrator/dispatch/backend.rs and orchestrator/dispatch/backend/. Daemon transport and lifecycle live under crates/cli/src/daemon/. See the operator references for cache-miss, cold-versus-warm, and active-versus-inactive daemon behavior.

The routing package keeps measurement, proof, and persistence separate:

BoundaryOwner
Candidate measurement and cross-backend parity probesbackend/calibration.rs
One-shot and warm-daemon route decision policybackend/evidence.rs
Statistical trial evidence and confidence intervalsbackend/evidence/timing.rs
Secret-safe, complete finding identity used for paritybackend/evidence/match_identity.rs
Workload identity and bucketingbackend/workload.rs
Host and accelerator identitybackend/host.rs
Cache schema, exact artifact/build identity, bounded codec, validation, inspection, and locked persistencethe matching modules under backend/store/

This separation is deliberate: persisted bytes cannot define routing policy, inspection cannot bypass cache validation, and performance evidence cannot silently weaken detection parity.

Failure and recovery contract

KeyHog separates trust failures from recoverable execution failures:

  • Complete: the selected backend covered the input normally.
  • Complete after recovery: an automatically selected accelerator failed or autoroute evidence was invalid. KeyHog warned visibly and counted every recovered range, chunk, and byte. Runtime faults retain completed dispatches and replay only unprocessed ranges through a proven recovery peer; invalid selection state scans the batch through the scalar correctness oracle. The result is complete, but it is not a healthy autoroute claim.
  • Incomplete: some requested bytes or transformation could not be recovered. The scan may report findings from covered input, but it cannot report clean.
  • Fatal trust or explicit-contract failure: invalid policy, corrupt or unauthenticated artifacts, or an explicitly required backend cannot be substituted.

Recovery is an owned execution path, not a silent fallback. It must operate on the same stable source snapshot, preserve finding parity, merge results deterministically, identify every replayed interval, and remain absent during autoroute calibration so a backend that needs recovery cannot be certified fastest-correct.

Process and exit ownership

The library crates do not terminate the process. core, scanner, sources, and verifier return values or errors to their caller. The CLI owns the operator-visible exit:

  1. crates/cli/src/main.rs installs the Unix SIGINT handler before starting the runtime. SIGINT writes the interruption diagnostic and exits 130.
  2. crates/cli/src/lib.rs::cli_main dispatches subcommands. Successful subcommands return std::process::ExitCode; setup and execution errors pass through cli_error_exit_code.
  3. crates/cli/src/orchestrator/run.rs::resolve_scan_exit owns completed scan precedence: scanner panic, live credentials, findings, incremental-cache failure, incomplete source coverage, then clean success. Autoroute calibration has its explicit success path.
  4. A scanner-thread panic sets the shared panic marker. The CLI flushes the diagnostic streams and exits 11 immediately instead of allowing a later accelerator teardown to replace the documented code.

Normal automatic autoroute recovery is part of a completed scan, so it keeps the ordinary finding or clean code. An explicit backend contract that cannot be honored is an error before completed-scan precedence applies. See Exit codes for every number and shell examples.

Finding identity and dedup

There is one identity contract with stage-specific keys, not interchangeable "same finding" guesses:

StageOwnerKeyWhy
Window overlap and raw collectorcrates/scanner/src/engine/windowed_support.rs::record_window_match; crates/scanner/src/scan_state.rs::ScanState::into_matches(detector_id, credential, source_offset)Adjacent 1 MiB windows overlap by 128 KiB, and more than one backend signal can surface the same span. The source-offset key removes duplicate raw hits without merging separate occurrences on different lines.
Raw-match correlation helpercrates/core/src/finding.rs::RawMatch::deduplication_key(detector_id, credential)Tests and internal correlation can ask whether two raw matches carry the same detector/value before a report scope is applied. It is not a report key because it intentionally excludes location.
User-selected report scopecrates/core/src/dedup.rs::dedup_matchesDedupScope::Credential: (detector_id, credential); DedupScope::File: (detector_id, credential, source + file_path + commit); DedupScope::None: no groupingThis is the operator-visible grouping. The primary location is the lowest source offset; additional locations use (source, file_path, line, commit) so structured/decode aliases on the same source line collapse.
Cross-detector report collapsecrates/core/src/dedup.rs::dedup_cross_detector(credential_hash, primary_file_path) after dedup_matchesOne secret value can match several detectors. This keeps one reported finding, chooses the best detector deterministically, and records alternate detector evidence as companions while preserving file-scoped reports.
Reporter-local location cleanupcrates/core/src/report/sarif.rs(file_path, line, offset) within one reported findingOutput adapters may remove repeated locations for format stability. They do not decide scan/report identity.

The required seam test is scan_windowed_overlap_dedups_end_to_end: a token placed wholly inside the 128 KiB overlap must scan as one raw match and one final reported finding.

Match adjudication: one policy, one chokepoint

Governing invariant. Whether a candidate match becomes a reported finding, and at what confidence, is a pure function of the value and its context, never of which emission path produced it. A value that is a ${} shell template, a name-name:v1 public identifier, or Config-Word-and-Word-only policy prose is not a secret no matter whether the entropy detector, the generic keyword bridge, the weak-anchor post-pass, or the hot-pattern fast path surfaced it. Phase-2 has several emission paths; they exist for speed and recall, not to each carry their own copy of policy.

Detector-local canonical and transport-decoded hexadecimal key-material rules follow the same boundary. Scanner construction compacts declared lengths, keywords, suffixes, and exclusions into detector-indexed programs. Named, generic, and entropy candidate paths execute those programs; only stable public compatibility helpers without a compiled scanner inspect DetectorSpec directly. Generic assignment processing resolves its entropy-policy owner and canonical-policy owner from one normalized key lookup.

The same construction step compiles hot scalar execution facts such as generic classification, minimum length and confidence, severity, structural password slots, exact detector keywords, and public-identifier assignment markers. Emission paths address that cache-local record by detector index. Once all matchers and policies are built, CompiledScanner drops DetectorSpec itself; the flexible structure remains a configuration and introspection schema, not a second runtime policy owner.

Every public scanner constructor reaches one full-corpus quality gate before it builds matchers or probes backends. This also applies when you construct DetectorSpec values in memory instead of loading TOML. The gate rejects invalid detector fields and duplicate IDs with detector-indexed configuration errors.

Each detector index addresses one compiled plan containing its interned primary and entropy-fallback metadata, execution facts, canonical/decoded key-material program, entropy floor and policy, ML policy, credential-shape gate, suppression policy, weak-anchor state, and compiled companions. Those policies remain separate modules by responsibility, but their runtime ownership and index alignment live in one structure rather than parallel vectors.

The same plan owner compiles detector decode_transforms declarations into one active-corpus reverse and Caesar admission program. The decoders do not read the scanner-global confidence prefix list. A custom corpus therefore changes both matching and evasion recovery through the same detector digest.

Scanner construction also snapshots the ordered decoder registry. Decode execution, decode admission, and autoroute workload sketches all read that same immutable snapshot. Each decoder supplies a stable name and version. Those descriptors contribute to the detector digest, so cached routing evidence does not survive a decoder-plan change. If you register a decoder after scanner construction, the existing scanner does not change. Compile another scanner to use the new decoder.

The rule. Emission paths produce CandidateMatch values and typed signals; adjudicate_match owns the ordered suppression verdict. Path owners may compute context-specific facts (entropy shape, generic bridge boundaries, named detector policy), but they do not invent an untyped final drop reason:

emission paths (entropy · generic/keyword bridge · weak-anchor · hot fast path · GPU)
        │  each yields CandidateMatch { detector, span, value }

adjudicate_match(CandidateMatch, MatchCtx)
   1. explicit/process signals
   2. generic/entropy/hot-pattern signals
   3. named-detector suppression
   4. final report-floor policy

   Verdict::Suppressed(stage_name)  |  Verdict::Reported(confidence)

MatchCtx carries one explicit signal family at a time. The Verdict names the deciding StageId, which is what dogfood telemetry records. Shared shape policy lives under suppression::shape; path-specific callers convert its result into the matching typed signal before adjudication.

Why this shape. Candidate discovery necessarily differs by detector family, but the final vocabulary and ordering of suppression decisions must not. Typed signals preserve the context each path needs while keeping one auditable verdict pipeline and one telemetry reason per decision.

The ML model (weights.bin)

The scanner serves a Mixture-of-Experts confidence model embedded at build time (crates/scanner/src/weights.bin, include_bytes!). It is trained out-of-band by the Python pipeline in ml/:

ml/harvest_corpus.py   real labelled candidates (CredData), harvested at a LOW
                       report floor so sub-floor hard negatives are captured

ml/train_classifier.py blend synthetic + real, file-grouped split (no leakage),
                       train the 55-feature detector-conditioned MoE, gate on
                       held-out F1 plus
                       aggregate plus recall-sensitive class/detector recall

ml/retrain_loop.sh     one command: harvest → train → (--write) ship weights.bin
                       → (--verify) rebuild + per-detector-FP bench gate,
                       fail-closed revert on any regression

Because the model is compile-time-embedded, a new model is only observable after a rebuild, which is why --verify rebuilds before benching. The adjacent crates/scanner/src/model_card.json carries the model hash, training inputs, and gate metrics; build.rs refuses a card/weights mismatch and embeds the summary shown by keyhog --version.

Scanner construction also compiles the detector-conditioned feature facts used by that model, including service identity, verifier and companion presence, generic/structural classification, phase-2 ownership, and entropy family. Inference indexes that compact immutable policy and does not reinterpret the loaded detector schema for each candidate. The public training oracle compiles the same facts from the supplied detector before extracting its feature row.

Detector-owned compiled validation

Offline validation is declared in each detector TOML's validators array. A declaration selects a typed shared primitive and supplies that secret type's prefixes, layout widths, bounds, and confidence floor. Scanner construction compiles those declarations into the same immutable detector plan as matching, entropy, suppression, companions, and ML policy.

Named matches dispatch directly to their detector plan. Generic and entropy candidates use a first-byte index compiled from the active corpus instead of walking a global validator registry. CRC32/base62 comparison is allocation-free; base64 validation reuses zeroed per-thread scratch storage. Boundary extension returns its validation decision with the final credential slice, and ML pending rows carry that decision to final reporting. No candidate is revalidated after model inference, and custom detector corpora never inherit an embedded service table.


Where do I find X?

I want to…Go to
Add/edit a detectordetectors/<name>.toml (data; see CONTRIBUTING.md for the schema)
Understand the scan flow at method levelcrates/scanner/src/engine/mod.rs header
Change how confidence is scoredcrates/scanner/src/confidence/, ml_scorer.rs
Add a suppression gate / change what counts as a non-secretthe one gate list public_noncredential_shape; see "Match adjudication" above (never inline a looks_like_* call in an emission path)
Retrain / improve the ML modelml/retrain_loop.sh (+ ml/README.md)
Change an entropy entry path or weak-anchor floorthe owning detector TOML (entropy_roles, entropy_floor, entropy_high)
Add or tune offline validationthe owning detector TOML validators declaration
Add or tune reverse or Caesar recoverythe owning detector TOML decode_transforms declaration
Add an input sourcecrates/sources/src/
Add live verification for a detector[detector.verify] in the TOML + crates/verifier/src/verify/
Change output formattingcrates/cli/src/format.rs, crates/cli/src/orchestrator/reporting.rs
Change process exit codes or precedencecrates/cli/src/exit_codes.rs, crates/cli/src/lib.rs::cli_error_exit_code, crates/cli/src/orchestrator/run.rs::resolve_scan_exit, and crates/cli/src/main.rs for Unix SIGINT
Add a benchmark / change the gatebenchmarks/bench/
Verify a perf or detection claimbenchmarks/ (the README numbers regenerate from here)