cribra 0.2.0

Privacy-first Rust core for detecting, querying, and safely transforming secrets and sensitive data
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

//! Contextual validation for generic secret-bearing fields.
//!
//! This validator is intentionally conservative and is used only when a field
//! explicitly identifies a token, API key or secret value.

use super::{
    context::ValidationContext,
    utils::{DEFAULT_KEY_WINDOW, key_matches_any, nearest_key},
};
use crate::validators::utils::is_obvious_placeholder;

/// Generic credential field recognized by contextual validation.
#[derive(Debug, Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Hash)]
pub(crate) enum GenericCredentialKind {
    ApiKey,
    Token,
    Secret,
}

/// Successful generic credential contextual validation.
#[derive(Debug, Copy, Clone, Eq, PartialEq)]
pub(crate) struct GenericValidation {
    kind: GenericCredentialKind,
}

impl GenericValidation {
    pub(crate) const fn kind(self) -> GenericCredentialKind {
        self.kind
    }
}

/// Validates an opaque value assigned to an explicit generic credential field.
pub(crate) fn validate_generic_credential(
    context: &ValidationContext<'_>,
) -> Option<GenericValidation> {
    let candidate = context.candidate();

    if !(16..=2048).contains(&candidate.len())
        || candidate.chars().any(char::is_control)
        || is_obvious_placeholder(candidate)
    {
        return None;
    }

    let key = nearest_key(context.before_window(DEFAULT_KEY_WINDOW))?;

    let kind = if key_matches_any(key, &["api_key", "apikey", "api_token", "access_key"]) {
        GenericCredentialKind::ApiKey
    } else if key_matches_any(
        key,
        &["token", "access_token", "auth_token", "bearer_token"],
    ) {
        GenericCredentialKind::Token
    } else if key_matches_any(
        key,
        &["secret", "secret_key", "signing_secret", "webhook_secret"],
    ) {
        GenericCredentialKind::Secret
    } else {
        return None;
    };

    Some(GenericValidation { kind })
}

#[cfg(test)]
mod tests {
    use super::*;

    fn context<'a>(source: &'a str, value: &'a str) -> ValidationContext<'a> {
        let start = source.find(value).expect("fixture must contain value");
        ValidationContext::new(source, start..start + value.len())
    }

    #[test]
    fn recognizes_explicit_api_key_field() {
        let value = "AbCdEfGhIjKlMnOpQrStUvWx";
        let source = format!("API_KEY={value}");

        assert_eq!(
            validate_generic_credential(&context(&source, value)).map(GenericValidation::kind),
            Some(GenericCredentialKind::ApiKey),
        );
    }

    #[test]
    fn rejects_unrelated_field_and_placeholder() {
        let value = "AbCdEfGhIjKlMnOpQrStUvWx";
        let source = format!("display_name={value}");
        assert!(validate_generic_credential(&context(&source, value)).is_none());

        let source = "API_KEY=your_api_key_here";
        assert!(validate_generic_credential(&context(source, "your_api_key_here")).is_none());
    }
}