Rust Engine¶

O ThreatDeflect possui um hybrid engine em Rust compilado via Maturin/PyO3 que oferece performance nativa para as operacoes mais pesadas da analise.

O engine e publicado como crate independente no crates.io: threatdeflect-core

O que o Rust engine acelera¶

Usando como crate Rust (independente do Python)¶

Adicione ao seu Cargo.toml:

[dependencies]
threatdeflect-core = "0.1"

use threatdeflect_core::SecretAnalyzer;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let rules = vec![
        ("AWS Key".to_string(), r"AKIA[0-9A-Z]{16}".to_string()),
        ("GitHub Token".to_string(), r"ghp_[a-zA-Z0-9]{36}".to_string()),
    ];

    let suspicious = vec![
        ("Reverse Shell".to_string(), r"bash\s+-i\s+>&\s+/dev/tcp".to_string()),
    ];

    let analyzer = SecretAnalyzer::new(rules, suspicious)?;
    let result = analyzer.analyze_content(
        "aws_key = AKIAIOSFODNN7EXAMPLE1",
        "src/config.py",
        "config.py",
    );

    for finding in &result.findings {
        println!(
            "[{:.0}%] {} in {} ({})",
            finding.confidence * 100.0,
            finding.finding_type,
            finding.file,
            finding.file_context.as_str(),
        );
    }

    Ok(())
}

Documentacao completa da crate: docs.rs/threatdeflect-core

Fallback automatico¶

O Rust engine e opcional no contexto Python. Se o modulo threatdeflect_rs nao estiver compilado, o ThreatDeflect usa automaticamente a implementacao em Python puro. Todas as funcionalidades permanecem disponiveis, apenas com menor performance em analises grandes.

Compilando para Python¶

Requisitos¶

• Rust toolchain (instale via rustup.rs)
• Maturin (uv tool install maturin)
• Python 3.11+ em um ambiente virtual

Compilacao¶

source .venv/bin/activate
maturin develop --release

O comando maturin develop --release compila o modulo Rust e instala diretamente no ambiente virtual ativo.

Verificando a instalacao¶

python -c "import threatdeflect_rs; print('Rust engine available')"

Arquitetura¶

threatdeflect-core/          Crate pura (publicada no crates.io)
  src/
    lib.rs                   Re-exports publicos
    analyzer.rs              SecretAnalyzer: orquestra deteccao
    confidence.rs            Entropia Shannon, scoring, ajustes
    context.rs               Classificacao de arquivo, comentarios
    types.rs                 Finding, Ioc, AnalysisResult (com zeroize)
    error.rs                 AnalyzerError (thiserror)

rust_core/                   Wrapper PyO3 (~50 linhas)
  src/
    lib.rs                   Converte tipos Rust -> PyDict/PyList

A classe RustAnalyzer e exposta para Python via PyO3 e disponibiliza o metodo principal process_file_content() que recebe o conteudo de um arquivo e retorna todos os achados (segredos, comandos suspeitos, IOCs, etc.).

Seguranca de memoria¶

Dependencias Rust¶