cipherrun 0.3.0

A fast, modular, and scalable TLS/SSL security scanner written in Rust
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

// Rating Record Model
// Represents SSL Labs-style rating components for a scan

use serde::{Deserialize, Serialize};
use sqlx::FromRow;

/// Rating record in database
#[derive(Debug, Clone, Serialize, Deserialize, FromRow)]
pub struct RatingRecord {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub rating_id: Option<i64>,
    pub scan_id: i64,
    pub category: String, // "protocol", "key_exchange", "cipher", "certificate"
    pub score: i32,       // 0-100
    pub grade: Option<String>, // "A+", "A", "B", etc.
    pub rationale: Option<String>,
}

impl RatingRecord {
    /// Create new rating record
    pub fn new(scan_id: i64, category: String, score: u8) -> Self {
        Self {
            rating_id: None,
            scan_id,
            category,
            score: score as i32,
            grade: None,
            rationale: None,
        }
    }

    /// Set grade
    pub fn with_grade(mut self, grade: String) -> Self {
        self.grade = Some(grade);
        self
    }

    /// Set rationale
    pub fn with_rationale(mut self, rationale: String) -> Self {
        self.rationale = Some(rationale);
        self
    }
}

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

    #[test]
    fn test_rating_record_creation() {
        let rating = RatingRecord::new(1, "certificate".to_string(), 90)
            .with_grade("A".to_string())
            .with_rationale("Strong certificate".to_string());

        assert_eq!(rating.scan_id, 1);
        assert_eq!(rating.category, "certificate");
        assert_eq!(rating.score, 90);
        assert_eq!(rating.grade, Some("A".to_string()));
    }
}