keyhog-core 0.5.40

keyhog-core: shared data model and detector specifications for the KeyHog secret scanner
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132

//! Tabular CSV findings reporter.

use std::io::Write;

use crate::VerifiedFinding;

use super::{ReportError, Reporter, WriterBackedReporter};

/// Tabular CSV output.
pub struct CsvReporter<W: Write + Send> {
    writer: W,
}

impl<W: Write + Send> CsvReporter<W> {
    /// Create a new CSV reporter and write headers.
    pub fn new(mut writer: W) -> Result<Self, ReportError> {
        writeln!(
            writer,
            "detector_id,detector_name,service,severity,credential_redacted,credential_hash,source,file_path,line,offset,commit,author,date,verification,confidence"
        )?;
        Ok(Self { writer })
    }
}

impl<W: Write + Send> Reporter for CsvReporter<W> {
    fn report(&mut self, finding: &VerifiedFinding) -> Result<(), ReportError> {
        let line_str = finding
            .location
            .line
            .map(|l| l.to_string())
            .unwrap_or_default();
        let commit_str = finding
            .location
            .commit
            .as_ref()
            .map(|c| c.as_ref())
            .unwrap_or_default();
        let author_str = finding
            .location
            .author
            .as_ref()
            .map(|a| a.as_ref())
            .unwrap_or_default();
        let date_str = finding
            .location
            .date
            .as_ref()
            .map(|d| d.as_ref())
            .unwrap_or_default();
        let file_path_str = finding
            .location
            .file_path
            .as_ref()
            .map(|f| f.as_ref())
            .unwrap_or_default();
        let confidence_str = finding
            .confidence
            .map(|c| c.to_string())
            .unwrap_or_default();

        let verification_str = match &finding.verification {
            crate::VerificationResult::Live => "live".to_string(),
            crate::VerificationResult::Revoked => "revoked".to_string(),
            crate::VerificationResult::Dead => "dead".to_string(),
            crate::VerificationResult::RateLimited => "rate_limited".to_string(),
            crate::VerificationResult::Error(err) => format!("error: {err}"),
            crate::VerificationResult::Unverifiable => "unverifiable".to_string(),
            crate::VerificationResult::Skipped => "skipped".to_string(),
        };

        writeln!(
            self.writer,
            "{},{},{},{},{},{},{},{},{},{},{},{},{},{},{}",
            escape_csv(&finding.detector_id),
            escape_csv(&finding.detector_name),
            escape_csv(&finding.service),
            escape_csv(&finding.severity.to_string()),
            escape_csv(&finding.credential_redacted),
            escape_csv(&crate::hex_encode(&finding.credential_hash)),
            escape_csv(&finding.location.source),
            escape_csv(file_path_str),
            escape_csv(&line_str),
            escape_csv(&finding.location.offset.to_string()),
            escape_csv(commit_str),
            escape_csv(author_str),
            escape_csv(date_str),
            escape_csv(&verification_str),
            escape_csv(&confidence_str)
        )?;
        Ok(())
    }

    fn finish(&mut self) -> Result<(), ReportError> {
        self.flush_writer()
    }
}

impl<W: Write + Send> WriterBackedReporter for CsvReporter<W> {
    type Writer = W;

    fn writer_mut(&mut self) -> &mut Self::Writer {
        &mut self.writer
    }
}

fn escape_csv(val: &str) -> String {
    // Neutralize spreadsheet formula injection (OWASP CSV-injection guidance):
    // a cell whose first character is `=`, `+`, `-`, `@`, or a leading tab/CR is
    // evaluated as a formula by Excel/LibreOffice/Sheets after CSV unquoting.
    // Prefix such attacker-controlled values with a single quote so the cell is
    // rendered as literal text, then apply the normal RFC-4180 quoting below.
    let neutralized = match val.as_bytes().first() {
        Some(b'=' | b'+' | b'-' | b'@' | b'\t' | b'\r') => {
            let mut guarded = String::with_capacity(val.len() + 1);
            guarded.push('\'');
            guarded.push_str(val);
            guarded
        }
        _ => val.to_string(),
    };

    if neutralized.contains(',')
        || neutralized.contains('"')
        || neutralized.contains('\n')
        || neutralized.contains('\r')
    {
        let escaped = neutralized.replace('"', "\"\"");
        format!("\"{}\"", escaped)
    } else {
        neutralized
    }
}